| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef NET_WEBSOCKETS_WEBSOCKET_STREAM_H_ | |
| 6 #define NET_WEBSOCKETS_WEBSOCKET_STREAM_H_ | |
| 7 | |
| 8 #include <string> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/callback_forward.h" | |
| 13 #include "base/memory/ref_counted.h" | |
| 14 #include "base/memory/scoped_ptr.h" | |
| 15 #include "base/memory/scoped_vector.h" | |
| 16 #include "base/time/time.h" | |
| 17 #include "net/base/completion_callback.h" | |
| 18 #include "net/base/net_export.h" | |
| 19 #include "net/websockets/websocket_event_interface.h" | |
| 20 #include "net/websockets/websocket_handshake_request_info.h" | |
| 21 #include "net/websockets/websocket_handshake_response_info.h" | |
| 22 | |
| 23 class GURL; | |
| 24 | |
| 25 namespace url { | |
| 26 class Origin; | |
| 27 } // namespace url | |
| 28 | |
| 29 namespace net { | |
| 30 | |
| 31 class BoundNetLog; | |
| 32 class URLRequestContext; | |
| 33 struct WebSocketFrame; | |
| 34 | |
| 35 // WebSocketStreamRequest is the caller's handle to the process of creation of a | |
| 36 // WebSocketStream. Deleting the object before the OnSuccess or OnFailure | |
| 37 // callbacks are called will cancel the request (and neither callback will be | |
| 38 // called). After OnSuccess or OnFailure have been called, this object may be | |
| 39 // safely deleted without side-effects. | |
| 40 class NET_EXPORT_PRIVATE WebSocketStreamRequest { | |
| 41 public: | |
| 42 virtual ~WebSocketStreamRequest(); | |
| 43 }; | |
| 44 | |
| 45 // WebSocketStream is a transport-agnostic interface for reading and writing | |
| 46 // WebSocket frames. This class provides an abstraction for WebSocket streams | |
| 47 // based on various transport layers, such as normal WebSocket connections | |
| 48 // (WebSocket protocol upgraded from HTTP handshake), SPDY transports, or | |
| 49 // WebSocket connections with multiplexing extension. Subtypes of | |
| 50 // WebSocketStream are responsible for managing the underlying transport | |
| 51 // appropriately. | |
| 52 // | |
| 53 // All functions except Close() can be asynchronous. If an operation cannot | |
| 54 // be finished synchronously, the function returns ERR_IO_PENDING, and | |
| 55 // |callback| will be called when the operation is finished. Non-null |callback| | |
| 56 // must be provided to these functions. | |
| 57 | |
| 58 class NET_EXPORT_PRIVATE WebSocketStream { | |
| 59 public: | |
| 60 // A concrete object derived from ConnectDelegate is supplied by the caller to | |
| 61 // CreateAndConnectStream() to receive the result of the connection. | |
| 62 class NET_EXPORT_PRIVATE ConnectDelegate { | |
| 63 public: | |
| 64 virtual ~ConnectDelegate(); | |
| 65 // Called on successful connection. The parameter is an object derived from | |
| 66 // WebSocketStream. | |
| 67 virtual void OnSuccess(scoped_ptr<WebSocketStream> stream) = 0; | |
| 68 | |
| 69 // Called on failure to connect. | |
| 70 // |message| contains defails of the failure. | |
| 71 virtual void OnFailure(const std::string& message) = 0; | |
| 72 | |
| 73 // Called when the WebSocket Opening Handshake starts. | |
| 74 virtual void OnStartOpeningHandshake( | |
| 75 scoped_ptr<WebSocketHandshakeRequestInfo> request) = 0; | |
| 76 | |
| 77 // Called when the WebSocket Opening Handshake ends. | |
| 78 virtual void OnFinishOpeningHandshake( | |
| 79 scoped_ptr<WebSocketHandshakeResponseInfo> response) = 0; | |
| 80 | |
| 81 // Called when there is an SSL certificate error. Should call | |
| 82 // ssl_error_callbacks->ContinueSSLRequest() or | |
| 83 // ssl_error_callbacks->CancelSSLRequest(). | |
| 84 virtual void OnSSLCertificateError( | |
| 85 scoped_ptr<WebSocketEventInterface::SSLErrorCallbacks> | |
| 86 ssl_error_callbacks, | |
| 87 const SSLInfo& ssl_info, | |
| 88 bool fatal) = 0; | |
| 89 }; | |
| 90 | |
| 91 // Create and connect a WebSocketStream of an appropriate type. The actual | |
| 92 // concrete type returned depends on whether multiplexing or SPDY are being | |
| 93 // used to communicate with the remote server. If the handshake completed | |
| 94 // successfully, then connect_delegate->OnSuccess() is called with a | |
| 95 // WebSocketStream instance. If it failed, then connect_delegate->OnFailure() | |
| 96 // is called with a WebSocket result code corresponding to the error. Deleting | |
| 97 // the returned WebSocketStreamRequest object will cancel the connection, in | |
| 98 // which case the |connect_delegate| object that the caller passed will be | |
| 99 // deleted without any of its methods being called. Unless cancellation is | |
| 100 // required, the caller should keep the WebSocketStreamRequest object alive | |
| 101 // until connect_delegate->OnSuccess() or OnFailure() have been called, then | |
| 102 // it is safe to delete. | |
| 103 static scoped_ptr<WebSocketStreamRequest> CreateAndConnectStream( | |
| 104 const GURL& socket_url, | |
| 105 const std::vector<std::string>& requested_subprotocols, | |
| 106 const url::Origin& origin, | |
| 107 URLRequestContext* url_request_context, | |
| 108 const BoundNetLog& net_log, | |
| 109 scoped_ptr<ConnectDelegate> connect_delegate); | |
| 110 | |
| 111 // Derived classes must make sure Close() is called when the stream is not | |
| 112 // closed on destruction. | |
| 113 virtual ~WebSocketStream(); | |
| 114 | |
| 115 // Reads WebSocket frame data. This operation finishes when new frame data | |
| 116 // becomes available. | |
| 117 // | |
| 118 // |frames| remains owned by the caller and must be valid until the | |
| 119 // operation completes or Close() is called. |frames| must be empty on | |
| 120 // calling. | |
| 121 // | |
| 122 // This function should not be called while the previous call of ReadFrames() | |
| 123 // is still pending. | |
| 124 // | |
| 125 // Returns net::OK or one of the net::ERR_* codes. | |
| 126 // | |
| 127 // frames->size() >= 1 if the result is OK. | |
| 128 // | |
| 129 // Only frames with complete header information are inserted into |frames|. If | |
| 130 // the currently available bytes of a new frame do not form a complete frame | |
| 131 // header, then the implementation will buffer them until all the fields in | |
| 132 // the WebSocketFrameHeader object can be filled. If ReadFrames() is freshly | |
| 133 // called in this situation, it will return ERR_IO_PENDING exactly as if no | |
| 134 // data was available. | |
| 135 // | |
| 136 // Original frame boundaries are not preserved. In particular, if only part of | |
| 137 // a frame is available, then the frame will be split, and the available data | |
| 138 // will be returned immediately. | |
| 139 // | |
| 140 // When the socket is closed on the remote side, this method will return | |
| 141 // ERR_CONNECTION_CLOSED. It will not return OK with an empty vector. | |
| 142 // | |
| 143 // If the connection is closed in the middle of receiving an incomplete frame, | |
| 144 // ReadFrames may discard the incomplete frame. Since the renderer will | |
| 145 // discard any incomplete messages when the connection is closed, this makes | |
| 146 // no difference to the overall semantics. | |
| 147 // | |
| 148 // Implementations of ReadFrames() must be able to handle being deleted during | |
| 149 // the execution of callback.Run(). In practice this means that the method | |
| 150 // calling callback.Run() (and any calling methods in the same object) must | |
| 151 // return immediately without any further method calls or access to member | |
| 152 // variables. Implementors should write test(s) for this case. | |
| 153 // | |
| 154 // Extensions which use reserved header bits should clear them when they are | |
| 155 // set correctly. If the reserved header bits are set incorrectly, it is okay | |
| 156 // to leave it to the caller to report the error. | |
| 157 virtual int ReadFrames(ScopedVector<WebSocketFrame>* frames, | |
| 158 const CompletionCallback& callback) = 0; | |
| 159 | |
| 160 // Writes WebSocket frame data. | |
| 161 // | |
| 162 // |frames| must be valid until the operation completes or Close() is called. | |
| 163 // | |
| 164 // This function must not be called while a previous call of WriteFrames() is | |
| 165 // still pending. | |
| 166 // | |
| 167 // This method will only return OK if all frames were written completely. | |
| 168 // Otherwise it will return an appropriate net error code. | |
| 169 // | |
| 170 // The callback implementation is permitted to delete this | |
| 171 // object. Implementations of WriteFrames() should be robust against | |
| 172 // this. This generally means returning to the event loop immediately after | |
| 173 // calling the callback. | |
| 174 virtual int WriteFrames(ScopedVector<WebSocketFrame>* frames, | |
| 175 const CompletionCallback& callback) = 0; | |
| 176 | |
| 177 // Closes the stream. All pending I/O operations (if any) are cancelled | |
| 178 // at this point, so |frames| can be freed. | |
| 179 virtual void Close() = 0; | |
| 180 | |
| 181 // The subprotocol that was negotiated for the stream. If no protocol was | |
| 182 // negotiated, then the empty string is returned. | |
| 183 virtual std::string GetSubProtocol() const = 0; | |
| 184 | |
| 185 // The extensions that were negotiated for the stream. Since WebSocketStreams | |
| 186 // can be layered, this may be different from what this particular | |
| 187 // WebSocketStream implements. The primary purpose of this accessor is to make | |
| 188 // the data available to Javascript. The format of the string is identical to | |
| 189 // the contents of the Sec-WebSocket-Extensions header supplied by the server, | |
| 190 // with some canonicalisations applied (leading and trailing whitespace | |
| 191 // removed, multiple headers concatenated into one comma-separated list). See | |
| 192 // RFC6455 section 9.1 for the exact format specification. If no | |
| 193 // extensions were negotiated, the empty string is returned. | |
| 194 virtual std::string GetExtensions() const = 0; | |
| 195 | |
| 196 protected: | |
| 197 WebSocketStream(); | |
| 198 | |
| 199 private: | |
| 200 DISALLOW_COPY_AND_ASSIGN(WebSocketStream); | |
| 201 }; | |
| 202 | |
| 203 // A helper function used in the implementation of CreateAndConnectStream() and | |
| 204 // WebSocketBasicHandshakeStream. It creates a WebSocketHandshakeResponseInfo | |
| 205 // object and dispatches it to the OnFinishOpeningHandshake() method of the | |
| 206 // supplied |connect_delegate|. | |
| 207 void WebSocketDispatchOnFinishOpeningHandshake( | |
| 208 WebSocketStream::ConnectDelegate* connect_delegate, | |
| 209 const GURL& gurl, | |
| 210 const scoped_refptr<HttpResponseHeaders>& headers, | |
| 211 base::Time response_time); | |
| 212 | |
| 213 } // namespace net | |
| 214 | |
| 215 #endif // NET_WEBSOCKETS_WEBSOCKET_STREAM_H_ | |
| OLD | NEW |