1 files changed, 42 insertions, 0 deletions
diff --git a/src/kernel/tests/include/tst_checkpoint_fn.h b/src/kernel/tests/include/tst_checkpoint_fn.h
new file mode 100644
index 0000000..57db905
--- /dev/null
+++ b/src/kernel/tests/include/tst_checkpoint_fn.h
@@ -0,0 +1,42 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later
+ * Copyright (c) 2015-2016 Cyril Hrubis <chrubis@suse.cz>
+ */
+
+#ifndef TST_CHECKPOINT_FN__
+#define TST_CHECKPOINT_FN__
+
+/*
+ * Checkpoint initializaton, must be done first.
+ *
+ * NOTE: tst_tmpdir() must be called beforehand.
+ */
+void tst_checkpoint_init(const char *file, const int lineno,
+			 void (*cleanup_fn)(void));
+
+/*
+ * Waits for wakeup.
+ *
+ * @id: Checkpoint id, possitive number
+ * @msec_timeout: Timeout in milliseconds, 0 == no timeout
+ */
+int tst_checkpoint_wait(unsigned int id, unsigned int msec_timeout);
+
+/*
+ * Wakes up sleeping process(es)/thread(s).
+ *
+ * @id: Checkpoint id, possitive number
+ * @nr_wake: Number of processes/threads to wake up
+ * @msec_timeout: Timeout in milliseconds, 0 == no timeout
+ */
+int tst_checkpoint_wake(unsigned int id, unsigned int nr_wake,
+                        unsigned int msec_timeout);
+
+void tst_safe_checkpoint_wait(const char *file, const int lineno,
+                              void (*cleanup_fn)(void), unsigned int id,
+			      unsigned int msec_timeout);
+
+void tst_safe_checkpoint_wake(const char *file, const int lineno,
+                              void (*cleanup_fn)(void), unsigned int id,
+                              unsigned int nr_wake);
+
+#endif /* TST_CHECKPOINT_FN__ */