Merge pull request #5 from projectdiscovery/feature-dialtls

Adding TLS support
projectdiscovery · Feb 21, 2021 · 2448775 · 2448775
2 parents 8a0407e + ea4a59e
commit 2448775
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/fastdialer/dialer.go b/fastdialer/dialer.go
@@ -2,6 +2,7 @@ package fastdialer
 
 import (
 	"context"
+	"crypto/tls"
 	"errors"
 	"fmt"
 	"net"
@@ -49,6 +50,17 @@ func NewDialer(options Options) (*Dialer, error) {
 
 // Dial function compatible with net/http
 func (d *Dialer) Dial(ctx context.Context, network, address string) (conn net.Conn, err error) {
+	conn, err = d.dial(ctx, network, address, false)
+	return
+}
+
+// DialTLS with encrypted connection
+func (d *Dialer) DialTLS(ctx context.Context, network, address string) (conn net.Conn, err error) {
+	conn, err = d.dial(ctx, network, address, true)
+	return
+}
+
+func (d *Dialer) dial(ctx context.Context, network, address string, shouldUseTLS bool) (conn net.Conn, err error) {
 	separator := strings.LastIndex(address, ":")
 
 	// check if data is in cache
@@ -68,7 +80,11 @@ func (d *Dialer) Dial(ctx context.Context, network, address string) (conn net.Co
 
 	// Dial to the IPs finally.
 	for _, ip := range append(data.A, data.AAAA...) {
-		conn, err = d.dialer.DialContext(ctx, network, ip+address[separator:])
+		if shouldUseTLS {
+			conn, err = tls.DialWithDialer(d.dialer, network, ip+address[separator:], &tls.Config{InsecureSkipVerify: true})
+		} else {
+			conn, err = d.dialer.DialContext(ctx, network, ip+address[separator:])
+		}
 		if err == nil {
 			setErr := d.dialerHistory.Set(hostname, []byte(ip))
 			if setErr != nil {

diff --git a/go.sum b/go.sum
@@ -8,6 +8,7 @@ github.com/golang/protobuf v1.2.0 h1:P3YflyNX/ehuJFLhxviNdFxQPkGK5cDcApsge1SqnvM
 github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db h1:woRePGFeVFfLKN/pOkfl+p/TAqKOfFu+7KPlMVpok/w=
 github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/golang/snappy v0.0.2 h1:aeE13tS0IiQgFjYdoL8qN3K1N2bXXtI6Vi51/y7BpMw=
 github.com/golang/snappy v0.0.2/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
 github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
@@ -27,6 +28,7 @@ github.com/projectdiscovery/hmap v0.0.1/go.mod h1:VDEfgzkKQdq7iGTKz8Ooul0NuYHQ8q
 github.com/projectdiscovery/retryabledns v1.0.6 h1:fz33puVeUKJJ5s2POSlxO4WA4iodW6Yzm/EVNuO/93w=
 github.com/projectdiscovery/retryabledns v1.0.6/go.mod h1:/UzJn4I+cPdQl6pKiiQfvVAT636YZvJQYZhYhGB0dUQ=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
 github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=