OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #ifndef NET_SERVER_WEB_SOCKET_H_ | 5 #ifndef NET_SERVER_WEB_SOCKET_H_ |
6 #define NET_SERVER_WEB_SOCKET_H_ | 6 #define NET_SERVER_WEB_SOCKET_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/strings/string_piece.h" | |
12 | 11 |
13 namespace net { | 12 namespace net { |
14 | 13 |
15 class HttpConnection; | 14 class HttpConnection; |
16 class HttpServer; | |
17 class HttpServerRequestInfo; | 15 class HttpServerRequestInfo; |
18 | 16 |
19 class WebSocket { | 17 class WebSocket { |
20 public: | 18 public: |
21 enum ParseResult { | 19 enum ParseResult { |
22 FRAME_OK, | 20 FRAME_OK, |
23 FRAME_INCOMPLETE, | 21 FRAME_INCOMPLETE, |
24 FRAME_CLOSE, | 22 FRAME_CLOSE, |
25 FRAME_ERROR | 23 FRAME_ERROR |
26 }; | 24 }; |
27 | 25 |
28 static WebSocket* CreateWebSocket(HttpServer* server, | 26 static WebSocket* CreateWebSocket(HttpConnection* connection, |
29 HttpConnection* connection, | |
30 const HttpServerRequestInfo& request, | 27 const HttpServerRequestInfo& request, |
31 size_t* pos); | 28 size_t* pos); |
32 | 29 |
33 static ParseResult DecodeFrameHybi17(const base::StringPiece& frame, | 30 static ParseResult DecodeFrameHybi17(const std::string& frame, |
34 bool client_frame, | 31 bool client_frame, |
35 int* bytes_consumed, | 32 int* bytes_consumed, |
36 std::string* output); | 33 std::string* output); |
37 | 34 |
38 static std::string EncodeFrameHybi17(const std::string& data, | 35 static std::string EncodeFrameHybi17(const std::string& data, |
39 int masking_key); | 36 int masking_key); |
40 | 37 |
41 virtual void Accept(const HttpServerRequestInfo& request) = 0; | 38 virtual void Accept(const HttpServerRequestInfo& request) = 0; |
42 virtual ParseResult Read(std::string* message) = 0; | 39 virtual ParseResult Read(std::string* message) = 0; |
43 virtual void Send(const std::string& message) = 0; | 40 virtual void Send(const std::string& message) = 0; |
44 virtual ~WebSocket() {} | 41 virtual ~WebSocket() {} |
45 | 42 |
46 protected: | 43 protected: |
47 WebSocket(HttpServer* server, HttpConnection* connection); | 44 explicit WebSocket(HttpConnection* connection); |
48 | 45 HttpConnection* connection_; |
49 HttpServer* const server_; | |
50 HttpConnection* const connection_; | |
51 | 46 |
52 private: | 47 private: |
53 DISALLOW_COPY_AND_ASSIGN(WebSocket); | 48 DISALLOW_COPY_AND_ASSIGN(WebSocket); |
54 }; | 49 }; |
55 | 50 |
56 } // namespace net | 51 } // namespace net |
57 | 52 |
58 #endif // NET_SERVER_WEB_SOCKET_H_ | 53 #endif // NET_SERVER_WEB_SOCKET_H_ |
OLD | NEW |