WorkerThreadableWebSocketChannel.h [plain text]
#ifndef WorkerThreadableWebSocketChannel_h
#define WorkerThreadableWebSocketChannel_h
#if ENABLE(WEB_SOCKETS)
#include "ThreadableWebSocketChannel.h"
#include "WebSocketChannelClient.h"
#include "WorkerGlobalScope.h"
#include <wtf/PassRefPtr.h>
#include <wtf/RefCounted.h>
#include <wtf/RefPtr.h>
#include <wtf/Threading.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
class URL;
class ScriptExecutionContext;
class ThreadableWebSocketChannelClientWrapper;
class WorkerGlobalScope;
class WorkerLoaderProxy;
class WorkerRunLoop;
class WorkerThreadableWebSocketChannel : public RefCounted<WorkerThreadableWebSocketChannel>, public ThreadableWebSocketChannel {
WTF_MAKE_FAST_ALLOCATED;
public:
static PassRefPtr<ThreadableWebSocketChannel> create(WorkerGlobalScope* workerGlobalScope, WebSocketChannelClient* client, const String& taskMode)
{
return adoptRef(new WorkerThreadableWebSocketChannel(workerGlobalScope, client, taskMode));
}
virtual ~WorkerThreadableWebSocketChannel();
virtual void connect(const URL&, const String& protocol) override;
virtual String subprotocol() override;
virtual String extensions() override;
virtual ThreadableWebSocketChannel::SendResult send(const String& message) override;
virtual ThreadableWebSocketChannel::SendResult send(const JSC::ArrayBuffer&, unsigned byteOffset, unsigned byteLength) override;
virtual ThreadableWebSocketChannel::SendResult send(Blob&) override;
virtual unsigned long bufferedAmount() const override;
virtual void close(int code, const String& reason) override;
virtual void fail(const String& reason) override;
virtual void disconnect() override; virtual void suspend() override;
virtual void resume() override;
class Peer : public WebSocketChannelClient {
WTF_MAKE_NONCOPYABLE(Peer); WTF_MAKE_FAST_ALLOCATED;
public:
static Peer* create(PassRefPtr<ThreadableWebSocketChannelClientWrapper> clientWrapper, WorkerLoaderProxy& loaderProxy, ScriptExecutionContext* context, const String& taskMode)
{
return new Peer(clientWrapper, loaderProxy, context, taskMode);
}
~Peer();
void connect(const URL&, const String& protocol);
void send(const String& message);
void send(const JSC::ArrayBuffer&);
void send(Blob&);
void bufferedAmount();
void close(int code, const String& reason);
void fail(const String& reason);
void disconnect();
void suspend();
void resume();
virtual void didConnect() override;
virtual void didReceiveMessage(const String& message) override;
virtual void didReceiveBinaryData(PassOwnPtr<Vector<char>>) override;
virtual void didUpdateBufferedAmount(unsigned long bufferedAmount) override;
virtual void didStartClosingHandshake() override;
virtual void didClose(unsigned long unhandledBufferedAmount, ClosingHandshakeCompletionStatus, unsigned short code, const String& reason) override;
virtual void didReceiveMessageError() override;
private:
Peer(PassRefPtr<ThreadableWebSocketChannelClientWrapper>, WorkerLoaderProxy&, ScriptExecutionContext*, const String& taskMode);
RefPtr<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
WorkerLoaderProxy& m_loaderProxy;
RefPtr<ThreadableWebSocketChannel> m_mainWebSocketChannel;
String m_taskMode;
};
using RefCounted<WorkerThreadableWebSocketChannel>::ref;
using RefCounted<WorkerThreadableWebSocketChannel>::deref;
protected:
virtual void refThreadableWebSocketChannel() { ref(); }
virtual void derefThreadableWebSocketChannel() { deref(); }
private:
class Bridge : public RefCounted<Bridge> {
public:
static PassRefPtr<Bridge> create(PassRefPtr<ThreadableWebSocketChannelClientWrapper> workerClientWrapper, PassRefPtr<WorkerGlobalScope> workerGlobalScope, const String& taskMode)
{
return adoptRef(new Bridge(workerClientWrapper, workerGlobalScope, taskMode));
}
~Bridge();
void initialize();
void connect(const URL&, const String& protocol);
ThreadableWebSocketChannel::SendResult send(const String& message);
ThreadableWebSocketChannel::SendResult send(const JSC::ArrayBuffer&, unsigned byteOffset, unsigned byteLength);
ThreadableWebSocketChannel::SendResult send(Blob&);
unsigned long bufferedAmount();
void close(int code, const String& reason);
void fail(const String& reason);
void disconnect();
void suspend();
void resume();
using RefCounted<Bridge>::ref;
using RefCounted<Bridge>::deref;
private:
Bridge(PassRefPtr<ThreadableWebSocketChannelClientWrapper>, PassRefPtr<WorkerGlobalScope>, const String& taskMode);
static void setWebSocketChannel(ScriptExecutionContext*, Bridge* thisPtr, Peer*, PassRefPtr<ThreadableWebSocketChannelClientWrapper>);
static void mainThreadInitialize(ScriptExecutionContext&, WorkerLoaderProxy*, PassRefPtr<ThreadableWebSocketChannelClientWrapper>, const String& taskMode);
void clearClientWrapper();
void setMethodNotCompleted();
void waitForMethodCompletion();
RefPtr<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
RefPtr<WorkerGlobalScope> m_workerGlobalScope;
WorkerLoaderProxy& m_loaderProxy;
String m_taskMode;
Peer* m_peer;
};
WorkerThreadableWebSocketChannel(WorkerGlobalScope*, WebSocketChannelClient*, const String& taskMode);
static void mainThreadConnect(ScriptExecutionContext&, Peer*, const URL&, const String& protocol);
static void mainThreadSend(ScriptExecutionContext&, Peer*, const String& message);
static void mainThreadSendArrayBuffer(ScriptExecutionContext&, Peer*, PassOwnPtr<Vector<char>>);
static void mainThreadSendBlob(ScriptExecutionContext&, Peer*, const URL&, const String& type, long long size);
static void mainThreadBufferedAmount(ScriptExecutionContext&, Peer*);
static void mainThreadClose(ScriptExecutionContext&, Peer*, int code, const String& reason);
static void mainThreadFail(ScriptExecutionContext&, Peer*, const String& reason);
static void mainThreadDestroy(ScriptExecutionContext&, Peer*);
static void mainThreadSuspend(ScriptExecutionContext&, Peer*);
static void mainThreadResume(ScriptExecutionContext&, Peer*);
class WorkerGlobalScopeDidInitializeTask;
RefPtr<WorkerGlobalScope> m_workerGlobalScope;
RefPtr<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
RefPtr<Bridge> m_bridge;
};
}
#endif // ENABLE(WEB_SOCKETS)
#endif // WorkerThreadableWebSocketChannel_h