ram-smalloc.c 5.93 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2
 * $Id: ram-smalloc.c,v 0.24 1998/07/08 13:42:00 ceder Exp $
Per Cederqvist's avatar
Per Cederqvist committed
3
 * Copyright (C) 1991, 1992, 1993, 1994, 1995, 1996  Lysator Academic Computer Association.
Linus Tolke Y's avatar
Linus Tolke Y committed
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 *
 * This file is part of the LysKOM server.
 * 
 * LysKOM is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by 
 * the Free Software Foundation; either version 1, or (at your option) 
 * any later version.
 * 
 * LysKOM is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with LysKOM; see the file COPYING.  If not, write to
 * Lysator, c/o ISY, Linkoping University, S-581 83 Linkoping, SWEDEN,
 * or the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, 
 * MA 02139, USA.
 *
 * Please mail bug reports to bug-lyskom@lysator.liu.se. 
 */
Per Cederqvist's avatar
Per Cederqvist committed
25 26 27 28 29 30 31 32
/*
 * smalloc.c
 *
 * Contains memory allocator routines
 *
 *	TEST VERSION by ceder
 */

33 34 35
/*
 * Define DEBUG_MALLOC to get traces from smalloc, srealloc and sfree.
 * Run the resulting lyskomd under gdb. Source trace-mem.gdb. Run
Per Cederqvist's avatar
Per Cederqvist committed
36
 * M-x resolve-trace (from handle-malloc-dump.el) on the resulting output.
37
 */
Per Cederqvist's avatar
Per Cederqvist committed
38
/* #define DEBUG_MALLOC */
39

40 41
static const char *
rcsid = "$Id: ram-smalloc.c,v 0.24 1998/07/08 13:42:00 ceder Exp $";
42 43
#include "rcs.h"
USE(rcsid);
Per Cederqvist's avatar
Per Cederqvist committed
44

45 46 47 48 49 50
#ifdef USE_GNU_MALLOC
#  include "malloc.h"
#else
#  ifdef HAVE_STDLIB_H
#    include <stdlib.h>
#  endif
Per Cederqvist's avatar
Per Cederqvist committed
51
#endif
Per Cederqvist's avatar
Per Cederqvist committed
52
#include <stdio.h>
Per Cederqvist's avatar
Per Cederqvist committed
53
#include <sys/types.h>
Per Cederqvist's avatar
Per Cederqvist committed
54

Per Cederqvist's avatar
Per Cederqvist committed
55
#include "exp.h"
56
#include "s-string.h"
Per Cederqvist's avatar
Per Cederqvist committed
57 58 59
#include "kom-types.h"
#include "lyskomd.h"
#include "server/smalloc.h"
Per Cederqvist's avatar
Per Cederqvist committed
60

61
static int no_of_allocated_blocks = 0;
Per Cederqvist's avatar
Per Cederqvist committed
62

63 64 65 66 67 68 69 70 71 72
#ifdef DEBUG_MALLOC
static void
trace_smalloc(size_t size,
	      void *result)
{
    printf("smalloc:\nArg: 0x%lx\nRes: 0x%lx\n", (long)size, (long)result);
    printf("==== end ====\n");
}
#endif

Per Cederqvist's avatar
Per Cederqvist committed
73 74 75 76
/*
 * "safe" malloc. Handles the case when malloc returns NULL.
 * smalloc cannot fail.
 */
