Home | History | Annotate | Line # | Download | only in sys 
      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 }
    135