Return to xmalloc.c CVS log

Up to [local] / src / usr.bin / cvs

Annotation of src/usr.bin/cvs/xmalloc.c, Revision 1.11

1.11    ! millert     1: /* $OpenBSD: xmalloc.c,v 1.10 2014/12/01 21:58:46 deraadt Exp $ */
1.1       joris       2: /*
                      3:  * Author: Tatu Ylonen <ylo@cs.hut.fi>
                      4:  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
                      5:  *                    All rights reserved
1.6       ray         6:  * Versions of malloc and friends that check their results, and never return
1.1       joris       7:  * failure (they call fatal if they encounter an error).
                      8:  *
                      9:  * As far as I am concerned, the code I have written for this software
                     10:  * can be used freely for any purpose.  Any derived versions of this
                     11:  * software must be clearly marked as such, and if the derived work is
                     12:  * incompatible with the protocol description in the RFC file, it must be
                     13:  * called by a name other than "ssh" or "Secure Shell".
                     14:  */
                     15:
1.11    ! millert    16: #include <stdint.h>
1.8       otto       17: #include <stdio.h>
                     18: #include <stdlib.h>
                     19: #include <string.h>
1.1       joris      20:
1.8       otto       21: #include "log.h"
1.3       xsa        22: #include "xmalloc.h"
1.1       joris      23:
                     24: void *
                     25: xmalloc(size_t size)
                     26: {
                     27:        void *ptr;
                     28:
                     29:        if (size == 0)
                     30:                fatal("xmalloc: zero size");
                     31:        ptr = malloc(size);
                     32:        if (ptr == NULL)
1.6       ray        33:                fatal("xmalloc: out of memory (allocating %lu bytes)", (u_long) size);
1.1       joris      34:        return ptr;
                     35: }
                     36:
                     37: void *
1.6       ray        38: xcalloc(size_t nmemb, size_t size)
                     39: {
                     40:        void *ptr;
                     41:
                     42:        if (size == 0 || nmemb == 0)
                     43:                fatal("xcalloc: zero size");
1.9       ray        44:        if (SIZE_MAX / nmemb < size)
                     45:                fatal("xcalloc: nmemb * size > SIZE_MAX");
1.6       ray        46:        ptr = calloc(nmemb, size);
                     47:        if (ptr == NULL)
                     48:                fatal("xcalloc: out of memory (allocating %lu bytes)",
                     49:                    (u_long)(size * nmemb));
                     50:        return ptr;
                     51: }
                     52:
                     53: void *
1.10      deraadt    54: xreallocarray(void *ptr, size_t nmemb, size_t size)
1.1       joris      55: {
                     56:        void *new_ptr;
1.6       ray        57:        size_t new_size = nmemb * size;
1.1       joris      58:
                     59:        if (new_size == 0)
                     60:                fatal("xrealloc: zero size");
1.9       ray        61:        if (SIZE_MAX / nmemb < size)
                     62:                fatal("xrealloc: nmemb * size > SIZE_MAX");
1.10      deraadt    63:        new_ptr = realloc(ptr, new_size);
1.1       joris      64:        if (new_ptr == NULL)
1.4       reyk       65:                fatal("xrealloc: out of memory (new_size %lu bytes)",
                     66:                    (u_long) new_size);
1.1       joris      67:        return new_ptr;
                     68: }
                     69:
                     70: void
                     71: xfree(void *ptr)
                     72: {
                     73:        if (ptr == NULL)
                     74:                fatal("xfree: NULL pointer given as argument");
                     75:        free(ptr);
                     76: }
                     77:
                     78: char *
                     79: xstrdup(const char *str)
                     80: {
                     81:        size_t len;
                     82:        char *cp;
                     83:
                     84:        len = strlen(str) + 1;
                     85:        cp = xmalloc(len);
                     86:        strlcpy(cp, str, len);
                     87:        return cp;
1.6       ray        88: }
                     89:
                     90: int
                     91: xasprintf(char **ret, const char *fmt, ...)
                     92: {
                     93:        va_list ap;
                     94:        int i;
                     95:
                     96:        va_start(ap, fmt);
                     97:        i = vasprintf(ret, fmt, ap);
                     98:        va_end(ap);
                     99:
                    100:        if (i < 0 || *ret == NULL)
                    101:                fatal("xasprintf: could not allocate memory");
1.7       xsa       102:
                    103:        return (i);
                    104: }
                    105:
                    106: int
                    107: xsnprintf(char *str, size_t size, const char *fmt, ...)
                    108: {
                    109:        va_list ap;
                    110:        int i;
                    111:
                    112:        va_start(ap, fmt);
                    113:        i = vsnprintf(str, size, fmt, ap);
                    114:        va_end(ap);
                    115:
                    116:        if (i == -1 || i >= (int)size)
                    117:                fatal("xsnprintf: overflow");
1.6       ray       118:
                    119:        return (i);
1.1       joris     120: }