/* * 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 #include #include #ifdef TARGET_WINDOWS namespace WIND { #include } #define EXPORT_CSYM extern "C" __declspec( dllexport ) #else #define EXPORT_CSYM extern "C" #endif #ifdef TARGET_WINDOWS #define ASMNAME(name) #else #define ASMNAME(name) asm(name) #endif #if defined(__cplusplus) #define EXTERN_C extern "C" #else #define EXTERN_C #endif EXTERN_C BOOL ProcessorSupportsAvx() ASMNAME("ProcessorSupportsAvx"); EXTERN_C BOOL SupportsAvx512f() ASMNAME("SupportsAvx512f"); EXTERN_C VOID FldzToTop3() ASMNAME("FldzToTop3"); EXTERN_C VOID SetXmmRegsToZero() ASMNAME("SetXmmRegsToZero"); EXTERN_C VOID SetIntRegsToZero() ASMNAME("SetIntRegsToZero"); EXTERN_C void UnMaskZeroDivideInMxcsr() ASMNAME("UnMaskZeroDivideInMxcsr"); EXTERN_C double var1; EXTERN_C double var2; EXTERN_C double var3; double var1=2.0; double var2=2.0; double var3=2.0; //ALIGN16 static UINT64 vals[]= {1,0,2,0,3,0,4,0,5,0,6,0,7,0,8,0,9,0,10,0,11,0,12,0,13,0,14,0,15,0,16,0}; extern "C" UINT64* val1; extern "C" UINT64* val2; extern "C" UINT64* val3; extern "C" UINT64* val4; extern "C" UINT64* val5; extern "C" UINT64* val6; extern "C" UINT64* val7; extern "C" UINT64* val8; extern "C" UINT64* val9; extern "C" UINT64* val10; extern "C" UINT64* val11; extern "C" UINT64* val12; extern "C" UINT64* val13; extern "C" UINT64* val14; extern "C" UINT64* val15; extern "C" UINT64* val16; extern "C" ADDRINT setFlagsX; UINT64* val1=&vals[0]; UINT64* val2=&vals[2]; UINT64* val3=&vals[4]; UINT64* val4=&vals[6]; UINT64* val5=&vals[8]; UINT64* val6=&vals[10]; UINT64* val7=&vals[12]; UINT64* val8=&vals[14]; UINT64* val9=&vals[16]; UINT64* val10=&vals[18]; UINT64* val11=&vals[20]; UINT64* val12=&vals[22]; UINT64* val13=&vals[24]; UINT64* val14=&vals[26]; UINT64* val15=&vals[28]; UINT64* val16=&vals[30]; ADDRINT setFlagsX=0; EXTERN_C VOID DummyProc() { } EXPORT_CSYM void TestIargPreserveInReplacement() { } EXPORT_CSYM void TestIargPreserveInReplacement1() { } EXPORT_CSYM void TestIargPreserveInReplacement2() { } EXPORT_CSYM void TestIargPreserveInProbed() { } EXPORT_CSYM void TestIargPreserveInProbed1() { } EXPORT_CSYM void TestIargPreserveInProbed2() { } VOID CallToFldzToTop3() { FldzToTop3(); } VOID CallToSetXmmRegsToZero() { SetXmmRegsToZero(); } VOID CallToSetIntRegsToZero() { SetIntRegsToZero(); } VOID CallToUnMaskZeroDivideInMxcsr() { UnMaskZeroDivideInMxcsr(); } #define NUM_TESTS 19 static int testNum=0; VOID Trace (TRACE trace, VOID *v) { for (BBL bbl = TRACE_BblHead(trace); BBL_Valid(bbl); bbl = BBL_Next(bbl)) { for (INS ins = BBL_InsHead(bbl); INS_Valid(ins); ins = INS_Next(ins)) { xed_iclass_enum_t iclass1 = static_cast(INS_Opcode(ins)); if (iclass1 == XED_ICLASS_FLD1 && INS_Valid(INS_Next(ins))) { xed_iclass_enum_t iclass2 = static_cast(INS_Opcode(INS_Next(ins))); if (iclass2 == XED_ICLASS_FLD1 && INS_Valid(INS_Next(INS_Next(ins)))) { xed_iclass_enum_t iclass3 = static_cast(INS_Opcode(INS_Next(INS_Next(ins)))); if (iclass3 == XED_ICLASS_FLD1) { printf ("found fld1 sequence at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); if (testNum == 0) { REGSET regsFP; REGSET_Clear(regsFP); REGSET_Insert (regsFP, REG_X87); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToFldzToTop3), IARG_PRESERVE, ®sFP, IARG_END); printf ("Inserted call1 to FldzToTop3 after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 1) { REGSET regsNone; REGSET_Clear(regsNone); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToFldzToTop3), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call2 to FldzToTop3 after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 2) { INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToFldzToTop3), IARG_END); printf ("Inserted call3 to FldzToTop3 after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 3) { REGSET regsNone; REGSET_Clear(regsNone); REGSET_Insert (regsNone, REG_X87); REGSET_Insert (regsNone, REG_MXCSR); REGSET_Remove(regsNone, REG_X87); REGSET_Remove(regsNone, REG_MXCSR); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToFldzToTop3), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call4 to FldzToTop3 after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 4) { REGSET regsXMM; REGSET_Clear(regsXMM); // need to declare X87 preserved also - because otherwise the fxsave/fxrstor will preserve the // xmm regs REGSET_Insert (regsXMM, REG_X87); REGSET_Insert (regsXMM, REG_XMM0); REGSET_Insert (regsXMM, REG_XMM1); REGSET_Insert (regsXMM, REG_XMM2); REGSET_Insert (regsXMM, REG_XMM3); REGSET_Insert (regsXMM, REG_XMM4); REGSET_Insert (regsXMM, REG_XMM5); REGSET_Insert (regsXMM, REG_XMM6); REGSET_Insert (regsXMM, REG_XMM7); if (ProcessorSupportsAvx()) { REGSET_Insert (regsXMM, REG_YMM0); REGSET_Insert (regsXMM, REG_YMM1); REGSET_Insert (regsXMM, REG_YMM2); REGSET_Insert (regsXMM, REG_YMM3); REGSET_Insert (regsXMM, REG_YMM4); REGSET_Insert (regsXMM, REG_YMM5); REGSET_Insert (regsXMM, REG_YMM6); REGSET_Insert (regsXMM, REG_YMM7); } if (SupportsAvx512f()) { REGSET_Insert (regsXMM, REG_ZMM0); REGSET_Insert (regsXMM, REG_ZMM1); REGSET_Insert (regsXMM, REG_ZMM2); REGSET_Insert (regsXMM, REG_ZMM3); REGSET_Insert (regsXMM, REG_ZMM4); REGSET_Insert (regsXMM, REG_ZMM5); REGSET_Insert (regsXMM, REG_ZMM6); REGSET_Insert (regsXMM, REG_ZMM7); } #ifdef TARGET_IA32E REGSET_Insert (regsXMM, REG_XMM8); REGSET_Insert (regsXMM, REG_XMM9); REGSET_Insert (regsXMM, REG_XMM10); REGSET_Insert (regsXMM, REG_XMM11); REGSET_Insert (regsXMM, REG_XMM12); REGSET_Insert (regsXMM, REG_XMM13); REGSET_Insert (regsXMM, REG_XMM14); REGSET_Insert (regsXMM, REG_XMM15); if (ProcessorSupportsAvx()) { REGSET_Insert (regsXMM, REG_YMM8); REGSET_Insert (regsXMM, REG_YMM9); REGSET_Insert (regsXMM, REG_YMM10); REGSET_Insert (regsXMM, REG_YMM11); REGSET_Insert (regsXMM, REG_YMM12); REGSET_Insert (regsXMM, REG_YMM13); REGSET_Insert (regsXMM, REG_YMM14); REGSET_Insert (regsXMM, REG_YMM15); } if (SupportsAvx512f()) { REGSET_Insert (regsXMM, REG_ZMM8); REGSET_Insert (regsXMM, REG_ZMM9); REGSET_Insert (regsXMM, REG_ZMM10); REGSET_Insert (regsXMM, REG_ZMM11); REGSET_Insert (regsXMM, REG_ZMM12); REGSET_Insert (regsXMM, REG_ZMM13); REGSET_Insert (regsXMM, REG_ZMM14); REGSET_Insert (regsXMM, REG_ZMM15); REGSET_Insert (regsXMM, REG_ZMM16); REGSET_Insert (regsXMM, REG_ZMM17); REGSET_Insert (regsXMM, REG_ZMM18); REGSET_Insert (regsXMM, REG_ZMM19); REGSET_Insert (regsXMM, REG_ZMM20); REGSET_Insert (regsXMM, REG_ZMM21); REGSET_Insert (regsXMM, REG_ZMM22); REGSET_Insert (regsXMM, REG_ZMM23); REGSET_Insert (regsXMM, REG_ZMM24); REGSET_Insert (regsXMM, REG_ZMM25); REGSET_Insert (regsXMM, REG_ZMM26); REGSET_Insert (regsXMM, REG_ZMM27); REGSET_Insert (regsXMM, REG_ZMM28); REGSET_Insert (regsXMM, REG_ZMM29); REGSET_Insert (regsXMM, REG_ZMM30); REGSET_Insert (regsXMM, REG_ZMM31); } #endif INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetXmmRegsToZero), IARG_PRESERVE, ®sXMM, IARG_END); printf ("Inserted call1 to CallToSetXmmRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 5) { REGSET regsNone; REGSET_Clear(regsNone); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetXmmRegsToZero), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call2 to CallToSetXmmRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 6) { INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetXmmRegsToZero), IARG_END); printf ("Inserted call3 to CallToSetXmmRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 7) { REGSET regsXMM; REGSET_Clear(regsXMM); REGSET_Insert (regsXMM, REG_X87); REGSET_Insert (regsXMM, REG_XMM0); REGSET_Insert (regsXMM, REG_XMM1); REGSET_Insert (regsXMM, REG_XMM2); REGSET_Insert (regsXMM, REG_XMM3); REGSET_Insert (regsXMM, REG_XMM4); REGSET_Insert (regsXMM, REG_XMM5); REGSET_Insert (regsXMM, REG_XMM6); REGSET_Insert (regsXMM, REG_XMM7); #ifdef TARGET_IA32E REGSET_Insert (regsXMM, REG_XMM8); REGSET_Insert (regsXMM, REG_XMM9); REGSET_Insert (regsXMM, REG_XMM10); REGSET_Insert (regsXMM, REG_XMM11); REGSET_Insert (regsXMM, REG_XMM12); REGSET_Insert (regsXMM, REG_XMM13); REGSET_Insert (regsXMM, REG_XMM14); REGSET_Insert (regsXMM, REG_XMM15); #endif REGSET_Remove (regsXMM, REG_X87); REGSET_Remove (regsXMM, REG_XMM0); REGSET_Remove (regsXMM, REG_XMM1); REGSET_Remove (regsXMM, REG_XMM2); REGSET_Remove (regsXMM, REG_XMM3); REGSET_Remove (regsXMM, REG_XMM4); REGSET_Remove (regsXMM, REG_XMM5); REGSET_Remove (regsXMM, REG_XMM6); REGSET_Remove (regsXMM, REG_XMM7); #ifdef TARGET_IA32E REGSET_Remove (regsXMM, REG_XMM8); REGSET_Remove (regsXMM, REG_XMM9); REGSET_Remove (regsXMM, REG_XMM10); REGSET_Remove (regsXMM, REG_XMM11); REGSET_Remove (regsXMM, REG_XMM12); REGSET_Remove (regsXMM, REG_XMM13); REGSET_Remove (regsXMM, REG_XMM14); REGSET_Remove (regsXMM, REG_XMM15); #endif INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetXmmRegsToZero), IARG_PRESERVE, ®sXMM, IARG_END); printf ("Inserted call4 to CallToSetXmmRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 8) { REGSET regsX87; REGSET_Clear(regsX87); REGSET_Insert (regsX87, REG_X87); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetXmmRegsToZero), IARG_PRESERVE, ®sX87, IARG_END); printf ("Inserted call5 to CallToSetXmmRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } #if 0 // this test is not run because overwriting the integer registers can cause Pin to fail else if(testNum == 9) { REGSET regsIntScratches; REGSET_Clear(regsIntScratches); REGSET_Insert (regsIntScratches, REG_GAX); REGSET_Insert (regsIntScratches, REG_GCX); REGSET_Insert (regsIntScratches, REG_GDX); #ifdef TARGET_IA32E REGSET_Insert (regsIntScratches, REG_R8); REGSET_Insert (regsIntScratches, REG_R9); REGSET_Insert (regsIntScratches, REG_R10); REGSET_Insert (regsIntScratches, REG_R11); #if defined( __GNUC__) REGSET_Insert (regsIntScratches, REG_GSI); REGSET_Insert (regsIntScratches, REG_GDI); #endif #endif INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetIntRegsToZero), IARG_PRESERVE, ®sIntScratches, IARG_END); printf ("Inserted call1 to CallToSetIntRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } #endif else if(testNum == 9) { REGSET regsNone; REGSET_Clear(regsNone); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetIntRegsToZero), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call2 to CallToSetIntRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 10) { INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetIntRegsToZero), IARG_END); printf ("Inserted call3 to CallToSetIntRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 11) { REGSET regsNone; REGSET_Clear(regsNone); REGSET_Insert (regsNone, REG_GAX); REGSET_Insert (regsNone, REG_GCX); REGSET_Insert (regsNone, REG_GDX); #ifdef TARGET_IA32E REGSET_Insert (regsNone, REG_R8); REGSET_Insert (regsNone, REG_R9); REGSET_Insert (regsNone, REG_R10); REGSET_Insert (regsNone, REG_R11); #if defined( __GNUC__) REGSET_Insert (regsNone, REG_GSI); REGSET_Insert (regsNone, REG_GDI); #endif #endif REGSET_Remove (regsNone, REG_GAX); REGSET_Remove (regsNone, REG_GCX); REGSET_Remove (regsNone, REG_GDX); #ifdef TARGET_IA32E REGSET_Remove (regsNone, REG_R8); REGSET_Remove (regsNone, REG_R9); REGSET_Remove (regsNone, REG_R10); REGSET_Remove (regsNone, REG_R11); #if defined( __GNUC__) REGSET_Remove (regsNone, REG_GSI); REGSET_Remove (regsNone, REG_GDI); #endif #endif INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToSetIntRegsToZero), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call4 to CallToSetIntRegsToZero after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 12) { REGSET regsFP; REGSET_Clear(regsFP); REGSET_Insert (regsFP, REG_X87); REGSET_Insert (regsFP, REG_MXCSR); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToUnMaskZeroDivideInMxcsr), IARG_PRESERVE, ®sFP, IARG_END); printf ("Inserted call1 to UnMaskZeroDivideInMxcsr after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 13) { INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToUnMaskZeroDivideInMxcsr), IARG_END); printf ("Inserted call2 to UnMaskZeroDivideInMxcsr after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 14) { REGSET regsFP; REGSET_Clear(regsFP); REGSET_Insert (regsFP, REG_X87); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToUnMaskZeroDivideInMxcsr), IARG_END); printf ("Inserted call3 to UnMaskZeroDivideInMxcsr after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; } else if(testNum == 15) { REGSET regsNone; REGSET_Clear(regsNone); INS_InsertCall(INS_Next(INS_Next(ins)), IPOINT_AFTER, AFUNPTR(CallToFldzToTop3), IARG_PRESERVE, ®sNone, IARG_END); printf ("Inserted call5 to FldzToTop3 after instruction at %lx\n", (unsigned long)(INS_Address(INS_Next(INS_Next(ins))))); testNum++; printf ("testNum %d\n", testNum); } return; } } } } } } #if defined(__cplusplus) extern "C" #endif VOID TestIargPreserveReplacement() { testNum++; printf ("TestIargPreserveReplacement testNum %d\n", testNum); FldzToTop3(); } #if defined(__cplusplus) extern "C" #endif VOID TestIargPreserveReplacement1() { testNum++; printf ("TestIargPreserveReplacement1 testNum %d\n", testNum); FldzToTop3(); } #if defined(__cplusplus) extern "C" #endif VOID TestIargPreserveReplacement2() { testNum++; printf ("TestIargPreserveReplacement2 testNum %d\n", testNum); FldzToTop3(); SetXmmRegsToZero(); } #if defined(__cplusplus) extern "C" #endif VOID TestIargPreserveProbedBefore() { testNum++; FldzToTop3(); } #if defined(__cplusplus) extern "C" #endif VOID TestIargPreserveProbedBefore1() { testNum++; FldzToTop3(); } VOID ImageLoad(IMG img, VOID *v) { RTN rtn = RTN_FindByName(img, "TestIargPreserveInReplacement"); if (RTN_Valid(rtn)) { printf ("found TestIargPreserveInReplacement\n"); fflush (stdout); PROTO proto = PROTO_Allocate( PIN_PARG(void), CALLINGSTD_DEFAULT, "TestIargPreserveInReplacementProto", PIN_PARG_END() ); REGSET regsFP; REGSET_Clear(regsFP); REGSET_Insert (regsFP, REG_X87); RTN_ReplaceSignature( rtn, AFUNPTR(TestIargPreserveReplacement), IARG_PROTOTYPE, proto, IARG_CONTEXT, IARG_ORIG_FUNCPTR, IARG_PRESERVE, ®sFP, IARG_END); PROTO_Free( proto ); } rtn = RTN_FindByName(img, "TestIargPreserveInReplacement1"); if (RTN_Valid(rtn)) { printf ("found TestIargPreserveInReplacement1\n"); fflush (stdout); PROTO proto = PROTO_Allocate( PIN_PARG(void), CALLINGSTD_DEFAULT, "TestIargPreserveInReplacementProto", PIN_PARG_END() ); RTN_ReplaceSignature( rtn, AFUNPTR(TestIargPreserveReplacement1), IARG_PROTOTYPE, proto, IARG_CONTEXT, IARG_ORIG_FUNCPTR, IARG_END); PROTO_Free( proto ); } rtn = RTN_FindByName(img, "TestIargPreserveInReplacement2"); if (RTN_Valid(rtn)) { printf ("found TestIargPreserveInReplacement2\n"); fflush (stdout); PROTO proto = PROTO_Allocate( PIN_PARG(void), CALLINGSTD_DEFAULT, "TestIargPreserveInReplacementProto", PIN_PARG_END() ); RTN_ReplaceSignature( rtn, AFUNPTR(TestIargPreserveReplacement2), IARG_PROTOTYPE, proto, IARG_CONTEXT, IARG_ORIG_FUNCPTR, IARG_END); PROTO_Free( proto ); } } VOID Fini(INT32 code, VOID *v) { if (testNum != NUM_TESTS) { printf ("***Error not all expected tests ran testNum %d\n", testNum); exit (-1); } printf ("all expected tests ran testNum %d\n", testNum); } int main(int argc, char *argv[]) { PIN_Init(argc, argv); PIN_InitSymbols(); IMG_AddInstrumentFunction(ImageLoad, 0); TRACE_AddInstrumentFunction(Trace, 0); PIN_AddFiniFunction(Fini, 0); // Never returns PIN_StartProgram(); return 0; }