#pragma once
#include "ArgumentCoders.h"
#include <wtf/CompletionHandler.h>
#include <wtf/StdLibExtras.h>
namespace IPC {
class Connection;
template <typename C, typename MF, typename ArgsTuple, size_t... ArgsIndex>
void callMemberFunctionImpl(C* object, MF function, ArgsTuple&& args, std::index_sequence<ArgsIndex...>)
{
(object->*function)(std::get<ArgsIndex>(std::forward<ArgsTuple>(args))...);
}
template<typename C, typename MF, typename ArgsTuple, typename ArgsIndicies = std::make_index_sequence<std::tuple_size<ArgsTuple>::value>>
void callMemberFunction(ArgsTuple&& args, C* object, MF function)
{
callMemberFunctionImpl(object, function, std::forward<ArgsTuple>(args), ArgsIndicies());
}
template <typename C, typename MF, typename ArgsTuple, size_t... ArgsIndex, typename ReplyArgsTuple, size_t... ReplyArgsIndex>
void callMemberFunctionImpl(C* object, MF function, ArgsTuple&& args, ReplyArgsTuple& replyArgs, std::index_sequence<ArgsIndex...>, std::index_sequence<ReplyArgsIndex...>)
{
(object->*function)(std::get<ArgsIndex>(std::forward<ArgsTuple>(args))..., std::get<ReplyArgsIndex>(replyArgs)...);
}
template <typename C, typename MF, typename ArgsTuple, typename ArgsIndicies = std::make_index_sequence<std::tuple_size<ArgsTuple>::value>, typename ReplyArgsTuple, typename ReplyArgsIndicies = std::make_index_sequence<std::tuple_size<ReplyArgsTuple>::value>>
void callMemberFunction(ArgsTuple&& args, ReplyArgsTuple& replyArgs, C* object, MF function)
{
callMemberFunctionImpl(object, function, std::forward<ArgsTuple>(args), replyArgs, ArgsIndicies(), ReplyArgsIndicies());
}
template <typename C, typename MF, typename CH, typename ArgsTuple, size_t... ArgsIndex>
void callMemberFunctionImpl(C* object, MF function, CompletionHandler<CH>&& completionHandler, ArgsTuple&& args, std::index_sequence<ArgsIndex...>)
{
(object->*function)(std::get<ArgsIndex>(std::forward<ArgsTuple>(args))..., WTFMove(completionHandler));
}
template<typename C, typename MF, typename CH, typename ArgsTuple, typename ArgsIndicies = std::make_index_sequence<std::tuple_size<ArgsTuple>::value>>
void callMemberFunction(ArgsTuple&& args, CompletionHandler<CH>&& completionHandler, C* object, MF function)
{
callMemberFunctionImpl(object, function, WTFMove(completionHandler), std::forward<ArgsTuple>(args), ArgsIndicies());
}
template <typename C, typename MF, typename ArgsTuple, size_t... ArgsIndex>
void callMemberFunctionImpl(C* object, MF function, Connection& connection, ArgsTuple&& args, std::index_sequence<ArgsIndex...>)
{
(object->*function)(connection, std::get<ArgsIndex>(std::forward<ArgsTuple>(args))...);
}
template<typename C, typename MF, typename ArgsTuple, typename ArgsIndicies = std::make_index_sequence<std::tuple_size<ArgsTuple>::value>>
void callMemberFunction(Connection& connection, ArgsTuple&& args, C* object, MF function)
{
callMemberFunctionImpl(object, function, connection, std::forward<ArgsTuple>(args), ArgsIndicies());
}
template <typename C, typename MF, typename ArgsTuple, size_t... ArgsIndex, typename ReplyArgsTuple, size_t... ReplyArgsIndex>
void callMemberFunctionImpl(C* object, MF function, Connection& connection, ArgsTuple&& args, ReplyArgsTuple& replyArgs, std::index_sequence<ArgsIndex...>, std::index_sequence<ReplyArgsIndex...>)
{
(object->*function)(connection, std::get<ArgsIndex>(std::forward<ArgsTuple>(args))..., std::get<ReplyArgsIndex>(replyArgs)...);
}
template <typename C, typename MF, typename ArgsTuple, typename ArgsIndicies = std::make_index_sequence<std::tuple_size<ArgsTuple>::value>, typename ReplyArgsTuple, typename ReplyArgsIndicies = std::make_index_sequence<std::tuple_size<ReplyArgsTuple>::value>>
void callMemberFunction(Connection& connection, ArgsTuple&& args, ReplyArgsTuple& replyArgs, C* object, MF function)
{
callMemberFunctionImpl(object, function, connection, std::forward<ArgsTuple>(args), replyArgs, ArgsIndicies(), ReplyArgsIndicies());
}
template<typename T>
struct CodingType {
typedef std::remove_const_t<std::remove_reference_t<T>> Type;
};
class DataReference;
class SharedBufferDataReference;
template<> struct CodingType<const SharedBufferDataReference&> {
typedef DataReference Type;
};
template<typename... Ts>
struct CodingType<std::tuple<Ts...>> {
typedef std::tuple<typename CodingType<Ts>::Type...> Type;
};
template<typename T, typename C, typename MF>
void handleMessage(Decoder& decoder, C* object, MF function)
{
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
callMemberFunction(WTFMove(arguments), object, function);
}
template<typename T, typename C, typename MF>
void handleMessageLegacySync(Decoder& decoder, Encoder& replyEncoder, C* object, MF function)
{
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
typename CodingType<typename T::Reply>::Type replyArguments;
callMemberFunction(WTFMove(arguments), replyArguments, object, function);
replyEncoder << replyArguments;
}
template<typename T, typename C, typename MF>
void handleMessageLegacySync(Connection& connection, Decoder& decoder, Encoder& replyEncoder, C* object, MF function)
{
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
typename CodingType<typename T::Reply>::Type replyArguments;
callMemberFunction(connection, WTFMove(arguments), replyArguments, object, function);
replyEncoder << replyArguments;
}
template<typename T, typename C, typename MF>
void handleMessage(Connection& connection, Decoder& decoder, C* object, MF function)
{
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
callMemberFunction(connection, WTFMove(arguments), object, function);
}
template<typename T, typename C, typename MF>
void handleMessageDelayed(Connection& connection, Decoder& decoder, std::unique_ptr<Encoder>& replyEncoder, C* object, MF function)
{
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
typename T::DelayedReply completionHandler = [replyEncoder = WTFMove(replyEncoder), connection = makeRef(connection)] (auto&&... args) mutable {
T::send(WTFMove(replyEncoder), WTFMove(connection), args...);
};
callMemberFunction(WTFMove(arguments), WTFMove(completionHandler), object, function);
}
template<typename T, typename C, typename MF>
void handleMessageAsync(Connection& connection, Decoder& decoder, C* object, MF function)
{
Optional<uint64_t> listenerID;
decoder >> listenerID;
if (!listenerID) {
ASSERT(decoder.isInvalid());
return;
}
typename CodingType<typename T::Arguments>::Type arguments;
if (!decoder.decode(arguments)) {
ASSERT(decoder.isInvalid());
return;
}
typename T::AsyncReply completionHandler = [listenerID = *listenerID, connection = makeRef(connection)] (auto&&... args) mutable {
auto encoder = std::make_unique<Encoder>("AsyncReply", T::asyncMessageReplyName(), 0);
*encoder << listenerID;
T::send(WTFMove(encoder), WTFMove(connection), args...);
};
callMemberFunction(WTFMove(arguments), WTFMove(completionHandler), object, function);
}
}