1 files changed, 73 insertions, 0 deletions
diff --git a/tests/virt-time.h b/tests/virt-time.h
new file mode 100644
index 0000000..7f7c374
--- /dev/null
+++ b/tests/virt-time.h
@@ -0,0 +1,73 @@
+/*
+ * Copyright (C) 2018 Red Hat, Inc.
+ *
+ * Author: Nikos Mavrogiannopoulos
+ *
+ * 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 Lesser General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>
+ */
+
+#ifndef GNUTLS_TESTS_VIRT_TIME_H
+#define GNUTLS_TESTS_VIRT_TIME_H
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <time.h>
+#include <gnutls/gnutls.h>
+
+/* copied from ../lib/system.h so not to include that header from
+ * every test program */
+typedef void (*gnutls_gettime_func) (struct timespec *);
+extern void _gnutls_global_set_gettime_function(gnutls_gettime_func gettime_func);
+
+/* virtualize time in a test. This freezes the time in the test, except for
+ * the advances due to calls to virt_sleep_sec(). This makes the test
+ * independent of the test system load, and avoids any long delays. */
+static time_t _now;
+static struct timespec _now_ts;
+
+#define virt_sec_sleep(s) { \
+		_now += s; \
+		_now_ts.tv_sec += s; \
+	}
+
+#define virt_time_init_at(d) { \
+		_now = (d); \
+		gnutls_global_set_time_function(mytime); \
+		_now_ts.tv_sec = _now; \
+		_now_ts.tv_nsec = 0; \
+		_gnutls_global_set_gettime_function(mygettime); \
+	}
+
+#define virt_time_init() virt_time_init_at(time(0))
+
+
+static time_t mytime(time_t * t)
+{
+	if (t)
+		*t = _now;
+
+	return _now;
+}
+
+static void mygettime(struct timespec * t)
+{
+	if (t)
+		*t = _now_ts;
+}
+
+#endif /* GNUTLS_TESTS_VIRT_TIME_H */