DefaultBtProgressInfoFileTest.cc 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579
  1. #include "DefaultBtProgressInfoFile.h"
  2. #include <fstream>
  3. #include <cppunit/extensions/HelperMacros.h>
  4. #include "Option.h"
  5. #include "util.h"
  6. #include "Exception.h"
  7. #include "MockPieceStorage.h"
  8. #include "prefs.h"
  9. #include "DownloadContext.h"
  10. #include "Piece.h"
  11. #include "FileEntry.h"
  12. #include "array_fun.h"
  13. #ifdef ENABLE_BITTORRENT
  14. # include "MockPeerStorage.h"
  15. # include "BtRuntime.h"
  16. # include "bittorrent_helper.h"
  17. #endif // ENABLE_BITTORRENT
  18. namespace aria2 {
  19. class DefaultBtProgressInfoFileTest:public CppUnit::TestFixture {
  20. CPPUNIT_TEST_SUITE(DefaultBtProgressInfoFileTest);
  21. #ifdef ENABLE_BITTORRENT
  22. CPPUNIT_TEST(testSave);
  23. CPPUNIT_TEST(testLoad);
  24. #ifndef WORDS_BIGENDIAN
  25. CPPUNIT_TEST(testLoad_compat);
  26. #endif // !WORDS_BIGENDIAN
  27. #endif // ENABLE_BITTORRENT
  28. CPPUNIT_TEST(testSave_nonBt);
  29. CPPUNIT_TEST(testLoad_nonBt);
  30. #ifndef WORDS_BIGENDIAN
  31. CPPUNIT_TEST(testLoad_nonBt_compat);
  32. #endif // !WORDS_BIGENDIAN
  33. CPPUNIT_TEST(testLoad_nonBt_pieceLengthShorter);
  34. CPPUNIT_TEST(testUpdateFilename);
  35. CPPUNIT_TEST_SUITE_END();
  36. private:
  37. #ifdef ENABLE_BITTORRENT
  38. SharedHandle<DownloadContext> dctx_;
  39. SharedHandle<MockPeerStorage> peerStorage_;
  40. SharedHandle<BtRuntime> btRuntime_;
  41. #endif // ENABLE_BITTORRENT
  42. SharedHandle<MockPieceStorage> pieceStorage_;
  43. SharedHandle<Option> option_;
  44. SharedHandle<BitfieldMan> bitfield_;
  45. public:
  46. void initializeMembers(int32_t pieceLength, int64_t totalLength)
  47. {
  48. option_.reset(new Option());
  49. option_->put(PREF_DIR, A2_TEST_OUT_DIR);
  50. bitfield_.reset(new BitfieldMan(pieceLength, totalLength));
  51. pieceStorage_.reset(new MockPieceStorage());
  52. pieceStorage_->setBitfield(bitfield_.get());
  53. #ifdef ENABLE_BITTORRENT
  54. static unsigned char infoHash[] = {
  55. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa,
  56. 0xbb, 0xcc, 0xdd, 0xee, 0xff, 0x00, 0xff, 0xff, 0xff, 0xff,
  57. };
  58. dctx_.reset(new DownloadContext());
  59. SharedHandle<TorrentAttribute> torrentAttrs(new TorrentAttribute());
  60. torrentAttrs->infoHash = std::string(vbegin(infoHash), vend(infoHash));
  61. dctx_->setAttribute(bittorrent::BITTORRENT, torrentAttrs);
  62. const SharedHandle<FileEntry> fileEntries[] = {
  63. SharedHandle<FileEntry>(new FileEntry("/path/to/file",totalLength,0))
  64. };
  65. dctx_->setFileEntries(vbegin(fileEntries), vend(fileEntries));
  66. dctx_->setPieceLength(pieceLength);
  67. peerStorage_.reset(new MockPeerStorage());
  68. btRuntime_.reset(new BtRuntime());
  69. #endif // ENABLE_BITTORRENT
  70. }
  71. #ifdef ENABLE_BITTORRENT
  72. void testSave();
  73. void testLoad();
  74. #ifndef WORDS_BIGENDIAN
  75. void testLoad_compat();
  76. #endif // !WORDS_BIGENDIAN
  77. #endif // ENABLE_BITTORRENT
  78. void testSave_nonBt();
  79. void testLoad_nonBt();
  80. #ifndef WORDS_BIGENDIAN
  81. void testLoad_nonBt_compat();
  82. #endif // !WORDS_BIGENDIAN
  83. void testLoad_nonBt_pieceLengthShorter();
  84. void testUpdateFilename();
  85. };
  86. #undef BLOCK_LENGTH
  87. #define BLOCK_LENGTH 256
  88. CPPUNIT_TEST_SUITE_REGISTRATION(DefaultBtProgressInfoFileTest);
  89. #ifdef ENABLE_BITTORRENT
  90. // Because load.aria2 is made for little endian systems, exclude
  91. // testLoad_compat() for big endian systems.
  92. #ifndef WORDS_BIGENDIAN
  93. void DefaultBtProgressInfoFileTest::testLoad_compat()
  94. {
  95. initializeMembers(1024, 81920);
  96. dctx_->setBasePath(A2_TEST_DIR"/load");
  97. DefaultBtProgressInfoFile infoFile(dctx_, pieceStorage_, option_.get());
  98. infoFile.setBtRuntime(btRuntime_);
  99. infoFile.setPeerStorage(peerStorage_);
  100. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load.aria2"),
  101. infoFile.getFilename());
  102. infoFile.load();
  103. // check the contents of objects
  104. // total length
  105. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx_->getTotalLength());
  106. // upload length
  107. CPPUNIT_ASSERT_EQUAL((int64_t)1024, btRuntime_->getUploadLengthAtStartup());
  108. // bitfield
  109. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  110. util::toHex(bitfield_->getBitfield(),
  111. bitfield_->getBitfieldLength()));
  112. // the number of in-flight pieces
  113. CPPUNIT_ASSERT_EQUAL((size_t)2,
  114. pieceStorage_->countInFlightPiece());
  115. // piece index 1
  116. std::vector<SharedHandle<Piece> > inFlightPieces;
  117. pieceStorage_->getInFlightPieces(inFlightPieces);
  118. SharedHandle<Piece> piece1 = inFlightPieces[0];
  119. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  120. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  121. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  122. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  123. piece1->getBitfieldLength()));
  124. // piece index 2
  125. SharedHandle<Piece> piece2 = inFlightPieces[1];
  126. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  127. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  128. }
  129. #endif // !WORDS_BIGENDIAN
  130. void DefaultBtProgressInfoFileTest::testLoad()
  131. {
  132. initializeMembers(1024, 81920);
  133. dctx_->setBasePath(A2_TEST_DIR"/load-v0001");
  134. DefaultBtProgressInfoFile infoFile(dctx_, pieceStorage_, option_.get());
  135. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-v0001.aria2"),
  136. infoFile.getFilename());
  137. infoFile.setBtRuntime(btRuntime_);
  138. infoFile.setPeerStorage(peerStorage_);
  139. infoFile.load();
  140. // check the contents of objects
  141. // total length
  142. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx_->getTotalLength());
  143. // upload length
  144. CPPUNIT_ASSERT_EQUAL((int64_t)1024, btRuntime_->getUploadLengthAtStartup());
  145. // bitfield
  146. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  147. util::toHex(bitfield_->getBitfield(),
  148. bitfield_->getBitfieldLength()));
  149. // the number of in-flight pieces
  150. CPPUNIT_ASSERT_EQUAL((size_t)2,
  151. pieceStorage_->countInFlightPiece());
  152. // piece index 1
  153. std::vector<SharedHandle<Piece> > inFlightPieces;
  154. pieceStorage_->getInFlightPieces(inFlightPieces);
  155. SharedHandle<Piece> piece1 = inFlightPieces[0];
  156. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  157. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  158. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  159. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  160. piece1->getBitfieldLength()));
  161. // piece index 2
  162. SharedHandle<Piece> piece2 = inFlightPieces[1];
  163. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  164. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  165. }
  166. void DefaultBtProgressInfoFileTest::testSave()
  167. {
  168. initializeMembers(1024, 81920);
  169. dctx_->setBasePath(A2_TEST_OUT_DIR"/save-temp");
  170. bitfield_->setAllBit();
  171. bitfield_->unsetBit(79);
  172. pieceStorage_->setCompletedLength(80896);
  173. TransferStat stat;
  174. stat.setAllTimeUploadLength(1024);
  175. peerStorage_->setStat(stat);
  176. SharedHandle<Piece> p1(new Piece(1, 1024));
  177. SharedHandle<Piece> p2(new Piece(2, 512));
  178. std::vector<SharedHandle<Piece> > inFlightPieces;
  179. inFlightPieces.push_back(p1);
  180. inFlightPieces.push_back(p2);
  181. pieceStorage_->addInFlightPiece(inFlightPieces);
  182. DefaultBtProgressInfoFile infoFile(dctx_, pieceStorage_, option_.get());
  183. infoFile.setBtRuntime(btRuntime_);
  184. infoFile.setPeerStorage(peerStorage_);
  185. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_OUT_DIR"/save-temp.aria2"),
  186. infoFile.getFilename());
  187. infoFile.save();
  188. // read and validate
  189. std::ifstream in(infoFile.getFilename().c_str(), std::ios::binary);
  190. //in.exceptions(ios::failbit);
  191. unsigned char version[2];
  192. in.read((char*)version, sizeof(version));
  193. CPPUNIT_ASSERT_EQUAL(std::string("0001"),
  194. util::toHex(version, sizeof(version)));
  195. unsigned char extension[4];
  196. in.read((char*)extension, sizeof(extension));
  197. CPPUNIT_ASSERT_EQUAL(std::string("00000001"),
  198. util::toHex(extension, sizeof(extension)));
  199. uint32_t infoHashLength;
  200. in.read(reinterpret_cast<char*>(&infoHashLength), sizeof(infoHashLength));
  201. infoHashLength = ntohl(infoHashLength);
  202. CPPUNIT_ASSERT_EQUAL((uint32_t)20, infoHashLength);
  203. unsigned char infoHashRead[20];
  204. in.read((char*)infoHashRead, sizeof(infoHashRead));
  205. CPPUNIT_ASSERT_EQUAL(std::string("112233445566778899aabbccddeeff00ffffffff"),
  206. util::toHex(infoHashRead, sizeof(infoHashRead)));
  207. uint32_t pieceLength;
  208. in.read((char*)&pieceLength, sizeof(pieceLength));
  209. pieceLength = ntohl(pieceLength);
  210. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength);
  211. uint64_t totalLength;
  212. in.read((char*)&totalLength, sizeof(totalLength));
  213. totalLength = ntoh64(totalLength);
  214. CPPUNIT_ASSERT_EQUAL((uint64_t)81920/* 80*1024 */, totalLength);
  215. uint64_t uploadLength;
  216. in.read((char*)&uploadLength, sizeof(uploadLength));
  217. uploadLength = ntoh64(uploadLength);
  218. CPPUNIT_ASSERT_EQUAL((uint64_t)1024, uploadLength);
  219. uint32_t bitfieldLength;
  220. in.read((char*)&bitfieldLength, sizeof(bitfieldLength));
  221. bitfieldLength = ntohl(bitfieldLength);
  222. CPPUNIT_ASSERT_EQUAL((uint32_t)10, bitfieldLength);
  223. unsigned char bitfieldRead[10];
  224. in.read((char*)bitfieldRead, sizeof(bitfieldRead));
  225. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  226. util::toHex(bitfieldRead, sizeof(bitfieldRead)));
  227. uint32_t numInFlightPiece;
  228. in.read((char*)&numInFlightPiece, sizeof(numInFlightPiece));
  229. numInFlightPiece = ntohl(numInFlightPiece);
  230. CPPUNIT_ASSERT_EQUAL((uint32_t)2, numInFlightPiece);
  231. // piece index 1
  232. uint32_t index1;
  233. in.read((char*)&index1, sizeof(index1));
  234. index1 = ntohl(index1);
  235. CPPUNIT_ASSERT_EQUAL((uint32_t)1, index1);
  236. uint32_t pieceLength1;
  237. in.read((char*)&pieceLength1, sizeof(pieceLength1));
  238. pieceLength1 = ntohl(pieceLength1);
  239. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength1);
  240. uint32_t pieceBitfieldLength1;
  241. in.read((char*)&pieceBitfieldLength1, sizeof(pieceBitfieldLength1));
  242. pieceBitfieldLength1 = ntohl(pieceBitfieldLength1);
  243. CPPUNIT_ASSERT_EQUAL((uint32_t)1, pieceBitfieldLength1);
  244. unsigned char pieceBitfield1[1];
  245. in.read((char*)pieceBitfield1, sizeof(pieceBitfield1));
  246. CPPUNIT_ASSERT_EQUAL(std::string("00"),
  247. util::toHex(pieceBitfield1, sizeof(pieceBitfield1)));
  248. // piece index 2
  249. uint32_t index2;
  250. in.read((char*)&index2, sizeof(index2));
  251. index2 = ntohl(index2);
  252. CPPUNIT_ASSERT_EQUAL((uint32_t)2, index2);
  253. uint32_t pieceLength2;
  254. in.read((char*)&pieceLength2, sizeof(pieceLength2));
  255. pieceLength2 = ntohl(pieceLength2);
  256. CPPUNIT_ASSERT_EQUAL((uint32_t)512, pieceLength2);
  257. }
  258. #endif // ENABLE_BITTORRENT
  259. // Because load-nonBt.aria2 is made for little endian systems, exclude
  260. // testLoad_nonBt_compat() for big endian systems.
  261. #ifndef WORDS_BIGENDIAN
  262. void DefaultBtProgressInfoFileTest::testLoad_nonBt_compat()
  263. {
  264. initializeMembers(1024, 81920);
  265. SharedHandle<DownloadContext> dctx
  266. (new DownloadContext(1024, 81920, A2_TEST_DIR"/load-nonBt"));
  267. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  268. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt.aria2"),
  269. infoFile.getFilename());
  270. infoFile.load();
  271. // check the contents of objects
  272. // total length
  273. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx->getTotalLength());
  274. // bitfield
  275. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  276. util::toHex(bitfield_->getBitfield(),
  277. bitfield_->getBitfieldLength()));
  278. // the number of in-flight pieces
  279. CPPUNIT_ASSERT_EQUAL((size_t)2,
  280. pieceStorage_->countInFlightPiece());
  281. // piece index 1
  282. std::vector<SharedHandle<Piece> > inFlightPieces;
  283. pieceStorage_->getInFlightPieces(inFlightPieces);
  284. SharedHandle<Piece> piece1 = inFlightPieces[0];
  285. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  286. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  287. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  288. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  289. piece1->getBitfieldLength()));
  290. // piece index 2
  291. SharedHandle<Piece> piece2 = inFlightPieces[1];
  292. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  293. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  294. }
  295. #endif // !WORDS_BIGENDIAN
  296. void DefaultBtProgressInfoFileTest::testLoad_nonBt()
  297. {
  298. initializeMembers(1024, 81920);
  299. SharedHandle<DownloadContext> dctx
  300. (new DownloadContext(1024, 81920, A2_TEST_DIR"/load-nonBt-v0001"));
  301. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  302. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt-v0001.aria2"),
  303. infoFile.getFilename());
  304. infoFile.load();
  305. // check the contents of objects
  306. // total length
  307. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx->getTotalLength());
  308. // bitfield
  309. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  310. util::toHex(bitfield_->getBitfield(),
  311. bitfield_->getBitfieldLength()));
  312. // the number of in-flight pieces
  313. CPPUNIT_ASSERT_EQUAL((size_t)2,
  314. pieceStorage_->countInFlightPiece());
  315. // piece index 1
  316. std::vector<SharedHandle<Piece> > inFlightPieces;
  317. pieceStorage_->getInFlightPieces(inFlightPieces);
  318. SharedHandle<Piece> piece1 = inFlightPieces[0];
  319. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  320. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  321. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  322. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  323. piece1->getBitfieldLength()));
  324. // piece index 2
  325. SharedHandle<Piece> piece2 = inFlightPieces[1];
  326. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  327. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  328. }
  329. void DefaultBtProgressInfoFileTest::testLoad_nonBt_pieceLengthShorter()
  330. {
  331. initializeMembers(512, 81920);
  332. option_->put(PREF_ALLOW_PIECE_LENGTH_CHANGE, A2_V_TRUE);
  333. SharedHandle<DownloadContext> dctx
  334. (new DownloadContext(512, 81920, A2_TEST_DIR"/load-nonBt-v0001"));
  335. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  336. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt-v0001.aria2"),
  337. infoFile.getFilename());
  338. infoFile.load();
  339. // check the contents of objects
  340. // bitfield
  341. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffffffffffffffffffffffc"),
  342. util::toHex(bitfield_->getBitfield(),
  343. bitfield_->getBitfieldLength()));
  344. // the number of in-flight pieces
  345. CPPUNIT_ASSERT_EQUAL((size_t)0,
  346. pieceStorage_->countInFlightPiece());
  347. }
  348. void DefaultBtProgressInfoFileTest::testSave_nonBt()
  349. {
  350. initializeMembers(1024, 81920);
  351. SharedHandle<DownloadContext> dctx
  352. (new DownloadContext(1024, 81920, A2_TEST_OUT_DIR"/save-temp"));
  353. bitfield_->setAllBit();
  354. bitfield_->unsetBit(79);
  355. pieceStorage_->setCompletedLength(80896);
  356. SharedHandle<Piece> p1(new Piece(1, 1024));
  357. SharedHandle<Piece> p2(new Piece(2, 512));
  358. std::vector<SharedHandle<Piece> > inFlightPieces;
  359. inFlightPieces.push_back(p1);
  360. inFlightPieces.push_back(p2);
  361. pieceStorage_->addInFlightPiece(inFlightPieces);
  362. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  363. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_OUT_DIR"/save-temp.aria2"),
  364. infoFile.getFilename());
  365. infoFile.save();
  366. // read and validate
  367. std::ifstream in(infoFile.getFilename().c_str(), std::ios::binary);
  368. //in.exceptions(ios::failbit);
  369. unsigned char version[2];
  370. in.read((char*)version, sizeof(version));
  371. CPPUNIT_ASSERT_EQUAL(std::string("0001"),
  372. util::toHex(version, sizeof(version)));
  373. unsigned char extension[4];
  374. in.read((char*)extension, sizeof(extension));
  375. CPPUNIT_ASSERT_EQUAL(std::string("00000000"),
  376. util::toHex(extension, sizeof(extension)));
  377. uint32_t infoHashLength;
  378. in.read(reinterpret_cast<char*>(&infoHashLength), sizeof(infoHashLength));
  379. infoHashLength = ntohl(infoHashLength);
  380. CPPUNIT_ASSERT_EQUAL((uint32_t)0, infoHashLength);
  381. uint32_t pieceLength;
  382. in.read((char*)&pieceLength, sizeof(pieceLength));
  383. pieceLength = ntohl(pieceLength);
  384. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength);
  385. uint64_t totalLength;
  386. in.read((char*)&totalLength, sizeof(totalLength));
  387. totalLength = ntoh64(totalLength);
  388. CPPUNIT_ASSERT_EQUAL((uint64_t)81920/* 80*1024 */, totalLength);
  389. uint64_t uploadLength;
  390. in.read((char*)&uploadLength, sizeof(uploadLength));
  391. uploadLength = ntoh64(uploadLength);
  392. CPPUNIT_ASSERT_EQUAL((uint64_t)0, uploadLength);
  393. uint32_t bitfieldLength;
  394. in.read((char*)&bitfieldLength, sizeof(bitfieldLength));
  395. bitfieldLength = ntohl(bitfieldLength);
  396. CPPUNIT_ASSERT_EQUAL((uint32_t)10, bitfieldLength);
  397. unsigned char bitfieldRead[10];
  398. in.read((char*)bitfieldRead, sizeof(bitfieldRead));
  399. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  400. util::toHex(bitfieldRead, sizeof(bitfieldRead)));
  401. uint32_t numInFlightPiece;
  402. in.read((char*)&numInFlightPiece, sizeof(numInFlightPiece));
  403. numInFlightPiece = ntohl(numInFlightPiece);
  404. CPPUNIT_ASSERT_EQUAL((uint32_t)2, numInFlightPiece);
  405. // piece index 1
  406. uint32_t index1;
  407. in.read((char*)&index1, sizeof(index1));
  408. index1 = ntohl(index1);
  409. CPPUNIT_ASSERT_EQUAL((uint32_t)1, index1);
  410. uint32_t pieceLength1;
  411. in.read((char*)&pieceLength1, sizeof(pieceLength1));
  412. pieceLength1 = ntohl(pieceLength1);
  413. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength1);
  414. uint32_t pieceBitfieldLength1;
  415. in.read((char*)&pieceBitfieldLength1, sizeof(pieceBitfieldLength1));
  416. pieceBitfieldLength1 = ntohl(pieceBitfieldLength1);
  417. CPPUNIT_ASSERT_EQUAL((uint32_t)1, pieceBitfieldLength1);
  418. unsigned char pieceBitfield1[1];
  419. in.read((char*)pieceBitfield1, sizeof(pieceBitfield1));
  420. CPPUNIT_ASSERT_EQUAL(std::string("00"),
  421. util::toHex(pieceBitfield1, sizeof(pieceBitfield1)));
  422. // piece index 2
  423. uint32_t index2;
  424. in.read((char*)&index2, sizeof(index2));
  425. index2 = ntohl(index2);
  426. CPPUNIT_ASSERT_EQUAL((uint32_t)2, index2);
  427. uint32_t pieceLength2;
  428. in.read((char*)&pieceLength2, sizeof(pieceLength2));
  429. pieceLength2 = ntohl(pieceLength2);
  430. CPPUNIT_ASSERT_EQUAL((uint32_t)512, pieceLength2);
  431. }
  432. void DefaultBtProgressInfoFileTest::testUpdateFilename()
  433. {
  434. SharedHandle<DownloadContext> dctx
  435. (new DownloadContext(1024, 81920, A2_TEST_DIR"/file1"));
  436. DefaultBtProgressInfoFile infoFile(dctx, SharedHandle<MockPieceStorage>(), 0);
  437. #ifdef ENABLE_BITTORRENT
  438. infoFile.setBtRuntime(btRuntime_);
  439. infoFile.setPeerStorage(peerStorage_);
  440. #endif // ENABLE_BITTORRENT
  441. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.aria2"),
  442. infoFile.getFilename());
  443. dctx->getFirstFileEntry()->setPath(A2_TEST_DIR"/file1.1");
  444. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.aria2"),
  445. infoFile.getFilename());
  446. infoFile.updateFilename();
  447. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.1.aria2"),
  448. infoFile.getFilename());
  449. }
  450. } // namespace aria2