MessageRecorder.cpp [plain text]
#include "config.h"
#include "MessageRecorder.h"
#if HAVE(DTRACE)
#include "Connection.h"
#include "MessageDecoder.h"
#include "MessageEncoder.h"
#include "MessageRecorderProbes.h"
#include <wtf/CurrentTime.h>
namespace IPC {
bool MessageRecorder::isEnabled()
{
return WEBKITMESSAGERECORDER_MESSAGE_RECEIVED_ENABLED() || WEBKITMESSAGERECORDER_MESSAGE_SENT_ENABLED();
}
std::unique_ptr<MessageRecorder::MessageProcessingToken> MessageRecorder::recordOutgoingMessage(Connection& connection, MessageEncoder& encoder)
{
if (!isEnabled() || !connection.isValid())
return nullptr;
WebKitMessageRecord record;
record.sourceProcessType = static_cast<uint64_t>(connection.client()->localProcessType());
record.destinationProcessType = static_cast<uint64_t>(connection.client()->remoteProcessType());
record.destinationID = encoder.destinationID();
record.isSyncMessage = encoder.isSyncMessage();
record.shouldDispatchMessageWhenWaitingForSyncReply = encoder.shouldDispatchMessageWhenWaitingForSyncReply();
record.sourceProcessID = getpid();
record.destinationProcessID = connection.remoteProcessID();
record.isIncoming = false;
record.messageReceiverName = MallocPtr<char>::malloc(sizeof(char) * (encoder.messageReceiverName().size() + 1));
strncpy(record.messageReceiverName.get(), encoder.messageReceiverName().data(), encoder.messageReceiverName().size());
record.messageReceiverName.get()[encoder.messageReceiverName().size()] = 0;
record.messageName = MallocPtr<char>::malloc(sizeof(char) * (encoder.messageName().size() + 1));
strncpy(record.messageName.get(), encoder.messageName().data(), encoder.messageName().size());
record.messageName.get()[encoder.messageName().size()] = 0;
uuid_copy(record.UUID, encoder.UUID());
return std::make_unique<MessageProcessingToken>(WTFMove(record));
}
void MessageRecorder::recordIncomingMessage(Connection& connection, MessageDecoder& decoder)
{
if (!isEnabled() || !connection.isValid())
return;
WebKitMessageRecord record;
record.sourceProcessType = static_cast<uint64_t>(connection.client()->remoteProcessType());
record.destinationProcessType = static_cast<uint64_t>(connection.client()->localProcessType());
record.destinationID = decoder.destinationID();
record.isSyncMessage = decoder.isSyncMessage();
record.shouldDispatchMessageWhenWaitingForSyncReply = decoder.shouldDispatchMessageWhenWaitingForSyncReply();
record.sourceProcessID = connection.remoteProcessID();
record.destinationProcessID = getpid();
record.isIncoming = true;
record.messageReceiverName = MallocPtr<char>::malloc(sizeof(char) * (decoder.messageReceiverName().size() + 1));
strncpy(record.messageReceiverName.get(), decoder.messageReceiverName().data(), decoder.messageReceiverName().size());
record.messageReceiverName.get()[decoder.messageReceiverName().size()] = 0;
record.messageName = MallocPtr<char>::malloc(sizeof(char) * (decoder.messageName().size() + 1));
strncpy(record.messageName.get(), decoder.messageName().data(), decoder.messageName().size());
record.messageName.get()[decoder.messageName().size()] = 0;
uuid_copy(record.UUID, decoder.UUID());
decoder.setMessageProcessingToken(std::make_unique<MessageProcessingToken>(WTFMove(record)));
}
#pragma mark MessageProcessingToken
MessageRecorder::MessageProcessingToken::MessageProcessingToken(WebKitMessageRecord record)
: m_record(WTFMove(record))
{
m_record.startTime = monotonicallyIncreasingTime();
}
MessageRecorder::MessageProcessingToken::~MessageProcessingToken()
{
m_record.endTime = monotonicallyIncreasingTime();
if (m_record.isIncoming)
WEBKITMESSAGERECORDER_MESSAGE_RECEIVED(&m_record);
else
WEBKITMESSAGERECORDER_MESSAGE_SENT(&m_record);
}
}
#endif // HAVE(DTRACE)