| /* |
| * |
| * 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(); |
| } |