Add packed IP list

[btrtrc.git] / dht / dht.go

diff --git a/dht/dht.go b/dht/dht.go
index ecae42fbd312430b46b16e772ed714fd7334aafa..d7b6cc171cd414b3172e20c1db45f9d3b885707e 100644 (file)
--- a/dht/dht.go
+++ b/dht/dht.go
@@ -50,7 +50,7 @@ type Server struct {
        nodes            map[string]*node // Keyed by dHTAddr.String().
        mu               sync.Mutex
        closed           chan struct{}
-       ipBlockList      *iplist.IPList
+       ipBlockList      iplist.Ranger
        badNodes         *boom.BloomFilter
 
        numConfirmedAnnounces int
@@ -70,7 +70,7 @@ type ServerConfig struct {
        NoSecurity bool
        // Initial IP blocklist to use. Applied before serving and bootstrapping
        // begins.
-       IPBlocklist *iplist.IPList
+       IPBlocklist iplist.Ranger
        // Used to secure the server's ID. Defaults to the Conn's LocalAddr().
        PublicIP net.IP
 }
@@ -595,13 +595,13 @@ func (s *Server) setDefaults() (err error) {
 }
 
 // Packets to and from any address matching a range in the list are dropped.
-func (s *Server) SetIPBlockList(list *iplist.IPList) {
+func (s *Server) SetIPBlockList(list iplist.Ranger) {
        s.mu.Lock()
        defer s.mu.Unlock()
        s.ipBlockList = list
 }
 
-func (s *Server) IPBlocklist() *iplist.IPList {
+func (s *Server) IPBlocklist() iplist.Ranger {
        return s.ipBlockList
 }