/*
*
* Copyright (c) International Business Machines Corp., 2001
*
* 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
/*
* NAME
* writev04.c
*
* DESCRIPTION
* The testcases are written calling writev() with partially valid data
* to overwrite the contents, to write in the beginning and to write in
* the end of the file. This is same as writev03.c, but the length of
* buffer used here is 8192 bytes.
*
* USAGE: <for command-line>
* writev04 [-c n] [-e] [-i n] [-I x] [-P x] [-t]
* where, -c n : Run n copies concurrently.
* -e : Turn on errno logging.
* -i n : Execute test n times.
* -I x : Execute test for x seconds.
* -P x : Pause for x seconds between iterations.
* -t : Turn on syscall timing.
*
* History
* 07/2001 John George
* -Ported
* 04/2002 wjhuie sigset cleanups
*
* Restrictions
* NONE
*/
#include <sys/types.h>
#include <sys/uio.h>
#include <sys/mman.h>
#include <unistd.h>
#include <signal.h>
#include <fcntl.h>
#include <memory.h>
#include <errno.h>
#include "test.h"
#define K_1 8192
#define NBUFS 4
#define CHUNK 64 /* single chunk */
#define MAX_IOVEC 4
#define DATA_FILE "writev_data_file"
char buf1[K_1], buf2[K_1], buf3[K_1];
char *bad_addr = 0;
struct iovec wr_iovec[MAX_IOVEC] = {
/* testcase #1 */
{buf1 + (CHUNK * 6), CHUNK},
{(caddr_t) - 1, CHUNK},
{buf1 + (CHUNK * 8), CHUNK},
{NULL, 0}
};
char name[K_1], f_name[K_1];
int fd[2], in_sighandler;
char *buf_list[NBUFS];
char *TCID = "writev04";
int TST_TOTAL = 1;
void sighandler(int);
long l_seek(int, long, int);
void setup(void);
void cleanup(void);
int fail;
#if !defined(UCLINUX)
int main(int argc, char **argv)
{
int lc;
int nbytes;
tst_parse_opts(argc, argv, NULL, NULL);
setup(); /* set "tstdir", and "testfile" vars */
/* The following loop checks looping state if -i option given */
for (lc = 0; TEST_LOOPING(lc); lc++) {
/* reset tst_count in case we are looping */
tst_count = 0;
buf_list[0] = buf1;
buf_list[1] = buf2;
buf_list[2] = buf3;
buf_list[3] = NULL;
fd[1] = -1; /* Invalid file descriptor */
if (signal(SIGTERM, sighandler) == SIG_ERR) {
perror("signal");
tst_resm(TFAIL, "signal() SIGTERM FAILED");
cleanup();
}
if (signal(SIGPIPE, sighandler) == SIG_ERR) {
perror("signal");
tst_resm(TFAIL, "signal() SIGPIPE FAILED");
cleanup();
}
memset(buf_list[0], 0, K_1);
memset(buf_list[1], 0, K_1);
if ((fd[0] = open(f_name, O_WRONLY | O_CREAT, 0666)) < 0) {
tst_resm(TFAIL, "open(2) failed: fname = %s, "
"errno = %d", f_name, errno);
cleanup();
} else {
if ((nbytes = write(fd[0], buf_list[1], K_1)) != K_1) {
tst_resm(TFAIL, "write(2) failed: nbytes "
"= %d, errno = %d", nbytes, errno);
cleanup();
}
}
if (close(fd[0]) < 0) {
tst_resm(TFAIL, "close failed: errno = %d", errno);
cleanup();
}
if ((fd[0] = open(f_name, O_RDWR, 0666)) < 0) {
tst_brkm(TFAIL, cleanup, "open failed: fname = %s, errno = %d",
f_name, errno);
}
//block1:
tst_resm(TINFO, "Enter block 1");
fail = 0;
/*
* In this block we are trying to call writev() with
* partially valid data. This should return the valid number
* of bytes written in the vector. If it returns EFAULT, it
* is an error. And after returning the number of valid
* bytes written, the check should be made to verify the
* contents of the first valid write() scheduled.
*/
if (writev(fd[0], wr_iovec, 3) < 0) {
fail = 1;
if (errno == EFAULT) {
tst_resm(TFAIL, "Got error EFAULT");
} else {
tst_resm(TFAIL, "Received unexpected error: %d",
errno);
}
} else {
l_seek(fd[0], 0, 0);
read(fd[0], buf_list[0], CHUNK);
if (memcmp(buf_list[0], buf_list[1], CHUNK) != 0) {
tst_resm(TFAIL, "writev overwrote the file");
fail = 1;
}
}
if (fail) {
tst_resm(TINFO, "block 1 FAILED");
} else {
tst_resm(TINFO, "block 1 PASSED");
}
tst_resm(TINFO, "Exit block 1");
//block2:
tst_resm(TINFO, "Enter block 2");
fail = 0;
/*
* In this block we are trying to over write the contents by
* calling writev() with partially valid data. It should
* return the valid number of bytes written but not EFAULT.
* Also the check should be made whether the initial write()
* scheduled is done correctly or not.
*/
l_seek(fd[0], 0, 0);
if (writev(fd[0], wr_iovec, 3) < 0) {
fail = 1;
if (errno == EFAULT) {
tst_resm(TFAIL, "Got error EFAULT");
} else {
tst_resm(TFAIL, "Received unexpected error: %d",
errno);
}
} else {
l_seek(fd[0], 0, 0);
read(fd[0], buf_list[0], CHUNK);
if (memcmp(buf_list[0], buf_list[1], CHUNK) != 0) {
tst_resm(TFAIL, "writev overwrote the file");
fail = 1;
}
}
if (fail) {
tst_resm(TINFO, "block 2 FAILED");
} else {
tst_resm(TINFO, "block 2 PASSED");
}
tst_resm(TINFO, "Exit block 2");
//block3:
tst_resm(TINFO, "Enter block 3");
fail = 0;
/*
* In this block, we are trying to call writev() by going to
* some end position of the file. Here writev() is called
* with partially valid data, and this will return the
* number of valid bytes written and not EFAULT. Also, the
* check should be made whether the inital write() that is
* scheduled with valid data is done correctly.
*/
l_seek(fd[0], 8192, 0);
if (writev(fd[0], wr_iovec, 3) < 0) {
fail = 1;
if (errno == EFAULT) {
tst_resm(TFAIL, "Got error EFAULT");
} else {
tst_resm(TFAIL, "Received unexpected error: %d",
errno);
}
} else {
l_seek(fd[0], 0, 0);
read(fd[0], buf_list[0], CHUNK);
if (memcmp(buf_list[0], buf_list[1], CHUNK) != 0) {
tst_resm(TFAIL, "writev overwrote the file");
fail = 1;
}
}
if (fail) {
tst_resm(TINFO, "block 3 FAILED");
} else {
tst_resm(TINFO, "block 3 PASSED");
}
tst_resm(TINFO, "Exit block 3");
}
close(fd[0]);
close(fd[1]);
cleanup();
tst_exit();
}
#else
int main(void)
{
tst_resm(TINFO, "test is not available on uClinux");
tst_exit();
}
#endif /* if !defined(UCLINUX) */
/*
* setup()
* performs all ONE TIME setup for this test
*/
void setup(void)
{
tst_sig(FORK, DEF_HANDLER, cleanup);
TEST_PAUSE;
/* Create a unique temporary directory and chdir() to it. */
tst_tmpdir();
strcpy(name, DATA_FILE);
sprintf(f_name, "%s.%d", name, getpid());
bad_addr = mmap(0, 1, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, 0, 0);
if (bad_addr == MAP_FAILED) {
tst_brkm(TBROK, cleanup, "mmap failed");
}
wr_iovec[1].iov_base = bad_addr;
}
/*
* cleanup()
* performs all ONE TIME cleanup for this test at
* completion or premature exit
*/
void cleanup(void)
{
if (unlink(f_name) < 0) {
tst_resm(TFAIL, "unlink Failed--file = %s, errno = %d",
f_name, errno);
}
tst_rmdir();
}
/*
* sighandler()
* Signal handler function for SIGTERM and SIGPIPE
*/
void sighandler(int sig)
{
switch (sig) {
case SIGTERM:
break;
case SIGPIPE:
++in_sighandler;
return;
default:
tst_resm(TFAIL, "sighandler() received invalid signal "
": %d", sig);
break;
}
if ((unlink(f_name) < 0) && (errno != ENOENT)) {
tst_resm(TFAIL, "unlink Failed--file = %s, errno = %d",
f_name, errno);
cleanup();
}
exit(sig);
}
/*
* l_seek()
* Wrap around for regular lseek function for giving error message
*/
long l_seek(int fdesc, long offset, int whence)
{
if (lseek(fdesc, offset, whence) < 0) {
tst_resm(TFAIL, "lseek Failed : errno = %d", errno);
fail = 1;
}
return 0;
}