BMOSP/kernel/mem.cpp

#include <fb.h>
#include <limine.h>
#include <lock.h>
#include <tool.h>

#define BLOCK_SIZE 4096
#define HHDM_OFFSET (hhdm_request.response->offset)

static volatile struct limine_memmap_request memmap_request = {
	.id = LIMINE_MEMMAP_REQUEST,
	.revision = 0,
	.response = (struct limine_memmap_response *)0
};
static volatile struct limine_hhdm_request hhdm_request = {
	.id = LIMINE_HHDM_REQUEST,
	.revision = 0,
	.response = (struct limine_hhdm_response *)0
};

struct mem_entry {
	struct mem_entry *next;
	bool free;
	size_t size;
	uint8_t data[0];
};
typedef struct mem_entry mem_entry_t;

// Битовая карта для отслеживания занятых и свободных фреймов памяти
uint8_t *bitmap;
// Объем доступных блоков
uint64_t bitmap_available = 0;
// Объем блоков
uint64_t bitmap_limit = 0;
// Верхняя граница доступной памяти
uint64_t limit;
// Объем всего доступного физического адресного пространства
uint64_t usable = 0;
// Объем доступной виртуальной памяти
uint64_t available = 0;
// Наивысший адрес в available space
uint64_t highest = 0;
// Количество записей в карте памяти
uint64_t mmmap_count = 0;

const char memory_types[8][82] = {
	"Доступно",      "Зарезервировано", "ACPI, можно освободить",
	"ACPI NVS",      "Плохая память",   "Загрузчик, можно освободить",
	"Ядро и модули", "Буфер кадра"
};

struct limine_memmap_response *memmap_response;

static mem_entry_t *first_node;

