DefaultBtProgressInfoFileTest.cc 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578
  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(CTX_ATTR_BT, 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. dctx_->getNetStat().updateUploadLength(768);
  171. btRuntime_->setUploadLengthAtStartup(256);
  172. bitfield_->setAllBit();
  173. bitfield_->unsetBit(79);
  174. pieceStorage_->setCompletedLength(80896);
  175. SharedHandle<Piece> p1(new Piece(1, 1024));
  176. SharedHandle<Piece> p2(new Piece(2, 512));
  177. std::vector<SharedHandle<Piece> > inFlightPieces;
  178. inFlightPieces.push_back(p1);
  179. inFlightPieces.push_back(p2);
  180. pieceStorage_->addInFlightPiece(inFlightPieces);
  181. DefaultBtProgressInfoFile infoFile(dctx_, pieceStorage_, option_.get());
  182. infoFile.setBtRuntime(btRuntime_);
  183. infoFile.setPeerStorage(peerStorage_);
  184. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_OUT_DIR"/save-temp.aria2"),
  185. infoFile.getFilename());
  186. infoFile.save();
  187. // read and validate
  188. std::ifstream in(infoFile.getFilename().c_str(), std::ios::binary);
  189. //in.exceptions(ios::failbit);
  190. unsigned char version[2];
  191. in.read((char*)version, sizeof(version));
  192. CPPUNIT_ASSERT_EQUAL(std::string("0001"),
  193. util::toHex(version, sizeof(version)));
  194. unsigned char extension[4];
  195. in.read((char*)extension, sizeof(extension));
  196. CPPUNIT_ASSERT_EQUAL(std::string("00000001"),
  197. util::toHex(extension, sizeof(extension)));
  198. uint32_t infoHashLength;
  199. in.read(reinterpret_cast<char*>(&infoHashLength), sizeof(infoHashLength));
  200. infoHashLength = ntohl(infoHashLength);
  201. CPPUNIT_ASSERT_EQUAL((uint32_t)20, infoHashLength);
  202. unsigned char infoHashRead[20];
  203. in.read((char*)infoHashRead, sizeof(infoHashRead));
  204. CPPUNIT_ASSERT_EQUAL(std::string("112233445566778899aabbccddeeff00ffffffff"),
  205. util::toHex(infoHashRead, sizeof(infoHashRead)));
  206. uint32_t pieceLength;
  207. in.read((char*)&pieceLength, sizeof(pieceLength));
  208. pieceLength = ntohl(pieceLength);
  209. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength);
  210. uint64_t totalLength;
  211. in.read((char*)&totalLength, sizeof(totalLength));
  212. totalLength = ntoh64(totalLength);
  213. CPPUNIT_ASSERT_EQUAL((uint64_t)81920/* 80*1024 */, totalLength);
  214. uint64_t uploadLength;
  215. in.read((char*)&uploadLength, sizeof(uploadLength));
  216. uploadLength = ntoh64(uploadLength);
  217. CPPUNIT_ASSERT_EQUAL((uint64_t)1024, uploadLength);
  218. uint32_t bitfieldLength;
  219. in.read((char*)&bitfieldLength, sizeof(bitfieldLength));
  220. bitfieldLength = ntohl(bitfieldLength);
  221. CPPUNIT_ASSERT_EQUAL((uint32_t)10, bitfieldLength);
  222. unsigned char bitfieldRead[10];
  223. in.read((char*)bitfieldRead, sizeof(bitfieldRead));
  224. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  225. util::toHex(bitfieldRead, sizeof(bitfieldRead)));
  226. uint32_t numInFlightPiece;
  227. in.read((char*)&numInFlightPiece, sizeof(numInFlightPiece));
  228. numInFlightPiece = ntohl(numInFlightPiece);
  229. CPPUNIT_ASSERT_EQUAL((uint32_t)2, numInFlightPiece);
  230. // piece index 1
  231. uint32_t index1;
  232. in.read((char*)&index1, sizeof(index1));
  233. index1 = ntohl(index1);
  234. CPPUNIT_ASSERT_EQUAL((uint32_t)1, index1);
  235. uint32_t pieceLength1;
  236. in.read((char*)&pieceLength1, sizeof(pieceLength1));
  237. pieceLength1 = ntohl(pieceLength1);
  238. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength1);
  239. uint32_t pieceBitfieldLength1;
  240. in.read((char*)&pieceBitfieldLength1, sizeof(pieceBitfieldLength1));
  241. pieceBitfieldLength1 = ntohl(pieceBitfieldLength1);
  242. CPPUNIT_ASSERT_EQUAL((uint32_t)1, pieceBitfieldLength1);
  243. unsigned char pieceBitfield1[1];
  244. in.read((char*)pieceBitfield1, sizeof(pieceBitfield1));
  245. CPPUNIT_ASSERT_EQUAL(std::string("00"),
  246. util::toHex(pieceBitfield1, sizeof(pieceBitfield1)));
  247. // piece index 2
  248. uint32_t index2;
  249. in.read((char*)&index2, sizeof(index2));
  250. index2 = ntohl(index2);
  251. CPPUNIT_ASSERT_EQUAL((uint32_t)2, index2);
  252. uint32_t pieceLength2;
  253. in.read((char*)&pieceLength2, sizeof(pieceLength2));
  254. pieceLength2 = ntohl(pieceLength2);
  255. CPPUNIT_ASSERT_EQUAL((uint32_t)512, pieceLength2);
  256. }
  257. #endif // ENABLE_BITTORRENT
  258. // Because load-nonBt.aria2 is made for little endian systems, exclude
  259. // testLoad_nonBt_compat() for big endian systems.
  260. #ifndef WORDS_BIGENDIAN
  261. void DefaultBtProgressInfoFileTest::testLoad_nonBt_compat()
  262. {
  263. initializeMembers(1024, 81920);
  264. SharedHandle<DownloadContext> dctx
  265. (new DownloadContext(1024, 81920, A2_TEST_DIR"/load-nonBt"));
  266. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  267. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt.aria2"),
  268. infoFile.getFilename());
  269. infoFile.load();
  270. // check the contents of objects
  271. // total length
  272. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx->getTotalLength());
  273. // bitfield
  274. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  275. util::toHex(bitfield_->getBitfield(),
  276. bitfield_->getBitfieldLength()));
  277. // the number of in-flight pieces
  278. CPPUNIT_ASSERT_EQUAL((size_t)2,
  279. pieceStorage_->countInFlightPiece());
  280. // piece index 1
  281. std::vector<SharedHandle<Piece> > inFlightPieces;
  282. pieceStorage_->getInFlightPieces(inFlightPieces);
  283. SharedHandle<Piece> piece1 = inFlightPieces[0];
  284. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  285. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  286. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  287. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  288. piece1->getBitfieldLength()));
  289. // piece index 2
  290. SharedHandle<Piece> piece2 = inFlightPieces[1];
  291. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  292. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  293. }
  294. #endif // !WORDS_BIGENDIAN
  295. void DefaultBtProgressInfoFileTest::testLoad_nonBt()
  296. {
  297. initializeMembers(1024, 81920);
  298. SharedHandle<DownloadContext> dctx
  299. (new DownloadContext(1024, 81920, A2_TEST_DIR"/load-nonBt-v0001"));
  300. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  301. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt-v0001.aria2"),
  302. infoFile.getFilename());
  303. infoFile.load();
  304. // check the contents of objects
  305. // total length
  306. CPPUNIT_ASSERT_EQUAL((int64_t)81920, dctx->getTotalLength());
  307. // bitfield
  308. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  309. util::toHex(bitfield_->getBitfield(),
  310. bitfield_->getBitfieldLength()));
  311. // the number of in-flight pieces
  312. CPPUNIT_ASSERT_EQUAL((size_t)2,
  313. pieceStorage_->countInFlightPiece());
  314. // piece index 1
  315. std::vector<SharedHandle<Piece> > inFlightPieces;
  316. pieceStorage_->getInFlightPieces(inFlightPieces);
  317. SharedHandle<Piece> piece1 = inFlightPieces[0];
  318. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getIndex());
  319. CPPUNIT_ASSERT_EQUAL(1024, piece1->getLength());
  320. CPPUNIT_ASSERT_EQUAL((size_t)1, piece1->getBitfieldLength());
  321. CPPUNIT_ASSERT_EQUAL(std::string("00"), util::toHex(piece1->getBitfield(),
  322. piece1->getBitfieldLength()));
  323. // piece index 2
  324. SharedHandle<Piece> piece2 = inFlightPieces[1];
  325. CPPUNIT_ASSERT_EQUAL((size_t)2, piece2->getIndex());
  326. CPPUNIT_ASSERT_EQUAL(512, piece2->getLength());
  327. }
  328. void DefaultBtProgressInfoFileTest::testLoad_nonBt_pieceLengthShorter()
  329. {
  330. initializeMembers(512, 81920);
  331. option_->put(PREF_ALLOW_PIECE_LENGTH_CHANGE, A2_V_TRUE);
  332. SharedHandle<DownloadContext> dctx
  333. (new DownloadContext(512, 81920, A2_TEST_DIR"/load-nonBt-v0001"));
  334. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  335. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/load-nonBt-v0001.aria2"),
  336. infoFile.getFilename());
  337. infoFile.load();
  338. // check the contents of objects
  339. // bitfield
  340. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffffffffffffffffffffffc"),
  341. util::toHex(bitfield_->getBitfield(),
  342. bitfield_->getBitfieldLength()));
  343. // the number of in-flight pieces
  344. CPPUNIT_ASSERT_EQUAL((size_t)0,
  345. pieceStorage_->countInFlightPiece());
  346. }
  347. void DefaultBtProgressInfoFileTest::testSave_nonBt()
  348. {
  349. initializeMembers(1024, 81920);
  350. SharedHandle<DownloadContext> dctx
  351. (new DownloadContext(1024, 81920, A2_TEST_OUT_DIR"/save-temp"));
  352. bitfield_->setAllBit();
  353. bitfield_->unsetBit(79);
  354. pieceStorage_->setCompletedLength(80896);
  355. SharedHandle<Piece> p1(new Piece(1, 1024));
  356. SharedHandle<Piece> p2(new Piece(2, 512));
  357. std::vector<SharedHandle<Piece> > inFlightPieces;
  358. inFlightPieces.push_back(p1);
  359. inFlightPieces.push_back(p2);
  360. pieceStorage_->addInFlightPiece(inFlightPieces);
  361. DefaultBtProgressInfoFile infoFile(dctx, pieceStorage_, option_.get());
  362. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_OUT_DIR"/save-temp.aria2"),
  363. infoFile.getFilename());
  364. infoFile.save();
  365. // read and validate
  366. std::ifstream in(infoFile.getFilename().c_str(), std::ios::binary);
  367. //in.exceptions(ios::failbit);
  368. unsigned char version[2];
  369. in.read((char*)version, sizeof(version));
  370. CPPUNIT_ASSERT_EQUAL(std::string("0001"),
  371. util::toHex(version, sizeof(version)));
  372. unsigned char extension[4];
  373. in.read((char*)extension, sizeof(extension));
  374. CPPUNIT_ASSERT_EQUAL(std::string("00000000"),
  375. util::toHex(extension, sizeof(extension)));
  376. uint32_t infoHashLength;
  377. in.read(reinterpret_cast<char*>(&infoHashLength), sizeof(infoHashLength));
  378. infoHashLength = ntohl(infoHashLength);
  379. CPPUNIT_ASSERT_EQUAL((uint32_t)0, infoHashLength);
  380. uint32_t pieceLength;
  381. in.read((char*)&pieceLength, sizeof(pieceLength));
  382. pieceLength = ntohl(pieceLength);
  383. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength);
  384. uint64_t totalLength;
  385. in.read((char*)&totalLength, sizeof(totalLength));
  386. totalLength = ntoh64(totalLength);
  387. CPPUNIT_ASSERT_EQUAL((uint64_t)81920/* 80*1024 */, totalLength);
  388. uint64_t uploadLength;
  389. in.read((char*)&uploadLength, sizeof(uploadLength));
  390. uploadLength = ntoh64(uploadLength);
  391. CPPUNIT_ASSERT_EQUAL((uint64_t)0, uploadLength);
  392. uint32_t bitfieldLength;
  393. in.read((char*)&bitfieldLength, sizeof(bitfieldLength));
  394. bitfieldLength = ntohl(bitfieldLength);
  395. CPPUNIT_ASSERT_EQUAL((uint32_t)10, bitfieldLength);
  396. unsigned char bitfieldRead[10];
  397. in.read((char*)bitfieldRead, sizeof(bitfieldRead));
  398. CPPUNIT_ASSERT_EQUAL(std::string("fffffffffffffffffffe"),
  399. util::toHex(bitfieldRead, sizeof(bitfieldRead)));
  400. uint32_t numInFlightPiece;
  401. in.read((char*)&numInFlightPiece, sizeof(numInFlightPiece));
  402. numInFlightPiece = ntohl(numInFlightPiece);
  403. CPPUNIT_ASSERT_EQUAL((uint32_t)2, numInFlightPiece);
  404. // piece index 1
  405. uint32_t index1;
  406. in.read((char*)&index1, sizeof(index1));
  407. index1 = ntohl(index1);
  408. CPPUNIT_ASSERT_EQUAL((uint32_t)1, index1);
  409. uint32_t pieceLength1;
  410. in.read((char*)&pieceLength1, sizeof(pieceLength1));
  411. pieceLength1 = ntohl(pieceLength1);
  412. CPPUNIT_ASSERT_EQUAL((uint32_t)1024, pieceLength1);
  413. uint32_t pieceBitfieldLength1;
  414. in.read((char*)&pieceBitfieldLength1, sizeof(pieceBitfieldLength1));
  415. pieceBitfieldLength1 = ntohl(pieceBitfieldLength1);
  416. CPPUNIT_ASSERT_EQUAL((uint32_t)1, pieceBitfieldLength1);
  417. unsigned char pieceBitfield1[1];
  418. in.read((char*)pieceBitfield1, sizeof(pieceBitfield1));
  419. CPPUNIT_ASSERT_EQUAL(std::string("00"),
  420. util::toHex(pieceBitfield1, sizeof(pieceBitfield1)));
  421. // piece index 2
  422. uint32_t index2;
  423. in.read((char*)&index2, sizeof(index2));
  424. index2 = ntohl(index2);
  425. CPPUNIT_ASSERT_EQUAL((uint32_t)2, index2);
  426. uint32_t pieceLength2;
  427. in.read((char*)&pieceLength2, sizeof(pieceLength2));
  428. pieceLength2 = ntohl(pieceLength2);
  429. CPPUNIT_ASSERT_EQUAL((uint32_t)512, pieceLength2);
  430. }
  431. void DefaultBtProgressInfoFileTest::testUpdateFilename()
  432. {
  433. SharedHandle<DownloadContext> dctx
  434. (new DownloadContext(1024, 81920, A2_TEST_DIR"/file1"));
  435. DefaultBtProgressInfoFile infoFile(dctx, SharedHandle<MockPieceStorage>(), 0);
  436. #ifdef ENABLE_BITTORRENT
  437. infoFile.setBtRuntime(btRuntime_);
  438. infoFile.setPeerStorage(peerStorage_);
  439. #endif // ENABLE_BITTORRENT
  440. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.aria2"),
  441. infoFile.getFilename());
  442. dctx->getFirstFileEntry()->setPath(A2_TEST_DIR"/file1.1");
  443. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.aria2"),
  444. infoFile.getFilename());
  445. infoFile.updateFilename();
  446. CPPUNIT_ASSERT_EQUAL(std::string(A2_TEST_DIR"/file1.1.aria2"),
  447. infoFile.getFilename());
  448. }
  449. } // namespace aria2