From 50d31a3c830db47f253d832f0f292c935176eea3 Mon Sep 17 00:00:00 2001 From: Todd Gruben Date: Fri, 10 Nov 2017 12:44:25 -0600 Subject: [PATCH] fix overflow in differenceRunBitmap --- roaring/roaring.go | 5 ++++- roaring/roaring_internal_test.go | 9 +++++++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/roaring/roaring.go b/roaring/roaring.go index 62f87d94b..e4d75a0d2 100644 --- a/roaring/roaring.go +++ b/roaring/roaring.go @@ -2508,7 +2508,7 @@ func differenceRunArray(a, b *container) *container { // differenceRunBitmap computes the difference of an run from a bitmap. func differenceRunBitmap(a, b *container) *container { // If a is full, difference is the flip of b. - if a.runs[0].start == 0 && a.runs[0].last == 65535 { + if len(a.runs) > 0 && a.runs[0].start == 0 && a.runs[0].last == 65535 { return b.flipBitmap() } output := &container{container_type: ContainerRun} @@ -2534,6 +2534,9 @@ func differenceRunBitmap(a, b *container) *container { break } } + if bit == 65535 { //overflow + break + } } if run.start <= run.last { output.runs = append(output.runs, run) diff --git a/roaring/roaring_internal_test.go b/roaring/roaring_internal_test.go index bdc08b649..a4b963a13 100644 --- a/roaring/roaring_internal_test.go +++ b/roaring/roaring_internal_test.go @@ -1524,14 +1524,19 @@ func TestDifferenceRunBitmap(t *testing.T) { bitmap: MakeBitmap([]uint64{0x0, 0x8000000000000000}), exp: []interval16{{start: 0, last: 65}}, }, + { + runs: []interval16{{start: 1, last: 65535}}, + bitmap: MakeBitmap([]uint64{0x0000000000000001}), + exp: []interval16{{start: 1, last: 65535}}, + }, } for i, test := range tests { a.runs = test.runs - a.n = a.runCountRange(0, 100) + a.n = a.runCountRange(0, 65535) for i, v := range test.bitmap { b.bitmap[i] = v } - b.n = b.bitmapCountRange(0, 100) + b.n = b.bitmapCountRange(0, 65535) ret := differenceRunBitmap(a, b) if !reflect.DeepEqual(ret.runs, test.exp) { t.Fatalf("test #%v expected %v, but got %v", i, test.exp, ret.runs)