/*
*
* 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
* sendfile02.c
*
* DESCRIPTION
* Testcase to test the basic functionality of the sendfile(2) system call.
*
* ALGORITHM
* 1. call sendfile(2) with offset = 0
* 2. call sendfile(2) with offset in the middle of the file
*
* USAGE: <for command-line>
* sendfile02 [-c n] [-f] [-i n] [-I x] [-P x] [-t]
* where,
* -f : Turn off functionality Testing.
* -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 Ported by Wayne Boyer
* 08/2002 Make it use a socket so it works with 2.5 kernel
*
* RESTRICTIONS
* NONE
*/
#include <stdio.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/sendfile.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <inttypes.h>
#include "test.h"
#include "safe_macros.h"
#ifndef OFF_T
#define OFF_T off_t
#endif /* Not def: OFF_T */
TCID_DEFINE(sendfile02);
int TST_TOTAL = 4;
char in_file[100];
char out_file[100];
int out_fd;
pid_t child_pid;
static int sockfd, s;
static struct sockaddr_in sin1; /* shared between do_child and create_server */
void cleanup(void);
void do_child(void);
void setup(void);
int create_server(void);
struct test_case_t {
char *desc;
int offset;
int exp_retval;
int exp_updated_offset;
} testcases[] = {
{
"Test sendfile(2) with offset = 0", 0, 26, 26}, {
"Test sendfile(2) with offset in the middle of file", 2, 24, 26}, {
"Test sendfile(2) with offset in the middle of file", 4, 22, 26}, {
"Test sendfile(2) with offset in the middle of file", 6, 20, 26}
};
#ifdef UCLINUX
static char *argv0;
#endif
void do_sendfile(OFF_T offset, int i)
{
int in_fd;
struct stat sb;
int wait_status;
int wait_stat;
off_t before_pos, after_pos;
out_fd = create_server();
if ((in_fd = open(in_file, O_RDONLY)) < 0) {
tst_brkm(TBROK, cleanup, "open failed: %d", errno);
}
SAFE_STAT(cleanup, in_file, &sb);
if ((before_pos = lseek(in_fd, 0, SEEK_CUR)) < 0) {
tst_brkm(TBROK, cleanup,
"lseek before invoking sendfile failed: %d", errno);
}
TEST(sendfile(out_fd, in_fd, &offset, sb.st_size - offset));
if ((after_pos = lseek(in_fd, 0, SEEK_CUR)) < 0) {
tst_brkm(TBROK, cleanup,
"lseek after invoking sendfile failed: %d", errno);
}
/* Close the sockets */
shutdown(sockfd, SHUT_RDWR);
shutdown(s, SHUT_RDWR);
if (TEST_RETURN != testcases[i].exp_retval) {
tst_resm(TFAIL, "sendfile(2) failed to return "
"expected value, expected: %d, "
"got: %ld", testcases[i].exp_retval,
TEST_RETURN);
kill(child_pid, SIGKILL);
} else if (offset != testcases[i].exp_updated_offset) {
tst_resm(TFAIL, "sendfile(2) failed to update "
"OFFSET parameter to expected value, "
"expected: %d, got: %" PRId64,
testcases[i].exp_updated_offset,
(int64_t) offset);
} else if (before_pos != after_pos) {
tst_resm(TFAIL, "sendfile(2) updated the file position "
" of in_fd unexpectedly, expected file position: %"
PRId64 ", " " actual file position %" PRId64,
(int64_t) before_pos, (int64_t) after_pos);
} else {
tst_resm(TPASS, "functionality of sendfile() is "
"correct");
wait_status = waitpid(-1, &wait_stat, 0);
}
close(in_fd);
}
/*
* do_child
*/
void do_child(void)
{
int lc;
socklen_t length;
char rbuf[4096];
for (lc = 0; TEST_LOOPING(lc); lc++) {
length = sizeof(sin1);
recvfrom(sockfd, rbuf, 4096, 0, (struct sockaddr *)&sin1,
&length);
}
exit(0);
}
/*
* setup() - performs all ONE TIME setup for this test.
*/
void setup(void)
{
int fd;
char buf[100];
tst_sig(FORK, DEF_HANDLER, cleanup);
TEST_PAUSE;
/* make a temporary directory and cd to it */
tst_tmpdir();
sprintf(in_file, "in.%d", getpid());
if ((fd = creat(in_file, 00700)) < 0) {
tst_brkm(TBROK, cleanup, "creat failed in setup, errno: %d",
errno);
}
sprintf(buf, "abcdefghijklmnopqrstuvwxyz");
if (write(fd, buf, strlen(buf)) < 0) {
tst_brkm(TBROK, cleanup, "write failed, errno: %d", errno);
}
close(fd);
sprintf(out_file, "out.%d", getpid());
}
/*
* cleanup() - performs all ONE TIME cleanup for this test at
* completion or premature exit.
*/
void cleanup(void)
{
close(out_fd);
/* delete the test directory created in setup() */
tst_rmdir();
}
int create_server(void)
{
static int count = 0;
socklen_t slen = sizeof(sin1);
sockfd = socket(PF_INET, SOCK_DGRAM, 0);
if (sockfd < 0) {
tst_brkm(TBROK, cleanup, "call to socket() failed: %s",
strerror(errno));
return -1;
}
sin1.sin_family = AF_INET;
sin1.sin_port = 0; /* pick random free port */
sin1.sin_addr.s_addr = INADDR_ANY;
count++;
if (bind(sockfd, (struct sockaddr *)&sin1, sizeof(sin1)) < 0) {
tst_brkm(TBROK, cleanup, "call to bind() failed: %s",
strerror(errno));
return -1;
}
SAFE_GETSOCKNAME(cleanup, sockfd, (struct sockaddr *)&sin1, &slen);
child_pid = FORK_OR_VFORK();
if (child_pid < 0) {
tst_brkm(TBROK, cleanup, "client/server fork failed: %s",
strerror(errno));
return -1;
}
if (!child_pid) { /* child */
#ifdef UCLINUX
if (self_exec(argv0, "") < 0) {
tst_brkm(TBROK, cleanup, "self_exec failed");
return -1;
}
#else
do_child();
#endif
}
s = socket(PF_INET, SOCK_DGRAM, 0);
inet_aton("127.0.0.1", &sin1.sin_addr);
if (s < 0) {
tst_brkm(TBROK, cleanup, "call to socket() failed: %s",
strerror(errno));
return -1;
}
SAFE_CONNECT(cleanup, s, (struct sockaddr *)&sin1, sizeof(sin1));
return s;
}
int main(int ac, char **av)
{
int i;
int lc;
tst_parse_opts(ac, av, NULL, NULL);
#ifdef UCLINUX
argv0 = av[0];
maybe_run_child(&do_child, "");
#endif
setup();
/*
* The following loop checks looping state if -c option given
*/
for (lc = 0; TEST_LOOPING(lc); lc++) {
tst_count = 0;
for (i = 0; i < TST_TOTAL; ++i) {
do_sendfile(testcases[i].offset, i);
}
}
cleanup();
tst_exit();
}