NetworkSocketChannel.h [plain text]
#pragma once
#include "MessageReceiver.h"
#include "MessageSender.h"
#include "WebSocketIdentifier.h"
#include <WebCore/Timer.h>
#include <pal/SessionID.h>
#include <wtf/CompletionHandler.h>
#include <wtf/WeakPtr.h>
namespace WebCore {
class ResourceRequest;
}
namespace IPC {
class Connection;
class Decoder;
class DataReference;
}
namespace WebKit {
class WebSocketTask;
class NetworkConnectionToWebProcess;
class NetworkProcess;
class NetworkSession;
class NetworkSocketChannel : public IPC::MessageSender, public IPC::MessageReceiver {
WTF_MAKE_FAST_ALLOCATED;
public:
static std::unique_ptr<NetworkSocketChannel> create(NetworkConnectionToWebProcess&, PAL::SessionID, const WebCore::ResourceRequest&, const String& protocol, WebSocketIdentifier);
NetworkSocketChannel(NetworkConnectionToWebProcess&, NetworkSession*, const WebCore::ResourceRequest&, const String& protocol, WebSocketIdentifier);
~NetworkSocketChannel();
void didReceiveMessage(IPC::Connection&, IPC::Decoder&);
friend class WebSocketTask;
private:
void didConnect(const String& subprotocol, const String& extensions);
void didReceiveText(const String&);
void didReceiveBinaryData(const uint8_t* data, size_t length);
void didClose(unsigned short code, const String& reason);
void didReceiveMessageError(const String&);
void didSendHandshakeRequest(WebCore::ResourceRequest&&);
void didReceiveHandshakeResponse(WebCore::ResourceResponse&&);
void sendString(const IPC::DataReference&, CompletionHandler<void()>&&);
void sendData(const IPC::DataReference&, CompletionHandler<void()>&&);
void close(int32_t code, const String& reason);
void sendDelayedError();
NetworkSession* session();
IPC::Connection* messageSenderConnection() const final;
uint64_t messageSenderDestinationID() const final { return m_identifier.toUInt64(); }
void finishClosingIfPossible();
NetworkConnectionToWebProcess& m_connectionToWebProcess;
WebSocketIdentifier m_identifier;
WeakPtr<NetworkSession> m_session;
std::unique_ptr<WebSocketTask> m_socket;
enum class State { Open, Closing, Closed };
State m_state { State::Open };
WebCore::Timer m_errorTimer;
String m_errorMessage;
Optional<std::pair<unsigned short, String>> m_closeInfo;
};
}