2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>

Renamed member variables
	* src/DefaultBtMessageFactory.cc
	* src/DefaultBtMessageFactory.h
pull/1/head
Tatsuhiro Tsujikawa 2010-06-12 08:23:49 +00:00
parent ee08678566
commit 4e071bfb62
3 changed files with 68 additions and 41 deletions

View File

@ -1,3 +1,9 @@
2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Renamed member variables
* src/DefaultBtMessageFactory.cc
* src/DefaultBtMessageFactory.h
2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net> 2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Renamed member variables Renamed member variables

View File

@ -70,7 +70,7 @@
namespace aria2 { namespace aria2 {
DefaultBtMessageFactory::DefaultBtMessageFactory():cuid(0), DefaultBtMessageFactory::DefaultBtMessageFactory():_cuid(0),
_dhtEnabled(false), _dhtEnabled(false),
_metadataGetMode(false) _metadataGetMode(false)
{} {}
@ -78,7 +78,8 @@ DefaultBtMessageFactory::DefaultBtMessageFactory():cuid(0),
DefaultBtMessageFactory::~DefaultBtMessageFactory() {} DefaultBtMessageFactory::~DefaultBtMessageFactory() {}
BtMessageHandle BtMessageHandle
DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataLength) DefaultBtMessageFactory::createBtMessage
(const unsigned char* data, size_t dataLength)
{ {
AbstractBtMessageHandle msg; AbstractBtMessageHandle msg;
if(dataLength == 0) { if(dataLength == 0) {
@ -162,7 +163,8 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
BtPieceMessageHandle temp = BtPieceMessage::create(data, dataLength); BtPieceMessageHandle temp = BtPieceMessage::create(data, dataLength);
if(!_metadataGetMode) { if(!_metadataGetMode) {
BtMessageValidatorHandle validator BtMessageValidatorHandle validator
(new BtPieceMessageValidator(temp.get(), (new BtPieceMessageValidator
(temp.get(),
_downloadContext->getNumPieces(), _downloadContext->getNumPieces(),
_pieceStorage->getPieceLength(temp->getIndex()))); _pieceStorage->getPieceLength(temp->getIndex())));
temp->setBtMessageValidator(validator); temp->setBtMessageValidator(validator);
@ -191,7 +193,8 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
break; break;
} }
case BtSuggestPieceMessage::ID: { case BtSuggestPieceMessage::ID: {
BtSuggestPieceMessageHandle temp = BtSuggestPieceMessage::create(data, dataLength); BtSuggestPieceMessageHandle temp =
BtSuggestPieceMessage::create(data, dataLength);
if(!_metadataGetMode) { if(!_metadataGetMode) {
SharedHandle<BtMessageValidator> validator SharedHandle<BtMessageValidator> validator
(new IndexBtMessageValidator(temp.get(), (new IndexBtMessageValidator(temp.get(),
@ -202,7 +205,8 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
break; break;
} }
case BtAllowedFastMessage::ID: { case BtAllowedFastMessage::ID: {
BtAllowedFastMessageHandle temp = BtAllowedFastMessage::create(data, dataLength); BtAllowedFastMessageHandle temp =
BtAllowedFastMessage::create(data, dataLength);
if(!_metadataGetMode) { if(!_metadataGetMode) {
SharedHandle<BtMessageValidator> validator SharedHandle<BtMessageValidator> validator
(new IndexBtMessageValidator(temp.get(), (new IndexBtMessageValidator(temp.get(),
@ -213,7 +217,8 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
break; break;
} }
case BtPortMessage::ID: { case BtPortMessage::ID: {
SharedHandle<BtPortMessage> temp = BtPortMessage::create(data, dataLength); SharedHandle<BtPortMessage> temp =
BtPortMessage::create(data, dataLength);
temp->setLocalNode(_localNode); temp->setLocalNode(_localNode);
temp->setRoutingTable(_routingTable); temp->setRoutingTable(_routingTable);
temp->setTaskQueue(_taskQueue); temp->setTaskQueue(_taskQueue);
@ -222,11 +227,12 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
break; break;
} }
case BtExtendedMessage::ID: { case BtExtendedMessage::ID: {
if(peer->isExtendedMessagingEnabled()) { if(_peer->isExtendedMessagingEnabled()) {
msg = BtExtendedMessage::create(_extensionMessageFactory, msg = BtExtendedMessage::create(_extensionMessageFactory,
peer, data, dataLength); _peer, data, dataLength);
} else { } else {
throw DL_ABORT_EX("Received extended message from peer during a session with extended messaging disabled."); throw DL_ABORT_EX("Received extended message from peer during"
" a session with extended messaging disabled.");
} }
break; break;
} }
@ -238,23 +244,26 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL
return msg; return msg;
} }
void DefaultBtMessageFactory::setCommonProperty(const AbstractBtMessageHandle& msg) { void DefaultBtMessageFactory::setCommonProperty
msg->setCuid(cuid); (const AbstractBtMessageHandle& msg) {
msg->setPeer(peer); msg->setCuid(_cuid);
msg->setPeer(_peer);
msg->setPieceStorage(_pieceStorage); msg->setPieceStorage(_pieceStorage);
msg->setBtMessageDispatcher(dispatcher); msg->setBtMessageDispatcher(_dispatcher);
msg->setBtRequestFactory(requestFactory); msg->setBtRequestFactory(_requestFactory);
msg->setBtMessageFactory(WeakHandle<BtMessageFactory>(this)); msg->setBtMessageFactory(WeakHandle<BtMessageFactory>(this));
msg->setPeerConnection(peerConnection); msg->setPeerConnection(_peerConnection);
if(_metadataGetMode) { if(_metadataGetMode) {
msg->enableMetadataGetMode(); msg->enableMetadataGetMode();
} }
} }
SharedHandle<BtHandshakeMessage> SharedHandle<BtHandshakeMessage>
DefaultBtMessageFactory::createHandshakeMessage(const unsigned char* data, size_t dataLength) DefaultBtMessageFactory::createHandshakeMessage
(const unsigned char* data, size_t dataLength)
{ {
SharedHandle<BtHandshakeMessage> msg = BtHandshakeMessage::create(data, dataLength); SharedHandle<BtHandshakeMessage> msg =
BtHandshakeMessage::create(data, dataLength);
BtMessageValidatorHandle validator BtMessageValidatorHandle validator
(new BtHandshakeMessageValidator (new BtHandshakeMessageValidator
(msg.get(), bittorrent::getInfoHash(_downloadContext))); (msg.get(), bittorrent::getInfoHash(_downloadContext)));
@ -267,7 +276,8 @@ SharedHandle<BtHandshakeMessage>
DefaultBtMessageFactory::createHandshakeMessage(const unsigned char* infoHash, DefaultBtMessageFactory::createHandshakeMessage(const unsigned char* infoHash,
const unsigned char* peerId) const unsigned char* peerId)
{ {
SharedHandle<BtHandshakeMessage> msg(new BtHandshakeMessage(infoHash, peerId)); SharedHandle<BtHandshakeMessage> msg
(new BtHandshakeMessage(infoHash, peerId));
BtMessageValidatorHandle validator BtMessageValidatorHandle validator
(new BtHandshakeMessageValidator (new BtHandshakeMessageValidator
(msg.get(), bittorrent::getInfoHash(_downloadContext))); (msg.get(), bittorrent::getInfoHash(_downloadContext)));
@ -297,7 +307,8 @@ DefaultBtMessageFactory::createRequestMessage
} }
BtMessageHandle BtMessageHandle
DefaultBtMessageFactory::createCancelMessage(size_t index, uint32_t begin, size_t length) DefaultBtMessageFactory::createCancelMessage
(size_t index, uint32_t begin, size_t length)
{ {
BtCancelMessageHandle msg(new BtCancelMessage(index, begin, length)); BtCancelMessageHandle msg(new BtCancelMessage(index, begin, length));
SharedHandle<BtMessageValidator> validator SharedHandle<BtMessageValidator> validator
@ -311,7 +322,8 @@ DefaultBtMessageFactory::createCancelMessage(size_t index, uint32_t begin, size_
} }
BtMessageHandle BtMessageHandle
DefaultBtMessageFactory::createPieceMessage(size_t index, uint32_t begin, size_t length) DefaultBtMessageFactory::createPieceMessage
(size_t index, uint32_t begin, size_t length)
{ {
BtPieceMessageHandle msg(new BtPieceMessage(index, begin, length)); BtPieceMessageHandle msg(new BtPieceMessage(index, begin, length));
BtMessageValidatorHandle validator BtMessageValidatorHandle validator
@ -406,7 +418,8 @@ DefaultBtMessageFactory::createHaveNoneMessage()
} }
BtMessageHandle BtMessageHandle
DefaultBtMessageFactory::createRejectMessage(size_t index, uint32_t begin, size_t length) DefaultBtMessageFactory::createRejectMessage
(size_t index, uint32_t begin, size_t length)
{ {
BtRejectMessageHandle msg(new BtRejectMessage(index, begin, length)); BtRejectMessageHandle msg(new BtRejectMessage(index, begin, length));
SharedHandle<BtMessageValidator> validator SharedHandle<BtMessageValidator> validator
@ -439,26 +452,29 @@ DefaultBtMessageFactory::createPortMessage(uint16_t port)
} }
BtMessageHandle BtMessageHandle
DefaultBtMessageFactory::createBtExtendedMessage(const ExtensionMessageHandle& msg) DefaultBtMessageFactory::createBtExtendedMessage
(const ExtensionMessageHandle& msg)
{ {
BtExtendedMessageHandle m(new BtExtendedMessage(msg)); BtExtendedMessageHandle m(new BtExtendedMessage(msg));
setCommonProperty(m); setCommonProperty(m);
return m; return m;
} }
void DefaultBtMessageFactory::setTaskQueue(const WeakHandle<DHTTaskQueue>& taskQueue) void DefaultBtMessageFactory::setTaskQueue
(const WeakHandle<DHTTaskQueue>& taskQueue)
{ {
_taskQueue = taskQueue; _taskQueue = taskQueue;
} }
void DefaultBtMessageFactory::setTaskFactory(const WeakHandle<DHTTaskFactory>& taskFactory) void DefaultBtMessageFactory::setTaskFactory
(const WeakHandle<DHTTaskFactory>& taskFactory)
{ {
_taskFactory = taskFactory; _taskFactory = taskFactory;
} }
void DefaultBtMessageFactory::setPeer(const SharedHandle<Peer>& peer) void DefaultBtMessageFactory::setPeer(const SharedHandle<Peer>& peer)
{ {
this->peer = peer; _peer = peer;
} }
void DefaultBtMessageFactory::setDownloadContext void DefaultBtMessageFactory::setDownloadContext
@ -479,9 +495,10 @@ void DefaultBtMessageFactory::setPeerStorage
_peerStorage = peerStorage; _peerStorage = peerStorage;
} }
void DefaultBtMessageFactory::setBtMessageDispatcher(const WeakHandle<BtMessageDispatcher>& dispatcher) void DefaultBtMessageFactory::setBtMessageDispatcher
(const WeakHandle<BtMessageDispatcher>& dispatcher)
{ {
this->dispatcher = dispatcher; _dispatcher = dispatcher;
} }
void DefaultBtMessageFactory::setExtensionMessageFactory void DefaultBtMessageFactory::setExtensionMessageFactory
@ -495,19 +512,22 @@ void DefaultBtMessageFactory::setLocalNode(const WeakHandle<DHTNode>& localNode)
_localNode = localNode; _localNode = localNode;
} }
void DefaultBtMessageFactory::setRoutingTable(const WeakHandle<DHTRoutingTable>& routingTable) void DefaultBtMessageFactory::setRoutingTable
(const WeakHandle<DHTRoutingTable>& routingTable)
{ {
_routingTable = routingTable; _routingTable = routingTable;
} }
void DefaultBtMessageFactory::setBtRequestFactory(const WeakHandle<BtRequestFactory>& factory) void DefaultBtMessageFactory::setBtRequestFactory
(const WeakHandle<BtRequestFactory>& factory)
{ {
this->requestFactory = factory; _requestFactory = factory;
} }
void DefaultBtMessageFactory::setPeerConnection(const WeakHandle<PeerConnection>& connection) void DefaultBtMessageFactory::setPeerConnection
(const WeakHandle<PeerConnection>& connection)
{ {
this->peerConnection = connection; _peerConnection = connection;
} }
} // namespace aria2 } // namespace aria2

View File

@ -56,19 +56,19 @@ class DHTTaskFactory;
class DefaultBtMessageFactory : public BtMessageFactory { class DefaultBtMessageFactory : public BtMessageFactory {
private: private:
cuid_t cuid; cuid_t _cuid;
SharedHandle<DownloadContext> _downloadContext; SharedHandle<DownloadContext> _downloadContext;
SharedHandle<PieceStorage> _pieceStorage; SharedHandle<PieceStorage> _pieceStorage;
SharedHandle<PeerStorage> _peerStorage; SharedHandle<PeerStorage> _peerStorage;
SharedHandle<Peer> peer; SharedHandle<Peer> _peer;
bool _dhtEnabled; bool _dhtEnabled;
WeakHandle<BtMessageDispatcher> dispatcher; WeakHandle<BtMessageDispatcher> _dispatcher;
WeakHandle<BtRequestFactory> requestFactory; WeakHandle<BtRequestFactory> _requestFactory;
WeakHandle<PeerConnection> peerConnection; WeakHandle<PeerConnection> _peerConnection;
SharedHandle<ExtensionMessageFactory> _extensionMessageFactory; SharedHandle<ExtensionMessageFactory> _extensionMessageFactory;
@ -145,14 +145,15 @@ public:
void setCuid(cuid_t cuid) void setCuid(cuid_t cuid)
{ {
this->cuid = cuid; _cuid = cuid;
} }
void setDHTEnabled(bool enabled) { void setDHTEnabled(bool enabled) {
_dhtEnabled = enabled; _dhtEnabled = enabled;
} }
void setBtMessageDispatcher(const WeakHandle<BtMessageDispatcher>& dispatcher); void setBtMessageDispatcher
(const WeakHandle<BtMessageDispatcher>& dispatcher);
void setBtRequestFactory(const WeakHandle<BtRequestFactory>& factory); void setBtRequestFactory(const WeakHandle<BtRequestFactory>& factory);