RemoteInspectorConnectionClient.cpp [plain text]
#include "config.h"
#include "RemoteInspectorConnectionClient.h"
#if ENABLE(REMOTE_INSPECTOR)
#include "RemoteInspectorSocketEndpoint.h"
#include <wtf/JSONValues.h>
#include <wtf/RunLoop.h>
namespace Inspector {
RemoteInspectorConnectionClient::~RemoteInspectorConnectionClient()
{
auto& endpoint = Inspector::RemoteInspectorSocketEndpoint::singleton();
endpoint.invalidateClient(*this);
}
Optional<ConnectionID> RemoteInspectorConnectionClient::connectInet(const char* serverAddr, uint16_t serverPort)
{
auto& endpoint = Inspector::RemoteInspectorSocketEndpoint::singleton();
return endpoint.connectInet(serverAddr, serverPort, *this);
}
Optional<ConnectionID> RemoteInspectorConnectionClient::createClient(PlatformSocketType socket)
{
auto& endpoint = Inspector::RemoteInspectorSocketEndpoint::singleton();
return endpoint.createClient(socket, *this);
}
void RemoteInspectorConnectionClient::send(ConnectionID id, const uint8_t* data, size_t size)
{
auto message = MessageParser::createMessage(data, size);
if (message.isEmpty())
return;
auto& endpoint = RemoteInspectorSocketEndpoint::singleton();
endpoint.send(id, message.data(), message.size());
}
void RemoteInspectorConnectionClient::didReceive(ConnectionID clientID, Vector<uint8_t>&& data)
{
ASSERT(!isMainThread());
LockHolder lock(m_parsersLock);
auto result = m_parsers.ensure(clientID, [this, clientID] {
return MessageParser([this, clientID](Vector<uint8_t>&& data) {
if (auto event = RemoteInspectorConnectionClient::extractEvent(clientID, WTFMove(data))) {
RunLoop::main().dispatch([this, event = WTFMove(*event)] {
const auto& methodName = event.methodName;
auto& methods = dispatchMap();
if (methods.contains(methodName)) {
auto call = methods.get(methodName);
(this->*call)(event);
} else
LOG_ERROR("Unknown event: %s", methodName.utf8().data());
});
}
});
});
result.iterator->value.pushReceivedData(data.data(), data.size());
}
Optional<RemoteInspectorConnectionClient::Event> RemoteInspectorConnectionClient::extractEvent(ConnectionID clientID, Vector<uint8_t>&& data)
{
if (data.isEmpty())
return WTF::nullopt;
String jsonData = String::fromUTF8(data);
RefPtr<JSON::Value> messageValue;
if (!JSON::Value::parseJSON(jsonData, messageValue))
return WTF::nullopt;
RefPtr<JSON::Object> messageObject;
if (!messageValue->asObject(messageObject))
return WTF::nullopt;
Event event;
if (!messageObject->getString("event"_s, event.methodName))
return WTF::nullopt;
event.clientID = clientID;
ConnectionID connectionID;
if (messageObject->getInteger("connectionID"_s, connectionID))
event.connectionID = connectionID;
TargetID targetID;
if (messageObject->getInteger("targetID"_s, targetID))
event.targetID = targetID;
String message;
if (messageObject->getString("message"_s, message))
event.message = message;
return event;
}
}
#endif // ENABLE(REMOTE_INSPECTOR)