From 93e5dbed325f29c1bf867a2847d3b839d545cfb2 Mon Sep 17 00:00:00 2001 From: Tatsuhiro Tsujikawa Date: Fri, 12 Nov 2010 12:48:48 +0000 Subject: 2010-11-12 Tatsuhiro Tsujikawa Removed SharedHandle::isNull(). Instead we added operator* and operator unspecified_bool_type. Removed use of WeakHandle and replaced with raw pointer. --- src/DHTMessageFactoryImpl.cc | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) (limited to 'src/DHTMessageFactoryImpl.cc') diff --git a/src/DHTMessageFactoryImpl.cc b/src/DHTMessageFactoryImpl.cc index 9910abe8..b3294800 100644 --- a/src/DHTMessageFactoryImpl.cc +++ b/src/DHTMessageFactoryImpl.cc @@ -66,6 +66,11 @@ namespace aria2 { DHTMessageFactoryImpl::DHTMessageFactoryImpl(int family): family_(family), + connection_(0), + dispatcher_(0), + routingTable_(0), + peerAnnounceStorage_(0), + tokenTracker_(0), logger_(LogFactory::getInstance()) {} DHTMessageFactoryImpl::~DHTMessageFactoryImpl() {} @@ -75,7 +80,7 @@ DHTMessageFactoryImpl::getRemoteNode (const unsigned char* id, const std::string& ipaddr, uint16_t port) const { SharedHandle node = routingTable_->getNode(id, ipaddr, port); - if(node.isNull()) { + if(!node) { node.reset(new DHTNode(id)); node->setIPAddress(ipaddr); node->setPort(port); @@ -309,8 +314,7 @@ void DHTMessageFactoryImpl::setCommonProperty m->setConnection(connection_); m->setMessageDispatcher(dispatcher_); m->setRoutingTable(routingTable_); - WeakHandle factory(this); - m->setMessageFactory(factory); + m->setMessageFactory(this); m->setVersion(getDefaultVersion()); } @@ -506,32 +510,29 @@ DHTMessageFactoryImpl::createUnknownMessage return m; } -void DHTMessageFactoryImpl::setRoutingTable -(const WeakHandle& routingTable) +void DHTMessageFactoryImpl::setRoutingTable(DHTRoutingTable* routingTable) { routingTable_ = routingTable; } -void DHTMessageFactoryImpl::setConnection -(const WeakHandle& connection) +void DHTMessageFactoryImpl::setConnection(DHTConnection* connection) { connection_ = connection; } void DHTMessageFactoryImpl::setMessageDispatcher -(const WeakHandle& dispatcher) +(DHTMessageDispatcher* dispatcher) { dispatcher_ = dispatcher; } void DHTMessageFactoryImpl::setPeerAnnounceStorage -(const WeakHandle& storage) +(DHTPeerAnnounceStorage* storage) { peerAnnounceStorage_ = storage; } -void DHTMessageFactoryImpl::setTokenTracker -(const WeakHandle& tokenTracker) +void DHTMessageFactoryImpl::setTokenTracker(DHTTokenTracker* tokenTracker) { tokenTracker_ = tokenTracker; } -- cgit