AsmHelper64.h

#pragma once

#include "AsmHelperBase.h"
#include "../Include/Macro.h"

namespace blackbone
{

class AsmHelper64 : public AsmHelperBase
{
public:
    BLACKBONE_API AsmHelper64();
    BLACKBONE_API ~AsmHelper64( void );

    virtual void GenPrologue( bool switchMode = false );

    virtual void GenEpilogue( bool switchMode = false, int retSize = 0 );

    virtual void GenCall( const AsmVariant& pFN, const std::vector<AsmVariant>& args, eCalligConvention cc = cc_stdcall );

    virtual void ExitThreadWithStatus( uint64_t pExitThread, size_t resultPtr = 0 );

    virtual void SaveRetValAndSignalEvent( size_t pSetEvent, 
                                           size_t ResultPtr, 
                                           size_t EventPtr, 
                                           size_t lastStatusPtr, 
                                           eReturnType rtype = rt_int32 );

    virtual void EnableX64CallStack( bool state );

    virtual void SetTebPtr()
    {
        // mov rdx, gs:[0x30]   
        _assembler.mov( asmjit::host::rdx, asmjit::host::dword_ptr_abs( 0x30 ).setSegment( asmjit::host::gs ) );
    }

private:
    AsmHelper64( const AsmHelper64& ) = delete;
    AsmHelper64& operator = (const AsmHelper64&) = delete;

    void PushArg( const AsmVariant& arg, size_t index );

    template<typename _Type>
    void PushArgp( const _Type& arg, size_t index, bool fpu = false );

private:
    bool _stackEnabled;     // if true - GenCall will allocate shadow stack space
};

}