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