LXR linux/tools/testing/selftests/kvm/lib/test

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