v811_spc009/external/llvm-project/compiler-rt/test/tsan/mmap_stress.cpp

// RUN: %clang_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
#include "test.h"
#include <errno.h>
#include <sys/mman.h>

void *SubWorker(void *arg) {
  (void)arg;
  const int kMmapSize =  65536;
  for (int i = 0; i < 500; i++) {
    int *ptr = (int*)mmap(0, kMmapSize, PROT_READ | PROT_WRITE,
                          MAP_PRIVATE | MAP_ANON, -1, 0);
    if (ptr == MAP_FAILED)
      exit(printf("mmap failed: %d\n", errno));
    *ptr = 42;
    if (munmap(ptr, kMmapSize))
      exit(printf("munmap failed: %d\n", errno));
  }
  return 0;
}

void *Worker1(void *arg) {
  (void)arg;
  pthread_t th[4];
  for (int i = 0; i < 4; i++) {
    if (pthread_create(&th[i], 0, SubWorker, 0))
      exit(printf("pthread_create failed: %d\n", errno));
  }
  for (int i = 0; i < 4; i++) {
    if (pthread_join(th[i], 0))
      exit(printf("pthread_join failed: %d\n", errno));
  }
  return 0;
}

void *Worker(void *arg) {
  (void)arg;
  pthread_t th[4];
  for (int i = 0; i < 4; i++) {
    if (pthread_create(&th[i], 0, Worker1, 0))
      exit(printf("pthread_create failed: %d\n", errno));
  }
  for (int i = 0; i < 4; i++) {
    if (pthread_join(th[i], 0))
      exit(printf("pthread_join failed: %d\n", errno));
  }
  return 0;
}

int main() {
  // This test is flaky on several builders:
  // https://groups.google.com/d/msg/llvm-dev/KUFPdLhBN3Q/L75rwW9xBgAJ
  // The cause is unknown (lit hides test output on failures).
#if 0
  pthread_t th[4];
  for (int i = 0; i < 4; i++) {
    if (pthread_create(&th[i], 0, Worker, 0))
      exit(printf("pthread_create failed: %d\n", errno));
  }
  for (int i = 0; i < 4; i++) {
    if (pthread_join(th[i], 0))
      exit(printf("pthread_join failed: %d\n", errno));
  }
#endif
  fprintf(stderr, "DONE\n");
}

// CHECK: DONE