OLD | NEW |
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 #ifndef NET_SERVER_WEB_SOCKET_ENCODER_H_ | 5 #ifndef NET_SERVER_WEB_SOCKET_ENCODER_H_ |
6 #define NET_SERVER_WEB_SOCKET_ENCODER_H_ | 6 #define NET_SERVER_WEB_SOCKET_ENCODER_H_ |
7 | 7 |
| 8 #include <memory> |
8 #include <string> | 9 #include <string> |
9 | 10 |
10 #include "base/macros.h" | 11 #include "base/macros.h" |
11 #include "base/memory/scoped_ptr.h" | |
12 #include "base/strings/string_piece.h" | 12 #include "base/strings/string_piece.h" |
13 #include "net/server/web_socket.h" | 13 #include "net/server/web_socket.h" |
14 #include "net/websockets/websocket_deflater.h" | 14 #include "net/websockets/websocket_deflater.h" |
15 #include "net/websockets/websocket_inflater.h" | 15 #include "net/websockets/websocket_inflater.h" |
16 | 16 |
17 namespace net { | 17 namespace net { |
18 | 18 |
19 class WebSocketDeflateParameters; | 19 class WebSocketDeflateParameters; |
20 | 20 |
21 class WebSocketEncoder final { | 21 class WebSocketEncoder final { |
22 public: | 22 public: |
23 static const char kClientExtensions[]; | 23 static const char kClientExtensions[]; |
24 | 24 |
25 ~WebSocketEncoder(); | 25 ~WebSocketEncoder(); |
26 | 26 |
27 // Creates and returns an encoder for a server without extensions. | 27 // Creates and returns an encoder for a server without extensions. |
28 static scoped_ptr<WebSocketEncoder> CreateServer(); | 28 static std::unique_ptr<WebSocketEncoder> CreateServer(); |
29 // Creates and returns an encoder. | 29 // Creates and returns an encoder. |
30 // |extensions| is the value of a Sec-WebSocket-Extensions header. | 30 // |extensions| is the value of a Sec-WebSocket-Extensions header. |
31 // Returns nullptr when there is an error. | 31 // Returns nullptr when there is an error. |
32 static scoped_ptr<WebSocketEncoder> CreateServer( | 32 static std::unique_ptr<WebSocketEncoder> CreateServer( |
33 const std::string& extensions, | 33 const std::string& extensions, |
34 WebSocketDeflateParameters* params); | 34 WebSocketDeflateParameters* params); |
35 static scoped_ptr<WebSocketEncoder> CreateClient( | 35 static std::unique_ptr<WebSocketEncoder> CreateClient( |
36 const std::string& response_extensions); | 36 const std::string& response_extensions); |
37 | 37 |
38 WebSocket::ParseResult DecodeFrame(const base::StringPiece& frame, | 38 WebSocket::ParseResult DecodeFrame(const base::StringPiece& frame, |
39 int* bytes_consumed, | 39 int* bytes_consumed, |
40 std::string* output); | 40 std::string* output); |
41 void EncodeFrame(const std::string& frame, | 41 void EncodeFrame(const std::string& frame, |
42 int masking_key, | 42 int masking_key, |
43 std::string* output); | 43 std::string* output); |
44 | 44 |
45 bool deflate_enabled() const { return !!deflater_; } | 45 bool deflate_enabled() const { return !!deflater_; } |
46 | 46 |
47 private: | 47 private: |
48 enum Type { | 48 enum Type { |
49 FOR_SERVER, | 49 FOR_SERVER, |
50 FOR_CLIENT, | 50 FOR_CLIENT, |
51 }; | 51 }; |
52 | 52 |
53 WebSocketEncoder(Type type, | 53 WebSocketEncoder(Type type, |
54 scoped_ptr<WebSocketDeflater> deflater, | 54 std::unique_ptr<WebSocketDeflater> deflater, |
55 scoped_ptr<WebSocketInflater> inflater); | 55 std::unique_ptr<WebSocketInflater> inflater); |
56 | 56 |
57 bool Inflate(std::string* message); | 57 bool Inflate(std::string* message); |
58 bool Deflate(const std::string& message, std::string* output); | 58 bool Deflate(const std::string& message, std::string* output); |
59 | 59 |
60 Type type_; | 60 Type type_; |
61 scoped_ptr<WebSocketDeflater> deflater_; | 61 std::unique_ptr<WebSocketDeflater> deflater_; |
62 scoped_ptr<WebSocketInflater> inflater_; | 62 std::unique_ptr<WebSocketInflater> inflater_; |
63 | 63 |
64 DISALLOW_COPY_AND_ASSIGN(WebSocketEncoder); | 64 DISALLOW_COPY_AND_ASSIGN(WebSocketEncoder); |
65 }; | 65 }; |
66 | 66 |
67 } // namespace net | 67 } // namespace net |
68 | 68 |
69 #endif // NET_SERVER_WEB_SOCKET_ENCODER_H_ | 69 #endif // NET_SERVER_WEB_SOCKET_ENCODER_H_ |
OLD | NEW |