| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 #include "net/websockets/websocket_stream_create_test_base.h" | |
| 6 | |
| 7 #include "base/callback.h" | |
| 8 #include "net/http/http_request_headers.h" | |
| 9 #include "net/http/http_response_headers.h" | |
| 10 #include "net/websockets/websocket_basic_handshake_stream.h" | |
| 11 #include "net/websockets/websocket_handshake_request_info.h" | |
| 12 #include "net/websockets/websocket_handshake_response_info.h" | |
| 13 #include "net/websockets/websocket_handshake_stream_create_helper.h" | |
| 14 #include "net/websockets/websocket_stream.h" | |
| 15 #include "url/gurl.h" | |
| 16 #include "url/origin.h" | |
| 17 | |
| 18 namespace net { | |
| 19 | |
| 20 using HeaderKeyValuePair = WebSocketStreamCreateTestBase::HeaderKeyValuePair; | |
| 21 | |
| 22 // A sub-class of WebSocketHandshakeStreamCreateHelper which always sets a | |
| 23 // deterministic key to use in the WebSocket handshake. | |
| 24 class DeterministicKeyWebSocketHandshakeStreamCreateHelper | |
| 25 : public WebSocketHandshakeStreamCreateHelper { | |
| 26 public: | |
| 27 DeterministicKeyWebSocketHandshakeStreamCreateHelper( | |
| 28 WebSocketStream::ConnectDelegate* connect_delegate, | |
| 29 const std::vector<std::string>& requested_subprotocols) | |
| 30 : WebSocketHandshakeStreamCreateHelper(connect_delegate, | |
| 31 requested_subprotocols) {} | |
| 32 | |
| 33 void OnStreamCreated(WebSocketBasicHandshakeStream* stream) override { | |
| 34 stream->SetWebSocketKeyForTesting("dGhlIHNhbXBsZSBub25jZQ=="); | |
| 35 } | |
| 36 | |
| 37 private: | |
| 38 DISALLOW_COPY_AND_ASSIGN( | |
| 39 DeterministicKeyWebSocketHandshakeStreamCreateHelper); | |
| 40 }; | |
| 41 | |
| 42 class WebSocketStreamCreateTestBase::TestConnectDelegate | |
| 43 : public WebSocketStream::ConnectDelegate { | |
| 44 public: | |
| 45 TestConnectDelegate(WebSocketStreamCreateTestBase* owner, | |
| 46 const base::Closure& done_callback) | |
| 47 : owner_(owner), done_callback_(done_callback) {} | |
| 48 | |
| 49 void OnSuccess(scoped_ptr<WebSocketStream> stream) override { | |
| 50 stream.swap(owner_->stream_); | |
| 51 done_callback_.Run(); | |
| 52 } | |
| 53 | |
| 54 void OnFailure(const std::string& message) override { | |
| 55 owner_->has_failed_ = true; | |
| 56 owner_->failure_message_ = message; | |
| 57 done_callback_.Run(); | |
| 58 } | |
| 59 | |
| 60 void OnStartOpeningHandshake( | |
| 61 scoped_ptr<WebSocketHandshakeRequestInfo> request) override { | |
| 62 // Can be called multiple times (in the case of HTTP auth). Last call | |
| 63 // wins. | |
| 64 owner_->request_info_ = request.Pass(); | |
| 65 } | |
| 66 | |
| 67 void OnFinishOpeningHandshake( | |
| 68 scoped_ptr<WebSocketHandshakeResponseInfo> response) override { | |
| 69 if (owner_->response_info_) | |
| 70 ADD_FAILURE(); | |
| 71 owner_->response_info_ = response.Pass(); | |
| 72 } | |
| 73 | |
| 74 void OnSSLCertificateError( | |
| 75 scoped_ptr<WebSocketEventInterface::SSLErrorCallbacks> | |
| 76 ssl_error_callbacks, | |
| 77 const SSLInfo& ssl_info, | |
| 78 bool fatal) override { | |
| 79 owner_->ssl_error_callbacks_ = ssl_error_callbacks.Pass(); | |
| 80 owner_->ssl_info_ = ssl_info; | |
| 81 owner_->ssl_fatal_ = fatal; | |
| 82 } | |
| 83 | |
| 84 private: | |
| 85 WebSocketStreamCreateTestBase* owner_; | |
| 86 base::Closure done_callback_; | |
| 87 DISALLOW_COPY_AND_ASSIGN(TestConnectDelegate); | |
| 88 }; | |
| 89 | |
| 90 WebSocketStreamCreateTestBase::WebSocketStreamCreateTestBase() | |
| 91 : has_failed_(false), ssl_fatal_(false) { | |
| 92 } | |
| 93 | |
| 94 WebSocketStreamCreateTestBase::~WebSocketStreamCreateTestBase() { | |
| 95 } | |
| 96 | |
| 97 void WebSocketStreamCreateTestBase::CreateAndConnectStream( | |
| 98 const std::string& socket_url, | |
| 99 const std::vector<std::string>& sub_protocols, | |
| 100 const std::string& origin, | |
| 101 scoped_ptr<base::Timer> timer) { | |
| 102 for (size_t i = 0; i < ssl_data_.size(); ++i) { | |
| 103 scoped_ptr<SSLSocketDataProvider> ssl_data(ssl_data_[i]); | |
| 104 url_request_context_host_.AddSSLSocketDataProvider(ssl_data.Pass()); | |
| 105 } | |
| 106 ssl_data_.weak_clear(); | |
| 107 scoped_ptr<WebSocketStream::ConnectDelegate> connect_delegate( | |
| 108 new TestConnectDelegate(this, connect_run_loop_.QuitClosure())); | |
| 109 WebSocketStream::ConnectDelegate* delegate = connect_delegate.get(); | |
| 110 scoped_ptr<WebSocketHandshakeStreamCreateHelper> create_helper( | |
| 111 new DeterministicKeyWebSocketHandshakeStreamCreateHelper(delegate, | |
| 112 sub_protocols)); | |
| 113 stream_request_ = CreateAndConnectStreamForTesting( | |
| 114 GURL(socket_url), create_helper.Pass(), url::Origin(origin), | |
| 115 url_request_context_host_.GetURLRequestContext(), BoundNetLog(), | |
| 116 connect_delegate.Pass(), | |
| 117 timer ? timer.Pass() | |
| 118 : scoped_ptr<base::Timer>(new base::Timer(false, false))); | |
| 119 } | |
| 120 | |
| 121 std::vector<HeaderKeyValuePair> | |
| 122 WebSocketStreamCreateTestBase::RequestHeadersToVector( | |
| 123 const HttpRequestHeaders& headers) { | |
| 124 HttpRequestHeaders::Iterator it(headers); | |
| 125 std::vector<HeaderKeyValuePair> result; | |
| 126 while (it.GetNext()) | |
| 127 result.push_back(HeaderKeyValuePair(it.name(), it.value())); | |
| 128 return result; | |
| 129 } | |
| 130 | |
| 131 std::vector<HeaderKeyValuePair> | |
| 132 WebSocketStreamCreateTestBase::ResponseHeadersToVector( | |
| 133 const HttpResponseHeaders& headers) { | |
| 134 void* iter = NULL; | |
| 135 std::string name, value; | |
| 136 std::vector<HeaderKeyValuePair> result; | |
| 137 while (headers.EnumerateHeaderLines(&iter, &name, &value)) | |
| 138 result.push_back(HeaderKeyValuePair(name, value)); | |
| 139 return result; | |
| 140 } | |
| 141 | |
| 142 void WebSocketStreamCreateTestBase::WaitUntilConnectDone() { | |
| 143 connect_run_loop_.Run(); | |
| 144 } | |
| 145 | |
| 146 std::vector<std::string> WebSocketStreamCreateTestBase::NoSubProtocols() { | |
| 147 return std::vector<std::string>(); | |
| 148 } | |
| 149 | |
| 150 } // namespace net | |
| OLD | NEW |