rcalloc0.c

开放源码的编译器open watcom 1.6.0版的源代码

/****************************************************************************
*
*                            Open Watcom Project
*
*    Portions Copyright (c) 1983-2002 Sybase, Inc. All Rights Reserved.
*
*  ========================================================================
*
*    This file contains Original Code and/or Modifications of Original
*    Code as defined in and that are subject to the Sybase Open Watcom
*    Public License version 1.0 (the 'License'). You may not use this file
*    except in compliance with the License. BY USING THIS FILE YOU AGREE TO
*    ALL TERMS AND CONDITIONS OF THE LICENSE. A copy of the License is
*    provided with the Original Code and Modifications, and is also
*    available at www.sybase.com/developer/opensource.
*
*    The Original Code and all software distributed under the License are
*    distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
*    EXPRESS OR IMPLIED, AND SYBASE AND ALL CONTRIBUTORS HEREBY DISCLAIM
*    ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF
*    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR
*    NON-INFRINGEMENT. Please see the License for the specific language
*    governing rights and limitations under the License.
*
*  ========================================================================
*
* Description:  Resource compiler memory layer 0.
*
****************************************************************************/


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "rcalloc0.h"
#include "errors.h"
#ifdef RCMEM_DEBUG
#include "errprt.h"
#endif

static void RCMemLayer0InitFreeList( FreeListInfo *freelist, int heapsize,
                                     int blocks_per_heap )
/************************************************************************/
{
    int                 i;

    for( i = 0; i < blocks_per_heap - 1; i++ ) {
#ifdef RCMEM_DEBUG
        freelist->next = (char *)freelist + heapsize + 1 +
                         sizeof( DebugMemInfo );
#else
        freelist->next = (char *)freelist + heapsize;
#endif
        freelist = (FreeListInfo *)freelist->next;
    }
    freelist->next = NULL;
}

static HeapList *RCMemLayer0AddToHeap( int heapsize, int blocks_per_heap )
/************************************************************************/
{
    FreeListInfo     *freelist;
    HeapList         *data;

#ifdef RCMEM_DEBUG
    unsigned long     allocsize;

    /* Add 1 in order to store endbyte */

    allocsize = ( heapsize + 1 + sizeof( DebugMemInfo ) ) * blocks_per_heap +
                  sizeof( HeapList );
    data = malloc( allocsize );
    memset( data, RCMEM_GARBAGEBYTE, allocsize );
#else
    data = malloc( heapsize * blocks_per_heap + sizeof( HeapList ) );
#endif
    if( data == NULL ) {
        RcFatalError( ERR_OUT_OF_MEMORY );
    }
    data->next = NULL;
    freelist = (FreeListInfo *)( (char *)data + sizeof( HeapList ) );
    RCMemLayer0InitFreeList( freelist, heapsize, blocks_per_heap );

    return( data );
}

extern HeapHandle *RCMemLayer0NewHeap( int heapsize, int blocks_per_heap )
/************************************************************************/
{
    HeapHandle         *heap;

    heap = malloc( sizeof( HeapHandle ) );
    if( heap == NULL ) {
        RcFatalError( ERR_OUT_OF_MEMORY );
    }
    heap->list = NULL;
    heap->freeList = NULL;
    heap->heapsize = heapsize;
    heap->blocksize = blocks_per_heap;

    return( heap );
}

#ifdef RCMEM_DEBUG
extern void *RCMemLayer0Malloc( HeapHandle *heap, size_t size )
#else
extern void *RCMemLayer0Malloc( HeapHandle *heap )
#endif
/************************************************/
{
    HeapList           *newheap;
    FreeListInfo       *freemem;
#ifdef RCMEM_DEBUG
    DebugMemInfo       *debugmem;
#endif

    if( heap->freeList == NULL ) {
        newheap = RCMemLayer0AddToHeap( heap->heapsize, heap->blocksize );
        newheap->next = heap->list;
        heap->list = newheap;
        heap->freeList = (char *)newheap + sizeof( HeapList );
    }
    freemem = (FreeListInfo *)heap->freeList;
    heap->freeList = (void *)( freemem->next );
#ifdef RCMEM_DEBUG
    debugmem = (DebugMemInfo *)freemem;
    debugmem->size = size;
    debugmem->startbyte = RCMEM_STARTBYTE;

    freemem = (FreeListInfo *)((char *)freemem + sizeof( DebugMemInfo ) );
    *( (char *)freemem + size ) = RCMEM_ENDBYTE;
#endif

    return( freemem );
}

extern void RCMemLayer0Free( void *mem, HeapHandle *heap )
/********************************************************/
{
#ifdef RCMEM_DEBUG
    DebugMemInfo        *debugmem;

    if( mem == NULL ) {
        RcFprintf( stderr, NULL, "Free NULL pointer\n" );
    }
    debugmem = (DebugMemInfo *)( (char *)mem - sizeof( DebugMemInfo ) );
    if( *((char*)mem + debugmem->size ) != RCMEM_ENDBYTE ) {
        RcFprintf( stderr, NULL, "(%x) Memory Overrun\n", mem );
    }

    debugmem->startbyte = !RCMEM_STARTBYTE;
    mem = (char *)mem - sizeof( DebugMemInfo );
#endif

    *( (char **)mem ) = heap->freeList;
    heap->freeList = mem;
}

#ifdef RCMEM_DEBUG

static void RCMemLayer0CheckUnfreed( HeapHandle *heap, char *freelist )
/*********************************************************************/
{
#ifdef RCMEM_DEBUG
    DebugMemInfo     *debugmem;
#endif
    int               i;

    for( i = 0; i < heap->blocksize; i++ ) {
        debugmem = (DebugMemInfo *)freelist;
        if( debugmem->startbyte == RCMEM_STARTBYTE ) {
            RcFprintf( stderr, NULL,
                        "Unfreed Memory Detected (0x%x bytes at 0x%x)\n",
                        debugmem->size, freelist + sizeof( DebugMemInfo ) );
            debugmem->startbyte = !RCMEM_STARTBYTE;
        }
        freelist += heap->heapsize + sizeof( DebugMemInfo ) + 1;
    }
}

#endif

extern void RCMemLayer0ShutDown( HeapHandle *heap )
/*************************************************/
{
    HeapList         *curnode;
    HeapList         *nextnode;

    curnode = heap->list;
    nextnode = heap->list;
    while( nextnode != NULL ) {
        nextnode = curnode->next;
#ifdef RCMEM_DEBUG
        RCMemLayer0CheckUnfreed( heap, (char *)curnode + sizeof( HeapList ) );
#endif
        free( curnode );
        curnode = nextnode;
    }
    free( heap );
}