diff options
Diffstat (limited to 'src/pkg/websocket/server.go')
| -rw-r--r-- | src/pkg/websocket/server.go | 203 |
1 files changed, 37 insertions, 166 deletions
diff --git a/src/pkg/websocket/server.go b/src/pkg/websocket/server.go index e0e7c872d..8f6a6a94f 100644 --- a/src/pkg/websocket/server.go +++ b/src/pkg/websocket/server.go @@ -5,11 +5,41 @@ package websocket import ( + "bufio" + "fmt" "http" "io" - "strings" + "os" ) +func newServerConn(rwc io.ReadWriteCloser, buf *bufio.ReadWriter, req *http.Request) (conn *Conn, err os.Error) { + config := new(Config) + var hs serverHandshaker = &hybiServerHandshaker{Config: config} + code, err := hs.ReadHandshake(buf.Reader, req) + if err != nil { + hs = &hixie76ServerHandshaker{Config: config} + code, err = hs.ReadHandshake(buf.Reader, req) + } + if err != nil { + hs = &hixie75ServerHandshaker{Config: config} + code, err = hs.ReadHandshake(buf.Reader, req) + } + if err != nil { + fmt.Fprintf(buf, "HTTP/1.1 %03d %s\r\n", code, http.StatusText(code)) + buf.WriteString("\r\n") + buf.WriteString(err.String()) + return + } + config.Protocol = nil + + err = hs.AcceptHandshake(buf.Writer) + if err != nil { + return + } + conn = hs.NewServerConn(buf, rwc, req) + return +} + /* Handler is an interface to a WebSocket. @@ -23,7 +53,7 @@ A trivial example server: "websocket" ) - // Echo the data received on the Web Socket. + // Echo the data received on the WebSocket. func EchoServer(ws *websocket.Conn) { io.Copy(ws, ws); } @@ -38,26 +68,8 @@ A trivial example server: */ type Handler func(*Conn) -/* -Gets key number from Sec-WebSocket-Key<n>: field as described -in 5.2 Sending the server's opening handshake, 4. -*/ -func getKeyNumber(s string) (r uint32) { - // 4. Let /key-number_n/ be the digits (characters in the range - // U+0030 DIGIT ZERO (0) to U+0039 DIGIT NINE (9)) in /key_1/, - // interpreted as a base ten integer, ignoring all other characters - // in /key_n/. - r = 0 - for i := 0; i < len(s); i++ { - if s[i] >= '0' && s[i] <= '9' { - r = r*10 + uint32(s[i]) - '0' - } - } - return -} - // ServeHTTP implements the http.Handler interface for a Web Socket -func (f Handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { +func (h Handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { rwc, buf, err := w.(http.Hijacker).Hijack() if err != nil { panic("Hijack failed: " + err.String()) @@ -67,153 +79,12 @@ func (f Handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { // the client did not send a handshake that matches with protocol // specification. defer rwc.Close() - - if req.Method != "GET" { - return - } - // HTTP version can be safely ignored. - - if strings.ToLower(req.Header.Get("Upgrade")) != "websocket" || - strings.ToLower(req.Header.Get("Connection")) != "upgrade" { - return - } - - // TODO(ukai): check Host - origin := req.Header.Get("Origin") - if origin == "" { - return - } - - key1 := req.Header.Get("Sec-Websocket-Key1") - if key1 == "" { - return - } - key2 := req.Header.Get("Sec-Websocket-Key2") - if key2 == "" { - return - } - key3 := make([]byte, 8) - if _, err := io.ReadFull(buf, key3); err != nil { - return - } - - var location string - if req.TLS != nil { - location = "wss://" + req.Host + req.URL.RawPath - } else { - location = "ws://" + req.Host + req.URL.RawPath - } - - // Step 4. get key number in Sec-WebSocket-Key<n> fields. - keyNumber1 := getKeyNumber(key1) - keyNumber2 := getKeyNumber(key2) - - // Step 5. get number of spaces in Sec-WebSocket-Key<n> fields. - space1 := uint32(strings.Count(key1, " ")) - space2 := uint32(strings.Count(key2, " ")) - if space1 == 0 || space2 == 0 { - return - } - - // Step 6. key number must be an integral multiple of spaces. - if keyNumber1%space1 != 0 || keyNumber2%space2 != 0 { - return - } - - // Step 7. let part be key number divided by spaces. - part1 := keyNumber1 / space1 - part2 := keyNumber2 / space2 - - // Step 8. let challenge be concatenation of part1, part2 and key3. - // Step 9. get MD5 fingerprint of challenge. - response, err := getChallengeResponse(part1, part2, key3) + conn, err := newServerConn(rwc, buf, req) if err != nil { return } - - // Step 10. send response status line. - buf.WriteString("HTTP/1.1 101 WebSocket Protocol Handshake\r\n") - // Step 11. send response headers. - buf.WriteString("Upgrade: WebSocket\r\n") - buf.WriteString("Connection: Upgrade\r\n") - buf.WriteString("Sec-WebSocket-Location: " + location + "\r\n") - buf.WriteString("Sec-WebSocket-Origin: " + origin + "\r\n") - protocol := strings.TrimSpace(req.Header.Get("Sec-Websocket-Protocol")) - if protocol != "" { - buf.WriteString("Sec-WebSocket-Protocol: " + protocol + "\r\n") - } - // Step 12. send CRLF. - buf.WriteString("\r\n") - // Step 13. send response data. - buf.Write(response) - if err := buf.Flush(); err != nil { - return - } - ws := newConn(origin, location, protocol, buf, rwc) - ws.Request = req - f(ws) -} - -/* -Draft75Handler is an interface to a WebSocket based on the -(soon obsolete) draft-hixie-thewebsocketprotocol-75. -*/ -type Draft75Handler func(*Conn) - -// ServeHTTP implements the http.Handler interface for a Web Socket. -func (f Draft75Handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { - if req.Method != "GET" || req.Proto != "HTTP/1.1" { - w.WriteHeader(http.StatusBadRequest) - io.WriteString(w, "Unexpected request") - return - } - if req.Header.Get("Upgrade") != "WebSocket" { - w.WriteHeader(http.StatusBadRequest) - io.WriteString(w, "missing Upgrade: WebSocket header") - return - } - if req.Header.Get("Connection") != "Upgrade" { - w.WriteHeader(http.StatusBadRequest) - io.WriteString(w, "missing Connection: Upgrade header") - return - } - origin := strings.TrimSpace(req.Header.Get("Origin")) - if origin == "" { - w.WriteHeader(http.StatusBadRequest) - io.WriteString(w, "missing Origin header") - return - } - - rwc, buf, err := w.(http.Hijacker).Hijack() - if err != nil { - panic("Hijack failed: " + err.String()) - return - } - defer rwc.Close() - - var location string - if req.TLS != nil { - location = "wss://" + req.Host + req.URL.RawPath - } else { - location = "ws://" + req.Host + req.URL.RawPath - } - - // TODO(ukai): verify origin,location,protocol. - - buf.WriteString("HTTP/1.1 101 Web Socket Protocol Handshake\r\n") - buf.WriteString("Upgrade: WebSocket\r\n") - buf.WriteString("Connection: Upgrade\r\n") - buf.WriteString("WebSocket-Origin: " + origin + "\r\n") - buf.WriteString("WebSocket-Location: " + location + "\r\n") - protocol := strings.TrimSpace(req.Header.Get("Websocket-Protocol")) - // canonical header key of WebSocket-Protocol. - if protocol != "" { - buf.WriteString("WebSocket-Protocol: " + protocol + "\r\n") - } - buf.WriteString("\r\n") - if err := buf.Flush(); err != nil { - return + if conn == nil { + panic("unepxected nil conn") } - ws := newConn(origin, location, protocol, buf, rwc) - f(ws) + h(conn) } |