rizin/libr/parse/p/parse_mreplace/mmemory.c
2016-09-19 14:44:47 +02:00

104 lines
2.5 KiB
C

//
// Library: Memory Manage Module v1.10, Copyleft, 2009-02-25
// Author : Mandingo, mandingo [ at ]yoire.com
//
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "mmemory.h"
static memInfo *mInfo;
void memStringRealloc(memChunk *chunk) {
memCheckState ();
memChunk *temp;
temp = memString (chunk->address);
//printf("Reallocating chunk size %d to size %d\n",chunk->size,temp->size);
memCopy (chunk,temp);
memFree (temp);
}
void memFree(memChunk *chunk) {
memCheckState ();
if (chunk && chunk->address){
mInfo->allocated-=chunk->size;
free (chunk->address);
free (chunk);
}
}
void memInit() {
mInfo = (memInfo*) malloc (sizeof (memInfo));
if (!mInfo) return;
memset (mInfo, 0, sizeof (memInfo));
mInfo->state = MEM_STATE_OK;
mInfo->allocated += sizeof (memInfo);
}
void memCheckState() {
if (!mInfo) memInit();
if (mInfo->state != MEM_STATE_OK){
fprintf (stderr,"\rMemory not initialized :p\n");
exit (0);
}
}
memInfo *memInformation() {
memCheckState ();
return mInfo;
}
long memAllocated() {
memCheckState ();
return mInfo->allocated;
}
memChunk *memReserve(long size) {
static memChunk *buffer;
memCheckState ();
buffer = (memChunk*)malloc (sizeof (memChunk));
if (!buffer || !(buffer->address = (char*)malloc (size))) {
perror ("memReserve");
exit (0);
}
//printf("- reservando %d bytes\n",size);
buffer->size = size;
memset (buffer->address, 0, buffer->size);
mInfo->allocated += buffer->size;
return buffer;
}
memChunk *memStringReserve(char *string, long nbytes) {
static memChunk *buffer;
buffer=memReserve (nbytes);
memCopy (buffer, memString (string));
return buffer;
}
memChunk *memString(char *string) {
static memChunk *buffer;
memCheckState ();
buffer = memReserve (strlen (string) + 1);
memcpy (buffer->address, string, strlen (string));
return buffer;
}
void memCopy(memChunk *dest,memChunk *source) {
long nbytes;
memCheckState ();
if ((!source->address) || (!dest->address)) return;
nbytes=dest->size > source->size ? source->size : dest->size;
#if DEBUG3
eprintf ("Copying %d bytes to dest (size %d)\n", nbytes, dest->address, dest->size);
#endif
memcpy (dest->address, source->address, nbytes);
}
void memStrCat(memChunk *dest, char *string) {
memChunk result,*temp;
temp = memReserve (dest->size + strlen (string)+1);
result.address = dest->address + strlen (dest->address);
result.size = dest->size - strlen (dest->address) + 1;
memCopy (temp, memString (string));
memCopy (&result, temp);
memFree (temp);
}