differentiate between dial back error and dial error

libp2p · Aug 27, 2023 · d4187ef · d4187ef
1 parent 367cbe1
commit d4187ef
Show file tree

Hide file tree

Showing 6 changed files with 416 additions and 326 deletions.
diff --git a/p2p/protocol/autonatv2/autonat.go b/p2p/protocol/autonatv2/autonat.go
@@ -64,8 +64,9 @@ type Result struct {
 	Status pb.DialStatus
 }
 
-// AutoNAT implements the AutoNAT v2 client and server. Users can check reachability
-// for their addresses using the CheckReachability method.
+// AutoNAT implements the AutoNAT v2 client and server.
+// Users can check reachability for their addresses using the CheckReachability method.
+// The server provides amplification attack prevention and rate limiting.
 type AutoNAT struct {
 	host host.Host
 	sub  event.Subscription
@@ -140,6 +141,7 @@ func (an *AutoNAT) background() {
 		select {
 		case <-an.ctx.Done():
 			an.srv.Disable()
+			an.srv.Close()
 			an.peers = nil
 			an.wg.Done()
 			return