t_io.c revision 1.9 1 /* $NetBSD: t_io.c,v 1.9 2012/03/20 05:21:45 jruoho Exp $ */
2
3 /*-
4 * Copyright (c) 2010 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/stat.h>
30 #include <sys/statvfs.h>
31
32 #include <atf-c.h>
33 #include <fcntl.h>
34 #include <libgen.h>
35 #include <stdlib.h>
36 #include <unistd.h>
37
38 #include <rump/rump_syscalls.h>
39 #include <rump/rump.h>
40
41 #include "../common/h_fsmacros.h"
42 #include "../../h_macros.h"
43
44 #define TESTSTR "this is a string. collect enough and you'll have Em"
45 #define TESTSZ sizeof(TESTSTR)
46
47 static void
48 holywrite(const atf_tc_t *tc, const char *mp)
49 {
50 char buf[1024];
51 char *b2, *b3;
52 size_t therange = getpagesize()+1;
53 int fd;
54
55 FSTEST_ENTER();
56
57 RL(fd = rump_sys_open("file", O_RDWR|O_CREAT|O_TRUNC, 0666));
58
59 memset(buf, 'A', sizeof(buf));
60 RL(rump_sys_pwrite(fd, buf, 1, getpagesize()));
61
62 memset(buf, 'B', sizeof(buf));
63 RL(rump_sys_pwrite(fd, buf, 2, getpagesize()-1));
64
65 REQUIRE_LIBC(b2 = malloc(2 * getpagesize()), NULL);
66 REQUIRE_LIBC(b3 = malloc(2 * getpagesize()), NULL);
67
68 RL(rump_sys_pread(fd, b2, therange, 0));
69
70 memset(b3, 0, therange);
71 memset(b3 + getpagesize() - 1, 'B', 2);
72
73 ATF_REQUIRE_EQ(memcmp(b2, b3, therange), 0);
74
75 rump_sys_close(fd);
76 FSTEST_EXIT();
77 }
78
79 static void
80 extendbody(const atf_tc_t *tc, off_t seekcnt)
81 {
82 char buf[TESTSZ+1];
83 struct stat sb;
84 int fd;
85
86 FSTEST_ENTER();
87 RL(fd = rump_sys_open("testfile",
88 O_CREAT | O_RDWR | (seekcnt ? O_APPEND : 0)));
89 RL(rump_sys_ftruncate(fd, seekcnt));
90 RL(rump_sys_fstat(fd, &sb));
91 ATF_REQUIRE_EQ(sb.st_size, seekcnt);
92 atf_tc_expect_pass();
93
94 ATF_REQUIRE_EQ(rump_sys_write(fd, TESTSTR, TESTSZ), TESTSZ);
95 ATF_REQUIRE_EQ(rump_sys_pread(fd, buf, TESTSZ, seekcnt), TESTSZ);
96 ATF_REQUIRE_STREQ(buf, TESTSTR);
97
98 RL(rump_sys_fstat(fd, &sb));
99 ATF_REQUIRE_EQ(sb.st_size, (off_t)TESTSZ + seekcnt);
100 RL(rump_sys_close(fd));
101 FSTEST_EXIT();
102 }
103
104 static void
105 extendfile(const atf_tc_t *tc, const char *mp)
106 {
107
108 extendbody(tc, 0);
109 }
110
111 static void
112 extendfile_append(const atf_tc_t *tc, const char *mp)
113 {
114
115 extendbody(tc, 37);
116 }
117
118 static void
119 overwritebody(const atf_tc_t *tc, off_t count, bool dotrunc)
120 {
121 char *buf;
122 int fd;
123
124 REQUIRE_LIBC(buf = malloc(count), NULL);
125 FSTEST_ENTER();
126 RL(fd = rump_sys_open("testi", O_CREAT | O_RDWR));
127 ATF_REQUIRE_EQ(rump_sys_write(fd, buf, count), count);
128 RL(rump_sys_close(fd));
129
130 RL(fd = rump_sys_open("testi", O_CREAT | O_RDWR));
131 if (dotrunc)
132 RL(rump_sys_ftruncate(fd, 0));
133 ATF_REQUIRE_EQ(rump_sys_write(fd, buf, count), count);
134 RL(rump_sys_close(fd));
135 FSTEST_EXIT();
136 }
137
138 static void
139 overwrite512(const atf_tc_t *tc, const char *mp)
140 {
141
142 overwritebody(tc, 512, false);
143 }
144
145 static void
146 overwrite64k(const atf_tc_t *tc, const char *mp)
147 {
148
149 overwritebody(tc, 1<<16, false);
150 }
151
152 static void
153 overwrite_trunc(const atf_tc_t *tc, const char *mp)
154 {
155
156 overwritebody(tc, 1<<16, true);
157 }
158
159 static void
160 shrinkfile(const atf_tc_t *tc, const char *mp)
161 {
162 int fd;
163
164 FSTEST_ENTER();
165 RL(fd = rump_sys_open("file", O_RDWR|O_CREAT|O_TRUNC, 0666));
166 RL(rump_sys_ftruncate(fd, 2));
167 RL(rump_sys_ftruncate(fd, 1));
168 rump_sys_close(fd);
169 FSTEST_EXIT();
170 }
171
172 ATF_TC_FSAPPLY(holywrite, "create a sparse file and fill hole");
173 ATF_TC_FSAPPLY(extendfile, "check that extending a file works");
174 ATF_TC_FSAPPLY(extendfile_append, "check that extending a file works "
175 "with a append-only fd (PR kern/44307)");
176 ATF_TC_FSAPPLY(overwrite512, "write a 512 byte file twice");
177 ATF_TC_FSAPPLY(overwrite64k, "write a 64k byte file twice");
178 ATF_TC_FSAPPLY(overwrite_trunc, "write 64k + truncate + rewrite");
179 ATF_TC_FSAPPLY(shrinkfile, "shrink file");
180
181 ATF_TP_ADD_TCS(tp)
182 {
183
184 ATF_TP_FSAPPLY(holywrite);
185 ATF_TP_FSAPPLY(extendfile);
186 ATF_TP_FSAPPLY(extendfile_append);
187 ATF_TP_FSAPPLY(overwrite512);
188 ATF_TP_FSAPPLY(overwrite64k);
189 ATF_TP_FSAPPLY(overwrite_trunc);
190 ATF_TP_FSAPPLY(shrinkfile);
191
192 return atf_no_error();
193 }
194