/*=============================================================================
   GNU UnRTF, a command-line program to convert RTF documents to other formats.
   Copyright (C) 2000,2001,2004 by Zachary Smith

   This program 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 3 of the License, or
   (at your option) any later version.

   This program 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 this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

   The maintainer is reachable by electronic mail at daved@physiol.usyd.edu.au
=============================================================================*/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <cstdio>
#include <cstdlib>
#include <cstring>

#include "error.h"
#include "malloc.h"

static unsigned long count = 0;

/*========================================================================
 * Name:    my_malloc
 * Purpose:    Internal version of malloc necessary for record keeping.
 * Args:    Amount.
 * Returns:    Pointer.
 *=======================================================================*/

char *
my_malloc(unsigned long size)
{
    char * ptr = reinterpret_cast<char*>(malloc(size));
    if (ptr)
    {
        count += size;
    }

    return ptr;
}

/*========================================================================
 * Name:    my_free
 * Purpose:    Internal version of free necessary for record keeping.
 * Args:    Pointer.
 * Returns:    None.
 *=======================================================================*/

void
my_free(char *ptr)
{
    CHECK_PARAM_NOT_NULL(ptr);

    free(ptr);
}

#if 1 /* AK3 - AF */
/*========================================================================
 * Name:    my_realloc
 * Purpose:    Internal version of realloc necessary for record keeping.
 * Args:    Pointer.
 * Returns:    None.
 *=======================================================================*/
char *
my_realloc(char *ptr, unsigned long old_size, unsigned long new_size)
{
    char *new_ptr = my_malloc(new_size);

    if (new_ptr != NULL)
    {
        memcpy(new_ptr, ptr, old_size);
    }

    my_free(ptr);

    return new_ptr;
}
#endif

/*========================================================================
 * Name:    total_malloced
 * Purpose:    Returns total amount of memory thus far allocated. Called at
 *             the end of main() when in debug mode.
 * Args:    None.
 * Returns:    Amount.
 *=======================================================================*/

unsigned long
total_malloced(void)
{
    return count;
}



/*========================================================================
 * Name:    my_strdup
 * Purpose:    Internal version of strdup necessary for record keeping.
 * Args:    String.
 * Returns:    String.
 *=======================================================================*/

char *
my_strdup(char *src)
{
    unsigned long len;
    char *ptr;

    CHECK_PARAM_NOT_NULL(src);

    len = strlen(src);
    ptr = my_malloc(len + 1);
    if (!ptr)
    {
        error_handler("out of memory in strdup()");
    }

    strcpy(ptr, src);
    return ptr;
}

#ifndef _WIN32
/* added by daved */
#include <sys/types.h>
#undef malloc

void *malloc();

/* Allocate an N-byte block of memory from the heap.
If N is zero, allocate a 1-byte block.  */

void *
rpl_malloc(size_t n)
{
    if (n == 0)
    {
        n = 1;
    }
    return malloc(n);
}
#endif
