Added few more stubs so that control reaches to DestroyDevice().

[mesa.git] / src / mapi / entry_x86_tsd.h

diff --git a/src/mapi/entry_x86_tsd.h b/src/mapi/entry_x86_tsd.h
index c479c199e238a8f4778e71d83a6d814b2ddf8924..f5d9c41253fbda38eb6ee19177f8a7b227fa5d90 100644 (file)
--- a/src/mapi/entry_x86_tsd.h
+++ b/src/mapi/entry_x86_tsd.h
@@ -1,6 +1,5 @@
 /*
  * Mesa 3-D graphics library
- * Version:  7.9
  *
  * Copyright (C) 2010 LunarG Inc.
  *
@@ -26,12 +25,30 @@
  *    Chia-I Wu <olv@lunarg.com>
  */
 
-#include "u_macros.h"
+#ifdef __CET__
+#define ENDBR "endbr32\n\t"
+#else
+#define ENDBR
+#endif
 
-#define X86_ENTRY_SIZE 32
+#ifdef HAVE_FUNC_ATTRIBUTE_VISIBILITY
+#define HIDDEN __attribute__((visibility("hidden")))
+#else
+#define HIDDEN
+#endif
 
-__asm__(".text\n"
-        ".balign 32\n"
+#define X86_ENTRY_SIZE 64
+
+__asm__(".text\n");
+
+__asm__("x86_got:\n\t"
+        "call 1f\n"
+        "1:\n\t"
+        "popl %eax\n\t"
+        "addl $_GLOBAL_OFFSET_TABLE_+[.-1b], %eax\n\t"
+        "ret");
+
+__asm__(".balign 32\n"
         "x86_entry_start:");
 
 #define STUB_ASM_ENTRY(func)        \
@@ -40,14 +57,29 @@ __asm__(".text\n"
    ".balign 32\n"                   \
    func ":"
 
+#define LOC_BEGIN_SET_ECX
+#define LOC_END_SET_ECX
+#define LOC_END_JMP
+
 #define STUB_ASM_CODE(slot)         \
-   "movl " ENTRY_CURRENT_TABLE ", %eax\n\t" \
+   ENDBR                            \
+   LOC_BEGIN_SET_ECX               \
+   "call 1f\n\t"                    \
+   "1:\n\t"                         \
+   "popl %ecx\n\t"                  \
+   "addl $_GLOBAL_OFFSET_TABLE_+[.-1b], %ecx\n\t" \
+   LOC_END_SET_ECX                                \
+   "movl " ENTRY_CURRENT_TABLE "@GOT(%ecx), %eax\n\t" \
+   "mov (%eax), %eax\n\t"           \
    "testl %eax, %eax\n\t"           \
-   "je 1f\n\t"                      \
-   "jmp *(4 * " slot ")(%eax)\n"    \
+   "jne 1f\n\t"                     \
+   "push %ebx\n\t"                  \
+   "movl %ecx, %ebx\n\t"            \
+   "call " ENTRY_CURRENT_TABLE_GET "@PLT\n\t" \
+   "popl %ebx\n\t"                  \
    "1:\n\t"                         \
-   "call " ENTRY_CURRENT_TABLE_GET "\n\t" \
-   "jmp *(4 * " slot ")(%eax)"
+   "jmp *(4 * " slot ")(%eax)\n\t" \
+   LOC_END_JMP
 
 #define MAPI_TMP_STUB_ASM_GCC
 #include "mapi_tmp.h"
@@ -57,11 +89,28 @@ __asm__(".text\n"
 __asm__(".balign 32\n"
         "x86_entry_end:");
 
+#undef LOC_BEGIN_SET_ECX
+#undef LOC_END_SET_ECX
+#undef LOC_END_JMP
+#define LOC_BEGIN_SET_ECX "jmp set_ecx\n\t"
+#define LOC_END_SET_ECX "set_ecx:movl $0x12345678, %ecx\n\tloc_end_set_ecx:\n\t"
+#define LOC_END_JMP "loc_end_jmp:"
+
+/* Any number big enough works. This is to make sure the final
+ * jmp is a long jmp */
+__asm__(STUB_ASM_CODE("10000"));
+
+extern const char loc_end_set_ecx[] HIDDEN;
+extern const char loc_end_jmp[] HIDDEN;
+
 #include <string.h>
 #include "u_execmem.h"
 
-static const char x86_entry_start[];
-static const char x86_entry_end[];
+extern unsigned long
+x86_got();
+
+extern const char x86_entry_start[] HIDDEN;
+extern const char x86_entry_end[] HIDDEN;
 
 void
 entry_patch_public(void)
@@ -78,16 +127,15 @@ void
 entry_patch(mapi_func entry, int slot)
 {
    char *code = (char *) entry;
-
-   *((unsigned long *) (code + 11)) = slot * sizeof(mapi_func);
-   *((unsigned long *) (code + 22)) = slot * sizeof(mapi_func);
+   int offset = loc_end_jmp - x86_entry_end - sizeof(unsigned long);
+   *((unsigned long *) (code + offset)) = slot * sizeof(mapi_func);
 }
 
 mapi_func
 entry_generate(int slot)
 {
-   const char *code_templ = x86_entry_end - X86_ENTRY_SIZE;
-   void *code;
+   const char *code_templ = x86_entry_end;
+   char *code;
    mapi_func entry;
 
    code = u_execmem_alloc(X86_ENTRY_SIZE);
@@ -96,6 +144,9 @@ entry_generate(int slot)
 
    memcpy(code, code_templ, X86_ENTRY_SIZE);
    entry = (mapi_func) code;
+   int ecx_value_off = loc_end_set_ecx - x86_entry_end - sizeof(unsigned long);
+   *((unsigned long *) (code + ecx_value_off)) = x86_got();
+
    entry_patch(entry, slot);
 
    return entry;