BitfieldManTest.cc 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565
  1. #include "BitfieldMan.h"
  2. #include "FixedNumberRandomizer.h"
  3. #include <cppunit/extensions/HelperMacros.h>
  4. namespace aria2 {
  5. class BitfieldManTest:public CppUnit::TestFixture {
  6. CPPUNIT_TEST_SUITE(BitfieldManTest);
  7. CPPUNIT_TEST(testGetBlockSize);
  8. CPPUNIT_TEST(testGetFirstMissingUnusedIndex);
  9. CPPUNIT_TEST(testGetFirstMissingIndex);
  10. CPPUNIT_TEST(testIsAllBitSet);
  11. CPPUNIT_TEST(testFilter);
  12. CPPUNIT_TEST(testGetMissingIndex);
  13. CPPUNIT_TEST(testGetSparceMissingUnusedIndex);
  14. CPPUNIT_TEST(testGetSparceMissingUnusedIndex_setBit);
  15. CPPUNIT_TEST(testIsBitSetOffsetRange);
  16. CPPUNIT_TEST(testGetMissingUnusedLength);
  17. CPPUNIT_TEST(testSetBitRange);
  18. CPPUNIT_TEST(testGetAllMissingIndexes);
  19. CPPUNIT_TEST(testGetAllMissingIndexes_noarg);
  20. CPPUNIT_TEST(testGetMissingUnusedIndex);
  21. CPPUNIT_TEST(testGetMissingIndex_noarg);
  22. CPPUNIT_TEST(testGetMissingUnusedIndex_noarg);
  23. CPPUNIT_TEST(testCountFilteredBlock);
  24. CPPUNIT_TEST_SUITE_END();
  25. private:
  26. SharedHandle<Randomizer> fixedNumberRandomizer;
  27. public:
  28. BitfieldManTest():fixedNumberRandomizer(0) {
  29. FixedNumberRandomizer* randomizer = new FixedNumberRandomizer();
  30. randomizer->setFixedNumber(0);
  31. this->fixedNumberRandomizer = randomizer;
  32. }
  33. void setUp() {
  34. }
  35. void testGetBlockSize();
  36. void testGetFirstMissingUnusedIndex();
  37. void testGetFirstMissingIndex();
  38. void testGetMissingIndex();
  39. void testGetMissingIndex_noarg();
  40. void testGetMissingUnusedIndex();
  41. void testGetMissingUnusedIndex_noarg();
  42. void testGetAllMissingIndexes();
  43. void testGetAllMissingIndexes_noarg();
  44. void testIsAllBitSet();
  45. void testFilter();
  46. void testGetSparceMissingUnusedIndex();
  47. void testGetSparceMissingUnusedIndex_setBit();
  48. void testIsBitSetOffsetRange();
  49. void testGetMissingUnusedLength();
  50. void testSetBitRange();
  51. void testCountFilteredBlock();
  52. };
  53. CPPUNIT_TEST_SUITE_REGISTRATION( BitfieldManTest );
  54. void BitfieldManTest::testGetBlockSize() {
  55. BitfieldMan bt1(1024, 1024*10);
  56. CPPUNIT_ASSERT_EQUAL((int32_t)1024, bt1.getBlockLength(9));
  57. BitfieldMan bt2(1024, 1024*10+1);
  58. CPPUNIT_ASSERT_EQUAL((int32_t)1024, bt2.getBlockLength(9));
  59. CPPUNIT_ASSERT_EQUAL((int32_t)1, bt2.getBlockLength(10));
  60. CPPUNIT_ASSERT_EQUAL((int32_t)0, bt2.getBlockLength(11));
  61. }
  62. void BitfieldManTest::testGetFirstMissingUnusedIndex()
  63. {
  64. {
  65. BitfieldMan bt1(1024, 1024*10);
  66. CPPUNIT_ASSERT_EQUAL(0, bt1.getFirstMissingUnusedIndex());
  67. bt1.setUseBit(0);
  68. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingUnusedIndex());
  69. bt1.unsetUseBit(0);
  70. bt1.setBit(0);
  71. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingUnusedIndex());
  72. bt1.setAllBit();
  73. CPPUNIT_ASSERT_EQUAL(-1, bt1.getFirstMissingUnusedIndex());
  74. }
  75. {
  76. BitfieldMan bt1(1024, 1024*10);
  77. bt1.addFilter(1024, 1024*10);
  78. bt1.enableFilter();
  79. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingUnusedIndex());
  80. bt1.setUseBit(1);
  81. CPPUNIT_ASSERT_EQUAL(2, bt1.getFirstMissingUnusedIndex());
  82. bt1.setBit(2);
  83. CPPUNIT_ASSERT_EQUAL(3, bt1.getFirstMissingUnusedIndex());
  84. }
  85. }
  86. void BitfieldManTest::testGetFirstMissingIndex()
  87. {
  88. {
  89. BitfieldMan bt1(1024, 1024*10);
  90. CPPUNIT_ASSERT_EQUAL(0, bt1.getFirstMissingIndex());
  91. bt1.setUseBit(0);
  92. CPPUNIT_ASSERT_EQUAL(0, bt1.getFirstMissingIndex());
  93. bt1.unsetUseBit(0);
  94. bt1.setBit(0);
  95. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingIndex());
  96. bt1.setAllBit();
  97. CPPUNIT_ASSERT_EQUAL(-1, bt1.getFirstMissingIndex());
  98. }
  99. {
  100. BitfieldMan bt1(1024, 1024*10);
  101. bt1.addFilter(1024, 1024*10);
  102. bt1.enableFilter();
  103. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingIndex());
  104. bt1.setUseBit(1);
  105. CPPUNIT_ASSERT_EQUAL(1, bt1.getFirstMissingIndex());
  106. bt1.setBit(1);
  107. CPPUNIT_ASSERT_EQUAL(2, bt1.getFirstMissingIndex());
  108. }
  109. }
  110. void BitfieldManTest::testGetMissingUnusedIndex_noarg()
  111. {
  112. {
  113. BitfieldMan bt1(1024, 1024*10);
  114. bt1.setRandomizer(fixedNumberRandomizer);
  115. CPPUNIT_ASSERT_EQUAL(0, bt1.getMissingUnusedIndex());
  116. bt1.setUseBit(0);
  117. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingUnusedIndex());
  118. bt1.unsetUseBit(0);
  119. bt1.setBit(0);
  120. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingUnusedIndex());
  121. bt1.setAllBit();
  122. CPPUNIT_ASSERT_EQUAL(-1, bt1.getMissingUnusedIndex());
  123. }
  124. {
  125. BitfieldMan bt1(1024, 1024*10);
  126. bt1.setRandomizer(fixedNumberRandomizer);
  127. bt1.addFilter(1024, 1024*10);
  128. bt1.enableFilter();
  129. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingUnusedIndex());
  130. bt1.setUseBit(1);
  131. CPPUNIT_ASSERT_EQUAL(2, bt1.getMissingUnusedIndex());
  132. bt1.setBit(2);
  133. CPPUNIT_ASSERT_EQUAL(3, bt1.getMissingUnusedIndex());
  134. }
  135. }
  136. void BitfieldManTest::testGetMissingIndex_noarg()
  137. {
  138. {
  139. BitfieldMan bt1(1024, 1024*10);
  140. bt1.setRandomizer(fixedNumberRandomizer);
  141. CPPUNIT_ASSERT_EQUAL(0, bt1.getMissingIndex());
  142. bt1.setUseBit(0);
  143. CPPUNIT_ASSERT_EQUAL(0, bt1.getMissingIndex());
  144. bt1.unsetUseBit(0);
  145. bt1.setBit(0);
  146. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingIndex());
  147. bt1.setAllBit();
  148. CPPUNIT_ASSERT_EQUAL(-1, bt1.getMissingIndex());
  149. }
  150. {
  151. BitfieldMan bt1(1024, 1024*10);
  152. bt1.setRandomizer(fixedNumberRandomizer);
  153. bt1.addFilter(1024, 1024*10);
  154. bt1.enableFilter();
  155. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingIndex());
  156. bt1.setUseBit(1);
  157. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingIndex());
  158. bt1.setBit(1);
  159. CPPUNIT_ASSERT_EQUAL(2, bt1.getMissingIndex());
  160. }
  161. }
  162. void BitfieldManTest::testIsAllBitSet() {
  163. BitfieldMan bt1(1024, 1024*10);
  164. CPPUNIT_ASSERT(!bt1.isAllBitSet());
  165. bt1.setBit(1);
  166. CPPUNIT_ASSERT(!bt1.isAllBitSet());
  167. for(int i = 0; i < 8; i++) {
  168. CPPUNIT_ASSERT(bt1.setBit(i));
  169. }
  170. CPPUNIT_ASSERT(!bt1.isAllBitSet());
  171. for(int32_t i = 0; i < bt1.countBlock(); i++) {
  172. CPPUNIT_ASSERT(bt1.setBit(i));
  173. }
  174. CPPUNIT_ASSERT(bt1.isAllBitSet());
  175. BitfieldMan btzero(1024, 0);
  176. CPPUNIT_ASSERT(btzero.isAllBitSet());
  177. }
  178. void BitfieldManTest::testFilter() {
  179. BitfieldMan btman(2, 32);
  180. btman.setRandomizer(fixedNumberRandomizer);
  181. // test offset=4, length=12
  182. btman.addFilter(4, 12);
  183. btman.enableFilter();
  184. unsigned char peerBt[2];
  185. memset(peerBt, 0xff, sizeof(peerBt));
  186. int index;
  187. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  188. btman.setUseBit(index);
  189. CPPUNIT_ASSERT_EQUAL(2, index);
  190. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  191. btman.setUseBit(index);
  192. CPPUNIT_ASSERT_EQUAL(3, index);
  193. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  194. btman.setUseBit(index);
  195. CPPUNIT_ASSERT_EQUAL(4, index);
  196. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  197. btman.setUseBit(index);
  198. CPPUNIT_ASSERT_EQUAL(5, index);
  199. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  200. btman.setUseBit(index);
  201. CPPUNIT_ASSERT_EQUAL(6, index);
  202. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  203. btman.setUseBit(index);
  204. CPPUNIT_ASSERT_EQUAL(7, index);
  205. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  206. btman.setUseBit(index);
  207. CPPUNIT_ASSERT_EQUAL(-1, index);
  208. CPPUNIT_ASSERT_EQUAL((int64_t)12, btman.getFilteredTotalLength());
  209. // test offset=5, length=2
  210. btman.clearAllBit();
  211. btman.clearAllUseBit();
  212. btman.clearFilter();
  213. btman.addFilter(5, 2);
  214. btman.enableFilter();
  215. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  216. btman.setUseBit(index);
  217. btman.setBit(index);
  218. CPPUNIT_ASSERT_EQUAL(2, index);
  219. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  220. btman.setUseBit(index);
  221. btman.setBit(index);
  222. CPPUNIT_ASSERT_EQUAL(3, index);
  223. index = btman.getMissingUnusedIndex(peerBt, sizeof(peerBt));
  224. btman.setUseBit(index);
  225. CPPUNIT_ASSERT_EQUAL(-1, index);
  226. CPPUNIT_ASSERT_EQUAL((int64_t)4, btman.getFilteredTotalLength());
  227. CPPUNIT_ASSERT(btman.isFilteredAllBitSet());
  228. BitfieldMan btman2(2, 31);
  229. btman2.addFilter(0, 31);
  230. btman2.enableFilter();
  231. CPPUNIT_ASSERT_EQUAL((int64_t)31, btman2.getFilteredTotalLength());
  232. }
  233. void BitfieldManTest::testGetMissingIndex() {
  234. BitfieldMan bt1(1024, 1024*256);
  235. bt1.setRandomizer(fixedNumberRandomizer);
  236. unsigned char bitArray[] = {
  237. 0xff, 0xff, 0xff, 0xff,
  238. 0xff, 0xff, 0xff, 0xff,
  239. 0xff, 0xff, 0xff, 0xff,
  240. 0xff, 0xff, 0xff, 0xff,
  241. 0xff, 0xff, 0xff, 0xff,
  242. 0xff, 0xff, 0xff, 0xff,
  243. 0xff, 0xff, 0xff, 0xff,
  244. 0xff, 0xff, 0xff, 0xff,
  245. };
  246. CPPUNIT_ASSERT_EQUAL((int32_t)0, bt1.getMissingIndex(bitArray, 32));
  247. bt1.addFilter(1024, 1024*256);
  248. bt1.enableFilter();
  249. CPPUNIT_ASSERT_EQUAL((int32_t)1, bt1.getMissingIndex(bitArray, 32));
  250. bt1.disableFilter();
  251. unsigned char bitArray2[] = {
  252. 0x0f, 0xff, 0xff, 0xff,
  253. 0xff, 0xff, 0xff, 0xff,
  254. 0xff, 0xff, 0xff, 0xff,
  255. 0xff, 0xff, 0xff, 0xff,
  256. 0xff, 0xff, 0xff, 0xff,
  257. 0xff, 0xff, 0xff, 0xff,
  258. 0xff, 0xff, 0xff, 0xff,
  259. 0xff, 0xff, 0xff, 0xff,
  260. };
  261. CPPUNIT_ASSERT_EQUAL((int32_t)4, bt1.getMissingIndex(bitArray2, 32));
  262. unsigned char bitArray3[] = {
  263. 0x00, 0xff, 0xff, 0xff,
  264. 0xff, 0xff, 0xff, 0xff,
  265. 0xff, 0xff, 0xff, 0xff,
  266. 0xff, 0xff, 0xff, 0xff,
  267. 0xff, 0xff, 0xff, 0xff,
  268. 0xff, 0xff, 0xff, 0xff,
  269. 0xff, 0xff, 0xff, 0xff,
  270. 0xff, 0xff, 0xff, 0xff,
  271. };
  272. CPPUNIT_ASSERT_EQUAL((int32_t)8, bt1.getMissingIndex(bitArray3, 32));
  273. unsigned char bitArray4[] = {
  274. 0x00, 0x00, 0x00, 0x00,
  275. 0x00, 0x00, 0x00, 0x00,
  276. 0x00, 0x00, 0x00, 0x00,
  277. 0x00, 0x00, 0x00, 0x00,
  278. 0x00, 0x00, 0x00, 0x00,
  279. 0x00, 0x00, 0x00, 0x00,
  280. 0x00, 0x00, 0x00, 0x00,
  281. 0x00, 0x00, 0x00, 0x00,
  282. };
  283. CPPUNIT_ASSERT_EQUAL((int32_t)-1, bt1.getMissingIndex(bitArray4, 32));
  284. }
  285. void BitfieldManTest::testGetSparceMissingUnusedIndex() {
  286. BitfieldMan bitfield(1024*1024, 10*1024*1024);
  287. CPPUNIT_ASSERT_EQUAL((int32_t)0, bitfield.getSparseMissingUnusedIndex());
  288. bitfield.setUseBit(0);
  289. CPPUNIT_ASSERT_EQUAL((int32_t)5, bitfield.getSparseMissingUnusedIndex());
  290. bitfield.setUseBit(5);
  291. CPPUNIT_ASSERT_EQUAL((int32_t)3, bitfield.getSparseMissingUnusedIndex());
  292. bitfield.setUseBit(3);
  293. CPPUNIT_ASSERT_EQUAL((int32_t)8, bitfield.getSparseMissingUnusedIndex());
  294. bitfield.setUseBit(8);
  295. CPPUNIT_ASSERT_EQUAL((int32_t)2, bitfield.getSparseMissingUnusedIndex());
  296. bitfield.setUseBit(2);
  297. CPPUNIT_ASSERT_EQUAL((int32_t)7, bitfield.getSparseMissingUnusedIndex());
  298. bitfield.setUseBit(7);
  299. CPPUNIT_ASSERT_EQUAL((int32_t)1, bitfield.getSparseMissingUnusedIndex());
  300. bitfield.setUseBit(1);
  301. CPPUNIT_ASSERT_EQUAL((int32_t)4, bitfield.getSparseMissingUnusedIndex());
  302. bitfield.setUseBit(4);
  303. CPPUNIT_ASSERT_EQUAL((int32_t)6, bitfield.getSparseMissingUnusedIndex());
  304. bitfield.setUseBit(6);
  305. CPPUNIT_ASSERT_EQUAL((int32_t)9, bitfield.getSparseMissingUnusedIndex());
  306. bitfield.setUseBit(9);
  307. CPPUNIT_ASSERT_EQUAL((int32_t)-1, bitfield.getSparseMissingUnusedIndex());
  308. }
  309. void BitfieldManTest::testGetSparceMissingUnusedIndex_setBit() {
  310. BitfieldMan bitfield(1024*1024, 10*1024*1024);
  311. CPPUNIT_ASSERT_EQUAL((int32_t)0, bitfield.getSparseMissingUnusedIndex());
  312. bitfield.setBit(0);
  313. CPPUNIT_ASSERT_EQUAL((int32_t)1, bitfield.getSparseMissingUnusedIndex());
  314. bitfield.setBit(1);
  315. CPPUNIT_ASSERT_EQUAL((int32_t)2, bitfield.getSparseMissingUnusedIndex());
  316. bitfield.setBit(2);
  317. CPPUNIT_ASSERT_EQUAL((int32_t)3, bitfield.getSparseMissingUnusedIndex());
  318. bitfield.setBit(3);
  319. CPPUNIT_ASSERT_EQUAL((int32_t)4, bitfield.getSparseMissingUnusedIndex());
  320. bitfield.setBit(4);
  321. CPPUNIT_ASSERT_EQUAL((int32_t)5, bitfield.getSparseMissingUnusedIndex());
  322. bitfield.setBit(5);
  323. CPPUNIT_ASSERT_EQUAL((int32_t)6, bitfield.getSparseMissingUnusedIndex());
  324. bitfield.setBit(6);
  325. CPPUNIT_ASSERT_EQUAL((int32_t)7, bitfield.getSparseMissingUnusedIndex());
  326. bitfield.setBit(7);
  327. CPPUNIT_ASSERT_EQUAL((int32_t)8, bitfield.getSparseMissingUnusedIndex());
  328. bitfield.setBit(8);
  329. CPPUNIT_ASSERT_EQUAL((int32_t)9, bitfield.getSparseMissingUnusedIndex());
  330. bitfield.setBit(9);
  331. CPPUNIT_ASSERT_EQUAL((int32_t)-1, bitfield.getSparseMissingUnusedIndex());
  332. }
  333. void BitfieldManTest::testIsBitSetOffsetRange()
  334. {
  335. int64_t totalLength = (int64_t)4*1024*1024*1024;
  336. int32_t pieceLength = 4*1024*1024;
  337. BitfieldMan bitfield(pieceLength, totalLength);
  338. bitfield.setAllBit();
  339. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(0, 0));
  340. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(0, -1));
  341. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(totalLength, 100));
  342. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(totalLength+1, 100));
  343. CPPUNIT_ASSERT(bitfield.isBitSetOffsetRange(0, totalLength));
  344. CPPUNIT_ASSERT(bitfield.isBitSetOffsetRange(0, totalLength+1));
  345. bitfield.clearAllBit();
  346. bitfield.setBit(100);
  347. bitfield.setBit(101);
  348. CPPUNIT_ASSERT(bitfield.isBitSetOffsetRange(pieceLength*100, pieceLength*2));
  349. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(pieceLength*100-10, pieceLength*2));
  350. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(pieceLength*100, pieceLength*2+1));
  351. bitfield.clearAllBit();
  352. bitfield.setBit(100);
  353. bitfield.setBit(102);
  354. CPPUNIT_ASSERT(!bitfield.isBitSetOffsetRange(pieceLength*100, pieceLength*3));
  355. }
  356. void BitfieldManTest::testGetMissingUnusedLength()
  357. {
  358. int64_t totalLength = 1024*10+10;
  359. int32_t blockLength = 1024;
  360. BitfieldMan bf(blockLength, totalLength);
  361. // from index 0 and all blocks are unused and not acquired.
  362. CPPUNIT_ASSERT_EQUAL((int64_t)totalLength, bf.getMissingUnusedLength(0));
  363. // from index 10 and all blocks are unused and not acquired.
  364. CPPUNIT_ASSERT_EQUAL((int64_t)10, bf.getMissingUnusedLength(10));
  365. // from index -1
  366. CPPUNIT_ASSERT_EQUAL((int64_t)0, bf.getMissingUnusedLength(-1));
  367. // from index 11
  368. CPPUNIT_ASSERT_EQUAL((int64_t)0, bf.getMissingUnusedLength(11));
  369. // from index 12
  370. CPPUNIT_ASSERT_EQUAL((int64_t)0, bf.getMissingUnusedLength(12));
  371. // from index 0 and 5th block is used.
  372. bf.setUseBit(5);
  373. CPPUNIT_ASSERT_EQUAL((int64_t)5*blockLength, bf.getMissingUnusedLength(0));
  374. // from index 0 and 4th block is acquired.
  375. bf.setBit(4);
  376. CPPUNIT_ASSERT_EQUAL((int64_t)4*blockLength, bf.getMissingUnusedLength(0));
  377. // from index 1
  378. CPPUNIT_ASSERT_EQUAL((int64_t)3*blockLength, bf.getMissingUnusedLength(1));
  379. }
  380. void BitfieldManTest::testSetBitRange()
  381. {
  382. int32_t blockLength = 1024*1024;
  383. int64_t totalLength = 10*blockLength;
  384. BitfieldMan bf(blockLength, totalLength);
  385. bf.setBitRange(0, 4);
  386. for(int32_t i = 0; i < 5; ++i) {
  387. CPPUNIT_ASSERT(bf.isBitSet(i));
  388. }
  389. for(int32_t i = 5; i < 10; ++i) {
  390. CPPUNIT_ASSERT(!bf.isBitSet(i));
  391. }
  392. CPPUNIT_ASSERT_EQUAL(int64_t(5*blockLength), bf.getCompletedLength());
  393. }
  394. void BitfieldManTest::testGetAllMissingIndexes_noarg()
  395. {
  396. int32_t blockLength = 16*1024;
  397. int64_t totalLength = 1024*1024;
  398. BitfieldMan bf(blockLength, totalLength);
  399. CPPUNIT_ASSERT_EQUAL((size_t)64, bf.getAllMissingIndexes().size());
  400. for(int32_t i = 0; i < 63; ++i) {
  401. bf.setBit(i);
  402. }
  403. CPPUNIT_ASSERT_EQUAL((size_t)1, bf.getAllMissingIndexes().size());
  404. CPPUNIT_ASSERT_EQUAL(63, bf.getAllMissingIndexes().front());
  405. }
  406. void BitfieldManTest::testGetAllMissingIndexes()
  407. {
  408. int32_t blockLength = 16*1024;
  409. int64_t totalLength = 1024*1024;
  410. BitfieldMan bf(blockLength, totalLength);
  411. BitfieldMan peerBf(blockLength, totalLength);
  412. peerBf.setAllBit();
  413. CPPUNIT_ASSERT_EQUAL((size_t)64, bf.getAllMissingIndexes(peerBf.getBitfield(),
  414. peerBf.getBitfieldLength()).size());
  415. for(int32_t i = 0; i < 62; ++i) {
  416. bf.setBit(i);
  417. }
  418. peerBf.unsetBit(62);
  419. {
  420. std::deque<int32_t> indexes = bf.getAllMissingIndexes(peerBf.getBitfield(),
  421. peerBf.getBitfieldLength());
  422. CPPUNIT_ASSERT_EQUAL((size_t)1, indexes.size());
  423. CPPUNIT_ASSERT_EQUAL(63, indexes.front());
  424. }
  425. }
  426. void BitfieldManTest::testGetMissingUnusedIndex()
  427. {
  428. BitfieldMan bt1(1024, 1024*256);
  429. bt1.setRandomizer(fixedNumberRandomizer);
  430. unsigned char bitArray[] = {
  431. 0xff, 0xff, 0xff, 0xff,
  432. 0xff, 0xff, 0xff, 0xff,
  433. 0xff, 0xff, 0xff, 0xff,
  434. 0xff, 0xff, 0xff, 0xff,
  435. 0xff, 0xff, 0xff, 0xff,
  436. 0xff, 0xff, 0xff, 0xff,
  437. 0xff, 0xff, 0xff, 0xff,
  438. 0xff, 0xff, 0xff, 0xff,
  439. };
  440. CPPUNIT_ASSERT_EQUAL(0, bt1.getMissingUnusedIndex(bitArray, 32));
  441. bt1.addFilter(1024, 1024*256);
  442. bt1.enableFilter();
  443. CPPUNIT_ASSERT_EQUAL(1, bt1.getMissingUnusedIndex(bitArray, 32));
  444. bt1.setUseBit(1);
  445. CPPUNIT_ASSERT_EQUAL(2, bt1.getMissingUnusedIndex(bitArray, 32));
  446. bt1.disableFilter();
  447. bt1.setBit(0);
  448. CPPUNIT_ASSERT_EQUAL(2, bt1.getMissingUnusedIndex(bitArray, 32));
  449. bt1.setAllBit();
  450. CPPUNIT_ASSERT_EQUAL(-1, bt1.getMissingUnusedIndex(bitArray, 32));
  451. bt1.clearAllBit();
  452. bt1.setAllUseBit();
  453. CPPUNIT_ASSERT_EQUAL(-1, bt1.getMissingUnusedIndex(bitArray, 32));
  454. unsigned char bitArray4[] = {
  455. 0x00, 0x00, 0x00, 0x00,
  456. 0x00, 0x00, 0x00, 0x00,
  457. 0x00, 0x00, 0x00, 0x00,
  458. 0x00, 0x00, 0x00, 0x00,
  459. 0x00, 0x00, 0x00, 0x00,
  460. 0x00, 0x00, 0x00, 0x00,
  461. 0x00, 0x00, 0x00, 0x00,
  462. 0x00, 0x00, 0x00, 0x00,
  463. };
  464. CPPUNIT_ASSERT_EQUAL(-1, bt1.getMissingUnusedIndex(bitArray4, 32));
  465. }
  466. void BitfieldManTest::testCountFilteredBlock()
  467. {
  468. BitfieldMan bt(1024, 1024*256);
  469. CPPUNIT_ASSERT_EQUAL(256, bt.countBlock());
  470. CPPUNIT_ASSERT_EQUAL(0, bt.countFilteredBlock());
  471. bt.addFilter(1024, 1024*256);
  472. bt.enableFilter();
  473. CPPUNIT_ASSERT_EQUAL(256, bt.countBlock());
  474. CPPUNIT_ASSERT_EQUAL(255, bt.countFilteredBlock());
  475. bt.disableFilter();
  476. CPPUNIT_ASSERT_EQUAL(256, bt.countBlock());
  477. CPPUNIT_ASSERT_EQUAL(0, bt.countFilteredBlock());
  478. }
  479. } // namespace aria2