Remove peer socket when disconnected

1 files changed, 2 insertions, 1 deletions

diff --git a/include/sibs/DirectConnection.hpp b/include/sibs/DirectConnection.hpp
index 0e8b961..45ba5d9 100644
--- a/include/sibs/DirectConnection.hpp
+++ b/include/sibs/DirectConnection.hpp
@@ -57,7 +57,8 @@ namespace sibs
     private:
         std::shared_ptr<DirectConnectionPeer> connect(const Ipv4 &address, bool rendezvous, bool reuseAddr, PubSubReceiveDataCallback receiveDataCallbackFunc);
         void receiveData();
-        bool receiveDataFromPeer(const int socket, char *output);
+        int receiveDataFromPeer(const int socket, char *output);
+        bool removePeer(int peerSocket);
     private:
         u16 port;
         int eid;