[add] http proxy (#19)

* [add] http proxy

* [fix]socks5

Co-authored-by: sxueck <[email protected]>

Author: sxueck

.gitignore

@@ -15,3 +15,4 @@
 
 dist
 *.tar.gz
+.idea/

go.mod

@@ -7,4 +7,5 @@ require (
 	github.com/smartystreets/goconvey v0.0.0-20190306220146-200a235640ff
 	github.com/spf13/cobra v0.0.3
 	github.com/spf13/pflag v1.0.3 // indirect
+	golang.org/x/net v0.0.0-20210224082022-3d97a244fca7
 )

go.sum

@@ -12,3 +12,9 @@ github.com/spf13/cobra v0.0.3 h1:ZlrZ4XsMRm04Fr5pSFxBgfND2EBVa1nLpiy1stUsX/8=
 github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=
 github.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg=
 github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
+golang.org/x/net v0.0.0-20210224082022-3d97a244fca7 h1:OgUuv8lsRpBibGNbSizVwKWlysjaNzmC9gYMhPVfqFM=
+golang.org/x/net v0.0.0-20210224082022-3d97a244fca7/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=

main.go

@@ -18,6 +18,7 @@ var (
 	version     string
 	gitCommit   string
 	counter     int
+	proxy       string
 	timeout     string
 	interval    string
 	sigs        chan os.Signal
@@ -123,6 +124,7 @@ var rootCmd = cobra.Command{
 			Host:     parseHost,
 			Port:     port,
 			Counter:  counter,
+			Proxy:    proxy,
 			Protocol: protocol,
 		}
 		var pinger ping.Pinger
@@ -161,6 +163,7 @@ var rootCmd = cobra.Command{
 func init() {
 	rootCmd.Flags().BoolVarP(&showVersion, "version", "v", false, "show the version and exit")
 	rootCmd.Flags().IntVarP(&counter, "counter", "c", 4, "ping counter")
+	rootCmd.Flags().StringVar(&proxy, "proxy", "", "Use HTTP proxy")
 	rootCmd.Flags().StringVarP(&timeout, "timeout", "T", "1s", `connect timeout, units are "ns", "us" (or "µs"), "ms", "s", "m", "h"`)
 	rootCmd.Flags().StringVarP(&interval, "interval", "I", "1s", `ping interval, units are "ns", "us" (or "µs"), "ms", "s", "m", "h"`)
 

ping/http.go

@@ -2,11 +2,16 @@ package ping
 
 import (
 	"bytes"
+	"context"
 	"fmt"
+	"golang.org/x/net/proxy"
 	"io"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"net/http/httptrace"
+	"net/url"
+	"strings"
 	"time"
 )
 
@@ -96,10 +101,12 @@ func (ping HTTPing) ping() (time.Duration, *http.Response, string, error) {
 		body = bytes.NewBufferString("{}")
 	}
 	req, err := http.NewRequest(ping.Method, ping.target.String(), body)
-	req.Header.Set(http.CanonicalHeaderKey("User-Agent"), "tcping")
 	if err != nil {
 		return 0, nil, "", err
 	}
+
+	req.Header.Set(http.CanonicalHeaderKey("User-Agent"), "tcping")
+
 	var remoteAddr string
 	trace := &httptrace.ClientTrace{
 		ConnectStart: func(network, addr string) {
@@ -108,7 +115,45 @@ func (ping HTTPing) ping() (time.Duration, *http.Response, string, error) {
 	}
 	req = req.WithContext(httptrace.WithClientTrace(req.Context(), trace))
 	duration, errIfce := timeIt(func() interface{} {
-		client := http.Client{Timeout: ping.target.Timeout}
+		client := http.Client{
+			Timeout: ping.target.Timeout,
+		}
+
+		proxyProco := ping.target.Proxy
+		if ping.target.Proxy != "" {
+
+			var (
+				parProxy *url.URL
+				dialer   proxy.Dialer
+			)
+
+			httpTransport := &http.Transport{}
+
+			if strings.HasPrefix(proxyProco, "http") {
+				parProxy, err = url.Parse(ping.target.Proxy)
+				if err != nil {
+					return err
+				}
+				httpTransport.Proxy = http.ProxyURL(parProxy)
+
+			} else if strings.HasPrefix(proxyProco, "sock") {
+
+				dialer, err = proxy.SOCKS5("tcp", proxyProco[strings.Index(proxyProco, "//")+2:], nil, proxy.Direct)
+				if err != nil {
+					return fmt.Errorf("can't connect to the proxy: %s", err)
+				}
+
+				httpTransport.DialContext = func(ctx context.Context, network string, addr string) (net.Conn, error) {
+					return dialer.Dial(network, addr)
+				}
+
+			} else {
+				return fmt.Errorf("%s", "please enter the correct agency procotol")
+			}
+
+			client.Transport = httpTransport
+		}
+
 		resp, err = client.Do(req)
 		return err
 	})

ping/ping.go

@@ -52,6 +52,7 @@ type Target struct {
 	Protocol Protocol
 	Host     string
 	Port     int
+	Proxy    string
 
 	Counter  int
 	Interval time.Duration