libexec/rtld-elf/xmalloc.c

   1 /*-
   2  * Copyright 1996-1998 John D. Polstra.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  *
  25  * $FreeBSD$
  26  */
  27
  28 #include <stddef.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <unistd.h>
  32 #include "rtld.h"
  33 #include "rtld_printf.h"
  34
  35 void *
  36 xcalloc(size_t number, size_t size)
  37 {
  38         void *p;
  39
  40         p = calloc(number, size);
  41         if (p == NULL) {
  42                 rtld_fdputstr(STDERR_FILENO, "Out of memory\n");
  43                 _exit(1);
  44         }
  45         return (p);
  46 }
  47
  48 void *
  49 xmalloc(size_t size)
  50 {
  51     void *p = malloc(size);
  52     if (p == NULL) {
  53         rtld_fdputstr(STDERR_FILENO, "Out of memory\n");
  54         _exit(1);
  55     }
  56     return p;
  57 }
  58
  59 char *
  60 xstrdup(const char *str)
  61 {
  62         char *copy;
  63         size_t len;
  64
  65         len = strlen(str) + 1;
  66         copy = xmalloc(len);
  67         memcpy(copy, str, len);
  68         return (copy);
  69 }
  70
  71 void *
  72 malloc_aligned(size_t size, size_t align)
  73 {
  74         void *mem, *res;
  75         uintptr_t x;
  76         size_t asize, r;
  77
  78         r = round(sizeof(void *), align);
  79         asize = round(size, align) + r;
  80         mem = xmalloc(asize);
  81         x = (uintptr_t)mem;
  82         res = (void *)round(x, align);
  83         *(void **)((uintptr_t)res - sizeof(void *)) = mem;
  84         return (res);
  85 }
  86
  87 void
  88 free_aligned(void *ptr)
  89 {
  90         void *mem;
  91         uintptr_t x;
  92
  93         if (ptr == NULL)
  94                 return;
  95         x = (uintptr_t)ptr;
  96         x -= sizeof(void *);
  97         mem = *(void **)x;
  98         free(mem);
  99 }