Adding proxy socks support to headless browser
package engine
import (
+ "context"
"crypto/tls"
+ "fmt"
+ "net"
"net/http"
"net/url"
"time"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/protocolstate"
"github.com/projectdiscovery/nuclei/v2/pkg/types"
+ "golang.org/x/net/proxy"
)
// newhttpClient creates a new http client for headless communication with a timeout
@@ -28,6 +32,22 @@ func newhttpClient(options *types.Options) *http.Client {
if proxyURL, err := url.Parse(options.ProxyURL); err == nil {
transport.Proxy = http.ProxyURL(proxyURL)
}
+ } else if options.ProxySocksURL != "" {
+ var proxyAuth *proxy.Auth
+
+ socksURL, proxyErr := url.Parse(options.ProxySocksURL)
+ if proxyErr == nil {
+ proxyAuth = &proxy.Auth{}
+ proxyAuth.User = socksURL.User.Username()
+ proxyAuth.Password, _ = socksURL.User.Password()
+ }
+ dialer, proxyErr := proxy.SOCKS5("tcp", fmt.Sprintf("%s:%s", socksURL.Hostname(), socksURL.Port()), proxyAuth, proxy.Direct)
+ dc := dialer.(interface {
+ DialContext(ctx context.Context, network, addr string) (net.Conn, error)
+ })
+ if proxyErr == nil {
+ transport.DialContext = dc.DialContext
+ }
}
return &http.Client{Transport: transport, Timeout: time.Duration(options.Timeout*3) * time.Second}
