sequence_test.go 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125
  1. package bitmap
  2. import (
  3. "math/rand"
  4. "testing"
  5. "time"
  6. )
  7. func TestSequenceGetAvailableBit(t *testing.T) {
  8. input := []struct {
  9. head *sequence
  10. from uint64
  11. bytePos uint64
  12. bitPos uint64
  13. }{
  14. {&sequence{block: 0x0, count: 0}, 0, invalidPos, invalidPos},
  15. {&sequence{block: 0x0, count: 1}, 0, 0, 0},
  16. {&sequence{block: 0x0, count: 100}, 0, 0, 0},
  17. {&sequence{block: 0x80000000, count: 0}, 0, invalidPos, invalidPos},
  18. {&sequence{block: 0x80000000, count: 1}, 0, 0, 1},
  19. {&sequence{block: 0x80000000, count: 100}, 0, 0, 1},
  20. {&sequence{block: 0xFF000000, count: 0}, 0, invalidPos, invalidPos},
  21. {&sequence{block: 0xFF000000, count: 1}, 0, 1, 0},
  22. {&sequence{block: 0xFF000000, count: 100}, 0, 1, 0},
  23. {&sequence{block: 0xFF800000, count: 0}, 0, invalidPos, invalidPos},
  24. {&sequence{block: 0xFF800000, count: 1}, 0, 1, 1},
  25. {&sequence{block: 0xFF800000, count: 100}, 0, 1, 1},
  26. {&sequence{block: 0xFFC0FF00, count: 0}, 0, invalidPos, invalidPos},
  27. {&sequence{block: 0xFFC0FF00, count: 1}, 0, 1, 2},
  28. {&sequence{block: 0xFFC0FF00, count: 100}, 0, 1, 2},
  29. {&sequence{block: 0xFFE0FF00, count: 0}, 0, invalidPos, invalidPos},
  30. {&sequence{block: 0xFFE0FF00, count: 1}, 0, 1, 3},
  31. {&sequence{block: 0xFFE0FF00, count: 100}, 0, 1, 3},
  32. {&sequence{block: 0xFFFEFF00, count: 0}, 0, invalidPos, invalidPos},
  33. {&sequence{block: 0xFFFEFF00, count: 1}, 0, 1, 7},
  34. {&sequence{block: 0xFFFEFF00, count: 100}, 0, 1, 7},
  35. {&sequence{block: 0xFFFFC0FF, count: 0}, 0, invalidPos, invalidPos},
  36. {&sequence{block: 0xFFFFC0FF, count: 1}, 0, 2, 2},
  37. {&sequence{block: 0xFFFFC0FF, count: 100}, 0, 2, 2},
  38. {&sequence{block: 0xFFFFFF00, count: 0}, 0, invalidPos, invalidPos},
  39. {&sequence{block: 0xFFFFFF00, count: 1}, 0, 3, 0},
  40. {&sequence{block: 0xFFFFFF00, count: 100}, 0, 3, 0},
  41. {&sequence{block: 0xFFFFFFFE, count: 0}, 0, invalidPos, invalidPos},
  42. {&sequence{block: 0xFFFFFFFE, count: 1}, 0, 3, 7},
  43. {&sequence{block: 0xFFFFFFFE, count: 100}, 0, 3, 7},
  44. {&sequence{block: 0xFFFFFFFF, count: 0}, 0, invalidPos, invalidPos},
  45. {&sequence{block: 0xFFFFFFFF, count: 1}, 0, invalidPos, invalidPos},
  46. {&sequence{block: 0xFFFFFFFF, count: 100}, 0, invalidPos, invalidPos},
  47. // now test with offset
  48. {&sequence{block: 0x0, count: 0}, 0, invalidPos, invalidPos},
  49. {&sequence{block: 0x0, count: 0}, 31, invalidPos, invalidPos},
  50. {&sequence{block: 0x0, count: 0}, 32, invalidPos, invalidPos},
  51. {&sequence{block: 0x0, count: 1}, 0, 0, 0},
  52. {&sequence{block: 0x0, count: 1}, 1, 0, 1},
  53. {&sequence{block: 0x0, count: 1}, 31, 3, 7},
  54. {&sequence{block: 0xF0FF0000, count: 1}, 0, 0, 4},
  55. {&sequence{block: 0xF0FF0000, count: 1}, 8, 2, 0},
  56. {&sequence{block: 0xFFFFFFFF, count: 1}, 0, invalidPos, invalidPos},
  57. {&sequence{block: 0xFFFFFFFF, count: 1}, 16, invalidPos, invalidPos},
  58. {&sequence{block: 0xFFFFFFFF, count: 1}, 31, invalidPos, invalidPos},
  59. {&sequence{block: 0xFFFFFFFE, count: 1}, 0, 3, 7},
  60. {&sequence{block: 0xFFFFFFFF, count: 2}, 0, invalidPos, invalidPos},
  61. {&sequence{block: 0xFFFFFFFF, count: 2}, 32, invalidPos, invalidPos},
  62. }
  63. for n, i := range input {
  64. b, bb, err := i.head.getAvailableBit(i.from)
  65. if b != i.bytePos || bb != i.bitPos {
  66. t.Fatalf("Error in sequence.getAvailableBit(%d) (%d).\nExp: (%d, %d)\nGot: (%d, %d), err: %v", i.from, n, i.bytePos, i.bitPos, b, bb, err)
  67. }
  68. }
  69. }
  70. func TestSequenceEqual(t *testing.T) {
  71. input := []struct {
  72. first *sequence
  73. second *sequence
  74. areEqual bool
  75. }{
  76. {&sequence{block: 0x0, count: 8, next: nil}, &sequence{block: 0x0, count: 8}, true},
  77. {&sequence{block: 0x0, count: 0, next: nil}, &sequence{block: 0x0, count: 0}, true},
  78. {&sequence{block: 0x0, count: 2, next: nil}, &sequence{block: 0x0, count: 1, next: &sequence{block: 0x0, count: 1}}, false},
  79. {&sequence{block: 0x0, count: 2, next: &sequence{block: 0x1, count: 1}}, &sequence{block: 0x0, count: 2}, false},
  80. {&sequence{block: 0x12345678, count: 8, next: nil}, &sequence{block: 0x12345678, count: 8}, true},
  81. {&sequence{block: 0x12345678, count: 8, next: nil}, &sequence{block: 0x12345678, count: 9}, false},
  82. {&sequence{block: 0x12345678, count: 1, next: &sequence{block: 0xFFFFFFFF, count: 1}}, &sequence{block: 0x12345678, count: 1}, false},
  83. {&sequence{block: 0x12345678, count: 1}, &sequence{block: 0x12345678, count: 1, next: &sequence{block: 0xFFFFFFFF, count: 1}}, false},
  84. }
  85. for n, i := range input {
  86. if i.areEqual != i.first.equal(i.second) {
  87. t.Fatalf("Error in sequence.equal() (%d).\nExp: %t\nGot: %t,", n, i.areEqual, !i.areEqual)
  88. }
  89. }
  90. }
  91. func TestSequenceCopy(t *testing.T) {
  92. s := getTestSequence()
  93. n := s.getCopy()
  94. if !s.equal(n) {
  95. t.Fatal("copy of s failed")
  96. }
  97. if n == s {
  98. t.Fatal("not true copy of s")
  99. }
  100. }
  101. func TestGetFirstAvailable(t *testing.T) {
  102. input := []struct {
  103. mask *sequence
  104. bytePos uint64
  105. bitPos uint64
  106. start uint64
  107. }{
  108. {&sequence{block: 0xffffffff, count: 2048}, invalidPos, invalidPos, 0},
  109. {&sequence{block: 0x0, count: 8}, 0, 0, 0},
  110. {&sequence{block: 0x80000000, count: 8}, 0, 1, 0},
  111. {&sequence{block: 0xC0000000, count: 8}, 0, 2, 0},
  112. {&sequence{block: 0xE0000000, count: 8}, 0, 3, 0},
  113. {&sequence{block: 0xF0000000, count: 8}, 0, 4, 0},
  114. {&sequence{block: 0xF8000000, count: 8}, 0, 5, 0},
  115. {&sequence{block: 0xFC000000, count: 8}, 0, 6, 0},
  116. {&sequence{block: 0xFE000000, count: 8}, 0, 7, 0},
  117. {&sequence{block: 0xFE000000, count: 8}, 3, 0, 24},
  118. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x00000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 0, 0},
  119. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 1, 0},
  120. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 2, 0},
  121. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xE0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 3, 0},
  122. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 4, 0},
  123. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF8000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 5, 0},
  124. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFC000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 6, 0},
  125. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFE000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 7, 0},
  126. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x0E000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 0, 16},
  127. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 0, 0},
  128. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF800000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 1, 0},
  129. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFC00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 2, 0},
  130. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFE00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 3, 0},
  131. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 4, 0},
  132. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF80000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 5, 0},
  133. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFC0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 6, 0},
  134. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFE0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 7, 0},
  135. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 7, 7, 0},
  136. {&sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0x0, count: 6}}, 8, 0, 0},
  137. {&sequence{block: 0xfffcffff, count: 1, next: &sequence{block: 0x0, count: 6}}, 4, 0, 16},
  138. {&sequence{block: 0xfffcffff, count: 1, next: &sequence{block: 0x0, count: 6}}, 1, 7, 15},
  139. {&sequence{block: 0xfffcffff, count: 1, next: &sequence{block: 0x0, count: 6}}, 1, 6, 10},
  140. {&sequence{block: 0xfffcfffe, count: 1, next: &sequence{block: 0x0, count: 6}}, 3, 7, 31},
  141. {&sequence{block: 0xfffcffff, count: 1, next: &sequence{block: 0xffffffff, count: 6}}, invalidPos, invalidPos, 31},
  142. }
  143. for n, i := range input {
  144. bytePos, bitPos, _ := getFirstAvailable(i.mask, i.start)
  145. if bytePos != i.bytePos || bitPos != i.bitPos {
  146. t.Fatalf("Error in (%d) getFirstAvailable(). Expected (%d, %d). Got (%d, %d)", n, i.bytePos, i.bitPos, bytePos, bitPos)
  147. }
  148. }
  149. }
  150. func TestFindSequence(t *testing.T) {
  151. input := []struct {
  152. head *sequence
  153. bytePos uint64
  154. precBlocks uint64
  155. inBlockBytePos uint64
  156. }{
  157. {&sequence{block: 0xffffffff, count: 0}, 0, 0, invalidPos},
  158. {&sequence{block: 0xffffffff, count: 0}, 31, 0, invalidPos},
  159. {&sequence{block: 0xffffffff, count: 0}, 100, 0, invalidPos},
  160. {&sequence{block: 0x0, count: 1}, 0, 0, 0},
  161. {&sequence{block: 0x0, count: 1}, 1, 0, 1},
  162. {&sequence{block: 0x0, count: 1}, 31, 0, invalidPos},
  163. {&sequence{block: 0x0, count: 1}, 60, 0, invalidPos},
  164. {&sequence{block: 0xffffffff, count: 10}, 0, 0, 0},
  165. {&sequence{block: 0xffffffff, count: 10}, 3, 0, 3},
  166. {&sequence{block: 0xffffffff, count: 10}, 4, 1, 0},
  167. {&sequence{block: 0xffffffff, count: 10}, 7, 1, 3},
  168. {&sequence{block: 0xffffffff, count: 10}, 8, 2, 0},
  169. {&sequence{block: 0xffffffff, count: 10}, 39, 9, 3},
  170. {&sequence{block: 0xffffffff, count: 10, next: &sequence{block: 0xcc000000, count: 10}}, 79, 9, 3},
  171. {&sequence{block: 0xffffffff, count: 10, next: &sequence{block: 0xcc000000, count: 10}}, 80, 0, invalidPos},
  172. }
  173. for n, i := range input {
  174. _, _, precBlocks, inBlockBytePos := findSequence(i.head, i.bytePos)
  175. if precBlocks != i.precBlocks || inBlockBytePos != i.inBlockBytePos {
  176. t.Fatalf("Error in (%d) findSequence(). Expected (%d, %d). Got (%d, %d)", n, i.precBlocks, i.inBlockBytePos, precBlocks, inBlockBytePos)
  177. }
  178. }
  179. }
  180. func TestCheckIfAvailable(t *testing.T) {
  181. input := []struct {
  182. head *sequence
  183. ordinal uint64
  184. bytePos uint64
  185. bitPos uint64
  186. }{
  187. {&sequence{block: 0xffffffff, count: 0}, 0, invalidPos, invalidPos},
  188. {&sequence{block: 0xffffffff, count: 0}, 31, invalidPos, invalidPos},
  189. {&sequence{block: 0xffffffff, count: 0}, 100, invalidPos, invalidPos},
  190. {&sequence{block: 0x0, count: 1}, 0, 0, 0},
  191. {&sequence{block: 0x0, count: 1}, 1, 0, 1},
  192. {&sequence{block: 0x0, count: 1}, 31, 3, 7},
  193. {&sequence{block: 0x0, count: 1}, 60, invalidPos, invalidPos},
  194. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x800000ff, count: 1}}, 31, invalidPos, invalidPos},
  195. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x800000ff, count: 1}}, 32, invalidPos, invalidPos},
  196. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x800000ff, count: 1}}, 33, 4, 1},
  197. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1}}, 33, invalidPos, invalidPos},
  198. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1}}, 34, 4, 2},
  199. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 55, 6, 7},
  200. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 56, invalidPos, invalidPos},
  201. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 63, invalidPos, invalidPos},
  202. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 64, 8, 0},
  203. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 95, 11, 7},
  204. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC00000ff, count: 1, next: &sequence{block: 0x0, count: 1}}}, 96, invalidPos, invalidPos},
  205. }
  206. for n, i := range input {
  207. bytePos, bitPos, err := checkIfAvailable(i.head, i.ordinal)
  208. if bytePos != i.bytePos || bitPos != i.bitPos {
  209. t.Fatalf("Error in (%d) checkIfAvailable(ord:%d). Expected (%d, %d). Got (%d, %d). err: %v", n, i.ordinal, i.bytePos, i.bitPos, bytePos, bitPos, err)
  210. }
  211. }
  212. }
  213. func TestMergeSequences(t *testing.T) {
  214. input := []struct {
  215. original *sequence
  216. merged *sequence
  217. }{
  218. {&sequence{block: 0xFE000000, count: 8, next: &sequence{block: 0xFE000000, count: 2}}, &sequence{block: 0xFE000000, count: 10}},
  219. {&sequence{block: 0xFFFFFFFF, count: 8, next: &sequence{block: 0xFFFFFFFF, count: 1}}, &sequence{block: 0xFFFFFFFF, count: 9}},
  220. {&sequence{block: 0xFFFFFFFF, count: 1, next: &sequence{block: 0xFFFFFFFF, count: 8}}, &sequence{block: 0xFFFFFFFF, count: 9}},
  221. {&sequence{block: 0xFFFFFFF0, count: 8, next: &sequence{block: 0xFFFFFFF0, count: 1}}, &sequence{block: 0xFFFFFFF0, count: 9}},
  222. {&sequence{block: 0xFFFFFFF0, count: 1, next: &sequence{block: 0xFFFFFFF0, count: 8}}, &sequence{block: 0xFFFFFFF0, count: 9}},
  223. {&sequence{block: 0xFE, count: 8, next: &sequence{block: 0xFE, count: 1, next: &sequence{block: 0xFE, count: 5}}}, &sequence{block: 0xFE, count: 14}},
  224. {&sequence{block: 0xFE, count: 8, next: &sequence{block: 0xFE, count: 1, next: &sequence{block: 0xFE, count: 5, next: &sequence{block: 0xFF, count: 1}}}},
  225. &sequence{block: 0xFE, count: 14, next: &sequence{block: 0xFF, count: 1}}},
  226. // No merge
  227. {&sequence{block: 0xFE, count: 8, next: &sequence{block: 0xF8, count: 1, next: &sequence{block: 0xFE, count: 5}}},
  228. &sequence{block: 0xFE, count: 8, next: &sequence{block: 0xF8, count: 1, next: &sequence{block: 0xFE, count: 5}}}},
  229. // No merge from head: // Merge function tries to merge from passed head. If it can't merge with next, it does not reattempt with next as head
  230. {&sequence{block: 0xFE, count: 8, next: &sequence{block: 0xFF, count: 1, next: &sequence{block: 0xFF, count: 5}}},
  231. &sequence{block: 0xFE, count: 8, next: &sequence{block: 0xFF, count: 6}}},
  232. }
  233. for n, i := range input {
  234. mergeSequences(i.original)
  235. for !i.merged.equal(i.original) {
  236. t.Fatalf("Error in (%d) mergeSequences().\nExp: %s\nGot: %s,", n, i.merged.toString(), i.original.toString())
  237. }
  238. }
  239. }
  240. func TestPushReservation(t *testing.T) {
  241. input := []struct {
  242. mask *sequence
  243. bytePos uint64
  244. bitPos uint64
  245. newMask *sequence
  246. }{
  247. // Create first sequence and fill in 8 addresses starting from address 0
  248. {&sequence{block: 0x0, count: 8, next: nil}, 0, 0, &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 7, next: nil}}},
  249. {&sequence{block: 0x80000000, count: 8}, 0, 1, &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0x80000000, count: 7, next: nil}}},
  250. {&sequence{block: 0xC0000000, count: 8}, 0, 2, &sequence{block: 0xE0000000, count: 1, next: &sequence{block: 0xC0000000, count: 7, next: nil}}},
  251. {&sequence{block: 0xE0000000, count: 8}, 0, 3, &sequence{block: 0xF0000000, count: 1, next: &sequence{block: 0xE0000000, count: 7, next: nil}}},
  252. {&sequence{block: 0xF0000000, count: 8}, 0, 4, &sequence{block: 0xF8000000, count: 1, next: &sequence{block: 0xF0000000, count: 7, next: nil}}},
  253. {&sequence{block: 0xF8000000, count: 8}, 0, 5, &sequence{block: 0xFC000000, count: 1, next: &sequence{block: 0xF8000000, count: 7, next: nil}}},
  254. {&sequence{block: 0xFC000000, count: 8}, 0, 6, &sequence{block: 0xFE000000, count: 1, next: &sequence{block: 0xFC000000, count: 7, next: nil}}},
  255. {&sequence{block: 0xFE000000, count: 8}, 0, 7, &sequence{block: 0xFF000000, count: 1, next: &sequence{block: 0xFE000000, count: 7, next: nil}}},
  256. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 7}}, 0, 1, &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0x0, count: 7, next: nil}}},
  257. // Create second sequence and fill in 8 addresses starting from address 32
  258. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x00000000, count: 1, next: &sequence{block: 0xffffffff, count: 6, next: nil}}}, 4, 0,
  259. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  260. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 1,
  261. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  262. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 2,
  263. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xE0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  264. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xE0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 3,
  265. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  266. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF0000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 4,
  267. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF8000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  268. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xF8000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 5,
  269. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFC000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  270. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFC000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 6,
  271. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFE000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  272. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFE000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 4, 7,
  273. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  274. // fill in 8 addresses starting from address 40
  275. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF000000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 0,
  276. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF800000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  277. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFF800000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 1,
  278. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFC00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  279. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFC00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 2,
  280. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFE00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  281. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFE00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 3,
  282. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  283. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF00000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 4,
  284. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF80000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  285. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFF80000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 5,
  286. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFC0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  287. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFC0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 6,
  288. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFE0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  289. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFE0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}, 5, 7,
  290. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xFFFF0000, count: 1, next: &sequence{block: 0xffffffff, count: 6}}}},
  291. // Insert new sequence
  292. {&sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0x0, count: 6}}, 8, 0,
  293. &sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5}}}},
  294. {&sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5}}}, 8, 1,
  295. &sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0xC0000000, count: 1, next: &sequence{block: 0x0, count: 5}}}},
  296. // Merge affected with next
  297. {&sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 2, next: &sequence{block: 0xffffffff, count: 1}}}, 31, 7,
  298. &sequence{block: 0xffffffff, count: 8, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 1}}}},
  299. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xfffffffc, count: 1, next: &sequence{block: 0xfffffffe, count: 6}}}, 7, 6,
  300. &sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xfffffffe, count: 7}}},
  301. // Merge affected with next and next.next
  302. {&sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 1}}}, 31, 7,
  303. &sequence{block: 0xffffffff, count: 9}},
  304. {&sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 1}}, 31, 7,
  305. &sequence{block: 0xffffffff, count: 8}},
  306. // Merge affected with previous and next
  307. {&sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 1}}}, 31, 7,
  308. &sequence{block: 0xffffffff, count: 9}},
  309. // Redundant push: No change
  310. {&sequence{block: 0xffff0000, count: 1}, 0, 0, &sequence{block: 0xffff0000, count: 1}},
  311. {&sequence{block: 0xffff0000, count: 7}, 25, 7, &sequence{block: 0xffff0000, count: 7}},
  312. {&sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 1}}}, 7, 7,
  313. &sequence{block: 0xffffffff, count: 7, next: &sequence{block: 0xfffffffe, count: 1, next: &sequence{block: 0xffffffff, count: 1}}}},
  314. // Set last bit
  315. {&sequence{block: 0x0, count: 8}, 31, 7, &sequence{block: 0x0, count: 7, next: &sequence{block: 0x1, count: 1}}},
  316. // Set bit in a middle sequence in the first block, first bit
  317. {&sequence{block: 0x40000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 4, 0,
  318. &sequence{block: 0x40000000, count: 1, next: &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5,
  319. next: &sequence{block: 0x1, count: 1}}}}},
  320. // Set bit in a middle sequence in the first block, first bit (merge involved)
  321. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 4, 0,
  322. &sequence{block: 0x80000000, count: 2, next: &sequence{block: 0x0, count: 5, next: &sequence{block: 0x1, count: 1}}}},
  323. // Set bit in a middle sequence in the first block, last bit
  324. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 4, 31,
  325. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x1, count: 1, next: &sequence{block: 0x0, count: 5,
  326. next: &sequence{block: 0x1, count: 1}}}}},
  327. // Set bit in a middle sequence in the first block, middle bit
  328. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 4, 16,
  329. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x8000, count: 1, next: &sequence{block: 0x0, count: 5,
  330. next: &sequence{block: 0x1, count: 1}}}}},
  331. // Set bit in a middle sequence in a middle block, first bit
  332. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 16, 0,
  333. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 3, next: &sequence{block: 0x80000000, count: 1,
  334. next: &sequence{block: 0x0, count: 2, next: &sequence{block: 0x1, count: 1}}}}}},
  335. // Set bit in a middle sequence in a middle block, last bit
  336. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 16, 31,
  337. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 3, next: &sequence{block: 0x1, count: 1,
  338. next: &sequence{block: 0x0, count: 2, next: &sequence{block: 0x1, count: 1}}}}}},
  339. // Set bit in a middle sequence in a middle block, middle bit
  340. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 16, 15,
  341. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 3, next: &sequence{block: 0x10000, count: 1,
  342. next: &sequence{block: 0x0, count: 2, next: &sequence{block: 0x1, count: 1}}}}}},
  343. // Set bit in a middle sequence in the last block, first bit
  344. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 24, 0,
  345. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5, next: &sequence{block: 0x80000000, count: 1,
  346. next: &sequence{block: 0x1, count: 1}}}}},
  347. // Set bit in a middle sequence in the last block, last bit
  348. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x4, count: 1}}}, 24, 31,
  349. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5, next: &sequence{block: 0x1, count: 1,
  350. next: &sequence{block: 0x4, count: 1}}}}},
  351. // Set bit in a middle sequence in the last block, last bit (merge involved)
  352. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 24, 31,
  353. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5, next: &sequence{block: 0x1, count: 2}}}},
  354. // Set bit in a middle sequence in the last block, middle bit
  355. {&sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 6, next: &sequence{block: 0x1, count: 1}}}, 24, 16,
  356. &sequence{block: 0x80000000, count: 1, next: &sequence{block: 0x0, count: 5, next: &sequence{block: 0x8000, count: 1,
  357. next: &sequence{block: 0x1, count: 1}}}}},
  358. }
  359. for n, i := range input {
  360. mask := pushReservation(i.bytePos, i.bitPos, i.mask, false)
  361. if !mask.equal(i.newMask) {
  362. t.Fatalf("Error in (%d) pushReservation():\n%s + (%d,%d):\nExp: %s\nGot: %s,",
  363. n, i.mask.toString(), i.bytePos, i.bitPos, i.newMask.toString(), mask.toString())
  364. }
  365. }
  366. }
  367. func TestSerializeDeserialize(t *testing.T) {
  368. s := getTestSequence()
  369. data, err := s.toByteArray()
  370. if err != nil {
  371. t.Fatal(err)
  372. }
  373. r := &sequence{}
  374. err = r.fromByteArray(data)
  375. if err != nil {
  376. t.Fatal(err)
  377. }
  378. if !s.equal(r) {
  379. t.Fatalf("Sequences are different: \n%v\n%v", s, r)
  380. }
  381. }
  382. func getTestSequence() *sequence {
  383. // Returns a custom sequence of 1024 * 32 bits
  384. return &sequence{
  385. block: 0xFFFFFFFF,
  386. count: 100,
  387. next: &sequence{
  388. block: 0xFFFFFFFE,
  389. count: 1,
  390. next: &sequence{
  391. block: 0xFF000000,
  392. count: 10,
  393. next: &sequence{
  394. block: 0xFFFFFFFF,
  395. count: 50,
  396. next: &sequence{
  397. block: 0xFFFFFFFC,
  398. count: 1,
  399. next: &sequence{
  400. block: 0xFF800000,
  401. count: 1,
  402. next: &sequence{
  403. block: 0xFFFFFFFF,
  404. count: 87,
  405. next: &sequence{
  406. block: 0x0,
  407. count: 150,
  408. next: &sequence{
  409. block: 0xFFFFFFFF,
  410. count: 200,
  411. next: &sequence{
  412. block: 0x0000FFFF,
  413. count: 1,
  414. next: &sequence{
  415. block: 0x0,
  416. count: 399,
  417. next: &sequence{
  418. block: 0xFFFFFFFF,
  419. count: 23,
  420. next: &sequence{
  421. block: 0x1,
  422. count: 1,
  423. },
  424. },
  425. },
  426. },
  427. },
  428. },
  429. },
  430. },
  431. },
  432. },
  433. },
  434. },
  435. }
  436. }
  437. func TestSet(t *testing.T) {
  438. hnd := New(1024 * 32)
  439. hnd.head = getTestSequence()
  440. firstAv := uint64(32*100 + 31)
  441. last := uint64(1024*32 - 1)
  442. if hnd.IsSet(100000) {
  443. t.Fatal("IsSet() returned wrong result")
  444. }
  445. if !hnd.IsSet(0) {
  446. t.Fatal("IsSet() returned wrong result")
  447. }
  448. if hnd.IsSet(firstAv) {
  449. t.Fatal("IsSet() returned wrong result")
  450. }
  451. if !hnd.IsSet(last) {
  452. t.Fatal("IsSet() returned wrong result")
  453. }
  454. if err := hnd.Set(0); err == nil {
  455. t.Fatal("Expected failure, but succeeded")
  456. }
  457. os, err := hnd.SetAny(false)
  458. if err != nil {
  459. t.Fatalf("Unexpected failure: %v", err)
  460. }
  461. if os != firstAv {
  462. t.Fatalf("SetAny returned unexpected ordinal. Expected %d. Got %d.", firstAv, os)
  463. }
  464. if !hnd.IsSet(firstAv) {
  465. t.Fatal("IsSet() returned wrong result")
  466. }
  467. if err := hnd.Unset(firstAv); err != nil {
  468. t.Fatalf("Unexpected failure: %v", err)
  469. }
  470. if hnd.IsSet(firstAv) {
  471. t.Fatal("IsSet() returned wrong result")
  472. }
  473. if err := hnd.Set(firstAv); err != nil {
  474. t.Fatalf("Unexpected failure: %v", err)
  475. }
  476. if err := hnd.Set(last); err == nil {
  477. t.Fatal("Expected failure, but succeeded")
  478. }
  479. }
  480. func TestSetUnset(t *testing.T) {
  481. numBits := uint64(32 * blockLen)
  482. hnd := New(numBits)
  483. if err := hnd.Set(uint64(32 * blockLen)); err == nil {
  484. t.Fatal("Expected failure, but succeeded")
  485. }
  486. if err := hnd.Unset(uint64(32 * blockLen)); err == nil {
  487. t.Fatal("Expected failure, but succeeded")
  488. }
  489. // set and unset all one by one
  490. for hnd.Unselected() > 0 {
  491. if _, err := hnd.SetAny(false); err != nil {
  492. t.Fatal(err)
  493. }
  494. }
  495. if _, err := hnd.SetAny(false); err != ErrNoBitAvailable {
  496. t.Fatal("Expected error. Got success")
  497. }
  498. if _, err := hnd.SetAnyInRange(10, 20, false); err != ErrNoBitAvailable {
  499. t.Fatal("Expected error. Got success")
  500. }
  501. if err := hnd.Set(50); err != ErrBitAllocated {
  502. t.Fatalf("Expected error. Got %v: %s", err, hnd)
  503. }
  504. i := uint64(0)
  505. for hnd.Unselected() < numBits {
  506. if err := hnd.Unset(i); err != nil {
  507. t.Fatal(err)
  508. }
  509. i++
  510. }
  511. }
  512. func TestOffsetSetUnset(t *testing.T) {
  513. numBits := uint64(32 * blockLen)
  514. hnd := New(numBits)
  515. // set and unset all one by one
  516. for hnd.Unselected() > 0 {
  517. if _, err := hnd.SetAny(false); err != nil {
  518. t.Fatal(err)
  519. }
  520. }
  521. if _, err := hnd.SetAny(false); err != ErrNoBitAvailable {
  522. t.Fatal("Expected error. Got success")
  523. }
  524. if _, err := hnd.SetAnyInRange(10, 20, false); err != ErrNoBitAvailable {
  525. t.Fatal("Expected error. Got success")
  526. }
  527. if err := hnd.Unset(288); err != nil {
  528. t.Fatal(err)
  529. }
  530. //At this point sequence is (0xffffffff, 9)->(0x7fffffff, 1)->(0xffffffff, 22)->end
  531. o, err := hnd.SetAnyInRange(32, 500, false)
  532. if err != nil {
  533. t.Fatal(err)
  534. }
  535. if o != 288 {
  536. t.Fatalf("Expected ordinal not received, Received:%d", o)
  537. }
  538. }
  539. func TestSetInRange(t *testing.T) {
  540. numBits := uint64(1024 * blockLen)
  541. hnd := New(numBits)
  542. hnd.head = getTestSequence()
  543. firstAv := uint64(100*blockLen + blockLen - 1)
  544. if o, err := hnd.SetAnyInRange(4, 3, false); err == nil {
  545. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  546. }
  547. if o, err := hnd.SetAnyInRange(0, numBits, false); err == nil {
  548. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  549. }
  550. o, err := hnd.SetAnyInRange(100*uint64(blockLen), 101*uint64(blockLen), false)
  551. if err != nil {
  552. t.Fatalf("Unexpected failure: (%d, %v)", o, err)
  553. }
  554. if o != firstAv {
  555. t.Fatalf("Unexpected ordinal: %d", o)
  556. }
  557. if o, err := hnd.SetAnyInRange(0, uint64(blockLen), false); err == nil {
  558. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  559. }
  560. if o, err := hnd.SetAnyInRange(0, firstAv-1, false); err == nil {
  561. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  562. }
  563. if o, err := hnd.SetAnyInRange(111*uint64(blockLen), 161*uint64(blockLen), false); err == nil {
  564. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  565. }
  566. o, err = hnd.SetAnyInRange(161*uint64(blockLen), 162*uint64(blockLen), false)
  567. if err != nil {
  568. t.Fatal(err)
  569. }
  570. if o != 161*uint64(blockLen)+30 {
  571. t.Fatalf("Unexpected ordinal: %d", o)
  572. }
  573. o, err = hnd.SetAnyInRange(161*uint64(blockLen), 162*uint64(blockLen), false)
  574. if err != nil {
  575. t.Fatal(err)
  576. }
  577. if o != 161*uint64(blockLen)+31 {
  578. t.Fatalf("Unexpected ordinal: %d", o)
  579. }
  580. o, err = hnd.SetAnyInRange(161*uint64(blockLen), 162*uint64(blockLen), false)
  581. if err == nil {
  582. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  583. }
  584. if _, err := hnd.SetAnyInRange(0, numBits-1, false); err != nil {
  585. t.Fatalf("Unexpected failure: %v", err)
  586. }
  587. // set one bit using the set range with 1 bit size range
  588. if _, err := hnd.SetAnyInRange(uint64(163*blockLen-1), uint64(163*blockLen-1), false); err != nil {
  589. t.Fatal(err)
  590. }
  591. // create a non multiple of 32 mask
  592. hnd = New(30)
  593. // set all bit in the first range
  594. for hnd.Unselected() > 22 {
  595. if o, err := hnd.SetAnyInRange(0, 7, false); err != nil {
  596. t.Fatalf("Unexpected failure: (%d, %v)", o, err)
  597. }
  598. }
  599. // try one more set, which should fail
  600. o, err = hnd.SetAnyInRange(0, 7, false)
  601. if err == nil {
  602. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  603. }
  604. if err != ErrNoBitAvailable {
  605. t.Fatalf("Unexpected error: %v", err)
  606. }
  607. // set all bit in a second range
  608. for hnd.Unselected() > 14 {
  609. if o, err := hnd.SetAnyInRange(8, 15, false); err != nil {
  610. t.Fatalf("Unexpected failure: (%d, %v)", o, err)
  611. }
  612. }
  613. // try one more set, which should fail
  614. o, err = hnd.SetAnyInRange(0, 15, false)
  615. if err == nil {
  616. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  617. }
  618. if err != ErrNoBitAvailable {
  619. t.Fatalf("Unexpected error: %v", err)
  620. }
  621. // set all bit in a range which includes the last bit
  622. for hnd.Unselected() > 12 {
  623. if o, err := hnd.SetAnyInRange(28, 29, false); err != nil {
  624. t.Fatalf("Unexpected failure: (%d, %v)", o, err)
  625. }
  626. }
  627. o, err = hnd.SetAnyInRange(28, 29, false)
  628. if err == nil {
  629. t.Fatalf("Expected failure. Got success with ordinal:%d", o)
  630. }
  631. if err != ErrNoBitAvailable {
  632. t.Fatalf("Unexpected error: %v", err)
  633. }
  634. }
  635. // This one tests an allocation pattern which unveiled an issue in pushReservation
  636. // Specifically a failure in detecting when we are in the (B) case (the bit to set
  637. // belongs to the last block of the current sequence). Because of a bug, code
  638. // was assuming the bit belonged to a block in the middle of the current sequence.
  639. // Which in turn caused an incorrect allocation when requesting a bit which is not
  640. // in the first or last sequence block.
  641. func TestSetAnyInRange(t *testing.T) {
  642. numBits := uint64(8 * blockLen)
  643. hnd := New(numBits)
  644. if err := hnd.Set(0); err != nil {
  645. t.Fatal(err)
  646. }
  647. if err := hnd.Set(255); err != nil {
  648. t.Fatal(err)
  649. }
  650. o, err := hnd.SetAnyInRange(128, 255, false)
  651. if err != nil {
  652. t.Fatal(err)
  653. }
  654. if o != 128 {
  655. t.Fatalf("Unexpected ordinal: %d", o)
  656. }
  657. o, err = hnd.SetAnyInRange(128, 255, false)
  658. if err != nil {
  659. t.Fatal(err)
  660. }
  661. if o != 129 {
  662. t.Fatalf("Unexpected ordinal: %d", o)
  663. }
  664. o, err = hnd.SetAnyInRange(246, 255, false)
  665. if err != nil {
  666. t.Fatal(err)
  667. }
  668. if o != 246 {
  669. t.Fatalf("Unexpected ordinal: %d", o)
  670. }
  671. o, err = hnd.SetAnyInRange(246, 255, false)
  672. if err != nil {
  673. t.Fatal(err)
  674. }
  675. if o != 247 {
  676. t.Fatalf("Unexpected ordinal: %d", o)
  677. }
  678. }
  679. func TestMethods(t *testing.T) {
  680. numBits := uint64(256 * blockLen)
  681. hnd := New(numBits)
  682. if hnd.Bits() != numBits {
  683. t.Fatalf("Unexpected bit number: %d", hnd.Bits())
  684. }
  685. if hnd.Unselected() != numBits {
  686. t.Fatalf("Unexpected bit number: %d", hnd.Unselected())
  687. }
  688. exp := "(0x0, 256)->end"
  689. if hnd.head.toString() != exp {
  690. t.Fatalf("Unexpected sequence string: %s", hnd.head.toString())
  691. }
  692. for i := 0; i < 192; i++ {
  693. _, err := hnd.SetAny(false)
  694. if err != nil {
  695. t.Fatal(err)
  696. }
  697. }
  698. exp = "(0xffffffff, 6)->(0x0, 250)->end"
  699. if hnd.head.toString() != exp {
  700. t.Fatalf("Unexpected sequence string: %s", hnd.head.toString())
  701. }
  702. }
  703. func TestRandomAllocateDeallocate(t *testing.T) {
  704. numBits := int(16 * blockLen)
  705. hnd := New(uint64(numBits))
  706. seed := time.Now().Unix()
  707. rand.Seed(seed)
  708. // Allocate all bits using a random pattern
  709. pattern := rand.Perm(numBits)
  710. for _, bit := range pattern {
  711. err := hnd.Set(uint64(bit))
  712. if err != nil {
  713. t.Fatalf("Unexpected failure on allocation of %d: %v.\nSeed: %d.\n%s", bit, err, seed, hnd)
  714. }
  715. }
  716. if hnd.Unselected() != 0 {
  717. t.Fatalf("Expected full sequence. Instead found %d free bits. Seed: %d.\n%s", hnd.unselected, seed, hnd)
  718. }
  719. if hnd.head.toString() != "(0xffffffff, 16)->end" {
  720. t.Fatalf("Unexpected db: %s", hnd.head.toString())
  721. }
  722. // Deallocate all bits using a random pattern
  723. pattern = rand.Perm(numBits)
  724. for _, bit := range pattern {
  725. err := hnd.Unset(uint64(bit))
  726. if err != nil {
  727. t.Fatalf("Unexpected failure on deallocation of %d: %v.\nSeed: %d.\n%s", bit, err, seed, hnd)
  728. }
  729. }
  730. if hnd.Unselected() != uint64(numBits) {
  731. t.Fatalf("Expected full sequence. Instead found %d free bits. Seed: %d.\n%s", hnd.unselected, seed, hnd)
  732. }
  733. if hnd.head.toString() != "(0x0, 16)->end" {
  734. t.Fatalf("Unexpected db: %s", hnd.head.toString())
  735. }
  736. }
  737. func TestAllocateRandomDeallocate(t *testing.T) {
  738. numBlocks := uint32(8)
  739. numBits := int(numBlocks * blockLen)
  740. hnd := New(uint64(numBits))
  741. expected := &sequence{block: 0xffffffff, count: uint64(numBlocks / 2), next: &sequence{block: 0x0, count: uint64(numBlocks / 2)}}
  742. // Allocate first half of the bits
  743. for i := 0; i < numBits/2; i++ {
  744. _, err := hnd.SetAny(false)
  745. if err != nil {
  746. t.Fatalf("Unexpected failure on allocation %d: %v\n%s", i, err, hnd)
  747. }
  748. }
  749. if hnd.Unselected() != uint64(numBits/2) {
  750. t.Fatalf("Expected full sequence. Instead found %d free bits. %s", hnd.unselected, hnd)
  751. }
  752. if !hnd.head.equal(expected) {
  753. t.Fatalf("Unexpected sequence. Got:\n%s", hnd)
  754. }
  755. seed := time.Now().Unix()
  756. rand.Seed(seed)
  757. // Deallocate half of the allocated bits following a random pattern
  758. pattern := rand.Perm(numBits / 2)
  759. for i := 0; i < numBits/4; i++ {
  760. bit := pattern[i]
  761. err := hnd.Unset(uint64(bit))
  762. if err != nil {
  763. t.Fatalf("Unexpected failure on deallocation of %d: %v.\nSeed: %d.\n%s", bit, err, seed, hnd)
  764. }
  765. }
  766. if hnd.Unselected() != uint64(3*numBits/4) {
  767. t.Fatalf("Expected full sequence. Instead found %d free bits.\nSeed: %d.\n%s", hnd.unselected, seed, hnd)
  768. }
  769. // Request a quarter of bits
  770. for i := 0; i < numBits/4; i++ {
  771. _, err := hnd.SetAny(false)
  772. if err != nil {
  773. t.Fatalf("Unexpected failure on allocation %d: %v\nSeed: %d\n%s", i, err, seed, hnd)
  774. }
  775. }
  776. if hnd.Unselected() != uint64(numBits/2) {
  777. t.Fatalf("Expected half sequence. Instead found %d free bits.\nSeed: %d\n%s", hnd.unselected, seed, hnd)
  778. }
  779. if !hnd.head.equal(expected) {
  780. t.Fatalf("Unexpected sequence. Got:\n%s", hnd)
  781. }
  782. }
  783. func TestAllocateRandomDeallocateSerialize(t *testing.T) {
  784. numBlocks := uint32(8)
  785. numBits := int(numBlocks * blockLen)
  786. hnd := New(uint64(numBits))
  787. expected := &sequence{block: 0xffffffff, count: uint64(numBlocks / 2), next: &sequence{block: 0x0, count: uint64(numBlocks / 2)}}
  788. // Allocate first half of the bits
  789. for i := 0; i < numBits/2; i++ {
  790. _, err := hnd.SetAny(true)
  791. if err != nil {
  792. t.Fatalf("Unexpected failure on allocation %d: %v\n%s", i, err, hnd)
  793. }
  794. }
  795. if hnd.Unselected() != uint64(numBits/2) {
  796. t.Fatalf("Expected full sequence. Instead found %d free bits. %s", hnd.unselected, hnd)
  797. }
  798. if !hnd.head.equal(expected) {
  799. t.Fatalf("Unexpected sequence. Got:\n%s", hnd)
  800. }
  801. seed := time.Now().Unix()
  802. rand.Seed(seed)
  803. // Deallocate half of the allocated bits following a random pattern
  804. pattern := rand.Perm(numBits / 2)
  805. for i := 0; i < numBits/4; i++ {
  806. bit := pattern[i]
  807. err := hnd.Unset(uint64(bit))
  808. if err != nil {
  809. t.Fatalf("Unexpected failure on deallocation of %d: %v.\nSeed: %d.\n%s", bit, err, seed, hnd)
  810. }
  811. }
  812. if hnd.Unselected() != uint64(3*numBits/4) {
  813. t.Fatalf("Expected full sequence. Instead found %d free bits.\nSeed: %d.\n%s", hnd.unselected, seed, hnd)
  814. }
  815. // Request a quarter of bits
  816. for i := 0; i < numBits/4; i++ {
  817. _, err := hnd.SetAny(true)
  818. if err != nil {
  819. t.Fatalf("Unexpected failure on allocation %d: %v\nSeed: %d\n%s", i, err, seed, hnd)
  820. }
  821. }
  822. if hnd.Unselected() != uint64(numBits/2) {
  823. t.Fatalf("Expected half sequence. Instead found %d free bits.\nSeed: %d\n%s", hnd.unselected, seed, hnd)
  824. }
  825. }
  826. func testSetRollover(t *testing.T, serial bool) {
  827. numBlocks := uint32(8)
  828. numBits := int(numBlocks * blockLen)
  829. hnd := New(uint64(numBits))
  830. // Allocate first half of the bits
  831. for i := 0; i < numBits/2; i++ {
  832. _, err := hnd.SetAny(serial)
  833. if err != nil {
  834. t.Fatalf("Unexpected failure on allocation %d: %v\n%s", i, err, hnd)
  835. }
  836. }
  837. if hnd.Unselected() != uint64(numBits/2) {
  838. t.Fatalf("Expected full sequence. Instead found %d free bits. %s", hnd.unselected, hnd)
  839. }
  840. seed := time.Now().Unix()
  841. rand.Seed(seed)
  842. // Deallocate half of the allocated bits following a random pattern
  843. pattern := rand.Perm(numBits / 2)
  844. for i := 0; i < numBits/4; i++ {
  845. bit := pattern[i]
  846. err := hnd.Unset(uint64(bit))
  847. if err != nil {
  848. t.Fatalf("Unexpected failure on deallocation of %d: %v.\nSeed: %d.\n%s", bit, err, seed, hnd)
  849. }
  850. }
  851. if hnd.Unselected() != uint64(3*numBits/4) {
  852. t.Fatalf("Unexpected free bits: found %d free bits.\nSeed: %d.\n%s", hnd.unselected, seed, hnd)
  853. }
  854. //request to allocate for remaining half of the bits
  855. for i := 0; i < numBits/2; i++ {
  856. _, err := hnd.SetAny(serial)
  857. if err != nil {
  858. t.Fatalf("Unexpected failure on allocation %d: %v\nSeed: %d\n%s", i, err, seed, hnd)
  859. }
  860. }
  861. //At this point all the bits must be allocated except the randomly unallocated bits
  862. //which were unallocated in the first half of the bit sequence
  863. if hnd.Unselected() != uint64(numBits/4) {
  864. t.Fatalf("Unexpected number of unselected bits %d, Expected %d", hnd.Unselected(), numBits/4)
  865. }
  866. for i := 0; i < numBits/4; i++ {
  867. _, err := hnd.SetAny(serial)
  868. if err != nil {
  869. t.Fatalf("Unexpected failure on allocation %d: %v\nSeed: %d\n%s", i, err, seed, hnd)
  870. }
  871. }
  872. //Now requesting to allocate the unallocated random bits (qurter of the number of bits) should
  873. //leave no more bits that can be allocated.
  874. if hnd.Unselected() != 0 {
  875. t.Fatalf("Unexpected number of unselected bits %d, Expected %d", hnd.Unselected(), 0)
  876. }
  877. }
  878. func TestSetRollover(t *testing.T) {
  879. testSetRollover(t, false)
  880. }
  881. func TestSetRolloverSerial(t *testing.T) {
  882. testSetRollover(t, true)
  883. }
  884. func TestGetFirstAvailableFromCurrent(t *testing.T) {
  885. input := []struct {
  886. mask *sequence
  887. bytePos uint64
  888. bitPos uint64
  889. start uint64
  890. curr uint64
  891. end uint64
  892. }{
  893. {&sequence{block: 0xffffffff, count: 2048}, invalidPos, invalidPos, 0, 0, 65536},
  894. {&sequence{block: 0x0, count: 8}, 0, 0, 0, 0, 256},
  895. {&sequence{block: 0x80000000, count: 8}, 1, 0, 0, 8, 256},
  896. {&sequence{block: 0xC0000000, count: 8}, 0, 2, 0, 2, 256},
  897. {&sequence{block: 0xE0000000, count: 8}, 0, 3, 0, 0, 256},
  898. {&sequence{block: 0xFFFB1FFF, count: 8}, 2, 0, 14, 0, 256},
  899. {&sequence{block: 0xFFFFFFFE, count: 8}, 3, 7, 0, 0, 256},
  900. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0x00000000, count: 1, next: &sequence{block: 0xffffffff, count: 14}}}, 4, 0, 0, 32, 512},
  901. {&sequence{block: 0xfffeffff, count: 1, next: &sequence{block: 0xffffffff, count: 15}}, 1, 7, 0, 16, 512},
  902. {&sequence{block: 0xfffeffff, count: 15, next: &sequence{block: 0xffffffff, count: 1}}, 5, 7, 0, 16, 512},
  903. {&sequence{block: 0xfffeffff, count: 15, next: &sequence{block: 0xffffffff, count: 1}}, 9, 7, 0, 48, 512},
  904. {&sequence{block: 0xffffffff, count: 2, next: &sequence{block: 0xffffffef, count: 14}}, 19, 3, 0, 124, 512},
  905. {&sequence{block: 0xfffeffff, count: 15, next: &sequence{block: 0x0fffffff, count: 1}}, 60, 0, 0, 480, 512},
  906. {&sequence{block: 0xffffffff, count: 1, next: &sequence{block: 0xfffeffff, count: 14, next: &sequence{block: 0xffffffff, count: 1}}}, 17, 7, 0, 124, 512},
  907. {&sequence{block: 0xfffffffb, count: 1, next: &sequence{block: 0xffffffff, count: 14, next: &sequence{block: 0xffffffff, count: 1}}}, 3, 5, 0, 124, 512},
  908. {&sequence{block: 0xfffffffb, count: 1, next: &sequence{block: 0xfffeffff, count: 14, next: &sequence{block: 0xffffffff, count: 1}}}, 13, 7, 0, 80, 512},
  909. }
  910. for n, i := range input {
  911. bytePos, bitPos, _ := getAvailableFromCurrent(i.mask, i.start, i.curr, i.end)
  912. if bytePos != i.bytePos || bitPos != i.bitPos {
  913. t.Fatalf("Error in (%d) getFirstAvailable(). Expected (%d, %d). Got (%d, %d)", n, i.bytePos, i.bitPos, bytePos, bitPos)
  914. }
  915. }
  916. }
  917. func TestMarshalJSON(t *testing.T) {
  918. expected := []byte("hello libnetwork")
  919. hnd := New(uint64(len(expected) * 8))
  920. for i, c := range expected {
  921. for j := 0; j < 8; j++ {
  922. if c&(1<<j) == 0 {
  923. continue
  924. }
  925. if err := hnd.Set(uint64(i*8 + j)); err != nil {
  926. t.Fatal(err)
  927. }
  928. }
  929. }
  930. hstr := hnd.String()
  931. t.Log(hstr)
  932. marshaled, err := hnd.MarshalJSON()
  933. if err != nil {
  934. t.Fatalf("MarshalJSON() err = %v", err)
  935. }
  936. t.Logf("%s", marshaled)
  937. // Serializations of hnd as would be marshaled by versions of the code
  938. // found in the wild. We need to support unmarshaling old versions to
  939. // maintain backwards compatibility with sequences persisted on disk.
  940. const (
  941. goldenV0 = `"AAAAAAAAAIAAAAAAAAAAPRamNjYAAAAAAAAAAfYENpYAAAAAAAAAAUZ2pi4AAAAAAAAAAe72TtYAAAAAAAAAAQ=="`
  942. )
  943. if string(marshaled) != goldenV0 {
  944. t.Errorf("MarshalJSON() output differs from golden. Please add a new golden case to this test.")
  945. }
  946. for _, tt := range []struct {
  947. name string
  948. data []byte
  949. }{
  950. {name: "Live", data: marshaled},
  951. {name: "Golden-v0", data: []byte(goldenV0)},
  952. } {
  953. tt := tt
  954. t.Run("UnmarshalJSON="+tt.name, func(t *testing.T) {
  955. hnd2 := New(0)
  956. if err := hnd2.UnmarshalJSON(tt.data); err != nil {
  957. t.Errorf("UnmarshalJSON() err = %v", err)
  958. }
  959. h2str := hnd2.String()
  960. t.Log(h2str)
  961. if hstr != h2str {
  962. t.Errorf("Unmarshaled a different bitmap: want %q, got %q", hstr, h2str)
  963. }
  964. })
  965. }
  966. }