]> granicus.if.org Git - strace/blobdiff - tests/sendfile.c
tests: extend TEST_NETLINK_OBJECT macro
[strace] / tests / sendfile.c
index fd02000af347fcb6a1ef9b355f3388dcbc2ec043..b86565b8e5eebcf6cf29830f968c2a24f0d88e2b 100644 (file)
@@ -1,5 +1,6 @@
 /*
- * Copyright (c) 2015 Dmitry V. Levin <ldv@altlinux.org>
+ * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
+ * Copyright (c) 2015-2017 The strace developers.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  */
 
 #include "tests.h"
-#include <assert.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <stdio.h>
-#include <stdint.h>
-#include <unistd.h>
-#include <sys/mman.h>
-#include <sys/socket.h>
-#include <sys/stat.h>
-#include <sys/syscall.h>
+#include <asm/unistd.h>
 
 #ifdef __NR_sendfile
 
+# include <assert.h>
+# include <errno.h>
+# include <fcntl.h>
+# include <stdio.h>
+# include <stdint.h>
+# include <stdlib.h>
+# include <unistd.h>
+# include <sys/socket.h>
+
 int
-main(int ac, const char **av)
+main(void)
 {
-       assert(ac == 2);
-
        (void) close(0);
        if (open("/dev/zero", O_RDONLY) != 0)
-               return 77;
+               perror_msg_and_skip("open: %s", "/dev/zero");
 
        int sv[2];
        if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv))
-               return 77;
-
-       int reg_in = open(av[1], O_RDONLY);
+               perror_msg_and_skip("socketpair");
+
+       const unsigned int page_size = get_page_size();
+       assert(syscall(__NR_sendfile, 0, 1, NULL, page_size) == -1);
+       if (EBADF != errno)
+               perror_msg_and_skip("sendfile");
+       printf("sendfile(0, 1, NULL, %u) = -1 EBADF (%m)\n", page_size);
+
+       unsigned int file_size = 0;
+       socklen_t optlen = sizeof(file_size);
+       if (getsockopt(sv[1], SOL_SOCKET, SO_SNDBUF, &file_size, &optlen))
+               perror_msg_and_fail("getsockopt");
+       if (file_size < 1024)
+               error_msg_and_skip("SO_SNDBUF too small: %u", file_size);
+
+       file_size /= 4;
+       if (file_size / 16 > page_size)
+               file_size = page_size * 16;
+       const unsigned int blen = file_size / 3;
+       const unsigned int alen = file_size - blen;
+
+       static const char fname[] = "sendfile-tmpfile";
+       int reg_in = open(fname, O_RDWR | O_CREAT | O_TRUNC, 0600);
        if (reg_in < 0)
