feat: net_response add DialTLSContext

conf/input.http_response/http_response.toml

@@ -57,9 +57,11 @@ targets = [
 # expect_response_status_codes = "200|301"
 
 ## Optional TLS Config
-# use_tls = false
+# use_tls = true
+# tls_remote_addr = "172.16.174.59:443"
 # tls_ca = "/etc/categraf/ca.pem"
 # tls_cert = "/etc/categraf/cert.pem"
 # tls_key = "/etc/categraf/key.pem"
 ## Use TLS but skip chain & host verification
 # insecure_skip_verify = false
+# tls_server_name = "www.baidu.com"

inputs/http_response/http_response.go

@@ -41,6 +41,7 @@ type Instance struct {
 	ResponseTimeout                 config.Duration `toml:"response_timeout"`
 	Headers                         []string        `toml:"headers"`
 	Body                            string          `toml:"body"`
+	TlsRemoteAddr                   string          `toml:"tls_remote_addr"`
 	ExpectResponseSubstring         string          `toml:"expect_response_substring"`
 	ExpectResponseRegularExpression string          `toml:"expect_response_regular_expression"`
 	ExpectResponseStatusCode        *int            `toml:"expect_response_status_code"`
@@ -105,7 +106,6 @@ func (ins *Instance) createHTTPClient() (*http.Client, error) {
 	}
 
 	dialer := &net.Dialer{}
-
 	if ins.Interface != "" {
 		dialer.LocalAddr, err = netx.LocalAddressByInterfaceName(ins.Interface)
 		if err != nil {
@@ -120,7 +120,7 @@ func (ins *Instance) createHTTPClient() (*http.Client, error) {
 
 	client := httpx.CreateHTTPClient(httpx.TlsConfig(tlsCfg),
 		httpx.NetDialer(dialer), httpx.Proxy(proxy),
-
+		httpx.SetTransportTlsRemoteAddr(ins.TlsRemoteAddr, tlsCfg),
 		httpx.DisableKeepAlives(*ins.DisableKeepAlives),
 		httpx.Timeout(time.Duration(ins.Timeout)),
 		httpx.FollowRedirects(*ins.FollowRedirects))
@@ -247,7 +247,6 @@ func (ins *Instance) httpGather(target string) (map[string]string, map[string]in
 		ins.HTTPCommonConfig.Headers[ins.Headers[i]] = ins.Headers[i+1]
 	}
 	ins.SetHeaders(request)
-
 	// Start Timer
 	start := time.Now()
 	resp, err := ins.client.Do(request)

pkg/httpx/client.go

@@ -112,6 +112,29 @@ func DisableKeepAlives(disableKeepAlives bool) Option {
 	}
 }
 
+
+func SetTransportTlsRemoteAddr(remoteAddr string, tlsConfig *tls.Config) Option {
+	return func(client *http.Client) {
+		if len(remoteAddr) > 0 && tlsConfig != nil {
+			client.Transport.(*http.Transport).DialTLSContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
+				conn, err := client.Transport.(*http.Transport).DialContext(ctx, network, remoteAddr)
+				if err != nil {
+					return nil, err
+				}
+
+				// 建立 TLS 连接
+				tlsConn := tls.Client(conn, tlsConfig)
+				if err := tlsConn.Handshake(); err != nil {
+					conn.Close()
+					return nil, err
+				}
+
+				return tlsConn, nil
+			}
+		}
+	}
+}
+
 func CreateHTTPClient(opts ...Option) *http.Client {
 	client := &http.Client{
 		Transport: &http.Transport{},