llvm/compiler-rt/test/asan/TestCases/Posix/fake_stack_gc.cpp

// RUN: %clangxx_asan -O0 -pthread %s -o %t && %env_asan_opts=use_sigaltstack=0 not --crash %run %t 2>&1 | FileCheck %s

// Check that fake stack does not discard frames on the main stack, when GC is
// triggered from high alt stack.

// This test does not work on iOS simulator
// (https://github.com/llvm/llvm-project/issues/64942).
// UNSUPPORTED: iossim

#include <algorithm>
#include <assert.h>
#include <csignal>
#include <cstdint>
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

const size_t kStackSize = 0x100000;

int *on_thread;
int *p;

template <size_t N> void Fn() {
  int t[N];
  p = t;
  if constexpr (N > 1)
    Fn<N - 1>();
}

static void Handler(int signo) {
  fprintf(stderr, "Handler Frame:%p\n", __builtin_frame_address(0));

  // Trigger GC and create a lot of frame to reuse "Thread" frame if it was
  // discarded.
  for (int i = 0; i < 1000; ++i)
    Fn<100>();
  // If we discarder and reused "Thread" frame, the next line will crash with
  // false report.
  *on_thread = 10;
  fprintf(stderr, "SUCCESS\n");
  // CHECK: SUCCESS
}

void *Thread(void *arg) {
  fprintf(stderr, "Thread Frame:%p\n", __builtin_frame_address(0));
  stack_t stack = {};
  stack.ss_sp = arg;
  stack.ss_flags = 0;
  stack.ss_size = kStackSize;
  assert(sigaltstack(&stack, nullptr) == 0);

  struct sigaction sa = {};
  sa.sa_handler = Handler;
  sa.sa_flags = SA_ONSTACK;
  sigaction(SIGABRT, &sa, nullptr);

  // Store pointer to the local var, so we can access this frame from the signal
  // handler when the frame is still alive.
  int n;
  on_thread = &n;

  // Abort should schedule FakeStack GC and call handler on alt stack.
  abort();
}

int main(void) {
  // Allocate main and alt stack for future thread.
  void *main_stack;
  void *alt_stack;
  size_t const kPageSize = sysconf(_SC_PAGESIZE);
  assert(posix_memalign(&main_stack, kPageSize, kStackSize) == 0);
  assert(posix_memalign(&alt_stack, kPageSize, kStackSize) == 0);

  // Pick the lower stack as the main stack, as we want to trigger GC in
  // FakeStack from alt stack in a such way that main stack is allocated below.
  if ((uintptr_t)main_stack > (uintptr_t)alt_stack)
    std::swap(alt_stack, main_stack);

  fprintf(stderr, "main_stack: %p-%p\n", main_stack,
          (char *)main_stack + kStackSize);
  fprintf(stderr, "alt_stack: %p-%p\n", alt_stack,
          (char *)alt_stack + kStackSize);

  pthread_attr_t attr;
  assert(pthread_attr_init(&attr) == 0);
  assert(pthread_attr_setstack(&attr, main_stack, kStackSize) == 0);

  pthread_t tid;
  assert(pthread_create(&tid, &attr, Thread, alt_stack) == 0);

  pthread_join(tid, nullptr);

  free(main_stack);
  free(alt_stack);

  return 0;
}