1 /*
2  * This file is part of vmsplice strace test.
3  *
4  * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org>
5  * Copyright (c) 2016-2017 The strace developers.
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. The name of the author may not be used to endorse or promote products
17  *    derived from this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30 
31 #include "tests.h"
32 #include <asm/unistd.h>
33 
34 #if defined __NR_vmsplice
35 
36 # include <assert.h>
37 # include <stdio.h>
38 # include <unistd.h>
39 # include <sys/uio.h>
40 
41 int
main(void)42 main(void)
43 {
44 	tprintf("%s", "");
45 
46 	int fds[2];
47 	if (pipe(fds))
48 		perror_msg_and_fail("pipe");
49 	assert(0 == fds[0]);
50 	assert(1 == fds[1]);
51 
52 	static const char w0_c[] = "012";
53 	const char *w0_d = hexdump_strdup(w0_c);
54 	void *w0 = tail_memdup(w0_c, LENGTH_OF(w0_c));
55 
56 	static const char w1_c[] = "34567";
57 	const char *w1_d = hexdump_strdup(w1_c);
58 	void *w1 = tail_memdup(w1_c, LENGTH_OF(w1_c));
59 
60 	static const char w2_c[] = "89abcde";
61 	const char *w2_d = hexdump_strdup(w2_c);
62 	void *w2 = tail_memdup(w2_c, LENGTH_OF(w2_c));
63 
64 	const struct iovec iov_[] = {
65 		{
66 			.iov_base = w0,
67 			.iov_len = LENGTH_OF(w0_c)
68 		}, {
69 			.iov_base = w1,
70 			.iov_len = LENGTH_OF(w1_c)
71 		}, {
72 			.iov_base = w2,
73 			.iov_len = LENGTH_OF(w2_c)
74 		}
75 	};
76 	const struct iovec *iov = tail_memdup(iov_, sizeof(iov_));
77 	const unsigned int len =
78 		LENGTH_OF(w0_c) + LENGTH_OF(w1_c) + LENGTH_OF(w2_c);
79 
80 	tprintf("vmsplice(1, [{iov_base=\"%s\", iov_len=%u}"
81 		", {iov_base=\"%s\", iov_len=%u}"
82 		", {iov_base=\"%s\", iov_len=%u}], %u, %s) = %u\n"
83 		" * %u bytes in buffer 0\n"
84 		" | 00000 %-49s  %-16s |\n"
85 		" * %u bytes in buffer 1\n"
86 		" | 00000 %-49s  %-16s |\n"
87 		" * %u bytes in buffer 2\n"
88 		" | 00000 %-49s  %-16s |\n",
89 		w0_c, LENGTH_OF(w0_c), w1_c, LENGTH_OF(w1_c),
90 		w2_c, LENGTH_OF(w2_c), (unsigned int) ARRAY_SIZE(iov_),
91 		"SPLICE_F_NONBLOCK", len,
92 		LENGTH_OF(w0_c), w0_d, w0_c,
93 		LENGTH_OF(w1_c), w1_d, w1_c, LENGTH_OF(w2_c), w2_d, w2_c);
94 
95 	const long rc = syscall(__NR_vmsplice, 1, iov, ARRAY_SIZE(iov_), 2);
96 	if (rc < 0)
97 		perror_msg_and_skip("vmsplice");
98 	assert(rc == (int) len);
99 
100 	tprintf("+++ exited with 0 +++\n");
101 	return 0;
102 }
103 
104 #else
105 
106 SKIP_MAIN_UNDEFINED("__NR_vmsplice")
107 
108 #endif
109