Merge pull request hashrocket#5 from 8tomat8/master

Added flag to skip ssl certificate validation

connection.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"crypto/tls"
 	"encoding/hex"
 	"fmt"
 	"net/http"
@@ -17,11 +18,17 @@ type session struct {
 	errChan chan error
 }
 
-func connect(url, origin string, rlConf *readline.Config) error {
+func connect(url, origin string, rlConf *readline.Config, allowInsecure bool) error {
 	headers := make(http.Header)
 	headers.Add("Origin", origin)
 
-	ws, _, err := websocket.DefaultDialer.Dial(url, headers)
+	dialer := websocket.Dialer{
+		Proxy: http.ProxyFromEnvironment,
+		TLSClientConfig:&tls.Config{
+			InsecureSkipVerify: allowInsecure,
+		},
+	}
+	ws, _, err := dialer.Dial(url, headers)
 	if err != nil {
 		return err
 	}

main.go

@@ -17,6 +17,7 @@ const Version = "0.2.1"
 var options struct {
 	origin       string
 	printVersion bool
+	insecure bool
 }
 
 func main() {
@@ -27,6 +28,7 @@ func main() {
 	}
 	rootCmd.Flags().StringVarP(&options.origin, "origin", "o", "", "websocket origin")
 	rootCmd.Flags().BoolVarP(&options.printVersion, "version", "v", false, "print version")
+	rootCmd.Flags().BoolVarP(&options.insecure, "insecure", "k", false, "skip ssl certificate check")
 
 	rootCmd.Execute()
 }
@@ -70,7 +72,7 @@ func root(cmd *cobra.Command, args []string) {
 	err = connect(dest.String(), origin, &readline.Config{
 		Prompt:      "> ",
 		HistoryFile: historyFile,
-	})
+	}, options.insecure)
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
 		if err != io.EOF && err != readline.ErrInterrupt {