usr/src/tools/cscope-fast/alloc.c

   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 }