git.ozlabs.org Git - ccan/blob - 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
  12         /* This is how many tests you plan to run */
  13         plan_tests(19);
  14
  15         ok1(memmem(haystack1, sizeof(haystack1), needle1, 2) == haystack1);
  16         ok1(memmem(haystack1, sizeof(haystack1), needle1, 3) == NULL);
  17         ok1(memmem(haystack1, sizeof(haystack1), needle2, 3) == (haystack1 + 3));
  18
  19         ok1(memmem(haystack2, sizeof(haystack2), needle1, sizeof(needle1))
  20             == haystack2);
  21         ok1(memmem(haystack2, sizeof(haystack2), needle2, 3) == NULL);
  22
  23         ok1(memrchr(haystack1, 'a', sizeof(haystack1)) == haystack1);
  24         ok1(memrchr(haystack1, 'b', sizeof(haystack1)) == haystack1 + 1);
  25         ok1(memrchr(haystack1, 'c', sizeof(haystack1)) == haystack1 + 2);
  26         ok1(memrchr(haystack1, 'd', sizeof(haystack1)) == haystack1 + 3);
  27         ok1(memrchr(haystack1, 'e', sizeof(haystack1)) == haystack1 + 5);
  28         ok1(memrchr(haystack1, 'f', sizeof(haystack1)) == haystack1 + 6);
  29         ok1(memrchr(haystack1, 'g', sizeof(haystack1)) == haystack1 + 7);
  30         ok1(memrchr(haystack1, 'h', sizeof(haystack1)) == haystack1 + 8);
  31         ok1(memrchr(haystack1, '\0', sizeof(haystack1)) == haystack1 + 9);
  32         ok1(memrchr(haystack1, 'i', sizeof(haystack1)) == NULL);
  33
  34         ok1(memrchr(haystack2, 'a', sizeof(haystack2)) == haystack2 + 9);
  35         ok1(memrchr(haystack2, 'b', sizeof(haystack2)) == haystack2 + 10);
  36         ok1(memrchr(haystack2, '\0', sizeof(haystack2)) == haystack2 + 11);
  37
  38         ok1(memrchr(needle1, '\0', 2) == NULL);
  39
  40         /* This exits depending on whether all tests passed */
  41         return exit_status();
  42 }