• 参考样例工具malloctrace的代码（下载解压Pin就有）
• 修改工程模板MyPinTool （下载解压Pin就有）
• 代码简介：
• 在main中使用IMG_AddInstrumentFunction(Image, 0);进行函数插桩
• 在 Image 函数总插桩了5个函数：LoadLibraryExW、LoadLibraryExA、LoadLibraryW、LoadLibraryA、GetProcAddress，都插桩在函数调用前。
• 插桩的函数（分ascii版和unicode版）打印被插桩函数的第一个参数：加载的库路径或者要获取地址的函数。
• 结果写在当前目录的LoadlibraryArguments.out内。
  

#include "pin.H"
#include <iostream>
#include <fstream>

#include <stdio.h>
#include <unistd.h>

#define LOADLIBRARYEXW "LoadLibraryExW"
#define LOADLIBRARYEXA "LoadLibraryExA"
#define LOADLIBRARYW "LoadLibraryW"
#define LOADLIBRARYA "LoadLibraryA"
#define GETPROCADDRESS "GetProcAddress"


/* ===================================================================== */
/* Global Variables */
/* ===================================================================== */

std::ofstream TraceFile;

/* ===================================================================== */
/* Commandline Switches */
/* ===================================================================== */

KNOB<string> KnobOutputFile(KNOB_MODE_WRITEONCE, "pintool",
    "o", "LoadlibraryArguments.out", "specify trace file name");

/* ===================================================================== */


/* ===================================================================== */
/* Analysis routines                                                     */
/* ===================================================================== */
 
VOID ArgBeforeA(CHAR * name, CHAR * size)
{
	//TraceFile << "in" << endl;
    TraceFile << name << "(" << size << ")" << endl;
}

VOID ArgBeforeA1(CHAR * name, CHAR * size)
{
	//TraceFile << "in" << endl;
	//这里判断是为了排除第一个参数为非法参数的情况，不判断的话，在读出现异常时会进程提出，暂时不知道Pin中如何做异常处理
	if((INT)size > 0xFFFF)
		TraceFile << "------" << name << "(" << size << ")" << endl;
}


VOID ArgBeforeW(CHAR * name, CHAR * size)
{
	//TraceFile << "in" << endl;
    TraceFile << name << "(" ;
	//*
	while(1)
	{
		if(size[0] == 0x00 && size[1] == 0x00)
		{
			break;
		}
		TraceFile << size;
		size++;
		size++;
	}
	//*/
	TraceFile << ")" << endl;
	return;	
	//<< size <<
}


/* ===================================================================== */
/* Instrumentation routines                                              */
/* ===================================================================== */
   
VOID Image(IMG img, VOID *v)
{
    // Instrument the malloc() and free() functions.  Print the input argument
    // of each malloc() or free(), and the return value of malloc().
    //
    //  Find the malloc() function.
	//*/
    RTN LoadLibraryExWRtn = RTN_FindByName(img, LOADLIBRARYEXW);
    if (RTN_Valid(LoadLibraryExWRtn))
    {
        RTN_Open(LoadLibraryExWRtn);

        // Instrument malloc() to print the input argument value and the return value.
        RTN_InsertCall(LoadLibraryExWRtn, IPOINT_BEFORE, (AFUNPTR)ArgBeforeW,
                       IARG_ADDRINT, LOADLIBRARYEXW,
                       IARG_FUNCARG_ENTRYPOINT_VALUE, 0,
                       IARG_END);

        RTN_Close(LoadLibraryExWRtn);
    }

	RTN LoadLibraryExARtn = RTN_FindByName(img, LOADLIBRARYEXA);
    if (RTN_Valid(LoadLibraryExARtn))
    {
        RTN_Open(LoadLibraryExARtn);

        // Instrument malloc() to print the input argument value and the return value.
        RTN_InsertCall(LoadLibraryExARtn, IPOINT_BEFORE, (AFUNPTR)ArgBeforeA,
                       IARG_ADDRINT, LOADLIBRARYEXA,
                       IARG_FUNCARG_ENTRYPOINT_VALUE, 0,
                       IARG_END);

        RTN_Close(LoadLibraryExARtn);
    }
	
	RTN LoadLibraryWRtn = RTN_FindByName(img, LOADLIBRARYW);
    if (RTN_Valid(LoadLibraryWRtn))
    {
        RTN_Open(LoadLibraryWRtn);

        // Instrument malloc() to print the input argument value and the return value.
        RTN_InsertCall(LoadLibraryWRtn, IPOINT_BEFORE, (AFUNPTR)ArgBeforeW,
                       IARG_ADDRINT, LOADLIBRARYW,
                       IARG_FUNCARG_ENTRYPOINT_VALUE, 0,
                       IARG_END);

        RTN_Close(LoadLibraryWRtn);
    }
	//*/

	RTN LoadLibraryARtn = RTN_FindByName(img, LOADLIBRARYA);
    if (RTN_Valid(LoadLibraryARtn))
    {
        RTN_Open(LoadLibraryARtn);

        // Instrument malloc() to print the input argument value and the return value.
        RTN_InsertCall(LoadLibraryARtn, IPOINT_BEFORE, (AFUNPTR)ArgBeforeA,
                       IARG_ADDRINT, LOADLIBRARYA,
                       IARG_FUNCARG_ENTRYPOINT_VALUE , 0,
                       IARG_END);

        RTN_Close(LoadLibraryARtn);
    }
	
	//*
	RTN GetProcAddressRtn = RTN_FindByName(img, GETPROCADDRESS);
    if (RTN_Valid(GetProcAddressRtn))
    {
        RTN_Open(GetProcAddressRtn);

        // Instrument malloc() to print the input argument value and the return value.
        RTN_InsertCall(GetProcAddressRtn, IPOINT_BEFORE, (AFUNPTR)ArgBeforeA1,
                       IARG_ADDRINT, GETPROCADDRESS,
                       IARG_FUNCARG_ENTRYPOINT_VALUE , 1,
                       IARG_END);

        RTN_Close(GetProcAddressRtn);
    }
	//*/
}

/* ===================================================================== */

VOID Fini(INT32 code, VOID *v)
{
    TraceFile.close();
}

/* ===================================================================== */
/* Print Help Message                                                    */
/* ===================================================================== */
   
INT32 Usage()
{
    cerr << "This tool produces a trace of calls to malloc." << endl;
    cerr << endl << KNOB_BASE::StringKnobSummary() << endl;
    return -1;
}

/* ===================================================================== */
/* Main                                                                  */
/* ===================================================================== */

int main(int argc, char *argv[])
{
    // Initialize pin & symbol manager
    PIN_InitSymbols();
    if( PIN_Init(argc,argv) )
    {
        return Usage();
    }
	
    // Write to a file since cout and cerr maybe closed by the application
    TraceFile.open(KnobOutputFile.Value().c_str());
    TraceFile << hex;
    TraceFile.setf(ios::showbase);
    
    // Register Image to be called to instrument functions.
    IMG_AddInstrumentFunction(Image, 0);
    PIN_AddFiniFunction(Fini, 0);

    // Never returns
    PIN_StartProgram();
    
    return 0;
}

/* ===================================================================== */
/* eof */
/* ===================================================================== */