WebSocketChannel.h [plain text]
#pragma once
#include "MessageReceiver.h"
#include "MessageSender.h"
#include "WebSocketIdentifier.h"
#include <WebCore/NetworkSendQueue.h>
#include <WebCore/ThreadableWebSocketChannel.h>
#include <WebCore/WebSocketChannelInspector.h>
#include <WebCore/WebSocketFrame.h>
#include <wtf/WeakPtr.h>
namespace IPC {
class Connection;
class Decoder;
class DataReference;
}
namespace WebKit {
class WebSocketChannel : public IPC::MessageSender, public IPC::MessageReceiver, public WebCore::ThreadableWebSocketChannel, public RefCounted<WebSocketChannel>, public CanMakeWeakPtr<WebSocketChannel> {
public:
static Ref<WebSocketChannel> create(WebCore::Document&, WebCore::WebSocketChannelClient&);
~WebSocketChannel();
WebSocketIdentifier identifier() const { return m_identifier; }
void didReceiveMessage(IPC::Connection&, IPC::Decoder&);
void networkProcessCrashed();
using RefCounted<WebSocketChannel>::ref;
using RefCounted<WebSocketChannel>::deref;
private:
WebSocketChannel(WebCore::Document&, WebCore::WebSocketChannelClient&);
static WebCore::NetworkSendQueue createMessageQueue(WebCore::Document&, WebSocketChannel&);
ConnectStatus connect(const URL&, const String& protocol) final;
String subprotocol() final;
String extensions() final;
SendResult send(const String& message) final;
SendResult send(const JSC::ArrayBuffer&, unsigned byteOffset, unsigned byteLength) final;
SendResult send(WebCore::Blob&) final;
unsigned bufferedAmount() const final;
void close(int code, const String& reason) final;
void fail(const String& reason) final;
void disconnect() final;
void suspend() final;
void resume() final;
void refThreadableWebSocketChannel() final { ref(); }
void derefThreadableWebSocketChannel() final { deref(); }
void notifySendFrame(WebCore::WebSocketFrame::OpCode, const char* data, size_t length);
void logErrorMessage(const String&);
void didConnect(String&& subprotocol, String&& extensions);
void didReceiveText(String&&);
void didReceiveBinaryData(IPC::DataReference&&);
void didClose(unsigned short code, String&&);
void didReceiveMessageError(String&&);
void didSendHandshakeRequest(WebCore::ResourceRequest&&);
void didReceiveHandshakeResponse(WebCore::ResourceResponse&&);
IPC::Connection* messageSenderConnection() const final;
uint64_t messageSenderDestinationID() const final;
bool increaseBufferedAmount(size_t);
void decreaseBufferedAmount(size_t);
template<typename T> void sendMessage(T&&, size_t byteLength);
void enqueueTask(Function<void()>&&);
WeakPtr<WebCore::Document> m_document;
WebSocketIdentifier m_identifier;
WeakPtr<WebCore::WebSocketChannelClient> m_client;
URL m_url;
String m_subprotocol;
String m_extensions;
size_t m_bufferedAmount { 0 };
bool m_isClosing { false };
bool m_isSuspended { false };
Deque<Function<void()>> m_pendingTasks;
WebCore::NetworkSendQueue m_messageQueue;
WebCore::WebSocketChannelInspector m_inspector;
};
}