[BACK]Return to memory.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / make

Annotation of src/usr.bin/make/memory.c, Revision 1.6

1.1       espie       1: /* $OpenPackages$ */
1.5       espie       2: /* $OpenBSD$ */
1.1       espie       3:
                      4: /*
                      5:  * Copyright (c) 1988, 1989, 1990, 1993
                      6:  *     The Regents of the University of California.  All rights reserved.
                      7:  * Copyright (c) 1989 by Berkeley Softworks
                      8:  * All rights reserved.
                      9:  *
                     10:  * This code is derived from software contributed to Berkeley by
                     11:  * Adam de Boor.
                     12:  *
                     13:  * Redistribution and use in source and binary forms, with or without
                     14:  * modification, are permitted provided that the following conditions
                     15:  * are met:
                     16:  * 1. Redistributions of source code must retain the above copyright
                     17:  *    notice, this list of conditions and the following disclaimer.
                     18:  * 2. Redistributions in binary form must reproduce the above copyright
                     19:  *    notice, this list of conditions and the following disclaimer in the
                     20:  *    documentation and/or other materials provided with the distribution.
1.2       millert    21:  * 3. Neither the name of the University nor the names of its contributors
1.1       espie      22:  *    may be used to endorse or promote products derived from this software
                     23:  *    without specific prior written permission.
                     24:  *
                     25:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     26:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     27:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     28:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     29:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     30:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     31:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     32:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     33:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     34:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     35:  * SUCH DAMAGE.
                     36:  */
                     37:
                     38: #include <sys/types.h>
                     39: #include <sys/stat.h>
                     40: #include <errno.h>
                     41: #include <stdio.h>
                     42: #include <stdlib.h>
                     43: #include <string.h>
                     44: #include <unistd.h>
1.4       espie      45: #include <stddef.h>
                     46: #include <stdint.h>
                     47: #include <ohash.h>
1.1       espie      48: #include "defines.h"
                     49: #include "memory.h"
                     50:
                     51: static void enomem(size_t);
1.5       espie      52: static void enocmem(size_t, size_t);
1.1       espie      53:
                     54: /*
                     55:  * emalloc --
                     56:  *     malloc, but die on error.
                     57:  */
                     58: void *
1.3       espie      59: emalloc(size_t size)
1.1       espie      60: {
                     61:        void *p;
                     62:
1.3       espie      63:        if ((p = malloc(size)) == NULL)
                     64:                enomem(size);
1.1       espie      65:        return p;
                     66: }
                     67:
                     68: /*
                     69:  * estrdup --
                     70:  *     strdup, but die on error.
                     71:  */
                     72: char *
1.3       espie      73: estrdup(const char *str)
1.1       espie      74: {
                     75:        char *p;
                     76:        size_t size;
                     77:
                     78:        size = strlen(str) + 1;
                     79:
                     80:        p = emalloc(size);
                     81:        memcpy(p, str, size);
                     82:        return p;
                     83: }
                     84:
                     85: /*
                     86:  * erealloc --
                     87:  *     realloc, but die on error.
                     88:  */
                     89: void *
1.3       espie      90: erealloc(void *ptr, size_t size)
1.1       espie      91: {
                     92:        if ((ptr = realloc(ptr, size)) == NULL)
                     93:                enomem(size);
                     94:        return ptr;
                     95: }
                     96:
                     97: void *
1.3       espie      98: ecalloc(size_t s1, size_t s2)
1.1       espie      99: {
                    100:        void *p;
                    101:
                    102:        if ((p = calloc(s1, s2)) == NULL)
1.5       espie     103:                enocmem(s1, s2);
1.1       espie     104:        return p;
                    105: }
                    106:
1.5       espie     107: void *
1.6     ! espie     108: emult_realloc(void *ptr, size_t s1, size_t s2)
1.5       espie     109: {
1.6     ! espie     110:        size_t size;
        !           111:
        !           112:        if (s1 && SIZE_MAX / s1 < s2) {
        !           113:                errno = ENOMEM;
        !           114:                enocmem(s1, s2);
        !           115:        }
        !           116:        size = s1 * s2;
        !           117:        if ((ptr = realloc(ptr, size)) == NULL)
1.5       espie     118:                enocmem(s1, s2);
                    119:        return ptr;
                    120: }
                    121:
1.1       espie     122: /* Support routines for hash tables.  */
                    123: void *
1.3       espie     124: hash_alloc(size_t s, void *u UNUSED)
1.1       espie     125: {
                    126:        return ecalloc(s, 1);
                    127: }
                    128:
                    129: void
1.3       espie     130: hash_free(void *p, size_t s UNUSED, void *u UNUSED)
1.1       espie     131: {
                    132:        free(p);
                    133: }
                    134:
                    135: void *
1.3       espie     136: element_alloc(size_t s, void *u UNUSED)
1.1       espie     137: {
                    138:        return emalloc(s);
                    139: }
                    140:
                    141:
                    142:
                    143: /*
                    144:  * enomem --
                    145:  *     die when out of memory.
                    146:  */
                    147: void
1.3       espie     148: enomem(size_t size)
1.1       espie     149: {
1.5       espie     150:        fprintf(stderr, "make: %s (%zu)\n", strerror(errno), size);
1.1       espie     151:        exit(2);
                    152: }
                    153:
1.5       espie     154: void
                    155: enocmem(size_t sz1, size_t sz2)
                    156: {
                    157:        fprintf(stderr, "make: %s (%zu * %zu)\n", strerror(errno), sz1, sz2);
                    158:        exit(2);
                    159: }
1.1       espie     160: /*
                    161:  * esetenv --
                    162:  *     change environment, die on error.
                    163:  */
                    164: void
1.3       espie     165: esetenv(const char *name, const char *value)
1.1       espie     166: {
                    167:        if (setenv(name, value, 1) == 0)
                    168:            return;
                    169:
                    170:        fprintf(stderr, "make: setenv failed (%s)\n", strerror(errno));
                    171:        exit(2);
                    172: }
                    173:
                    174:
                    175: /*
                    176:  * enunlink --
                    177:  *     Remove a file carefully, avoiding directories.
                    178:  */
                    179: int
1.3       espie     180: eunlink(const char *file)
1.1       espie     181: {
                    182:        struct stat st;
                    183:
                    184:        if (lstat(file, &st) == -1)
                    185:                return -1;
                    186:
                    187:        if (S_ISDIR(st.st_mode)) {
                    188:                errno = EISDIR;
                    189:                return -1;
                    190:        }
                    191:        return unlink(file);
1.4       espie     192: }
                    193:
                    194: void
                    195: free_hash(struct ohash *h)
                    196: {
                    197:        void *e;
                    198:        unsigned int i;
                    199:
                    200:        for (e = ohash_first(h, &i); e != NULL; e = ohash_next(h, &i))
                    201:                free(e);
                    202:        ohash_delete(h);
1.1       espie     203: }
                    204: