Explorar o código

Check socket is null or not before calling (add|delete)SocketForReadCheck.

Tatsuhiro Tsujikawa %!s(int64=14) %!d(string=hai) anos
pai
achega
b775b19366
Modificáronse 1 ficheiros con 6 adicións e 2 borrados
  1. 6 2
      src/DHTInteractionCommand.cc

+ 6 - 2
src/DHTInteractionCommand.cc

@@ -63,12 +63,16 @@ DHTInteractionCommand::~DHTInteractionCommand()
 void DHTInteractionCommand::setReadCheckSocket(const SocketHandle& socket)
 {
   readCheckSocket_ = socket;
-  e_->addSocketForReadCheck(socket, this);
+  if(socket) {
+    e_->addSocketForReadCheck(socket, this);
+  }
 }
 
 void DHTInteractionCommand::disableReadCheckSocket(const SocketHandle& socket)
 {
-  e_->deleteSocketForReadCheck(socket, this);
+  if(socket) {
+    e_->deleteSocketForReadCheck(socket, this);
+  }
 }
 
 bool DHTInteractionCommand::execute()