File: close_range.c

package info (click to toggle)
strace 5.10-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 49,332 kB
  • sloc: ansic: 113,177; sh: 8,831; makefile: 3,108; awk: 364; perl: 267; sed: 9
file content (147 lines) | stat: -rw-r--r-- 3,523 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
/*
 * Check decoding of close_range syscall.
 *
 * Copyright (c) 2020 The strace developers.
 * All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "tests.h"
#include "scno.h"

#ifdef __NR_close_range

# include <fcntl.h>
# include <stdio.h>
# include <unistd.h>

# ifdef HAVE_LINUX_CLOSE_RANGE_H
#  include <linux/close_range.h>
# else
#  define CLOSE_RANGE_UNSHARE	(1U << 1)
# endif

# ifndef FD0_PATH
#  define FD0_PATH ""
# endif
# ifndef FD7_PATH
#  define FD7_PATH ""
# endif
# ifndef SKIP_IF_PROC_IS_UNAVAILABLE
#  define SKIP_IF_PROC_IS_UNAVAILABLE
# endif

static const char *errstr;

static long
k_close_range(const unsigned int fd1, const unsigned int fd2, const unsigned int flags)
{
	const kernel_ulong_t fill = (kernel_ulong_t) 0xdefaced00000000ULL;
	const kernel_ulong_t bad = (kernel_ulong_t) 0xbadc0dedbadc0dedULL;
	const kernel_ulong_t arg1 = fill | fd1;
	const kernel_ulong_t arg2 = fill | fd2;
	const kernel_ulong_t arg3 = fill | flags;
	const long rc = syscall(__NR_close_range, arg1, arg2, arg3, bad, bad, bad);
	errstr = sprintrc(rc);
	return rc;
}

static void
xdup2(int fd1, int fd2)
{
	if (dup2(fd1, fd2) != fd2)
		perror_msg_and_fail("dup2(%d, %d)", fd1, fd2);
}

int
main(void)
{
	SKIP_IF_PROC_IS_UNAVAILABLE;

	const int fd0 = dup(0);
	const int fd7 = dup(7);
	const int fd7_min = MIN(7, fd7);
	const int fd7_max = MAX(7, fd7);

	k_close_range(-2, -1, 1);
# ifndef PATH_TRACING
	printf("close_range(-2, -1, 0x1 /* CLOSE_RANGE_??? */) = %s\n", errstr);
# endif

	k_close_range(-1, -2, 2);
# ifndef PATH_TRACING
	printf("close_range(-1, -2, CLOSE_RANGE_UNSHARE) = %s\n", errstr);
# endif

	k_close_range(-3, 0, 4);
# ifndef PATH_TRACING
	printf("close_range(-3, 0" FD0_PATH ", 0x4 /* CLOSE_RANGE_??? */)"
	       " = %s\n", errstr);
# endif

	k_close_range(0, -4, -1);
# ifndef PATH_TRACING
	printf("close_range(0" FD0_PATH ", -4, CLOSE_RANGE_UNSHARE|%#x) = %s\n",
	       (-1U & ~CLOSE_RANGE_UNSHARE), errstr);
# endif

	k_close_range(-5, 7, 0);
	printf("close_range(-5, 7" FD7_PATH ", 0) = %s\n", errstr);

	k_close_range(7, -6, 1);
	printf("close_range(7" FD7_PATH ", -6, 0x1 /* CLOSE_RANGE_??? */)"
	       " = %s\n", errstr);

	k_close_range(7, 7, 4);
	printf("close_range(7" FD7_PATH ", 7" FD7_PATH
	       ", 0x4 /* CLOSE_RANGE_??? */) = %s\n", errstr);

	k_close_range(-7, -7, 7);
# ifndef PATH_TRACING
	printf("close_range(-7, -7, CLOSE_RANGE_UNSHARE|0x5) = %s\n", errstr);
# endif

	k_close_range(7, 0, 0);
	printf("close_range(7" FD7_PATH ", 0" FD0_PATH ", 0) = %s\n", errstr);

	k_close_range(fd7, fd0, 0);
	printf("close_range(%d" FD7_PATH ", %d" FD0_PATH ", 0) = %s\n",
	       fd7, fd0, errstr);

	if (k_close_range(0, 0, 0) == 0)
		xdup2(fd0, 0);
# ifndef PATH_TRACING
	printf("close_range(0" FD0_PATH ", 0" FD0_PATH ", 0) = %s\n", errstr);
# endif

	if (k_close_range(fd0, fd0, 0) == 0)
		xdup2(0, fd0);
# ifndef PATH_TRACING
	printf("close_range(%d" FD0_PATH ", %d" FD0_PATH ", 0) = %s\n",
	       fd0, fd0, errstr);
# endif

	if (k_close_range(7, 7, 0) == 0)
		xdup2(fd7, 7);
	printf("close_range(7" FD7_PATH ", 7" FD7_PATH ", 0) = %s\n", errstr);

	if (k_close_range(fd7, fd7, 0) == 0)
		xdup2(7, fd7);
	printf("close_range(%d" FD7_PATH ", %d" FD7_PATH ", 0) = %s\n",
	       fd7, fd7, errstr);

	if (k_close_range(fd7_max, -1, 0) == 0)
		xdup2(fd7_min, fd7_max);
	printf("close_range(%d" FD7_PATH ", -1, 0) = %s\n",
	       fd7_max, errstr);

	puts("+++ exited with 0 +++");
	return 0;
}

#else

SKIP_MAIN_UNDEFINED("__NR_close_range")

#endif