git.ozlabs.org Git - ccan/blob - ccan/tal/test/run-overflow.c

   1 #include <ccan/tal/tal.h>
   2 #include <ccan/tal/tal.c>
   3 #include <ccan/tap/tap.h>
   4
   5 static int error_count;
   6
   7 static void my_error(const char *msg)
   8 {
   9         error_count++;
  10 }
  11
  12 int main(void)
  13 {
  14         void *p;
  15         int *pi, *origpi;
  16
  17         plan_tests(26);
  18
  19         tal_set_backend(NULL, NULL, NULL, my_error);
  20
  21         p = tal_arr(NULL, int, (size_t)-1);
  22         ok1(!p);
  23         ok1(error_count == 1);
  24
  25         p = tal_arr(NULL, char, (size_t)-2);
  26         ok1(!p);
  27         ok1(error_count == 2);
  28
  29         /* Now try overflow cases for tal_dup. */
  30         error_count = 0;
  31         pi = origpi = tal_arr(NULL, int, 100);
  32         ok1(pi);
  33         ok1(error_count == 0);
  34         pi = tal_dup(NULL, int, pi, (size_t)-1, 0);
  35         ok1(!pi);
  36         ok1(error_count == 1);
  37         pi = tal_dup(NULL, int, pi, 0, (size_t)-1);
  38         ok1(!pi);
  39         ok1(error_count == 2);
  40
  41         pi = tal_dup(NULL, int, pi, (size_t)-1UL / sizeof(int),
  42                      (size_t)-1UL / sizeof(int));
  43         ok1(!pi);
  44         ok1(error_count == 3);
  45         /* This will still overflow when tal_hdr is added. */
  46         pi = tal_dup(NULL, int, pi, (size_t)-1UL / sizeof(int) / 2,
  47                      (size_t)-1UL / sizeof(int) / 2);
  48         ok1(!pi);
  49         ok1(error_count == 4);
  50
  51         /* Now, check that with TAL_TAKE we free old one on failure. */
  52         pi = tal_arr(NULL, int, 100);
  53         error_count = 0;
  54         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1, 0);
  55         ok1(!pi);
  56         ok1(error_count == 1);
  57         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
  58
  59         pi = tal_arr(NULL, int, 100);
  60         error_count = 0;
  61         pi = tal_dup(TAL_TAKE, int, pi, 0, (size_t)-1);
  62         ok1(!pi);
  63         ok1(error_count == 1);
  64         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
  65
  66         pi = tal_arr(NULL, int, 100);
  67         error_count = 0;
  68         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1UL / sizeof(int),
  69                      (size_t)-1UL / sizeof(int));
  70         ok1(!pi);
  71         ok1(error_count == 1);
  72         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
  73
  74         pi = tal_arr(NULL, int, 100);
  75         error_count = 0;
  76         /* This will still overflow when tal_hdr is added. */
  77         pi = tal_dup(TAL_TAKE, int, pi, (size_t)-1UL / sizeof(int) / 2,
  78                      (size_t)-1UL / sizeof(int) / 2);
  79         ok1(!pi);
  80         ok1(error_count == 1);
  81         ok1(tal_first(NULL) == origpi && !tal_next(NULL, origpi));
  82
  83         return exit_status();
  84 }