github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity

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

Renamed member variables
	* src/BtBitfieldMessageValidator.h
	* src/BtHandshakeMessageValidator.h
	* src/BtPieceMessageValidator.h
pull/1/head
Tatsuhiro Tsujikawa 2010-06-11 14:27:09 +00:00
parent 9ebf141fb6
commit 0901dfcfc1
4 changed files with 35 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ChangeLog
View File

@ -1,3 +1,10 @@
2010-06-11 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Renamed member variables
* src/BtBitfieldMessageValidator.h
* src/BtHandshakeMessageValidator.h
* src/BtPieceMessageValidator.h
2010-06-11 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Renamed member variables

14
src/BtBitfieldMessageValidator.h
View File

@ -43,19 +43,19 @@ namespace aria2 {
class BtBitfieldMessageValidator : public BtMessageValidator {
private:
const BtBitfieldMessage* message;
size_t numPiece;
const BtBitfieldMessage* _message;
size_t _numPiece;
public:
BtBitfieldMessageValidator(const BtBitfieldMessage* message,
size_t numPiece):
message(message),
numPiece(numPiece) {}
_message(message),
_numPiece(numPiece) {}
virtual void validate()
{
bittorrent::checkBitfield(message->getBitfield(),
message->getBitfieldLength(),
numPiece);
bittorrent::checkBitfield(_message->getBitfield(),
_message->getBitfieldLength(),
_numPiece);
}
};

24
src/BtHandshakeMessageValidator.h
View File

@ -42,38 +42,40 @@
#include "BtHandshakeMessage.h"
#include "util.h"
#include "StringFormat.h"
#include "BtConstants.h"
namespace aria2 {
class BtHandshakeMessageValidator : public BtMessageValidator {
private:
const BtHandshakeMessage* message;
unsigned char infoHash[20];
const BtHandshakeMessage* _message;
unsigned char _infoHash[INFO_HASH_LENGTH];
public:
BtHandshakeMessageValidator(const BtHandshakeMessage* message,
const unsigned char* infoHash):
message(message)
_message(message)
{
memcpy(this->infoHash, infoHash, sizeof(this->infoHash));
memcpy(_infoHash, infoHash, sizeof(_infoHash));
}
virtual void validate()
{
if(message->getPstrlen() != 19) {
if(_message->getPstrlen() != 19) {
throw DL_ABORT_EX(StringFormat("invalid handshake pstrlen=%u",
message->getPstrlen()).str());
_message->getPstrlen()).str());
}
if(memcmp(BtHandshakeMessage::BT_PSTR, message->getPstr(), 19) != 0) {
if(memcmp(BtHandshakeMessage::BT_PSTR, _message->getPstr(), 19) != 0) {
throw DL_ABORT_EX
(StringFormat("invalid handshake pstr=%s",
util::percentEncode
(message->getPstr(), 19).c_str()).str());
(_message->getPstr(), 19).c_str()).str());
}
if(memcmp(infoHash, message->getInfoHash(), 20) != 0) {
if(memcmp(_infoHash, _message->getInfoHash(), sizeof(_infoHash)) != 0) {
throw DL_ABORT_EX
(StringFormat("invalid handshake info hash: expected:%s, actual:%s",
util::toHex(infoHash, 20).c_str(),
util::toHex(message->getInfoHash(), 20).c_str()).str());
util::toHex(_infoHash, sizeof(_infoHash)).c_str(),
util::toHex(_message->getInfoHash(),
INFO_HASH_LENGTH).c_str()).str());
}
}
};

16
src/BtPieceMessageValidator.h
View File

@ -43,21 +43,21 @@ namespace aria2 {
class BtPieceMessageValidator : public BtMessageValidator {
private:
const BtPieceMessage* message;
size_t numPiece;
size_t pieceLength;
const BtPieceMessage* _message;
size_t _numPiece;
size_t _pieceLength;
public:
BtPieceMessageValidator(const BtPieceMessage* message,
size_t numPiece,
size_t pieceLength):
message(message),
numPiece(numPiece),
pieceLength(pieceLength) {}
_message(message),
_numPiece(numPiece),
_pieceLength(pieceLength) {}
virtual void validate()
{
bittorrent::checkIndex(message->getIndex(), numPiece);
bittorrent::checkBegin(message->getBegin(), pieceLength);
bittorrent::checkIndex(_message->getIndex(), _numPiece);
bittorrent::checkBegin(_message->getBegin(), _pieceLength);
}
};