Add functions to send/receive custom messages

1 files changed, 7 insertions, 2 deletions

diff --git a/src/DhtKey.cpp b/src/DhtKey.cpp
index 7bd6acf..422b715 100644
--- a/src/DhtKey.cpp
+++ b/src/DhtKey.cpp
@@ -8,13 +8,18 @@ namespace odhtdb
         firstByteOriginalValue = infoHash[0];
     }
     
-    const dht::InfoHash& DhtKey::getNewDataListenerKey()
+    DhtKey::DhtKey(const dht::InfoHash &_infoHash) : infoHash(_infoHash)
+    {
+        firstByteOriginalValue = infoHash[0];
+    }
+    
+    dht::InfoHash DhtKey::getNewDataListenerKey()
     {
         infoHash[0] = firstByteOriginalValue;
         return infoHash;
     }
     
-    const dht::InfoHash& DhtKey::getRequestOldDataKey()
+    dht::InfoHash DhtKey::getRequestOldDataKey()
     {
         infoHash[0] = firstByteOriginalValue + 1;
         return infoHash;