mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-14 09:06:35 +00:00
77 lines
2.7 KiB
C++
77 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include <mesosphere.hpp>
|
|
|
|
namespace ams::kern {
|
|
|
|
Result KThreadLocalPage::Initialize(KProcess *process) {
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
/* Set that this process owns us. */
|
|
m_owner = process;
|
|
|
|
/* Allocate a new page. */
|
|
KPageBuffer *page_buf = KPageBuffer::AllocateChecked<PageSize>();
|
|
R_UNLESS(page_buf != nullptr, svc::ResultOutOfMemory());
|
|
ON_RESULT_FAILURE { KPageBuffer::Free(page_buf); };
|
|
|
|
/* Map the address in. */
|
|
R_RETURN(m_owner->GetPageTable().MapPages(std::addressof(m_virt_addr), 1, PageSize, page_buf->GetPhysicalAddress(), KMemoryState_ThreadLocal, KMemoryPermission_UserReadWrite));
|
|
}
|
|
|
|
Result KThreadLocalPage::Finalize() {
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
/* Get the physical address of the page. */
|
|
KPhysicalAddress phys_addr = Null<KPhysicalAddress>;
|
|
MESOSPHERE_ABORT_UNLESS(m_owner->GetPageTable().GetPhysicalAddress(std::addressof(phys_addr), this->GetAddress()));
|
|
|
|
/* Unmap the page. */
|
|
R_TRY(m_owner->GetPageTable().UnmapPages(this->GetAddress(), 1, KMemoryState_ThreadLocal));
|
|
|
|
/* Free the page. */
|
|
KPageBuffer::FreeChecked<PageSize>(KPageBuffer::FromPhysicalAddress(phys_addr));
|
|
R_SUCCEED();
|
|
}
|
|
|
|
KProcessAddress KThreadLocalPage::Reserve() {
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
for (size_t i = 0; i < util::size(m_is_region_free); i++) {
|
|
if (m_is_region_free[i]) {
|
|
m_is_region_free[i] = false;
|
|
return this->GetRegionAddress(i);
|
|
}
|
|
}
|
|
|
|
return Null<KProcessAddress>;
|
|
}
|
|
|
|
void KThreadLocalPage::Release(KProcessAddress addr) {
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
m_is_region_free[this->GetRegionIndex(addr)] = true;
|
|
}
|
|
|
|
void *KThreadLocalPage::GetPointer() const {
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
KPhysicalAddress phys_addr;
|
|
MESOSPHERE_ABORT_UNLESS(m_owner->GetPageTable().GetPhysicalAddress(std::addressof(phys_addr), this->GetAddress()));
|
|
return static_cast<void *>(KPageBuffer::FromPhysicalAddress(phys_addr));
|
|
}
|
|
|
|
}
|