Home | History | Annotate | Download | only in tests
      1 /*
      2  * Copyright (c) 2015-2016 Dmitry V. Levin <ldv (at) altlinux.org>
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  * 1. Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  * 2. Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in the
     12  *    documentation and/or other materials provided with the distribution.
     13  * 3. The name of the author may not be used to endorse or promote products
     14  *    derived from this software without specific prior written permission.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 
     28 #include "tests.h"
     29 #include <asm/unistd.h>
     30 
     31 #ifdef __NR_sendfile64
     32 
     33 # include <assert.h>
     34 # include <errno.h>
     35 # include <fcntl.h>
     36 # include <stdio.h>
     37 # include <stdint.h>
     38 # include <unistd.h>
     39 # include <sys/socket.h>
     40 # include <sys/stat.h>
     41 
     42 int
     43 main(int ac, const char **av)
     44 {
     45 	assert(ac == 1);
     46 
     47 	(void) close(0);
     48 	if (open("/dev/zero", O_RDONLY) != 0)
     49 		perror_msg_and_skip("open: %s", "/dev/zero");
     50 
     51 	int sv[2];
     52 	if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv))
     53 		perror_msg_and_skip("socketpair");
     54 
     55 	int reg_in = open(av[0], O_RDONLY);
     56 	if (reg_in < 0)
     57 		perror_msg_and_fail("open: %s", av[0]);
     58 
     59 	struct stat stb;
     60 	assert(fstat(reg_in, &stb) == 0);
     61 	const size_t blen = stb.st_size / 3;
     62 	const size_t alen = stb.st_size - blen;
     63 	assert(S_ISREG(stb.st_mode) && blen > 0);
     64 
     65 	const size_t page_len = get_page_size();
     66 	assert(syscall(__NR_sendfile64, 0, 1, NULL, page_len) == -1);
     67 	if (EBADF != errno)
     68 		perror_msg_and_skip("sendfile64");
     69 	printf("sendfile64(0, 1, NULL, %lu) = -1 EBADF (%m)\n",
     70 	       (unsigned long) page_len);
     71 
     72 	uint64_t *p_off = tail_alloc(sizeof(uint64_t));
     73 	void *p = p_off + 1;
     74 	*p_off = 0;
     75 
     76 	assert(syscall(__NR_sendfile64, 0, 1, p, page_len) == -1);
     77 	printf("sendfile64(0, 1, %#lx, %lu) = -1 EFAULT (%m)\n",
     78 	       (unsigned long) p, (unsigned long) page_len);
     79 
     80 	assert(syscall(__NR_sendfile64, sv[1], reg_in, NULL, alen)
     81 	       == (long) alen);
     82 	printf("sendfile64(%d, %d, NULL, %lu) = %lu\n",
     83 	       sv[1], reg_in, (unsigned long) alen,
     84 	       (unsigned long) alen);
     85 
     86 	assert(syscall(__NR_sendfile64, sv[1], reg_in, p_off, alen)
     87 	       == (long) alen);
     88 	printf("sendfile64(%d, %d, [0] => [%lu], %lu) = %lu\n",
     89 	       sv[1], reg_in, (unsigned long) alen,
     90 	       (unsigned long) alen, (unsigned long) alen);
     91 
     92 	assert(syscall(__NR_sendfile64, sv[1], reg_in, p_off, stb.st_size + 1)
     93 	       == (long) blen);
     94 	printf("sendfile64(%d, %d, [%lu] => [%lu], %lu) = %lu\n",
     95 	       sv[1], reg_in, (unsigned long) alen,
     96 	       (unsigned long) stb.st_size,
     97 	       (unsigned long) stb.st_size + 1,
     98 	       (unsigned long) blen);
     99 
    100 	*p_off = 0xcafef00dfacefeedULL;
    101 	assert(syscall(__NR_sendfile64, sv[1], reg_in, p_off, 1) == -1);
    102 	printf("sendfile64(%d, %d, [14627392582579060461], 1)"
    103 		" = -1 EINVAL (%m)\n", sv[1], reg_in);
    104 
    105 	*p_off = 0xfacefeed;
    106 	assert(syscall(__NR_sendfile64, sv[1], reg_in, p_off, 1) == 0);
    107 	printf("sendfile64(%d, %d, [4207869677], 1) = 0\n", sv[1], reg_in);
    108 
    109 	puts("+++ exited with 0 +++");
    110 	return 0;
    111 }
    112 
    113 #else
    114 
    115 SKIP_MAIN_UNDEFINED("__NR_sendfile64")
    116 
    117 #endif
    118