18a272653SPeter Holm#!/bin/sh 28a272653SPeter Holm 38a272653SPeter Holm# 48a272653SPeter Holm# Copyright (c) 2011 Peter Holm <pho@FreeBSD.org> 58a272653SPeter Holm# All rights reserved. 68a272653SPeter Holm# 78a272653SPeter Holm# Redistribution and use in source and binary forms, with or without 88a272653SPeter Holm# modification, are permitted provided that the following conditions 98a272653SPeter Holm# are met: 108a272653SPeter Holm# 1. Redistributions of source code must retain the above copyright 118a272653SPeter Holm# notice, this list of conditions and the following disclaimer. 128a272653SPeter Holm# 2. Redistributions in binary form must reproduce the above copyright 138a272653SPeter Holm# notice, this list of conditions and the following disclaimer in the 148a272653SPeter Holm# documentation and/or other materials provided with the distribution. 158a272653SPeter Holm# 168a272653SPeter Holm# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 178a272653SPeter Holm# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 188a272653SPeter Holm# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 198a272653SPeter Holm# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 208a272653SPeter Holm# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 218a272653SPeter Holm# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 228a272653SPeter Holm# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 238a272653SPeter Holm# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 248a272653SPeter Holm# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 258a272653SPeter Holm# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 268a272653SPeter Holm# SUCH DAMAGE. 278a272653SPeter Holm# 288a272653SPeter Holm 298a272653SPeter Holm# sendfile(2) test by kib@ 308a272653SPeter Holm# Both processes will sit in the sbwait state forever: 318a272653SPeter Holm 328a272653SPeter Holm. ../default.cfg 338a272653SPeter Holm 348a272653SPeter Holmhere=`pwd` 358a272653SPeter Holmcd /tmp 368a272653SPeter Holmsed '1,/^EOF/d' < $here/$0 > sendfile4.c 378a272653SPeter Holmmycc -o sendfile4 -Wall -Wextra -O2 sendfile4.c || exit 388a272653SPeter Holmrm -f sendfile4.c 398a272653SPeter Holmcd $here 408a272653SPeter Holm 418a272653SPeter Holmdd if=/dev/zero of=$diskimage bs=1m count=6 status=none 428a272653SPeter Holm/tmp/sendfile4 $diskimage || echo FAIL 438a272653SPeter Holm 448a272653SPeter Holmrm -f /tmp/sendfile4 $diskimage 458a272653SPeter Holmexit 468a272653SPeter HolmEOF 478a272653SPeter Holm#include <sys/types.h> 488a272653SPeter Holm#include <sys/fcntl.h> 498a272653SPeter Holm#include <sys/socket.h> 508a272653SPeter Holm#include <sys/stat.h> 518a272653SPeter Holm#include <sys/uio.h> 528a272653SPeter Holm#include <sys/wait.h> 538a272653SPeter Holm#include <err.h> 548a272653SPeter Holm#include <errno.h> 558a272653SPeter Holm#include <signal.h> 568a272653SPeter Holm#include <stdlib.h> 578a272653SPeter Holm#include <unistd.h> 588a272653SPeter Holm 598a272653SPeter Holmstatic void 608a272653SPeter Holmhandler(int i __unused) { 618a272653SPeter Holm _exit(1); 628a272653SPeter Holm} 638a272653SPeter Holm 648a272653SPeter Holmint 658a272653SPeter Holmmain(int argc, char *argv[]) 668a272653SPeter Holm{ 678a272653SPeter Holm const char *from_name; 688a272653SPeter Holm char *buf; 698a272653SPeter Holm int sv[2]; 708a272653SPeter Holm struct stat st; 718a272653SPeter Holm off_t written, pos; 728a272653SPeter Holm int child, error, from, status; 738a272653SPeter Holm 748a272653SPeter Holm if (argc != 2) 758a272653SPeter Holm errx(1, "Usage: %s from", argv[0]); 768a272653SPeter Holm from_name = argv[1]; 778a272653SPeter Holm 788a272653SPeter Holm from = open(from_name, O_RDONLY); 798a272653SPeter Holm if (from == -1) 808a272653SPeter Holm err(1, "open read %s", from_name); 818a272653SPeter Holm 828a272653SPeter Holm error = fstat(from, &st); 838a272653SPeter Holm if (error == -1) 848a272653SPeter Holm err(1, "stat %s", from_name); 858a272653SPeter Holm 868a272653SPeter Holm error = socketpair(AF_UNIX, SOCK_STREAM, 0, sv); 878a272653SPeter Holm if (error == -1) 888a272653SPeter Holm err(1, "socketpair"); 898a272653SPeter Holm 908a272653SPeter Holm signal(SIGALRM, handler); 918a272653SPeter Holm alarm(120); 928a272653SPeter Holm child = fork(); 938a272653SPeter Holm if (child == -1) 948a272653SPeter Holm err(1, "fork"); 958a272653SPeter Holm else if (child != 0) { 968a272653SPeter Holm close(sv[1]); 978a272653SPeter Holm pos = 0; 988a272653SPeter Holm for (;;) { 998a272653SPeter Holm error = sendfile(from, sv[0], pos, st.st_size - pos, 1008a272653SPeter Holm NULL, &written, 0); 1018a272653SPeter Holm if (error == -1) { 1028a272653SPeter Holm if (errno != EAGAIN) 1038a272653SPeter Holm err(1, "sendfile"); 1048a272653SPeter Holm } 1058a272653SPeter Holm pos += written; 1068a272653SPeter Holm if (pos == st.st_size) 1078a272653SPeter Holm break; 1088a272653SPeter Holm } 1098a272653SPeter Holm close(sv[0]); 1108a272653SPeter Holm waitpid(child, &status, 0); 1118a272653SPeter Holm } else { 1128a272653SPeter Holm close(sv[0]); 1138a272653SPeter Holm buf = malloc(st.st_size); 1148a272653SPeter Holm if (buf == NULL) 1158a272653SPeter Holm err(1, "malloc %jd", st.st_size); 1168a272653SPeter Holm pos = 0; 1178a272653SPeter Holm for (;;) { 1188a272653SPeter Holm written = 413; 1198a272653SPeter Holm if (written > st.st_size - pos) 1208a272653SPeter Holm written = st.st_size - pos; 1218a272653SPeter Holm#if 0 1228a272653SPeter Holm written = st.st_size - pos; 1238a272653SPeter Holm if (written > 1000) 1248a272653SPeter Holm written = 1000; 1258a272653SPeter Holm written = arc4random_uniform(written) + 1; 1268a272653SPeter Holm#endif 1278a272653SPeter Holm error = read(sv[1], buf + pos, written); 1288a272653SPeter Holm if (error == -1) 1298a272653SPeter Holm err(1, "read"); 1308a272653SPeter Holm else if (error == 0) 1318a272653SPeter Holm errx(1, "short read"); 1328a272653SPeter Holm pos += error; 1338a272653SPeter Holm if (pos == st.st_size) 1348a272653SPeter Holm break; 1358a272653SPeter Holm } 1368a272653SPeter Holm close(sv[1]); 1378a272653SPeter Holm _exit(0); 1388a272653SPeter Holm } 1398a272653SPeter Holm 1408a272653SPeter Holm return (0); 1418a272653SPeter Holm} 142