122 lines
3.0 KiB
C++
122 lines
3.0 KiB
C++
/**
|
|
@file assemble.cpp
|
|
|
|
@brief Implements the assemble class.
|
|
*/
|
|
|
|
#include "assemble.h"
|
|
#include "memory.h"
|
|
#include "debugger.h"
|
|
#include "XEDParse\XEDParse.h"
|
|
#include "value.h"
|
|
#include "disasm_helper.h"
|
|
#include "console.h"
|
|
|
|
/**
|
|
@fn static bool cbUnknown(const char* text, ULONGLONG* value)
|
|
|
|
@brief Unknowns.
|
|
|
|
@param text The text.
|
|
@param [in,out] value If non-null, the value.
|
|
|
|
@return true if it succeeds, false if it fails.
|
|
*/
|
|
|
|
static bool cbUnknown(const char* text, ULONGLONG* value)
|
|
{
|
|
if(!text or !value)
|
|
return false;
|
|
uint val;
|
|
if(!valfromstring(text, &val))
|
|
return false;
|
|
*value = val;
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
@fn bool assemble(uint addr, unsigned char* dest, int* size, const char* instruction, char* error)
|
|
|
|
@brief Assembles.
|
|
|
|
@param addr The address.
|
|
@param [in,out] dest If non-null, destination for the.
|
|
@param [in,out] size If non-null, the size.
|
|
@param instruction The instruction.
|
|
@param [in,out] error If non-null, the error.
|
|
|
|
@return true if it succeeds, false if it fails.
|
|
*/
|
|
|
|
bool assemble(uint addr, unsigned char* dest, int* size, const char* instruction, char* error)
|
|
{
|
|
if(strlen(instruction) >= XEDPARSE_MAXBUFSIZE)
|
|
return false;
|
|
XEDPARSE parse;
|
|
memset(&parse, 0, sizeof(parse));
|
|
#ifdef _WIN64
|
|
parse.x64 = true;
|
|
#else //x86
|
|
parse.x64 = false;
|
|
#endif
|
|
parse.cbUnknown = cbUnknown;
|
|
parse.cip = addr;
|
|
strcpy(parse.instr, instruction);
|
|
if(XEDParseAssemble(&parse) == XEDPARSE_ERROR)
|
|
{
|
|
if(error)
|
|
strcpy(error, parse.error);
|
|
return false;
|
|
}
|
|
|
|
if(dest)
|
|
memcpy(dest, parse.dest, parse.dest_size);
|
|
if(size)
|
|
*size = parse.dest_size;
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
@fn bool assembleat(uint addr, const char* instruction, int* size, char* error, bool fillnop)
|
|
|
|
@brief Assembleats.
|
|
|
|
@param addr The address.
|
|
@param instruction The instruction.
|
|
@param [in,out] size If non-null, the size.
|
|
@param [in,out] error If non-null, the error.
|
|
@param fillnop true to fillnop.
|
|
|
|
@return true if it succeeds, false if it fails.
|
|
*/
|
|
|
|
bool assembleat(uint addr, const char* instruction, int* size, char* error, bool fillnop)
|
|
{
|
|
int destSize;
|
|
unsigned char dest[16];
|
|
if(!assemble(addr, dest, &destSize, instruction, error))
|
|
return false;
|
|
//calculate the number of NOPs to insert
|
|
int origLen = disasmgetsize(addr);
|
|
while(origLen < destSize)
|
|
origLen += disasmgetsize(addr + origLen);
|
|
int nopsize = origLen - destSize;
|
|
unsigned char nops[16];
|
|
memset(nops, 0x90, sizeof(nops));
|
|
|
|
if(size)
|
|
*size = destSize;
|
|
|
|
bool ret = mempatch(fdProcessInfo->hProcess, (void*)addr, dest, destSize, 0);
|
|
if(ret && fillnop && nopsize)
|
|
{
|
|
if(size)
|
|
*size += nopsize;
|
|
if(!mempatch(fdProcessInfo->hProcess, (void*)(addr + destSize), nops, nopsize, 0))
|
|
ret = false;
|
|
}
|
|
GuiUpdatePatches();
|
|
return true;
|
|
}
|