| /* |
| * |
| * Copyright (c) International Business Machines Corp., 2002 |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See |
| * the GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| |
| /* |
| * NAME |
| * pwrite04.c (ported from SPIE, section2/filesuite/pread_pwrite.c, |
| * by Airong Zhang) |
| * |
| * TEST SUMMARY |
| * Test the pwrite() system call with O_APPEND. |
| * |
| * USAGE |
| * pwrite04 |
| * |
| */ |
| |
| #define _XOPEN_SOURCE 500 |
| #include <stdio.h> |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <unistd.h> |
| #include <sys/fcntl.h> |
| #include <memory.h> |
| #include <errno.h> |
| #include "usctest.h" |
| #include "test.h" |
| |
| char *TCID = "pwrite04"; |
| int TST_TOTAL = 1; |
| extern int Tst_count; |
| int local_flag; |
| |
| #define PASSED 1 |
| #define FAILED 0 |
| |
| int block_cnt = 0; |
| |
| #define K1 1024 |
| #define K2 (K1 * 2) |
| #define K3 (K1 * 3) |
| #define K4 (K1 * 4) |
| #define K5 (K1 * 5) |
| #define NBUFS 4 |
| #define DATA_FILE "pwrite04_file" |
| |
| char name[256], fname[256]; |
| |
| void init_buffers(char *[]); |
| void l_seek(int, off_t, int, off_t); |
| static void cleanup(void); |
| |
| int main(int ac, char *av[]) |
| { |
| int fd; |
| int nbytes; |
| char *wbuf[NBUFS]; |
| struct stat statbuf; |
| int lc; /* loop counter */ |
| char *msg; /* message returned from parse_opts */ |
| |
| strcpy(name, DATA_FILE); |
| sprintf(fname, "%s.%d", name, getpid()); |
| |
| /* |
| * parse standard options |
| */ |
| if ((msg = parse_opts(ac, av, NULL, NULL)) != NULL) { |
| tst_resm(TBROK, "OPTION PARSING ERROR - %s", msg); |
| |
| } |
| tst_tmpdir(); |
| for (lc = 0; TEST_LOOPING(lc); lc++) { |
| |
| init_buffers(wbuf); |
| local_flag = PASSED; |
| |
| if ((fd = open(fname, O_RDWR | O_CREAT, 0666)) < 0) { |
| tst_resm(TBROK, "open failed: fname = %s, errno = %d", |
| fname, errno); |
| cleanup(); |
| } |
| /* |
| * pwrite() K1 of data (0's) at offset 0. |
| */ |
| if ((nbytes = pwrite(fd, wbuf[0], K1, 0)) != K1) { |
| tst_resm(TFAIL, |
| "pwrite at 0 failed: nbytes=%d, errno=%d", |
| nbytes, errno); |
| cleanup(); |
| } |
| |
| /* |
| * We should still be at offset 0. |
| */ |
| l_seek(fd, 0, SEEK_CUR, 0); |
| |
| /* |
| * lseek() to a non K boundary, just to be different. |
| */ |
| l_seek(fd, K1 / 2, SEEK_SET, K1 / 2); |
| |
| /* |
| * pwrite() K1 of data (2's) at offset K2. |
| */ |
| if ((nbytes = pwrite(fd, wbuf[2], K1, K2)) != K1) { |
| tst_resm(TFAIL, |
| "pwrite at K2 failed: nbytes=%d, errno=%d", |
| nbytes, errno); |
| cleanup(); |
| } |
| |
| /* |
| * We should still be at our non K boundary. |
| */ |
| l_seek(fd, 0, SEEK_CUR, K1 / 2); |
| |
| /* |
| * lseek() to an offset of K3. |
| */ |
| l_seek(fd, K3, SEEK_SET, K3); |
| |
| /* |
| * This time use a normal write() of K1 of data (3's) which should |
| * take place at an offset of K3, moving the file pointer to K4. |
| */ |
| if ((nbytes = write(fd, wbuf[3], K1)) != K1) { |
| tst_resm(TFAIL, "write failed: nbytes=%d, errno=%d", |
| nbytes, errno); |
| cleanup(); |
| } |
| |
| /* |
| * We should be at offset K4. |
| */ |
| l_seek(fd, 0, SEEK_CUR, K4); |
| |
| /* |
| * pwrite() K1 of data (1's) at offset K1. |
| */ |
| if ((nbytes = pwrite(fd, wbuf[1], K1, K1)) != K1) { |
| tst_resm(TFAIL, "pwrite failed: nbytes=%d, errno=%d", |
| nbytes, errno); |
| cleanup(); |
| } |
| |
| /*--------------------------------------------------------------*/ |
| |
| /* |
| * Now test that O_APPEND takes precedence over any |
| * offset specified by pwrite(), but that the file |
| * pointer remains unchanged. First, close then reopen |
| * the file and ensure it is already K4 in length and |
| * set the file pointer to it's midpoint, K2. |
| */ |
| close(fd); |
| if ((fd = open(fname, O_RDWR | O_APPEND, 0666)) < 0) { |
| tst_resm(TBROK, "open failed: fname = %s, errno = %d", |
| fname, errno); |
| cleanup(); |
| } |
| if (fstat(fd, &statbuf) == -1) { |
| tst_resm(TFAIL, "fstat failed: errno = %d", errno); |
| cleanup(); |
| } |
| if (statbuf.st_size != K4) { |
| tst_resm(TFAIL, "file size is %ld != K4", |
| statbuf.st_size); |
| cleanup(); |
| } |
| l_seek(fd, K2, SEEK_SET, K2); |
| |
| /* |
| * Finally, pwrite() some K1 of data at offset 0. |
| * What we should end up with is: |
| * -The file pointer should still be at K2. |
| * -The data should have been written to the end |
| * of the file (O_APPEND) and should be K5 in size. |
| */ |
| if ((nbytes = pwrite(fd, wbuf[0], K1, 0)) != K1) { |
| tst_resm(TFAIL, |
| "pwrite at 0 failed: nbytes=%d, errno=%d", |
| nbytes, errno); |
| |
| } |
| l_seek(fd, 0, SEEK_CUR, K2); |
| if (fstat(fd, &statbuf) == -1) { |
| tst_resm(TFAIL, "fstat failed: errno = %d", errno); |
| |
| } |
| if (statbuf.st_size != K5) { |
| tst_resm(TFAIL, "file size is %ld != K4", |
| statbuf.st_size); |
| |
| } |
| tst_resm(TPASS, "O_APPEND test passed."); |
| |
| /*------------------------------------------------------------------------*/ |
| |
| close(fd); |
| unlink(fname); |
| } /* end for */ |
| cleanup(); |
| |
| } |
| |
| /*------------------------------------------------------------------------*/ |
| |
| /* |
| * init_buffers() allocates wbuf[] array |
| * as follows: |
| * wbuf[0] has 0's, wbuf[1] has 1's, wbuf[2] has 2's, and wbuf[3] has 3's. |
| */ |
| void init_buffers(char *wbuf[]) |
| { |
| int i; |
| |
| for (i = 0; i < NBUFS; i++) { |
| wbuf[i] = (char *)malloc(K1); |
| if (wbuf[i] == NULL) { |
| tst_resm(TBROK, "ib: malloc failed: errno=%d", errno); |
| tst_exit(); |
| } |
| memset(wbuf[i], i, K1); |
| } |
| } |
| |
| /* |
| * l_seek() is a local front end to lseek(). |
| * "checkoff" is the offset at which we believe we should be at. |
| * Used to validate pwrite doesn't move the offset. |
| */ |
| void l_seek(int fdesc, off_t offset, int whence, off_t checkoff) |
| { |
| off_t offloc; |
| |
| if ((offloc = lseek(fdesc, offset, whence)) != checkoff) { |
| tst_resm(TFAIL, "(%ld = lseek(%d, %ld, %d)) != %ld) errno = %d", |
| offloc, fdesc, offset, whence, checkoff, errno); |
| tst_exit(); |
| } |
| } |
| |
| /* |
| * cleanup() - Performs all ONE TIME cleanup for this test at |
| * completion or premature exit. |
| * |
| * Print test timing stats and errno log if test executed with options. |
| * Close the testfile if still opened. |
| * Remove temporary directory and sub-directories/files under it |
| * created during setup(). |
| * Exit the test program with normal exit code. |
| */ |
| void cleanup() |
| { |
| /* |
| * print timing stats if that option was specified. |
| * print errno log if that option was specified. |
| */ |
| TEST_CLEANUP; |
| |
| tst_rmdir(); |
| |
| } |