52 lines
1.1 KiB
C
52 lines
1.1 KiB
C
|
#ifndef ELFSPY_ARG_H
|
||
|
|
#define ELFSPY_ARG_H
|
||
|
|
|
||
|
|
#include "elfspy/Variadic.h"
|
||
|
|
#include "elfspy/Thunk.h"
|
||
|
|
#include "elfspy/ThunkHandle.h"
|
||
|
|
#include "elfspy/Capture.h"
|
||
|
|
|
||
|
|
namespace spy
|
||
|
|
{
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @namespace spy
|
||
|
|
* @class Arg
|
||
|
|
* This captures one argument of an invocation
|
||
|
|
*/
|
||
|
|
|
||
|
|
template <size_t N, typename... ArgTypes>
|
||
|
|
class Arg : public Thunk<ArgTypes...>,
|
||
|
|
public Capture<typename Variadic<N, ArgTypes...>::Type>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
virtual void invoke(ArgTypes&&... args) override;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <size_t N, typename... ArgTypes>
|
||
|
|
void Arg<N, ArgTypes...>::invoke(ArgTypes&&... args)
|
||
|
|
{
|
||
|
|
using Var = Variadic<N, ArgTypes...>;
|
||
|
|
// forward arguments to avoid copying
|
||
|
|
this->captures_.push_back(Var::get(std::forward<ArgTypes>(args)...));
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t N, typename H>
|
||
|
|
inline auto arg(H& hook)
|
||
|
|
-> ThunkHandle<typename H::template ExportN<Arg>::template Type<N>>
|
||
|
|
{
|
||
|
|
return hook.thunks();
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t N, typename H>
|
||
|
|
inline auto new_arg(H& hook)
|
||
|
|
-> ThunkHandle<typename H::template ExportN<Arg>::template Type<N>>*
|
||
|
|
{
|
||
|
|
using TP = ThunkHandle<typename H::template ExportN<Arg>::template Type<N>>;
|
||
|
|
return new TP(hook.thunks());
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace elfspy
|
||
|
|
|
||
|
|
#endif
|