Merge pull request #76 from nathan0/master

Add optional HOST header

Author: jpillora

client/client.go

@@ -28,6 +28,7 @@ type Config struct {
 	Server           string
 	HTTPProxy        string
 	Remotes          []string
+	HostHeader       string
 }
 
 //Client represents a client instance
@@ -198,7 +199,13 @@ func (c *Client) connectionLoop() {
 				return c.httpProxyURL, nil
 			}
 		}
-		wsConn, _, err := d.Dial(c.server, nil)
+		hostHeader := http.Header{}
+		if c.config.HostHeader == "" {
+			hostHeader = nil
+		} else {
+			hostHeader = http.Header{"Host": {c.config.HostHeader}}
+		}
+		wsConn, _, err := d.Dial(c.server, hostHeader)
 		if err != nil {
 			connerr = err
 			continue

main.go

@@ -263,6 +263,8 @@ var clientHelp = `
     --proxy, An optional HTTP CONNECT proxy which will be used reach
     the chisel server. Authentication can be specified inside the URL.
     For example, http://admin:[email protected]:8081
+
+    --hostname, An optional HOST header to send, can be left blank.
 ` + commonHelp
 
 func client(args []string) {
@@ -276,6 +278,7 @@ func client(args []string) {
 	maxRetryInterval := flags.Duration("max-retry-interval", 0, "")
 	proxy := flags.String("proxy", "", "")
 	pid := flags.Bool("pid", false, "")
+	hostname := flags.String("hostname", "", "")
 	verbose := flags.Bool("v", false, "")
 	flags.Usage = func() {
 		fmt.Print(clientHelp)
@@ -299,6 +302,7 @@ func client(args []string) {
 		HTTPProxy:        *proxy,
 		Server:           args[0],
 		Remotes:          args[1:],
+		HostHeader:       *hostname,
 	})
 	if err != nil {
 		log.Fatal(err)