TheRock/compiler/amd-llvm/compiler-rt/test/asan/TestCases/Posix/coverage-fork.cpp

// RUN: %clangxx_asan -fsanitize-coverage=func,trace-pc-guard %s -o %t
// RUN: rm -rf %t-dir
// RUN: mkdir -p %t-dir && cd %t-dir
// RUN: %env_asan_opts=coverage=1:verbosity=1 %run %t 2>&1 | FileCheck %s
//
// UNSUPPORTED: android
// UNSUPPORTED: iossim
//
// Ideally a forked-subprocess should only report it's own coverage,
// not parent's one. But trace-pc-guard currently does nothing special for fork,
// and thus this test is relaxed.

#include <stdio.h>
#include <string.h>
#include <sys/wait.h>
#include <unistd.h>

__attribute__((noinline))
void foo() { printf("foo\n"); }

__attribute__((noinline))
void bar() { printf("bar\n"); }

__attribute__((noinline))
void baz() { printf("baz\n"); }

int main(int argc, char **argv) {
  pid_t child_pid = fork();
  char buf[100];
  if (child_pid == 0) {
    snprintf(buf, sizeof(buf), "Child PID: %ld\n", (long)getpid());
    write(2, buf, strlen(buf));
    baz();
  } else {
    snprintf(buf, sizeof(buf), "Parent PID: %ld\n", (long)getpid());
    write(2, buf, strlen(buf));
    foo();
    bar();

    // Wait for the child process(s) to finish
    while (wait(NULL) > 0)
      ;
  }
  return 0;
}

// CHECK-DAG: Child PID: [[ChildPID:[0-9]+]]
// CHECK-DAG: [[ChildPID]].sancov: {{.*}} PCs written
// CHECK-DAG: Parent PID: [[ParentPID:[0-9]+]]
// CHECK-DAG: [[ParentPID]].sancov: 3 PCs written