3b70c4d128
If we want to have some tests about transparent hugepages, the system configured THP hugepage size should better be known by the tests, which can be used for kinds of alignment or guest memory accessing of vcpus... So it makes sense to add a helper to get the transparent hugepage size. With VM_MEM_SRC_ANONYMOUS_THP specified in vm_userspace_mem_region_add(), we now stat /sys/kernel/mm/transparent_hugepage to check whether THP is configured in the host kernel before madvise(). Based on this, we can also read file /sys/kernel/mm/transparent_hugepage/hpage_pmd_size to get THP hugepage size. Signed-off-by: Yanan Wang <wangyanan55@huawei.com> Reviewed-by: Ben Gardon <bgardon@google.com> Reviewed-by: Andrew Jones <drjones@redhat.com> Message-Id: <20210330080856.14940-7-wangyanan55@huawei.com> Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
86 lines
2.3 KiB
C
86 lines
2.3 KiB
C
/* SPDX-License-Identifier: GPL-2.0-only */
|
|
/*
|
|
* tools/testing/selftests/kvm/include/test_util.h
|
|
*
|
|
* Copyright (C) 2018, Google LLC.
|
|
*/
|
|
|
|
#ifndef SELFTEST_KVM_TEST_UTIL_H
|
|
#define SELFTEST_KVM_TEST_UTIL_H
|
|
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <inttypes.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include "kselftest.h"
|
|
|
|
static inline int _no_printf(const char *format, ...) { return 0; }
|
|
|
|
#ifdef DEBUG
|
|
#define pr_debug(...) printf(__VA_ARGS__)
|
|
#else
|
|
#define pr_debug(...) _no_printf(__VA_ARGS__)
|
|
#endif
|
|
#ifndef QUIET
|
|
#define pr_info(...) printf(__VA_ARGS__)
|
|
#else
|
|
#define pr_info(...) _no_printf(__VA_ARGS__)
|
|
#endif
|
|
|
|
void print_skip(const char *fmt, ...) __attribute__((format(printf, 1, 2)));
|
|
|
|
ssize_t test_write(int fd, const void *buf, size_t count);
|
|
ssize_t test_read(int fd, void *buf, size_t count);
|
|
int test_seq_read(const char *path, char **bufp, size_t *sizep);
|
|
|
|
void test_assert(bool exp, const char *exp_str,
|
|
const char *file, unsigned int line, const char *fmt, ...)
|
|
__attribute__((format(printf, 5, 6)));
|
|
|
|
#define TEST_ASSERT(e, fmt, ...) \
|
|
test_assert((e), #e, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
|
|
|
|
#define ASSERT_EQ(a, b) do { \
|
|
typeof(a) __a = (a); \
|
|
typeof(b) __b = (b); \
|
|
TEST_ASSERT(__a == __b, \
|
|
"ASSERT_EQ(%s, %s) failed.\n" \
|
|
"\t%s is %#lx\n" \
|
|
"\t%s is %#lx", \
|
|
#a, #b, #a, (unsigned long) __a, #b, (unsigned long) __b); \
|
|
} while (0)
|
|
|
|
#define TEST_FAIL(fmt, ...) \
|
|
TEST_ASSERT(false, fmt, ##__VA_ARGS__)
|
|
|
|
size_t parse_size(const char *size);
|
|
|
|
int64_t timespec_to_ns(struct timespec ts);
|
|
struct timespec timespec_add_ns(struct timespec ts, int64_t ns);
|
|
struct timespec timespec_add(struct timespec ts1, struct timespec ts2);
|
|
struct timespec timespec_sub(struct timespec ts1, struct timespec ts2);
|
|
struct timespec timespec_elapsed(struct timespec start);
|
|
struct timespec timespec_div(struct timespec ts, int divisor);
|
|
|
|
enum vm_mem_backing_src_type {
|
|
VM_MEM_SRC_ANONYMOUS,
|
|
VM_MEM_SRC_ANONYMOUS_THP,
|
|
VM_MEM_SRC_ANONYMOUS_HUGETLB,
|
|
};
|
|
|
|
struct vm_mem_backing_src_alias {
|
|
const char *name;
|
|
enum vm_mem_backing_src_type type;
|
|
};
|
|
|
|
bool thp_configured(void);
|
|
size_t get_trans_hugepagesz(void);
|
|
void backing_src_help(void);
|
|
enum vm_mem_backing_src_type parse_backing_src_type(const char *type_name);
|
|
|
|
#endif /* SELFTEST_KVM_TEST_UTIL_H */
|