You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.8 KiB
124 lines
3.8 KiB
/*
|
|
* Copyright 2002-2019 Intel Corporation.
|
|
*
|
|
* This software is provided to you as Sample Source Code as defined in the accompanying
|
|
* End User License Agreement for the Intel(R) Software Development Products ("Agreement")
|
|
* section 1.L.
|
|
*
|
|
* This software and the related documents are provided as is, with no express or implied
|
|
* warranties, other than those that are expressly stated in the License.
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include "pin.H"
|
|
using std::hex;
|
|
using std::cerr;
|
|
using std::endl;
|
|
using std::cout;
|
|
|
|
/*
|
|
* This tool tests correctness of the IARG_MEMORYOP_PTR value in instructions
|
|
* with rewritten memory operands.
|
|
* Note, this tool does not use PIN_GetMemoryAddressTransFunction to register
|
|
* memory translation callback.
|
|
*/
|
|
|
|
// Temporary registers used to rewrite memory operands
|
|
REG rewriteReg[2];
|
|
|
|
/*
|
|
* @return two's complement of the specified value
|
|
*/
|
|
static ADDRINT NegValue(ADDRINT val)
|
|
{
|
|
return 0 - val;
|
|
}
|
|
|
|
/*
|
|
* Verify correctness of the IARG_MEMORYOP_PTR calculation
|
|
* @param[in] ea IARG_MEMORYOP_EA value
|
|
* @param[in] ptr IARG_MEMORYOP_PTR value
|
|
* @pre (ptr == NegValue(ea))
|
|
*/
|
|
static VOID CheckMemPtr(ADDRINT ea, ADDRINT ptr)
|
|
{
|
|
if (ptr != NegValue(ea))
|
|
{
|
|
cerr << "Wrong IARG_MEMORYOP_PTR value: " << hex << ptr << " != " << NegValue(ea) << endl;
|
|
PIN_ExitProcess(1);
|
|
}
|
|
}
|
|
|
|
// Pin calls this function every time a new instruction is encountered
|
|
VOID Instruction(INS ins, VOID *v)
|
|
{
|
|
UINT32 memOps = INS_MemoryOperandCount(ins);
|
|
if (memOps > 2) { memOps = 2; } // maximum two rewritten operands
|
|
|
|
for (UINT32 i = 0 ; i < memOps; i++)
|
|
{
|
|
// Set rewriteReg[i] = -IARG_MEMORYOP_EA(i)
|
|
INS_InsertCall(ins, IPOINT_BEFORE, (AFUNPTR)NegValue,
|
|
IARG_CALL_ORDER, CALL_ORDER_DEFAULT,
|
|
IARG_MEMORYOP_EA, (ADDRINT)i,
|
|
IARG_RETURN_REGS, rewriteReg[i],
|
|
IARG_END);
|
|
|
|
// Check to see that IARG_MEMORYOP_PTR(i) == -IARG_MEMORYOP_EA(i)
|
|
INS_InsertCall(ins, IPOINT_BEFORE, (AFUNPTR)CheckMemPtr,
|
|
IARG_CALL_ORDER, CALL_ORDER_DEFAULT + 1,
|
|
IARG_MEMORYOP_EA, (ADDRINT)i,
|
|
IARG_MEMORYOP_PTR, (ADDRINT)i,
|
|
IARG_END);
|
|
|
|
// Revert the rewriteReg[i] value back to IARG_MEMORYOP_EA(i)
|
|
INS_InsertCall(ins, IPOINT_BEFORE, (AFUNPTR)NegValue,
|
|
IARG_CALL_ORDER, CALL_ORDER_DEFAULT + 2,
|
|
IARG_REG_VALUE, rewriteReg[i],
|
|
IARG_RETURN_REGS, rewriteReg[i],
|
|
IARG_END);
|
|
|
|
// Rewrite i-th memory operand (with no EA change)
|
|
INS_RewriteMemoryOperand(ins, i, rewriteReg[i]);
|
|
}
|
|
}
|
|
|
|
|
|
// This function is called when the application exits
|
|
VOID Fini(INT32 code, VOID *v)
|
|
{
|
|
cout << "Done!" << endl;
|
|
}
|
|
|
|
/* ===================================================================== */
|
|
/* Print Help Message */
|
|
/* ===================================================================== */
|
|
|
|
INT32 Usage()
|
|
{
|
|
cerr << "This tool tests IARG_MEMORY*_PTR arguments for rewritten memory operands" << endl;
|
|
return -1;
|
|
}
|
|
|
|
/* ===================================================================== */
|
|
/* Main */
|
|
/* ===================================================================== */
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
// Initialize pin
|
|
if (PIN_Init(argc, argv)) return Usage();
|
|
|
|
rewriteReg[0] = PIN_ClaimToolRegister();
|
|
rewriteReg[1] = PIN_ClaimToolRegister();
|
|
|
|
// Register Instruction to be called to instrument instructions
|
|
INS_AddInstrumentFunction(Instruction, 0);
|
|
|
|
// Register Fini to be called when the application exits
|
|
PIN_AddFiniFunction(Fini, 0);
|
|
|
|
// Start the program, never returns
|
|
PIN_StartProgram();
|
|
}
|