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.

94 lines
2.1 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.
*/
/*! @file
*/
#include "pin.H"
#include <iostream>
#include <fstream>
#include <stdlib.h>
#include <tool_macros.h>
using std::string;
using std::ios;
using std::hex;
using std::cerr;
using std::ofstream;
using std::endl;
/* ===================================================================== */
/* Commandline Switches */
/* ===================================================================== */
KNOB<string> KnobOutputFile(KNOB_MODE_WRITEONCE, "pintool",
"o", "attach_tool.out", "specify file name");
ofstream TraceFile;
/* ===================================================================== */
INT32 Usage()
{
cerr <<
"This pin tool tests attach to single-threaded application JIT mode.\n"
"\n";
cerr << KNOB_BASE::StringKnobSummary();
cerr << endl;
return -1;
}
BOOL PinAttached()
{
return TRUE;
}
VOID ImageLoad(IMG img, void *v)
{
RTN rtn = RTN_FindByName(img, C_MANGLE("PinAttached"));
if (RTN_Valid(rtn))
{
TraceFile << "Replacing PinAttached" << endl;
RTN_Replace(rtn, AFUNPTR(PinAttached));
}
}
void Fini(INT32 code, VOID *v)
{
TraceFile << "Fini" << endl;
}
/* ===================================================================== */
int main(int argc, CHAR *argv[])
{
PIN_InitSymbols();
if( PIN_Init(argc,argv) )
{
return Usage();
}
TraceFile.open(KnobOutputFile.Value().c_str());
TraceFile << hex;
TraceFile.setf(ios::showbase);
IMG_AddInstrumentFunction(ImageLoad, 0);
PIN_AddFiniFunction(Fini, 0);
PIN_StartProgram();
return 0;
}
/* ===================================================================== */
/* eof */
/* ===================================================================== */