negentropy: fuzz testing, move accumulator to vector package.

This commit is contained in:
fiatjaf
2024-09-20 10:56:15 -03:00
parent e9e96be95e
commit 9813d1776f
13 changed files with 205 additions and 68 deletions

View File

@@ -13,6 +13,8 @@ import (
type Vector struct {
items []negentropy.Item
sealed bool
acc Accumulator
}
func New() *Vector {
@@ -21,14 +23,13 @@ func New() *Vector {
}
}
func (v *Vector) Insert(createdAt nostr.Timestamp, id string) error {
func (v *Vector) Insert(createdAt nostr.Timestamp, id string) {
if len(id) != 64 {
return fmt.Errorf("bad id size for added item: expected %d bytes, got %d", 32, len(id)/2)
panic(fmt.Errorf("bad id size for added item: expected %d bytes, got %d", 32, len(id)/2))
}
item := negentropy.Item{Timestamp: createdAt, ID: id}
v.items = append(v.items, item)
return nil
}
func (v *Vector) Size() int { return len(v.items) }
@@ -63,15 +64,14 @@ func (v *Vector) FindLowerBound(begin, end int, bound negentropy.Bound) int {
return begin + idx
}
func (v *Vector) Fingerprint(begin, end int) [negentropy.FingerprintSize]byte {
var out negentropy.Accumulator
out.SetToZero()
func (v *Vector) Fingerprint(begin, end int) string {
v.acc.Reset()
tmp := make([]byte, 32)
for _, item := range v.Range(begin, end) {
hex.Decode(tmp, []byte(item.ID))
out.AddBytes(tmp)
v.acc.AddBytes(tmp)
}
return out.GetFingerprint(end - begin)
return v.acc.GetFingerprint(end - begin)
}