Changeset 4838 for zzufTweet

Timestamp:

Aug 23, 2012, 1:21:31 PM (9 years ago)

Author:

Sam Hocevar

Message:

win32: add more explicit error messages and add support for 0xb8 opcode.

File:

• zzuf/trunk/src/libzzuf/sys.c (modified) (7 diffs)

zzuf/trunk/src/libzzuf/sys.c

@@ -49 +49 @@
 #if defined HAVE_WINDOWS_H
 static void insert_funcs(void);
-
-/* TODO: get rid of this later */
-HINSTANCE (WINAPI *LoadLibraryA_orig)(LPCSTR);
-HINSTANCE WINAPI LoadLibraryA_new(LPCSTR path)
-{
-    return LoadLibraryA_orig(path);
-}
-
-BOOL (WINAPI *AllocConsole_orig)(void);
-BOOL WINAPI AllocConsole_new(void)
-{
-    return AllocConsole_orig();
-}
-
-BOOL (WINAPI *AttachConsole_orig)(DWORD);
-BOOL WINAPI AttachConsole_new(DWORD d)
-{
-    return AttachConsole_orig(d);
-}
 #endif
 
@@ -130 +111 @@
     switch (opcd)
     {
-    case 0x68: return (insn_size + 4); /* PUSH Iv */
-    case 0x6a: return (insn_size + 1); /* PUSH Ib */
-    case 0x90: return insn_size;       /* NOP     */
-    default: break;
+    case 0x68:
+        return (insn_size + 4); /* PUSH Iv */
+    case 0x6a:
+        return (insn_size + 1); /* PUSH Ib */
+    case 0x90:
+        return insn_size;       /* NOP     */
+    case 0xb8:
+    case 0xb9:
+    case 0xba:
+    case 0xbb:
+    case 0xbc:
+    case 0xbd:
+    case 0xbe:
+    case 0xbf:
+        return insn_size + 5;   /* MOV immediate */
+    default:
+        break;
     }
 
     /* PUSH/POP rv */
-    if ((opcd & 0xf0) == 0x50) return insn_size;
+    if ((opcd & 0xf0) == 0x50)
+        return insn_size;
 
     /* MNEM E?, G? or G?, E? */
@@ -159 +154 @@
         break;
 
-    default: break;
+    default:
+        fprintf(stderr, "unknown opcode %02x\n", opcd);
+        break;
     }
 
@@ -172 +169 @@
     {
         int insn_size = zz_lde(code);
-        if (insn_size == 0) return -1;
+        if (insn_size == 0)
+            return -1;
         patch_size += insn_size;
     }
@@ -206 +204 @@
 
         trampoline = malloc(patch_size + reloc_size + 13); /* Worst case */
-        if (trampoline == NULL) return -1;
+        if (trampoline == NULL)
+            return -1;
         memset(trampoline, 0xcc, patch_size + 13);
 
@@ -212 +211 @@
         {
             int insn_size = zz_lde(code + code_offset);
-            if (insn_size == 0) return -1;
+            if (insn_size == 0)
+                return -1;
 
             /* mov rax, [rip + ...] is the signature for stack cookie */
@@ -335 +335 @@
     /* if we can't get enough byte, we quit */
     if ((patch_size = compute_patch_size(old_api, required_size)) == -1)
+    {
+        fprintf(stderr, "cannot compute patch size\n");
         return -1;
-
-    if (make_trampoline(old_api, patch_size, &trampoline, &trampoline_size) < 0) goto _out;
+    }
+
+    if (make_trampoline(old_api, patch_size, &trampoline, &trampoline_size) < 0)
+    {
+        fprintf(stderr, "cannot make trampoline\n");
+        goto _out;
+    }
 
     /* We must make the trampoline executable, this line is required because of DEP */
     /* NOTE: We _must_ set the write protection, otherwise the heap allocator will crash ! */
-    if (!VirtualProtect(trampoline, trampoline_size, PAGE_EXECUTE_READWRITE, &old_prot)) goto _out;
+    if (!VirtualProtect(trampoline, trampoline_size, PAGE_EXECUTE_READWRITE, &old_prot))
+    {
+        fprintf(stderr, "cannot make the trampoline writable\n");
+        goto _out;
+    }
 
     /* We patch the targeted API, so we must set it as writable */
-    if (!VirtualProtect(old_api, patch_size, PAGE_EXECUTE_READWRITE, &old_prot)) goto _out;
+    if (!VirtualProtect(old_api, patch_size, PAGE_EXECUTE_READWRITE, &old_prot))
+    {
+        fprintf(stderr, "cannot make old API writable\n");
+        goto _out;
+    }
     memcpy(old_api, jmp_prolog, patch_size);
     VirtualProtect(old_api, patch_size, old_prot, &old_prot); /* we don't care if this functon fails */