1 /* $NetBSD: t_mktemp.c,v 1.4 2021/01/11 20:31:34 christos Exp $ */
2 
3 /*-
4  * Copyright (c) 2013, 2020 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Christos Zoulas and 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_mktemp.c,v 1.4 2021/01/11 20:31:34 christos Exp $");
33 
34 #include <sys/stat.h>
35 
36 #include <atf-c.h>
37 #include <fcntl.h>
38 #include <limits.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <unistd.h>
42 
43 static int
check_mode(struct stat sa,const int mode,const int dir)44 check_mode(struct stat sa, const int mode, const int dir)
45 {
46 
47           if (dir == 0 && S_ISREG(sa.st_mode) == 0)
48                     return -1;
49 
50           if (dir != 0 && S_ISDIR(sa.st_mode) == 0)
51                     return -1;
52 
53           if ((sa.st_mode & mode) == 0)
54                     return -1;
55 
56           return 0;
57 }
58 
59 ATF_TC(mktemp_not_exist);
ATF_TC_HEAD(mktemp_not_exist,tc)60 ATF_TC_HEAD(mktemp_not_exist, tc)
61 {
62           atf_tc_set_md_var(tc, "descr", "Test that mktemp(3)"
63               " does not fail on a path that does not exist");
64 }
65 
ATF_TC_BODY(mktemp_not_exist,tc)66 ATF_TC_BODY(mktemp_not_exist, tc)
67 {
68           char template[] = "I will barf/XXXXXX";
69           ATF_REQUIRE(mktemp(template) != NULL);
70 }
71 
72 ATF_TC(mktemp_large_template);
ATF_TC_HEAD(mktemp_large_template,tc)73 ATF_TC_HEAD(mktemp_large_template, tc)
74 {
75           atf_tc_set_md_var(tc, "descr", "Test that mktemp "
76               "accepts arbitrarily large templates (PR lib/55441)");
77 }
78 
ATF_TC_BODY(mktemp_large_template,tc)79 ATF_TC_BODY(mktemp_large_template, tc)
80 {
81           const char *path = "/tmp/mktemp.XXXXXX";
82           char *template;
83           size_t i;
84           long name_max;
85           size_t tlen;
86 
87           name_max = pathconf("/tmp/", _PC_NAME_MAX);
88           ATF_REQUIRE(name_max > 0);
89 
90           tlen = (size_t)name_max + sizeof("/tmp/");
91           template = malloc(tlen);
92           ATF_REQUIRE(template != NULL);
93 
94           ATF_REQUIRE(strcpy(template, path) != NULL);
95           ATF_REQUIRE(mktemp(template) != NULL);
96           ATF_REQUIRE(strlen(template) == strlen(path));
97           ATF_REQUIRE(strcmp(template, path) != 0);
98           ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
99 
100           (void)memset(template, '\0', tlen);
101           (void)strcpy(template, "/tmp/mktemp.");
102 
103           for (i = 12; i < tlen - 1; i++)
104                     template[i] = 'X';
105 
106           ATF_REQUIRE(mktemp(template) != NULL);
107           ATF_REQUIRE(strlen(template) == tlen - 1);
108           ATF_REQUIRE(strcmp(template, path) != 0);
109           ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
110           free(template);
111 }
112 
113 ATF_TC(mkstemp_basic);
ATF_TC_HEAD(mkstemp_basic,tc)114 ATF_TC_HEAD(mkstemp_basic, tc)
115 {
116           atf_tc_set_md_var(tc, "descr", "A basic test of mkstemp(3)");
117 }
118 
ATF_TC_BODY(mkstemp_basic,tc)119 ATF_TC_BODY(mkstemp_basic, tc)
120 {
121           char template[] = "/tmp/mktemp.XXXXXX";
122           struct stat sa;
123           int fd;
124 
125           (void)memset(&sa, 0, sizeof(struct stat));
126 
127           fd = mkstemp(template);
128 
129           ATF_REQUIRE(fd != -1);
130           ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
131           ATF_REQUIRE(write(fd, "X", 1) == 1);
132           ATF_REQUIRE(fstat(fd, &sa) == 0);
133           ATF_REQUIRE(check_mode(sa, 0600, 0) == 0);
134           ATF_REQUIRE(close(fd) == 0);
135           ATF_REQUIRE(unlink(template) == 0);
136 }
137 
138 ATF_TC(mkstemps_basic);
ATF_TC_HEAD(mkstemps_basic,tc)139 ATF_TC_HEAD(mkstemps_basic, tc)
140 {
141           atf_tc_set_md_var(tc, "descr", "A basic test of mkstemps(3)");
142 }
143 
ATF_TC_BODY(mkstemps_basic,tc)144 ATF_TC_BODY(mkstemps_basic, tc)
145 {
146           char template[] = "/tmp/mktemp.XXXyyy";
147           char *suffix = strchr(template, 'y');
148           struct stat sa;
149           int fd;
150 
151           (void)memset(&sa, 0, sizeof(struct stat));
152 
153           fd = mkstemps(template, 3);
154 
155           ATF_REQUIRE(fd != -1);
156           ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
157           ATF_REQUIRE(strcmp(suffix, "yyy") == 0);
158           ATF_REQUIRE(write(fd, "X", 1) == 1);
159           ATF_REQUIRE(fstat(fd, &sa) == 0);
160           ATF_REQUIRE(check_mode(sa, 0600, 0) == 0);
161           ATF_REQUIRE(close(fd) == 0);
162           ATF_REQUIRE(unlink(template) == 0);
163 }
164 
165 ATF_TC(mkdtemp_basic);
ATF_TC_HEAD(mkdtemp_basic,tc)166 ATF_TC_HEAD(mkdtemp_basic, tc)
167 {
168           atf_tc_set_md_var(tc, "descr", "A basic test of mkdtemp(3)");
169 }
170 
ATF_TC_BODY(mkdtemp_basic,tc)171 ATF_TC_BODY(mkdtemp_basic, tc)
172 {
173           char template[] = "/tmp/mktemp.XXXXXX";
174           char *path = mkdtemp(template);
175           struct stat sa;
176 
177           (void)memset(&sa, 0, sizeof(struct stat));
178 
179           ATF_REQUIRE(path != NULL);
180           ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
181           ATF_REQUIRE(stat(path, &sa) == 0);
182           ATF_REQUIRE(check_mode(sa, 0700, 1) == 0);
183           ATF_REQUIRE(rmdir(path) == 0);
184 }
185 
186 ATF_TC(mkostemp_basic);
ATF_TC_HEAD(mkostemp_basic,tc)187 ATF_TC_HEAD(mkostemp_basic, tc)
188 {
189           atf_tc_set_md_var(tc, "descr", "A basic test of mkostemp(3)");
190 }
191 
ATF_TC_BODY(mkostemp_basic,tc)192 ATF_TC_BODY(mkostemp_basic, tc)
193 {
194           static const int flags[] = {
195                     O_APPEND, O_DIRECT,
196                     O_SHLOCK, O_EXLOCK,
197                     O_SYNC,   O_CLOEXEC
198           };
199 
200           char template[] = "/tmp/mktemp.XXXXXX";
201           struct stat sa;
202           size_t i;
203           int fd;
204 
205           for (i = 0; i < __arraycount(flags); i++) {
206 
207                     (void)memset(&sa, 0, sizeof(struct stat));
208 
209                     fd = mkostemp(template, flags[i]);
210 
211                     ATF_REQUIRE(fd != -1);
212                     ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
213                     ATF_REQUIRE(write(fd, "X", 1) == 1);
214                     ATF_REQUIRE(fstat(fd, &sa) == 0);
215                     ATF_REQUIRE(check_mode(sa, 0600 | flags[i], 0) == 0);
216                     ATF_REQUIRE(close(fd) == 0);
217                     ATF_REQUIRE(unlink(template) == 0);
218           }
219 }
220 
221 ATF_TC(mkostemps_basic);
ATF_TC_HEAD(mkostemps_basic,tc)222 ATF_TC_HEAD(mkostemps_basic, tc)
223 {
224           atf_tc_set_md_var(tc, "descr", "A basic test of mkostemps(3)");
225 }
226 
ATF_TC_BODY(mkostemps_basic,tc)227 ATF_TC_BODY(mkostemps_basic, tc)
228 {
229           static const int flags[] = {
230                     O_APPEND, O_DIRECT,
231                     O_SHLOCK, O_EXLOCK,
232                     O_SYNC,   O_CLOEXEC
233           };
234 
235           char template[] = "/tmp/mktemp.XXXyyy";
236           char *suffix = strchr(template, 'y');
237           struct stat sa;
238           size_t i;
239           int fd;
240 
241           for (i = 0; i < __arraycount(flags); i++) {
242 
243                     (void)memset(&sa, 0, sizeof(struct stat));
244 
245                     fd = mkostemps(template, 3, flags[i]);
246 
247                     ATF_REQUIRE(fd != -1);
248                     ATF_REQUIRE(strncmp(template, "/tmp/mktemp.", 12) == 0);
249                     ATF_REQUIRE(strcmp(suffix, "yyy") == 0);
250                     ATF_REQUIRE(write(fd, "X", 1) == 1);
251                     ATF_REQUIRE(fstat(fd, &sa) == 0);
252                     ATF_REQUIRE(check_mode(sa, 0600 | flags[i], 0) == 0);
253                     ATF_REQUIRE(close(fd) == 0);
254                     ATF_REQUIRE(unlink(template) == 0);
255           }
256 }
257 
ATF_TP_ADD_TCS(tp)258 ATF_TP_ADD_TCS(tp)
259 {
260 
261           ATF_TP_ADD_TC(tp, mktemp_not_exist);
262           ATF_TP_ADD_TC(tp, mktemp_large_template);
263           ATF_TP_ADD_TC(tp, mkstemp_basic);
264           ATF_TP_ADD_TC(tp, mkstemps_basic);
265           ATF_TP_ADD_TC(tp, mkdtemp_basic);
266           ATF_TP_ADD_TC(tp, mkostemp_basic);
267           ATF_TP_ADD_TC(tp, mkostemps_basic);
268 
269           return atf_no_error();
270 }
271