Merge pull request #121 from kaistshadow/issue/120

issue #120 관련. yg-amazon-exp branch 정리를 위한 cherry-picking
kaistshadow · Jan 9, 2021 · 9ae9712 · 9ae9712
2 parents e05f597 + c098a24
commit 9ae9712
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/BLEEPlib/src/mainmodules/EstimateRTTModule.cpp b/BLEEPlib/src/mainmodules/EstimateRTTModule.cpp
@@ -607,6 +607,10 @@ bool EstimateRTTModule::MulticastMessage(std::shared_ptr<Message> message) {
     int sentOut = 0;
     for (int i : idxs) {
         if (message->GetSource().GetId() != dests[i].GetId() && sentOut < fanOut) {
+            // check whether there exists a data socket for the destination peer
+            if (!peerManager.HasEstablishedDataSocket(dests[i]))
+                continue;
+
             // get datasocket
             int socketFD = peerManager.GetConnectedSocketFD(dests[i]);
             std::shared_ptr<RTTModule_DataSocket> dataSocket = socketManager.GetDataSocket(socketFD);
@@ -667,6 +671,10 @@ bool EstimateRTTModule::ForwardMessage(std::shared_ptr<RTTModule_MessageHeader>
     int sentOut = 0;
     for (int i : idxs) {
         if (message->GetSource().GetId() != dests[i].GetId() && sentOut < fanOut){
+            // check whether there exists a data socket for the destination peer
+            if (!peerManager.HasEstablishedDataSocket(dests[i]))
+                continue;
+
             // get datasocket
             int socketFD = peerManager.GetConnectedSocketFD(dests[i]);
             std::shared_ptr<RTTModule_DataSocket> dataSocket = socketManager.GetDataSocket(socketFD);