1 files changed, 80 insertions, 0 deletions
diff --git a/src/benchmark.h b/src/benchmark.h
new file mode 100644
index 0000000..a5e2aff
--- /dev/null
+++ b/src/benchmark.h
@@ -0,0 +1,80 @@
+/*
+ * 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 */