contrib/netbsd-tests/lib/libc/sys/t_getpid.c

   1 /* $NetBSD: t_getpid.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $ */
   2
   3 /*-
   4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Jukka Ruohonen.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 __RCSID("$NetBSD: t_getpid.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $");
  33
  34 #include <sys/wait.h>
  35
  36 #include <stdlib.h>
  37 #include <pthread.h>
  38 #include <unistd.h>
  39
  40 #include <atf-c.h>
  41
  42 static int       maxiter = 10;
  43 static void     *threadfunc(void *);
  44
  45 static void *
  46 threadfunc(void *arg)
  47 {
  48         *(pid_t *)arg = getpid();
  49
  50         return NULL;
  51 }
  52
  53 ATF_TC(getpid_process);
  54 ATF_TC_HEAD(getpid_process, tc)
  55 {
  56         atf_tc_set_md_var(tc, "descr", "Test getpid(2) with processes");
  57 }
  58
  59 ATF_TC_BODY(getpid_process, tc)
  60 {
  61         pid_t ppid, fpid, cpid, tpid, wpid;
  62         int i, sta;
  63
  64         for (i = 0; i < maxiter; i++) {
  65
  66                 tpid = getpid();
  67                 fpid = fork();
  68
  69                 ATF_REQUIRE(fpid >= 0);
  70
  71                 if (fpid == 0) {
  72
  73                         cpid = getpid();
  74                         ppid = getppid();
  75
  76                         if (tpid != ppid)
  77                                 _exit(EXIT_FAILURE);
  78
  79                         if (cpid == ppid)
  80                                 _exit(EXIT_FAILURE);
  81
  82                         if (tpid == fpid)
  83                                 _exit(EXIT_FAILURE);
  84
  85                         _exit(EXIT_SUCCESS);
  86                 }
  87
  88                 wpid = wait(&sta);
  89
  90                 if (wpid != fpid)
  91                         atf_tc_fail("PID mismatch");
  92
  93                 ATF_REQUIRE(WIFEXITED(sta) != 0);
  94
  95                 if (WEXITSTATUS(sta) != EXIT_SUCCESS)
  96                         atf_tc_fail("PID mismatch");
  97         }
  98 }
  99
 100 ATF_TC(getpid_thread);
 101 ATF_TC_HEAD(getpid_thread, tc)
 102 {
 103         atf_tc_set_md_var(tc, "descr", "Test getpid(2) with threads");
 104 }
 105
 106 ATF_TC_BODY(getpid_thread, tc)
 107 {
 108         pid_t pid, tpid;
 109         pthread_t tid;
 110         int i, rv;
 111
 112         for (i = 0; i < maxiter; i++) {
 113
 114                 pid = getpid();
 115
 116                 rv = pthread_create(&tid, NULL, threadfunc, &tpid);
 117                 ATF_REQUIRE(rv == 0);
 118
 119                 rv = pthread_join(tid, NULL);
 120                 ATF_REQUIRE(rv == 0);
 121
 122                 if (pid != tpid)
 123                         atf_tc_fail("Unequal PIDs");
 124         }
 125 }
 126
 127 ATF_TP_ADD_TCS(tp)
 128 {
 129
 130         ATF_TP_ADD_TC(tp, getpid_process);
 131         ATF_TP_ADD_TC(tp, getpid_thread);
 132
 133         return atf_no_error();
 134 }