Add option for using websocket binary frames instead of text frames

Fixes #2

Author: isobit

README.md

@@ -10,6 +10,9 @@ instead of `STDIN` and `STDOUT`.
 ## Usage
 ```
 Usage: ws-tcp-relay <tcpTargetAddress>
+  -b	Use binary frames instead of text frames
+  -binary
+    	Use binary frames instead of text frames
   -p uint
     	The port to listen on (default 4223)
   -port uint
@@ -20,6 +23,10 @@ Usage: ws-tcp-relay <tcpTargetAddress>
     	TLS key file path
 ```
 
+### Binary Data
+By default, `golang.org/x/net/websocket` uses text frames to deliver payload
+data. To use binary frames instead, use either the `b` or `binary` flags.
+
 ### WSS Support
 To use secure WebSockets simply specify both the `tlscert` and `tlskey` flags.
 

ws-tcp-relay.go

@@ -13,6 +13,7 @@ import (
 )
 
 var tcpAddress string
+var binaryMode bool
 
 func copyWorker(dst io.Writer, src io.Reader, doneCh chan<- bool) {
 	io.Copy(dst, src)
@@ -26,6 +27,10 @@ func relayHandler(ws *websocket.Conn) {
 		return
 	}
 
+	if binaryMode {
+		ws.PayloadType = websocket.BinaryFrame
+	}
+
 	doneCh := make(chan bool)
 
 	go copyWorker(conn, ws, doneCh)
@@ -51,6 +56,8 @@ func main() {
 	flag.UintVar(&port, "port", 4223, "The port to listen on")
 	flag.StringVar(&certFile, "tlscert", "", "TLS cert file path")
 	flag.StringVar(&keyFile, "tlskey", "", "TLS key file path")
+	flag.BoolVar(&binaryMode, "b", false, "Use binary frames instead of text frames")
+	flag.BoolVar(&binaryMode, "binary", false, "Use binary frames instead of text frames")
 	flag.Usage = usage
 	flag.Parse()