1  /*
       2   * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@strace.io>
       3   * Copyright (c) 2015-2021 The strace developers.
       4   * All rights reserved.
       5   *
       6   * SPDX-License-Identifier: GPL-2.0-or-later
       7   */
       8  
       9  #include "tests.h"
      10  #include "scno.h"
      11  
      12  #ifdef __NR_sendfile
      13  
      14  # include <assert.h>
      15  # include <errno.h>
      16  # include <fcntl.h>
      17  # include <stdio.h>
      18  # include <stdint.h>
      19  # include <stdlib.h>
      20  # include <unistd.h>
      21  # include <sys/socket.h>
      22  
      23  int
      24  main(void)
      25  {
      26  	(void) close(0);
      27  	if (open("/dev/zero", O_RDONLY) != 0)
      28  		perror_msg_and_skip("open: %s", "/dev/zero");
      29  
      30  	int sv[2];
      31  	if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv))
      32  		perror_msg_and_skip("socketpair");
      33  
      34  	const unsigned int page_size = get_page_size();
      35  	assert(syscall(__NR_sendfile, 0, 1, NULL, page_size) == -1);
      36  	if (EBADF != errno)
      37  		perror_msg_and_skip("sendfile");
      38  	printf("sendfile(0, 1, NULL, %u) = -1 EBADF (%m)\n", page_size);
      39  
      40  	unsigned int file_size = 0;
      41  	socklen_t optlen = sizeof(file_size);
      42  	if (getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &file_size, &optlen))
      43  		perror_msg_and_fail("getsockopt");
      44  	if (file_size < 1024)
      45  		error_msg_and_skip("SO_SNDBUF too small: %u", file_size);
      46  
      47  	file_size /= 4;
      48  	if (file_size / 16 > page_size)
      49  		file_size = page_size * 16;
      50  	const unsigned int blen = file_size / 3;
      51  	const unsigned int alen = file_size - blen;
      52  
      53  	int reg_in = create_tmpfile(O_RDWR);
      54  	if (ftruncate(reg_in, file_size))
      55  		perror_msg_and_fail("ftruncate(%d, %u)", reg_in, file_size);
      56  
      57  	TAIL_ALLOC_OBJECT_CONST_PTR(kernel_ulong_t, p_off);
      58  	void *const efault = p_off + 1;
      59  	*p_off = 0;
      60  
      61  	assert(syscall(__NR_sendfile, 0, 1, efault, page_size) == -1);
      62  	printf("sendfile(0, 1, %p, %u) = -1 EFAULT (%m)\n", efault, page_size);
      63  
      64  	assert(syscall(__NR_sendfile, sv[1], reg_in, NULL, alen)
      65  	       == (long) alen);
      66  	printf("sendfile(%d, %d, NULL, %u) = %u\n",
      67  	       sv[1], reg_in, alen, alen);
      68  
      69  	assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, alen)
      70  	       == (long) alen);
      71  	printf("sendfile(%d, %d, [0] => [%u], %u) = %u\n",
      72  	       sv[1], reg_in, alen, alen, alen);
      73  
      74  	assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, file_size + 1)
      75  	       == (long) blen);
      76  	printf("sendfile(%d, %d, [%u] => [%u], %u) = %u\n",
      77  	       sv[1], reg_in, alen, file_size, file_size + 1, blen);
      78  
      79  	*p_off = (typeof(*p_off)) 0xcafef00dfacefeedULL;
      80  	assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, 1) == -1);
      81  	printf("sendfile(%d, %d, [%llu], 1) = -1 EINVAL (%m)\n",
      82  	       sv[1], reg_in, (unsigned long long) *p_off);
      83  
      84  	*p_off = 0xdefaced;
      85  	assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, 1) == 0);
      86  	printf("sendfile(%d, %d, [%u], 1) = 0\n",
      87  	       sv[1], reg_in, 0xdefaced);
      88  
      89  	puts("+++ exited with 0 +++");
      90  	return 0;
      91  }
      92  
      93  #else
      94  
      95  SKIP_MAIN_UNDEFINED("__NR_sendfile")
      96  
      97  #endif