[X86] Add test suite for stack realignment.

Reland patch D145926. Constrain the test to X86 only.

Author: LuoYuanke

SingleSource/UnitTests/CMakeLists.txt

@@ -7,6 +7,7 @@ add_subdirectory(SignlessTypes)
 add_subdirectory(Threads)
 add_subdirectory(Vector)
 add_subdirectory(Vectorizer)
+add_subdirectory(X86)
 
 list(APPEND CFLAGS -Wno-implicit-function-declaration -Wno-implicit-int)
 

SingleSource/UnitTests/X86/CMakeLists.txt

@@ -0,0 +1,5 @@
+if(CMAKE_C_COMPILER_ID STREQUAL "Clang")
+  if(ARCH STREQUAL "x86")
+    llvm_singlesource(PREFIX "x86-")
+  endif()
+endif()

SingleSource/UnitTests/X86/dyn_stack_alloc_realign.cpp

@@ -0,0 +1,59 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+int checkInt(int *I, int Align) {
+  *I = 20;
+  if ((((size_t)I) & (Align - 1)) != 0) {
+    printf("\nUnalign address (%d): %p!\n", Align, I);
+    abort();
+  }
+  return *I;
+}
+
+#ifndef ALIGNMENT
+#define ALIGNMENT 64
+#endif
+
+typedef int aligned __attribute__((aligned(ALIGNMENT)));
+
+void bar(char *p, int size) { __builtin_strncpy(p, "good", size); }
+
+class Base {};
+
+struct A : virtual public Base {
+  A() {}
+};
+
+struct B {};
+
+void foo(int size) {
+  aligned Var;
+  char *Ptr = (char *)__builtin_alloca(size + 1);
+  aligned I;
+
+  // clobber 32-bit base pointer.
+  asm volatile("nop" ::"S"(405) :);
+  // clobber 64-bit base pointer.
+  asm volatile("nop" ::"b"(405) :);
+  bar(Ptr, size);
+  if (__builtin_strncmp(Ptr, "good", size) != 0) {
+    Ptr[size] = '\0';
+    printf("Failed: %s != good\n", Ptr);
+    abort();
+  }
+
+  if (checkInt(&I, __alignof__(I)) != I)
+    abort();
+
+  // access argument and local variable.
+  asm volatile("movl %0, %1" ::"r"(size), "m"(Var) :);
+  throw A();
+}
+
+int main() {
+  try {
+    foo(5);
+  } catch (A &a) {
+  }
+  return 0;
+}

SingleSource/UnitTests/X86/dyn_stack_alloc_realign.reference_output

@@ -0,0 +1 @@
+exit 0

SingleSource/UnitTests/X86/dyn_stack_alloc_realign2.cpp

@@ -0,0 +1,59 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+int checkInt(int *I, int Align) {
+  *I = 20;
+  if ((((size_t)I) & (Align - 1)) != 0) {
+    printf("\nUnalign address (%d): %p!\n", Align, I);
+    abort();
+  }
+  return *I;
+}
+
+#ifndef ALIGNMENT
+#define ALIGNMENT 64
+#endif
+
+typedef int aligned __attribute__((aligned(ALIGNMENT)));
+
+void bar(char *p, int size) { __builtin_strncpy(p, "good", size); }
+
+class Base {};
+
+struct A : virtual public Base {
+  A() {}
+};
+
+struct B {};
+
+__attribute__((noinline))
+void foo(int size) {
+  aligned Var;
+  char *Ptr = (char *)__builtin_alloca(size + 1);
+  aligned I;
+
+  // clobber 32-bit base pointer.
+  asm volatile("nop" ::"S"(405) :);
+  // clobber 64-bit base pointer.
+  asm volatile("nop" ::"b"(405) :);
+  bar(Ptr, size);
+  if (__builtin_strncmp(Ptr, "good", size) != 0) {
+    Ptr[size] = '\0';
+    printf("Failed: %s != good\n", Ptr);
+    abort();
+  }
+
+  if (checkInt(&I, __alignof__(I)) != I)
+    abort();
+
+  // access argument and local variable.
+  asm volatile("movl %0, %1" ::"r"(size), "m"(Var) :);
+}
+
+int main() {
+  try {
+    foo(5);
+  } catch (A &a) {
+  }
+  return 0;
+}

SingleSource/UnitTests/X86/dyn_stack_alloc_realign2.reference_output

@@ -0,0 +1 @@
+exit 0