Fix sockes5

dara/core.go

@@ -323,7 +323,7 @@ func DoRequest(request *Request, runtimeObject *RuntimeObject) (response *Respon
 		if !defaultClient.ifInit || defaultClient.httpClient.Transport == nil {
 			defaultClient.httpClient.Transport = trans
 		}
-		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ConnectTimeout)+IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
+		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
 		defaultClient.ifInit = true
 		defaultClient.Unlock()
 	}
@@ -442,7 +442,7 @@ func DoRequestWithCtx(ctx context.Context, request *Request, runtimeObject *Runt
 		if !defaultClient.ifInit || defaultClient.httpClient.Transport == nil {
 			defaultClient.httpClient.Transport = trans
 		}
-		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ConnectTimeout)+IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
+		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
 		defaultClient.ifInit = true
 		defaultClient.Unlock()
 	}
@@ -504,7 +504,6 @@ func DoRequestWithCtx(ctx context.Context, request *Request, runtimeObject *Runt
 
 func getHttpTransport(req *Request, runtime *RuntimeObject) (*http.Transport, error) {
 	trans := new(http.Transport)
-	trans.ResponseHeaderTimeout = time.Duration(IntValue(runtime.ReadTimeout)) * time.Millisecond
 	httpProxy, err := getHttpProxy(StringValue(req.Protocol), StringValue(req.Domain), runtime)
 	if err != nil {
 		return nil, err
@@ -697,7 +696,7 @@ func getSocks5Proxy(runtime *RuntimeObject) (proxy *url.URL, err error) {
 func getLocalAddr(localAddr string) (addr *net.TCPAddr) {
 	if localAddr != "" {
 		addr = &net.TCPAddr{
-			IP: net.ParseIP(localAddr),
+			IP: []byte(localAddr),
 		}
 	}
 	return addr
@@ -705,17 +704,20 @@ func getLocalAddr(localAddr string) (addr *net.TCPAddr) {
 
 func setDialContext(runtime *RuntimeObject) func(cxt context.Context, net, addr string) (c net.Conn, err error) {
 	return func(ctx context.Context, network, address string) (net.Conn, error) {
-		timeout := time.Duration(IntValue(runtime.ConnectTimeout)) * time.Millisecond
-		dialer := &net.Dialer{
-			Timeout: timeout,
-			Resolver: &net.Resolver{
-				PreferGo: false,
-			},
-		}
 		if runtime.LocalAddr != nil && StringValue(runtime.LocalAddr) != "" {
-			dialer.LocalAddr = getLocalAddr(StringValue(runtime.LocalAddr))
+			netAddr := &net.TCPAddr{
+				IP: []byte(StringValue(runtime.LocalAddr)),
+			}
+			return (&net.Dialer{
+				Timeout:   time.Duration(IntValue(runtime.ConnectTimeout)) * time.Second,
+				DualStack: true,
+				LocalAddr: netAddr,
+			}).DialContext(ctx, network, address)
 		}
-		return dialer.DialContext(ctx, network, address)
+		return (&net.Dialer{
+			Timeout:   time.Duration(IntValue(runtime.ConnectTimeout)) * time.Second,
+			DualStack: true,
+		}).DialContext(ctx, network, address)
 	}
 }
 

dara/core_test.go

@@ -1531,49 +1531,3 @@ func TestForceUint64(t *testing.T) {
 		}
 	}
 }
-
-func Test_getLocalAddr(t *testing.T) {
-	// Test empty address
-	addr := getLocalAddr("")
-	utils.AssertNil(t, addr)
-
-	// Test valid IPv4 address
-	addr = getLocalAddr("127.0.0.1")
-	utils.AssertNotNil(t, addr)
-	utils.AssertEqual(t, "127.0.0.1", addr.IP.String())
-
-	// Test valid IPv6 address
-	addr = getLocalAddr("::1")
-	utils.AssertNotNil(t, addr)
-	utils.AssertEqual(t, "::1", addr.IP.String())
-
-	// Test invalid IP
-	addr = getLocalAddr("invalid")
-	utils.AssertNotNil(t, addr)
-	utils.AssertNil(t, addr.IP)
-}
-
-func Test_setDialContext(t *testing.T) {
-	runtime := &RuntimeObject{
-		ConnectTimeout: Int(1000),
-		LocalAddr:      String("127.0.0.1"),
-	}
-	dialerFunc := setDialContext(runtime)
-	utils.AssertNotNil(t, dialerFunc)
-}
-
-func Test_TimeoutLogic(t *testing.T) {
-	runtime := &RuntimeObject{
-		ConnectTimeout: Int(1000),
-		ReadTimeout:    Int(2000),
-	}
-	
-	req := &Request{
-		Protocol: String("http"),
-		Domain:   String("localhost"),
-	}
-	trans, err := getHttpTransport(req, runtime)
-	utils.AssertNil(t, err)
-	utils.AssertNotNil(t, trans)
-	utils.AssertEqual(t, time.Duration(2000)*time.Millisecond, trans.ResponseHeaderTimeout)
-}

tea/tea.go

@@ -386,7 +386,7 @@ func DoRequest(request *Request, requestRuntime map[string]interface{}) (respons
 		if !defaultClient.ifInit || defaultClient.httpClient.Transport == nil {
 			defaultClient.httpClient.Transport = trans
 		}
-		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ConnectTimeout)+IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
+		defaultClient.httpClient.Timeout = time.Duration(IntValue(runtimeObject.ReadTimeout)) * time.Millisecond
 		defaultClient.ifInit = true
 		defaultClient.Unlock()
 	}
@@ -442,7 +442,6 @@ func DoRequest(request *Request, requestRuntime map[string]interface{}) (respons
 
 func getHttpTransport(req *Request, runtime *RuntimeObject) (*http.Transport, error) {
 	trans := new(http.Transport)
-	trans.ResponseHeaderTimeout = time.Duration(IntValue(runtime.ReadTimeout)) * time.Millisecond
 	httpProxy, err := getHttpProxy(StringValue(req.Protocol), StringValue(req.Domain), runtime)
 	if err != nil {
 		return nil, err
@@ -603,7 +602,7 @@ func getSocks5Proxy(runtime *RuntimeObject) (proxy *url.URL, err error) {
 func getLocalAddr(localAddr string) (addr *net.TCPAddr) {
 	if localAddr != "" {
 		addr = &net.TCPAddr{
-			IP: net.ParseIP(localAddr),
+			IP: []byte(localAddr),
 		}
 	}
 	return addr
@@ -611,18 +610,20 @@ func getLocalAddr(localAddr string) (addr *net.TCPAddr) {
 
 func setDialContext(runtime *RuntimeObject) func(cxt context.Context, net, addr string) (c net.Conn, err error) {
 	return func(ctx context.Context, network, address string) (net.Conn, error) {
-		timeout := time.Duration(IntValue(runtime.ConnectTimeout)) * time.Millisecond
-		dialer := &net.Dialer{
-			Timeout: timeout,
-			Resolver: &net.Resolver{
-				PreferGo: false,
-			},
-			DualStack: true,
-		}
 		if runtime.LocalAddr != nil && StringValue(runtime.LocalAddr) != "" {
-			dialer.LocalAddr = getLocalAddr(StringValue(runtime.LocalAddr))
+			netAddr := &net.TCPAddr{
+				IP: []byte(StringValue(runtime.LocalAddr)),
+			}
+			return (&net.Dialer{
+				Timeout:   time.Duration(IntValue(runtime.ConnectTimeout)) * time.Second,
+				DualStack: true,
+				LocalAddr: netAddr,
+			}).DialContext(ctx, network, address)
 		}
-		return dialer.DialContext(ctx, network, address)
+		return (&net.Dialer{
+			Timeout:   time.Duration(IntValue(runtime.ConnectTimeout)) * time.Second,
+			DualStack: true,
+		}).DialContext(ctx, network, address)
 	}
 }
 

tea/tea_test.go

@@ -967,49 +967,3 @@ func Test_TransInt32AndInt(t *testing.T) {
 	b := ToInt32(a)
 	utils.AssertEqual(t, Int32Value(b), int32(10))
 }
-
-func Test_getLocalAddr(t *testing.T) {
-	// Test empty address
-	addr := getLocalAddr("")
-	utils.AssertNil(t, addr)
-
-	// Test valid IPv4 address
-	addr = getLocalAddr("127.0.0.1")
-	utils.AssertNotNil(t, addr)
-	utils.AssertEqual(t, "127.0.0.1", addr.IP.String())
-
-	// Test valid IPv6 address
-	addr = getLocalAddr("::1")
-	utils.AssertNotNil(t, addr)
-	utils.AssertEqual(t, "::1", addr.IP.String())
-
-	// Test invalid IP
-	addr = getLocalAddr("invalid")
-	utils.AssertNotNil(t, addr)
-	utils.AssertNil(t, addr.IP)
-}
-
-func Test_setDialContext(t *testing.T) {
-	runtime := &RuntimeObject{
-		ConnectTimeout: Int(1000),
-		LocalAddr:      String("127.0.0.1"),
-	}
-	dialerFunc := setDialContext(runtime)
-	utils.AssertNotNil(t, dialerFunc)
-}
-
-func Test_TimeoutLogic(t *testing.T) {
-	runtime := &RuntimeObject{
-		ConnectTimeout: Int(1000),
-		ReadTimeout:    Int(2000),
-	}
-	
-	req := &Request{
-		Protocol: String("http"),
-		Domain:   String("localhost"),
-	}
-	trans, err := getHttpTransport(req, runtime)
-	utils.AssertNil(t, err)
-	utils.AssertNotNil(t, trans)
-	utils.AssertEqual(t, time.Duration(2000)*time.Millisecond, trans.ResponseHeaderTimeout)
-}