Demonstrate continued correct behavior on invalid memory access

regress/ro_mem_test.c

@@ -29,18 +29,59 @@ static void hook_code(uch handle, uint64_t address, uint32_t size, void *user_da
 
 }
 
+// callback for tracing memory access (READ or WRITE)
+static bool hook_mem_invalid(uch handle, uc_mem_type type,
+        uint64_t address, int size, int64_t value, void *user_data)
+{
+    uint32_t esp;
+    uc_reg_read(handle, UC_X86_REG_ESP, &esp);
+
+    switch(type) {
+        default:
+            // return false to indicate we want to stop emulation
+            return false;
+        case UC_MEM_WRITE:
+            //if this is a push, esp has not been adjusted yet
+            if (esp == (address + size)) {
+                 uint32_t upper;
+                 upper = (esp + 0xfff) & ~0xfff;
+                 printf(">>> Stack appears to be missing at 0x%"PRIx64 ", allocating now\n", address);
+                 // map this memory in with 2MB in size
+                 uc_mem_map(handle, upper - 0x8000, 0x8000);
+                 // return true to indicate we want to continue
+                 return true;
+            }
+            printf(">>> Missing memory is being WRITE at 0x%"PRIx64 ", data size = %u, data value = 0x%"PRIx64 "\n",
+                   address, size, value);
+            return false;
+    }
+}
+
+
 #define STACK 0x500000
 #define STACK_SIZE 0x5000
 
 int main(int argc, char **argv, char **envp) {
-    uch handle, trace2;
+    uch handle, trace1, trace2;
     uc_err err;
     uint8_t bytes[8];
     uint32_t esp;
     int result;
+    int map_stack = 0;
+    
+    if (argc == 2 && strcmp(argv[1], "--map-stack") == 0) {
+        map_stack = 1;
+    }
 
     printf("Memory mapping test\n");
 
+    if (map_stack) {
+        printf("Pre-mapping stack\n");
+    }
+    else {
+        printf("Mapping stack on first invalid memory access\n");
+    }
+
     // Initialize emulator in X86-32bit mode
     err = uc_open(UC_ARCH_X86, UC_MODE_32, &handle);
     if (err) {
@@ -52,7 +93,10 @@ int main(int argc, char **argv, char **envp) {
     uc_mem_map(handle, 0x200000, 0x2000);
     uc_mem_map(handle, 0x300000, 0x3000);
     uc_mem_map_ex(handle, 0x400000, 0x4000, UC_PROT_READ | UC_PROT_EXEC);
+    
+    if (map_stack) {
        uc_mem_map_ex(handle, STACK, STACK_SIZE, UC_PROT_READ | UC_PROT_WRITE);
+    }
 
     esp = STACK + STACK_SIZE;
 
@@ -69,6 +113,9 @@ int main(int argc, char **argv, char **envp) {
 
     //uc_hook_add(handle, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)0x400000, (uint64_t)0x400fff);
 
+    // intercept invalid memory events
+    uc_hook_add(handle, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL);
+
     // emulate machine code in infinite time
     printf("BEGIN execution\n");
     err = uc_emu_start(handle, 0x400000, 0x400000 + sizeof(PROGRAM), 0, 5);
@@ -79,11 +126,12 @@ int main(int argc, char **argv, char **envp) {
     }
     printf("END execution\n");
 
-    if (!uc_mem_read(handle, 0x400000 + sizeof(PROGRAM) - 1, bytes, 4)) {
+    printf("Verifying content at bottom of stack is readable and correct\n");
-        printf(">>> Read 4 bytes from [0x%x] = 0x%x\n", (uint32_t)(0x400000 + sizeof(PROGRAM) - 1),*(uint32_t*) bytes);
+    if (!uc_mem_read(handle, esp - 4, bytes, 4)) {
+        printf(">>> Read 4 bytes from [0x%x] = 0x%x\n", (uint32_t)(esp - 4), *(uint32_t*) bytes);
     }
     else {
-        printf(">>> Failed to read 4 bytes from [0x%x]\n", (uint32_t)(0x400000 + sizeof(PROGRAM) - 1));
+        printf(">>> Failed to read 4 bytes from [0x%x]\n", (uint32_t)(esp - 4));
         return 4;
     }