githubmirror/gnutls

src/benchmark.h
80 строк2 KB
Nikos Mavrogiannopoulos
gnutls-cli: benchmark-tls-kx can work with sub-ms accuracy
20 дек 2019, 22:52
20 дек 2019, 22:526f314f1
/*
 * Copyright (C) 2011-2012 Free Software Foundation, Inc.
 *
 * This file is part of GnuTLS.
 *
 * GnuTLS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * GnuTLS is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef GNUTLS_SRC_BENCHMARK_H
#define GNUTLS_SRC_BENCHMARK_H

#include <sys/time.h>
#include <time.h>
#include <signal.h>
#if defined(_WIN32)
#include <windows.h>
#endif

/* for uint64_t */
# include <stdint.h>

#if defined(HAVE_CLOCK_GETTIME) && defined(CLOCK_PROCESS_CPUTIME_ID)
#undef gettime
#define gettime(x) clock_gettime(CLOCK_PROCESS_CPUTIME_ID, x)
#else
inline static void gettime(struct timespec *ts)
{
	struct timeval tv;
	gettimeofday(&tv, NULL);
	ts->tv_sec = tv.tv_sec;
	ts->tv_nsec = tv.tv_usec * 1000;
}
#endif

typedef void (*sighandler_t) (int);

void benchmark_cipher(int debug_level);
void benchmark_tls(int debug_level, int ciphers);

struct benchmark_st {
	struct timespec start;
	uint64_t size;
	sighandler_t old_handler;
#if defined(_WIN32)
	HANDLE wtimer;
	HANDLE wthread;
	LARGE_INTEGER alarm_timeout;
#endif
};

extern volatile int benchmark_must_finish;

void start_benchmark(struct benchmark_st *st);
double stop_benchmark(struct benchmark_st *st, const char *metric,
		      int quiet);

inline static unsigned int
timespec_sub_ms(struct timespec *a, struct timespec *b)
{
	return (a->tv_sec - b->tv_sec) * 1000 + (a->tv_nsec - b->tv_nsec) / (1000 * 1000);
}

inline static unsigned long
timespec_sub_ns(struct timespec *a, struct timespec *b)
{
	return (a->tv_sec - b->tv_sec) * 1000000000 + (a->tv_nsec - b->tv_nsec);
}

#endif /* GNUTLS_SRC_BENCHMARK_H */