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.

110 lines
3.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.
*/
/*
* This test checks SYM_Address() interface on Linux
*/
#include <iostream>
#include <fstream>
#include <stdio.h>
#include <stdlib.h>
#include "pin.H"
using std::hex;
using std::ofstream;
using std::cerr;
using std::string;
using std::endl;
KNOB<string> KnobOutputFile(KNOB_MODE_WRITEONCE, "pintool",
"o", "elf_symbols.out", "specify output file name");
ofstream outfile;
VOID ImageLoad(IMG img, void *v)
{
for( SYM sym = IMG_RegsymHead(img); SYM_Valid(sym); sym = SYM_Next(sym) )
{
string symPureName = PIN_UndecorateSymbolName(SYM_Name(sym), UNDECORATION_NAME_ONLY);
if (symPureName == "fopen")
{
RTN fopenRtn = RTN_FindByAddress( SYM_Address(sym));
if (!RTN_Valid(fopenRtn))
{
cerr << "Routine " << symPureName << " is not found at address 0x"
<< hex << SYM_Address(sym) << endl;
exit(-1);
}
else
{
outfile << "Routine " << SYM_Name(sym) << " is found at address 0x"
<< hex << SYM_Address(sym) << endl;
}
}
if (symPureName == "malloc")
{
RTN mallocRtn = RTN_FindByAddress( SYM_Address(sym));
if (!RTN_Valid(mallocRtn))
{
cerr << "Routine " << symPureName << " is not found at address 0x"
<< hex << SYM_Address(sym) << endl;
exit(-1);
}
else
{
outfile << "Routine " << SYM_Name(sym) << " is found at address 0x"
<< hex << SYM_Address(sym) << endl;
}
}
}
}
VOID Fini(INT32 code, VOID *v)
{
outfile << "Symbol test passed successfully" << endl;
outfile.close();
}
/* ===================================================================== */
/* Print Help Message. */
/* ===================================================================== */
INT32 Usage()
{
cerr << "This is the invocation pintool" << endl;
cerr << endl << KNOB_BASE::StringKnobSummary() << endl;
return -1;
}
/* ===================================================================== */
/* Main. */
/* ===================================================================== */
int main(int argc, char * argv[])
{
PIN_InitSymbols();
if (PIN_Init(argc, argv)) return Usage();
outfile.open(KnobOutputFile.Value().c_str());
IMG_AddInstrumentFunction(ImageLoad, 0);
PIN_AddFiniFunction(Fini, 0);
PIN_StartProgram();
return 0;
}
/* ================================================================== */