1 files changed, 70 insertions, 0 deletions
diff --git a/chromium/mojo/spy/websocket_server.cc b/chromium/mojo/spy/websocket_server.cc
new file mode 100644
index 00000000000..649a135c13e
--- /dev/null
+++ b/chromium/mojo/spy/websocket_server.cc
@@ -0,0 +1,70 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "mojo/spy/websocket_server.h"
+
+#include "base/bind.h"
+#include "base/strings/stringprintf.h"
+#include "net/base/ip_endpoint.h"
+#include "net/base/net_errors.h"
+#include "net/server/http_server_request_info.h"
+#include "net/server/http_server_response_info.h"
+#include "net/socket/tcp_listen_socket.h"
+
+namespace spy {
+
+const int kNotConnected = -1;
+
+WebSocketServer::WebSocketServer(int port)
+    : port_(port), connection_id_(kNotConnected) {
+}
+
+WebSocketServer::~WebSocketServer() {
+}
+
+bool WebSocketServer::Start() {
+  net::TCPListenSocketFactory factory("0.0.0.0", port_);
+  server_ = new net::HttpServer(factory, this);
+  net::IPEndPoint address;
+  int error = server_->GetLocalAddress(&address);
+  port_ = address.port();
+  return (error == net::OK);
+}
+
+void WebSocketServer::OnHttpRequest(
+    int connection_id,
+    const net::HttpServerRequestInfo& info) {
+  server_->Send500(connection_id, "websockets protocol only");
+}
+
+void WebSocketServer::OnWebSocketRequest(
+    int connection_id,
+    const net::HttpServerRequestInfo& info) {
+  if (connection_id_ != kNotConnected) {
+    // Reject connection since we already have our client.
+    base::MessageLoop::current()->PostTask(
+        FROM_HERE,
+        base::Bind(&net::HttpServer::Close, server_, connection_id));
+    return;
+  }
+  // Accept the connection.
+  server_->AcceptWebSocket(connection_id, info);
+  connection_id_ = connection_id;
+}
+
+void WebSocketServer::OnWebSocketMessage(
+    int connection_id,
+    const std::string& data) {
+  // TODO(cpu): remove this test code soon.
+  if (data == "\"hello\"")
+    server_->SendOverWebSocket(connection_id, "\"hi there!\"");
+}
+
+void WebSocketServer::OnClose(
+    int connection_id) {
+  if (connection_id == connection_id_)
+    connection_id_ = kNotConnected;
+}
+
+}  // namespace spy