#ifndef WebSocket_h
#define WebSocket_h
#if ENABLE(WEB_SOCKETS)
#include "ActiveDOMObject.h"
#include "EventListener.h"
#include "EventNames.h"
#include "EventTarget.h"
#include "URL.h"
#include "WebSocketChannel.h"
#include "WebSocketChannelClient.h"
#include <wtf/Forward.h>
#include <wtf/RefCounted.h>
#include <wtf/text/AtomicStringHash.h>
namespace WebCore {
class Blob;
class ThreadableWebSocketChannel;
class WebSocket final : public RefCounted<WebSocket>, public EventTargetWithInlineData, public ActiveDOMObject, public WebSocketChannelClient {
public:
static void setIsAvailable(bool);
static bool isAvailable();
static const char* subProtocolSeperator();
static PassRefPtr<WebSocket> create(ScriptExecutionContext&);
static PassRefPtr<WebSocket> create(ScriptExecutionContext&, const String& url, ExceptionCode&);
static PassRefPtr<WebSocket> create(ScriptExecutionContext&, const String& url, const String& protocol, ExceptionCode&);
static PassRefPtr<WebSocket> create(ScriptExecutionContext&, const String& url, const Vector<String>& protocols, ExceptionCode&);
virtual ~WebSocket();
enum State {
CONNECTING = 0,
OPEN = 1,
CLOSING = 2,
CLOSED = 3
};
void connect(const String& url, ExceptionCode&);
void connect(const String& url, const String& protocol, ExceptionCode&);
void connect(const String& url, const Vector<String>& protocols, ExceptionCode&);
void send(const String& message, ExceptionCode&);
void send(JSC::ArrayBuffer*, ExceptionCode&);
void send(JSC::ArrayBufferView*, ExceptionCode&);
void send(Blob*, ExceptionCode&);
void close(int code, const String& reason, ExceptionCode&);
void close(ExceptionCode& ec) { close(WebSocketChannel::CloseEventCodeNotSpecified, String(), ec); }
void close(int code, ExceptionCode& ec) { close(code, String(), ec); }
const URL& url() const;
State readyState() const;
unsigned long bufferedAmount() const;
String protocol() const;
String extensions() const;
String binaryType() const;
void setBinaryType(const String&);
DEFINE_ATTRIBUTE_EVENT_LISTENER(open);
DEFINE_ATTRIBUTE_EVENT_LISTENER(message);
DEFINE_ATTRIBUTE_EVENT_LISTENER(error);
DEFINE_ATTRIBUTE_EVENT_LISTENER(close);
virtual EventTargetInterface eventTargetInterface() const override;
virtual ScriptExecutionContext* scriptExecutionContext() const override;
using RefCounted<WebSocket>::ref;
using RefCounted<WebSocket>::deref;
virtual void didConnect() override;
virtual void didReceiveMessage(const String& message) override;
virtual void didReceiveBinaryData(PassOwnPtr<Vector<char>>) override;
virtual void didReceiveMessageError() 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;
private:
explicit WebSocket(ScriptExecutionContext&);
virtual void contextDestroyed() override;
virtual bool canSuspend() const override;
virtual void suspend(ReasonForSuspension) override;
virtual void resume() override;
virtual void stop() override;
virtual void refEventTarget() override { ref(); }
virtual void derefEventTarget() override { deref(); }
size_t getFramingOverhead(size_t payloadSize);
enum BinaryType {
BinaryTypeBlob,
BinaryTypeArrayBuffer
};
RefPtr<ThreadableWebSocketChannel> m_channel;
State m_state;
URL m_url;
unsigned long m_bufferedAmount;
unsigned long m_bufferedAmountAfterClose;
BinaryType m_binaryType;
String m_subprotocol;
String m_extensions;
};
}
#endif // ENABLE(WEB_SOCKETS)
#endif // WebSocket_h