ptr_valid: test whether a ptr is valid.

[ccan] / ccan / ptr_valid / test / run-string.c

diff --git a/ccan/ptr_valid/test/run-string.c b/ccan/ptr_valid/test/run-string.c
new file mode 100644 (file)
index 0000000..59d36a4
--- /dev/null
+++ b/ccan/ptr_valid/test/run-string.c
@@ -0,0 +1,56 @@
+#include <ccan/ptr_valid/ptr_valid.h>
+/* Include the C files directly. */
+#include <ccan/ptr_valid/ptr_valid.c>
+#include <ccan/tap/tap.h>
+#include <sys/mman.h>
+
+int main(void)
+{
+       char *page;
+       struct ptr_valid_batch *batch = malloc(sizeof *batch);
+
+       /* This is how many tests you plan to run */
+       plan_tests(14);
+
+       page = mmap(NULL, getpagesize(), PROT_READ|PROT_WRITE,
+                   MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
+       strcpy(page, "hello");
+       ok1(ptr_valid_read(page));
+       ok1(ptr_valid_write(page));
+       ok1(ptr_valid_string(page));
+
+       ok1(ptr_valid_batch_start(batch));
+       ok1(ptr_valid_batch_string(batch, page));
+       ptr_valid_batch_end(batch);
+
+       /* Check invalid case. */
+       munmap(page, getpagesize());
+       ok1(!ptr_valid_string(page));
+
+       ok1(ptr_valid_batch_start(batch));
+       ok1(!ptr_valid_batch_string(batch, page));
+       ptr_valid_batch_end(batch);
+
+       /* Check for overrun. */
+       page = mmap(NULL, getpagesize()*2, PROT_READ|PROT_WRITE,
+                   MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
+       munmap(page + getpagesize(), getpagesize());
+
+       memset(page, 'a', getpagesize());
+       ok1(!ptr_valid_string(page));
+       ok1(ptr_valid_batch_start(batch));
+       ok1(!ptr_valid_batch_string(batch, page));
+       ptr_valid_batch_end(batch);
+
+       page[getpagesize()-1] = '\0';
+       ok1(ptr_valid_string(page));
+
+       ok1(ptr_valid_batch_start(batch));
+       ok1(ptr_valid_batch_string(batch, page));
+       ptr_valid_batch_end(batch);
+       munmap(page, getpagesize());
+
+       free(batch);
+       /* This exits depending on whether all tests passed */
+       return exit_status();
+}