mirror of https://github.com/aria2/aria2
353 lines
10 KiB
C++
353 lines
10 KiB
C++
#include "PeerMessageUtil.h"
|
|
#include <netinet/in.h>
|
|
#include <string>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
using namespace std;
|
|
|
|
class PeerMessageUtilTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(PeerMessageUtilTest);
|
|
CPPUNIT_TEST(testCreatePeerMessageKeepAlive);
|
|
CPPUNIT_TEST(testCreatePeerMessageChoke);
|
|
CPPUNIT_TEST(testCreatePeerMessageUnchoke);
|
|
CPPUNIT_TEST(testCreatePeerMessageInterested);
|
|
CPPUNIT_TEST(testCreatePeerMessageNotInterested);
|
|
CPPUNIT_TEST(testCreatePeerMessageHave);
|
|
CPPUNIT_TEST(testCreatePeerMessageBitfield);
|
|
CPPUNIT_TEST(testCreatePeerMessageRequest);
|
|
CPPUNIT_TEST(testCreatePeerMessagePiece);
|
|
CPPUNIT_TEST(testCreatePeerMessageCancel);
|
|
CPPUNIT_TEST(testCheckIntegrityHave);
|
|
CPPUNIT_TEST(testCheckIntegrityBitfield);
|
|
CPPUNIT_TEST(testCheckIntegrityRequest);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testCreatePeerMessageKeepAlive();
|
|
void testCreatePeerMessageChoke();
|
|
void testCreatePeerMessageUnchoke();
|
|
void testCreatePeerMessageInterested();
|
|
void testCreatePeerMessageNotInterested();
|
|
void testCreatePeerMessageHave();
|
|
void testCreatePeerMessageBitfield();
|
|
void testCreatePeerMessageRequest();
|
|
void testCreatePeerMessagePiece();
|
|
void testCreatePeerMessageCancel();
|
|
|
|
void testCheckIntegrityHave();
|
|
void testCheckIntegrityBitfield();
|
|
void testCheckIntegrityRequest();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( PeerMessageUtilTest );
|
|
|
|
void setIntParam(char* dest, int param) {
|
|
int nParam = htonl(param);
|
|
memcpy(dest, &nParam, 4);
|
|
}
|
|
|
|
void createNLengthMessage(char* msg, int msgLen, int payloadLen, int id) {
|
|
memset(msg, 0, msgLen);
|
|
setIntParam(msg, payloadLen);
|
|
msg[4] = (char)id;
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageKeepAlive() {
|
|
char msg[4];
|
|
memset(msg, 0, sizeof(msg));
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(NULL, 0);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::KEEP_ALIVE, pm->id);
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageChoke() {
|
|
char msg[5];
|
|
createNLengthMessage(msg, sizeof(msg), 1, 0);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::CHOKE, pm->id);
|
|
|
|
try {
|
|
char msg[6];
|
|
createNLengthMessage(msg, sizeof(msg), 2, 0);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 2);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageUnchoke() {
|
|
char msg[5];
|
|
createNLengthMessage(msg, sizeof(msg), 1, 1);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::UNCHOKE, pm->id);
|
|
|
|
try {
|
|
char msg[6];
|
|
createNLengthMessage(msg, sizeof(msg), 2, 1);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 2);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageInterested() {
|
|
char msg[5];
|
|
createNLengthMessage(msg, sizeof(msg), 1, 2);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::INTERESTED, pm->id);
|
|
|
|
try {
|
|
char msg[6];
|
|
createNLengthMessage(msg, sizeof(msg), 2, 2);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 2);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageNotInterested() {
|
|
char msg[5];
|
|
createNLengthMessage(msg, sizeof(msg), 1, 3);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::NOT_INTERESTED, pm->id);
|
|
|
|
try {
|
|
char msg[6];
|
|
createNLengthMessage(msg, sizeof(msg), 2, 3);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 2);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageHave() {
|
|
char msg[9];
|
|
createNLengthMessage(msg, sizeof(msg), 5, 4);
|
|
setIntParam(&msg[5], 100);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 5);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::HAVE, pm->id);
|
|
CPPUNIT_ASSERT_EQUAL(100, pm->index);
|
|
|
|
try {
|
|
char msg[8];
|
|
createNLengthMessage(msg, sizeof(msg), 4, 4);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 4);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
|
|
try {
|
|
char msg[5];
|
|
createNLengthMessage(msg, sizeof(msg), 1, 4);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageBitfield() {
|
|
int msgLen = 5+2;
|
|
char* msg = new char[msgLen];
|
|
createNLengthMessage(msg, msgLen, 3, 5);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 3);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::BITFIELD, pm->id);
|
|
CPPUNIT_ASSERT_EQUAL((unsigned char)0, pm->bitfield[0]);
|
|
CPPUNIT_ASSERT_EQUAL((unsigned char)0, pm->bitfield[1]);
|
|
CPPUNIT_ASSERT_EQUAL(2, pm->bitfieldLen);
|
|
|
|
try {
|
|
int msgLen = 5;
|
|
char* msg = new char[msgLen];
|
|
createNLengthMessage(msg, msgLen, 1, 5);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 1);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageRequest() {
|
|
char msg[17];
|
|
createNLengthMessage(msg, sizeof(msg), 13, 6);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
setIntParam(&msg[13], 16*1024-1);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 13);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::REQUEST, pm->id);
|
|
CPPUNIT_ASSERT_EQUAL(1, pm->index);
|
|
CPPUNIT_ASSERT_EQUAL(16*1024, pm->begin);
|
|
CPPUNIT_ASSERT_EQUAL(16*1024-1, pm->length);
|
|
|
|
try {
|
|
char msg[13];
|
|
createNLengthMessage(msg, sizeof(msg), 9, 6);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 9);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessagePiece() {
|
|
char msg[23];
|
|
createNLengthMessage(msg, sizeof(msg), 9+10, 7);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 19);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::PIECE, pm->id);
|
|
CPPUNIT_ASSERT_EQUAL(1, pm->index);
|
|
CPPUNIT_ASSERT_EQUAL(16*1024, pm->begin);
|
|
CPPUNIT_ASSERT_EQUAL(10, pm->blockLen);
|
|
for(int i = 0; i < 10; i++) {
|
|
CPPUNIT_ASSERT_EQUAL((char)0, pm->block[i]);
|
|
}
|
|
|
|
try {
|
|
char msg[13];
|
|
createNLengthMessage(msg, sizeof(msg), 9, 7);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 9);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCreatePeerMessageCancel() {
|
|
char msg[17];
|
|
createNLengthMessage(msg, sizeof(msg), 13, 8);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
setIntParam(&msg[13], 16*1024-1);
|
|
PeerMessage* pm = PeerMessageUtil::createPeerMessage(&msg[4], 13);
|
|
CPPUNIT_ASSERT_EQUAL((int)PeerMessage::CANCEL, pm->id);
|
|
CPPUNIT_ASSERT_EQUAL(1, pm->index);
|
|
CPPUNIT_ASSERT_EQUAL(16*1024, pm->begin);
|
|
CPPUNIT_ASSERT_EQUAL(16*1024-1, pm->length);
|
|
|
|
try {
|
|
char msg[13];
|
|
createNLengthMessage(msg, sizeof(msg), 9, 8);
|
|
setIntParam(&msg[5], 1);
|
|
setIntParam(&msg[9], 16*1024);
|
|
PeerMessageUtil::createPeerMessage(&msg[4], 9);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCheckIntegrityHave() {
|
|
PeerMessage* pm = new PeerMessage();
|
|
pm->id = PeerMessage::HAVE;
|
|
pm->index = 119;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
} catch(Exception* ex) {
|
|
cerr << ex->getMsg() << endl;
|
|
CPPUNIT_FAIL("");
|
|
}
|
|
|
|
pm->index = 120;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
|
|
void PeerMessageUtilTest::testCheckIntegrityBitfield() {
|
|
PeerMessage* pm = new PeerMessage();
|
|
pm->id = PeerMessage::BITFIELD;
|
|
pm->bitfieldLen = 15;
|
|
pm->bitfield = new unsigned char[pm->bitfieldLen];
|
|
memset(pm->bitfield, 1, pm->bitfieldLen);
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
} catch(Exception* ex) {
|
|
cerr << ex->getMsg() << endl;
|
|
CPPUNIT_FAIL("");
|
|
}
|
|
|
|
pm->bitfieldLen = 16;
|
|
pm->bitfield = new unsigned char[pm->bitfieldLen];
|
|
memset(pm->bitfield, 1, pm->bitfieldLen);
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {
|
|
}
|
|
|
|
pm->bitfieldLen = 14;
|
|
pm->bitfield = new unsigned char[pm->bitfieldLen];
|
|
memset(pm->bitfield, 1, pm->bitfieldLen);
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {
|
|
}
|
|
|
|
pm->bitfieldLen = 15;
|
|
pm->bitfield = new unsigned char[pm->bitfieldLen];
|
|
memset(pm->bitfield, 1, pm->bitfieldLen);
|
|
pm->bitfield[pm->bitfieldLen-1] &= 0xfe;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 119, 256*1024*120);
|
|
} catch(Exception* ex) {
|
|
cerr << ex->getMsg() << endl;
|
|
CPPUNIT_FAIL("");
|
|
}
|
|
|
|
pm->bitfieldLen = 15;
|
|
pm->bitfield = new unsigned char[pm->bitfieldLen];
|
|
memset(pm->bitfield, 1, pm->bitfieldLen);
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 119, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {
|
|
}
|
|
}
|
|
|
|
void PeerMessageUtilTest::testCheckIntegrityRequest() {
|
|
PeerMessage* pm = new PeerMessage();
|
|
pm->id = PeerMessage::REQUEST;
|
|
pm->index = 119;
|
|
pm->begin = 0;
|
|
pm->length = 16*1024;
|
|
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
} catch(Exception* ex) {
|
|
cerr << ex->getMsg() << endl;
|
|
CPPUNIT_FAIL("");
|
|
}
|
|
|
|
pm->begin = 256*1024;
|
|
pm->length = 16*1024;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {}
|
|
|
|
pm->begin = 0;
|
|
pm->length = 256*1024+1;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {}
|
|
|
|
pm->begin = 0;
|
|
pm->length = 256*1024;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120);
|
|
} catch(Exception* ex) {
|
|
cerr << ex->getMsg() << endl;
|
|
CPPUNIT_FAIL("");
|
|
}
|
|
|
|
pm->begin = 0;
|
|
pm->length = 256*1024;
|
|
try {
|
|
PeerMessageUtil::checkIntegrity(pm, 256*1024, 120, 256*1024*120-1);
|
|
CPPUNIT_FAIL("exception must be throwed.");
|
|
} catch(Exception* ex) {}
|
|
}
|