| 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 |