tools/testing/selftests/kvm/lib/test_util.c

   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  */
  19 size_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
  57 int64_t timespec_to_ns(struct timespec ts)
  58 {
  59         return (int64_t)ts.tv_nsec + 1000000000LL * (int64_t)ts.tv_sec;
  60 }
  61
  62 struct 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
  73 struct 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
  80 struct 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
  87 struct 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
  95 struct 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
 102 void 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 }