mirror of
https://github.com/aria2/aria2.git
synced 2025-02-26 08:22:11 +00:00
7d63daed22
Calculate piece hash when data is arrived if the data is arrived in order. This removes additional read operation for hash calculation. If the data is arrived in out of order, the hash is calucated when the piece is completed. This is the same behavior as the old implementation. * src/BtPieceMessage.cc * src/DefaultBtProgressInfoFile.cc * src/DefaultPieceStorage.cc * src/DownloadCommand.cc * src/DownloadCommand.h * src/DownloadEngine.cc * src/GrowSegment.cc * src/GrowSegment.h * src/Piece.cc * src/Piece.h * src/PiecedSegment.cc * src/PiecedSegment.h * src/Segment.h * test/PieceTest.cc
91 lines
1.8 KiB
C++
91 lines
1.8 KiB
C++
#include "Piece.h"
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
# include "messageDigest.h"
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
#include <string>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class PieceTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(PieceTest);
|
|
CPPUNIT_TEST(testCompleteBlock);
|
|
CPPUNIT_TEST(testGetCompletedLength);
|
|
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
CPPUNIT_TEST(testUpdateHash);
|
|
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {}
|
|
|
|
void testCompleteBlock();
|
|
void testGetCompletedLength();
|
|
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
void testUpdateHash();
|
|
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( PieceTest );
|
|
|
|
void PieceTest::testCompleteBlock()
|
|
{
|
|
size_t blockLength = 32*1024;
|
|
Piece p(0, blockLength*10, blockLength);
|
|
|
|
p.completeBlock(5);
|
|
|
|
CPPUNIT_ASSERT(p.hasBlock(5));
|
|
}
|
|
|
|
void PieceTest::testGetCompletedLength()
|
|
{
|
|
size_t blockLength = 16*1024;
|
|
Piece p(0, blockLength*10+100, blockLength);
|
|
|
|
p.completeBlock(1);
|
|
p.completeBlock(2);
|
|
p.completeBlock(9);
|
|
p.completeBlock(10); // <-- 100 bytes
|
|
|
|
CPPUNIT_ASSERT_EQUAL(blockLength*3+100, p.getCompletedLength());
|
|
}
|
|
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
void PieceTest::testUpdateHash()
|
|
{
|
|
Piece p(0, 16, 2*1024*1024);
|
|
p.setHashAlgo(MessageDigestContext::SHA1);
|
|
|
|
std::string spam("SPAM!");
|
|
CPPUNIT_ASSERT(p.updateHash
|
|
(0, reinterpret_cast<const unsigned char*>(spam.c_str()),
|
|
spam.size()));
|
|
CPPUNIT_ASSERT(!p.isHashCalculated());
|
|
|
|
std::string spamspam("SPAM!SPAM!!");
|
|
CPPUNIT_ASSERT(p.updateHash
|
|
(spam.size(),
|
|
reinterpret_cast<const unsigned char*>(spamspam.c_str()),
|
|
spamspam.size()));
|
|
CPPUNIT_ASSERT(p.isHashCalculated());
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("d9189aff79e075a2e60271b9556a710dc1bc7de7"),
|
|
p.getHashString());
|
|
}
|
|
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
|
|
} // namespace aria2
|