9ec226da9c74b3e7dde17f9982358770fdc5b3c1
[ccan] / ccan / mem / test / api.c
1 #include <ccan/array_size/array_size.h>
2 #include <ccan/mem/mem.h>
3 #include <ccan/tap/tap.h>
4
5 int main(void)
6 {
7         char haystack1[] = "abcd\0efgh";
8         char haystack2[] = "ab\0ab\0ab\0ab";
9         char needle1[] = "ab";
10         char needle2[] = "d\0e";
11         char scan1[] = "aaaab";
12         char scan2[] = "\0\0\0b";
13
14         /* This is how many tests you plan to run */
15         plan_tests(46);
16
17         ok1(memmem(haystack1, sizeof(haystack1), needle1, 2) == haystack1);
18         ok1(memmem(haystack1, sizeof(haystack1), needle1, 3) == NULL);
19         ok1(memmem(haystack1, sizeof(haystack1), needle2, 3) == (haystack1 + 3));
20
21         ok1(memmem(haystack2, sizeof(haystack2), needle1, sizeof(needle1))
22             == haystack2);
23         ok1(memmem(haystack2, sizeof(haystack2), needle2, 3) == NULL);
24
25         ok1(memrchr(haystack1, 'a', sizeof(haystack1)) == haystack1);
26         ok1(memrchr(haystack1, 'b', sizeof(haystack1)) == haystack1 + 1);
27         ok1(memrchr(haystack1, 'c', sizeof(haystack1)) == haystack1 + 2);
28         ok1(memrchr(haystack1, 'd', sizeof(haystack1)) == haystack1 + 3);
29         ok1(memrchr(haystack1, 'e', sizeof(haystack1)) == haystack1 + 5);
30         ok1(memrchr(haystack1, 'f', sizeof(haystack1)) == haystack1 + 6);
31         ok1(memrchr(haystack1, 'g', sizeof(haystack1)) == haystack1 + 7);
32         ok1(memrchr(haystack1, 'h', sizeof(haystack1)) == haystack1 + 8);
33         ok1(memrchr(haystack1, '\0', sizeof(haystack1)) == haystack1 + 9);
34         ok1(memrchr(haystack1, 'i', sizeof(haystack1)) == NULL);
35
36         ok1(memrchr(haystack2, 'a', sizeof(haystack2)) == haystack2 + 9);
37         ok1(memrchr(haystack2, 'b', sizeof(haystack2)) == haystack2 + 10);
38         ok1(memrchr(haystack2, '\0', sizeof(haystack2)) == haystack2 + 11);
39
40         ok1(memrchr(needle1, '\0', 2) == NULL);
41
42 #define S(x) (x), sizeof(x) - 1
43         ok1(mempbrkm(S(haystack1), S("\0efgh")) == haystack1 + 4);
44         ok1(mempbrkm(S(haystack1), S("jklmn")) == NULL);
45         ok1(mempbrkm(S(haystack1), S("sd\0a")) == haystack1 + 0);
46
47         ok1(mempbrk(haystack1, sizeof(haystack1), "bcd\0a") == haystack1 + 1);
48         ok1(mempbrk(haystack1, sizeof(haystack1), "\0") == NULL);
49
50         ok1(memcchr(scan1, 'a', sizeof(scan1)) == scan1 + 4);
51         ok1(memcchr(scan1, 'b', sizeof(scan1)) == scan1);
52         ok1(memcchr(scan2, '\0', sizeof(scan2)) == scan2 + 3);
53         ok1(memcchr(scan2, '\0', sizeof(scan2) - 2) == NULL);
54
55         ok1(memeq(haystack1, sizeof(haystack1), haystack1, sizeof(haystack1)));
56         ok1(!memeq(haystack1, sizeof(haystack1), haystack2, sizeof(haystack2)));
57
58         ok1(memeqstr(scan1, sizeof(scan1) - 1, scan1));
59         ok1(!memeqstr(scan1, sizeof(scan1), scan1));
60         ok1(!memeqstr(scan1, sizeof(scan1), "aaaa"));
61
62         ok1(memstarts(S("a\0bcdef"), S("a\0bc")));
63         ok1(!memstarts(S("a\0bcdef"), S("a\0bcG")));
64         ok1(!memstarts(S("a\0bcdef"), S("a\0bcdefg")));
65
66         ok1(memstarts_str(scan1, sizeof(scan1), scan1));
67         ok1(!memstarts_str(scan1, sizeof(scan1), "ab"));
68
69         ok1(memends(S("abcdef"), S("abcdef")));
70         ok1(!memends(S("abcdef"), S("abcdefg")));
71         ok1(!memends(S("a\0bcdef"), S("a\0b")));
72         ok1(memends(S("a\0bcdef"), S("ef")));
73
74         ok1(memends_str(S("abcdef"), "abcdef"));
75         ok1(!memends_str(S("abcde\0f"), "d\0f"));
76         ok1(!memends_str(S("a\0bcdef"), "a"));
77         ok1(memends_str(S("a\0bcdef"), "ef"));
78
79         /* This exits depending on whether all tests passed */
80         return exit_status();
81 }