Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(238)

Side by Side Diff: mojo/spy/websocket_server.cc

Issue 296053012: Replace StreamListenSocket with StreamSocket in HttpServer. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2014 The Chromium Authors. All rights reserved. 1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "mojo/spy/websocket_server.h" 5 #include "mojo/spy/websocket_server.h"
6 6
7 #include "base/bind.h" 7 #include "base/bind.h"
8 #include "base/strings/stringprintf.h" 8 #include "base/strings/stringprintf.h"
9 #include "net/base/ip_endpoint.h" 9 #include "net/base/ip_endpoint.h"
10 #include "net/base/net_errors.h" 10 #include "net/base/net_errors.h"
11 #include "net/base/net_util.h"
11 #include "net/server/http_server_request_info.h" 12 #include "net/server/http_server_request_info.h"
12 #include "net/server/http_server_response_info.h" 13 #include "net/server/http_server_response_info.h"
13 #include "net/socket/tcp_listen_socket.h"
14 14
15 namespace spy { 15 namespace spy {
16 16
17 const int kNotConnected = -1; 17 const int kNotConnected = -1;
18 18
19 WebSocketServer::WebSocketServer(int port) 19 WebSocketServer::WebSocketServer(int port)
20 : port_(port), connection_id_(kNotConnected) { 20 : port_(port), connection_id_(kNotConnected) {
21 } 21 }
22 22
23 WebSocketServer::~WebSocketServer() { 23 WebSocketServer::~WebSocketServer() {
24 } 24 }
25 25
26 bool WebSocketServer::Start() { 26 bool WebSocketServer::Start() {
27 net::TCPListenSocketFactory factory("0.0.0.0", port_); 27 server_.reset(net::HttpServer::Create("0.0.0.0", 0, this));
28 server_ = new net::HttpServer(factory, this); 28 if (!server_) {
29 return false;
30 }
29 net::IPEndPoint address; 31 net::IPEndPoint address;
30 int error = server_->GetLocalAddress(&address); 32 int error = server_->GetLocalAddress(&address);
31 port_ = address.port(); 33 port_ = address.port();
32 return (error == net::OK); 34 return (error == net::OK);
33 } 35 }
34 36
35 void WebSocketServer::OnHttpRequest( 37 void WebSocketServer::OnHttpRequest(
36 int connection_id, 38 int connection_id,
37 const net::HttpServerRequestInfo& info) { 39 const net::HttpServerRequestInfo& info) {
38 server_->Send500(connection_id, "websockets protocol only"); 40 server_->Send500(connection_id, "websockets protocol only");
39 } 41 }
40 42
41 void WebSocketServer::OnWebSocketRequest( 43 void WebSocketServer::OnWebSocketRequest(
42 int connection_id, 44 int connection_id,
43 const net::HttpServerRequestInfo& info) { 45 const net::HttpServerRequestInfo& info) {
44 if (connection_id_ != kNotConnected) { 46 if (connection_id_ != kNotConnected) {
45 // Reject connection since we already have our client. 47 // Reject connection since we already have our client.
46 base::MessageLoop::current()->PostTask( 48 base::MessageLoop::current()->PostTask(
47 FROM_HERE, 49 FROM_HERE,
48 base::Bind(&net::HttpServer::Close, server_, connection_id)); 50 base::Bind(&net::HttpServer::Close, server_->AsWeakPtr(),
51 connection_id));
49 return; 52 return;
50 } 53 }
51 // Accept the connection. 54 // Accept the connection.
52 server_->AcceptWebSocket(connection_id, info); 55 server_->AcceptWebSocket(connection_id, info);
53 connection_id_ = connection_id; 56 connection_id_ = connection_id;
54 } 57 }
55 58
56 void WebSocketServer::OnWebSocketMessage( 59 void WebSocketServer::OnWebSocketMessage(
57 int connection_id, 60 int connection_id,
58 const std::string& data) { 61 const std::string& data) {
59 // TODO(cpu): remove this test code soon. 62 // TODO(cpu): remove this test code soon.
60 if (data == "\"hello\"") 63 if (data == "\"hello\"")
61 server_->SendOverWebSocket(connection_id, "\"hi there!\""); 64 server_->SendOverWebSocket(connection_id, "\"hi there!\"");
62 } 65 }
63 66
64 void WebSocketServer::OnClose( 67 void WebSocketServer::OnClose(
65 int connection_id) { 68 int connection_id) {
66 if (connection_id == connection_id_) 69 if (connection_id == connection_id_)
67 connection_id_ = kNotConnected; 70 connection_id_ = kNotConnected;
68 } 71 }
69 72
70 } // namespace spy 73 } // namespace spy
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698