OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #include "content/browser/websockets/websocket_impl.h" | 5 #include "content/browser/websockets/websocket_impl.h" |
6 | 6 |
7 #include <inttypes.h> | 7 #include <inttypes.h> |
8 | 8 |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
(...skipping 25 matching lines...) Expand all Loading... |
36 #include "net/websockets/websocket_frame.h" // for WebSocketFrameHeader::OpCode | 36 #include "net/websockets/websocket_frame.h" // for WebSocketFrameHeader::OpCode |
37 #include "net/websockets/websocket_handshake_request_info.h" | 37 #include "net/websockets/websocket_handshake_request_info.h" |
38 #include "net/websockets/websocket_handshake_response_info.h" | 38 #include "net/websockets/websocket_handshake_response_info.h" |
39 #include "url/origin.h" | 39 #include "url/origin.h" |
40 | 40 |
41 namespace content { | 41 namespace content { |
42 namespace { | 42 namespace { |
43 | 43 |
44 typedef net::WebSocketEventInterface::ChannelState ChannelState; | 44 typedef net::WebSocketEventInterface::ChannelState ChannelState; |
45 | 45 |
46 // Convert a blink::mojom::WebSocketMessageType to a | 46 // Convert a mojom::WebSocketMessageType to a |
47 // net::WebSocketFrameHeader::OpCode | 47 // net::WebSocketFrameHeader::OpCode |
48 net::WebSocketFrameHeader::OpCode MessageTypeToOpCode( | 48 net::WebSocketFrameHeader::OpCode MessageTypeToOpCode( |
49 blink::mojom::WebSocketMessageType type) { | 49 mojom::WebSocketMessageType type) { |
50 DCHECK(type == blink::mojom::WebSocketMessageType::CONTINUATION || | 50 DCHECK(type == mojom::WebSocketMessageType::CONTINUATION || |
51 type == blink::mojom::WebSocketMessageType::TEXT || | 51 type == mojom::WebSocketMessageType::TEXT || |
52 type == blink::mojom::WebSocketMessageType::BINARY); | 52 type == mojom::WebSocketMessageType::BINARY); |
53 typedef net::WebSocketFrameHeader::OpCode OpCode; | 53 typedef net::WebSocketFrameHeader::OpCode OpCode; |
54 // These compile asserts verify that the same underlying values are used for | 54 // These compile asserts verify that the same underlying values are used for |
55 // both types, so we can simply cast between them. | 55 // both types, so we can simply cast between them. |
56 static_assert( | 56 static_assert( |
57 static_cast<OpCode>(blink::mojom::WebSocketMessageType::CONTINUATION) == | 57 static_cast<OpCode>(mojom::WebSocketMessageType::CONTINUATION) == |
58 net::WebSocketFrameHeader::kOpCodeContinuation, | 58 net::WebSocketFrameHeader::kOpCodeContinuation, |
59 "enum values must match for opcode continuation"); | 59 "enum values must match for opcode continuation"); |
60 static_assert( | 60 static_assert( |
61 static_cast<OpCode>(blink::mojom::WebSocketMessageType::TEXT) == | 61 static_cast<OpCode>(mojom::WebSocketMessageType::TEXT) == |
62 net::WebSocketFrameHeader::kOpCodeText, | 62 net::WebSocketFrameHeader::kOpCodeText, |
63 "enum values must match for opcode text"); | 63 "enum values must match for opcode text"); |
64 static_assert( | 64 static_assert( |
65 static_cast<OpCode>(blink::mojom::WebSocketMessageType::BINARY) == | 65 static_cast<OpCode>(mojom::WebSocketMessageType::BINARY) == |
66 net::WebSocketFrameHeader::kOpCodeBinary, | 66 net::WebSocketFrameHeader::kOpCodeBinary, |
67 "enum values must match for opcode binary"); | 67 "enum values must match for opcode binary"); |
68 return static_cast<OpCode>(type); | 68 return static_cast<OpCode>(type); |
69 } | 69 } |
70 | 70 |
71 blink::mojom::WebSocketMessageType OpCodeToMessageType( | 71 mojom::WebSocketMessageType OpCodeToMessageType( |
72 net::WebSocketFrameHeader::OpCode opCode) { | 72 net::WebSocketFrameHeader::OpCode opCode) { |
73 DCHECK(opCode == net::WebSocketFrameHeader::kOpCodeContinuation || | 73 DCHECK(opCode == net::WebSocketFrameHeader::kOpCodeContinuation || |
74 opCode == net::WebSocketFrameHeader::kOpCodeText || | 74 opCode == net::WebSocketFrameHeader::kOpCodeText || |
75 opCode == net::WebSocketFrameHeader::kOpCodeBinary); | 75 opCode == net::WebSocketFrameHeader::kOpCodeBinary); |
76 // This cast is guaranteed valid by the static_assert() statements above. | 76 // This cast is guaranteed valid by the static_assert() statements above. |
77 return static_cast<blink::mojom::WebSocketMessageType>(opCode); | 77 return static_cast<mojom::WebSocketMessageType>(opCode); |
78 } | 78 } |
79 | 79 |
80 } // namespace | 80 } // namespace |
81 | 81 |
82 // Implementation of net::WebSocketEventInterface. Receives events from our | 82 // Implementation of net::WebSocketEventInterface. Receives events from our |
83 // WebSocketChannel object. | 83 // WebSocketChannel object. |
84 class WebSocketImpl::WebSocketEventHandler final | 84 class WebSocketImpl::WebSocketEventHandler final |
85 : public net::WebSocketEventInterface { | 85 : public net::WebSocketEventInterface { |
86 public: | 86 public: |
87 explicit WebSocketEventHandler(WebSocketImpl* impl); | 87 explicit WebSocketEventHandler(WebSocketImpl* impl); |
(...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
237 bool should_send = | 237 bool should_send = |
238 ChildProcessSecurityPolicyImpl::GetInstance()->CanReadRawCookies( | 238 ChildProcessSecurityPolicyImpl::GetInstance()->CanReadRawCookies( |
239 impl_->delegate_->GetClientProcessId()); | 239 impl_->delegate_->GetClientProcessId()); |
240 | 240 |
241 DVLOG(3) << "WebSocketEventHandler::OnStartOpeningHandshake @" | 241 DVLOG(3) << "WebSocketEventHandler::OnStartOpeningHandshake @" |
242 << reinterpret_cast<void*>(this) << " should_send=" << should_send; | 242 << reinterpret_cast<void*>(this) << " should_send=" << should_send; |
243 | 243 |
244 if (!should_send) | 244 if (!should_send) |
245 return WebSocketEventInterface::CHANNEL_ALIVE; | 245 return WebSocketEventInterface::CHANNEL_ALIVE; |
246 | 246 |
247 blink::mojom::WebSocketHandshakeRequestPtr request_to_pass( | 247 mojom::WebSocketHandshakeRequestPtr request_to_pass( |
248 blink::mojom::WebSocketHandshakeRequest::New()); | 248 mojom::WebSocketHandshakeRequest::New()); |
249 request_to_pass->url.Swap(&request->url); | 249 request_to_pass->url.Swap(&request->url); |
250 net::HttpRequestHeaders::Iterator it(request->headers); | 250 net::HttpRequestHeaders::Iterator it(request->headers); |
251 while (it.GetNext()) { | 251 while (it.GetNext()) { |
252 blink::mojom::HttpHeaderPtr header(blink::mojom::HttpHeader::New()); | 252 mojom::HttpHeaderPtr header(mojom::HttpHeader::New()); |
253 header->name = it.name(); | 253 header->name = it.name(); |
254 header->value = it.value(); | 254 header->value = it.value(); |
255 request_to_pass->headers.push_back(std::move(header)); | 255 request_to_pass->headers.push_back(std::move(header)); |
256 } | 256 } |
257 request_to_pass->headers_text = | 257 request_to_pass->headers_text = |
258 base::StringPrintf("GET %s HTTP/1.1\r\n", | 258 base::StringPrintf("GET %s HTTP/1.1\r\n", |
259 request_to_pass->url.spec().c_str()) + | 259 request_to_pass->url.spec().c_str()) + |
260 request->headers.ToString(); | 260 request->headers.ToString(); |
261 | 261 |
262 impl_->client_->OnStartOpeningHandshake(std::move(request_to_pass)); | 262 impl_->client_->OnStartOpeningHandshake(std::move(request_to_pass)); |
263 | 263 |
264 return WebSocketEventInterface::CHANNEL_ALIVE; | 264 return WebSocketEventInterface::CHANNEL_ALIVE; |
265 } | 265 } |
266 | 266 |
267 ChannelState WebSocketImpl::WebSocketEventHandler::OnFinishOpeningHandshake( | 267 ChannelState WebSocketImpl::WebSocketEventHandler::OnFinishOpeningHandshake( |
268 std::unique_ptr<net::WebSocketHandshakeResponseInfo> response) { | 268 std::unique_ptr<net::WebSocketHandshakeResponseInfo> response) { |
269 bool should_send = | 269 bool should_send = |
270 ChildProcessSecurityPolicyImpl::GetInstance()->CanReadRawCookies( | 270 ChildProcessSecurityPolicyImpl::GetInstance()->CanReadRawCookies( |
271 impl_->delegate_->GetClientProcessId()); | 271 impl_->delegate_->GetClientProcessId()); |
272 | 272 |
273 DVLOG(3) << "WebSocketEventHandler::OnFinishOpeningHandshake " | 273 DVLOG(3) << "WebSocketEventHandler::OnFinishOpeningHandshake " |
274 << reinterpret_cast<void*>(this) << " should_send=" << should_send; | 274 << reinterpret_cast<void*>(this) << " should_send=" << should_send; |
275 | 275 |
276 if (!should_send) | 276 if (!should_send) |
277 return WebSocketEventInterface::CHANNEL_ALIVE; | 277 return WebSocketEventInterface::CHANNEL_ALIVE; |
278 | 278 |
279 blink::mojom::WebSocketHandshakeResponsePtr response_to_pass( | 279 mojom::WebSocketHandshakeResponsePtr response_to_pass( |
280 blink::mojom::WebSocketHandshakeResponse::New()); | 280 mojom::WebSocketHandshakeResponse::New()); |
281 response_to_pass->url.Swap(&response->url); | 281 response_to_pass->url.Swap(&response->url); |
282 response_to_pass->status_code = response->status_code; | 282 response_to_pass->status_code = response->status_code; |
283 response_to_pass->status_text = response->status_text; | 283 response_to_pass->status_text = response->status_text; |
284 size_t iter = 0; | 284 size_t iter = 0; |
285 std::string name, value; | 285 std::string name, value; |
286 while (response->headers->EnumerateHeaderLines(&iter, &name, &value)) { | 286 while (response->headers->EnumerateHeaderLines(&iter, &name, &value)) { |
287 blink::mojom::HttpHeaderPtr header(blink::mojom::HttpHeader::New()); | 287 mojom::HttpHeaderPtr header(mojom::HttpHeader::New()); |
288 header->name = name; | 288 header->name = name; |
289 header->value = value; | 289 header->value = value; |
290 response_to_pass->headers.push_back(std::move(header)); | 290 response_to_pass->headers.push_back(std::move(header)); |
291 } | 291 } |
292 response_to_pass->headers_text = | 292 response_to_pass->headers_text = |
293 net::HttpUtil::ConvertHeadersBackToHTTPResponse( | 293 net::HttpUtil::ConvertHeadersBackToHTTPResponse( |
294 response->headers->raw_headers()); | 294 response->headers->raw_headers()); |
295 | 295 |
296 impl_->client_->OnFinishOpeningHandshake(std::move(response_to_pass)); | 296 impl_->client_->OnFinishOpeningHandshake(std::move(response_to_pass)); |
297 | 297 |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
343 } | 343 } |
344 | 344 |
345 void WebSocketImpl::WebSocketEventHandler::SSLErrorHandlerDelegate:: | 345 void WebSocketImpl::WebSocketEventHandler::SSLErrorHandlerDelegate:: |
346 ContinueSSLRequest() { | 346 ContinueSSLRequest() { |
347 DVLOG(3) << "SSLErrorHandlerDelegate::ContinueSSLRequest"; | 347 DVLOG(3) << "SSLErrorHandlerDelegate::ContinueSSLRequest"; |
348 callbacks_->ContinueSSLRequest(); | 348 callbacks_->ContinueSSLRequest(); |
349 } | 349 } |
350 | 350 |
351 WebSocketImpl::WebSocketImpl( | 351 WebSocketImpl::WebSocketImpl( |
352 Delegate* delegate, | 352 Delegate* delegate, |
353 blink::mojom::WebSocketRequest request, | 353 mojom::WebSocketRequest request, |
354 int frame_id, | 354 int frame_id, |
355 base::TimeDelta delay) | 355 base::TimeDelta delay) |
356 : delegate_(delegate), | 356 : delegate_(delegate), |
357 binding_(this, std::move(request)), | 357 binding_(this, std::move(request)), |
358 delay_(delay), | 358 delay_(delay), |
359 pending_flow_control_quota_(0), | 359 pending_flow_control_quota_(0), |
360 frame_id_(frame_id), | 360 frame_id_(frame_id), |
361 handshake_succeeded_(false), | 361 handshake_succeeded_(false), |
362 weak_ptr_factory_(this) { | 362 weak_ptr_factory_(this) { |
363 binding_.set_connection_error_handler( | 363 binding_.set_connection_error_handler( |
364 base::Bind(&WebSocketImpl::OnConnectionError, base::Unretained(this))); | 364 base::Bind(&WebSocketImpl::OnConnectionError, base::Unretained(this))); |
365 } | 365 } |
366 | 366 |
367 WebSocketImpl::~WebSocketImpl() {} | 367 WebSocketImpl::~WebSocketImpl() {} |
368 | 368 |
369 void WebSocketImpl::GoAway() { | 369 void WebSocketImpl::GoAway() { |
370 StartClosingHandshake(static_cast<uint16_t>(net::kWebSocketErrorGoingAway), | 370 StartClosingHandshake(static_cast<uint16_t>(net::kWebSocketErrorGoingAway), |
371 ""); | 371 ""); |
372 } | 372 } |
373 | 373 |
374 void WebSocketImpl::AddChannelRequest( | 374 void WebSocketImpl::AddChannelRequest( |
375 const GURL& socket_url, | 375 const GURL& socket_url, |
376 const std::vector<std::string>& requested_protocols, | 376 const std::vector<std::string>& requested_protocols, |
377 const url::Origin& origin, | 377 const url::Origin& origin, |
378 const GURL& first_party_for_cookies, | 378 const GURL& first_party_for_cookies, |
379 const std::string& user_agent_override, | 379 const std::string& user_agent_override, |
380 blink::mojom::WebSocketClientPtr client) { | 380 mojom::WebSocketClientPtr client) { |
381 DVLOG(3) << "WebSocketImpl::AddChannelRequest @" | 381 DVLOG(3) << "WebSocketImpl::AddChannelRequest @" |
382 << reinterpret_cast<void*>(this) | 382 << reinterpret_cast<void*>(this) |
383 << " socket_url=\"" << socket_url << "\" requested_protocols=\"" | 383 << " socket_url=\"" << socket_url << "\" requested_protocols=\"" |
384 << base::JoinString(requested_protocols, ", ") | 384 << base::JoinString(requested_protocols, ", ") |
385 << "\" origin=\"" << origin | 385 << "\" origin=\"" << origin |
386 << "\" first_party_for_cookies=\"" << first_party_for_cookies | 386 << "\" first_party_for_cookies=\"" << first_party_for_cookies |
387 << "\" user_agent_override=\"" << user_agent_override | 387 << "\" user_agent_override=\"" << user_agent_override |
388 << "\""; | 388 << "\""; |
389 | 389 |
390 if (client_ || !client) { | 390 if (client_ || !client) { |
(...skipping 17 matching lines...) Expand all Loading... |
408 first_party_for_cookies, | 408 first_party_for_cookies, |
409 user_agent_override), | 409 user_agent_override), |
410 delay_); | 410 delay_); |
411 } else { | 411 } else { |
412 AddChannel(socket_url, requested_protocols, origin, first_party_for_cookies, | 412 AddChannel(socket_url, requested_protocols, origin, first_party_for_cookies, |
413 user_agent_override); | 413 user_agent_override); |
414 } | 414 } |
415 } | 415 } |
416 | 416 |
417 void WebSocketImpl::SendFrame(bool fin, | 417 void WebSocketImpl::SendFrame(bool fin, |
418 blink::mojom::WebSocketMessageType type, | 418 mojom::WebSocketMessageType type, |
419 const std::vector<uint8_t>& data) { | 419 const std::vector<uint8_t>& data) { |
420 DVLOG(3) << "WebSocketImpl::SendFrame @" | 420 DVLOG(3) << "WebSocketImpl::SendFrame @" |
421 << reinterpret_cast<void*>(this) << " fin=" << fin | 421 << reinterpret_cast<void*>(this) << " fin=" << fin |
422 << " type=" << type << " data is " << data.size() << " bytes"; | 422 << " type=" << type << " data is " << data.size() << " bytes"; |
423 | 423 |
424 if (!channel_) { | 424 if (!channel_) { |
425 // The client should not be sending us frames until after we've informed | 425 // The client should not be sending us frames until after we've informed |
426 // it that the channel has been opened (OnAddChannelResponse). | 426 // it that the channel has been opened (OnAddChannelResponse). |
427 if (handshake_succeeded_) { | 427 if (handshake_succeeded_) { |
428 DVLOG(1) << "Dropping frame sent to closed websocket"; | 428 DVLOG(1) << "Dropping frame sent to closed websocket"; |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
522 net::HttpRequestHeaders::kUserAgent, | 522 net::HttpRequestHeaders::kUserAgent, |
523 user_agent_override.c_str()); | 523 user_agent_override.c_str()); |
524 } | 524 } |
525 channel_->SendAddChannelRequest(socket_url, requested_protocols, origin, | 525 channel_->SendAddChannelRequest(socket_url, requested_protocols, origin, |
526 first_party_for_cookies, additional_headers); | 526 first_party_for_cookies, additional_headers); |
527 if (quota > 0) | 527 if (quota > 0) |
528 SendFlowControl(quota); | 528 SendFlowControl(quota); |
529 } | 529 } |
530 | 530 |
531 } // namespace content | 531 } // namespace content |
OLD | NEW |