mirrors/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-04-06 22:17:38 +03:00
Code Issues Projects Releases Packages Wiki Activity

2007-09-02 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>

Browse source 
Now *.aria2 contorol file is first saved to *.aria2__temp and if
	it is successful, then renamed to *.aria2.
	This prevents *.aria2 file from being truncated	or corrupted 
when
	file system becomes out of space.
	* src/DefaultBtProgressInfoFile.cc (save)
	* src/SegmentMan.cc (save)
	* test/DefaultBtProgressInfoFileTest.cc (testSave): Implemented.
This commit is contained in:
Tatsuhiro Tsujikawa 2007-09-01 16:10:30 +00:00
parent 2bea8759c4
commit 57471aac9c
6 changed files with 103 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

59
test/DefaultBtProgressInfoFileTest.cc
View file

@ -3,6 +3,10 @@
#include "Option.h"
#include "Util.h"
#include "Exception.h"
#include "MockBtContext.h"
#include "MockPeerStorage.h"
#include "MockPieceStorage.h"
#include "prefs.h"
#include <cppunit/extensions/HelperMacros.h>
using namespace std;
@ -31,4 +35,59 @@ public:
CPPUNIT_TEST_SUITE_REGISTRATION(DefaultBtProgressInfoFileTest);
void DefaultBtProgressInfoFileTest::testSave() {
unsigned char infoHash[] = {
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa,
0xbb, 0xcc, 0xdd, 0xee, 0xff, 0x00, 0xff, 0xff, 0xff, 0xff,
};
Option option;
option.put(PREF_DIR, ".");
MockBtContextHandle btContext = new MockBtContext();
btContext->setInfoHash(infoHash);
btContext->setName("save-temp");
BitfieldMan bitfield(1024, 80*1024);
bitfield.setAllBit();
bitfield.unsetBit(79);
MockPieceStorageHandle pieceStorage = new MockPieceStorage();
pieceStorage->setBitfield(&bitfield);
pieceStorage->setCompletedLength(80896);
MockPeerStorageHandle peerStorage = new MockPeerStorage();
TransferStat stat;
stat.sessionUploadLength = 1024;
peerStorage->setStat(stat);
BtRuntimeHandle btRuntime = new BtRuntime();
DefaultBtProgressInfoFile infoFile(btContext, &option);
infoFile.setPieceStorage(pieceStorage);
infoFile.setPeerStorage(peerStorage);
infoFile.setBtRuntime(btRuntime);
infoFile.save();
// read and validate
ifstream in(string(option.get(PREF_DIR)+"/"+btContext->getName()+".aria2").c_str());
unsigned char infoHashRead[20];
in.read((char*)infoHashRead, sizeof(infoHashRead));
CPPUNIT_ASSERT_EQUAL(string("112233445566778899aabbccddeeff00ffffffff"),
Util::toHex(infoHashRead, sizeof(infoHashRead)));
unsigned char bitfieldRead[10];
in.read((char*)bitfieldRead, sizeof(bitfieldRead));
CPPUNIT_ASSERT_EQUAL(string("fffffffffffffffffffe"),
Util::toHex(bitfieldRead, sizeof(bitfieldRead)));
int64_t allTimeDownloadLengthRead = 0;
in.read((char*)&allTimeDownloadLengthRead, sizeof(allTimeDownloadLengthRead));
CPPUNIT_ASSERT_EQUAL((int64_t)80896, allTimeDownloadLengthRead);
int64_t allTimeUploadLengthRead = 0;
in.read((char*)&allTimeUploadLengthRead, sizeof(allTimeUploadLengthRead));
CPPUNIT_ASSERT_EQUAL((int64_t)1024, allTimeUploadLengthRead);
string temp;
getline(in, temp);
CPPUNIT_ASSERT_EQUAL(string(""), temp);
CPPUNIT_ASSERT(in.eof());
}