1 files changed, 47 insertions, 0 deletions
diff --git a/src/kernel/tests/include/tst_safe_prw.h b/src/kernel/tests/include/tst_safe_prw.h
new file mode 100644
index 0000000..01a684d
--- /dev/null
+++ b/src/kernel/tests/include/tst_safe_prw.h
@@ -0,0 +1,47 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later
+ * Copyright (c) 2010-2017 Linux Test Project
+ */
+
+#ifndef TST_SAFE_PRW_H__
+#define TST_SAFE_PRW_H__
+
+static inline ssize_t safe_pread(const char *file, const int lineno,
+		char len_strict, int fildes, void *buf, size_t nbyte,
+		off_t offset)
+{
+	ssize_t rval;
+
+	rval = pread(fildes, buf, nbyte, offset);
+
+	if (rval == -1 || (len_strict && (size_t)rval != nbyte)) {
+		tst_brk_(file, lineno, TBROK | TERRNO,
+			 "pread(%d,%p,%zu,%lld) failed",
+			 fildes, buf, nbyte, (long long)offset);
+	}
+
+	return rval;
+}
+#define SAFE_PREAD(len_strict, fildes, buf, nbyte, offset) \
+	safe_pread(__FILE__, __LINE__, (len_strict), (fildes), \
+	           (buf), (nbyte), (offset))
+
+static inline ssize_t safe_pwrite(const char *file, const int lineno,
+		char len_strict, int fildes, const void *buf, size_t nbyte,
+		off_t offset)
+{
+	ssize_t rval;
+
+	rval = pwrite(fildes, buf, nbyte, offset);
+	if (rval == -1 || (len_strict && (size_t)rval != nbyte)) {
+		tst_brk_(file, lineno, TBROK | TERRNO,
+			 "pwrite(%d,%p,%zu,%lld) failed",
+			 fildes, buf, nbyte, (long long)offset);
+	}
+
+	return rval;
+}
+#define SAFE_PWRITE(len_strict, fildes, buf, nbyte, offset) \
+	safe_pwrite(__FILE__, __LINE__, (len_strict), (fildes), \
+	            (buf), (nbyte), (offset))
+
+#endif /* SAFE_PRW_H__ */