1ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 5c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// An implementation of WebSocketStreamHandle. 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 7c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "webkit/glue/websocketstreamhandle_impl.h" 8c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include <vector> 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/compiler_specific.h" 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/logging.h" 13ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "base/memory/ref_counted.h" 14ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "base/memory/scoped_ptr.h" 1572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "third_party/WebKit/Source/WebKit/chromium/public/WebData.h" 1672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "third_party/WebKit/Source/WebKit/chromium/public/WebURL.h" 1772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "third_party/WebKit/Source/WebKit/chromium/public/WebSocketStreamHandleClient.h" 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "webkit/glue/websocketstreamhandle_bridge.h" 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "webkit/glue/websocketstreamhandle_delegate.h" 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace webkit_glue { 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// WebSocketStreamHandleImpl::Context ----------------------------------------- 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochclass WebSocketStreamHandleImpl::Context 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : public base::RefCounted<Context>, 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public WebSocketStreamHandleDelegate { 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public: 29c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch explicit Context(WebSocketStreamHandleImpl* handle); 30c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 31c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandleClient* client() const { return client_; } 32c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch void set_client(WebKit::WebSocketStreamHandleClient* client) { 33c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_ = client; 34c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 35c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch void Connect(const WebKit::WebURL& url); 37c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bool Send(const WebKit::WebData& data); 38c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch void Close(); 39c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 40c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Must be called before |handle_| or |client_| is deleted. 41c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Once detached, it never calls |client_| back. 42c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch void Detach(); 43c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // WebSocketStreamHandleDelegate methods: 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void DidOpenStream(WebKit::WebSocketStreamHandle*, int); 46c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void DidSendData(WebKit::WebSocketStreamHandle*, int); 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void DidReceiveData( 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandle*, const char*, int); 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void DidClose(WebKit::WebSocketStreamHandle*); 50c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch private: 52c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch friend class base::RefCounted<Context>; 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ~Context() { 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!handle_); 55c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!client_); 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!bridge_); 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 59c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebSocketStreamHandleImpl* handle_; 60c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandleClient* client_; 61c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // |bridge_| is alive from Connect to DidClose, so Context must be alive 62c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // in the time period. 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch scoped_refptr<WebSocketStreamHandleBridge> bridge_; 64c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 65c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DISALLOW_COPY_AND_ASSIGN(Context); 66c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 67c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 68c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochWebSocketStreamHandleImpl::Context::Context(WebSocketStreamHandleImpl* handle) 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : handle_(handle), 70c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_(NULL), 71c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_(NULL) { 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 73c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 74c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::Connect(const WebKit::WebURL& url) { 75731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "Connect url=" << url; 76c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!bridge_); 77c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_ = WebSocketStreamHandleBridge::Create(handle_, this); 78c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch AddRef(); // Will be released by DidClose(). 79c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_->Connect(url); 80c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 81c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 82c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool WebSocketStreamHandleImpl::Context::Send(const WebKit::WebData& data) { 83731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "Send data.size=" << data.size(); 84c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(bridge_); 85c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return bridge_->Send( 86c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::vector<char>(data.data(), data.data() + data.size())); 87c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 88c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 89c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::Close() { 90731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "Close"; 91c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (bridge_) 92c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_->Close(); 93c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 94c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 95c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::Detach() { 96c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch handle_ = NULL; 97c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_ = NULL; 98c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // If Connect was called, |bridge_| is not NULL, so that this Context closes 99c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // the |bridge_| here. Then |bridge_| will call back DidClose, and will 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // be released by itself. 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Otherwise, |bridge_| is NULL. 102c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (bridge_) 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_->Close(); 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 105c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::DidOpenStream( 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandle* web_handle, int max_amount_send_allowed) { 108731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "DidOpen"; 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (client_) 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_->didOpenStream(handle_, max_amount_send_allowed); 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::DidSendData( 114c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandle* web_handle, int amount_sent) { 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (client_) 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_->didSendData(handle_, amount_sent); 117c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 118c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::DidReceiveData( 120c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandle* web_handle, const char* data, int size) { 121c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (client_) 122c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_->didReceiveData(handle_, WebKit::WebData(data, size)); 123c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 124c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 125c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::Context::DidClose( 126c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandle* web_handle) { 127731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "DidClose"; 128c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch bridge_ = NULL; 129c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebSocketStreamHandleImpl* handle = handle_; 130c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch handle_ = NULL; 131c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (client_) { 132c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch WebKit::WebSocketStreamHandleClient* client = client_; 133c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client_ = NULL; 134c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch client->didClose(handle); 135c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 136c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch Release(); 137c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 138c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 139c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// WebSocketStreamHandleImpl ------------------------------------------------ 140c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 141c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochWebSocketStreamHandleImpl::WebSocketStreamHandleImpl() 142c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : ALLOW_THIS_IN_INITIALIZER_LIST(context_(new Context(this))) { 143c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 144c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 145c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochWebSocketStreamHandleImpl::~WebSocketStreamHandleImpl() { 146c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // We won't receive any events from |context_|. 147c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // |context_| is ref counted, and will be released when it received 148c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // DidClose. 149c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch context_->Detach(); 150c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 151c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 152c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::connect( 153c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const WebKit::WebURL& url, WebKit::WebSocketStreamHandleClient* client) { 154731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick VLOG(1) << "connect url=" << url; 155c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK(!context_->client()); 156c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch context_->set_client(client); 157c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 158c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch context_->Connect(url); 159c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 160c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 161c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool WebSocketStreamHandleImpl::send(const WebKit::WebData& data) { 162c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return context_->Send(data); 163c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 164c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 165c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid WebSocketStreamHandleImpl::close() { 166c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch context_->Close(); 167c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 168c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 169c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace webkit_glue 170