Per Cederqvist's avatar
Per Cederqvist committed
77
EXPORT  void *
Per Cederqvist's avatar
Per Cederqvist committed
78 79 80 81 82 83
smalloc(size_t size)
{
   unsigned int *p;
   
   p = (unsigned int *) malloc(size + 2*sizeof(unsigned int) + 2);
   if (p == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
84
       restart_kom("Can't allocate %lu bytes.\n", (unsigned long)size);
Per Cederqvist's avatar
Per Cederqvist committed
85 86 87 88 89 90 91

   ++no_of_allocated_blocks;

   *p++ = SMALLOC_MAGIC_ALLOC;
   *p++ = size;
   ((unsigned char *) p)[size]   = 0x89;
   ((unsigned char *) p)[size+1] = 0xA7;
92 93 94 95 96

#ifdef DEBUG_MALLOC
   trace_smalloc(size, p);
#endif

Per Cederqvist's avatar
Per Cederqvist committed
97 98 99
   return (void *) p;
}

100 101 102 103 104 105 106 107
#ifdef DEBUG_MALLOC
static void
trace_free(void *block)
{
    printf("sfree:\nArg: 0x%lx\n", (long)block);
    printf("==== end ====\n");
}
#endif
Per Cederqvist's avatar
Per Cederqvist committed
108

Per Cederqvist's avatar
Per Cederqvist committed
109
EXPORT  void
Per Cederqvist's avatar
Per Cederqvist committed
110 111 112 113 114 115
sfree(void * ptr)	/* it is legal to sfree a NULL pointer */
{
    unsigned int *ip;
    
    if ( ptr != NULL )
    {
116 117 118
#ifdef DEBUG_MALLOC
	trace_free(ptr);
#endif
Per Cederqvist's avatar
Per Cederqvist committed
119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
        ip = (unsigned int *) ptr;
	ip -= 2;
	switch (*ip)
	{
	    case SMALLOC_MAGIC_ALLOC:
		if (((unsigned char *) (ip+2))[ip[1]] != 0x89 ||
		    ((unsigned char *) (ip+2))[ip[1]+1] != 0xA7)
		  restart_kom("SFREE: Buffer overflow, bsize = %ul\n", ip[1]);
	        --no_of_allocated_blocks;
		*ip = SMALLOC_MAGIC_FREE;
		free( ip );
		break;

	    case SMALLOC_MAGIC_FREE:
		restart_kom("SFREE: Trying to free already freed block\n");

	    default:
		restart_kom("SFREE: Illegal magic number\n");
	}
    }
}

141 142 143 144 145 146 147 148 149 150 151
#ifdef DEBUG_MALLOC
static void
trace_srealloc(size_t size,
	       void *arg,		   
	       void *result)
{
    printf("srealloc:\nSize: 0x%lx\nArg: 0x%lx\nRes: 0x%lx\n",
	   (long)size, (long)arg, (long)result);
    printf("==== end ====\n");
}
#endif
Per Cederqvist's avatar
Per Cederqvist committed
152
EXPORT  void *
Per Cederqvist's avatar
Per Cederqvist committed
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
srealloc(void * ptr, size_t size) /* Never fails. It is legal to */
{				    /* realloc the NULL ptr. */
    unsigned int   * ip;
    unsigned int   * new_ptr;

    if ( ptr == NULL )
	return smalloc(size);

    ip = (unsigned int *) ptr;
    ip -= 2;
    switch (*ip)
    {
        case SMALLOC_MAGIC_ALLOC:
            break;

	case SMALLOC_MAGIC_FREE:
	    restart_kom("SREALLOC: Trying to realloc freed block\n");

	default:
	    restart_kom("SREALLOC: Illegal magic number\n");
    }

    if (((unsigned char *) (ip+2))[ip[1]] != 0x89 ||
	((unsigned char *) (ip+2))[ip[1]+1] != 0xA7)
177 178
      restart_kom("SREALLOC: Buffer overflow, osize = %ul, nsize = %lu.\n",
		  ip[1], (unsigned long)size);
Per Cederqvist's avatar
Per Cederqvist committed
179 180 181 182 183

    *ip = SMALLOC_MAGIC_FREE;
    if ( (new_ptr = (unsigned int *) realloc((void *) ip,
				size+2*sizeof(unsigned int)+2) ) == NULL )
    {
184 185
	restart_kom("Out of memory - can't realloc. ptr = %lu size = %lu.\n",
		    (unsigned long)ptr, (unsigned long)size);
Per Cederqvist's avatar
Per Cederqvist committed
186 187 188 189 190 191 192
    }

    *new_ptr++ = SMALLOC_MAGIC_ALLOC;
    *new_ptr++ = size;

    ((unsigned char *) new_ptr)[size]   = 0x89;
    ((unsigned char *) new_ptr)[size+1] = 0xA7;
193 194 195 196 197

#ifdef DEBUG_MALLOC
    trace_srealloc(size, ptr, new_ptr);
#endif

Per Cederqvist's avatar
Per Cederqvist committed
198 199 200 201 202 203 204 205 206 207 208 209 210
    return (void *) new_ptr;
}


/*
 * Allocate temporary memory, which is automatically freed after this
 * atomic call.
 */

static void **tmp_alloc_table   = NULL;
static int tmp_alloc_table_size = 0; /* Size */
static int tmp_alloc_table_use  = 0; /* Used size */

Per Cederqvist's avatar
Per Cederqvist committed
211
EXPORT  void *
Per Cederqvist's avatar
Per Cederqvist committed
212
tmp_alloc(unsigned long size)
Per Cederqvist's avatar
Per Cederqvist committed
213 214 215 216 217 218 219 220
{
    if ( tmp_alloc_table_size <= tmp_alloc_table_use )
    {
	/* Need to increas table. */
	tmp_alloc_table = srealloc (tmp_alloc_table,
				    ((++tmp_alloc_table_size)
				     * sizeof (void *)));
    }
221

Per Cederqvist's avatar
Per Cederqvist committed
222 223 224 225 226 227 228 229 230
    return (tmp_alloc_table[ tmp_alloc_table_use++ ]
	    = smalloc (size));
}


/*
 * Free all core which is allocated with tmp_alloc(). This is called from
 * end_of_atomic().
 */
Per Cederqvist's avatar
Per Cederqvist committed
231
EXPORT void
Per Cederqvist's avatar
Per Cederqvist committed
232 233 234 235 236 237 238 239 240 241 242 243 244
free_tmp(void)
{
    int i;

    for ( i = 0; i < tmp_alloc_table_use; i++ )
    {
	sfree ( tmp_alloc_table[ i ] );
	tmp_alloc_table[ i ] = NULL;
    }

    tmp_alloc_table_use = 0;
}

Per Cederqvist's avatar
Per Cederqvist committed
245
EXPORT  void
Per Cederqvist's avatar
Per Cederqvist committed
246 247 248 249 250 251 252
free_all_tmp(void)
{
    free_tmp();
    sfree( tmp_alloc_table );
    tmp_alloc_table = NULL;
    tmp_alloc_table_size = 0;
}
Per Cederqvist's avatar
Per Cederqvist committed
253 254 255 256

EXPORT void
dump_smalloc_counts(FILE *stat_file)
{
Per Cederqvist's avatar
Per Cederqvist committed
257 258
    fprintf(stat_file, "---ram-smalloc.c:\n%s%d\n",
	    "\tAllocated blocks (grand total): ",
Per Cederqvist's avatar
Per Cederqvist committed
259 260
	    no_of_allocated_blocks);
}