Provide convenient way to disable CFI stuff for old/broken assemblers.

[oota-llvm.git] / lib / Target / X86 / X86JITInfo.cpp

diff --git a/lib/Target/X86/X86JITInfo.cpp b/lib/Target/X86/X86JITInfo.cpp
index 3fb3da788db7e8431aafbb81e365df7e505cc3a3..5f88da5f89b6104aa425f7b421a1ec64ca554d4e 100644 (file)
--- a/lib/Target/X86/X86JITInfo.cpp
+++ b/lib/Target/X86/X86JITInfo.cpp
@@ -47,6 +47,12 @@ static TargetJITInfo::JITCompilerFn JITCompilerFunction;
 #define GETASMPREFIX(X) GETASMPREFIX2(X)
 #define ASMPREFIX GETASMPREFIX(__USER_LABEL_PREFIX__)
 
+#if defined(__APPLE__)
+# define CFI(x)
+#else
+# define CFI(x) x
+#endif
+
 // Provide a wrapper for X86CompilationCallback2 that saves non-traditional
 // callee saved registers, for the fastcc calling convention.
 extern "C" {
@@ -58,17 +64,27 @@ extern "C" {
     ".align 8\n"
     ".globl " ASMPREFIX "X86CompilationCallback\n"
   ASMPREFIX "X86CompilationCallback:\n"
+    CFI(".cfi_startproc\n")
     // Save RBP
     "pushq   %rbp\n"
+    CFI(".cfi_def_cfa_offset 16\n")
+    CFI(".cfi_offset %rbp, -16\n")
     // Save RSP
     "movq    %rsp, %rbp\n"
+    CFI(".cfi_def_cfa_register %rbp\n")
     // Save all int arg registers
     "pushq   %rdi\n"
+    CFI(".cfi_rel_offset %rdi, 0\n")
     "pushq   %rsi\n"
+    CFI(".cfi_rel_offset %rsi, 8\n")
     "pushq   %rdx\n"
+    CFI(".cfi_rel_offset %rdx, 16\n")
     "pushq   %rcx\n"
+    CFI(".cfi_rel_offset %rcx, 24\n")
     "pushq   %r8\n"
+    CFI(".cfi_rel_offset %r8, 32\n")
     "pushq   %r9\n"
+    CFI(".cfi_rel_offset %r9, 40\n")
     // Align stack on 16-byte boundary. ESP might not be properly aligned
     // (8 byte) if this is called from an indirect stub.
     "andq    $-16, %rsp\n"
@@ -97,17 +113,35 @@ extern "C" {
     "movaps  (%rsp), %xmm0\n"
     // Restore RSP
     "movq    %rbp, %rsp\n"
+    CFI(".cfi_def_cfa_register esp\n")
     // Restore all int arg registers
     "subq    $48, %rsp\n"
+    CFI(".cfi_adjust_cfa_offset 48\n")
     "popq    %r9\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %r9\n")
     "popq    %r8\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %r8\n")
     "popq    %rcx\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %rcx\n")
     "popq    %rdx\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %rdx\n")
     "popq    %rsi\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %rsi\n")
     "popq    %rdi\n"
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %rdi\n")
     // Restore RBP
     "popq    %rbp\n"
-    "ret\n");
+    CFI(".cfi_adjust_cfa_offset -8\n")
+    CFI(".cfi_restore %rbp\n")
+    "ret\n"
+    CFI(".cfi_endproc\n")
+  );
 #elif defined(__i386__) || defined(i386) || defined(_M_IX86)
 #ifndef _MSC_VER
   void X86CompilationCallback(void);
