Add peer to listen callback

1 files changed, 2 insertions, 2 deletions

diff --git a/src/BootstrapConnection.cpp b/src/BootstrapConnection.cpp
index 9a19943..421dfe1 100644
--- a/src/BootstrapConnection.cpp
+++ b/src/BootstrapConnection.cpp
@@ -85,7 +85,7 @@ namespace sibs
     void BootstrapConnection::receiveDataFromPeer(BoostrapConnectionListenCallbackFunc listenCallbackFunc, std::shared_ptr<DirectConnectionPeer> peer, const void *data, const usize size)
     {
         if(listenCallbackFunc)
-            listenCallbackFunc(data, size);
+            listenCallbackFunc(peer.get(), data, size);
     }
     
     void BootstrapConnection::listen(const PubsubKey &pubsubKey, BoostrapConnectionListenCallbackFunc callbackFunc)
@@ -109,7 +109,7 @@ namespace sibs
             std::lock_guard<std::mutex> lock(listenerCallbackFuncMutex);
             auto listenCallbackFuncIt = listenCallbackFuncs.find(pubsubKey);
             if(listenCallbackFuncIt != listenCallbackFuncs.end() && listenCallbackFuncIt->second)
-                listenCallbackFuncIt->second(data->data(), data->size());
+                listenCallbackFuncIt->second(nullptr, data->data(), data->size());
         }
         
         std::lock_guard<std::mutex> lock(subscribedPeersMutex);