nsz Git - musl/blob - src/malloc/memalign.c

   1 #include <stdlib.h>
   2 #include <stdint.h>
   3 #include <errno.h>
   4 #include "libc.h"
   5
   6 void *__memalign(size_t align, size_t len)
   7 {
   8         unsigned char *mem, *new, *end;
   9         size_t header, footer;
  10
  11         if ((align & -align) != align) {
  12                 errno = EINVAL;
  13                 return NULL;
  14         }
  15
  16         if (len > SIZE_MAX - align) {
  17                 errno = ENOMEM;
  18                 return NULL;
  19         }
  20
  21         if (align <= 4*sizeof(size_t)) {
  22                 if (!(mem = malloc(len)))
  23                         return NULL;
  24                 return mem;
  25         }
  26
  27         if (!(mem = malloc(len + align-1)))
  28                 return NULL;
  29
  30         new = (void *)((uintptr_t)mem + align-1 & -align);
  31         if (new == mem) return mem;
  32
  33         header = ((size_t *)mem)[-1];
  34
  35         if (!(header & 7)) {
  36                 ((size_t *)new)[-2] = ((size_t *)mem)[-2] + (new-mem);
  37                 ((size_t *)new)[-1] = ((size_t *)mem)[-1] - (new-mem);
  38                 return new;
  39         }
  40
  41         end = mem + (header & -8);
  42         footer = ((size_t *)end)[-2];
  43
  44         ((size_t *)mem)[-1] = header&7 | new-mem;
  45         ((size_t *)new)[-2] = footer&7 | new-mem;
  46         ((size_t *)new)[-1] = header&7 | end-new;
  47         ((size_t *)end)[-2] = footer&7 | end-new;
  48
  49         free(mem);
  50         return new;
  51 }
  52
  53 weak_alias(__memalign, memalign);