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 58a46f8a63SWainer dos Santos Moschetta #define TEST_FAIL(fmt, ...) \ 59a46f8a63SWainer dos Santos Moschetta TEST_ASSERT(false, fmt, ##__VA_ARGS__) 60a46f8a63SWainer dos Santos Moschetta 61af99e1adSBen Gardon size_t parse_size(const char *size); 62af99e1adSBen Gardon 63f09205b9SBen Gardon int64_t timespec_to_ns(struct timespec ts); 6494c4b76bSAndrew Jones struct timespec timespec_add_ns(struct timespec ts, int64_t ns); 65bfcaa849SAndrew Jones struct timespec timespec_add(struct timespec ts1, struct timespec ts2); 66bfcaa849SAndrew Jones struct timespec timespec_sub(struct timespec ts1, struct timespec ts2); 67f9224a52SBen Gardon struct timespec timespec_elapsed(struct timespec start); 684fd94ec7SBen Gardon struct timespec timespec_div(struct timespec ts, int divisor); 69f09205b9SBen Gardon 709e965bb7SBen Gardon enum vm_mem_backing_src_type { 719e965bb7SBen Gardon VM_MEM_SRC_ANONYMOUS, 729e965bb7SBen Gardon VM_MEM_SRC_ANONYMOUS_THP, 739e965bb7SBen Gardon VM_MEM_SRC_ANONYMOUS_HUGETLB, 74*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_16KB, 75*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_64KB, 76*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_512KB, 77*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_1MB, 78*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_2MB, 79*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_8MB, 80*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_16MB, 81*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_32MB, 82*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_256MB, 83*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_512MB, 84*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_1GB, 85*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_2GB, 86*623653b7SYanan Wang VM_MEM_SRC_ANONYMOUS_HUGETLB_16GB, 87*623653b7SYanan Wang NUM_SRC_TYPES, 889e965bb7SBen Gardon }; 899e965bb7SBen Gardon 909e965bb7SBen Gardon struct vm_mem_backing_src_alias { 919e965bb7SBen Gardon const char *name; 92*623653b7SYanan Wang uint32_t flag; 939e965bb7SBen Gardon }; 949e965bb7SBen Gardon 953b70c4d1SYanan Wang bool thp_configured(void); 963b70c4d1SYanan Wang size_t get_trans_hugepagesz(void); 975579fa68SYanan Wang size_t get_def_hugetlb_pagesz(void); 98*623653b7SYanan Wang const struct vm_mem_backing_src_alias *vm_mem_backing_src_alias(uint32_t i); 99*623653b7SYanan Wang size_t get_backing_src_pagesz(uint32_t i); 1009e965bb7SBen Gardon void backing_src_help(void); 1019e965bb7SBen Gardon enum vm_mem_backing_src_type parse_backing_src_type(const char *type_name); 1029e965bb7SBen Gardon 103cc68765dSAndrew Jones #endif /* SELFTEST_KVM_TEST_UTIL_H */ 104