From c2187ca6b61c701c281cc528db43f6b97c50f3d8 Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Thu, 7 Jun 2018 22:00:42 +0200
Subject: Add bootstrap node, listen method

---
 include/sibs/DirectConnection.hpp | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

(limited to 'include/sibs/DirectConnection.hpp')

diff --git a/include/sibs/DirectConnection.hpp b/include/sibs/DirectConnection.hpp
index c4431a5..0e8b961 100644
--- a/include/sibs/DirectConnection.hpp
+++ b/include/sibs/DirectConnection.hpp
@@ -7,6 +7,7 @@
 #include <memory>
 #include <thread>
 #include <mutex>
+#include <vector>
 #include "IpAddress.hpp"
 #include "../types.hpp"
 #include "../utils.hpp"
@@ -19,25 +20,22 @@ 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) {}
-    };
-    
-    
-    enum class PubSubConnectResult
+    enum class PubSubResult
     {
         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;
+    
+    using PubSubConnectCallback = std::function<void(PubSubResult result, const std::string &resultStr)>;
+    using PubSubReceiveDataCallback = std::function<void(std::shared_ptr<DirectConnectionPeer> peer, const void *data, const usize size)>;
+    using PubSubSendDataCallback = std::function<void(PubSubResult result, const std::string &resultStr)>;
     
     struct DirectConnectionPeer
     {
         int socket;
+        Ipv4 address;
         PubSubReceiveDataCallback receiveDataCallbackFunc;
     };
     
@@ -49,13 +47,15 @@ namespace sibs
         DirectConnections(u16 port = 27137);
         ~DirectConnections();
         
+        // Throws ConnectionException on error
+        std::shared_ptr<DirectConnectionPeer> connectServer(const Ipv4 &address, PubSubReceiveDataCallback receiveDataCallbackFunc);
         // 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);
+        void send(const std::shared_ptr<DirectConnectionPeer> &peer, std::shared_ptr<std::vector<u8>> data, PubSubSendDataCallback sendDataCallbackFunc = nullptr);
     protected:
         int createSocket(const Ipv4 &addressToBind, bool rendezvous, bool reuseAddr);
     private:
+        std::shared_ptr<DirectConnectionPeer> connect(const Ipv4 &address, bool rendezvous, bool reuseAddr, PubSubReceiveDataCallback receiveDataCallbackFunc);
         void receiveData();
         bool receiveDataFromPeer(const int socket, char *output);
     private:
-- 
cgit v1.2.3