BitfieldManTest.cc 17 KB

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