mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 21:47:37 +03:00
f314719618
This option enables disk cache. If SIZE is 0, the disk cache is disabled. This feature caches the downloaded data in memory, which grows to at most SIZE bytes. The cache storage is created for aria2 instance and shared by all downloads. The one advantage of the disk cache is reduce the disk seek time because the data is written in larger unit and it is reordered by the offset of the file. If the underlying file is heavily fragmented it is not the case.
153 lines
3.7 KiB
C++
153 lines
3.7 KiB
C++
#include "Piece.h"
|
|
|
|
#include <string>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "util.h"
|
|
#include "DirectDiskAdaptor.h"
|
|
#include "ByteArrayDiskWriter.h"
|
|
#include "WrDiskCache.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class PieceTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(PieceTest);
|
|
CPPUNIT_TEST(testCompleteBlock);
|
|
CPPUNIT_TEST(testGetCompletedLength);
|
|
CPPUNIT_TEST(testFlushWrCache);
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
CPPUNIT_TEST(testGetDigestWithWrCache);
|
|
CPPUNIT_TEST(testUpdateHash);
|
|
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<DirectDiskAdaptor> adaptor_;
|
|
SharedHandle<ByteArrayDiskWriter> writer_;
|
|
public:
|
|
void setUp()
|
|
{
|
|
adaptor_.reset(new DirectDiskAdaptor());
|
|
writer_.reset(new ByteArrayDiskWriter());
|
|
adaptor_->setDiskWriter(writer_);
|
|
}
|
|
|
|
void testCompleteBlock();
|
|
void testGetCompletedLength();
|
|
void testFlushWrCache();
|
|
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
void testGetDigestWithWrCache();
|
|
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()
|
|
{
|
|
int32_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());
|
|
}
|
|
|
|
void PieceTest::testFlushWrCache()
|
|
{
|
|
unsigned char* data;
|
|
Piece p(0, 1024);
|
|
WrDiskCache dc(64);
|
|
p.initWrCache(&dc, adaptor_);
|
|
data = new unsigned char[3];
|
|
memcpy(data, "foo", 3);
|
|
p.updateWrCache(&dc, data, 0, 3, 0);
|
|
data = new unsigned char[4];
|
|
memcpy(data, " bar", 4);
|
|
p.updateWrCache(&dc, data, 0, 4, 3);
|
|
p.flushWrCache(&dc);
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("foo bar"), writer_->getString());
|
|
|
|
data = new unsigned char[3];
|
|
memcpy(data, "foo", 3);
|
|
p.updateWrCache(&dc, data, 0, 3, 0);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)3, dc.getSize());
|
|
p.clearWrCache(&dc);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, dc.getSize());
|
|
p.releaseWrCache(&dc);
|
|
CPPUNIT_ASSERT(!p.getWrDiskCacheEntry());
|
|
}
|
|
|
|
#ifdef ENABLE_MESSAGE_DIGEST
|
|
|
|
void PieceTest::testGetDigestWithWrCache()
|
|
{
|
|
unsigned char* data;
|
|
Piece p(0, 26);
|
|
p.setHashType("sha-1");
|
|
WrDiskCache dc(64);
|
|
// 012345678901234567890123456
|
|
writer_->setString("abcde...ijklmnopq...uvwx.z");
|
|
p.initWrCache(&dc, adaptor_);
|
|
data = new unsigned char[3];
|
|
memcpy(data, "fgh", 3);
|
|
p.updateWrCache(&dc, data, 0, 3, 5);
|
|
data = new unsigned char[3];
|
|
memcpy(data, "rst", 3);
|
|
p.updateWrCache(&dc, data, 0, 3, 17);
|
|
data = new unsigned char[1];
|
|
memcpy(data, "y", 1);
|
|
p.updateWrCache(&dc, data, 0, 1, 24);
|
|
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("32d10c7b8cf96570ca04ce37f2a19d84240d3a89"),
|
|
util::toHex(p.getDigestWithWrCache(p.getLength(), adaptor_)));
|
|
}
|
|
|
|
void PieceTest::testUpdateHash()
|
|
{
|
|
Piece p(0, 16, 2*1024*1024);
|
|
p.setHashType("sha-1");
|
|
|
|
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"),
|
|
util::toHex(p.getDigest()));
|
|
}
|
|
|
|
#endif // ENABLE_MESSAGE_DIGEST
|
|
|
|
} // namespace aria2
|