Merge pull request #35 from projectdiscovery/feat-custom-sni-name

Adding custom SNI name

fastdialer/context.go

@@ -0,0 +1,8 @@
+package fastdialer
+
+type ContextOption string
+
+const (
+	// SniName to use in tls connection
+	SniName ContextOption = "sni-name"
+)

fastdialer/dialer.go

@@ -202,13 +202,25 @@ func (d *Dialer) dial(ctx context.Context, network, address string, shouldUseTLS
 		hostPort := net.JoinHostPort(ip, port)
 		if shouldUseTLS {
 			tlsconfigCopy := tlsconfig.Clone()
-			if !iputil.IsIP(hostname) {
+			switch {
+			case d.options.SNIName != "":
+				tlsconfigCopy.ServerName = d.options.SNIName
+			case ctx.Value(SniName) != nil:
+				sniName := ctx.Value(SniName).(string)
+				tlsconfigCopy.ServerName = sniName
+			case !iputil.IsIP(hostname):
 				tlsconfigCopy.ServerName = hostname
 			}
-			conn, err = tls.DialWithDialer(d.dialer, network, hostPort, tlsconfig)
+			conn, err = tls.DialWithDialer(d.dialer, network, hostPort, tlsconfigCopy)
 		} else if shouldUseZTLS {
 			ztlsconfigCopy := ztlsconfig.Clone()
-			if !iputil.IsIP(hostname) {
+			switch {
+			case d.options.SNIName != "":
+				ztlsconfigCopy.ServerName = d.options.SNIName
+			case ctx.Value(SniName) != nil:
+				sniName := ctx.Value(SniName).(string)
+				ztlsconfigCopy.ServerName = sniName
+			case !iputil.IsIP(hostname):
 				ztlsconfigCopy.ServerName = hostname
 			}
 			conn, err = ztls.DialWithDialer(d.dialer, network, hostPort, ztlsconfigCopy)

fastdialer/options.go

@@ -46,6 +46,7 @@ type Options struct {
 	DialerKeepAlive     time.Duration
 	Dialer              *net.Dialer
 	WithZTLS            bool
+	SNIName             string
 }
 
 // DefaultOptions of the cache