//----------------------------------------------------------------------------//
//                                OS on Kaleid                                //
//                                                                            //
//  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;
}