1 /* $NetBSD: t_io.c,v 1.8 2011/02/27 15:16:31 njoly 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 if (FSTYPE_SYSVBFS(tc) && seekcnt) 92 atf_tc_expect_fail("PR kern/44307"); 93 ATF_REQUIRE_EQ(sb.st_size, seekcnt); 94 atf_tc_expect_pass(); 95 96 ATF_REQUIRE_EQ(rump_sys_write(fd, TESTSTR, TESTSZ), TESTSZ); 97 ATF_REQUIRE_EQ(rump_sys_pread(fd, buf, TESTSZ, seekcnt), TESTSZ); 98 ATF_REQUIRE_STREQ(buf, TESTSTR); 99 100 RL(rump_sys_fstat(fd, &sb)); 101 ATF_REQUIRE_EQ(sb.st_size, (off_t)TESTSZ + seekcnt); 102 RL(rump_sys_close(fd)); 103 FSTEST_EXIT(); 104 } 105 106 static void 107 extendfile(const atf_tc_t *tc, const char *mp) 108 { 109 110 extendbody(tc, 0); 111 } 112 113 static void 114 extendfile_append(const atf_tc_t *tc, const char *mp) 115 { 116 117 extendbody(tc, 37); 118 } 119 120 static void 121 overwritebody(const atf_tc_t *tc, off_t count, bool dotrunc) 122 { 123 char *buf; 124 int fd; 125 126 REQUIRE_LIBC(buf = malloc(count), NULL); 127 FSTEST_ENTER(); 128 RL(fd = rump_sys_open("testi", O_CREAT | O_RDWR)); 129 ATF_REQUIRE_EQ(rump_sys_write(fd, buf, count), count); 130 RL(rump_sys_close(fd)); 131 132 RL(fd = rump_sys_open("testi", O_CREAT | O_RDWR)); 133 if (dotrunc) 134 RL(rump_sys_ftruncate(fd, 0)); 135 ATF_REQUIRE_EQ(rump_sys_write(fd, buf, count), count); 136 RL(rump_sys_close(fd)); 137 FSTEST_EXIT(); 138 } 139 140 static void 141 overwrite512(const atf_tc_t *tc, const char *mp) 142 { 143 144 overwritebody(tc, 512, false); 145 } 146 147 static void 148 overwrite64k(const atf_tc_t *tc, const char *mp) 149 { 150 151 overwritebody(tc, 1<<16, false); 152 } 153 154 static void 155 overwrite_trunc(const atf_tc_t *tc, const char *mp) 156 { 157 158 overwritebody(tc, 1<<16, true); 159 } 160 161 static void 162 shrinkfile(const atf_tc_t *tc, const char *mp) 163 { 164 int fd; 165 166 FSTEST_ENTER(); 167 RL(fd = rump_sys_open("file", O_RDWR|O_CREAT|O_TRUNC, 0666)); 168 RL(rump_sys_ftruncate(fd, 2)); 169 RL(rump_sys_ftruncate(fd, 1)); 170 rump_sys_close(fd); 171 FSTEST_EXIT(); 172 } 173 174 ATF_TC_FSAPPLY(holywrite, "create a sparse file and fill hole"); 175 ATF_TC_FSAPPLY(extendfile, "check that extending a file works"); 176 ATF_TC_FSAPPLY(extendfile_append, "check that extending a file works " 177 "with a append-only fd"); 178 ATF_TC_FSAPPLY(overwrite512, "write a 512 byte file twice"); 179 ATF_TC_FSAPPLY(overwrite64k, "write a 64k byte file twice"); 180 ATF_TC_FSAPPLY(overwrite_trunc, "write 64k + truncate + rewrite"); 181 ATF_TC_FSAPPLY(shrinkfile, "shrink file"); 182 183 ATF_TP_ADD_TCS(tp) 184 { 185 186 ATF_TP_FSAPPLY(holywrite); 187 ATF_TP_FSAPPLY(extendfile); 188 ATF_TP_FSAPPLY(extendfile_append); 189 ATF_TP_FSAPPLY(overwrite512); 190 ATF_TP_FSAPPLY(overwrite64k); 191 ATF_TP_FSAPPLY(overwrite_trunc); 192 ATF_TP_FSAPPLY(shrinkfile); 193 194 return atf_no_error(); 195 } 196