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.

90 lines
2.3 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 "pin.H"
#include <stdio.h>
FILE *fp;
bool instrumented = FALSE;
unsigned int xmmInitVals[64];
extern "C" int SetXmmScratchesFun(unsigned int *values);
// Insert a call to an analysis routine that sets the scratch xmm registers, the call is inserted just after the
// movdqa instruction of DoXmm (see xmm-asm-*.s)
static VOID InstrumentRoutine(RTN rtn, VOID *)
{
if (RTN_Name(rtn) == "DoXmm")
{
RTN_Open(rtn);
for (INS ins = RTN_InsHead(rtn); INS_Valid(ins); ins = INS_Next(ins))
{
if (INS_Opcode(ins)==XED_ICLASS_MOVDQA)
{
fprintf (fp, "instrumenting ins %p %s\n", (void *)INS_Address(ins), INS_Disassemble(ins).c_str());
instrumented = TRUE;
INS_InsertCall(ins, IPOINT_AFTER, (AFUNPTR)SetXmmScratchesFun, IARG_PTR, xmmInitVals, IARG_END);
}
}
RTN_Close(rtn);
}
}
static void OnExit(INT32, VOID *)
{
if (!instrumented)
{
fprintf (fp, "***Error tool did not instrument the movdqa instruction of DoXmm\n");
fflush (fp);
PIN_ExitProcess(1);
}
else
{
fprintf (fp, "instrumented the movdqa instruction of DoXmm\n");
fflush (fp);
}
}
// argc, argv are the entire command line, including pin -t <toolname> -- ...
int main(int argc, char * argv[])
{
fp = fopen ("set_xmm_scratches.out", "w");
// initialize memory area used to set values in ymm regs
for (int i =0; i<64; i++)
{
xmmInitVals[i] = 0xdeadbeef;
}
PIN_InitSymbols();
// Initialize pin
PIN_Init(argc, argv);
// Register Instruction to be called to instrument the movdqa instruction of DoXmm
RTN_AddInstrumentFunction(InstrumentRoutine, 0);
PIN_AddFiniFunction(OnExit, 0);
// Start the program, never returns
PIN_StartProgram();
return 0;
}