83 lines
1.5 KiB
C
83 lines
1.5 KiB
C
|
/* 7zAlloc.c -- Allocation functions
|
||
|
|
2010-10-29 : Igor Pavlov : Public domain */
|
||
|
|
|
||
|
|
#include "7zAlloc.h"
|
||
|
|
|
||
|
|
#if defined(_WIN32)
|
||
|
|
#include <WinSock2.h>
|
||
|
|
#include <Windows.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/* #define _SZ_ALLOC_DEBUG */
|
||
|
|
/* use _SZ_ALLOC_DEBUG to debug alloc/free operations */
|
||
|
|
|
||
|
|
#ifdef _SZ_ALLOC_DEBUG
|
||
|
|
|
||
|
|
#ifdef _WIN32
|
||
|
|
#include <windows.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
int g_allocCount = 0;
|
||
|
|
int g_allocCountTemp = 0;
|
||
|
|
|
||
|
|
#endif
|
||
|
|
#include "clamav.h"
|
||
|
|
|
||
|
|
void *SzAlloc(void *p, size_t size)
|
||
|
|
{
|
||
|
|
UNUSEDPARAM(p);
|
||
|
|
if (size == 0)
|
||
|
|
return 0;
|
||
|
|
#ifdef _SZ_ALLOC_DEBUG
|
||
|
|
fprintf(stderr, "\nAlloc %10d bytes; count = %10d", size, g_allocCount);
|
||
|
|
g_allocCount++;
|
||
|
|
#endif
|
||
|
|
return malloc(size);
|
||
|
|
}
|
||
|
|
|
||
|
|
void SzFree(void *p, void *address)
|
||
|
|
{
|
||
|
|
UNUSEDPARAM(p);
|
||
|
|
#ifdef _SZ_ALLOC_DEBUG
|
||
|
|
if (address != 0)
|
||
|
|
{
|
||
|
|
g_allocCount--;
|
||
|
|
fprintf(stderr, "\nFree; count = %10d", g_allocCount);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
free(address);
|
||
|
|
}
|
||
|
|
|
||
|
|
void *SzAllocTemp(void *p, size_t size)
|
||
|
|
{
|
||
|
|
UNUSEDPARAM(p);
|
||
|
|
if (size == 0)
|
||
|
|
return 0;
|
||
|
|
#ifdef _SZ_ALLOC_DEBUG
|
||
|
|
fprintf(stderr, "\nAlloc_temp %10d bytes; count = %10d", size, g_allocCountTemp);
|
||
|
|
g_allocCountTemp++;
|
||
|
|
#ifdef _WIN32
|
||
|
|
return HeapAlloc(GetProcessHeap(), 0, size);
|
||
|
|
#endif
|
||
|
|
#endif
|
||
|
|
return malloc(size);
|
||
|
|
}
|
||
|
|
|
||
|
|
void SzFreeTemp(void *p, void *address)
|
||
|
|
{
|
||
|
|
UNUSEDPARAM(p);
|
||
|
|
#ifdef _SZ_ALLOC_DEBUG
|
||
|
|
if (address != 0)
|
||
|
|
{
|
||
|
|
g_allocCountTemp--;
|
||
|
|
fprintf(stderr, "\nFree_temp; count = %10d", g_allocCountTemp);
|
||
|
|
}
|
||
|
|
#ifdef _WIN32
|
||
|
|
HeapFree(GetProcessHeap(), 0, address);
|
||
|
|
return;
|
||
|
|
#endif
|
||
|
|
#endif
|
||
|
|
free(address);
|
||
|
|
}
|