auto merge of #15999 : Kimundi/rust/fix_folder, r=nikomatsakis

[rust.git] / src / rt / arch / x86_64 / morestack.S

diff --git a/src/rt/arch/x86_64/morestack.S b/src/rt/arch/x86_64/morestack.S
index 1113194affc1444053b181d558a2c060b44afb7e..c82da57c12847af48603e9bbc1660d0a57a4e36f 100644 (file)
--- a/src/rt/arch/x86_64/morestack.S
+++ b/src/rt/arch/x86_64/morestack.S
@@ -1,42 +1,27 @@
-    .text
+// Mark stack as non-executable
+#if defined(__linux__) && defined(__ELF__)
+.section       .note.GNU-stack, "", @progbits
+#endif
+
+/* See i386/morestack.S for the lengthy, general explanation. */
 
-// __morestack
-//
-// LLVM generates a call to this to allocate more stack space in a functiono
-// prolog when we run out.
+.text
 
-#if defined(__APPLE__) || defined(_WIN32)
-#define RUST_NEW_STACK2     _rust_new_stack2
-#define RUST_DEL_STACK      _rust_del_stack
-#define UPCALL_CALL_C       _upcall_call_shim_on_c_stack
-#define MORESTACK           ___morestack
+#if defined(__APPLE__)
+#define MORESTACK               ___morestack
 #else
-#define RUST_NEW_STACK2     rust_new_stack2
-#define RUST_DEL_STACK      rust_del_stack
-#define UPCALL_CALL_C       upcall_call_shim_on_c_stack
-#define MORESTACK           __morestack
+#define MORESTACK               __morestack
 #endif
 
-        // Naturally, nobody can agree as to
-        // which arguments should go in which
-        // registers:
-#if defined(_WIN32)
-#  define ARG0 %rcx
-#  define ARG1 %rdx
-#  define ARG2 %r8
+#if defined(__APPLE__)
+#define EXHAUSTED               _rust_stack_exhausted
+#elif defined(__linux__) || defined(__FreeBSD__) || defined(__DragonFly__)
+#define EXHAUSTED               rust_stack_exhausted@PLT
 #else
-#  define ARG0 %rdi
-#  define ARG1 %rsi
-#  define ARG2 %rdx
+#define EXHAUSTED               rust_stack_exhausted
 #endif
 
-.globl RUST_NEW_STACK2
-.globl RUST_DEL_STACK
-.globl UPCALL_CALL_C
-.globl MORESTACK
-
-// FIXME: What about _WIN32?   
-#if defined(__linux__)
+#if defined(__linux__) || defined(__FreeBSD__) || defined(__DragonFly__)
        .hidden MORESTACK
 #else
 #if defined(__APPLE__)
@@ -48,42 +33,37 @@
        .type MORESTACK,@function
 #endif
 
+.globl MORESTACK
 MORESTACK:
        .cfi_startproc
 
-       # Set up a normal backtrace
        pushq %rbp
-       .cfi_def_cfa_offset 16
-       .cfi_offset %rbp, -16
+       // The CFA is 24 bytes above the register that it will
+       // be associated with for this frame (%rbp). That is 8
+       // bytes greater than a normal frame, to allow the unwinder
+       // to skip the partial frame of the original function.
+       .cfi_def_cfa_offset 24
+
+#if defined(__APPLE__)
+       // The pattern of the return address being saved twice to the same location
+       // tells the OS X linker that it should not attempt to convert the DWARF
+       // unwind information to the compact format.
+       .cfi_offset %rip, -8
+       .cfi_offset %rip, -8
+#endif
+
+       // %rbp is -24 bytes from the CFA
+       .cfi_offset %rbp, -24
        movq %rsp, %rbp
+       // Calculate the CFA as on offset from %ebp
        .cfi_def_cfa_register %rbp
 
-       // Save argument registers
-       pushq   %rdi
-       pushq   %rsi
-       pushq   %rdx
-       pushq   %rcx
-       pushq   %r8
-       pushq   %r9
-
-       // Calculate the address of the stack arguments
-       movq %rbp, %rcx
-       addq $16, %rcx   // Add the saved %rbp, and return address
-       addq %r11, %rcx // Add the size of stack arguments
+        // re-align the stack
+        subq $8, %rsp
 
-       pushq %r10 // The amount of stack needed
-       pushq %rcx // Address of stack arguments
-       pushq %r11 // Size of stack arguments
-       pushq %rbp // Save the Rust stack pointer
+        // kill this program
+        call EXHAUSTED
 
-       // FIXME: Don't understand why I have to use the PLT here
-       lea RUST_NEW_STACK2@PLT(%rip), %rsi
-       lea 24(%rsp), %rdi
-       call UPCALL_CALL_C@PLT
-       
-        mov (%rsp),%rdx        // Grab the return pointer.
-        inc %rdx               // Skip past the `ret`.
-        mov %rax,%rsp          // Switch to the new stack.
-        call *%rdx             // Enter the new function.
+        // the exhaustion function guarantees that it can't return
 
-       .cfi_endproc
\ No newline at end of file
+       .cfi_endproc