mirror of
https://github.com/Fennix-Project/Kernel.git
synced 2025-07-11 07:19:20 +00:00
Refactor memory allocation implementation and add KernelReserve flag to memory mappings
This commit is contained in:
@ -19,7 +19,7 @@
|
||||
#define __FENNIX_KERNEL_MEMORY_STACK_GUARD_H__
|
||||
|
||||
#include <types.h>
|
||||
#include <vector>
|
||||
#include <list>
|
||||
|
||||
#include <memory/table.hpp>
|
||||
|
||||
@ -42,10 +42,10 @@ namespace Memory
|
||||
bool UserMode = false;
|
||||
bool Expanded = false;
|
||||
VirtualMemoryArea *vma = nullptr;
|
||||
std::vector<AllocatedPages> AllocatedPagesList;
|
||||
std::list<AllocatedPages> AllocatedPagesList;
|
||||
|
||||
public:
|
||||
std::vector<AllocatedPages> GetAllocatedPages()
|
||||
std::list<AllocatedPages> GetAllocatedPages()
|
||||
{
|
||||
return AllocatedPagesList;
|
||||
}
|
||||
|
@ -59,7 +59,7 @@ namespace Memory
|
||||
CoW = 1 << 9,
|
||||
|
||||
/** @brief Available 1 */
|
||||
AVL1 = 1 << 10,
|
||||
KRsv = 1 << 10,
|
||||
|
||||
/** @brief Available 2 */
|
||||
AVL2 = 1 << 11,
|
||||
@ -119,7 +119,7 @@ namespace Memory
|
||||
uintptr_t PageAttributeTable : 1; // 7
|
||||
uintptr_t Global : 1; // 8
|
||||
uintptr_t CopyOnWrite : 1; // 9
|
||||
uintptr_t Available1 : 1; // 10
|
||||
uintptr_t KernelReserve : 1; // 10
|
||||
uintptr_t Available2 : 1; // 11
|
||||
uintptr_t Address : 40; // 12-51
|
||||
uintptr_t Available3 : 1; // 52
|
||||
@ -142,7 +142,7 @@ namespace Memory
|
||||
uintptr_t PageAttributeTable : 1; // 7
|
||||
uintptr_t Global : 1; // 8
|
||||
uintptr_t CopyOnWrite : 1; // 9
|
||||
uintptr_t Available1 : 1; // 10
|
||||
uintptr_t KernelReserve : 1; // 10
|
||||
uintptr_t Available2 : 1; // 11
|
||||
uintptr_t Address : 20; // 12-31
|
||||
#elif defined(aa64)
|
||||
@ -204,9 +204,22 @@ namespace Memory
|
||||
uintptr_t Accessed : 1; // 5
|
||||
uintptr_t CopyOnWrite : 1; // 6
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Available1 : 4; // 8-11
|
||||
uintptr_t KernelReserve : 1; // 8
|
||||
uintptr_t Available2 : 1; // 9
|
||||
uintptr_t Available3 : 1; // 10
|
||||
uintptr_t Available4 : 1; // 11
|
||||
uintptr_t Address : 40; // 12-51
|
||||
uintptr_t Available2 : 11; // 52-62
|
||||
uintptr_t Available5 : 1; // 52
|
||||
uintptr_t Available6 : 1; // 53
|
||||
uintptr_t Available7 : 1; // 54
|
||||
uintptr_t Available8 : 1; // 55
|
||||
uintptr_t Available9 : 1; // 56
|
||||
uintptr_t Available10 : 1; // 57
|
||||
uintptr_t Available11 : 1; // 58
|
||||
uintptr_t Available12 : 1; // 59
|
||||
uintptr_t Available13 : 1; // 60
|
||||
uintptr_t Available14 : 1; // 61
|
||||
uintptr_t Available15 : 1; // 62
|
||||
uintptr_t ExecuteDisable : 1; // 63
|
||||
};
|
||||
|
||||
@ -222,7 +235,7 @@ namespace Memory
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Global : 1; // 8
|
||||
uintptr_t CopyOnWrite : 1; // 9
|
||||
uintptr_t Available1 : 1; // 10
|
||||
uintptr_t KernelReserve : 1; // 10
|
||||
uintptr_t Available2 : 1; // 11
|
||||
uintptr_t PageAttributeTable : 1; // 12
|
||||
uintptr_t Reserved0 : 8; // 13-20
|
||||
@ -248,7 +261,10 @@ namespace Memory
|
||||
uintptr_t Accessed : 1; // 5
|
||||
uintptr_t CopyOnWrite : 1; // 6
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Available1 : 4; // 8-11
|
||||
uintptr_t KernelReserve : 1; // 8
|
||||
uintptr_t Available2 : 1; // 9
|
||||
uintptr_t Available3 : 1; // 10
|
||||
uintptr_t Available4 : 1; // 11
|
||||
uintptr_t Address : 20; // 12-31
|
||||
};
|
||||
|
||||
@ -264,7 +280,7 @@ namespace Memory
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Global : 1; // 8
|
||||
uintptr_t CopyOnWrite : 1; // 9
|
||||
uintptr_t Available1 : 1; // 10
|
||||
uintptr_t KernelReserve : 1; // 10
|
||||
uintptr_t Available2 : 1; // 11
|
||||
uintptr_t PageAttributeTable : 1; // 12
|
||||
uintptr_t Address0 : 8; // 13-20
|
||||
@ -324,9 +340,22 @@ namespace Memory
|
||||
uintptr_t Accessed : 1; // 5
|
||||
uintptr_t CopyOnWrite : 1; // 6
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Available1 : 4; // 8-11
|
||||
uintptr_t KernelReserve : 1; // 8
|
||||
uintptr_t Available2 : 1; // 9
|
||||
uintptr_t Available3 : 1; // 10
|
||||
uintptr_t Available4 : 1; // 11
|
||||
uintptr_t Address : 40; // 12-51
|
||||
uintptr_t Available2 : 11; // 52-62
|
||||
uintptr_t Available5 : 1; // 52
|
||||
uintptr_t Available6 : 1; // 53
|
||||
uintptr_t Available7 : 1; // 54
|
||||
uintptr_t Available8 : 1; // 55
|
||||
uintptr_t Available9 : 1; // 56
|
||||
uintptr_t Available10 : 1; // 57
|
||||
uintptr_t Available11 : 1; // 58
|
||||
uintptr_t Available12 : 1; // 59
|
||||
uintptr_t Available13 : 1; // 60
|
||||
uintptr_t Available14 : 1; // 61
|
||||
uintptr_t Available15 : 1; // 62
|
||||
uintptr_t ExecuteDisable : 1; // 63
|
||||
};
|
||||
|
||||
@ -342,7 +371,7 @@ namespace Memory
|
||||
uintptr_t PageSize : 1; // 7
|
||||
uintptr_t Global : 1; // 8
|
||||
uintptr_t CopyOnWrite : 1; // 9
|
||||
uintptr_t Available1 : 1; // 10
|
||||
uintptr_t KernelReserve : 1; // 10
|
||||
uintptr_t Available2 : 1; // 11
|
||||
uintptr_t PageAttributeTable : 1; // 12
|
||||
uintptr_t Reserved0 : 17; // 13-29
|
||||
@ -406,9 +435,22 @@ namespace Memory
|
||||
uintptr_t Accessed : 1; // 5
|
||||
uintptr_t CopyOnWrite : 1; // 6
|
||||
uintptr_t Reserved0 : 1; // 7
|
||||
uintptr_t Available1 : 4; // 8-11
|
||||
uintptr_t KernelReserve : 1; // 8
|
||||
uintptr_t Available2 : 1; // 9
|
||||
uintptr_t Available3 : 1; // 10
|
||||
uintptr_t Available4 : 1; // 11
|
||||
uintptr_t Address : 40; // 12-51
|
||||
uintptr_t Available2 : 11; // 52-62
|
||||
uintptr_t Available5 : 1; // 52
|
||||
uintptr_t Available6 : 1; // 53
|
||||
uintptr_t Available7 : 1; // 54
|
||||
uintptr_t Available8 : 1; // 55
|
||||
uintptr_t Available9 : 1; // 56
|
||||
uintptr_t Available10 : 1; // 57
|
||||
uintptr_t Available11 : 1; // 58
|
||||
uintptr_t Available12 : 1; // 59
|
||||
uintptr_t Available13 : 1; // 60
|
||||
uintptr_t Available14 : 1; // 61
|
||||
uintptr_t Available15 : 1; // 62
|
||||
uintptr_t ExecuteDisable : 1; // 63
|
||||
};
|
||||
#elif defined(aa64)
|
||||
@ -460,9 +502,22 @@ namespace Memory
|
||||
uintptr_t Accessed : 1; // 5
|
||||
uintptr_t Available0 : 1; // 6
|
||||
uintptr_t Reserved0 : 1; // 7
|
||||
uintptr_t Available1 : 4; // 8-11
|
||||
uintptr_t KernelReserve : 1; // 8
|
||||
uintptr_t Available1 : 1; // 9
|
||||
uintptr_t Available2 : 1; // 10
|
||||
uintptr_t Available3 : 1; // 11
|
||||
uintptr_t Address : 40; // 12-51
|
||||
uintptr_t Available2 : 11; // 52-62
|
||||
uintptr_t Available4 : 1; // 52
|
||||
uintptr_t Available5 : 1; // 53
|
||||
uintptr_t Available6 : 1; // 54
|
||||
uintptr_t Available7 : 1; // 55
|
||||
uintptr_t Available8 : 1; // 56
|
||||
uintptr_t Available9 : 1; // 57
|
||||
uintptr_t Available10 : 1; // 58
|
||||
uintptr_t Available11 : 1; // 59
|
||||
uintptr_t Available12 : 1; // 60
|
||||
uintptr_t Available13 : 1; // 61
|
||||
uintptr_t Available14 : 1; // 62
|
||||
uintptr_t ExecuteDisable : 1; // 63
|
||||
};
|
||||
#elif defined(aa64)
|
||||
|
@ -22,12 +22,13 @@
|
||||
#include <filesystem.hpp>
|
||||
#include <bitmap.hpp>
|
||||
#include <lock.hpp>
|
||||
#include <vector>
|
||||
#include <list>
|
||||
|
||||
#include <memory/table.hpp>
|
||||
|
||||
namespace Memory
|
||||
{
|
||||
|
||||
class VirtualMemoryArea
|
||||
{
|
||||
public:
|
||||
@ -35,6 +36,7 @@ namespace Memory
|
||||
{
|
||||
void *Address;
|
||||
size_t PageCount;
|
||||
bool Protected;
|
||||
};
|
||||
|
||||
struct SharedRegion
|
||||
@ -49,30 +51,15 @@ namespace Memory
|
||||
private:
|
||||
NewLock(MgrLock);
|
||||
Bitmap PageBitmap;
|
||||
PageTable *Table;
|
||||
|
||||
std::vector<AllocatedPages> AllocatedPagesList;
|
||||
std::vector<SharedRegion> SharedRegions;
|
||||
std::list<AllocatedPages> AllocatedPagesList;
|
||||
std::list<SharedRegion> SharedRegions;
|
||||
|
||||
public:
|
||||
PageTable *GetTable() { return Table; }
|
||||
void SetTable(PageTable *Table) { this->Table = Table; }
|
||||
|
||||
std::vector<AllocatedPages> &GetAllocatedPagesList()
|
||||
{
|
||||
return AllocatedPagesList;
|
||||
}
|
||||
|
||||
std::vector<SharedRegion> &GetSharedRegions()
|
||||
{
|
||||
return SharedRegions;
|
||||
}
|
||||
|
||||
PageTable *Table = nullptr;
|
||||
uint64_t GetAllocatedMemorySize();
|
||||
|
||||
bool Add(void *Address, size_t Count);
|
||||
|
||||
void *RequestPages(size_t Count, bool User = false);
|
||||
void *RequestPages(size_t Count, bool User = false, bool Protect = false);
|
||||
void FreePages(void *Address, size_t Count);
|
||||
void DetachAddress(void *Address);
|
||||
|
||||
@ -88,18 +75,45 @@ namespace Memory
|
||||
* @param Shared Shared region
|
||||
* @return Address of the region
|
||||
*/
|
||||
void *CreateCoWRegion(void *Address,
|
||||
size_t Length,
|
||||
void *CreateCoWRegion(void *Address, size_t Length,
|
||||
bool Read, bool Write, bool Exec,
|
||||
bool Fixed, bool Shared);
|
||||
|
||||
bool HandleCoW(uintptr_t PFA);
|
||||
|
||||
void FreeAllPages();
|
||||
|
||||
void Fork(VirtualMemoryArea *Parent);
|
||||
|
||||
VirtualMemoryArea(PageTable *Table = nullptr);
|
||||
void Reserve(void *Address, size_t Length);
|
||||
void Unreserve(void *Address, size_t Length);
|
||||
|
||||
int Map(void *VirtualAddress, void *PhysicalAddress,
|
||||
size_t Length, uint64_t Flags);
|
||||
int Remap(void *VirtualAddress, void *PhysicalAddress, uint64_t Flags);
|
||||
int Unmap(void *VirtualAddress, size_t Length);
|
||||
void *__UserCheckAndGetAddress(void *Address, size_t Length);
|
||||
int __UserCheck(void *Address, size_t Length);
|
||||
|
||||
template <typename T>
|
||||
T UserCheckAndGetAddress(T Address, size_t Length = 0)
|
||||
{
|
||||
if (Length == 0)
|
||||
Length = sizeof(T);
|
||||
|
||||
void *PhysAddr = __UserCheckAndGetAddress((void *)Address, Length);
|
||||
if (PhysAddr == nullptr)
|
||||
return {};
|
||||
return T(PhysAddr);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
int UserCheck(T Address, size_t Length = 0)
|
||||
{
|
||||
if (Length == 0)
|
||||
Length = sizeof(T);
|
||||
return __UserCheck((void *)Address, Length);
|
||||
}
|
||||
|
||||
VirtualMemoryArea(PageTable *Table);
|
||||
~VirtualMemoryArea();
|
||||
};
|
||||
}
|
||||
|
Reference in New Issue
Block a user