2009-11-22 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>

Added _metadataGetMode member variable. It toggles the action when
	messages are received: usually they almost do nothing if
	_metadataGetMode is true.
	* src/AbstractBtMessage.cc
	* src/AbstractBtMessage.h
	* src/BtAllowedFastMessage.cc
	* src/BtBitfieldMessage.cc
	* src/BtCancelMessage.cc
	* src/BtChokeMessage.cc
	* src/BtHaveAllMessage.cc
	* src/BtHaveMessage.cc
	* src/BtInterestedMessage.cc
	* src/BtNotInterestedMessage.cc
	* src/BtPieceMessage.cc
	* src/BtRejectMessage.cc
	* src/BtRequestMessage.cc
	* src/BtUnchokeMessage.cc

ChangeLog

@@ -1,3 +1,23 @@
+2009-11-22  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
+
+	Added _metadataGetMode member variable. It toggles the action when
+	messages are received: usually they almost do nothing if
+	_metadataGetMode is true.
+	* src/AbstractBtMessage.cc
+	* src/AbstractBtMessage.h
+	* src/BtAllowedFastMessage.cc
+	* src/BtBitfieldMessage.cc
+	* src/BtCancelMessage.cc
+	* src/BtChokeMessage.cc
+	* src/BtHaveAllMessage.cc
+	* src/BtHaveMessage.cc
+	* src/BtInterestedMessage.cc
+	* src/BtNotInterestedMessage.cc
+	* src/BtPieceMessage.cc
+	* src/BtRejectMessage.cc
+	* src/BtRequestMessage.cc
+	* src/BtUnchokeMessage.cc
+
 2009-11-22  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
 
 	Added bencode::decode overload functions. They have extra argument

src/AbstractBtMessage.cc

@@ -48,6 +48,7 @@ AbstractBtMessage::AbstractBtMessage(uint8_t id, const std::string& name):
   uploading(false),
   cuid(0),
   _name(name),
+  _metadataGetMode(false),
   logger(LogFactory::getInstance())
 {}
 

src/AbstractBtMessage.h

@@ -73,6 +73,8 @@ protected:
 
   SharedHandle<BtMessageValidator> validator;
 
+  bool _metadataGetMode;
+
   Logger* logger;
 public:
   AbstractBtMessage(uint8_t id, const std::string& name);
@@ -148,6 +150,11 @@ public:
   {
     return _name;
   }
+
+  void enableMetadataGetMode()
+  {
+    _metadataGetMode = true;
+  }
 };
 
 typedef SharedHandle<AbstractBtMessage> AbstractBtMessageHandle;

src/BtAllowedFastMessage.cc

@@ -53,6 +53,9 @@ void BtAllowedFastMessage::doReceivedAction() {
       (StringFormat("%s received while fast extension is disabled",
 		    toString().c_str()).str());
   }
+  if(_metadataGetMode) {
+    return;
+  }
   peer->addPeerAllowedIndex(getIndex());
 }
 

src/BtBitfieldMessage.cc

@@ -71,6 +71,9 @@ BtBitfieldMessage::create(const unsigned char* data, size_t dataLength)
 }
 
 void BtBitfieldMessage::doReceivedAction() {
+  if(_metadataGetMode) {
+    return;
+  }
   pieceStorage->updatePieceStats(bitfield, bitfieldLength, peer->getBitfield());
   peer->setBitfield(bitfield, bitfieldLength);
   if(peer->isSeeder() && pieceStorage->downloadFinished()) {

src/BtCancelMessage.cc

@@ -47,6 +47,9 @@ SharedHandle<BtCancelMessage> BtCancelMessage::create
 
 void BtCancelMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   dispatcher->doCancelSendingPieceAction(getIndex(), getBegin(), getLength());
 }
 

src/BtChokeMessage.cc

@@ -49,6 +49,9 @@ SharedHandle<BtChokeMessage> BtChokeMessage::create
 
 void BtChokeMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   peer->peerChoking(true);
   dispatcher->doChokedAction();
   requestFactory->doChokedAction();

src/BtHaveAllMessage.cc

@@ -56,6 +56,9 @@ void BtHaveAllMessage::doReceivedAction()
       (StringFormat("%s received while fast extension is disabled",
 		    toString().c_str()).str());
   }
+  if(_metadataGetMode) {
+    return;
+  }
   pieceStorage->subtractPieceStats(peer->getBitfield(),
 				   peer->getBitfieldLength());
   peer->setAllBitfield();

src/BtHaveMessage.cc

@@ -50,6 +50,9 @@ SharedHandle<BtHaveMessage> BtHaveMessage::create
 
 void BtHaveMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   peer->updateBitfield(getIndex(), 1);
   pieceStorage->addPieceStats(getIndex());
   if(peer->isSeeder() && pieceStorage->downloadFinished()) {

src/BtInterestedMessage.cc

@@ -48,6 +48,9 @@ SharedHandle<BtInterestedMessage> BtInterestedMessage::create
 
 void BtInterestedMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   peer->peerInterested(true);
   if(!peer->amChoking()) {
     _peerStorage->executeChoke();

src/BtNotInterestedMessage.cc

@@ -48,6 +48,9 @@ SharedHandle<BtNotInterestedMessage> BtNotInterestedMessage::create
 
 void BtNotInterestedMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   peer->peerInterested(false);
   if(!peer->amChoking()) {
     _peerStorage->executeChoke();

src/BtPieceMessage.cc

@@ -77,6 +77,9 @@ BtPieceMessageHandle BtPieceMessage::create(const unsigned char* data, size_t da
 }
 
 void BtPieceMessage::doReceivedAction() {
+  if(_metadataGetMode) {
+    return;
+  }
   RequestSlot slot = dispatcher->getOutstandingRequest(index,
 						       begin,
 						       blockLength);

src/BtRejectMessage.cc

@@ -56,6 +56,9 @@ void BtRejectMessage::doReceivedAction()
       (StringFormat("%s received while fast extension is disabled.",
 		    toString().c_str()).str());
   }
+  if(_metadataGetMode) {
+    return;
+  }
   // TODO Current implementation does not close a connection even if
   // a request for this reject message has never sent.
   RequestSlot slot =

src/BtRequestMessage.cc

@@ -51,6 +51,9 @@ SharedHandle<BtRequestMessage> BtRequestMessage::create
 
 void BtRequestMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   if(pieceStorage->hasPiece(getIndex()) &&
      (!peer->amChoking() ||
       (peer->amChoking() && peer->isInAmAllowedIndexSet(getIndex())))) {

src/BtUnchokeMessage.cc

@@ -47,6 +47,9 @@ SharedHandle<BtUnchokeMessage> BtUnchokeMessage::create
 
 void BtUnchokeMessage::doReceivedAction()
 {
+  if(_metadataGetMode) {
+    return;
+  }
   peer->peerChoking(false);
 }