packedArray check if wordId belongs to valid boundaries of values

FinalRep · Jul 26, 2020 · ffde11a · ffde11a
1 parent 6200ec3
commit ffde11a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/lm/packed_array.go b/pkg/lm/packed_array.go
@@ -178,6 +178,10 @@ func (s *packedArray) find(wordID WordID, context ContextOffset) (packedValue, C
 		values = s.values[from:to]
 	}
 
+	if getWordID(values[0]) > wordID || getWordID(values[len(values)-1]) < wordID {
+		return 0, InvalidContextOffset
+	}
+
 	target := newPackedValue(wordID, 0)
 	j := sort.Search(len(values), func(i int) bool { return values[i] >= target })