Fennix/Kernel
1
0
Fork 0
mirror of https://github.com/Fennix-Project/Kernel.git synced 2025-07-11 15:29:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

memory: Refactor SwapPT class

Browse Source
This commit is contained in:
EnderIce2
2024-10-26 03:25:36 +03:00
parent b6006e379d
commit 29896bb0a4
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
include/memory/swap_pt.hpp
View File

@ -31,12 +31,19 @@ namespace Memory
private:
PageTable *Replace = nullptr;
PageTable *Restore = nullptr;
bool Ignore;
public:
SwapPT(PageTable *ReplaceWith,
PageTable *RestoreWith = nullptr)
: Replace(ReplaceWith)
SwapPT(auto ReplaceWith,
auto _RestoreWith = nullptr)
: Replace((PageTable *)ReplaceWith)
{
PageTable *RestoreWith = (PageTable *)_RestoreWith;
this->Ignore = Replace == RestoreWith;
if (this->Ignore)
return;
if (RestoreWith)
Restore = RestoreWith;
else
@ -45,7 +52,13 @@ namespace Memory
Replace->Update();
}
~SwapPT() { Restore->Update(); }
~SwapPT()
{
if (this->Ignore)
return;
Restore->Update();
}
};
}