1 /* 2 * 3 * Copyright (c) International Business Machines Corp., 2001 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 * sendfile02.c 23 * 24 * DESCRIPTION 25 * Testcase to test the basic functionality of the sendfile(2) system call. 26 * 27 * ALGORITHM 28 * 1. call sendfile(2) with offset = 0 29 * 2. call sendfile(2) with offset in the middle of the file 30 * 31 * USAGE: <for command-line> 32 * sendfile02 [-c n] [-f] [-i n] [-I x] [-P x] [-t] 33 * where, 34 * -f : Turn off functionality Testing. 35 * -i n : Execute test n times. 36 * -I x : Execute test for x seconds. 37 * -P x : Pause for x seconds between iterations. 38 * -t : Turn on syscall timing. 39 * 40 * HISTORY 41 * 07/2001 Ported by Wayne Boyer 42 * 08/2002 Make it use a socket so it works with 2.5 kernel 43 * 44 * RESTRICTIONS 45 * NONE 46 */ 47 #include <stdio.h> 48 #include <errno.h> 49 #include <fcntl.h> 50 #include <sys/stat.h> 51 #include <sys/sendfile.h> 52 #include <sys/types.h> 53 #include <sys/wait.h> 54 #include <sys/socket.h> 55 #include <netinet/in.h> 56 #include <arpa/inet.h> 57 #include <unistd.h> 58 #include <inttypes.h> 59 #include "test.h" 60 #include "safe_macros.h" 61 62 #ifndef OFF_T 63 #define OFF_T off_t 64 #endif /* Not def: OFF_T */ 65 66 TCID_DEFINE(sendfile02); 67 int TST_TOTAL = 4; 68 69 char in_file[100]; 70 char out_file[100]; 71 int out_fd; 72 pid_t child_pid; 73 static int sockfd, s; 74 static struct sockaddr_in sin1; /* shared between do_child and create_server */ 75 76 void cleanup(void); 77 void do_child(void); 78 void setup(void); 79 int create_server(void); 80 81 struct test_case_t { 82 char *desc; 83 int offset; 84 int exp_retval; 85 int exp_updated_offset; 86 } testcases[] = { 87 { 88 "Test sendfile(2) with offset = 0", 0, 26, 26}, { 89 "Test sendfile(2) with offset in the middle of file", 2, 24, 26}, { 90 "Test sendfile(2) with offset in the middle of file", 4, 22, 26}, { 91 "Test sendfile(2) with offset in the middle of file", 6, 20, 26} 92 }; 93 94 #ifdef UCLINUX 95 static char *argv0; 96 #endif 97 98 void do_sendfile(OFF_T offset, int i) 99 { 100 int in_fd; 101 struct stat sb; 102 int wait_status; 103 int wait_stat; 104 off_t before_pos, after_pos; 105 106 out_fd = create_server(); 107 108 if ((in_fd = open(in_file, O_RDONLY)) < 0) { 109 tst_brkm(TBROK, cleanup, "open failed: %d", errno); 110 } 111 SAFE_STAT(cleanup, in_file, &sb); 112 113 if ((before_pos = lseek(in_fd, 0, SEEK_CUR)) < 0) { 114 tst_brkm(TBROK, cleanup, 115 "lseek before invoking sendfile failed: %d", errno); 116 } 117 118 TEST(sendfile(out_fd, in_fd, &offset, sb.st_size - offset)); 119 120 if ((after_pos = lseek(in_fd, 0, SEEK_CUR)) < 0) { 121 tst_brkm(TBROK, cleanup, 122 "lseek after invoking sendfile failed: %d", errno); 123 } 124 125 /* Close the sockets */ 126 shutdown(sockfd, SHUT_RDWR); 127 shutdown(s, SHUT_RDWR); 128 if (TEST_RETURN != testcases[i].exp_retval) { 129 tst_resm(TFAIL, "sendfile(2) failed to return " 130 "expected value, expected: %d, " 131 "got: %ld", testcases[i].exp_retval, 132 TEST_RETURN); 133 kill(child_pid, SIGKILL); 134 } else if (offset != testcases[i].exp_updated_offset) { 135 tst_resm(TFAIL, "sendfile(2) failed to update " 136 "OFFSET parameter to expected value, " 137 "expected: %d, got: %" PRId64, 138 testcases[i].exp_updated_offset, 139 (int64_t) offset); 140 } else if (before_pos != after_pos) { 141 tst_resm(TFAIL, "sendfile(2) updated the file position " 142 " of in_fd unexpectedly, expected file position: %" 143 PRId64 ", " " actual file position %" PRId64, 144 (int64_t) before_pos, (int64_t) after_pos); 145 } else { 146 tst_resm(TPASS, "functionality of sendfile() is " 147 "correct"); 148 wait_status = waitpid(-1, &wait_stat, 0); 149 } 150 151 close(in_fd); 152 } 153 154 /* 155 * do_child 156 */ 157 void do_child(void) 158 { 159 int lc; 160 socklen_t length; 161 char rbuf[4096]; 162 163 for (lc = 0; TEST_LOOPING(lc); lc++) { 164 length = sizeof(sin1); 165 recvfrom(sockfd, rbuf, 4096, 0, (struct sockaddr *)&sin1, 166 &length); 167 } 168 exit(0); 169 } 170 171 /* 172 * setup() - performs all ONE TIME setup for this test. 173 */ 174 void setup(void) 175 { 176 int fd; 177 char buf[100]; 178 179 tst_sig(FORK, DEF_HANDLER, cleanup); 180 181 TEST_PAUSE; 182 183 /* make a temporary directory and cd to it */ 184 tst_tmpdir(); 185 sprintf(in_file, "in.%d", getpid()); 186 if ((fd = creat(in_file, 00700)) < 0) { 187 tst_brkm(TBROK, cleanup, "creat failed in setup, errno: %d", 188 errno); 189 } 190 sprintf(buf, "abcdefghijklmnopqrstuvwxyz"); 191 if (write(fd, buf, strlen(buf)) < 0) { 192 tst_brkm(TBROK, cleanup, "write failed, errno: %d", errno); 193 } 194 close(fd); 195 sprintf(out_file, "out.%d", getpid()); 196 } 197 198 /* 199 * cleanup() - performs all ONE TIME cleanup for this test at 200 * completion or premature exit. 201 */ 202 void cleanup(void) 203 { 204 205 close(out_fd); 206 /* delete the test directory created in setup() */ 207 tst_rmdir(); 208 209 } 210 211 int create_server(void) 212 { 213 static int count = 0; 214 socklen_t slen = sizeof(sin1); 215 216 sockfd = socket(PF_INET, SOCK_DGRAM, 0); 217 if (sockfd < 0) { 218 tst_brkm(TBROK, cleanup, "call to socket() failed: %s", 219 strerror(errno)); 220 return -1; 221 } 222 sin1.sin_family = AF_INET; 223 sin1.sin_port = 0; /* pick random free port */ 224 sin1.sin_addr.s_addr = INADDR_ANY; 225 count++; 226 if (bind(sockfd, (struct sockaddr *)&sin1, sizeof(sin1)) < 0) { 227 tst_brkm(TBROK, cleanup, "call to bind() failed: %s", 228 strerror(errno)); 229 return -1; 230 } 231 SAFE_GETSOCKNAME(cleanup, sockfd, (struct sockaddr *)&sin1, &slen); 232 233 child_pid = FORK_OR_VFORK(); 234 if (child_pid < 0) { 235 tst_brkm(TBROK, cleanup, "client/server fork failed: %s", 236 strerror(errno)); 237 return -1; 238 } 239 if (!child_pid) { /* child */ 240 #ifdef UCLINUX 241 if (self_exec(argv0, "") < 0) { 242 tst_brkm(TBROK, cleanup, "self_exec failed"); 243 return -1; 244 245 } 246 #else 247 do_child(); 248 #endif 249 } 250 251 s = socket(PF_INET, SOCK_DGRAM, 0); 252 inet_aton("127.0.0.1", &sin1.sin_addr); 253 if (s < 0) { 254 tst_brkm(TBROK, cleanup, "call to socket() failed: %s", 255 strerror(errno)); 256 return -1; 257 } 258 SAFE_CONNECT(cleanup, s, (struct sockaddr *)&sin1, sizeof(sin1)); 259 return s; 260 261 } 262 263 int main(int ac, char **av) 264 { 265 int i; 266 int lc; 267 268 tst_parse_opts(ac, av, NULL, NULL); 269 #ifdef UCLINUX 270 argv0 = av[0]; 271 maybe_run_child(&do_child, ""); 272 #endif 273 274 setup(); 275 276 /* 277 * The following loop checks looping state if -c option given 278 */ 279 for (lc = 0; TEST_LOOPING(lc); lc++) { 280 tst_count = 0; 281 282 for (i = 0; i < TST_TOTAL; ++i) { 283 do_sendfile(testcases[i].offset, i); 284 } 285 } 286 cleanup(); 287 288 tst_exit(); 289 } 290