mem: add memeqzero.

[ccan] / ccan / mem / test / api.c

diff --git a/ccan/mem/test/api.c b/ccan/mem/test/api.c
index 89b6acb41c49f5558435233a94afb4a800561ec6..59b25947ab0a96d5c15de578608d7a244f331708 100644 (file)
--- a/ccan/mem/test/api.c
+++ b/ccan/mem/test/api.c
@@ -1,7 +1,12 @@
-#include <ccan/array_size/array_size.h>
+#include "config.h"
+
+#include <assert.h>
+
 #include <ccan/mem/mem.h>
 #include <ccan/tap/tap.h>
 
+#define SWAPSIZE       12
+
 int main(void)
 {
        char haystack1[] = "abcd\0efgh";
@@ -10,9 +15,10 @@ int main(void)
        char needle2[] = "d\0e";
        char scan1[] = "aaaab";
        char scan2[] = "\0\0\0b";
+       char tmp1[SWAPSIZE], tmp2[SWAPSIZE];
 
        /* This is how many tests you plan to run */
-       plan_tests(42);
+       plan_tests(65);
 
        ok1(memmem(haystack1, sizeof(haystack1), needle1, 2) == haystack1);
        ok1(memmem(haystack1, sizeof(haystack1), needle1, 3) == NULL);
@@ -71,6 +77,45 @@ int main(void)
        ok1(!memends(S("a\0bcdef"), S("a\0b")));
        ok1(memends(S("a\0bcdef"), S("ef")));
 
+       ok1(memends_str(S("abcdef"), "abcdef"));
+       ok1(!memends_str(S("abcde\0f"), "d\0f"));
+       ok1(!memends_str(S("a\0bcdef"), "a"));
+       ok1(memends_str(S("a\0bcdef"), "ef"));
+
+       ok1(!memoverlaps(haystack1, sizeof(haystack1),
+                        haystack2, sizeof(haystack2)));
+       ok1(!memoverlaps(haystack2, sizeof(haystack2),
+                        haystack1, sizeof(haystack1)));
+       ok1(memoverlaps(haystack1, sizeof(haystack1), haystack1, 1));
+       ok1(memoverlaps(haystack1, 1, haystack1, sizeof(haystack1)));
+       ok1(memoverlaps(haystack1, sizeof(haystack1),
+                       haystack1 + sizeof(haystack1) - 1, 1));
+       ok1(memoverlaps(haystack1 + sizeof(haystack1) - 1, 1,
+                       haystack1, sizeof(haystack1)));
+       ok1(!memoverlaps(haystack1, sizeof(haystack1),
+                        haystack1 + sizeof(haystack1), 1));
+       ok1(!memoverlaps(haystack1 + sizeof(haystack1), 1,
+                        haystack1, sizeof(haystack1)));
+       ok1(!memoverlaps(haystack1, sizeof(haystack1), haystack1 - 1, 1));
+       ok1(!memoverlaps(haystack1 - 1, 1, haystack1, sizeof(haystack1)));
+       ok1(memoverlaps(haystack1, 5, haystack1 + 4, 7));
+       ok1(!memoverlaps(haystack1, 5, haystack1 + 5, 6));
+       ok1(memoverlaps(haystack1 + 4, 7, haystack1, 5));
+       ok1(!memoverlaps(haystack1 + 5, 6, haystack1, 5));
+
+       assert(sizeof(haystack1) <= SWAPSIZE);
+       assert(sizeof(haystack2) <= SWAPSIZE);
+       memset(tmp1, 0, sizeof(tmp1));
+       memset(tmp2, 0, sizeof(tmp2));
+       memcpy(tmp1, haystack1, sizeof(haystack1));
+       memcpy(tmp2, haystack2, sizeof(haystack2));
+       memswap(tmp1, tmp2, SWAPSIZE);
+       ok1(memcmp(tmp1, haystack2, sizeof(haystack2)) == 0);
+       ok1(memcmp(tmp2, haystack1, sizeof(haystack1)) == 0);
+
+       ok1(memeqzero(NULL, 0));
+       ok1(memeqzero(scan2, 3));
+       ok1(!memeqzero(scan2, 4));
 
        /* This exits depending on whether all tests passed */
        return exit_status();