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