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 #pragma ident "%Z%%M% %I% %E% SMI" 32 33 /* memory allocation functions */ 34 35 #include <stdio.h> 36 #include <string.h> 37 #include <stdlib.h> 38 39 extern char *argv0; /* command name (must be set in main function) */ 40 41 char *stralloc(char *s); 42 void *mymalloc(size_t size); 43 void *mycalloc(size_t nelem, size_t size); 44 void *myrealloc(void *p, size_t size); 45 static void *alloctest(void *p); 46 47 /* allocate a string */ 48 49 char * 50 stralloc(char *s) 51 { 52 return (strcpy(mymalloc(strlen(s) + 1), s)); 53 } 54 55 /* version of malloc that only returns if successful */ 56 57 void * 58 mymalloc(size_t size) 59 { 60 return (alloctest(malloc(size))); 61 } 62 63 /* version of calloc that only returns if successful */ 64 65 void * 66 mycalloc(size_t nelem, size_t size) 67 { 68 return (alloctest(calloc(nelem, size))); 69 } 70 71 /* version of realloc that only returns if successful */ 72 73 void * 74 myrealloc(void *p, size_t size) 75 { 76 return (alloctest(realloc(p, size))); 77 } 78 79 /* check for memory allocation failure */ 80 81 static void * 82 alloctest(void *p) 83 { 84 if (p == NULL) { 85 (void) fprintf(stderr, "\n%s: out of storage\n", argv0); 86 exit(1); 87 } 88 return (p); 89 } 90