5 #include <ccan/mem/mem.h>
6 #include <ccan/tap/tap.h>
12 char haystack1[] = "abcd\0efgh";
13 char haystack2[] = "ab\0ab\0ab\0ab";
14 char needle1[] = "ab";
15 char needle2[] = "d\0e";
16 char scan1[] = "aaaab";
17 char scan2[] = "\0\0\0b";
18 char tmp1[SWAPSIZE], tmp2[SWAPSIZE];
20 /* This is how many tests you plan to run */
23 ok1(memmem(haystack1, sizeof(haystack1), needle1, 2) == haystack1);
24 ok1(memmem(haystack1, sizeof(haystack1), needle1, 3) == NULL);
25 ok1(memmem(haystack1, sizeof(haystack1), needle2, 3) == (haystack1 + 3));
27 ok1(memmem(haystack2, sizeof(haystack2), needle1, sizeof(needle1))
29 ok1(memmem(haystack2, sizeof(haystack2), needle2, 3) == NULL);
31 ok1(memrchr(haystack1, 'a', sizeof(haystack1)) == haystack1);
32 ok1(memrchr(haystack1, 'b', sizeof(haystack1)) == haystack1 + 1);
33 ok1(memrchr(haystack1, 'c', sizeof(haystack1)) == haystack1 + 2);
34 ok1(memrchr(haystack1, 'd', sizeof(haystack1)) == haystack1 + 3);
35 ok1(memrchr(haystack1, 'e', sizeof(haystack1)) == haystack1 + 5);
36 ok1(memrchr(haystack1, 'f', sizeof(haystack1)) == haystack1 + 6);
37 ok1(memrchr(haystack1, 'g', sizeof(haystack1)) == haystack1 + 7);
38 ok1(memrchr(haystack1, 'h', sizeof(haystack1)) == haystack1 + 8);
39 ok1(memrchr(haystack1, '\0', sizeof(haystack1)) == haystack1 + 9);
40 ok1(memrchr(haystack1, 'i', sizeof(haystack1)) == NULL);
42 ok1(memrchr(haystack2, 'a', sizeof(haystack2)) == haystack2 + 9);
43 ok1(memrchr(haystack2, 'b', sizeof(haystack2)) == haystack2 + 10);
44 ok1(memrchr(haystack2, '\0', sizeof(haystack2)) == haystack2 + 11);
46 ok1(memrchr(needle1, '\0', 2) == NULL);
48 #define S(x) (x), sizeof(x) - 1
49 ok1(mempbrkm(S(haystack1), S("\0efgh")) == haystack1 + 4);
50 ok1(mempbrkm(S(haystack1), S("jklmn")) == NULL);
51 ok1(mempbrkm(S(haystack1), S("sd\0a")) == haystack1 + 0);
53 ok1(mempbrk(haystack1, sizeof(haystack1), "bcd\0a") == haystack1 + 1);
54 ok1(mempbrk(haystack1, sizeof(haystack1), "\0") == NULL);
56 ok1(memcchr(scan1, 'a', sizeof(scan1)) == scan1 + 4);
57 ok1(memcchr(scan1, 'b', sizeof(scan1)) == scan1);
58 ok1(memcchr(scan2, '\0', sizeof(scan2)) == scan2 + 3);
59 ok1(memcchr(scan2, '\0', sizeof(scan2) - 2) == NULL);
61 ok1(memeq(haystack1, sizeof(haystack1), haystack1, sizeof(haystack1)));
62 ok1(!memeq(haystack1, sizeof(haystack1), haystack2, sizeof(haystack2)));
64 ok1(memeqstr(scan1, sizeof(scan1) - 1, scan1));
65 ok1(!memeqstr(scan1, sizeof(scan1), scan1));
66 ok1(!memeqstr(scan1, sizeof(scan1), "aaaa"));
68 ok1(memstarts(S("a\0bcdef"), S("a\0bc")));
69 ok1(!memstarts(S("a\0bcdef"), S("a\0bcG")));
70 ok1(!memstarts(S("a\0bcdef"), S("a\0bcdefg")));
72 ok1(memstarts_str(scan1, sizeof(scan1), scan1));
73 ok1(!memstarts_str(scan1, sizeof(scan1), "ab"));
75 ok1(memends(S("abcdef"), S("abcdef")));
76 ok1(!memends(S("abcdef"), S("abcdefg")));
77 ok1(!memends(S("a\0bcdef"), S("a\0b")));
78 ok1(memends(S("a\0bcdef"), S("ef")));
80 ok1(memends_str(S("abcdef"), "abcdef"));
81 ok1(!memends_str(S("abcde\0f"), "d\0f"));
82 ok1(!memends_str(S("a\0bcdef"), "a"));
83 ok1(memends_str(S("a\0bcdef"), "ef"));
85 ok1(!memoverlaps(haystack1, sizeof(haystack1),
86 haystack2, sizeof(haystack2)));
87 ok1(!memoverlaps(haystack2, sizeof(haystack2),
88 haystack1, sizeof(haystack1)));
89 ok1(memoverlaps(haystack1, sizeof(haystack1), haystack1, 1));
90 ok1(memoverlaps(haystack1, 1, haystack1, sizeof(haystack1)));
91 ok1(memoverlaps(haystack1, sizeof(haystack1),
92 haystack1 + sizeof(haystack1) - 1, 1));
93 ok1(memoverlaps(haystack1 + sizeof(haystack1) - 1, 1,
94 haystack1, sizeof(haystack1)));
95 ok1(!memoverlaps(haystack1, sizeof(haystack1),
96 haystack1 + sizeof(haystack1), 1));
97 ok1(!memoverlaps(haystack1 + sizeof(haystack1), 1,
98 haystack1, sizeof(haystack1)));
99 ok1(!memoverlaps(haystack1, sizeof(haystack1), haystack1 - 1, 1));
100 ok1(!memoverlaps(haystack1 - 1, 1, haystack1, sizeof(haystack1)));
101 ok1(memoverlaps(haystack1, 5, haystack1 + 4, 7));
102 ok1(!memoverlaps(haystack1, 5, haystack1 + 5, 6));
103 ok1(memoverlaps(haystack1 + 4, 7, haystack1, 5));
104 ok1(!memoverlaps(haystack1 + 5, 6, haystack1, 5));
106 assert(sizeof(haystack1) <= SWAPSIZE);
107 assert(sizeof(haystack2) <= SWAPSIZE);
108 memset(tmp1, 0, sizeof(tmp1));
109 memset(tmp2, 0, sizeof(tmp2));
110 memcpy(tmp1, haystack1, sizeof(haystack1));
111 memcpy(tmp2, haystack2, sizeof(haystack2));
112 memswap(tmp1, tmp2, SWAPSIZE);
113 ok1(memcmp(tmp1, haystack2, sizeof(haystack2)) == 0);
114 ok1(memcmp(tmp2, haystack1, sizeof(haystack1)) == 0);
116 ok1(memeqzero(NULL, 0));
117 ok1(memeqzero(scan2, 3));
118 ok1(!memeqzero(scan2, 4));
120 /* This exits depending on whether all tests passed */
121 return exit_status();