Starting...

1 files changed, 112 insertions, 0 deletions

diff --git a/src/DirectConnection.cpp b/src/DirectConnection.cpp
new file mode 100644
index 0000000..12ae761
--- /dev/null
+++ b/src/DirectConnection.cpp
@@ -0,0 +1,112 @@
+#include "../include/DirectConnection.hpp"
+#include <cstdio>
+#include <cstring>
+
+#ifndef WIN32
+   #include <arpa/inet.h>
+   #include <netdb.h>
+#else
+   #include <winsock2.h>
+   #include <ws2tcpip.h>
+#endif
+#include <udt/udt.h>
+
+namespace sibs
+{
+    Ipv4::Ipv4(const char *ip, u16 port)
+    {        
+        struct addrinfo hints = {};
+        hints.ai_flags = AI_PASSIVE;
+        hints.ai_family = AF_INET;
+        hints.ai_socktype = SOCK_STREAM;
+        
+        char portStr[6];
+        sprintf(portStr, "%d", port);
+        int addrInfoResult = getaddrinfo(ip, portStr, &hints, &address);
+        if(addrInfoResult != 0)
+        {
+            std::string errMsg = "Ip ";
+            errMsg += ip;
+            errMsg += " is not a valid ip";
+            throw InvalidAddressException(errMsg);
+        }
+    }
+    
+    Ipv4::~Ipv4()
+    {
+        freeaddrinfo(address);
+    }
+    
+    DirectConnections::DirectConnections() : 
+        mySocket(0)
+    {
+        try
+        {
+            init();
+        }
+        catch(...)
+        {
+            cleanup();
+        }
+    }
+    
+    DirectConnections::~DirectConnections()
+    {
+        cleanup();
+    }
+    
+    void DirectConnections::init()
+    {
+        UDT::startup();
+        eid = UDT::epoll_create();
+        mySocket = UDT::socket(AI_PASSIVE, AF_INET, SOCK_STREAM);
+        bool rendezvous = true;
+        UDT::setsockopt(mySocket, 0, UDT_RENDEZVOUS, &rendezvous, sizeof(bool));
+        bool reuseAddr = true;
+        UDT::setsockopt(mySocket, 0, UDT_REUSEADDR, &reuseAddr, sizeof(bool));
+        
+        // Windows UDP issue
+        // For better performance, modify HKLM\System\CurrentControlSet\Services\Afd\Parameters\FastSendDatagramThreshold
+        #ifdef WIN32
+        int mss = 1052;
+        UDT::setsockopt(socket, 0, UDT_MSS, &mss, sizeof(mss));
+        #endif
+        
+        sockaddr_in myAddr = {};
+        myAddr.sin_family = AF_INET;
+        myAddr.sin_port = htons(9000);
+        myAddr.sin_addr.s_addr = INADDR_ANY;
+        memset(&myAddr.sin_zero, '\0', 8);
+
+        if(UDT::bind(mySocket, (sockaddr*)&myAddr, sizeof(myAddr)) == UDT::ERROR)
+        {
+            // TODO: Add ip and port to error
+            std::string errMsg = "UDT: Failed to bind, error: ";
+            errMsg += UDT::getlasterror().getErrorMessage();
+            throw ConnectionException(errMsg);
+        }
+    }
+    
+    void DirectConnections::cleanup()
+    {
+        UDT::epoll_release(eid);
+        
+        if(mySocket != 0)
+            UDT::close(mySocket);
+        
+        UDT::cleanup();
+    }
+    
+    void DirectConnections::connect(const Ipv4 &address)
+    {
+        if(UDT::connect(mySocket, address.address->ai_addr, address.address->ai_addrlen) == UDT::ERROR)
+        {
+            // TODO: Add ip and port to error
+            std::string errMsg = "UDT: Failed to connect, error: ";
+            errMsg += UDT::getlasterror().getErrorMessage();
+            throw ConnectionException(errMsg);
+        }
+        
+        //UDT::epoll_add_usock(eid, 2);
+    } 
+}