Add argument validation to most libpsn00b functions

1 files changed, 19 insertions, 0 deletions

diff --git a/libpsn00b/include/assert.h b/libpsn00b/include/assert.h
index d18a56f..26b2023 100644
--- a/libpsn00b/include/assert.h
+++ b/libpsn00b/include/assert.h
@@ -25,6 +25,9 @@ void _assert_abort(const char *file, int line, const char *expr);
 
 #define assert(expr)
 #define _sdk_log(fmt, ...)
+#define _sdk_assert(expr, fmt, ...)
+#define _sdk_validate_args_void(expr)
+#define _sdk_validate_args(expr, ret)
 
 #else
 
@@ -39,6 +42,22 @@ void _assert_abort(const char *file, int line, const char *expr);
 	printf(fmt __VA_OPT__(,) __VA_ARGS__)
 #endif
 
+#define _sdk_assert(expr, ret, fmt, ...) \
+	if (!(expr)) { \
+		_sdk_log(fmt, __VA_ARGS__); \
+		return ret; \
+	}
+#define _sdk_validate_args_void(expr) \
+	if (!(expr)) { \
+		_sdk_log("invalid args to %s() (%s)\n", __func__, #expr); \
+		return; \
+	}
+#define _sdk_validate_args(expr, ret) \
+	if (!(expr)) { \
+		_sdk_log("invalid args to %s() (%s)\n", __func__, #expr); \
+		return ret; \
+	}
+
 #endif
 
 #endif