[compiler-rt] r321151 - [builtins] Implement __chkstk for arm64 windows

[Martin Storsjo via llvm-commits]

Author: mstorsjo
Date: Tue Dec 19 22:52:52 2017
New Revision: 321151

URL: http://llvm.org/viewvc/llvm-project?rev=321151&view=rev
Log:
[builtins] Implement __chkstk for arm64 windows

Differential Revision: https://reviews.llvm.org/D41134

Added:
    compiler-rt/trunk/lib/builtins/aarch64/
    compiler-rt/trunk/lib/builtins/aarch64/chkstk.S
Modified:
    compiler-rt/trunk/lib/builtins/CMakeLists.txt

Modified: compiler-rt/trunk/lib/builtins/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/lib/builtins/CMakeLists.txt?rev=321151&r1=321150&r2=321151&view=diff
==============================================================================
--- compiler-rt/trunk/lib/builtins/CMakeLists.txt (original)
+++ compiler-rt/trunk/lib/builtins/CMakeLists.txt Tue Dec 19 22:52:52 2017
@@ -446,6 +446,12 @@ set(aarch64_SOURCES
   ${GENERIC_TF_SOURCES}
   ${GENERIC_SOURCES})
 
+if (MINGW)
+  set(aarch64_SOURCES
+      ${aarch64_SOURCES}
+      aarch64/chkstk.S)
+endif()
+
 set(armhf_SOURCES ${arm_SOURCES})
 set(armv7_SOURCES ${arm_SOURCES})
 set(armv7s_SOURCES ${arm_SOURCES})

Added: compiler-rt/trunk/lib/builtins/aarch64/chkstk.S
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/lib/builtins/aarch64/chkstk.S?rev=321151&view=auto
==============================================================================
--- compiler-rt/trunk/lib/builtins/aarch64/chkstk.S (added)
+++ compiler-rt/trunk/lib/builtins/aarch64/chkstk.S Tue Dec 19 22:52:52 2017
@@ -0,0 +1,34 @@
+// This file is dual licensed under the MIT and the University of Illinois Open
+// Source Licenses. See LICENSE.TXT for details.
+
+#include "../assembly.h"
+
+// __chkstk routine
+// This routine is windows specific.
+// http://msdn.microsoft.com/en-us/library/ms648426.aspx
+
+// This clobbers registers x16 and x17.
+// Does not modify any memory or the stack pointer.
+
+//      mov     x15, #256 // Number of bytes of stack, in units of 16 byte
+//      bl      __chkstk
+//      sub     sp, sp, x15, lsl #4
+
+#ifdef __aarch64__
+
+#define PAGE_SIZE 4096
+
+        .p2align 2
+DEFINE_COMPILERRT_FUNCTION(__chkstk)
+        lsl    x16, x15, #4
+        mov    x17, sp
+1:
+        sub    x17, x17, #PAGE_SIZE
+        subs   x16, x16, #PAGE_SIZE
+        ldr    xzr, [x17]
+        b.gt   1b
+
+        ret
+END_COMPILERRT_FUNCTION(__chkstk)
+
+#endif // __aarch64__