fix proxy when http

lqqyt2423 · May 27, 2024 · 8f97773 · 8f97773
1 parent 5f4ec97
commit 8f97773
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/proxy/attacker.go b/proxy/attacker.go
@@ -150,9 +150,8 @@ func (a *attacker) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 func (a *attacker) initHttpDialFn(req *http.Request) {
 	connCtx := req.Context().Value(connContextKey).(*ConnContext)
 	connCtx.dialFn = func(ctx context.Context) error {
-		// todo: proxy
 		addr := helper.CanonicalAddr(req.URL)
-		c, err := (&net.Dialer{}).DialContext(ctx, "tcp", addr)
+		c, err := a.proxy.getUpstreamConn(ctx, req)
 		if err != nil {
 			return err
 		}

diff --git a/proxy/proxy.go b/proxy/proxy.go
@@ -113,10 +113,11 @@ func (proxy *Proxy) getUpstreamConn(ctx context.Context, req *http.Request) (net
 		return nil, err
 	}
 	var conn net.Conn
+	address := helper.CanonicalAddr(req.URL)
 	if proxyUrl != nil {
-		conn, err = helper.GetProxyConn(ctx, proxyUrl, req.Host, proxy.Opts.SslInsecure)
+		conn, err = helper.GetProxyConn(ctx, proxyUrl, address, proxy.Opts.SslInsecure)
 	} else {
-		conn, err = (&net.Dialer{}).DialContext(ctx, "tcp", req.Host)
+		conn, err = (&net.Dialer{}).DialContext(ctx, "tcp", address)
 	}
 	return conn, err
 }