From 04a6c82730cc941d377d70ec810b733113b901c6 Mon Sep 17 00:00:00 2001
From: Shubham Rasal <shubham@projectdiscovery.io>
Date: Mon, 17 Mar 2025 16:44:33 +0530
Subject: [PATCH] add setDialer function to TCP protocol (#6101)

- add ability to override setDialer function for tcp
- socks proxy is not used incase of tcp protocol
- TCP uses global tcp dialer shared, we need template request level ability to set dialer
---
 pkg/protocols/network/network.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkg/protocols/network/network.go b/pkg/protocols/network/network.go
index c90f5e701..5c072affc 100644
--- a/pkg/protocols/network/network.go
+++ b/pkg/protocols/network/network.go
@@ -259,3 +259,7 @@ func (request *Request) Compile(options *protocols.ExecutorOptions) error {
 func (request *Request) Requests() int {
 	return len(request.Address)
 }
+
+func (request *Request) SetDialer(dialer *fastdialer.Dialer) {
+	request.dialer = dialer
+}