namespace mem {

void dump_memory( ) {
	mem_entry_t *curr = first_node;

	while (curr) {
		fb::printf("->0x%x | %u.%u kb | %u | 0x%x\n", &curr->data,
		           (curr->size) / 1024, (curr->size) % 1024, curr->free,
		           curr->next);
		curr = curr->next;
	}
}

void frame_free(void *addr, uint64_t frames) {
	// Проход по фреймам памяти и очистка битов в битовой карте
	uint64_t frame = (uint64_t)addr / BLOCK_SIZE;
	for (uint64_t i = frame; i < frames + frame; i++) { BIT_CLEAR(i); }
	bitmap_available += frames;
}

// Функция выделения памяти
void *frame_alloc(uint64_t wanted_frames) {
	void *addr;

	uint64_t available_frames = 0;
	for (uint64_t frame = 1; frame < limit; frame++) {
		if (!BIT_GET(frame)) {
			available_frames++;
		} else if (available_frames != wanted_frames) {
			available_frames = 0;
			continue;
		}
		if (available_frames == wanted_frames) {
			uint64_t i;
			for (i = 0; i < wanted_frames; i++) { BIT_SET(frame - i); }
			frame -= i - 1;

			addr = (void *)(BLOCK_SIZE * frame);
			bitmap_available -= wanted_frames;
			return addr;
		}
	}
	return NULL;
}

void *frame_calloc(uint64_t frames) {
	void *addr = frame_alloc(frames);
	tool::memset(addr + HHDM_OFFSET, 0, frames * BLOCK_SIZE);
	return addr;
}

void merge_blocks(mem_entry_t *start) {
	if (!start->free) return;
	mem_entry_t *block = start;
	while (block->next && block->next->free) {
		block->size += block->next->size + sizeof(mem_entry_t);
		block->next = block->next->next;
	}
}

void merge_all_blocks( ) {
	mem_entry_t *curr = first_node;

	while (curr) {
		merge_blocks(curr);
		curr = curr->next;
	}
}

void add_block(void *addr, size_t size) {
	mem_entry_t *new_entry = (mem_entry_t *)addr;

	new_entry->size = size - sizeof(mem_entry_t);
	new_entry->free = true;

	if (first_node == NULL) {
		first_node = new_entry;
		new_entry->next = NULL;
	} else {
		mem_entry_t *curr = first_node;
		while (curr->next != NULL) { curr = curr->next; }

		curr->next = new_entry;
		new_entry->next = NULL;
	}
}

void alloc_init(void *address, size_t length) {
	first_node = (mem_entry_t *)address;

	first_node->size = length - sizeof(mem_entry_t);
	first_node->free = true;
	first_node->next = NULL;
}

void *alloc_align(size_t size, size_t alignment) {
	mem_entry_t *curr = first_node;

	while (curr) {
		if (curr->free) {
			void *addr = curr->data + alignment - 1;
			addr -= (uintptr_t)addr % alignment + sizeof(mem_entry_t);
			mem_entry_t *second = (mem_entry_t *)addr;
			if (curr->size >= (second->data - curr->data + size)) {
				mem_entry_t *third = (mem_entry_t *)(second->data + size);

				third->size = curr->size - (third->data - curr->data);
				third->next = curr->next;
				third->free = 1;

				second->size = size;
				second->next = third;
				second->free = 0;

				if (curr != second) {
					curr->next = second;
					curr->size = (uintptr_t)second - (uintptr_t)curr->data;
					curr->free = 1;
				}

				return second->data;
			}
		}

		curr = curr->next;
	}
	return NULL;
}

void *alloc(size_t size) {
	return alloc_align(size, 1);
}

void free(void *addr) {
	mem_entry_t *curr = first_node, *prev = NULL;
	while (curr != NULL) {
		if (curr->data == addr) {
			curr->free = 1;
			merge_blocks(prev ? prev : curr);
			return;
		}
		prev = curr;
		curr = curr->next;
	}
}

void *realloc(void *addr, size_t size) {
	if (size == 0) {
		free(addr);
		return NULL;
	}

	if (addr == NULL) { return alloc(size); }

	void *new_addr = alloc(size);

	if (new_addr == NULL) { return NULL; }

	tool::memcpy(new_addr, addr, size);
	free(addr);

	return new_addr;
}

// Инициализация менеджера памяти
void init( ) {
	// Получение информации о доступной памяти из Limine bootloader
	memmap_response = memmap_request.response;
	mmmap_count = memmap_response->entry_count;
	struct limine_memmap_entry **mmaps = memmap_response->entries;

	fb::printf("Записей в карте памяти: %u\n", memmap_response->entry_count);

	// Обработка каждой записи в карте памяти
	for (int i = 0; i < mmmap_count; i++) {
		available += mmaps[i]->length;

		// fb::printf("\t%d: 0x%x\tдлина: 0x%x\tтип: %s\n", i + 1,
		// mmaps[i]->base, mmaps[i]->length, memory_types[mmaps[i]->type]);
		if (mmaps[i]->type == LIMINE_MEMMAP_FRAMEBUFFER) {
			fb::printf("На видеопамять BIOS/UEFI выделено: %u мегабайт + %u "
			           "килобайт\n",
			           mmaps[i]->length / 1024 / 1024,
			           (mmaps[i]->length / 1024) % 1024);
		}
		if (!(mmaps[i]->type == LIMINE_MEMMAP_USABLE)) { continue; }

		usable += mmaps[i]->length;
		uint64_t top = mmaps[i]->base + mmaps[i]->length;
		if (top > highest) highest = top;
	}

	limit = highest / BLOCK_SIZE;
	uint64_t bitmap_size = ALIGN_UP(highest / BLOCK_SIZE / 8, BLOCK_SIZE);

	// Находим доступное место для битовой карты и устанавливаем ее
	for (uint64_t i = 0; i < mmmap_count; i++) {
		if (!mmaps[i]->type == LIMINE_MEMMAP_USABLE) continue;

		if (mmaps[i]->length >= bitmap_size) {
			bitmap = (uint8_t *)mmaps[i]->base;
			tool::memset(bitmap, 0xFF, bitmap_size);
			mmaps[i]->length -= bitmap_size;
			mmaps[i]->base += bitmap_size;
			available -= bitmap_size;
			break;
		}
	}

	// Освобождаем все доступные фреймы памяти
	for (uint64_t i = 0; i < mmmap_count; i++) {
		for (uint64_t t = 0; t < mmaps[i]->length; t += BLOCK_SIZE) {
			bitmap_limit++;
		}

		if (!(mmaps[i]->type == LIMINE_MEMMAP_USABLE)) { continue; }

		for (uint64_t t = 0; t < mmaps[i]->length; t += BLOCK_SIZE) {
			frame_free((void *)mmaps[i]->base + t, 1);
		}
	}

	fb::printf("%u / %u блоков доступно\n", bitmap_available, bitmap_limit);
	fb::printf("Размер битовой карты: %u\n", bitmap_size);
	alloc_init(frame_alloc(1), BLOCK_SIZE);
	for (uint64_t i = 256 * 1024; i > 0; i -= BLOCK_SIZE) {
		add_block(frame_alloc(1024), 1024 * BLOCK_SIZE);
	}
	merge_all_blocks( );
	mem::dump_memory( );
	fb::printf("%u мегабайт выделено в динамичную память\n",
	           (256 * 1024 * BLOCK_SIZE + BLOCK_SIZE) / 1024 / 1024);
	fb::printf("%u МБ объем доступной памяти, %u МБ объем виртуальной памяти\n",
	           (bitmap_available * BLOCK_SIZE) / 1024 / 1024,
	           available / 1024 / 1024);

	fb::printf("%u / %u блоков доступно\n", bitmap_available, bitmap_limit);
	fb::printf("Проверка менеджера памяти\n");
}

} // namespace mem