From 3e2c7e59672b282f8e089679ae7f05cfc03e405b Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Sun, 14 Oct 2018 04:17:54 +0200
Subject: Add peer to listen callback

---
 src/BootstrapConnection.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/BootstrapConnection.cpp')

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);
-- 
cgit v1.2.3