Преглед изворни кода

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

Tatsuhiro Tsujikawa пре 14 година
родитељ
комит
b775b19366
1 измењених фајлова са 6 додато и 2 уклоњено
  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()