1 /* 2 * 3 * Copyright (c) International Business Machines Corp., 2002 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of the GNU General Public License as published by 7 * the Free Software Foundation; either version 2 of the License, or 8 * (at your option) any later version. 9 * 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See 13 * the GNU General Public License for more details. 14 * 15 * You should have received a copy of the GNU General Public License 16 * along with this program; if not, write to the Free Software 17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 18 */ 19 20 /* 21 * NAME 22 * pwrite04.c (ported from SPIE, section2/filesuite/pread_pwrite.c, 23 * by Airong Zhang) 24 * 25 * TEST SUMMARY 26 * Test the pwrite() system call with O_APPEND. 27 * 28 * USAGE 29 * pwrite04 30 * 31 */ 32 33 #define _XOPEN_SOURCE 500 34 #include <stdio.h> 35 #include <sys/types.h> 36 #include <sys/stat.h> 37 #include <unistd.h> 38 #include <fcntl.h> 39 #include <memory.h> 40 #include <errno.h> 41 #include "test.h" 42 43 char *TCID = "pwrite04"; 44 int TST_TOTAL = 1; 45 int local_flag; 46 47 #define PASSED 1 48 #define FAILED 0 49 50 int block_cnt = 0; 51 52 #define K1 1024 53 #define K2 (K1 * 2) 54 #define K3 (K1 * 3) 55 #define K4 (K1 * 4) 56 #define K5 (K1 * 5) 57 #define NBUFS 4 58 #define DATA_FILE "pwrite04_file" 59 60 char name[256], fname[256]; 61 62 void init_buffers(char *[]); 63 void l_seek(int, off_t, int, off_t); 64 static void cleanup(void); 65 66 int main(int ac, char *av[]) 67 { 68 int fd; 69 int nbytes; 70 char *wbuf[NBUFS]; 71 struct stat statbuf; 72 int lc; 73 74 strcpy(name, DATA_FILE); 75 sprintf(fname, "%s.%d", name, getpid()); 76 77 tst_parse_opts(ac, av, NULL, NULL); 78 79 tst_tmpdir(); 80 for (lc = 0; TEST_LOOPING(lc); lc++) { 81 82 init_buffers(wbuf); 83 local_flag = PASSED; 84 85 if ((fd = open(fname, O_RDWR | O_CREAT, 0666)) < 0) { 86 tst_resm(TBROK, "open failed: fname = %s, errno = %d", 87 fname, errno); 88 cleanup(); 89 } 90 /* 91 * pwrite() K1 of data (0's) at offset 0. 92 */ 93 if ((nbytes = pwrite(fd, wbuf[0], K1, 0)) != K1) { 94 tst_resm(TFAIL, 95 "pwrite at 0 failed: nbytes=%d, errno=%d", 96 nbytes, errno); 97 cleanup(); 98 } 99 100 /* 101 * We should still be at offset 0. 102 */ 103 l_seek(fd, 0, SEEK_CUR, 0); 104 105 /* 106 * lseek() to a non K boundary, just to be different. 107 */ 108 l_seek(fd, K1 / 2, SEEK_SET, K1 / 2); 109 110 /* 111 * pwrite() K1 of data (2's) at offset K2. 112 */ 113 if ((nbytes = pwrite(fd, wbuf[2], K1, K2)) != K1) { 114 tst_resm(TFAIL, 115 "pwrite at K2 failed: nbytes=%d, errno=%d", 116 nbytes, errno); 117 cleanup(); 118 } 119 120 /* 121 * We should still be at our non K boundary. 122 */ 123 l_seek(fd, 0, SEEK_CUR, K1 / 2); 124 125 /* 126 * lseek() to an offset of K3. 127 */ 128 l_seek(fd, K3, SEEK_SET, K3); 129 130 /* 131 * This time use a normal write() of K1 of data (3's) which should 132 * take place at an offset of K3, moving the file pointer to K4. 133 */ 134 if ((nbytes = write(fd, wbuf[3], K1)) != K1) { 135 tst_resm(TFAIL, "write failed: nbytes=%d, errno=%d", 136 nbytes, errno); 137 cleanup(); 138 } 139 140 /* 141 * We should be at offset K4. 142 */ 143 l_seek(fd, 0, SEEK_CUR, K4); 144 145 /* 146 * pwrite() K1 of data (1's) at offset K1. 147 */ 148 if ((nbytes = pwrite(fd, wbuf[1], K1, K1)) != K1) { 149 tst_resm(TFAIL, "pwrite failed: nbytes=%d, errno=%d", 150 nbytes, errno); 151 cleanup(); 152 } 153 154 /*--------------------------------------------------------------*/ 155 156 /* 157 * Now test that O_APPEND takes precedence over any 158 * offset specified by pwrite(), but that the file 159 * pointer remains unchanged. First, close then reopen 160 * the file and ensure it is already K4 in length and 161 * set the file pointer to it's midpoint, K2. 162 */ 163 close(fd); 164 if ((fd = open(fname, O_RDWR | O_APPEND, 0666)) < 0) { 165 tst_resm(TBROK, "open failed: fname = %s, errno = %d", 166 fname, errno); 167 cleanup(); 168 } 169 if (fstat(fd, &statbuf) == -1) { 170 tst_resm(TFAIL, "fstat failed: errno = %d", errno); 171 cleanup(); 172 } 173 if (statbuf.st_size != K4) { 174 tst_resm(TFAIL, "file size is %ld != K4", 175 statbuf.st_size); 176 cleanup(); 177 } 178 l_seek(fd, K2, SEEK_SET, K2); 179 180 /* 181 * Finally, pwrite() some K1 of data at offset 0. 182 * What we should end up with is: 183 * -The file pointer should still be at K2. 184 * -The data should have been written to the end 185 * of the file (O_APPEND) and should be K5 in size. 186 */ 187 if ((nbytes = pwrite(fd, wbuf[0], K1, 0)) != K1) { 188 tst_resm(TFAIL, 189 "pwrite at 0 failed: nbytes=%d, errno=%d", 190 nbytes, errno); 191 192 } 193 l_seek(fd, 0, SEEK_CUR, K2); 194 if (fstat(fd, &statbuf) == -1) { 195 tst_resm(TFAIL, "fstat failed: errno = %d", errno); 196 197 } 198 if (statbuf.st_size != K5) { 199 tst_resm(TFAIL, "file size is %ld != K4", 200 statbuf.st_size); 201 202 } 203 tst_resm(TPASS, "O_APPEND test passed."); 204 205 /*------------------------------------------------------------------------*/ 206 207 close(fd); 208 unlink(fname); 209 } /* end for */ 210 cleanup(); 211 tst_exit(); 212 213 } 214 215 /*------------------------------------------------------------------------*/ 216 217 /* 218 * init_buffers() allocates wbuf[] array 219 * as follows: 220 * wbuf[0] has 0's, wbuf[1] has 1's, wbuf[2] has 2's, and wbuf[3] has 3's. 221 */ 222 void init_buffers(char *wbuf[]) 223 { 224 int i; 225 226 for (i = 0; i < NBUFS; i++) { 227 wbuf[i] = malloc(K1); 228 if (wbuf[i] == NULL) { 229 tst_brkm(TBROK, NULL, "ib: malloc failed: errno=%d", 230 errno); 231 } 232 memset(wbuf[i], i, K1); 233 } 234 } 235 236 /* 237 * l_seek() is a local front end to lseek(). 238 * "checkoff" is the offset at which we believe we should be at. 239 * Used to validate pwrite doesn't move the offset. 240 */ 241 void l_seek(int fdesc, off_t offset, int whence, off_t checkoff) 242 { 243 off_t offloc; 244 245 if ((offloc = lseek(fdesc, offset, whence)) != checkoff) { 246 tst_brkm(TFAIL, NULL, 247 "(%ld = lseek(%d, %ld, %d)) != %ld) errno = %d", 248 offloc, fdesc, offset, whence, checkoff, errno); 249 } 250 } 251 252 /* 253 * cleanup() - Performs all ONE TIME cleanup for this test at 254 * completion or premature exit. 255 * 256 * Print test timing stats and errno log if test executed with options. 257 * Close the testfile if still opened. 258 * Remove temporary directory and sub-directories/files under it 259 * created during setup(). 260 * Exit the test program with normal exit code. 261 */ 262 void cleanup(void) 263 { 264 265 tst_rmdir(); 266 267 } 268