1 /* $NetBSD: t_getrusage.c,v 1.9 2024/04/12 07:16:50 gson 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_getrusage.c,v 1.9 2024/04/12 07:16:50 gson Exp $");
33
34 #include <sys/resource.h>
35 #include <sys/time.h>
36
37 #include <atf-c.h>
38 #include <stdio.h>
39 #include <errno.h>
40 #include <limits.h>
41 #include <signal.h>
42 #include <stdint.h>
43 #include <stdlib.h>
44 #include <string.h>
45 #include <fcntl.h>
46 #include <sys/socket.h>
47 #include <netinet/in.h>
48
49 static void work(void);
50 static void sighandler(int);
51
52 static const size_t maxiter = 2000;
53
54 static void
sighandler(int signo __unused)55 sighandler(int signo __unused)
56 {
57 /* Nothing. */
58 }
59
60 static void
work(void)61 work(void)
62 {
63 size_t n = UINT16_MAX * 10;
64
65 while (n > 0) {
66 #ifdef __or1k__
67 asm volatile("l.nop"); /* Do something. */
68 #elif defined(__ia64__)
69 asm volatile("nop 0"); /* Do something. */
70 #else
71 asm volatile("nop"); /* Do something. */
72 #endif
73 n--;
74 }
75 }
76
77 ATF_TC(getrusage_err);
ATF_TC_HEAD(getrusage_err,tc)78 ATF_TC_HEAD(getrusage_err, tc)
79 {
80 atf_tc_set_md_var(tc, "descr", "Test error conditions");
81 }
82
ATF_TC_BODY(getrusage_err,tc)83 ATF_TC_BODY(getrusage_err, tc)
84 {
85 struct rusage ru;
86
87 errno = 0;
88
89 ATF_REQUIRE(getrusage(INT_MAX, &ru) != 0);
90 ATF_REQUIRE(errno == EINVAL);
91
92 errno = 0;
93
94 ATF_REQUIRE(getrusage(RUSAGE_SELF, (void *)0) != 0);
95 ATF_REQUIRE(errno == EFAULT);
96 }
97
98 ATF_TC(getrusage_sig);
ATF_TC_HEAD(getrusage_sig,tc)99 ATF_TC_HEAD(getrusage_sig, tc)
100 {
101 atf_tc_set_md_var(tc, "descr", "Test signal count with getrusage(2)");
102 }
103
ATF_TC_BODY(getrusage_sig,tc)104 ATF_TC_BODY(getrusage_sig, tc)
105 {
106 struct rusage ru;
107 const long n = 5;
108 int i;
109
110 /*
111 * Test that signals are recorded.
112 */
113 ATF_REQUIRE(signal(SIGUSR1, sighandler) != SIG_ERR);
114
115 for (i = 0; i < n; i++)
116 ATF_REQUIRE(raise(SIGUSR1) == 0);
117
118 (void)memset(&ru, 0, sizeof(struct rusage));
119 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
120
121 if (n != ru.ru_nsignals)
122 atf_tc_fail("getrusage(2) did not record signals");
123 }
124
125 ATF_TC(getrusage_maxrss);
ATF_TC_HEAD(getrusage_maxrss,tc)126 ATF_TC_HEAD(getrusage_maxrss, tc)
127 {
128 atf_tc_set_md_var(tc, "descr", "Test maxrss growing with getrusage(2)");
129 atf_tc_set_md_var(tc, "require.memory", "64M");
130 }
131
ATF_TC_BODY(getrusage_maxrss,tc)132 ATF_TC_BODY(getrusage_maxrss, tc)
133 {
134 struct rusage ru;
135 long maxrss;
136 int i, fd;
137
138 #define DUMP_FILE "dump"
139
140 fd = open(DUMP_FILE, O_WRONLY|O_CREAT|O_TRUNC, 0222);
141 ATF_REQUIRE(fd != -1);
142
143 (void)memset(&ru, 0, sizeof(struct rusage));
144 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
145 maxrss = ru.ru_maxrss;
146
147 #define CHUNK (1024 * 1024)
148 for (i = 0; i < 40; i++) {
149 void *p = malloc(CHUNK);
150 memset(p, 0, CHUNK);
151 write(fd, p, CHUNK);
152 }
153 close(fd);
154 unlink(DUMP_FILE);
155
156 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
157 ATF_REQUIRE_MSG(maxrss < ru.ru_maxrss,
158 "maxrss: %ld, ru.ru_maxrss: %ld", maxrss, ru.ru_maxrss);
159 }
160
161 ATF_TC(getrusage_msgsnd);
ATF_TC_HEAD(getrusage_msgsnd,tc)162 ATF_TC_HEAD(getrusage_msgsnd, tc)
163 {
164 atf_tc_set_md_var(tc, "descr", "Test send growing with getrusage(2)");
165 }
166
ATF_TC_BODY(getrusage_msgsnd,tc)167 ATF_TC_BODY(getrusage_msgsnd, tc)
168 {
169 struct rusage ru;
170 long msgsnd;
171 int s, i;
172 struct sockaddr_in sin;
173
174 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
175 msgsnd = ru.ru_msgsnd;
176
177 s = socket(AF_INET, SOCK_DGRAM, 0);
178 ATF_REQUIRE(s >= 0);
179 memset(&sin, 0, sizeof(sin));
180 sin.sin_family = AF_INET;
181 sin.sin_len = sizeof(sin);
182 sin.sin_addr.s_addr = ntohl(INADDR_LOOPBACK);
183 sin.sin_port = htons(3333);
184
185 for (i = 0; i < 10; i++)
186 ATF_REQUIRE(sendto(s, &sin, sizeof(sin), 0, (void *)&sin,
187 (socklen_t)sizeof(sin)) != -1);
188
189 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
190 ATF_REQUIRE(msgsnd + 10 == ru.ru_msgsnd);
191 close(s);
192 }
193
194 ATF_TC(getrusage_utime_back);
ATF_TC_HEAD(getrusage_utime_back,tc)195 ATF_TC_HEAD(getrusage_utime_back, tc)
196 {
197 atf_tc_set_md_var(tc, "descr", "Test bogus values from getrusage(2)");
198 }
199
ATF_TC_BODY(getrusage_utime_back,tc)200 ATF_TC_BODY(getrusage_utime_back, tc)
201 {
202 struct rusage ru1, ru2;
203 size_t i;
204
205 /*
206 * Test that two consecutive calls are sane.
207 */
208 atf_tc_expect_fail("PR kern/30115");
209
210 for (i = 0; i < maxiter; i++) {
211
212 (void)memset(&ru1, 0, sizeof(struct rusage));
213 (void)memset(&ru2, 0, sizeof(struct rusage));
214
215 work();
216
217 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru1) == 0);
218
219 work();
220
221 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru2) == 0);
222
223 if (timercmp(&ru2.ru_utime, &ru1.ru_utime, <) != 0)
224 atf_tc_fail("user time went backwards");
225 }
226
227 atf_tc_fail("anticipated error did not occur");
228 }
229
230 ATF_TC(getrusage_utime_zero);
ATF_TC_HEAD(getrusage_utime_zero,tc)231 ATF_TC_HEAD(getrusage_utime_zero, tc)
232 {
233 atf_tc_set_md_var(tc, "descr", "Test zero utime from getrusage(2)");
234 }
235
ATF_TC_BODY(getrusage_utime_zero,tc)236 ATF_TC_BODY(getrusage_utime_zero, tc)
237 {
238 struct rusage ru;
239 size_t i;
240
241 /*
242 * Test that getrusage(2) does not return
243 * zero user time for the calling process.
244 *
245 * See also (duplicate) PR port-amd64/41734.
246 */
247 atf_tc_expect_fail("PR kern/30115");
248
249 for (i = 0; i < maxiter; i++) {
250
251 work();
252
253 (void)memset(&ru, 0, sizeof(struct rusage));
254
255 ATF_REQUIRE(getrusage(RUSAGE_SELF, &ru) == 0);
256
257 if (ru.ru_utime.tv_sec == 0 && ru.ru_utime.tv_usec == 0)
258 atf_tc_fail("zero user time from getrusage(2)");
259 }
260
261 atf_tc_fail("anticipated error did not occur");
262 }
263
ATF_TP_ADD_TCS(tp)264 ATF_TP_ADD_TCS(tp)
265 {
266
267 ATF_TP_ADD_TC(tp, getrusage_err);
268 ATF_TP_ADD_TC(tp, getrusage_sig);
269 ATF_TP_ADD_TC(tp, getrusage_maxrss);
270 ATF_TP_ADD_TC(tp, getrusage_msgsnd);
271 ATF_TP_ADD_TC(tp, getrusage_utime_back);
272 ATF_TP_ADD_TC(tp, getrusage_utime_zero);
273
274 return atf_no_error();
275 }
276