1 /* $NetBSD: t_umask.c,v 1.1 2011/07/07 06:57:54 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_umask.c,v 1.1 2011/07/07 06:57:54 jruoho 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 43 static const char path[] = "umask"; 44 static 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 59 ATF_TC_WITH_CLEANUP(umask_fork); 60 ATF_TC_HEAD(umask_fork, tc) 61 { 62 atf_tc_set_md_var(tc, "descr", "Check that umask(2) is inherited"); 63 } 64 65 ATF_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 99 fail: 100 (void)umask(S_IWGRP | S_IWOTH); 101 102 atf_tc_fail("umask(2) was not inherited"); 103 } 104 105 ATF_TC_CLEANUP(umask_fork, tc) 106 { 107 (void)umask(S_IWGRP | S_IWOTH); 108 } 109 110 ATF_TC_WITH_CLEANUP(umask_open); 111 ATF_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 116 ATF_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 #ifdef __FreeBSD__ 133 (void)close(fd); 134 #endif 135 (void)memset(&st, 0, sizeof(struct stat)); 136 137 if (stat(path, &st) != 0) { 138 str = "failed to stat(2)"; 139 goto out; 140 } 141 142 if ((st.st_mode & mask[i]) != 0) { 143 str = "invalid umask(2)"; 144 goto out; 145 } 146 147 if (unlink(path) != 0) { 148 str = "failed to unlink(2)"; 149 goto out; 150 } 151 152 } 153 154 out: 155 (void)umask(S_IWGRP | S_IWOTH); 156 157 if (str != NULL) 158 atf_tc_fail("%s", str); 159 } 160 161 ATF_TC_CLEANUP(umask_open, tc) 162 { 163 (void)umask(S_IWGRP | S_IWOTH); 164 (void)unlink(path); 165 } 166 167 ATF_TC_WITH_CLEANUP(umask_previous); 168 ATF_TC_HEAD(umask_previous, tc) 169 { 170 atf_tc_set_md_var(tc, "descr", "Test the return value from umask(2)"); 171 } 172 173 ATF_TC_BODY(umask_previous, tc) 174 { 175 mode_t mode; 176 size_t i; 177 178 for (i = 0; i < __arraycount(mask); i++) { 179 180 mode = umask(mask[i]); 181 mode = umask(mask[i]); 182 183 if (mode != mask[i]) 184 goto fail; 185 } 186 187 return; 188 189 fail: 190 (void)umask(S_IWGRP | S_IWOTH); 191 192 atf_tc_fail("umask(2) did not return the previous mask"); 193 } 194 195 ATF_TC_CLEANUP(umask_previous, tc) 196 { 197 (void)umask(S_IWGRP | S_IWOTH); 198 } 199 200 ATF_TP_ADD_TCS(tp) 201 { 202 203 ATF_TP_ADD_TC(tp, umask_fork); 204 ATF_TP_ADD_TC(tp, umask_open); 205 ATF_TP_ADD_TC(tp, umask_previous); 206 207 return atf_no_error(); 208 } 209