1 /*-
2 * Copyright (c) 2007-2009 Dag-Erling Smørgrav
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer
10 * in this position and unchanged.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28 #include <sys/types.h>
29 #include <sys/fcntl.h>
30
31 #include <errno.h>
32 #include <signal.h>
33 #include <stdint.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <unistd.h>
38
39 #include <libutil.h>
40
41 /*
42 * Test that flopen() can create a file.
43 */
44 const char *
test_flopen_create(void)45 test_flopen_create(void)
46 {
47 const char *fn = "test_flopen_create";
48 const char *result = NULL;
49 int fd;
50
51 unlink(fn);
52 fd = flopen(fn, O_RDWR|O_CREAT, 0640);
53 if (fd < 0) {
54 result = strerror(errno);
55 } else {
56 close(fd);
57 }
58 unlink(fn);
59 return (result);
60 }
61
62 /*
63 * Test that flopen() can open an existing file.
64 */
65 const char *
test_flopen_open(void)66 test_flopen_open(void)
67 {
68 const char *fn = "test_flopen_open";
69 const char *result = NULL;
70 int fd;
71
72 fd = open(fn, O_RDWR|O_CREAT, 0640);
73 if (fd < 0) {
74 result = strerror(errno);
75 } else {
76 close(fd);
77 fd = flopen(fn, O_RDWR);
78 if (fd < 0) {
79 result = strerror(errno);
80 } else {
81 close(fd);
82 }
83 }
84 unlink(fn);
85 return (result);
86 }
87
88 /*
89 * Test that flopen() can lock against itself
90 */
91 const char *
test_flopen_lock_self(void)92 test_flopen_lock_self(void)
93 {
94 const char *fn = "test_flopen_lock_self";
95 const char *result = NULL;
96 int fd1, fd2;
97
98 unlink(fn);
99 fd1 = flopen(fn, O_RDWR|O_CREAT, 0640);
100 if (fd1 < 0) {
101 result = strerror(errno);
102 } else {
103 fd2 = flopen(fn, O_RDWR|O_NONBLOCK);
104 if (fd2 >= 0) {
105 result = "second open succeeded";
106 close(fd2);
107 }
108 close(fd1);
109 }
110 unlink(fn);
111 return (result);
112 }
113
114 /*
115 * Test that flopen() can lock against other processes
116 */
117 const char *
test_flopen_lock_other(void)118 test_flopen_lock_other(void)
119 {
120 const char *fn = "test_flopen_lock_other";
121 const char *result = NULL;
122 volatile int fd1, fd2;
123
124 unlink(fn);
125 fd1 = flopen(fn, O_RDWR|O_CREAT, 0640);
126 if (fd1 < 0) {
127 result = strerror(errno);
128 } else {
129 fd2 = -42;
130 if (vfork() == 0) {
131 fd2 = flopen(fn, O_RDWR|O_NONBLOCK);
132 close(fd2);
133 _exit(0);
134 }
135 if (fd2 == -42)
136 result = "vfork() doesn't work as expected";
137 if (fd2 >= 0)
138 result = "second open succeeded";
139 close(fd1);
140 }
141 unlink(fn);
142 return (result);
143 }
144
145 /*
146 * Test that child processes inherit the lock
147 */
148 const char *
test_flopen_lock_child(void)149 test_flopen_lock_child(void)
150 {
151 const char *fn = "test_flopen_lock_child";
152 const char *result = NULL;
153 pid_t pid;
154 volatile int fd1, fd2;
155
156 unlink(fn);
157 fd1 = flopen(fn, O_RDWR|O_CREAT, 0640);
158 if (fd1 < 0) {
159 result = strerror(errno);
160 } else {
161 pid = fork();
162 if (pid == -1) {
163 result = strerror(errno);
164 } else if (pid == 0) {
165 select(0, 0, 0, 0, 0);
166 _exit(0);
167 }
168 close(fd1);
169 if ((fd2 = flopen(fn, O_RDWR|O_NONBLOCK)) != -1) {
170 result = "second open succeeded";
171 close(fd2);
172 }
173 kill(pid, SIGINT);
174 }
175 unlink(fn);
176 return (result);
177 }
178
179 static struct test {
180 const char *name;
181 const char *(*func)(void);
182 } t[] = {
183 { "flopen_create", test_flopen_create },
184 { "flopen_open", test_flopen_open },
185 { "flopen_lock_self", test_flopen_lock_self },
186 { "flopen_lock_other", test_flopen_lock_other },
187 { "flopen_lock_child", test_flopen_lock_child },
188 };
189
190 int
main(void)191 main(void)
192 {
193 const char *result;
194 int i, nt;
195
196 nt = sizeof(t) / sizeof(*t);
197 printf("1..%d\n", nt);
198 for (i = 0; i < nt; ++i) {
199 if ((result = t[i].func()) != NULL)
200 printf("not ok %d - %s # %s\n", i + 1,
201 t[i].name, result);
202 else
203 printf("ok %d - %s\n", i + 1,
204 t[i].name);
205 }
206 exit(0);
207 }
208