1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License, Version 1.0 only 6 * (the "License"). You may not use this file except in compliance 7 * with the License. 8 * 9 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 * or http://www.opensolaris.org/os/licensing. 11 * See the License for the specific language governing permissions 12 * and limitations under the License. 13 * 14 * When distributing Covered Code, include this CDDL HEADER in each 15 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 * If applicable, add the following below this CDDL HEADER, with the 17 * fields enclosed by brackets "[]" replaced with your own identifying 18 * information: Portions Copyright [yyyy] [name of copyright owner] 19 * 20 * CDDL HEADER END 21 */ 22 /* Copyright (c) 1988 AT&T */ 23 /* All Rights Reserved */ 24 25 26 /* 27 * Copyright (c) 1999 by Sun Microsystems, Inc. 28 * All rights reserved. 29 */ 30 31 /* memory allocation functions */ 32 33 #include <stdio.h> 34 #include <string.h> 35 #include <stdlib.h> 36 37 extern char *argv0; /* command name (must be set in main function) */ 38 39 char *stralloc(char *s); 40 void *mymalloc(size_t size); 41 void *mycalloc(size_t nelem, size_t size); 42 void *myrealloc(void *p, size_t size); 43 static void *alloctest(void *p); 44 45 /* allocate a string */ 46 47 char * 48 stralloc(char *s) 49 { 50 return (strcpy(mymalloc(strlen(s) + 1), s)); 51 } 52 53 /* version of malloc that only returns if successful */ 54 55 void * 56 mymalloc(size_t size) 57 { 58 return (alloctest(malloc(size))); 59 } 60 61 /* version of calloc that only returns if successful */ 62 63 void * 64 mycalloc(size_t nelem, size_t size) 65 { 66 return (alloctest(calloc(nelem, size))); 67 } 68 69 /* version of realloc that only returns if successful */ 70 71 void * 72 myrealloc(void *p, size_t size) 73 { 74 return (alloctest(realloc(p, size))); 75 } 76 77 /* check for memory allocation failure */ 78 79 static void * 80 alloctest(void *p) 81 { 82 if (p == NULL) { 83 (void) fprintf(stderr, "\n%s: out of storage\n", argv0); 84 exit(1); 85 } 86 return (p); 87 } 88