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

pull/2/head
Tatsuhiro Tsujikawa 2011-09-11 00:12:18 +09:00
parent c408dbba72
commit b775b19366
1 changed files with 6 additions and 2 deletions

View File

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