/* * 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 tool verifies that the registered image unload callbacks are being called * before the image was unloaded. */ #include #include #include #include #include #include #include "pin.H" using std::endl; using std::string; using std::cout; const string SO_NAME = "one.so"; const int BUFF_SIZE = 1024; VOID ImageUnload(IMG img, VOID *v) { size_t found1=IMG_Name(img).find(SO_NAME); if (found1!=string::npos) { // verify that "one.so" is in the mapped memory regions of the process. string mapFile = string("/proc/") + decstr(getpid()) + "/maps"; FILE *fp = fopen(mapFile.c_str(), "r"); char buff[BUFF_SIZE]; while(fgets(buff, BUFF_SIZE, fp) != NULL) { if (strstr(buff, SO_NAME.c_str())!=0) { cout << "one.so is mapped in the memory." << endl; } } fclose(fp); } } /* ===================================================================== */ /* Print Help Message */ /* ===================================================================== */ static INT32 Usage() { PIN_ERROR(" This tool verifies that the registered image unload callback is called when an image has been unloaded\n"); return -1; } /* ===================================================================== */ /* Main */ /* ===================================================================== */ int main(int argc, char * argv[]) { // Initialize symbol processing PIN_InitSymbols(); // Initialize pin if (PIN_Init(argc, argv)) return Usage(); // Register ImageUnload to be called when an image is unloaded IMG_AddUnloadFunction(ImageUnload, 0); // Start the program, never returns PIN_StartProgram(); return 0; }