remove closed dialer reference

projectdiscovery · May 22, 2024 · b8c2f94 · b8c2f94
1 parent c20d42b
commit b8c2f94
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pkg/protocols/common/protocolinit/init.go b/pkg/protocols/common/protocolinit/init.go
@@ -13,6 +13,9 @@ import (
 
 // Init initializes the client pools for the protocols
 func Init(options *types.Options) error {
+	if err := protocolstate.Init(options); err != nil {
+		return err
+	}
 	if err := dnsclientpool.Init(options); err != nil {
 		return err
 	}
@@ -36,4 +39,5 @@ func Init(options *types.Options) error {
 
 func Close() {
 	protocolstate.Dialer.Close()
+	protocolstate.Dialer = nil
 }
diff --git a/pkg/protocols/common/protocolstate/state.go b/pkg/protocols/common/protocolstate/state.go
@@ -210,6 +210,7 @@ func interfaceAddresses(interfaceName string) ([]net.Addr, error) {
 func Close() {
 	if Dialer != nil {
 		Dialer.Close()
+		Dialer = nil
 	}
 	StopActiveMemGuardian()
 }