File: fork_multithreaded.cpp

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (66 lines) | stat: -rw-r--r-- 1,904 bytes parent folder | download | duplicates (24)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// RUN: %clangxx_tsan -O1 %s -o %t && %run %t 66 2>&1 | FileCheck %s -check-prefix=CHECK-DIE
// RUN: %clangxx_tsan -O1 %s -o %t && %env_tsan_opts=die_after_fork=0 %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-NODIE
#include "test.h"
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>

static void *sleeper(void *p) {
  barrier_wait(&barrier);
  return 0;
}

static void *nop(void *p) {
  return 0;
}

int main(int argc, const char **argv) {
  barrier_init(&barrier, 3);
  const int kSleeperThreads = 2;
  barrier_init(&barrier, kSleeperThreads + 1);
  pthread_t th0[kSleeperThreads];
  for (int i = 0; i < kSleeperThreads; i++)
    pthread_create(&th0[i], 0, sleeper, 0);
  const int kNopThreads = 5;
  pthread_t th1[kNopThreads];
  for (int i = 0; i < kNopThreads; i++)
    pthread_create(&th1[i], 0, nop, 0);
  for (int i = 0; i < kNopThreads; i++)
    pthread_join(th1[i], 0);
  int pid = fork();
  if (pid < 0) {
    fprintf(stderr, "failed to fork (%d)\n", errno);
    exit(1);
  }
  if (pid == 0) {
    // child
    const int kChildThreads = 4;
    pthread_t th2[kChildThreads];
    for (int i = 0; i < kChildThreads; i++)
      pthread_create(&th2[i], 0, nop, 0);
    for (int i = 0; i < kChildThreads; i++)
      pthread_join(th2[i], 0);
    exit(0);
    return 0;
  }
  // parent
  int expect = argc > 1 ? atoi(argv[1]) : 0;
  int status = 0;
  while (waitpid(pid, &status, 0) != pid) {
  }
  if (!WIFEXITED(status) || WEXITSTATUS(status) != expect) {
    fprintf(stderr, "subprocess exited with %d, expected %d\n", status, expect);
    exit(1);
  }
  barrier_wait(&barrier);
  for (int i = 0; i < kSleeperThreads; i++)
    pthread_join(th0[i], 0);
  fprintf(stderr, "OK\n");
  return 0;
}

// CHECK-DIE: ThreadSanitizer: starting new threads after multi-threaded fork is not supported

// CHECK-NODIE-NOT: ThreadSanitizer:
// CHECK-NODIE: OK
// CHECK-NODIE-NOT: ThreadSanitizer: