You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.3 KiB
62 lines
1.3 KiB
/*======================*
|
|
%$ /!\ yolomalloc.c /!\ $%
|
|
%$ DANGER Swag overflow $%
|
|
*======================*/
|
|
|
|
/* Feature test macros */
|
|
#define _DEFAULT_SOURCE /* for sbrk */
|
|
|
|
/* Includes */
|
|
|
|
#include <unistd.h>
|
|
#include <stdio.h>
|
|
|
|
#include "yolomalloc.h"
|
|
|
|
/* Defines */
|
|
|
|
#define YOLO_WORD (sizeof(intptr_t))
|
|
#define YOLO_DWORD (2 * YOLO_WORD)
|
|
#define YOLO_CHUNK 8192
|
|
|
|
/* Globals */
|
|
|
|
void *yolo_start = NULL;
|
|
void *yolo_end = NULL;
|
|
|
|
int yolo_init() {
|
|
yolo_start = sbrk(0);
|
|
if (yolo_start == (void*) -1) {
|
|
yolo_start = NULL;
|
|
yolo_end = NULL;
|
|
return -1;
|
|
}
|
|
yolo_end = yolo_start;
|
|
return 0;
|
|
}
|
|
|
|
/* Ya know, allocating stuff and all */
|
|
void *yolomalloc(size_t size) {
|
|
if (size == 0) {
|
|
return NULL;
|
|
}
|
|
if (yolo_start == NULL || yolo_end == NULL) {
|
|
if (yolo_init() != 0) {
|
|
return NULL;
|
|
}
|
|
}
|
|
intptr_t size_w = (size + (YOLO_WORD-1)) / YOLO_WORD; // size in words
|
|
intptr_t align_dw = ((size_w + 1) / 2) * 2; // round size_w to dword
|
|
void *ptr = sbrk(align_dw);
|
|
if (ptr == (void*) -1) {
|
|
return NULL;
|
|
}
|
|
yolo_end = ptr + align_dw;
|
|
return ptr;
|
|
}
|
|
|
|
void yolofree(void *ptr) {
|
|
/*----------------------------------*
|
|
* THE YOLO IS STRONG WITH THIS ONE *
|
|
*----------------------------------*/
|
|
}
|