Merge pull request #139 from projectdiscovery/feat-ctx-ip-arg

Adding ip context arg type
projectdiscovery · Jun 22, 2023 · 937580e · 937580e
2 parents 9af36f0 + 5cc8977
commit 937580e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/fastdialer/context.go b/fastdialer/context.go
@@ -5,4 +5,5 @@ type ContextOption string
 const (
 	// SniName to use in tls connection
 	SniName ContextOption = "sni-name"
+	IP      ContextOption = "ip"
 )
diff --git a/fastdialer/dialer.go b/fastdialer/dialer.go
@@ -180,7 +180,7 @@ func (d *Dialer) dial(ctx context.Context, network, address string, shouldUseTLS
 				fixedIP = addressParts[2]
 			}
 			// check if the ip is within the context
-			if ctxIP := ctx.Value("ip"); ctxIP != nil {
+			if ctxIP := ctx.Value(IP); ctxIP != nil {
 				fixedIP = fmt.Sprint(ctxIP)
 			}
 		} else {