xref: /linux/tools/testing/selftests/vfio/lib/include/libvfio/assert.h (revision a3ebb59eee2e558e8f8f27fc3f75cd367f17cd8e)
1 /* SPDX-License-Identifier: GPL-2.0-only */
2 #ifndef SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_ASSERT_H
3 #define SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_ASSERT_H
4 
5 #include <stdio.h>
6 #include <string.h>
7 #include <sys/ioctl.h>
8 
9 #include "../../../../kselftest.h"
10 
11 #define VFIO_LOG_AND_EXIT(...) do {		\
12 	fprintf(stderr, "  " __VA_ARGS__);	\
13 	fprintf(stderr, "\n");			\
14 	exit(KSFT_FAIL);			\
15 } while (0)
16 
17 #define VFIO_ASSERT_OP(_lhs, _rhs, _op, ...) do {				\
18 	typeof(_lhs) __lhs = (_lhs);						\
19 	typeof(_rhs) __rhs = (_rhs);						\
20 										\
21 	if (__lhs _op __rhs)							\
22 		break;								\
23 										\
24 	fprintf(stderr, "%s:%u: Assertion Failure\n\n", __FILE__, __LINE__);	\
25 	fprintf(stderr, "  Expression: " #_lhs " " #_op " " #_rhs "\n");	\
26 	fprintf(stderr, "  Observed: %#lx %s %#lx\n",				\
27 			(u64)__lhs, #_op, (u64)__rhs);				\
28 	fprintf(stderr, "  [errno: %d - %s]\n", errno, strerror(errno));	\
29 	VFIO_LOG_AND_EXIT(__VA_ARGS__);						\
30 } while (0)
31 
32 #define VFIO_ASSERT_EQ(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, ==, ##__VA_ARGS__)
33 #define VFIO_ASSERT_NE(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, !=, ##__VA_ARGS__)
34 #define VFIO_ASSERT_LT(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, <, ##__VA_ARGS__)
35 #define VFIO_ASSERT_LE(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, <=, ##__VA_ARGS__)
36 #define VFIO_ASSERT_GT(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, >, ##__VA_ARGS__)
37 #define VFIO_ASSERT_GE(_a, _b, ...) VFIO_ASSERT_OP(_a, _b, >=, ##__VA_ARGS__)
38 #define VFIO_ASSERT_TRUE(_a, ...) VFIO_ASSERT_NE(false, (_a), ##__VA_ARGS__)
39 #define VFIO_ASSERT_FALSE(_a, ...) VFIO_ASSERT_EQ(false, (_a), ##__VA_ARGS__)
40 #define VFIO_ASSERT_NULL(_a, ...) VFIO_ASSERT_EQ(NULL, _a, ##__VA_ARGS__)
41 #define VFIO_ASSERT_NOT_NULL(_a, ...) VFIO_ASSERT_NE(NULL, _a, ##__VA_ARGS__)
42 
43 #define VFIO_FAIL(_fmt, ...) do {				\
44 	fprintf(stderr, "%s:%u: FAIL\n\n", __FILE__, __LINE__);	\
45 	VFIO_LOG_AND_EXIT(_fmt, ##__VA_ARGS__);			\
46 } while (0)
47 
48 #define ioctl_assert(_fd, _op, _arg) do {						       \
49 	void *__arg = (_arg);								       \
50 	int __ret = ioctl((_fd), (_op), (__arg));					       \
51 	VFIO_ASSERT_EQ(__ret, 0, "ioctl(%s, %s, %s) returned %d\n", #_fd, #_op, #_arg, __ret); \
52 } while (0)
53 
54 #endif /* SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_ASSERT_H */
55