1 /*
2 * Copyright (c) International Business Machines Corp., 2001
3 * Copyright (c) Red Hat Inc., 2007
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 Foundation,
17 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 */
19
20 /*
21 * DESCRIPTION
22 * Testcase to test that sendfile(2) system call updates file
23 * position of in_fd correctly when passing NULL as offset.
24 *
25 * HISTORY
26 * 11/2007 Copyed from sendfile02.c by Masatake YAMATO
27 */
28
29 #include <inttypes.h>
30 #include <stdio.h>
31 #include <errno.h>
32 #include <fcntl.h>
33 #include <sys/stat.h>
34 #include <sys/sendfile.h>
35 #include <sys/types.h>
36 #include <sys/wait.h>
37 #include <sys/socket.h>
38 #include <sys/mman.h>
39 #include <netinet/in.h>
40 #include <arpa/inet.h>
41 #include <string.h>
42 #include "test.h"
43 #include "safe_macros.h"
44
45 TCID_DEFINE(sendfile06);
46
47 #define IN_FILE "infile"
48 #define OUT_FILE "outfile"
49
50 static pid_t child_pid;
51 static int sockfd;
52 static struct sockaddr_in sin1;
53 static struct stat sb;
54
55 static void cleanup(void);
56 static void do_child(void);
57 static void setup(void);
58 static int create_server(void);
59
60 int TST_TOTAL = 1;
61
62 #ifdef UCLINUX
63 static char *argv0;
64 #endif
65
do_sendfile(void)66 static void do_sendfile(void)
67 {
68 int in_fd, out_fd;
69 off_t after_pos;
70 int wait_stat;
71
72 out_fd = create_server();
73
74 in_fd = SAFE_OPEN(cleanup, IN_FILE, O_RDONLY);
75
76 TEST(sendfile(out_fd, in_fd, NULL, sb.st_size));
77 if ((after_pos = lseek(in_fd, 0, SEEK_CUR)) < 0) {
78 tst_brkm(TBROK, cleanup,
79 "lseek after invoking sendfile failed: %d", errno);
80 }
81
82 /* Close the sockets */
83 shutdown(sockfd, SHUT_RDWR);
84 shutdown(out_fd, SHUT_RDWR);
85 if (TEST_RETURN != sb.st_size) {
86 tst_resm(TFAIL, "sendfile(2) failed to return "
87 "expected value, expected: %" PRId64 ", "
88 "got: %ld", (int64_t) sb.st_size, TEST_RETURN);
89 SAFE_KILL(cleanup, child_pid, SIGKILL);
90 } else if (after_pos != sb.st_size) {
91 tst_resm(TFAIL, "sendfile(2) failed to update "
92 " the file position of in_fd, "
93 "expected file position: %" PRId64 ", "
94 "actual file position %" PRId64,
95 (int64_t) sb.st_size, (int64_t) after_pos);
96 SAFE_KILL(cleanup, child_pid, SIGKILL);
97 } else {
98 tst_resm(TPASS, "functionality of sendfile() is "
99 "correct");
100 waitpid(-1, &wait_stat, 0);
101 }
102
103 SAFE_CLOSE(cleanup, in_fd);
104 SAFE_CLOSE(cleanup, out_fd);
105 SAFE_CLOSE(cleanup, sockfd);
106 }
107
do_child(void)108 static void do_child(void)
109 {
110 socklen_t length = sizeof(sin1);
111 char rbuf[4096];
112 ssize_t ret, bytes_total_received = 0;
113
114 while (bytes_total_received < sb.st_size) {
115 ret = recvfrom(sockfd, rbuf, 4096, 0, (struct sockaddr *)&sin1,
116 &length);
117 if (ret < 0) {
118 fprintf(stderr, "child process recvfrom failed: %s\n",
119 strerror(errno));
120 exit(1);
121 }
122 bytes_total_received += ret;
123 }
124
125 exit(0);
126 }
127
setup(void)128 static void setup(void)
129 {
130 int fd;
131
132 tst_sig(FORK, DEF_HANDLER, cleanup);
133
134 TEST_PAUSE;
135
136 tst_tmpdir();
137
138 fd = SAFE_CREAT(cleanup, IN_FILE, 0600);
139
140 SAFE_WRITE(cleanup, 1, fd, "abcdefghijklmnopqrstuvwxyz", 26);
141
142 SAFE_FSTAT(cleanup, fd, &sb);
143
144 SAFE_CLOSE(cleanup, fd);
145 }
146
cleanup(void)147 static void cleanup(void)
148 {
149 tst_rmdir();
150 }
151
create_server(void)152 static int create_server(void)
153 {
154 int s;
155 socklen_t slen = sizeof(sin1);
156
157 sockfd = socket(PF_INET, SOCK_DGRAM, 0);
158 if (sockfd < 0) {
159 tst_brkm(TBROK, cleanup, "call to socket() failed: %s",
160 strerror(errno));
161 return -1;
162 }
163 sin1.sin_family = AF_INET;
164 sin1.sin_port = 0; /* pick random free port */
165 sin1.sin_addr.s_addr = INADDR_ANY;
166
167 if (bind(sockfd, (struct sockaddr *)&sin1, sizeof(sin1)) < 0) {
168 tst_brkm(TBROK, cleanup, "call to bind() failed: %s",
169 strerror(errno));
170 return -1;
171 }
172 if (getsockname(sockfd, (struct sockaddr *)&sin1, &slen) == -1)
173 tst_brkm(TBROK | TERRNO, cleanup, "getsockname failed");
174
175 child_pid = FORK_OR_VFORK();
176 if (child_pid < 0) {
177 tst_brkm(TBROK, cleanup, "client/server fork failed: %s",
178 strerror(errno));
179 return -1;
180 }
181
182 if (!child_pid) {
183 #ifdef UCLINUX
184 if (self_exec(argv0, "") < 0) {
185 tst_brkm(TBROK, cleanup, "self_exec failed");
186 return -1;
187
188 }
189 #else
190 do_child();
191 #endif
192 }
193
194 s = socket(PF_INET, SOCK_DGRAM, 0);
195 inet_aton("127.0.0.1", &sin1.sin_addr);
196
197 if (s < 0) {
198 tst_brkm(TBROK, cleanup, "call to socket() failed: %s",
199 strerror(errno));
200 return -1;
201 }
202
203 if (connect(s, (struct sockaddr *)&sin1, sizeof(sin1)) < 0) {
204 tst_brkm(TBROK, cleanup, "call to connect() failed: %s",
205 strerror(errno));
206 }
207
208 return s;
209 }
210
main(int ac,char ** av)211 int main(int ac, char **av)
212 {
213 int lc;
214
215 tst_parse_opts(ac, av, NULL, NULL);
216
217 #ifdef UCLINUX
218 argv0 = av[0];
219 maybe_run_child(&do_child, "");
220 #endif
221
222 setup();
223
224 for (lc = 0; TEST_LOOPING(lc); lc++) {
225 tst_count = 0;
226
227 do_sendfile();
228 }
229
230 cleanup();
231 tst_exit();
232 }
233