1 files changed, 3 insertions, 2 deletions
diff --git a/include/sibs/BootstrapNode.hpp b/include/sibs/BootstrapNode.hpp
index ab3d6b3..c824a84 100644
--- a/include/sibs/BootstrapNode.hpp
+++ b/include/sibs/BootstrapNode.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "Socket.hpp"
 #include "DirectConnection.hpp"
 #include "IpAddress.hpp"
 #include "PubsubKey.hpp"
@@ -23,10 +24,10 @@ namespace sibs
         ~BootstrapNode();
     private:
         void acceptConnections();
-        void peerSubscribe(std::shared_ptr<DirectConnectionPeer> peer, const void *data, const usize size);
+        void messageFromClient(std::shared_ptr<DirectConnectionPeer> peer, MessageType messageType, const void *data, const usize size);
     private:
         DirectConnections connections;
-        int socket;
+        std::unique_ptr<Socket> socket;
         std::thread acceptConnectionsThread;
         PubsubKeyMap<std::vector<std::shared_ptr<DirectConnectionPeer>>> subscribedPeers;
         std::mutex subscribedPeersMutex;