]> granicus.if.org Git - strace/blob - tests/tests.h
tests: introduce midtail_alloc and use it in netlink tests
[strace] / tests / tests.h
1 /*
2  * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org>
3  * Copyright (c) 2016-2018 The strace developers.
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. The name of the author may not be used to endorse or promote products
15  *    derived from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #ifndef STRACE_TESTS_H
30 #define STRACE_TESTS_H
31
32 #ifdef HAVE_CONFIG_H
33 # include "config.h"
34 #endif
35
36 #ifdef TESTS_SIZEOF_KERNEL_LONG_T
37 # undef SIZEOF_KERNEL_LONG_T
38 # define SIZEOF_KERNEL_LONG_T TESTS_SIZEOF_KERNEL_LONG_T
39 #endif
40
41 #ifdef TESTS_SIZEOF_LONG
42 # undef SIZEOF_LONG
43 # define SIZEOF_LONG TESTS_SIZEOF_LONG
44 #endif
45
46 #include <stdbool.h>
47 #include <sys/types.h>
48 #include "kernel_types.h"
49 #include "gcc_compat.h"
50 #include "macros.h"
51
52 /*
53  * The printf-like function to use in header files
54  * shared between strace and its tests.
55  */
56 #ifndef STRACE_PRINTF
57 # define STRACE_PRINTF printf
58 #endif
59
60 /* Tests of "strace -v" are expected to define VERBOSE to 1. */
61 #ifndef VERBOSE
62 # define VERBOSE 0
63 #endif
64
65 /* xlat verbosity defaults */
66 #ifndef XLAT_RAW
67 # define XLAT_RAW 0
68 #endif
69 #ifndef XLAT_VERBOSE
70 # define XLAT_VERBOSE 0
71 #endif
72
73 #ifndef DEFAULT_STRLEN
74 /* Default maximum # of bytes printed in printstr et al. */
75 # define DEFAULT_STRLEN 32
76 #endif
77
78 /* Cached sysconf(_SC_PAGESIZE). */
79 size_t get_page_size(void);
80
81 /* The size of kernel's sigset_t. */
82 unsigned int get_sigset_size(void);
83
84 /* Print message and strerror(errno) to stderr, then exit(1). */
85 void perror_msg_and_fail(const char *, ...)
86         ATTRIBUTE_FORMAT((printf, 1, 2)) ATTRIBUTE_NORETURN;
87 /* Print message to stderr, then exit(1). */
88 void error_msg_and_fail(const char *, ...)
89         ATTRIBUTE_FORMAT((printf, 1, 2)) ATTRIBUTE_NORETURN;
90 /* Print message to stderr, then exit(77). */
91 void error_msg_and_skip(const char *, ...)
92         ATTRIBUTE_FORMAT((printf, 1, 2)) ATTRIBUTE_NORETURN;
93 /* Print message and strerror(errno) to stderr, then exit(77). */
94 void perror_msg_and_skip(const char *, ...)
95         ATTRIBUTE_FORMAT((printf, 1, 2)) ATTRIBUTE_NORETURN;
96
97 #ifndef perror_msg_and_fail
98 # define perror_msg_and_fail(fmt_, ...) \
99         perror_msg_and_fail("%s:%d: " fmt_, __FILE__, __LINE__, ##__VA_ARGS__)
100 #endif
101 #ifndef perror_msg_and_fail
102 # define error_msg_and_fail(fmt_, ...) \
103         error_msg_and_fail("%s:%d: " fmt_, __FILE__, __LINE__, ##__VA_ARGS__)
104 #endif
105
106 /* Stat the specified file and skip the test if the stat call failed. */
107 void skip_if_unavailable(const char *);
108
109 /*
110  * Allocate memory that ends on the page boundary.
111  * Pages allocated by this call are preceded by an unmapped page
112  * and followed also by an unmapped page.
113  */
114 void *tail_alloc(const size_t)
115         ATTRIBUTE_MALLOC ATTRIBUTE_ALLOC_SIZE((1));
116 /* Allocate memory using tail_alloc, then memcpy. */
117 void *tail_memdup(const void *, const size_t)
118         ATTRIBUTE_MALLOC ATTRIBUTE_ALLOC_SIZE((2));
119
120 #define midtail_alloc(after_, before_) \
121         ((void *) ((char *) tail_alloc(((before_) + (after_))) + (before_)))
122
123 /*
124  * Allocate an object of the specified type at the end
125  * of a mapped memory region.
126  * Assign its address to the specified constant pointer.
127  */
128 #define TAIL_ALLOC_OBJECT_CONST_PTR(type_name, type_ptr)        \
129         type_name *const type_ptr = tail_alloc(sizeof(*type_ptr))
130
131 /*
132  * Allocate an object of the specified type at the end
133  * of a mapped memory region.
134  * Assign its address to the specified variable pointer.
135  */
136 #define TAIL_ALLOC_OBJECT_VAR_PTR(type_name, type_ptr)          \
137         type_name *type_ptr = tail_alloc(sizeof(*type_ptr))
138
139 /*
140  * Fill memory (pointed by ptr, having size bytes) with different bytes (with
141  * values starting with start and resetting every period) in order to catch
142  * sign, byte order and/or alignment errors.
143  */
144 void fill_memory_ex(void *ptr, size_t size, unsigned char start,
145                     unsigned char period);
146 /* Shortcut for fill_memory_ex(ptr, size, 0x80, 0x80) */
147 void fill_memory(void *ptr, size_t size);
148
149 /* Close stdin, move stdout to a non-standard descriptor, and print. */
150 void tprintf(const char *, ...)
151         ATTRIBUTE_FORMAT((printf, 1, 2));
152
153 /* Make a hexdump copy of C string */
154 const char *hexdump_strdup(const char *);
155
156 /* Make a hexdump copy of memory */
157 const char *hexdump_memdup(const char *, size_t);
158
159 /* Make a hexquoted copy of a string */
160 const char *hexquote_strndup(const char *, size_t);
161
162 /* Return inode number of socket descriptor. */
163 unsigned long inode_of_sockfd(int);
164
165 /* Print string in a quoted form with optional escape characters. */
166 void print_quoted_string_ex(const char *, bool quote, const char *escape_str);
167
168 /* Print string in a quoted form. */
169 void print_quoted_string(const char *);
170
171 /*
172  * Print a NUL-terminated string `str' of length up to `size' - 1
173  * in a quoted form.
174  */
175 void print_quoted_cstring(const char *str, size_t size);
176
177 /* Print memory in a quoted form with optional escape characters. */
178 void print_quoted_memory_ex(const void *, size_t, bool quote,
179                             const char *escape_chars);
180
181 /* Print memory in a quoted form. */
182 void print_quoted_memory(const void *, size_t);
183
184 /* Print memory in a hexquoted form. */
185 void print_quoted_hex(const void *, size_t);
186
187 /* Print time_t and nanoseconds in symbolic format. */
188 void print_time_t_nsec(time_t, unsigned long long, int);
189
190 /* Print time_t and microseconds in symbolic format. */
191 void print_time_t_usec(time_t, unsigned long long, int);
192
193 /* Read an int from the file. */
194 int read_int_from_file(const char *, int *);
195
196 /* Check whether given uid matches kernel overflowuid. */
197 void check_overflowuid(const int);
198
199 /* Check whether given gid matches kernel overflowgid. */
200 void check_overflowgid(const int);
201
202 /* Translate errno to its name. */
203 const char *errno2name(void);
204
205 /* Translate signal number to its name. */
206 const char *signal2name(int);
207
208 /* Print return code and, in case return code is -1, errno information. */
209 const char *sprintrc(long rc);
210 /* sprintrc variant suitable for usage as part of grep pattern. */
211 const char *sprintrc_grep(long rc);
212
213 struct xlat;
214
215 /* Print flags in symbolic form according to xlat table. */
216 int printflags(const struct xlat *, const unsigned long long, const char *);
217
218 /* Print constant in symbolic form according to xlat table. */
219 int printxval(const struct xlat *, const unsigned long long, const char *);
220
221 /* Invoke a socket syscall, either directly or via __NR_socketcall. */
222 int socketcall(const int nr, const int call,
223                long a1, long a2, long a3, long a4, long a5);
224
225 /* Wrappers for recvmmsg and sendmmsg syscalls. */
226 struct mmsghdr;
227 struct timespec;
228 int recv_mmsg(int, struct mmsghdr *, unsigned int, unsigned int, struct timespec *);
229 int send_mmsg(int, struct mmsghdr *, unsigned int, unsigned int);
230
231 /* Create a netlink socket. */
232 int create_nl_socket_ext(int proto, const char *name);
233 #define create_nl_socket(proto) create_nl_socket_ext((proto), #proto)
234
235 /* Create a pipe with maximized descriptor numbers. */
236 void pipe_maxfd(int pipefd[2]);
237
238 /* if_nametoindex("lo") */
239 unsigned int ifindex_lo(void);
240
241 #ifdef HAVE_IF_INDEXTONAME
242 # define IFINDEX_LO_STR "if_nametoindex(\"lo\")"
243 #else
244 # define IFINDEX_LO_STR "1"
245 #endif
246
247 #define F8ILL_KULONG_SUPPORTED  (sizeof(void *) < sizeof(kernel_ulong_t))
248 #define F8ILL_KULONG_MASK       ((kernel_ulong_t) 0xffffffff00000000ULL)
249
250 /*
251  * For 64-bit kernel_ulong_t and 32-bit pointer,
252  * return a kernel_ulong_t value by filling higher bits.
253  * For other architertures, return the original pointer.
254  */
255 static inline kernel_ulong_t
256 f8ill_ptr_to_kulong(const void *const ptr)
257 {
258         const unsigned long uptr = (unsigned long) ptr;
259         return F8ILL_KULONG_SUPPORTED
260                ? F8ILL_KULONG_MASK | uptr : (kernel_ulong_t) uptr;
261 }
262
263 # define LENGTH_OF(arg) ((unsigned int) sizeof(arg) - 1)
264
265 /* Zero-extend a signed integer type to unsigned long long. */
266 #define zero_extend_signed_to_ull(v) \
267         (sizeof(v) == sizeof(char) ? (unsigned long long) (unsigned char) (v) : \
268          sizeof(v) == sizeof(short) ? (unsigned long long) (unsigned short) (v) : \
269          sizeof(v) == sizeof(int) ? (unsigned long long) (unsigned int) (v) : \
270          sizeof(v) == sizeof(long) ? (unsigned long long) (unsigned long) (v) : \
271          (unsigned long long) (v))
272
273 /* Sign-extend an unsigned integer type to long long. */
274 #define sign_extend_unsigned_to_ll(v) \
275         (sizeof(v) == sizeof(char) ? (long long) (char) (v) : \
276          sizeof(v) == sizeof(short) ? (long long) (short) (v) : \
277          sizeof(v) == sizeof(int) ? (long long) (int) (v) : \
278          sizeof(v) == sizeof(long) ? (long long) (long) (v) : \
279          (long long) (v))
280
281 #define SKIP_MAIN_UNDEFINED(arg) \
282         int main(void) { error_msg_and_skip("undefined: %s", arg); }
283
284 #if WORDS_BIGENDIAN
285 # define LL_PAIR(HI, LO) (HI), (LO)
286 #else
287 # define LL_PAIR(HI, LO) (LO), (HI)
288 #endif
289 #define LL_VAL_TO_PAIR(llval) LL_PAIR((long) ((llval) >> 32), (long) (llval))
290
291 #define _STR(_arg) #_arg
292 #define ARG_STR(_arg) (_arg), #_arg
293 #define ARG_ULL_STR(_arg) _arg##ULL, #_arg
294
295 /*
296  * Assign an object of type DEST_TYPE at address DEST_ADDR
297  * using memcpy to avoid potential unaligned access.
298  */
299 #define SET_STRUCT(DEST_TYPE, DEST_ADDR, ...)                                           \
300         do {                                                                            \
301                 DEST_TYPE dest_type_tmp_var = { __VA_ARGS__ };                          \
302                 memcpy(DEST_ADDR, &dest_type_tmp_var, sizeof(dest_type_tmp_var));       \
303         } while (0)
304
305 #define NLMSG_ATTR(nlh, hdrlen) ((void *)(nlh) + NLMSG_SPACE(hdrlen))
306
307 #endif /* !STRACE_TESTS_H */