Browse Source

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

Tatsuhiro Tsujikawa 14 năm trước cách đây
mục cha
commit
b775b19366
1 tập tin đã thay đổi với 6 bổ sung2 xóa
  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()