contrib/netbsd-tests/lib/libpthread/t_fork.c

   1 /* $NetBSD: t_fork.c,v 1.2 2017/01/16 16:28:27 christos Exp $ */
   2
   3 /*
   4  * Copyright (c) 2008 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  26  * POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include <sys/cdefs.h>
  30 __COPYRIGHT("@(#) Copyright (c) 2008\
  31  The NetBSD Foundation, inc. All rights reserved.");
  32 __RCSID("$NetBSD: t_fork.c,v 1.2 2017/01/16 16:28:27 christos Exp $");
  33
  34 /*
  35  * Written by Love Hörnquist Åstrand <lha@NetBSD.org>, March 2003.
  36  * Public domain.
  37  */
  38
  39 #include <sys/types.h>
  40 #include <sys/wait.h>
  41
  42 #include <errno.h>
  43 #include <pthread.h>
  44 #include <signal.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <string.h>
  48 #include <unistd.h>
  49
  50 #include <atf-c.h>
  51
  52 #include "h_common.h"
  53
  54 static pid_t parent;
  55 static int thread_survived = 0;
  56
  57 static void *
  58 print_pid(void *arg)
  59 {
  60         sleep(3);
  61
  62         thread_survived = 1;
  63         if (parent != getpid()) {
  64                 _exit(1);
  65         }
  66         return NULL;
  67 }
  68
  69 ATF_TC(fork);
  70 ATF_TC_HEAD(fork, tc)
  71 {
  72         atf_tc_set_md_var(tc, "descr",
  73                 "Checks that child process doesn't get threads");
  74 }
  75 ATF_TC_BODY(fork, tc)
  76 {
  77         pthread_t p;
  78         pid_t fork_pid;
  79
  80         parent = getpid();
  81
  82         PTHREAD_REQUIRE(pthread_create(&p, NULL, print_pid, NULL));
  83
  84         fork_pid = fork();
  85         ATF_REQUIRE(fork_pid != -1);
  86
  87         if (fork_pid) {
  88                 int status;
  89
  90                 PTHREAD_REQUIRE(pthread_join(p, NULL));
  91                 ATF_REQUIRE_MSG(thread_survived, "thread did not survive in parent");
  92
  93                 waitpid(fork_pid, &status, 0);
  94                 ATF_REQUIRE_MSG(WIFEXITED(status), "child died wrongly");
  95                 ATF_REQUIRE_EQ_MSG(WEXITSTATUS(status), 0, "thread survived in child");
  96         } else {
  97                 sleep(5);
  98                 _exit(thread_survived ? 1 : 0);
  99         }
 100 }
 101
 102 ATF_TP_ADD_TCS(tp)
 103 {
 104         ATF_TP_ADD_TC(tp, fork);
 105
 106         return atf_no_error();
 107 }