17a338472SThomas Gleixner /* SPDX-License-Identifier: GPL-2.0-only */
2783e9e51SPaolo Bonzini /*
3783e9e51SPaolo Bonzini  * tools/testing/selftests/kvm/include/test_util.h
4783e9e51SPaolo Bonzini  *
5783e9e51SPaolo Bonzini  * Copyright (C) 2018, Google LLC.
6783e9e51SPaolo Bonzini  */
7783e9e51SPaolo Bonzini 
8cc68765dSAndrew Jones #ifndef SELFTEST_KVM_TEST_UTIL_H
9cc68765dSAndrew Jones #define SELFTEST_KVM_TEST_UTIL_H
10783e9e51SPaolo Bonzini 
11783e9e51SPaolo Bonzini #include <stdlib.h>
12783e9e51SPaolo Bonzini #include <stdarg.h>
13783e9e51SPaolo Bonzini #include <stdbool.h>
14783e9e51SPaolo Bonzini #include <stdio.h>
15783e9e51SPaolo Bonzini #include <string.h>
16783e9e51SPaolo Bonzini #include <inttypes.h>
17783e9e51SPaolo Bonzini #include <errno.h>
18783e9e51SPaolo Bonzini #include <unistd.h>
19783e9e51SPaolo Bonzini #include <fcntl.h>
20bcb2b94aSPaolo Bonzini #include "kselftest.h"
21783e9e51SPaolo Bonzini 
223439d886SAndrew Jones static inline int _no_printf(const char *format, ...) { return 0; }
233439d886SAndrew Jones 
243439d886SAndrew Jones #ifdef DEBUG
253439d886SAndrew Jones #define pr_debug(...) printf(__VA_ARGS__)
263439d886SAndrew Jones #else
273439d886SAndrew Jones #define pr_debug(...) _no_printf(__VA_ARGS__)
283439d886SAndrew Jones #endif
293439d886SAndrew Jones #ifndef QUIET
303439d886SAndrew Jones #define pr_info(...) printf(__VA_ARGS__)
313439d886SAndrew Jones #else
323439d886SAndrew Jones #define pr_info(...) _no_printf(__VA_ARGS__)
333439d886SAndrew Jones #endif
343439d886SAndrew Jones 
35d0aac332SAndrew Jones void print_skip(const char *fmt, ...) __attribute__((format(printf, 1, 2)));
36d0aac332SAndrew Jones 
37783e9e51SPaolo Bonzini ssize_t test_write(int fd, const void *buf, size_t count);
38783e9e51SPaolo Bonzini ssize_t test_read(int fd, void *buf, size_t count);
39783e9e51SPaolo Bonzini int test_seq_read(const char *path, char **bufp, size_t *sizep);
40783e9e51SPaolo Bonzini 
41783e9e51SPaolo Bonzini void test_assert(bool exp, const char *exp_str,
42d9eaf19eSAndrew Jones 		 const char *file, unsigned int line, const char *fmt, ...)
43d9eaf19eSAndrew Jones 		__attribute__((format(printf, 5, 6)));
44783e9e51SPaolo Bonzini 
45783e9e51SPaolo Bonzini #define TEST_ASSERT(e, fmt, ...) \
46783e9e51SPaolo Bonzini 	test_assert((e), #e, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
47783e9e51SPaolo Bonzini 
48783e9e51SPaolo Bonzini #define ASSERT_EQ(a, b) do { \
49783e9e51SPaolo Bonzini 	typeof(a) __a = (a); \
50783e9e51SPaolo Bonzini 	typeof(b) __b = (b); \
51783e9e51SPaolo Bonzini 	TEST_ASSERT(__a == __b, \
52783e9e51SPaolo Bonzini 		    "ASSERT_EQ(%s, %s) failed.\n" \
53783e9e51SPaolo Bonzini 		    "\t%s is %#lx\n" \
54783e9e51SPaolo Bonzini 		    "\t%s is %#lx", \
55783e9e51SPaolo Bonzini 		    #a, #b, #a, (unsigned long) __a, #b, (unsigned long) __b); \
56783e9e51SPaolo Bonzini } while (0)
57783e9e51SPaolo Bonzini 
58af99e1adSBen Gardon size_t parse_size(const char *size);
59af99e1adSBen Gardon 
60f09205b9SBen Gardon int64_t timespec_to_ns(struct timespec ts);
61f09205b9SBen Gardon struct timespec timespec_diff(struct timespec start, struct timespec end);
6294c4b76bSAndrew Jones struct timespec timespec_add_ns(struct timespec ts, int64_t ns);
63f09205b9SBen Gardon 
64cc68765dSAndrew Jones #endif /* SELFTEST_KVM_TEST_UTIL_H */
65