BitfieldManTest.cc 17 KB

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