1/* $NetBSD: t_write.c,v 1.3 2017/01/13 19:27:23 christos Exp $ */
2
3/*-
4 * Copyright (c) 2001, 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30__COPYRIGHT("@(#) Copyright (c) 2008\
31 The NetBSD Foundation, inc. All rights reserved.");
32__RCSID("$NetBSD: t_write.c,v 1.3 2017/01/13 19:27:23 christos Exp $");
33
34#include <sys/uio.h>
35
36#include <atf-c.h>
37#include <errno.h>
38#include <fcntl.h>
39#include <signal.h>
40#include <limits.h>
41#include <stdio.h>
42#include <stdint.h>
43#include <string.h>
44#include <unistd.h>
45
46static void		 sighandler(int);
47
48static bool		 fail = false;
49static const char	*path = "write";
50
51static void
52sighandler(int signo __unused)
53{
54	fail = false;
55}
56
57ATF_TC_WITH_CLEANUP(write_err);
58ATF_TC_HEAD(write_err, tc)
59{
60	atf_tc_set_md_var(tc, "descr", "Checks errors from write(2)");
61}
62
63ATF_TC_BODY(write_err, tc)
64{
65	char rbuf[3] = { 'a', 'b', 'c' };
66	char wbuf[3] = { 'x', 'y', 'z' };
67	int fd;
68
69	errno = 0;
70	ATF_REQUIRE_ERRNO(EBADF, write(-1, wbuf, sizeof(wbuf)) == -1);
71
72	fd = open(path, O_RDWR | O_CREAT, 0600);
73
74	if (fd >= 0) {
75
76		errno = 0;
77		ATF_REQUIRE_ERRNO(0, write(fd, wbuf, 3) == 3);
78
79		errno = 0;
80		ATF_REQUIRE_ERRNO(EINVAL, write(fd, wbuf, SIZE_MAX) == -1);
81
82		errno = 0;
83		ATF_REQUIRE_ERRNO(EFAULT, write(fd, (void *)-1, 1) == -1);
84
85		/*
86		 * Check that the above bogus write(2)
87		 * calls did not corrupt the file.
88		 */
89		ATF_REQUIRE(lseek(fd, 0, SEEK_SET) == 0);
90		ATF_REQUIRE(read(fd, rbuf, 3) == 3);
91		ATF_REQUIRE(memcmp(rbuf, wbuf, 3) == 0);
92
93		(void)close(fd);
94		(void)unlink(path);
95	}
96}
97
98ATF_TC_CLEANUP(write_err, tc)
99{
100	(void)unlink(path);
101}
102
103ATF_TC(write_pipe);
104ATF_TC_HEAD(write_pipe, tc)
105{
106	atf_tc_set_md_var(tc, "descr", "Checks for EPIPE from write(2)");
107}
108
109ATF_TC_BODY(write_pipe, tc)
110{
111	int fds[2];
112
113	ATF_REQUIRE(pipe(fds) == 0);
114	ATF_REQUIRE(signal(SIGPIPE, sighandler) == 0);
115
116	ATF_REQUIRE(write(fds[1], "x", 1) != -1);
117	ATF_REQUIRE(close(fds[0]) == 0);
118
119	errno = 0;
120	fail = true;
121
122	if (write(fds[1], "x", 1) != -1 || errno != EPIPE)
123		atf_tc_fail_nonfatal("expected EPIPE but write(2) succeeded");
124
125	ATF_REQUIRE(close(fds[1]) == 0);
126
127	if (fail != false)
128		atf_tc_fail_nonfatal("SIGPIPE was not raised");
129}
130
131ATF_TC_WITH_CLEANUP(write_pos);
132ATF_TC_HEAD(write_pos, tc)
133{
134	atf_tc_set_md_var(tc, "descr", "Checks that write(2) "
135	    "updates the file position");
136}
137
138ATF_TC_BODY(write_pos, tc)
139{
140	const size_t n = 123;
141	size_t i;
142	int fd;
143
144	fd = open(path, O_RDWR | O_CREAT, 0600);
145	ATF_REQUIRE(fd >= 0);
146
147	for (i = 0; i < n; i++) {
148		ATF_REQUIRE(write(fd, "x", 1) == 1);
149		ATF_REQUIRE(lseek(fd, 0, SEEK_CUR) == (off_t)(i + 1));
150	}
151
152	ATF_REQUIRE(close(fd) == 0);
153	ATF_REQUIRE(unlink(path) == 0);
154}
155
156ATF_TC_CLEANUP(write_pos, tc)
157{
158	(void)unlink(path);
159}
160
161ATF_TC_WITH_CLEANUP(write_ret);
162ATF_TC_HEAD(write_ret, tc)
163{
164	atf_tc_set_md_var(tc, "descr", "Checks return values from write(2)");
165}
166
167ATF_TC_BODY(write_ret, tc)
168{
169	const size_t n = 99;
170	char buf[123];
171	size_t i, j;
172	int fd;
173
174	fd = open(path, O_WRONLY | O_CREAT, 0600);
175	ATF_REQUIRE(fd >= 0);
176
177	(void)memset(buf, 'x', sizeof(buf));
178
179	for (i = j = 0; i < n; i++)
180		j += write(fd, buf, sizeof(buf));
181
182	if (j != n * 123)
183		atf_tc_fail("inconsistent return values from write(2)");
184
185	(void)close(fd);
186	(void)unlink(path);
187}
188
189ATF_TC_CLEANUP(write_ret, tc)
190{
191	(void)unlink(path);
192}
193
194ATF_TC(writev_iovmax);
195ATF_TC_HEAD(writev_iovmax, tc)
196{
197	atf_tc_set_md_var(tc, "timeout", "10");
198	atf_tc_set_md_var(tc, "descr",
199	    "Checks that file descriptor is properly FILE_UNUSE()d "
200	    "when iovcnt is greater than IOV_MAX");
201}
202
203ATF_TC_BODY(writev_iovmax, tc)
204{
205	ssize_t retval;
206
207	(void)printf("Calling writev(2, NULL, IOV_MAX + 1)...\n");
208
209	errno = 0;
210	retval = writev(2, NULL, IOV_MAX + 1);
211
212	ATF_REQUIRE_EQ_MSG(retval, -1, "got: %zd", retval);
213	ATF_REQUIRE_EQ_MSG(errno, EINVAL, "got: %s", strerror(errno));
214}
215
216ATF_TP_ADD_TCS(tp)
217{
218
219	ATF_TP_ADD_TC(tp, write_err);
220	ATF_TP_ADD_TC(tp, write_pipe);
221	ATF_TP_ADD_TC(tp, write_pos);
222	ATF_TP_ADD_TC(tp, write_ret);
223	ATF_TP_ADD_TC(tp, writev_iovmax);
224
225	return atf_no_error();
226}
227