Explorar el Código

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

Tatsuhiro Tsujikawa hace 14 años
padre
commit
b775b19366
Se han modificado 1 ficheros con 6 adiciones y 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()