WebSocket implementation in WebKit

WebCore/websockets/WebSocketResponse.cpp

+/*
+ * Copyright (C) 2009 Google Inc.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *     * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *     * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *     * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "WebSocketResponse.h"
+
+#include "CString.h"
+#include "Logging.h"
+
+#undef LOG
+#define LOG(channel, ...) do { fprintf(stderr, __VA_ARGS__); fprintf(stderr, "\n"); } while (0);
+
+namespace {
+
+bool hadResponseHeader(const char *start, const char *end)
+{
+    for (const char *p = start; p && p < end; p++) {
+        p = static_cast<const char *>(memchr(p, '\r', end - p));
+        if (!p) {
+            return false;
+        }
+        if (p + 3 < end && *(p + 1) == '\n' && *(p + 2) == '\r' && *(p + 3) == '\n') {
+            return true;
+        }
+    }
+    return false;
+}
+
+}
+
+namespace WebCore {
+
+const char webSocketResponseHeader[] = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n"
+        "Upgrade: WebSocket\r\n"
+        "Connection: Upgrade\r\n";
+
+WebSocketResponse::WebSocketResponse()
+        : m_valid(false)
+{
+}
+
+WebSocketResponse::~WebSocketResponse()
+{
+}
+
+const char* WebSocketResponse::parseHeader(const char* start, const char* end, Vector<std::pair<String, String> >* headers)
+{
+    Vector<char> name;
+    Vector<char> value;
+    for (const char* p = start; p < end; p++) {
+        name.clear();
+        value.clear();
+        // LOG(Network, "parseHeader next %s", p);
+        for (; p < end; p++) {
+            switch (*p) {
+                case '\r':
+                    if (name.isEmpty()) {
+                        if (p + 1 < end && *(p + 1) == '\n')
+                            return p + 2;
+                        else {
+                            LOG(Network, "CF doesn't follow LF p=%p end=%p", p, end);
+                            return NULL;
+                        }
+                    }
+                    LOG(Network, "Unexpected CR in name");
+                    return NULL;
+                case '\n':
+                    LOG(Network, "Unexpected LF in name");
+                    return NULL;
+                case ':':
+                    break;
+                default:
+                    if (*p >= 0x41 && *p <= 0x5a) {
+                        name.append(*p + 0x20);
+                    } else {
+                        name.append(*p);
+                    }
+                    continue;
+            }
+            if (*p == ':') {
+                // LOG(Network, "colon found after name=%s", name.data());
+                ++p;
+                break;
+            }
+        }
+        // LOG(Network, "skip space: %s", p);
+        for (; p < end && *p == 0x20; p++)
+            ;
+        // LOG(Network, "parse value: %s", p);
+        for (; p < end; p++) {
+            switch (*p) {
+                case '\r':
+                    break;
+                case '\n':
+                    LOG(Network, "Unexpected LF in value");
+                    return NULL;
+                default:
+                    value.append(*p);
+            }
+            if (*p == '\r') {
+                // LOG(Network, "CR found after value=%s", value.data());
+                ++p;
+                break;
+            }
+        }
+        if (p >= end || *p != '\n') {
+            LOG(Network, "CR doesn't follow LF after value p=%p end=%p", p, end);
+            return NULL;
+        }
+        String name_str = String::fromUTF8(name.data(), name.size());
+        String value_str = String::fromUTF8(value.data(), value.size());
+        LOG(Network, "name=%s value=%s", name_str.utf8().data(), value_str.utf8().data());
+        headers->append(std::make_pair(name_str, value_str));
+    }
+    LOG(Network, "Unexpected end of header");
+    return NULL;
+}
+
+int WebSocketResponse::readHandshakeResponse(const char* header, int len)
+{
+    m_valid = false;
+    if (len < sizeof(webSocketResponseHeader)) {
+        LOG(Network, "short response header len=%d", len);
+        return 0;
+    }
+    if (memcmp(header, webSocketResponseHeader, sizeof(webSocketResponseHeader) - 1) != 0) {
+        LOG(Network, "Mismatch response header: %s", header);
+        return len;
+    }
+    const char *p = header + sizeof(webSocketResponseHeader) - 1;
+    const char *end = header + len + 1;
+    if (!hadResponseHeader(p, end)) {
+        LOG(Network, "incomplete response header len=%d", len);
+        return 0;
+    }
+    Vector<std::pair<String, String> > headers;
+    p = parseHeader(p, end, &headers);
+    if (!p) {
+        LOG(Network, "parseHeader failed");
+        return len;
+    }
+    // _Headers processing_
+    for (Vector<std::pair<String, String> >::iterator it = headers.begin(); it != headers.end(); ++it) {
+        if (it->first == "websocket-origin") {
+            if (!m_origin.isEmpty())
+                return len;
+            m_origin = it->second;
+        } else if (it->first == "websocket-location") {
+            if (!m_location.isEmpty())
+                return len;
+            m_location = it->second;
+        } else if (it->first == "websocket-protocol") {
+            if (!m_protocol.isEmpty())
+                return len;
+            m_protocol = it->second;
+        }
+    }
+    if (m_origin.isEmpty() || m_location.isEmpty())
+        return len;
+    m_valid = true;
+    return p - header;
+}
+
+bool WebSocketResponse::isValidHeader() const
+{
+    return m_valid;
+}
+
+void WebSocketResponse::setIsValidHeader(bool valid)
+{
+    m_valid = valid;
+}
+
+const String& WebSocketResponse::websocket_origin() const
+{
+    return m_origin;
+}
+
+void WebSocketResponse::setWebSocketOrigin(const String& websocket_origin)
+{
+    m_origin = websocket_origin;
+}
+
+const String& WebSocketResponse::websocket_location() const
+{
+    return m_location;
+}
+
+void WebSocketResponse::setWebSocketLocation(const String& websocket_location)
+{
+    m_location = websocket_location;
+}
+
+const String& WebSocketResponse::websocket_protocol() const
+{
+    return m_protocol;
+}
+
+void WebSocketResponse::setWebSocketProtocol(const String& websocket_protocol)
+{
+    m_protocol = websocket_protocol;
+}
+
+}  // namespace WebCore