1 files changed, 312 insertions, 0 deletions
diff --git a/ltp_framework/lib/safe_macros.c b/ltp_framework/lib/safe_macros.c
new file mode 100644
index 0000000..11f5fe6
--- /dev/null
+++ b/ltp_framework/lib/safe_macros.c
@@ -0,0 +1,312 @@
+#include <sys/types.h>
+#include <sys/mman.h>
+#include <sys/resource.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <libgen.h>
+#include <pwd.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include "test.h"
+#include "safe_macros.h"
+
+char *
+safe_basename(const char *file, const int lineno, void (*cleanup_fn)(void),
+    char *path)
+{
+	char *rval;
+
+	rval = basename(path);
+	if (rval == NULL)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "basename failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_chdir(const char *file, const int lineno, void (*cleanup_fn)(void),
+    const char *path)
+{
+	int rval;
+
+	rval = chdir(path);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "chdir failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_close(const char *file, const int lineno, void (*cleanup_fn)(void),
+    int fildes)
+{
+	int rval;
+
+	rval = close(fildes);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "close failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_creat(const char *file, const int lineno, void (*cleanup_fn)(void),
+    char *pathname, mode_t mode)
+{
+	int rval;
+
+	rval = creat(pathname, mode);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "pipe failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+char *
+safe_dirname(const char *file, const int lineno, void (*cleanup_fn)(void),
+    char *path)
+{
+	char *rval;
+
+	rval = dirname(path);
+	if (rval == NULL)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "dirname failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+char *
+safe_getcwd(const char *file, const int lineno, void (*cleanup_fn)(void),
+    char *buf, size_t size)
+{
+	char *rval;
+
+	rval = getcwd(buf, size);
+	if (rval == NULL)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "getcwd failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+struct passwd*
+safe_getpwnam(const char *file, const int lineno, void (*cleanup_fn)(void),
+    const char *name)
+{
+	struct passwd *rval;
+
+	rval = getpwnam(name);
+	if (rval == NULL)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "getpwnam failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_getrusage(const char *file, const int lineno, void (*cleanup_fn)(void),
+	    int who, struct rusage *usage)
+{
+	int rval;
+
+	rval = getrusage(who, usage);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "getrusage failed at %s:%d",
+		    file, lineno);
+
+	return rval;
+}
+
+void*
+safe_malloc(const char *file, const int lineno, void (*cleanup_fn)(void),
+    size_t size)
+{
+	void *rval;
+
+	rval = malloc(size);
+	if (rval == NULL)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "malloc failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_mkdir(const char *file, const int lineno, void (*cleanup_fn)(void),
+    const char *pathname, mode_t mode)
+{
+	int rval;
+
+	rval = mkdir(pathname, mode);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "mkdir failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+void*
+safe_mmap(const char *file, const int lineno, void (*cleanup_fn)(void),
+    void *addr, size_t length, int prot, int flags, int fd, off_t offset)
+{
+	void *rval;
+
+	rval = mmap(addr, length, prot, flags, fd, offset);
+	if (rval == MAP_FAILED)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "mmap failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_munmap(const char *file, const int lineno, void (*cleanup_fn)(void),
+    void *addr, size_t length)
+{
+	int rval;
+
+	rval = munmap(addr, length);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "munmap failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_open(const char *file, const int lineno, void (*cleanup_fn)(void),
+    const char *pathname, int oflags, ...)
+{
+	va_list ap;
+	int rval;
+	mode_t mode;
+
+	va_start(ap, oflags);
+	mode = va_arg(ap, mode_t);
+	va_end(ap);
+
+	rval = open(pathname, oflags, mode);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "open failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_pipe(const char *file, const int lineno, void (*cleanup_fn)(void),
+    int fildes[2])
+{
+	int rval;
+
+	rval = pipe(fildes);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "pipe failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+ssize_t
+safe_read(const char *file, const int lineno, void (*cleanup_fn)(void),
+    char len_strict, int fildes, void *buf, size_t nbyte)
+{
+	ssize_t rval;
+
+	rval = read(fildes, buf, nbyte);
+	if ((len_strict == 0 && rval == -1) || rval != nbyte)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "read failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+	int
+safe_setegid(const char *file, const int lineno, void (*cleanup_fn)(void),
+    gid_t egid)
+{
+	int rval;
+
+	rval = setegid(egid);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "setegid failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_seteuid(const char *file, const int lineno, void (*cleanup_fn)(void),
+    uid_t euid)
+{
+	int rval;
+
+	rval = seteuid(euid);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "seteuid failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_setgid(const char *file, const int lineno, void (*cleanup_fn)(void),
+    gid_t gid)
+{
+	int rval;
+
+	rval = setgid(gid);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "setgid failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_setuid(const char *file, const int lineno, void (*cleanup_fn)(void),
+    uid_t uid)
+{
+	int rval;
+
+	rval = setuid(uid);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "setuid failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+int
+safe_unlink(const char *file, const int lineno, void (*cleanup_fn)(void),
+    const char *pathname)
+{
+	int rval;
+
+	rval = unlink(pathname);
+	if (rval == -1)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "unlink failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}
+
+ssize_t
+safe_write(const char *file, const int lineno, void (cleanup_fn)(void),
+    char len_strict, int fildes, const void *buf, size_t nbyte)
+{
+	ssize_t rval;
+
+	rval = write(fildes, buf, nbyte);
+	if ((len_strict == 0 && rval == -1) || rval != nbyte)
+		tst_brkm(TBROK|TERRNO, cleanup_fn, "write failed at %s:%d",
+		    file, lineno);
+
+	return (rval);
+}