8 "github.com/anacrolix/generics"
10 "github.com/anacrolix/torrent/smartban"
13 type bannableAddr = netip.Addr
15 type smartBanCache = smartban.Cache[bannableAddr, RequestIndex, [sha1.Size]byte]
17 type blockCheckingWriter struct {
19 requestIndex RequestIndex
20 // Peers that didn't match blocks written now.
21 badPeers map[bannableAddr]struct{}
22 blockBuffer bytes.Buffer
26 func (me *blockCheckingWriter) checkBlock() {
27 b := me.blockBuffer.Next(me.chunkSize)
28 for _, peer := range me.cache.CheckBlock(me.requestIndex, b) {
29 generics.MakeMapIfNilAndSet(&me.badPeers, peer, struct{}{})
34 func (me *blockCheckingWriter) checkFullBlocks() {
35 for me.blockBuffer.Len() >= me.chunkSize {
40 func (me *blockCheckingWriter) Write(b []byte) (int, error) {
41 n, err := me.blockBuffer.Write(b)
43 // bytes.Buffer.Write should never fail.
50 // Check any remaining block data. Terminal pieces or piece sizes that don't divide into the chunk
51 // size cleanly may leave fragments that should be checked.
52 func (me *blockCheckingWriter) Flush() {
53 for me.blockBuffer.Len() != 0 {