src/jemalloc/test/integration/mallocx.c

   1 #include "test/jemalloc_test.h"
   2
   3 #define CHUNK 0x400000
   4 #define MAXALIGN (((size_t)1) << 25)
   5 #define NITER 4
   6
   7 TEST_BEGIN(test_basic)
   8 {
   9         size_t nsz, rsz, sz;
  10         void *p;
  11
  12         sz = 42;
  13         nsz = nallocx(sz, 0);
  14         assert_zu_ne(nsz, 0, "Unexpected nallocx() error");
  15         p = mallocx(sz, 0);
  16         assert_ptr_not_null(p, "Unexpected mallocx() error");
  17         rsz = sallocx(p, 0);
  18         assert_zu_ge(rsz, sz, "Real size smaller than expected");
  19         assert_zu_eq(nsz, rsz, "nallocx()/sallocx() size mismatch");
  20         dallocx(p, 0);
  21
  22         p = mallocx(sz, 0);
  23         assert_ptr_not_null(p, "Unexpected mallocx() error");
  24         dallocx(p, 0);
  25
  26         nsz = nallocx(sz, MALLOCX_ZERO);
  27         assert_zu_ne(nsz, 0, "Unexpected nallocx() error");
  28         p = mallocx(sz, MALLOCX_ZERO);
  29         assert_ptr_not_null(p, "Unexpected mallocx() error");
  30         rsz = sallocx(p, 0);
  31         assert_zu_eq(nsz, rsz, "nallocx()/sallocx() rsize mismatch");
  32         dallocx(p, 0);
  33 }
  34 TEST_END
  35
  36 TEST_BEGIN(test_alignment_and_size)
  37 {
  38         size_t nsz, rsz, sz, alignment, total;
  39         unsigned i;
  40         void *ps[NITER];
  41
  42         for (i = 0; i < NITER; i++)
  43                 ps[i] = NULL;
  44
  45         for (alignment = 8;
  46             alignment <= MAXALIGN;
  47             alignment <<= 1) {
  48                 total = 0;
  49                 for (sz = 1;
  50                     sz < 3 * alignment && sz < (1U << 31);
  51                     sz += (alignment >> (LG_SIZEOF_PTR-1)) - 1) {
  52                         for (i = 0; i < NITER; i++) {
  53                                 nsz = nallocx(sz, MALLOCX_ALIGN(alignment) |
  54                                     MALLOCX_ZERO);
  55                                 assert_zu_ne(nsz, 0,
  56                                     "nallocx() error for alignment=%zu, "
  57                                     "size=%zu (%#zx)", alignment, sz, sz);
  58                                 ps[i] = mallocx(sz, MALLOCX_ALIGN(alignment) |
  59                                     MALLOCX_ZERO);
  60                                 assert_ptr_not_null(ps[i],
  61                                     "mallocx() error for alignment=%zu, "
  62                                     "size=%zu (%#zx)", alignment, sz, sz);
  63                                 rsz = sallocx(ps[i], 0);
  64                                 assert_zu_ge(rsz, sz,
  65                                     "Real size smaller than expected for "
  66                                     "alignment=%zu, size=%zu", alignment, sz);
  67                                 assert_zu_eq(nsz, rsz,
  68                                     "nallocx()/sallocx() size mismatch for "
  69                                     "alignment=%zu, size=%zu", alignment, sz);
  70                                 assert_ptr_null(
  71                                     (void *)((uintptr_t)ps[i] & (alignment-1)),
  72                                     "%p inadequately aligned for"
  73                                     " alignment=%zu, size=%zu", ps[i],
  74                                     alignment, sz);
  75                                 total += rsz;
  76                                 if (total >= (MAXALIGN << 1))
  77                                         break;
  78                         }
  79                         for (i = 0; i < NITER; i++) {
  80                                 if (ps[i] != NULL) {
  81                                         dallocx(ps[i], 0);
  82                                         ps[i] = NULL;
  83                                 }
  84                         }
  85                 }
  86         }
  87 }
  88 TEST_END
  89
  90 int
  91 main(void)
  92 {
  93
  94         return (test(
  95             test_basic,
  96             test_alignment_and_size));
  97 }