blob: eef6a96a7f61eb82539bbbbe05e10e50e16533d8 [file] [log] [blame]
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_
#define MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_
#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "mojo/common/handle_watcher.h"
#include "mojo/services/network/public/interfaces/web_socket.mojom.h"
#include "third_party/WebKit/public/platform/WebSocketHandle.h"
namespace mojo {
class NetworkService;
class WebSocketWriteQueue;
}
namespace html_viewer {
class WebSocketClientImpl;
// Implements WebSocketHandle by talking to the mojo WebSocket interface.
class WebSocketHandleImpl : public blink::WebSocketHandle {
public:
explicit WebSocketHandleImpl(mojo::NetworkService* network_service);
private:
friend class WebSocketClientImpl;
virtual ~WebSocketHandleImpl();
// blink::WebSocketHandle methods:
virtual void connect(const blink::WebURL& url,
const blink::WebVector<blink::WebString>& protocols,
const blink::WebSerializedOrigin& origin,
blink::WebSocketHandleClient*) override;
virtual void send(bool fin,
MessageType,
const char* data,
size_t size) override;
virtual void flowControl(int64_t quota) override;
virtual void close(unsigned short code,
const blink::WebString& reason) override;
// Called when we finished writing to |send_stream_|.
void DidWriteToSendStream(bool fin,
WebSocketHandle::MessageType type,
uint32_t num_bytes,
const char* data);
// Called when the socket is closed.
void Disconnect();
mojo::WebSocketPtr web_socket_;
scoped_ptr<WebSocketClientImpl> client_;
mojo::ScopedDataPipeProducerHandle send_stream_;
scoped_ptr<mojo::WebSocketWriteQueue> write_queue_;
// True if close() was called.
bool did_close_;
DISALLOW_COPY_AND_ASSIGN(WebSocketHandleImpl);
};
} // namespace html_viewer
#endif // MOJO_SERVICES_HTML_VIEWER_WEBSOCKETHANDLE_IMPL_H_