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

   1 /* $NetBSD: t_fork.c,v 1.1 2010/07/16 15:42:53 jmmv 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.1 2010/07/16 15:42:53 jmmv 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 #ifdef __FreeBSD__
  65                 _exit(1);
  66 #else
  67                 exit(1);
  68 #endif
  69         }
  70         return NULL;
  71 }
  72
  73 ATF_TC(fork);
  74 ATF_TC_HEAD(fork, tc)
  75 {
  76         atf_tc_set_md_var(tc, "descr",
  77                 "Checks that child process doesn't get threads");
  78 }
  79 ATF_TC_BODY(fork, tc)
  80 {
  81         pthread_t p;
  82         pid_t fork_pid;
  83
  84         parent = getpid();
  85
  86         PTHREAD_REQUIRE(pthread_create(&p, NULL, print_pid, NULL));
  87
  88         fork_pid = fork();
  89         ATF_REQUIRE(fork_pid != -1);
  90
  91         if (fork_pid) {
  92                 int status;
  93
  94                 PTHREAD_REQUIRE(pthread_join(p, NULL));
  95                 ATF_REQUIRE_MSG(thread_survived, "thread did not survive in parent");
  96
  97                 waitpid(fork_pid, &status, 0);
  98                 ATF_REQUIRE_MSG(WIFEXITED(status), "child died wrongly");
  99                 ATF_REQUIRE_EQ_MSG(WEXITSTATUS(status), 0, "thread survived in child");
 100         } else {
 101                 sleep(5);
 102 #ifdef __FreeBSD__
 103                 _exit(thread_survived ? 1 : 0);
 104 #else
 105                 exit(thread_survived ? 1 : 0);
 106 #endif
 107         }
 108 }
 109
 110 ATF_TP_ADD_TCS(tp)
 111 {
 112         ATF_TP_ADD_TC(tp, fork);
 113
 114         return atf_no_error();
 115 }