Changeset 2833 for trunk/kLdr/kLdrHlp.h

Timestamp:

Oct 26, 2006, 2:08:09 AM (19 years ago)

Author:

bird

Message:

Wrote the kLdrDyld verification routines and made it build on win32.

File:

• trunk/kLdr/kLdrHlp.h (modified) (1 diff)

trunk/kLdr/kLdrHlp.h

@@ -160 +160 @@
 
 
+/** @name Parameter validation macros
+ * @{ */
+
+/** Crash validation of a string argument. */
+#define KLDRHLP_VALIDATE_STRING(str) \
+    do { strlen(str); } while (0)
+
+/** Crash validation of an optional string argument. */
+#define KLDRHLP_VALIDATE_OPTIONAL_STRING(str) \
+    do { if (str) { KLDRHLP_VALIDATE_STRING(str); } } while (0)
+
+/** Return/Crash validation of an output buffer. */
+#define KLDRHLP_VALIDATE_BUFFER(buf, cb) \
+    do { \
+        if ((cb)) \
+        { \
+            uint8_t             __b; \
+            uint8_t volatile *  __pb = (uint8_t volatile *)(buf); \
+            size_t              __cbPage1 = 0x1000 - ((uintptr_t)(__pb) & 0xfff); /* ASSUMES page size! */ \
+            __b = *__pb; *__pb = 0xff; *__pb = __b; \
+            if ((cb) > __cbPage1) \
+            { \
+                size_t  __cb = (cb) - __cbPage1; \
+                __pb -= __cbPage1; \
+                for (;;) \
+                { \
+                    __b = *__pb; *__pb = 0xff; *__pb = __b; \
+                    if (__cb < 0x1000) \
+                        break; \
+                    __pb += 0x1000; \
+                    __cb -= 0x1000; \
+                } \
+            } \
+        } \
+        else \
+            return KLDR_ERR_INVALID_PARAMETER; \
+    } while (0)
+
+/** Crash validation of an optional output buffer. */
+#define KLDRHLP_VALIDATE_OPTIONAL_BUFFER(buf, cb) \
+    do { \
+        if ((buf) != NULL && (cb) != 0) \
+        { \
+            KLDRHLP_VALIDATE_BUFFER(buf, cb); \
+        } \
+    } while (0)
+
+/** Return validation of an enum argument. */
+#define KLDRHLP_VALIDATE_ENUM(arg, enumname) \
+    do { \
+        if ((arg) <= enumname##_INVALID || (arg) >= enumname##_END) \
+        {  \
+            return KLDR_ERR_INVALID_PARAMETER; \
+        } \
+    } while (0)
+
 /** @} */
 
 
+/** @} */
+
 #endif /* __kLdrHlp_h__ */