examples/timing.c

   1 /* timing.c */
   2
   3 /* nettle, low-level cryptographics library
   4  *
   5  * Copyright (C) 2013 Niels Möller
   6  *
   7  * The nettle library is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU Lesser General Public License as published by
   9  * the Free Software Foundation; either version 2.1 of the License, or (at your
  10  * option) any later version.
  11  *
  12  * The nettle library is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  15  * License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public License
  18  * along with the nettle library; see the file COPYING.LIB.  If not, write to
  19  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  20  * MA 02111-1301, USA.
  21  */
  22
  23 #if HAVE_CONFIG_H
  24 # include "config.h"
  25 #endif
  26
  27 #include "timing.h"
  28
  29 #include <errno.h>
  30 #include <stdarg.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <time.h>
  35
  36 #if HAVE_CLOCK_GETTIME && defined CLOCK_PROCESS_CPUTIME_ID
  37 #define TRY_CLOCK_GETTIME 1
  38 struct timespec cgt_start;
  39
  40 static void NORETURN PRINTF_STYLE(1,2)
  41 die(const char *format, ...)
  42 {
  43   va_list args;
  44   va_start(args, format);
  45   vfprintf(stderr, format, args);
  46   va_end(args);
  47
  48   exit(EXIT_FAILURE);
  49 }
  50
  51 static int
  52 cgt_works_p(void)
  53 {
  54   struct timespec now;
  55   return clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &now) == 0;
  56 }
  57
  58 static void
  59 cgt_time_start(void)
  60 {
  61   if (clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &cgt_start) < 0)
  62     die("clock_gettime failed: %s\n", strerror(errno));
  63 }
  64
  65 static double
  66 cgt_time_end(void)
  67 {
  68   struct timespec end;
  69   if (clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &end) < 0)
  70     die("clock_gettime failed: %s\n", strerror(errno));
  71
  72   return end.tv_sec - cgt_start.tv_sec
  73     + 1e-9 * (end.tv_nsec - cgt_start.tv_nsec);
  74 }
  75 #endif /* !HAVE_CLOCK_GETTIME */
  76
  77 static clock_t clock_start;
  78
  79 static void
  80 clock_time_start(void)
  81 {
  82   clock_start = clock();
  83 }
  84
  85 static double
  86 clock_time_end(void)
  87 {
  88   return (double) (clock() - (clock_start)) / CLOCKS_PER_SEC;
  89 }
  90
  91 void (*time_start)(void) = clock_time_start;
  92 double (*time_end)(void) = clock_time_end;
  93
  94 void time_init(void)
  95 {
  96   /* Choose timing function */
  97 #if TRY_CLOCK_GETTIME
  98   if (cgt_works_p())
  99     {
 100       time_start = cgt_time_start;
 101       time_end = cgt_time_end;
 102     }
 103   else
 104     {
 105       fprintf(stderr, "clock_gettime not working, falling back to clock\n");
 106       time_start = clock_time_start;
 107       time_end = clock_time_end;
 108     }
 109 #endif
 110 }