Handle more PeerRemoteAddr variants when calculating dial addr

[btrtrc.git] / smartban.go

diff --git a/smartban.go b/smartban.go
index 96cc2acfdbc717e2be5e0a0284eb86884d498b71..034a702d950372400c8dddb002fa1121304073e5 100644 (file)
--- a/smartban.go
+++ b/smartban.go
@@ -5,7 +5,7 @@ import (
        "crypto/sha1"
        "net/netip"
 
-       "github.com/anacrolix/generics"
+       g "github.com/anacrolix/generics"
 
        "github.com/anacrolix/torrent/smartban"
 )
@@ -26,7 +26,7 @@ type blockCheckingWriter struct {
 func (me *blockCheckingWriter) checkBlock() {
        b := me.blockBuffer.Next(me.chunkSize)
        for _, peer := range me.cache.CheckBlock(me.requestIndex, b) {
-               generics.MakeMapIfNilAndSet(&me.badPeers, peer, struct{}{})
+               g.MakeMapIfNilAndSet(&me.badPeers, peer, struct{}{})
        }
        me.requestIndex++
 }