1 files changed, 21 insertions, 0 deletions
diff --git a/malloc/beware_overflow/main.c b/malloc/beware_overflow/main.c
new file mode 100644
index 0000000..42fd689
--- /dev/null
+++ b/malloc/beware_overflow/main.c
@@ -0,0 +1,21 @@
+#include <stddef.h>
+#include <stdio.h>
+
+#include "beware_overflow.h"
+
+void print_overflow(void *ptr, size_t nmemb, size_t size)
+{
+    void *begin = ptr;
+    void *res = beware_overflow(ptr, nmemb, size);
+    if (res)
+        printf("Pointer was incremented from %p to %p.\n", begin, res);
+    else
+        printf("Overflow detected between %ld and %ld.\n", nmemb, size);
+}
+
+int main(void)
+{
+    print_overflow((void *)0x1000, 25, 6);
+    print_overflow((void *)0x40000, 12345678904, 12345678904);
+    return 0;
+}