[REFACT] Verified read PEB size

hasherezade · Aug 23, 2024 · eeed2da · eeed2da
1 parent bda723b
commit eeed2da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AntiDebug.cpp b/AntiDebug.cpp
@@ -478,15 +478,15 @@ BOOL getPEB(CONTEXT* ctxt, ADDRINT& pebAddr)
     PIN_GetContextRegval(ctxt, REG_SEG_GS_BASE, reinterpret_cast<UINT8*>(&gsValue));
     gsValue += 0x60;
     // Save PEB Address
-    if (PIN_SafeCopy(&pebAddr, reinterpret_cast<VOID*>(gsValue), sizeof(pebAddr)))is_ok = TRUE;
+    if (PIN_SafeCopy(&pebAddr, reinterpret_cast<VOID*>(gsValue), sizeof(pebAddr)) == sizeof(pebAddr)) is_ok = TRUE;
 #else
     // Read the value from the memory address pointed by FS:[30h] and save it in the global variable
     ADDRINT fsValue;
     PIN_GetContextRegval(ctxt, REG_SEG_FS_BASE, reinterpret_cast<UINT8*>(&fsValue));
     fsValue += 0x30;
 
     // Save PEB Address
-    if (PIN_SafeCopy(&pebAddr, reinterpret_cast<VOID*>(fsValue), sizeof(pebAddr))) is_ok = TRUE;
+    if (PIN_SafeCopy(&pebAddr, reinterpret_cast<VOID*>(fsValue), sizeof(pebAddr)) == sizeof(pebAddr)) is_ok = TRUE;
 #endif
     return is_ok;
 }