1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * tools/testing/selftests/kvm/lib/test_util.c
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Copyright (C) 2020, Google LLC.
6*4882a593Smuzhiyun */
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun #include <assert.h>
9*4882a593Smuzhiyun #include <ctype.h>
10*4882a593Smuzhiyun #include <limits.h>
11*4882a593Smuzhiyun #include <stdlib.h>
12*4882a593Smuzhiyun #include <time.h>
13*4882a593Smuzhiyun
14*4882a593Smuzhiyun #include "test_util.h"
15*4882a593Smuzhiyun
16*4882a593Smuzhiyun /*
17*4882a593Smuzhiyun * Parses "[0-9]+[kmgt]?".
18*4882a593Smuzhiyun */
parse_size(const char * size)19*4882a593Smuzhiyun size_t parse_size(const char *size)
20*4882a593Smuzhiyun {
21*4882a593Smuzhiyun size_t base;
22*4882a593Smuzhiyun char *scale;
23*4882a593Smuzhiyun int shift = 0;
24*4882a593Smuzhiyun
25*4882a593Smuzhiyun TEST_ASSERT(size && isdigit(size[0]), "Need at least one digit in '%s'", size);
26*4882a593Smuzhiyun
27*4882a593Smuzhiyun base = strtoull(size, &scale, 0);
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun TEST_ASSERT(base != ULLONG_MAX, "Overflow parsing size!");
30*4882a593Smuzhiyun
31*4882a593Smuzhiyun switch (tolower(*scale)) {
32*4882a593Smuzhiyun case 't':
33*4882a593Smuzhiyun shift = 40;
34*4882a593Smuzhiyun break;
35*4882a593Smuzhiyun case 'g':
36*4882a593Smuzhiyun shift = 30;
37*4882a593Smuzhiyun break;
38*4882a593Smuzhiyun case 'm':
39*4882a593Smuzhiyun shift = 20;
40*4882a593Smuzhiyun break;
41*4882a593Smuzhiyun case 'k':
42*4882a593Smuzhiyun shift = 10;
43*4882a593Smuzhiyun break;
44*4882a593Smuzhiyun case 'b':
45*4882a593Smuzhiyun case '\0':
46*4882a593Smuzhiyun shift = 0;
47*4882a593Smuzhiyun break;
48*4882a593Smuzhiyun default:
49*4882a593Smuzhiyun TEST_ASSERT(false, "Unknown size letter %c", *scale);
50*4882a593Smuzhiyun }
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun TEST_ASSERT((base << shift) >> shift == base, "Overflow scaling size!");
53*4882a593Smuzhiyun
54*4882a593Smuzhiyun return base << shift;
55*4882a593Smuzhiyun }
56*4882a593Smuzhiyun
timespec_to_ns(struct timespec ts)57*4882a593Smuzhiyun int64_t timespec_to_ns(struct timespec ts)
58*4882a593Smuzhiyun {
59*4882a593Smuzhiyun return (int64_t)ts.tv_nsec + 1000000000LL * (int64_t)ts.tv_sec;
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun
timespec_add_ns(struct timespec ts,int64_t ns)62*4882a593Smuzhiyun struct timespec timespec_add_ns(struct timespec ts, int64_t ns)
63*4882a593Smuzhiyun {
64*4882a593Smuzhiyun struct timespec res;
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun res.tv_nsec = ts.tv_nsec + ns;
67*4882a593Smuzhiyun res.tv_sec = ts.tv_sec + res.tv_nsec / 1000000000LL;
68*4882a593Smuzhiyun res.tv_nsec %= 1000000000LL;
69*4882a593Smuzhiyun
70*4882a593Smuzhiyun return res;
71*4882a593Smuzhiyun }
72*4882a593Smuzhiyun
timespec_add(struct timespec ts1,struct timespec ts2)73*4882a593Smuzhiyun struct timespec timespec_add(struct timespec ts1, struct timespec ts2)
74*4882a593Smuzhiyun {
75*4882a593Smuzhiyun int64_t ns1 = timespec_to_ns(ts1);
76*4882a593Smuzhiyun int64_t ns2 = timespec_to_ns(ts2);
77*4882a593Smuzhiyun return timespec_add_ns((struct timespec){0}, ns1 + ns2);
78*4882a593Smuzhiyun }
79*4882a593Smuzhiyun
timespec_sub(struct timespec ts1,struct timespec ts2)80*4882a593Smuzhiyun struct timespec timespec_sub(struct timespec ts1, struct timespec ts2)
81*4882a593Smuzhiyun {
82*4882a593Smuzhiyun int64_t ns1 = timespec_to_ns(ts1);
83*4882a593Smuzhiyun int64_t ns2 = timespec_to_ns(ts2);
84*4882a593Smuzhiyun return timespec_add_ns((struct timespec){0}, ns1 - ns2);
85*4882a593Smuzhiyun }
86*4882a593Smuzhiyun
timespec_diff_now(struct timespec start)87*4882a593Smuzhiyun struct timespec timespec_diff_now(struct timespec start)
88*4882a593Smuzhiyun {
89*4882a593Smuzhiyun struct timespec end;
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun clock_gettime(CLOCK_MONOTONIC, &end);
92*4882a593Smuzhiyun return timespec_sub(end, start);
93*4882a593Smuzhiyun }
94*4882a593Smuzhiyun
timespec_div(struct timespec ts,int divisor)95*4882a593Smuzhiyun struct timespec timespec_div(struct timespec ts, int divisor)
96*4882a593Smuzhiyun {
97*4882a593Smuzhiyun int64_t ns = timespec_to_ns(ts) / divisor;
98*4882a593Smuzhiyun
99*4882a593Smuzhiyun return timespec_add_ns((struct timespec){0}, ns);
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun
print_skip(const char * fmt,...)102*4882a593Smuzhiyun void print_skip(const char *fmt, ...)
103*4882a593Smuzhiyun {
104*4882a593Smuzhiyun va_list ap;
105*4882a593Smuzhiyun
106*4882a593Smuzhiyun assert(fmt);
107*4882a593Smuzhiyun va_start(ap, fmt);
108*4882a593Smuzhiyun vprintf(fmt, ap);
109*4882a593Smuzhiyun va_end(ap);
110*4882a593Smuzhiyun puts(", skipping test");
111*4882a593Smuzhiyun }
112