t_umask.c revision 314817 
1/* $NetBSD: t_umask.c,v 1.2 2017/01/13 19:34:19 christos 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_umask.c,v 1.2 2017/01/13 19:34:19 christos Exp $");
33
34#include <sys/stat.h>
35#include <sys/wait.h>
36
37#include <atf-c.h>
38#include <fcntl.h>
39#include <stdlib.h>
40#include <string.h>
41#include <unistd.h>
42
43static const char path[] = "umask";
44static const mode_t mask[] = {
45	S_IRWXU,
46	S_IRUSR,
47	S_IWUSR,
48	S_IXUSR,
49	S_IRWXG,
50	S_IRGRP,
51	S_IWGRP,
52	S_IXGRP,
53	S_IRWXO,
54	S_IROTH,
55	S_IWOTH,
56	S_IXOTH
57};
58
59ATF_TC_WITH_CLEANUP(umask_fork);
60ATF_TC_HEAD(umask_fork, tc)
61{
62	atf_tc_set_md_var(tc, "descr", "Check that umask(2) is inherited");
63}
64
65ATF_TC_BODY(umask_fork, tc)
66{
67	mode_t mode;
68	pid_t pid;
69	size_t i;
70	int sta;
71
72	for (i = 0; i < __arraycount(mask) - 1; i++) {
73
74		(void)umask(mask[i] | mask[i + 1]);
75
76		pid = fork();
77
78		if (pid < 0)
79			continue;
80
81		if (pid == 0) {
82
83			mode = umask(mask[i]);
84
85			if (mode != (mask[i] | mask[i + 1]))
86				_exit(EXIT_FAILURE);
87
88			_exit(EXIT_SUCCESS);
89		}
90
91		(void)wait(&sta);
92
93		if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
94			goto fail;
95	}
96
97	return;
98
99fail:
100	(void)umask(S_IWGRP | S_IWOTH);
101
102	atf_tc_fail("umask(2) was not inherited");
103}
104
105ATF_TC_CLEANUP(umask_fork, tc)
106{
107	(void)umask(S_IWGRP | S_IWOTH);
108}
109
110ATF_TC_WITH_CLEANUP(umask_open);
111ATF_TC_HEAD(umask_open, tc)
112{
113	atf_tc_set_md_var(tc, "descr", "A basic test of open(2) and umask(2)");
114}
115
116ATF_TC_BODY(umask_open, tc)
117{
118	const char *str = NULL;
119	struct stat st;
120	size_t i;
121	int fd;
122
123	for (i = 0; i < __arraycount(mask); i++) {
124
125		(void)umask(mask[i]);
126
127		fd = open(path, O_RDWR | O_CREAT, 0777);
128
129		if (fd < 0)
130			continue;
131
132		(void)close(fd);
133		(void)memset(&st, 0, sizeof(struct stat));
134
135		if (stat(path, &st) != 0) {
136			str = "failed to stat(2)";
137			goto out;
138		}
139
140		if ((st.st_mode & mask[i]) != 0) {
141			str = "invalid umask(2)";
142			goto out;
143		}
144
145		if (unlink(path) != 0) {
146			str = "failed to unlink(2)";
147			goto out;
148		}
149
150	}
151
152out:
153	(void)umask(S_IWGRP | S_IWOTH);
154
155	if (str != NULL)
156		atf_tc_fail("%s", str);
157}
158
159ATF_TC_CLEANUP(umask_open, tc)
160{
161	(void)umask(S_IWGRP | S_IWOTH);
162	(void)unlink(path);
163}
164
165ATF_TC_WITH_CLEANUP(umask_previous);
166ATF_TC_HEAD(umask_previous, tc)
167{
168	atf_tc_set_md_var(tc, "descr", "Test the return value from umask(2)");
169}
170
171ATF_TC_BODY(umask_previous, tc)
172{
173	mode_t mode;
174	size_t i;
175
176	for (i = 0; i < __arraycount(mask); i++) {
177
178		mode = umask(mask[i]);
179		mode = umask(mask[i]);
180
181		if (mode != mask[i])
182			goto fail;
183	}
184
185	return;
186
187fail:
188	(void)umask(S_IWGRP | S_IWOTH);
189
190	atf_tc_fail("umask(2) did not return the previous mask");
191}
192
193ATF_TC_CLEANUP(umask_previous, tc)
194{
195	(void)umask(S_IWGRP | S_IWOTH);
196}
197
198ATF_TP_ADD_TCS(tp)
199{
200
201	ATF_TP_ADD_TC(tp, umask_fork);
202	ATF_TP_ADD_TC(tp, umask_open);
203	ATF_TP_ADD_TC(tp, umask_previous);
204
205	return atf_no_error();
206}
207