websocket: use URL.RawPath to construct WebSocket-Location: header

R=rsc
CC=golang-dev
http://codereview.appspot.com/651041

Committer: Russ Cox <rsc@golang.org>

2 files changed, 18 insertions, 1 deletions

diff --git a/src/pkg/websocket/server.go b/src/pkg/websocket/server.go
index 0ccb31e8a..93d8b7afd 100644
--- a/src/pkg/websocket/server.go
+++ b/src/pkg/websocket/server.go
@@ -66,7 +66,7 @@ func (f Handler) ServeHTTP(c *http.Conn, req *http.Request) {
 		return
 	}
 	defer rwc.Close()
-	location := "ws://" + req.Host + req.URL.Path
+	location := "ws://" + req.Host + req.URL.RawPath
 
 	// TODO(ukai): verify origin,location,protocol.
 
diff --git a/src/pkg/websocket/websocket_test.go b/src/pkg/websocket/websocket_test.go
index 44fda8aaa..92582b1ef 100644
--- a/src/pkg/websocket/websocket_test.go
+++ b/src/pkg/websocket/websocket_test.go
@@ -60,6 +60,23 @@ func TestEcho(t *testing.T) {
 	ws.Close()
 }
 
+func TestWithQuery(t *testing.T) {
+	once.Do(startServer)
+
+	client, err := net.Dial("tcp", "", serverAddr)
+	if err != nil {
+		t.Fatal("dialing", err)
+	}
+
+	ws, err := newClient("/echo?q=v", "localhost", "http://localhost",
+		"ws://localhost/echo?q=v", "", client)
+	if err != nil {
+		t.Errorf("WebSocket handshake error", err)
+		return
+	}
+	ws.Close()
+}
+
 func TestHTTP(t *testing.T) {
 	once.Do(startServer)