154 lines
4.0 KiB
C
154 lines
4.0 KiB
C
//----------------------------------------------------------------------------//
|
|
// GNU GPL OS/K //
|
|
// //
|
|
// Desc: Early and very dumb heap managment //
|
|
// //
|
|
// //
|
|
// Copyright © 2018-2020 The OS/K Team //
|
|
// //
|
|
// This file is part of OS/K. //
|
|
// //
|
|
// OS/K is free software: you can redistribute it and/or modify //
|
|
// it under the terms of the GNU General Public License as published by //
|
|
// the Free Software Foundation, either version 3 of the License, or //
|
|
// any later version. //
|
|
// //
|
|
// OS/K is distributed in the hope that it will be useful, //
|
|
// but WITHOUT ANY WARRANTY//without even the implied warranty of //
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
|
|
// GNU General Public License for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with OS/K. If not, see <https://www.gnu.org/licenses/>. //
|
|
//----------------------------------------------------------------------------//
|
|
|
|
#include <mm/map.h>
|
|
#include <mm/heap.h>
|
|
#include <ex/lock.h>
|
|
#include <init/boot.h>
|
|
|
|
// Start address of the heap
|
|
void *_heap_start;
|
|
|
|
// End of the heap
|
|
/*static*/ void *_heap_end;
|
|
|
|
// Maximal value of the heap
|
|
/*static*/ size_t _heap_max;
|
|
|
|
// Lock NOT used internally, but used by KalAllocMemory() & co.
|
|
static Lock_t _heap_lock = 0;
|
|
|
|
//
|
|
// Initializes heap managment
|
|
//
|
|
void MmInitHeap(void)
|
|
{
|
|
assert(_heap_end == NULL);
|
|
|
|
// Get the first available zone address
|
|
_heap_start = MmGetFirstAvailZone((void*)0);
|
|
|
|
// Align it
|
|
_heap_start = (void *)_ALIGN_UP((size_t)_heap_start, alignof(QWORD));
|
|
|
|
// Initialize the heap
|
|
_heap_end = _heap_start;
|
|
_heap_max = lmin(MM_HEAP_MAX, MmGetAvailZoneSize(_heap_end));
|
|
|
|
if ((ulong)_heap_start == _heap_max)
|
|
KeStartPanic("The heap failed to initialize ! (Not enough memory)");
|
|
|
|
DebugLog("Heap initialized from %p to %p\n", _heap_start, _heap_max);
|
|
}
|
|
|
|
//
|
|
// Acquires control of the heap's lock
|
|
//
|
|
void MmLockHeap(void)
|
|
{
|
|
ExAcquireLock(&_heap_lock);
|
|
}
|
|
|
|
//
|
|
// Releases control of the heap's lock
|
|
//
|
|
void MmUnlockHeap(void)
|
|
{
|
|
ExReleaseLock(&_heap_lock);
|
|
}
|
|
|
|
//
|
|
// Returns the heap's current size
|
|
//
|
|
size_t MmGetHeapSize(void)
|
|
{
|
|
return (size_t)_heap_end - (size_t)_heap_start;
|
|
}
|
|
|
|
//
|
|
// Returns the heap's maximum size
|
|
//
|
|
size_t MmGetMaxHeapSize(void)
|
|
{
|
|
|
|
return _heap_max;
|
|
}
|
|
|
|
//
|
|
// Changes the heap's maximal size
|
|
//
|
|
error_t MmSetMaxHeapSize(size_t new)
|
|
{
|
|
|
|
if (new > MmGetAvailZoneSize((void *)_heap_start)) {
|
|
return ENOMEM;
|
|
}
|
|
|
|
if (new < (size_t)_heap_end - (size_t)_heap_start) {
|
|
return EADDRINUSE;
|
|
}
|
|
|
|
_heap_max = new;
|
|
|
|
return EOK;
|
|
}
|
|
|
|
//
|
|
// Extends the heap's size
|
|
//
|
|
error_t MmGrowHeap(size_t req)
|
|
{
|
|
|
|
assert(req % alignof(QWORD) == 0);
|
|
|
|
if ((size_t)_heap_end + req > (size_t)_heap_start + _heap_max) {
|
|
assert(!"Can't extend heap that much");
|
|
return ENOMEM;
|
|
}
|
|
|
|
_heap_end += req;
|
|
|
|
return EOK;
|
|
}
|
|
|
|
//
|
|
// Reduces the heap's size
|
|
//
|
|
error_t MmShrinkHeap(size_t req)
|
|
{
|
|
|
|
assert(req % alignof(QWORD) == 0);
|
|
|
|
if (req > (size_t)_heap_end - (size_t)_heap_start) {
|
|
assert(!"Can't shrink heap that much");
|
|
return EADDRINUSE;
|
|
}
|
|
|
|
_heap_end -= req;
|
|
|
|
return EOK;
|
|
}
|
|
|
|
|