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

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: Use base::StringPiece in some places not to copy data. Created 6 years, 6 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/server/http_server_request_info.h" 11 #include "net/server/http_server_request_info.h"
12 #include "net/server/http_server_response_info.h" 12 #include "net/server/http_server_response_info.h"
13 #include "net/socket/tcp_listen_socket.h" 13 #include "net/socket/tcp_server_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 net::TCPServerSocketFactory factory;
28 server_ = new net::HttpServer(factory, this); 28 factory.SetAddressAndPort("0.0.0.0", 0);
29 server_.reset(new net::HttpServer(factory.CreateAndListen(), this));
29 net::IPEndPoint address; 30 net::IPEndPoint address;
30 int error = server_->GetLocalAddress(&address); 31 int error = server_->GetLocalAddress(&address);
31 port_ = address.port(); 32 port_ = address.port();
32 return (error == net::OK); 33 return (error == net::OK);
33 } 34 }
34 35
35 void WebSocketServer::OnHttpRequest( 36 void WebSocketServer::OnHttpRequest(
36 int connection_id, 37 int connection_id,
37 const net::HttpServerRequestInfo& info) { 38 const net::HttpServerRequestInfo& info) {
38 server_->Send500(connection_id, "websockets protocol only"); 39 server_->Send500(connection_id, "websockets protocol only");
39 } 40 }
40 41
41 void WebSocketServer::OnWebSocketRequest( 42 void WebSocketServer::OnWebSocketRequest(
42 int connection_id, 43 int connection_id,
43 const net::HttpServerRequestInfo& info) { 44 const net::HttpServerRequestInfo& info) {
44 if (connection_id_ != kNotConnected) { 45 if (connection_id_ != kNotConnected) {
45 // Reject connection since we already have our client. 46 server_->Close(connection_id);
46 base::MessageLoop::current()->PostTask(
47 FROM_HERE,
48 base::Bind(&net::HttpServer::Close, server_, connection_id));
49 return; 47 return;
50 } 48 }
51 // Accept the connection. 49 // Accept the connection.
52 server_->AcceptWebSocket(connection_id, info); 50 server_->AcceptWebSocket(connection_id, info);
53 connection_id_ = connection_id; 51 connection_id_ = connection_id;
54 } 52 }
55 53
56 void WebSocketServer::OnWebSocketMessage( 54 void WebSocketServer::OnWebSocketMessage(
57 int connection_id, 55 int connection_id,
58 const std::string& data) { 56 const std::string& data) {
59 // TODO(cpu): remove this test code soon. 57 // TODO(cpu): remove this test code soon.
60 if (data == "\"hello\"") 58 if (data == "\"hello\"")
61 server_->SendOverWebSocket(connection_id, "\"hi there!\""); 59 server_->SendOverWebSocket(connection_id, "\"hi there!\"");
62 } 60 }
63 61
64 void WebSocketServer::OnClose( 62 void WebSocketServer::OnClose(
65 int connection_id) { 63 int connection_id) {
66 if (connection_id == connection_id_) 64 if (connection_id == connection_id_)
67 connection_id_ = kNotConnected; 65 connection_id_ = kNotConnected;
68 } 66 }
69 67
70 } // namespace spy 68 } // namespace spy
OLDNEW
« no previous file with comments | « mojo/spy/websocket_server.h ('k') | net/net.gypi » ('j') | net/server/http_connection.h » ('J')

Powered by Google App Engine
This is Rietveld 408576698