@@ -116,11 +150,18 @@ extern "C" {
     ".align 8\n"
     ".globl " ASMPREFIX  "X86CompilationCallback\n"
   ASMPREFIX "X86CompilationCallback:\n"
+    CFI(".cfi_startproc\n")
     "pushl   %ebp\n"
+    CFI(".cfi_def_cfa_offset 8\n")
+    CFI(".cfi_offset %ebp, -8\n")
     "movl    %esp, %ebp\n"    // Standard prologue
+    CFI(".cfi_def_cfa_register %ebp\n")
     "pushl   %eax\n"
+    CFI(".cfi_rel_offset %eax, 0\n")
     "pushl   %edx\n"          // Save EAX/EDX/ECX
+    CFI(".cfi_rel_offset %edx, 4\n")
     "pushl   %ecx\n"
+    CFI(".cfi_rel_offset %ecx, 8\n")
 #if defined(__APPLE__)
     "andl    $-16, %esp\n"    // Align ESP on 16-byte boundary
 #endif
@@ -130,12 +171,24 @@ extern "C" {
     "movl    %ebp, (%esp)\n"
     "call    " ASMPREFIX "X86CompilationCallback2\n"
     "movl    %ebp, %esp\n"    // Restore ESP
+    CFI(".cfi_def_cfa_register %esp\n")
     "subl    $12, %esp\n"
+    CFI(".cfi_adjust_cfa_offset 12\n")
     "popl    %ecx\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %ecx\n")
     "popl    %edx\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %edx\n")
     "popl    %eax\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %eax\n")
     "popl    %ebp\n"
-    "ret\n");
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %ebp\n")
+    "ret\n"
+    CFI(".cfi_endproc\n")
+  );
 
   // Same as X86CompilationCallback but also saves XMM argument registers.
   void X86CompilationCallback_SSE(void);
@@ -144,14 +197,24 @@ extern "C" {
     ".align 8\n"
     ".globl " ASMPREFIX  "X86CompilationCallback_SSE\n"
   ASMPREFIX "X86CompilationCallback_SSE:\n"
+    CFI(".cfi_startproc\n")
     "pushl   %ebp\n"
+    CFI(".cfi_def_cfa_offset 8\n")
+    CFI(".cfi_offset %ebp, -8\n")
     "movl    %esp, %ebp\n"    // Standard prologue
+    CFI(".cfi_def_cfa_register %ebp\n")
     "pushl   %eax\n"
+    CFI(".cfi_rel_offset %eax, 0\n")
     "pushl   %edx\n"          // Save EAX/EDX/ECX
+    CFI(".cfi_rel_offset %edx, 4\n")
     "pushl   %ecx\n"
+    CFI(".cfi_rel_offset %ecx, 8\n")
     "andl    $-16, %esp\n"    // Align ESP on 16-byte boundary
     // Save all XMM arg registers
     "subl    $64, %esp\n"
+    // FIXME: provide frame move information for xmm registers.
+    // This can be tricky, because CFA register is ebp (unaligned)
+    // and we need to produce offsets relative to it.
     "movaps  %xmm0, (%esp)\n"
     "movaps  %xmm1, 16(%esp)\n"
     "movaps  %xmm2, 32(%esp)\n"
@@ -163,16 +226,32 @@ extern "C" {
     "call    " ASMPREFIX "X86CompilationCallback2\n"
     "addl    $16, %esp\n"
     "movaps  48(%esp), %xmm3\n"
+    CFI(".cfi_restore %xmm3\n")
     "movaps  32(%esp), %xmm2\n"
+    CFI(".cfi_restore %xmm2\n")
     "movaps  16(%esp), %xmm1\n"
+    CFI(".cfi_restore %xmm1\n")
     "movaps  (%esp), %xmm0\n"
+    CFI(".cfi_restore %xmm0\n")
     "movl    %ebp, %esp\n"    // Restore ESP
+    CFI(".cfi_def_cfa_register esp\n")
     "subl    $12, %esp\n"
+    CFI(".cfi_adjust_cfa_offset 12\n")
     "popl    %ecx\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %ecx\n")
     "popl    %edx\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %edx\n")
     "popl    %eax\n"
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %eax\n")
     "popl    %ebp\n"
-    "ret\n");
+    CFI(".cfi_adjust_cfa_offset -4\n")
+    CFI(".cfi_restore %ebp\n")
+    "ret\n"
+    CFI(".cfi_endproc\n")
+  );
 #else
   void X86CompilationCallback2(void);
 
@@ -319,8 +398,8 @@ void *X86JITInfo::emitFunctionStub(void *Fn, MachineCodeEmitter &MCE) {
     MCE.startFunctionStub(5, 4);
     MCE.emitByte(0xE9);
     MCE.emitWordLE((intptr_t)Fn-MCE.getCurrentPCValue()-4);
-    return MCE.finishFunctionStub(0);
 #endif
+    return MCE.finishFunctionStub(0);
   }
 
 #ifdef __x86_64__