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