tests/fuzzers/txfetcher: pull in suggestion from Marius

This commit is contained in:
Péter Szilágyi 2023-10-06 12:39:05 +03:00
parent 4a0d6cac97
commit c44bcfb71b

View file

@ -117,6 +117,8 @@ func Fuzz(input []byte) int {
var ( var (
announceIdxs = make([]int, announce) announceIdxs = make([]int, announce)
announces = make([]common.Hash, announce) announces = make([]common.Hash, announce)
types = make([]byte, announce)
sizes = make([]uint32, announce)
) )
for i := 0; i < len(announces); i++ { for i := 0; i < len(announces); i++ {
annBuf := make([]byte, 2) annBuf := make([]byte, 2)
@ -125,11 +127,13 @@ func Fuzz(input []byte) int {
} }
announceIdxs[i] = (int(annBuf[0])*256 + int(annBuf[1])) % len(txs) announceIdxs[i] = (int(annBuf[0])*256 + int(annBuf[1])) % len(txs)
announces[i] = txs[announceIdxs[i]].Hash() announces[i] = txs[announceIdxs[i]].Hash()
types[i] = txs[announceIdxs[i]].Type()
sizes[i] = uint32(txs[announceIdxs[i]].Size())
} }
if verbose { if verbose {
fmt.Println("Notify", peer, announceIdxs) fmt.Println("Notify", peer, announceIdxs)
} }
if err := f.Notify(peer, nil, nil, announces); err != nil { if err := f.Notify(peer, types, sizes, announces); err != nil {
panic(err) panic(err)
} }