+network

1 files changed, 189 insertions, 0 deletions

diff --git a/src/libjin/Net/Socket.cpp b/src/libjin/Net/Socket.cpp
new file mode 100644
index 0000000..b7c621e
--- /dev/null
+++ b/src/libjin/Net/Socket.cpp
@@ -0,0 +1,189 @@
+#include "Socket.h"
+
+namespace jin
+{
+namespace net
+{
+
+    Socket::Socket(SocketInformation info)
+        : tcpHandle(nullptr)
+        , udpHandle(nullptr)
+    {
+        type = info.type;
+        if (type == SocketType::TCP)
+        {
+            tk_IPaddress ip;
+            ip.host = info.address;
+            ip.port = info.port;
+            tcpHandle = tk_tcp_open(ip);
+        }
+        else if (type == SocketType::UDP)
+        {
+            udpHandle = tk_udp_open(info.port);
+        }
+    }
+
+    Socket::Socket(SocketType type, const char* address, unsigned short port)
+    {
+        this->type = type;
+        if (type == SocketType::TCP)
+        {
+            tk_IPaddress ip;
+        #if JIN_NET_TEKCOS
+            ip.host = tk_strtohl(address);
+            ip.port = port;
+            tcpHandle = tk_tcp_open(ip);
+        #endif
+        }
+        else if (type == SocketType::UDP)
+        {
+            udpHandle = tk_udp_open(port);
+        }
+    }
+
+    Socket::Socket(SocketType type, unsigned int address, unsigned short port)
+    {
+        this->type = type;
+        if (type == SocketType::TCP)
+        {
+            tk_IPaddress ip;
+            ip.host = address;
+            ip.port = port;
+            tcpHandle = tk_tcp_open(ip);
+        }
+        else if (type == SocketType::UDP)
+        {
+            udpHandle = tk_udp_open(port);
+        }
+    }
+
+    Socket::Socket(SocketType type, unsigned short port)
+    {
+        this->type = type;
+        if (type == SocketType::TCP)
+        {
+            tk_IPaddress ip;
+            ip.host = 0;
+            ip.port = port;
+            tcpHandle = tk_tcp_open(ip);
+        }
+        else if (type == SocketType::UDP)
+        {
+            udpHandle = tk_udp_open(port);
+        }
+    }
+
+#if JIN_NET_TEKCOS
+
+    Socket::Socket(tk_TCPsocket* handle)
+        : tcpHandle(handle)
+        , udpHandle(nullptr)
+    {
+    }
+
+    Socket::Socket(tk_UDPsocket* handle)
+        : tcpHandle(nullptr)
+        , udpHandle(handle)
+    {
+    }
+
+#endif // JIN_NET_TEKCOS
+
+    Socket::~Socket()
+    {
+    }
+
+    void Socket::configureBlocking(bool blocking)
+    {
+        if (type != SocketType::TCP)
+            return;
+    #if JIN_NET_TEKCOS
+        if (blocking)
+            tk_tcp_blocking(tcpHandle);
+        else
+            tk_tcp_nonblocking(tcpHandle);
+    #endif
+    }
+
+    Socket* Socket::accept()
+    {
+        if (type != SocketType::TCP)
+            return nullptr;
+        Socket* client;
+    #if JIN_NET_TEKCOS
+        tk_TCPsocket* socket = tk_tcp_accept(tcpHandle);
+        client = new Socket(socket);
+    #endif
+        return client;
+    }
+
+    int Socket::receive(char* buffer, int size)
+    {
+        if (type != SocketType::TCP)
+            return 0;
+    #if JIN_NET_TEKCOS
+        int len;
+        tk_tcp_recv(tcpHandle, buffer, size, &len);
+        return len;
+    #endif
+    }
+    
+    int Socket::send(char* buffer, int size)
+    {
+        if (type != SocketType::TCP)
+            return 0;
+    #if JIN_NET_TEKCOS
+        int len;
+        tk_tcp_send(tcpHandle, buffer, size, &len);
+        return len;
+    #endif
+    }
+
+    void Socket::sendTo(char* buffer, int size, unsigned int address, unsigned int port)
+    {
+        if (type != SocketType::UDP)
+            return;
+    #if JIN_NET_TEKCOS
+        tk_UDPpack pack;
+        pack.data = buffer;
+        pack.len = size; 
+        pack.ip.host = address;
+        pack.ip.port = port;
+        tk_udp_sendto(udpHandle, &pack);
+    #endif
+    }
+    
+    int Socket::receiveFrom(char* buffer, int size, unsigned int address, unsigned int port)
+    {
+        if (type != SocketType::UDP)
+            return 0;
+        int len;
+    #if JIN_NET_TEKCOS
+        tk_UDPpack pack;
+        pack.data = buffer;
+        pack.len = size;
+        pack.ip.host = address;
+        pack.ip.port = port;
+        tk_udp_recvfrom(udpHandle, &pack);
+        return pack.len;
+    #endif
+    }
+
+    void Socket::close()
+    {
+        if (type == SocketType::TCP)
+        {
+        #if JIN_NET_TEKCOS
+            tk_tcp_close(tcpHandle);
+        #endif
+        }
+        else if (type == SocketType::UDP)
+        {
+        #if JIN_NET_TEKCOS
+            tk_udp_close(udpHandle);
+        #endif
+        }
+    }
+
+}
+}
\ No newline at end of file