#ifndef _KMALLOC_H
#define _KMALLOC_H

#include "types.h"

// This function allocates size bytes of contiguous memory.
// Returns a pointer to allocated memory or NULL.
void* kmalloc(size_t size);

// This function allocates size bytes of contiguous, zeroed-out memory.
// Returns a pointer to allocated memory or NULL.
void* kzalloc(size_t size);

// This function allocates memory for count elements each of size bytes,
// zeroed-out memory. Returns a pointer to allocated memory or NULL.
void* kcalloc(size_t count, size_t size);

// powers of 2
#define KMALLOC_MAX_SIZE 17
#define LOWEST_POWER 3

union per_cpu_freelist_cache {
	void** freelists[KMALLOC_MAX_SIZE];
	struct {
		void** kmalloc_8;
		void** kmalloc_16;
		void** kmalloc_32;
		void** kmalloc_64;
		void** kmalloc_128;
		void** kmalloc_256;
		void** kmalloc_512;
		void** kmalloc_1024;
		void** kmalloc_2048;
		void** kmalloc_4096;
		void** kmalloc_8192;
		void** kmalloc_16384;
		void** kmalloc_32768;
		void** kmalloc_65536;
		void** kmalloc_131072;
	};
};

union partial_list {
	void** freelists[KMALLOC_MAX_SIZE];
	struct {
		void** kmalloc_8;
		void** kmalloc_16;
		void** kmalloc_32;
		void** kmalloc_64;
		void** kmalloc_128;
		void** kmalloc_256;
		void** kmalloc_512;
		void** kmalloc_1024;
		void** kmalloc_2048;
		void** kmalloc_4096;
		void** kmalloc_8192;
		void** kmalloc_16384;
		void** kmalloc_32768;
		void** kmalloc_65536;
		void** kmalloc_131072;
	};
};

#endif