#include "config.h"
#include "Arena.h"
#include <algorithm>
#include <stdlib.h>
#include <string.h>
#include <wtf/Assertions.h>
#include <wtf/FastMalloc.h>
using namespace std;
namespace WebCore {
#ifdef DEBUG_ARENA_MALLOC
static int i = 0;
#endif
#define ARENA_DEFAULT_ALIGN sizeof(double)
#define BIT(n) ((unsigned int)1 << (n))
#define BITMASK(n) (BIT(n) - 1)
#define CEILING_LOG2(_log2, _n) \
unsigned int j_ = (unsigned int)(_n); \
(_log2) = 0; \
if ((j_) & ((j_)-1)) \
(_log2) += 1; \
if ((j_) >> 16) \
(_log2) += 16, (j_) >>= 16; \
if ((j_) >> 8) \
(_log2) += 8, (j_) >>= 8; \
if ((j_) >> 4) \
(_log2) += 4, (j_) >>= 4; \
if ((j_) >> 2) \
(_log2) += 2, (j_) >>= 2; \
if ((j_) >> 1) \
(_log2) += 1;
#define FREE_PATTERN 0xDA
static int CeilingLog2(unsigned int i) {
int log2;
CEILING_LOG2(log2, i);
return log2;
}
void InitArenaPool(ArenaPool* pool, const char*, unsigned size, unsigned align)
{
if (align == 0)
align = ARENA_DEFAULT_ALIGN;
pool->mask = BITMASK(CeilingLog2(align));
pool->first.next = NULL;
pool->first.base = pool->first.avail = pool->first.limit = (uword)ARENA_ALIGN(&pool->first + 1);
pool->current = &pool->first;
pool->arenasize = size;
}
void* ArenaAllocate(ArenaPool* pool, unsigned int numBytes, unsigned int& bytesAllocated)
{
Arena* arena;
char* returnPointer;
ASSERT((numBytes & pool->mask) == 0);
numBytes = (uword)ARENA_ALIGN(numBytes);
{
arena = pool->current;
do {
if (arena->avail + numBytes <= arena->limit) {
pool->current = arena;
returnPointer = (char *)arena->avail;
arena->avail += numBytes;
return returnPointer;
}
} while (NULL != (arena = arena->next));
}
{
unsigned int size = max(pool->arenasize, numBytes);
size += sizeof *arena + pool->mask;
#ifdef DEBUG_ARENA_MALLOC
i++;
printf("Malloc: %d\n", i);
#endif
bytesAllocated = size;
arena = (Arena*)fastMalloc(size);
arena->limit = (uword)arena + size;
arena->base = arena->avail = (uword)ARENA_ALIGN(arena + 1);
returnPointer = (char *)arena->avail;
arena->avail += numBytes;
arena->next = pool->current->next;
pool->current->next = arena;
pool->current = arena;
if (!pool->first.next)
pool->first.next = arena;
return(returnPointer);
}
}
static void FreeArenaList(ArenaPool* pool, Arena* head)
{
Arena** arenaPointer = &head->next;
Arena* arena = *arenaPointer;
if (!arena)
return;
#ifdef DEBUG
do {
ASSERT(arena->base <= arena->avail && arena->avail <= arena->limit);
arena->avail = arena->base;
memset((void*)(arena)->avail, FREE_PATTERN, (arena)->limit - (arena)->avail)
} while ((arena = arena->next) != 0);
arena = *arenaPointer;
#endif
do {
*arenaPointer = arena->next;
#ifdef DEBUG
memset((void*)(arena), FREE_PATTERN, (arena)->limit - (uword)(arena));
#endif
#ifdef DEBUG_ARENA_MALLOC
if (arena) {
i--;
printf("Free: %d\n", i);
}
#endif
fastFree(arena);
arena = 0;
} while ((arena = *arenaPointer) != 0);
pool->current = head;
}
void FinishArenaPool(ArenaPool* pool)
{
FreeArenaList(pool, &pool->first);
}
}