-               return 77;
-
-       struct stat stb;
-       if (fstat(reg_in, &stb))
-               return 77;
-       const size_t blen = stb.st_size / 3;
-       const size_t alen = stb.st_size - blen;
-       assert(S_ISREG(stb.st_mode) && blen > 0);
-
-       const size_t page_len = sysconf(_SC_PAGESIZE);
-       if (!syscall(__NR_sendfile, 0, 1, NULL, page_len) ||
-           EBADF != errno)
-               return 77;
-       printf("sendfile(0, 1, NULL, %lu) = -1 EBADF (Bad file descriptor)\n",
-              (unsigned long) page_len);
-
-       void *p = mmap(NULL, page_len * 2, PROT_READ | PROT_WRITE,
-                      MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
-       if (MAP_FAILED == p || munmap(p + page_len, page_len))
-               return 77;
-
-       if (!syscall(__NR_sendfile, 0, 1, p + page_len, page_len))
-               return 77;
-       printf("sendfile(0, 1, %#lx, %lu) = -1 EFAULT (Bad address)\n",
-              (unsigned long) p + page_len, (unsigned long) page_len);
-
-       if (syscall(__NR_sendfile, sv[1], reg_in, NULL, alen) != (long) alen)
-               return 77;
-       printf("sendfile(%d, %d, NULL, %lu) = %lu\n",
-              sv[1], reg_in, (unsigned long) alen,
-              (unsigned long) alen);
-
-       uint32_t *p_off = p + page_len - sizeof(uint32_t);
+               perror_msg_and_fail("open: %s", fname);
+       if (unlink(fname))
+               perror_msg_and_fail("unlink: %s", fname);
+       if (ftruncate(reg_in, file_size))
+               perror_msg_and_fail("ftruncate: %s", fname);
+
+       TAIL_ALLOC_OBJECT_VAR_PTR(uint32_t, p_off);
+       void *p = p_off + 1;
+       *p_off = 0;
+
+       assert(syscall(__NR_sendfile, 0, 1, p, page_size) == -1);
+       printf("sendfile(0, 1, %p, %u) = -1 EFAULT (%m)\n", p, page_size);
+
+       assert(syscall(__NR_sendfile, sv[1], reg_in, NULL, alen)
+              == (long) alen);
+       printf("sendfile(%d, %d, NULL, %u) = %u\n",
+              sv[1], reg_in, alen, alen);
+
+       p = p_off;
        if (syscall(__NR_sendfile, sv[1], reg_in, p_off, alen) != (long) alen) {
-               printf("sendfile(%d, %d, %#lx, %lu) = -1 EFAULT (Bad address)\n",
-                      sv[1], reg_in, (unsigned long) p_off,
-                      (unsigned long) alen);
+               printf("sendfile(%d, %d, %#lx, %u) = -1 EFAULT (%m)\n",
+                      sv[1], reg_in, (unsigned long) p_off, alen);
                --p_off;
-               if (syscall(__NR_sendfile, sv[1], reg_in, p_off, alen)
-                   != (long) alen)
-                       return 77;
+               *p_off = 0;
+               assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, alen)
+                      == (long) alen);
        }
-       printf("sendfile(%d, %d, [0] => [%lu], %lu) = %lu\n",
-              sv[1], reg_in, (unsigned long) alen,
-              (unsigned long) alen, (unsigned long) alen);
-
-       if (syscall(__NR_sendfile, sv[1], reg_in, p_off, stb.st_size + 1)
-           != (long) blen)
-               return 77;
-       printf("sendfile(%d, %d, [%lu] => [%lu], %lu) = %lu\n",
-              sv[1], reg_in, (unsigned long) alen,
-              (unsigned long) stb.st_size,
-              (unsigned long) stb.st_size + 1,
-              (unsigned long) blen);
-
-       if (p_off == p + page_len - sizeof(uint64_t)) {
+       printf("sendfile(%d, %d, [0] => [%u], %u) = %u\n",
+              sv[1], reg_in, alen, alen, alen);
+
+       assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, file_size + 1)
+              == (long) blen);
+       printf("sendfile(%d, %d, [%u] => [%u], %u) = %u\n",
+              sv[1], reg_in, alen, file_size, file_size + 1, blen);
+
+       if (p_off != p) {
                uint64_t *p_off64 = (uint64_t *) p_off;
-               *p_off64 = 0xcafef00dfacefeed;
-               if (!syscall(__NR_sendfile, sv[1], reg_in, p_off64, 1))
-                       return 77;
-               printf("sendfile(%d, %d, [14627392582579060461], 1) = -1 EINVAL (Invalid argument)\n",
-                      sv[1], reg_in);
+               *p_off64 = 0xcafef00dfacefeedULL;
+               assert(syscall(__NR_sendfile, sv[1], reg_in, p_off64, 1) == -1);
+               printf("sendfile(%d, %d, [14627392582579060461], 1)"
+                      " = -1 EINVAL (%m)\n", sv[1], reg_in);
                *p_off64 = 0xdefaced;
        } else {
                *p_off = 0xdefaced;
        }
-       if (syscall(__NR_sendfile, sv[1], reg_in, p_off, 1))
-               return 77;
+       assert(syscall(__NR_sendfile, sv[1], reg_in, p_off, 1) == 0);
        printf("sendfile(%d, %d, [233811181], 1) = 0\n",
               sv[1], reg_in);
 
@@ -131,10 +128,6 @@ main(int ac, const char **av)
 
 #else
 
-int
-main(void)
-{
-       return 77;
-}
+SKIP_MAIN_UNDEFINED("__NR_sendfile")
 
 #endif