From d2dde382e6423afd88217128dbc66dde74415dca Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Sat, 2 Jun 2018 01:39:25 +0200
Subject: Receive data with epoll_wait

---
 include/DirectConnection.hpp | 44 +++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 39 insertions(+), 5 deletions(-)

(limited to 'include/DirectConnection.hpp')

diff --git a/include/DirectConnection.hpp b/include/DirectConnection.hpp
index c11871f..4075d12 100644
--- a/include/DirectConnection.hpp
+++ b/include/DirectConnection.hpp
@@ -1,6 +1,12 @@
 #pragma once
 
 #include <stdexcept>
+#include <unordered_map>
+#include <functional>
+#include <string>
+#include <memory>
+#include <thread>
+#include <mutex>
 #include "types.hpp"
 #include "utils.hpp"
 
@@ -20,6 +26,12 @@ namespace sibs
         ConnectionException(const std::string &errMsg) : std::runtime_error(errMsg) {}
     };
     
+    class SendException : public std::runtime_error
+    {
+    public:
+        SendException(const std::string &errMsg) : std::runtime_error(errMsg) {}
+    };
+    
     class Ipv4
     {
         DISABLE_COPY(Ipv4)
@@ -31,19 +43,41 @@ namespace sibs
         struct addrinfo *address;
     };
     
+    enum class PubSubConnectResult
+    {
+        OK,
+        ERROR
+    };
+    
+    using PubSubConnectCallback = std::function<void(PubSubConnectResult result, const std::string &resultStr)>;
+    using PubSubReceiveDataCallback = std::function<void(const void *data, const usize size)>;
+    
+    struct DirectConnectionPeer
+    {
+        int socket;
+        PubSubReceiveDataCallback receiveDataCallbackFunc;
+    };
+    
     class DirectConnections
     {
         DISABLE_COPY(DirectConnections)
     public:
-        DirectConnections();
+        DirectConnections(u16 port = 27137);
         ~DirectConnections();
         
-        void connect(const Ipv4 &address);
+        // Throws ConnectionException on error
+        std::shared_ptr<DirectConnectionPeer> connect(const Ipv4 &address, PubSubReceiveDataCallback receiveDataCallbackFunc);
+        // Throws SendException on error
+        void send(const std::shared_ptr<DirectConnectionPeer> &peer, const void *data, const usize size);
     private:
-        void init();
-        void cleanup();
+        void receiveData();
+        bool receiveDataFromPeer(const int socket, char *output);
     private:
+        u16 port;
         int eid;
-        int mySocket;
+        std::unordered_map<int, std::shared_ptr<DirectConnectionPeer>> peers;
+        std::thread receiveDataThread;
+        std::mutex peersMutex;
+        bool alive;
     };
 }
-- 
cgit v1.2.3