mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-18 00:12:03 +00:00
sysupdater: implement (untested) rest of the api
This commit is contained in:
parent
f223c27bb0
commit
73a6aeed15
22 changed files with 957 additions and 57 deletions
|
@ -29,6 +29,7 @@
|
||||||
#include <stratosphere/os/os_condition_variable.hpp>
|
#include <stratosphere/os/os_condition_variable.hpp>
|
||||||
#include <stratosphere/os/os_sdk_mutex.hpp>
|
#include <stratosphere/os/os_sdk_mutex.hpp>
|
||||||
#include <stratosphere/os/os_rw_lock.hpp>
|
#include <stratosphere/os/os_rw_lock.hpp>
|
||||||
|
#include <stratosphere/os/os_transfer_memory.hpp>
|
||||||
#include <stratosphere/os/os_semaphore.hpp>
|
#include <stratosphere/os/os_semaphore.hpp>
|
||||||
#include <stratosphere/os/os_event.hpp>
|
#include <stratosphere/os/os_event.hpp>
|
||||||
#include <stratosphere/os/os_system_event.hpp>
|
#include <stratosphere/os/os_system_event.hpp>
|
||||||
|
|
|
@ -70,6 +70,11 @@ namespace ams::os {
|
||||||
return h;
|
return h;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Detach() {
|
||||||
|
const Handle h = this->Move();
|
||||||
|
AMS_UNUSED(h);
|
||||||
|
}
|
||||||
|
|
||||||
void Reset(Handle h) {
|
void Reset(Handle h) {
|
||||||
ManagedHandle(h).Swap(*this);
|
ManagedHandle(h).Swap(*this);
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,78 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#include <vapours.hpp>
|
||||||
|
#include <stratosphere/os/os_transfer_memory_types.hpp>
|
||||||
|
#include <stratosphere/os/os_transfer_memory_api.hpp>
|
||||||
|
|
||||||
|
namespace ams::os {
|
||||||
|
|
||||||
|
class TransferMemory {
|
||||||
|
NON_COPYABLE(TransferMemory);
|
||||||
|
NON_MOVEABLE(TransferMemory);
|
||||||
|
private:
|
||||||
|
TransferMemoryType tmem;
|
||||||
|
public:
|
||||||
|
constexpr TransferMemory() : tmem{ .state = TransferMemoryType::State_NotInitialized } {
|
||||||
|
/* ... */
|
||||||
|
}
|
||||||
|
|
||||||
|
TransferMemory(void *address, size_t size, MemoryPermission perm) {
|
||||||
|
R_ABORT_UNLESS(CreateTransferMemory(std::addressof(this->tmem), address, size, perm));
|
||||||
|
}
|
||||||
|
|
||||||
|
TransferMemory(size_t size, Handle handle, bool managed) {
|
||||||
|
this->Attach(size, handle, managed);
|
||||||
|
}
|
||||||
|
|
||||||
|
~TransferMemory() {
|
||||||
|
if (this->tmem.state == TransferMemoryType::State_NotInitialized) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
DestroyTransferMemory(std::addressof(this->tmem));
|
||||||
|
}
|
||||||
|
|
||||||
|
void Attach(size_t size, Handle handle, bool managed) {
|
||||||
|
AttachTransferMemory(std::addressof(this->tmem), size, handle, managed);
|
||||||
|
}
|
||||||
|
|
||||||
|
Handle Detach() {
|
||||||
|
return DetachTransferMemory(std::addressof(this->tmem));
|
||||||
|
}
|
||||||
|
|
||||||
|
Result Map(void **out, MemoryPermission owner_perm) {
|
||||||
|
return MapTransferMemory(out, std::addressof(this->tmem), owner_perm);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Unmap() {
|
||||||
|
UnmapTransferMemory(std::addressof(this->tmem));
|
||||||
|
}
|
||||||
|
|
||||||
|
operator TransferMemoryType &() {
|
||||||
|
return this->tmem;
|
||||||
|
}
|
||||||
|
|
||||||
|
operator const TransferMemoryType &() const {
|
||||||
|
return this->tmem;
|
||||||
|
}
|
||||||
|
|
||||||
|
TransferMemoryType *GetBase() {
|
||||||
|
return std::addressof(this->tmem);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,35 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#include <vapours.hpp>
|
||||||
|
#include <stratosphere/os/os_memory_permission.hpp>
|
||||||
|
|
||||||
|
namespace ams::os {
|
||||||
|
|
||||||
|
struct TransferMemoryType;
|
||||||
|
|
||||||
|
Result CreateTransferMemory(TransferMemoryType *tmem, void *address, size_t size, MemoryPermission perm);
|
||||||
|
|
||||||
|
Result AttachTransferMemory(TransferMemoryType *tmem, size_t size, Handle handle, bool managed);
|
||||||
|
Handle DetachTransferMemory(TransferMemoryType *tmem);
|
||||||
|
|
||||||
|
void DestroyTransferMemory(TransferMemoryType *tmem);
|
||||||
|
|
||||||
|
Result MapTransferMemory(void **out, TransferMemoryType *tmem, MemoryPermission owner_perm);
|
||||||
|
void UnmapTransferMemory(TransferMemoryType *tmem);
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,43 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#include <vapours.hpp>
|
||||||
|
#include <stratosphere/os/impl/os_internal_critical_section.hpp>
|
||||||
|
|
||||||
|
namespace ams::os {
|
||||||
|
|
||||||
|
struct TransferMemoryType {
|
||||||
|
enum State {
|
||||||
|
State_NotInitialized = 0,
|
||||||
|
State_Created = 1,
|
||||||
|
State_Mapped = 2,
|
||||||
|
State_Detached = 3,
|
||||||
|
};
|
||||||
|
|
||||||
|
u8 state;
|
||||||
|
bool handle_managed;
|
||||||
|
bool allocated;
|
||||||
|
|
||||||
|
void *address;
|
||||||
|
size_t size;
|
||||||
|
Handle handle;
|
||||||
|
|
||||||
|
mutable impl::InternalCriticalSectionStorage cs_transfer_memory;
|
||||||
|
};
|
||||||
|
static_assert(std::is_trivial<TransferMemoryType>::value);
|
||||||
|
|
||||||
|
}
|
|
@ -23,6 +23,7 @@
|
||||||
#include <stratosphere/settings/factory/settings_device_certificate.hpp>
|
#include <stratosphere/settings/factory/settings_device_certificate.hpp>
|
||||||
#include <stratosphere/settings/system/settings_error_report.hpp>
|
#include <stratosphere/settings/system/settings_error_report.hpp>
|
||||||
#include <stratosphere/settings/system/settings_firmware_version.hpp>
|
#include <stratosphere/settings/system/settings_firmware_version.hpp>
|
||||||
|
#include <stratosphere/settings/system/settings_platform_region.hpp>
|
||||||
#include <stratosphere/settings/system/settings_product_model.hpp>
|
#include <stratosphere/settings/system/settings_product_model.hpp>
|
||||||
#include <stratosphere/settings/system/settings_region.hpp>
|
#include <stratosphere/settings/system/settings_region.hpp>
|
||||||
#include <stratosphere/settings/system/settings_serial_number.hpp>
|
#include <stratosphere/settings/system/settings_serial_number.hpp>
|
||||||
|
|
|
@ -0,0 +1,31 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
#include <vapours.hpp>
|
||||||
|
#include <stratosphere/settings/settings_types.hpp>
|
||||||
|
|
||||||
|
namespace ams::settings::system {
|
||||||
|
|
||||||
|
enum PlatformRegion {
|
||||||
|
PlatformRegion_Invalid = 0,
|
||||||
|
PlatformRegion_Global = 1,
|
||||||
|
PlatformRegion_China = 2,
|
||||||
|
};
|
||||||
|
|
||||||
|
PlatformRegion GetPlatformRegion();
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,30 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#include <stratosphere.hpp>
|
||||||
|
|
||||||
|
namespace ams::os::impl {
|
||||||
|
|
||||||
|
class TransferMemoryImpl {
|
||||||
|
public:
|
||||||
|
static Result Create(Handle *out, void *address, size_t size, MemoryPermission perm);
|
||||||
|
static void Close(Handle handle);
|
||||||
|
|
||||||
|
static Result Map(void **out, Handle handle, void *address, size_t size, MemoryPermission owner_perm);
|
||||||
|
static void Unmap(Handle handle, void *address, size_t size);
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,77 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||||
|
#include "os_transfer_memory_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::os::impl {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
svc::MemoryPermission ConvertToSvcMemoryPermission(os::MemoryPermission perm) {
|
||||||
|
switch (perm) {
|
||||||
|
case os::MemoryPermission_None: return svc::MemoryPermission_None;
|
||||||
|
case os::MemoryPermission_ReadOnly: return svc::MemoryPermission_Read;
|
||||||
|
case os::MemoryPermission_WriteOnly: return svc::MemoryPermission_Write;
|
||||||
|
case os::MemoryPermission_ReadWrite: return svc::MemoryPermission_ReadWrite;
|
||||||
|
AMS_UNREACHABLE_DEFAULT_CASE();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
Result TransferMemoryImpl::Create(Handle *out, void *address, size_t size, MemoryPermission perm) {
|
||||||
|
/* Convert memory permission. */
|
||||||
|
auto svc_perm = ConvertToSvcMemoryPermission(perm);
|
||||||
|
|
||||||
|
/* Create the memory. */
|
||||||
|
svc::Handle handle;
|
||||||
|
R_TRY_CATCH(svc::CreateTransferMemory(std::addressof(handle), reinterpret_cast<uintptr_t>(address), size, svc_perm)) {
|
||||||
|
R_CONVERT(svc::ResultOutOfHandles, os::ResultOutOfHandles())
|
||||||
|
R_CONVERT(svc::ResultOutOfResource, os::ResultOutOfTransferMemory())
|
||||||
|
} R_END_TRY_CATCH_WITH_ABORT_UNLESS;
|
||||||
|
|
||||||
|
*out = handle;
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
void TransferMemoryImpl::Close(Handle handle) {
|
||||||
|
R_ABORT_UNLESS(svc::CloseHandle(handle));
|
||||||
|
}
|
||||||
|
|
||||||
|
Result TransferMemoryImpl::Map(void **out, Handle handle, void *address, size_t size, MemoryPermission owner_perm) {
|
||||||
|
AMS_ASSERT(address != nullptr);
|
||||||
|
|
||||||
|
/* Convert memory permission. */
|
||||||
|
auto svc_owner_perm = ConvertToSvcMemoryPermission(owner_perm);
|
||||||
|
|
||||||
|
/* Map the memory. */
|
||||||
|
R_TRY_CATCH(svc::MapTransferMemory(handle, reinterpret_cast<uintptr_t>(address), size, svc_owner_perm)) {
|
||||||
|
R_CONVERT(svc::ResultInvalidHandle, os::ResultInvalidHandle())
|
||||||
|
R_CONVERT(svc::ResultInvalidSize, os::ResultInvalidTransferMemorySize())
|
||||||
|
R_CONVERT(svc::ResultInvalidState, os::ResultInvalidTransferMemoryState())
|
||||||
|
R_CONVERT(svc::ResultInvalidCurrentMemory, os::ResultInvalidCurrentMemoryState())
|
||||||
|
R_CONVERT(svc::ResultInvalidMemoryRegion, os::ResultInvalidCurrentMemoryState())
|
||||||
|
} R_END_TRY_CATCH_WITH_ABORT_UNLESS;
|
||||||
|
|
||||||
|
*out = address;
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
void TransferMemoryImpl::Unmap(Handle handle, void *address, size_t size) {
|
||||||
|
R_ABORT_UNLESS(svc::UnmapTransferMemory(handle, reinterpret_cast<uintptr_t>(address), size));
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
185
libraries/libstratosphere/source/os/os_transfer_memory_api.cpp
Normal file
185
libraries/libstratosphere/source/os/os_transfer_memory_api.cpp
Normal file
|
@ -0,0 +1,185 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||||
|
#include "impl/os_thread_manager.hpp"
|
||||||
|
#include "impl/os_transfer_memory_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::os {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
Result MapTransferMemoryWithAddressUnsafe(TransferMemoryType *tmem, void *address, os::MemoryPermission owner_perm) {
|
||||||
|
/* Map the transfer memory. */
|
||||||
|
void *mapped_address = nullptr;
|
||||||
|
R_TRY(impl::TransferMemoryImpl::Map(std::addressof(mapped_address), tmem->handle, address, tmem->size, owner_perm));
|
||||||
|
|
||||||
|
/* Set fields now that we've mapped. */
|
||||||
|
tmem->address = mapped_address;
|
||||||
|
tmem->state = TransferMemoryType::State_Mapped;
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void SetupTransferMemoryType(TransferMemoryType *tmem, size_t size, Handle handle, bool managed) {
|
||||||
|
/* Set members. */
|
||||||
|
tmem->handle = handle;
|
||||||
|
tmem->size = size;
|
||||||
|
tmem->address = nullptr;
|
||||||
|
tmem->allocated = false;
|
||||||
|
|
||||||
|
/* Set managed. */
|
||||||
|
tmem->handle_managed = managed;
|
||||||
|
|
||||||
|
/* Create the critical section. */
|
||||||
|
new (GetPointer(tmem->cs_transfer_memory)) impl::InternalCriticalSection;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
Result CreateTransferMemory(TransferMemoryType *tmem, void *address, size_t size, MemoryPermission perm) {
|
||||||
|
/* Validate pre-conditions. */
|
||||||
|
AMS_ASSERT(size > 0);
|
||||||
|
AMS_ASSERT(util::IsAligned(size, os::MemoryPageSize));
|
||||||
|
AMS_ASSERT(address != nullptr);
|
||||||
|
AMS_ASSERT(util::IsAligned(reinterpret_cast<uintptr_t>(address), os::MemoryPageSize));
|
||||||
|
|
||||||
|
/* Create the memory. */
|
||||||
|
Handle handle;
|
||||||
|
R_TRY(impl::TransferMemoryImpl::Create(std::addressof(handle), address, size, perm));
|
||||||
|
|
||||||
|
/* Setup the object. */
|
||||||
|
SetupTransferMemoryType(tmem, size, handle, true);
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result AttachTransferMemory(TransferMemoryType *tmem, size_t size, Handle handle, bool managed) {
|
||||||
|
AMS_ASSERT(size > 0);
|
||||||
|
AMS_ASSERT(util::IsAligned(size, os::MemoryPageSize));
|
||||||
|
AMS_ASSERT(handle != svc::InvalidHandle);
|
||||||
|
|
||||||
|
/* Setup the object. */
|
||||||
|
SetupTransferMemoryType(tmem, size, handle, managed);
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Handle DetachTransferMemory(TransferMemoryType *tmem) {
|
||||||
|
AMS_ASSERT(tmem->state == TransferMemoryType::State_Created);
|
||||||
|
|
||||||
|
/* Set state to detached. */
|
||||||
|
tmem->state = TransferMemoryType::State_Detached;
|
||||||
|
|
||||||
|
/* Clear handle. */
|
||||||
|
Handle handle = tmem->handle;
|
||||||
|
|
||||||
|
tmem->handle = svc::InvalidHandle;
|
||||||
|
tmem->handle_managed = false;
|
||||||
|
|
||||||
|
return handle;
|
||||||
|
}
|
||||||
|
|
||||||
|
void DestroyTransferMemory(TransferMemoryType *tmem) {
|
||||||
|
/* Unmap the transfer memory, if required. */
|
||||||
|
if (tmem->state == TransferMemoryType::State_Mapped) {
|
||||||
|
UnmapTransferMemory(tmem);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Check the state is valid. */
|
||||||
|
AMS_ASSERT(tmem->state == TransferMemoryType::State_Created || tmem->state == TransferMemoryType::State_Detached);
|
||||||
|
|
||||||
|
/* Set state to not initialized. */
|
||||||
|
tmem->state = TransferMemoryType::State_NotInitialized;
|
||||||
|
|
||||||
|
/* Close the handle, if it's managed. */
|
||||||
|
if (tmem->handle_managed) {
|
||||||
|
impl::TransferMemoryImpl::Close(tmem->handle);
|
||||||
|
}
|
||||||
|
tmem->handle_managed = false;
|
||||||
|
|
||||||
|
/* Clear members. */
|
||||||
|
tmem->address = nullptr;
|
||||||
|
tmem->size = 0;
|
||||||
|
tmem->handle = svc::InvalidHandle;
|
||||||
|
|
||||||
|
/* Destroy the critical section. */
|
||||||
|
GetReference(tmem->cs_transfer_memory).~InternalCriticalSection();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result MapTransferMemory(void **out, TransferMemoryType *tmem, MemoryPermission owner_perm) {
|
||||||
|
/* Lock the current thread, and then the transfer memory. */
|
||||||
|
std::scoped_lock thread_lk(GetReference(impl::GetCurrentThread()->cs_thread));
|
||||||
|
std::scoped_lock lk(GetReference(tmem->cs_transfer_memory));
|
||||||
|
|
||||||
|
/* Ensure we're in a mappable state. */
|
||||||
|
AMS_ASSERT(tmem->state == TransferMemoryType::State_Created);
|
||||||
|
|
||||||
|
/* Try to map up to 64 times. */
|
||||||
|
for (int i = 0; i < 64; ++i) {
|
||||||
|
/* Reserve space to map the memory. */
|
||||||
|
/* TODO: os::AslrSpaceManager */
|
||||||
|
void *map_address = ::virtmemReserve(tmem->size);
|
||||||
|
R_UNLESS(map_address != nullptr, os::ResultOutOfAddressSpace());
|
||||||
|
|
||||||
|
/* Mark allocated. */
|
||||||
|
tmem->allocated = true;
|
||||||
|
auto alloc_guard = SCOPE_GUARD { tmem->allocated = false; };
|
||||||
|
|
||||||
|
/* Try to map. */
|
||||||
|
R_TRY_CATCH(MapTransferMemoryWithAddressUnsafe(tmem, map_address, owner_perm)) {
|
||||||
|
/* If we failed to map at the address, retry. */
|
||||||
|
R_CATCH(os::ResultInvalidCurrentMemoryState) { continue; }
|
||||||
|
} R_END_TRY_CATCH;
|
||||||
|
|
||||||
|
/* TODO: Check guard space via aslr manager. */
|
||||||
|
if (false /* !impl::GetAslrSpaceManager()->CheckGuardSpace(reinterpret_cast<uintptr_t>(tmem->address), tmem->size) */) {
|
||||||
|
impl::TransferMemoryImpl::Unmap(tmem->handle, tmem->address, tmem->size);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* We mapped successfully. */
|
||||||
|
alloc_guard.Cancel();
|
||||||
|
*out = tmem->address;
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
/* We failed to map. */
|
||||||
|
return os::ResultOutOfAddressSpace();
|
||||||
|
}
|
||||||
|
|
||||||
|
void UnmapTransferMemory(TransferMemoryType *tmem) {
|
||||||
|
/* Lock the memory. */
|
||||||
|
std::scoped_lock lk(GetReference(tmem->cs_transfer_memory));
|
||||||
|
|
||||||
|
/* If the memory isn't mapped, we can't unmap it. */
|
||||||
|
if (tmem->state != TransferMemoryType::State_Mapped) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Unmap the memory. */
|
||||||
|
impl::TransferMemoryImpl::Unmap(tmem->handle, tmem->address, tmem->size);
|
||||||
|
|
||||||
|
/* Unmapped memory is necessarily not allocated. */
|
||||||
|
if (tmem->allocated) {
|
||||||
|
tmem->allocated = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Clear the address. */
|
||||||
|
tmem->address = nullptr;
|
||||||
|
tmem->state = TransferMemoryType::State_Created;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,26 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||||
|
#include "settings_platform_region_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::settings::impl {
|
||||||
|
|
||||||
|
Result GetPlatformRegion(s32 *out) {
|
||||||
|
static_assert(sizeof(*out) == sizeof(::SetSysPlatformRegion));
|
||||||
|
return ::setsysGetPlatformRegion(reinterpret_cast<::SetSysPlatformRegion *>(out));
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,23 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#include <stratosphere.hpp>
|
||||||
|
|
||||||
|
namespace ams::settings::impl {
|
||||||
|
|
||||||
|
Result GetPlatformRegion(s32 *out);
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,31 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||||
|
#include "impl/settings_platform_region_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::settings::system {
|
||||||
|
|
||||||
|
PlatformRegion GetPlatformRegion() {
|
||||||
|
if (hos::GetVersion() >= hos::Version_9_0_0) {
|
||||||
|
s32 region = 0;
|
||||||
|
R_ABORT_UNLESS(settings::impl::GetPlatformRegion(std::addressof(region)));
|
||||||
|
return static_cast<PlatformRegion>(region);
|
||||||
|
} else {
|
||||||
|
return PlatformRegion_Global;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -21,7 +21,11 @@ namespace ams::ns {
|
||||||
|
|
||||||
R_DEFINE_NAMESPACE_RESULT_MODULE(16);
|
R_DEFINE_NAMESPACE_RESULT_MODULE(16);
|
||||||
|
|
||||||
R_DEFINE_ERROR_RESULT(Canceled, 90);
|
R_DEFINE_ERROR_RESULT(Canceled, 90);
|
||||||
R_DEFINE_ERROR_RESULT(OutOfMaxRunningTask, 110);
|
R_DEFINE_ERROR_RESULT(OutOfMaxRunningTask, 110);
|
||||||
|
R_DEFINE_ERROR_RESULT(CardUpdateNotSetup, 270);
|
||||||
|
R_DEFINE_ERROR_RESULT(CardUpdateNotPrepared, 280);
|
||||||
|
R_DEFINE_ERROR_RESULT(CardUpdateAlreadySetup, 290);
|
||||||
|
R_DEFINE_ERROR_RESULT(PrepareCardUpdateAlreadyRequested, 460);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -21,12 +21,20 @@ namespace ams::os {
|
||||||
|
|
||||||
R_DEFINE_NAMESPACE_RESULT_MODULE(3);
|
R_DEFINE_NAMESPACE_RESULT_MODULE(3);
|
||||||
|
|
||||||
R_DEFINE_ERROR_RESULT(Busy, 4);
|
R_DEFINE_ERROR_RESULT(Busy, 4);
|
||||||
|
|
||||||
R_DEFINE_ERROR_RESULT(OutOfMemory, 8);
|
R_DEFINE_ERROR_RESULT(OutOfMemory, 8);
|
||||||
R_DEFINE_ERROR_RESULT(OutOfResource, 9);
|
R_DEFINE_ERROR_RESULT(OutOfResource, 9);
|
||||||
|
|
||||||
R_DEFINE_ERROR_RESULT(OutOfVirtualAddressSpace, 12);
|
R_DEFINE_ERROR_RESULT(OutOfVirtualAddressSpace, 12);
|
||||||
R_DEFINE_ERROR_RESULT(ResourceLimit, 13);
|
R_DEFINE_ERROR_RESULT(ResourceLimit, 13);
|
||||||
|
|
||||||
|
R_DEFINE_ERROR_RESULT(OutOfHandles, 500);
|
||||||
|
R_DEFINE_ERROR_RESULT(InvalidHandle, 501);
|
||||||
|
R_DEFINE_ERROR_RESULT(InvalidCurrentMemoryState, 502);
|
||||||
|
R_DEFINE_ERROR_RESULT(InvalidTransferMemoryState, 503);
|
||||||
|
R_DEFINE_ERROR_RESULT(InvalidTransferMemorySize, 504);
|
||||||
|
R_DEFINE_ERROR_RESULT(OutOfTransferMemory, 505);
|
||||||
|
R_DEFINE_ERROR_RESULT(OutOfAddressSpace, 506);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,102 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||||
|
#include "sysupdater_apply_manager.hpp"
|
||||||
|
|
||||||
|
namespace ams::mitm::sysupdater {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
alignas(os::MemoryPageSize) u8 g_boot_image_update_buffer[64_KB];
|
||||||
|
|
||||||
|
updater::BootImageUpdateType GetBootImageUpdateType() {
|
||||||
|
int boot_image_update_type;
|
||||||
|
auto size = settings::fwdbg::GetSettingsItemValue(std::addressof(boot_image_update_type), sizeof(boot_image_update_type), "systeminitializer", "boot_image_update_type");
|
||||||
|
if (size != sizeof(boot_image_update_type)) {
|
||||||
|
return updater::BootImageUpdateType::Erista;
|
||||||
|
}
|
||||||
|
return updater::GetBootImageUpdateType(boot_image_update_type);
|
||||||
|
}
|
||||||
|
|
||||||
|
Result MarkPreCommitForBootImages() {
|
||||||
|
/* Set verification required for both normal and safe mode. */
|
||||||
|
R_TRY(updater::MarkVerifyingRequired(updater::BootModeType::Normal, g_boot_image_update_buffer, sizeof(g_boot_image_update_buffer)));
|
||||||
|
R_TRY(updater::MarkVerifyingRequired(updater::BootModeType::Safe, g_boot_image_update_buffer, sizeof(g_boot_image_update_buffer)));
|
||||||
|
|
||||||
|
/* Pre-commit is now marked. */
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result UpdateBootImages() {
|
||||||
|
/* Define a helper to update the images. */
|
||||||
|
auto UpdateBootImageImpl = [](updater::BootModeType boot_mode, updater::BootImageUpdateType boot_image_update_type) -> Result {
|
||||||
|
/* Get the boot image package id. */
|
||||||
|
ncm::SystemDataId boot_image_package_id = {};
|
||||||
|
R_TRY_CATCH(updater::GetBootImagePackageId(std::addressof(boot_image_package_id), boot_mode, g_boot_image_update_buffer, sizeof(g_boot_image_update_buffer))) {
|
||||||
|
R_CATCH(updater::ResultBootImagePackageNotFound) {
|
||||||
|
/* Nintendo simply falls through when the package is not found. */
|
||||||
|
}
|
||||||
|
} R_END_TRY_CATCH;
|
||||||
|
|
||||||
|
|
||||||
|
/* Update the boot images. */
|
||||||
|
R_TRY_CATCH(updater::UpdateBootImagesFromPackage(boot_image_package_id, boot_mode, g_boot_image_update_buffer, sizeof(g_boot_image_update_buffer), boot_image_update_type)) {
|
||||||
|
R_CATCH(updater::ResultBootImagePackageNotFound) {
|
||||||
|
/* Nintendo simply falls through when the package is not found. */
|
||||||
|
}
|
||||||
|
} R_END_TRY_CATCH;
|
||||||
|
|
||||||
|
/* Mark the images verified. */
|
||||||
|
R_TRY(updater::MarkVerified(boot_mode, g_boot_image_update_buffer, sizeof(g_boot_image_update_buffer)));
|
||||||
|
|
||||||
|
/* The boot images are updated. */
|
||||||
|
return ResultSuccess();
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Get the boot image update type. */
|
||||||
|
auto boot_image_update_type = GetBootImageUpdateType();
|
||||||
|
|
||||||
|
/* Update boot images for safe mode. */
|
||||||
|
R_TRY(UpdateBootImageImpl(updater::BootModeType::Safe, boot_image_update_type));
|
||||||
|
|
||||||
|
/* Update boot images for normal mode. */
|
||||||
|
R_TRY(UpdateBootImageImpl(updater::BootModeType::Normal, boot_image_update_type));
|
||||||
|
|
||||||
|
/* Both sets of images are updated. */
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateApplyManager::ApplyPackageTask(ncm::PackageSystemDowngradeTask *task) {
|
||||||
|
/* Lock the apply mutex. */
|
||||||
|
std::scoped_lock lk(this->apply_mutex);
|
||||||
|
|
||||||
|
/* NOTE: Here, Nintendo creates a system report for the update. */
|
||||||
|
|
||||||
|
/* Mark boot images to note that we're updating. */
|
||||||
|
R_TRY(MarkPreCommitForBootImages());
|
||||||
|
|
||||||
|
/* Commit the task. */
|
||||||
|
R_TRY(task->Commit());
|
||||||
|
|
||||||
|
/* Update the boot images. */
|
||||||
|
R_TRY(UpdateBootImages());
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,30 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#include <stratosphere.hpp>
|
||||||
|
|
||||||
|
namespace ams::mitm::sysupdater {
|
||||||
|
|
||||||
|
class SystemUpdateApplyManager {
|
||||||
|
private:
|
||||||
|
os::Mutex apply_mutex;
|
||||||
|
public:
|
||||||
|
constexpr SystemUpdateApplyManager() : apply_mutex(false) { /* ... */ }
|
||||||
|
|
||||||
|
Result ApplyPackageTask(ncm::PackageSystemDowngradeTask *task);
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -15,6 +15,7 @@
|
||||||
*/
|
*/
|
||||||
#pragma once
|
#pragma once
|
||||||
#include <stratosphere.hpp>
|
#include <stratosphere.hpp>
|
||||||
|
#include "sysupdater_i_async_result.hpp"
|
||||||
#include "sysupdater_thread_allocator.hpp"
|
#include "sysupdater_thread_allocator.hpp"
|
||||||
|
|
||||||
namespace ams::mitm::sysupdater {
|
namespace ams::mitm::sysupdater {
|
||||||
|
@ -58,18 +59,18 @@ namespace ams::mitm::sysupdater {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
class AsyncBase {
|
class AsyncBase : public IAsyncBase {
|
||||||
public:
|
public:
|
||||||
virtual ~AsyncBase() { /* ... */ }
|
virtual ~AsyncBase() { /* ... */ }
|
||||||
|
|
||||||
Result Cancel() {
|
static Result ToAsyncResult(Result result);
|
||||||
|
|
||||||
|
virtual Result Cancel() override final {
|
||||||
this->CancelImpl();
|
this->CancelImpl();
|
||||||
return ResultSuccess();
|
return ResultSuccess();
|
||||||
}
|
}
|
||||||
|
|
||||||
static Result ToAsyncResult(Result result);
|
virtual Result GetErrorContext(sf::Out<err::ErrorContext> out) override {
|
||||||
|
|
||||||
virtual Result GetErrorContext(sf::Out<err::ErrorContext> out) {
|
|
||||||
*out = {};
|
*out = {};
|
||||||
return ResultSuccess();
|
return ResultSuccess();
|
||||||
}
|
}
|
||||||
|
@ -77,12 +78,27 @@ namespace ams::mitm::sysupdater {
|
||||||
virtual void CancelImpl() = 0;
|
virtual void CancelImpl() = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
class AsyncResultBase : public AsyncBase {
|
class AsyncResultBase : public IAsyncResult {
|
||||||
public:
|
public:
|
||||||
Result Get() {
|
virtual ~AsyncResultBase() { /* ... */ }
|
||||||
|
|
||||||
|
static Result ToAsyncResult(Result result) { return AsyncBase::ToAsyncResult(result); }
|
||||||
|
|
||||||
|
virtual Result Cancel() override final {
|
||||||
|
this->CancelImpl();
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual Result Get() override final {
|
||||||
return ToAsyncResult(this->GetImpl());
|
return ToAsyncResult(this->GetImpl());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
virtual Result GetErrorContext(sf::Out<err::ErrorContext> out) override {
|
||||||
|
*out = {};
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
private:
|
private:
|
||||||
|
virtual void CancelImpl() = 0;
|
||||||
virtual Result GetImpl() = 0;
|
virtual Result GetImpl() = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,44 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2018-2020 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/>.
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
#include <stratosphere.hpp>
|
||||||
|
|
||||||
|
namespace ams::mitm::sysupdater {
|
||||||
|
|
||||||
|
class IAsyncBase : public sf::IServiceObject {
|
||||||
|
public:
|
||||||
|
virtual Result Cancel() = 0;
|
||||||
|
virtual Result GetErrorContext(sf::Out<err::ErrorContext> out) = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
class IAsyncResult : public IAsyncBase {
|
||||||
|
private:
|
||||||
|
enum class CommandId {
|
||||||
|
Get = 0,
|
||||||
|
Cancel = 1,
|
||||||
|
GetErrorContext = 2,
|
||||||
|
};
|
||||||
|
public:
|
||||||
|
virtual Result Get() = 0;
|
||||||
|
public:
|
||||||
|
DEFINE_SERVICE_DISPATCH_TABLE {
|
||||||
|
MAKE_SERVICE_COMMAND_META(Get),
|
||||||
|
MAKE_SERVICE_COMMAND_META(Cancel),
|
||||||
|
MAKE_SERVICE_COMMAND_META(GetErrorContext),
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -27,7 +27,7 @@ namespace ams::mitm::sysupdater {
|
||||||
constexpr size_t SystemUpdateMaxSessions = 1;
|
constexpr size_t SystemUpdateMaxSessions = 1;
|
||||||
|
|
||||||
constexpr size_t MaxServers = 1;
|
constexpr size_t MaxServers = 1;
|
||||||
constexpr size_t MaxSessions = SystemUpdateMaxSessions;
|
constexpr size_t MaxSessions = SystemUpdateMaxSessions + 3;
|
||||||
|
|
||||||
struct ServerOptions {
|
struct ServerOptions {
|
||||||
static constexpr size_t PointerBufferSize = 1_KB;
|
static constexpr size_t PointerBufferSize = 1_KB;
|
||||||
|
@ -37,6 +37,8 @@ namespace ams::mitm::sysupdater {
|
||||||
|
|
||||||
sf::hipc::ServerManager<MaxServers, ServerOptions, MaxSessions> g_server_manager;
|
sf::hipc::ServerManager<MaxServers, ServerOptions, MaxSessions> g_server_manager;
|
||||||
|
|
||||||
|
constinit sysupdater::SystemUpdateService g_system_update_service_object;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void MitmModule::ThreadFunction(void *arg) {
|
void MitmModule::ThreadFunction(void *arg) {
|
||||||
|
@ -48,7 +50,7 @@ namespace ams::mitm::sysupdater {
|
||||||
ON_SCOPE_EXIT { nim::FinalizeForNetworkInstallManager(); };
|
ON_SCOPE_EXIT { nim::FinalizeForNetworkInstallManager(); };
|
||||||
|
|
||||||
/* Register ams:su. */
|
/* Register ams:su. */
|
||||||
R_ABORT_UNLESS((g_server_manager.RegisterServer<sysupdater::SystemUpdateService>(SystemUpdateServiceName, SystemUpdateMaxSessions)));
|
R_ABORT_UNLESS((g_server_manager.RegisterServer<sysupdater::SystemUpdateService>(SystemUpdateServiceName, SystemUpdateMaxSessions, sf::ServiceObjectTraits<sysupdater::SystemUpdateService>::SharedPointerHelper::GetEmptyDeleteSharedPointer(std::addressof(g_system_update_service_object)))));
|
||||||
|
|
||||||
/* Loop forever, servicing our services. */
|
/* Loop forever, servicing our services. */
|
||||||
g_server_manager.LoopProcess();
|
g_server_manager.LoopProcess();
|
||||||
|
|
|
@ -15,6 +15,7 @@
|
||||||
*/
|
*/
|
||||||
#include <stratosphere.hpp>
|
#include <stratosphere.hpp>
|
||||||
#include "sysupdater_service.hpp"
|
#include "sysupdater_service.hpp"
|
||||||
|
#include "sysupdater_async_impl.hpp"
|
||||||
#include "sysupdater_fs_utils.hpp"
|
#include "sysupdater_fs_utils.hpp"
|
||||||
|
|
||||||
namespace ams::mitm::sysupdater {
|
namespace ams::mitm::sysupdater {
|
||||||
|
@ -275,45 +276,6 @@ namespace ams::mitm::sysupdater {
|
||||||
return ResultSuccess();
|
return ResultSuccess();
|
||||||
}
|
}
|
||||||
|
|
||||||
Result ActivateSystemUpdateContentMetaDatabase() {
|
|
||||||
/* TODO: Don't use gamecard db. */
|
|
||||||
return ncm::ActivateContentMetaDatabase(ncm::StorageId::GameCard);
|
|
||||||
}
|
|
||||||
|
|
||||||
void InactivateSystemUpdateContentMetaDatabase() {
|
|
||||||
/* TODO: Don't use gamecard db. */
|
|
||||||
ncm::InactivateContentMetaDatabase(ncm::StorageId::GameCard);
|
|
||||||
}
|
|
||||||
|
|
||||||
Result OpenSystemUpdateContentMetaDatabase(ncm::ContentMetaDatabase *out) {
|
|
||||||
/* TODO: Don't use gamecard db. */
|
|
||||||
return ncm::OpenContentMetaDatabase(out, ncm::StorageId::GameCard);
|
|
||||||
}
|
|
||||||
|
|
||||||
Result GetContentInfoOfContentMeta(ncm::ContentInfo *out, ncm::ContentMetaDatabase &db, const ncm::ContentMetaKey &key) {
|
|
||||||
s32 ofs = 0;
|
|
||||||
while (true) {
|
|
||||||
/* List content infos. */
|
|
||||||
s32 count;
|
|
||||||
ncm::ContentInfo info;
|
|
||||||
R_TRY(db.ListContentInfo(std::addressof(count), std::addressof(info), 1, key, ofs++));
|
|
||||||
|
|
||||||
/* No content infos left to list. */
|
|
||||||
if (count == 0) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Check if the info is for meta content. */
|
|
||||||
if (info.GetType() == ncm::ContentType::Meta) {
|
|
||||||
*out = info;
|
|
||||||
return ResultSuccess();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Not found. */
|
|
||||||
return ncm::ResultContentInfoNotFound();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool IsExFatDriverSupported(const ncm::ContentMetaInfo &info) {
|
bool IsExFatDriverSupported(const ncm::ContentMetaInfo &info) {
|
||||||
return info.version >= MinimumVersionForExFatDriver && ((info.attributes & ncm::ContentMetaAttribute_IncludesExFatDriver) != 0);
|
return info.version >= MinimumVersionForExFatDriver && ((info.attributes & ncm::ContentMetaAttribute_IncludesExFatDriver) != 0);
|
||||||
}
|
}
|
||||||
|
@ -334,6 +296,25 @@ namespace ams::mitm::sysupdater {
|
||||||
return ResultSuccess();
|
return ResultSuccess();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const char *GetFirmwareVariationSettingName(settings::system::PlatformRegion region) {
|
||||||
|
switch (region) {
|
||||||
|
case settings::system::PlatformRegion_Global: return "firmware_variation";
|
||||||
|
case settings::system::PlatformRegion_China: return "t_firmware_variation";
|
||||||
|
AMS_UNREACHABLE_DEFAULT_CASE();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ncm::FirmwareVariationId GetFirmwareVariationId() {
|
||||||
|
/* Get the firmware variation setting name. */
|
||||||
|
const char * const setting_name = GetFirmwareVariationSettingName(settings::system::GetPlatformRegion());
|
||||||
|
|
||||||
|
/* Retrieve the firmware variation id. */
|
||||||
|
ncm::FirmwareVariationId id = {};
|
||||||
|
settings::fwdbg::GetSettingsItemValue(std::addressof(id.value), sizeof(u8), "ns.systemupdate", setting_name);
|
||||||
|
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Result SystemUpdateService::GetUpdateInformation(sf::Out<UpdateInformation> out, const ncm::Path &path) {
|
Result SystemUpdateService::GetUpdateInformation(sf::Out<UpdateInformation> out, const ncm::Path &path) {
|
||||||
|
@ -422,4 +403,115 @@ namespace ams::mitm::sysupdater {
|
||||||
return ResultSuccess();
|
return ResultSuccess();
|
||||||
};
|
};
|
||||||
|
|
||||||
|
Result SystemUpdateService::SetupUpdate(sf::MoveHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat) {
|
||||||
|
return this->SetupUpdateImpl(transfer_memory.GetValue(), transfer_memory_size, path, exfat, GetFirmwareVariationId());
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::SetupUpdateWithVariation(sf::MoveHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id) {
|
||||||
|
return this->SetupUpdateImpl(transfer_memory.GetValue(), transfer_memory_size, path, exfat, firmware_variation_id);
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::RequestPrepareUpdate(sf::OutCopyHandle out_event_handle, sf::Out<std::shared_ptr<IAsyncResult>> out_async) {
|
||||||
|
/* Ensure the update is setup but not prepared. */
|
||||||
|
R_UNLESS(this->setup_update, ns::ResultCardUpdateNotSetup());
|
||||||
|
R_UNLESS(!this->requested_update, ns::ResultPrepareCardUpdateAlreadyRequested());
|
||||||
|
|
||||||
|
/* Create the async result. */
|
||||||
|
auto async_result = std::make_shared<AsyncPrepareSdCardUpdateImpl>(std::addressof(*this->update_task));
|
||||||
|
R_UNLESS(async_result != nullptr, ns::ResultOutOfMaxRunningTask());
|
||||||
|
|
||||||
|
/* Run the task. */
|
||||||
|
R_TRY(async_result->Run());
|
||||||
|
|
||||||
|
/* We prepared the task! */
|
||||||
|
this->requested_update = true;
|
||||||
|
out_event_handle.SetValue(async_result->GetEvent().GetReadableHandle());
|
||||||
|
out_async.SetValue(std::move(async_result));
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::GetPrepareUpdateProgress(sf::Out<SystemUpdateProgress> out) {
|
||||||
|
/* Ensure the update is setup. */
|
||||||
|
R_UNLESS(this->setup_update, ns::ResultCardUpdateNotSetup());
|
||||||
|
|
||||||
|
/* Get the progress. */
|
||||||
|
auto install_progress = this->update_task->GetProgress();
|
||||||
|
out.SetValue({ .current_size = install_progress.installed_size, .total_size = install_progress.total_size });
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::HasPreparedUpdate(sf::Out<bool> out) {
|
||||||
|
/* Ensure the update is setup. */
|
||||||
|
R_UNLESS(this->setup_update, ns::ResultCardUpdateNotSetup());
|
||||||
|
|
||||||
|
out.SetValue(this->update_task->GetProgress().state == ncm::InstallProgressState::Downloaded);
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::ApplyPreparedUpdate() {
|
||||||
|
/* Ensure the update is setup. */
|
||||||
|
R_UNLESS(this->setup_update, ns::ResultCardUpdateNotSetup());
|
||||||
|
|
||||||
|
/* Ensure the update is prepared. */
|
||||||
|
R_UNLESS(this->update_task->GetProgress().state == ncm::InstallProgressState::Downloaded, ns::ResultCardUpdateNotPrepared());
|
||||||
|
|
||||||
|
/* Apply the task. */
|
||||||
|
R_TRY(this->apply_manager.ApplyPackageTask(std::addressof(*this->update_task)));
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::SetupUpdateImpl(os::ManagedHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id) {
|
||||||
|
/* Ensure we don't already have an update set up. */
|
||||||
|
R_UNLESS(!this->setup_update, ns::ResultCardUpdateAlreadySetup());
|
||||||
|
|
||||||
|
/* Destroy any existing update tasks. */
|
||||||
|
nim::SystemUpdateTaskId id;
|
||||||
|
auto count = nim::ListSystemUpdateTask(std::addressof(id), 1);
|
||||||
|
if (count > 0) {
|
||||||
|
R_TRY(nim::DestroySystemUpdateTask(id));
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Initialize the update task. */
|
||||||
|
R_TRY(InitializeUpdateTask(transfer_memory, transfer_memory_size, path, exfat, firmware_variation_id));
|
||||||
|
|
||||||
|
/* The update is now set up. */
|
||||||
|
this->setup_update = true;
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
|
Result SystemUpdateService::InitializeUpdateTask(os::ManagedHandle &transfer_memory_handle, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id) {
|
||||||
|
/* Map the transfer memory. */
|
||||||
|
const size_t tmem_buffer_size = static_cast<size_t>(transfer_memory_size);
|
||||||
|
this->update_transfer_memory.emplace(tmem_buffer_size, transfer_memory_handle.Get(), true);
|
||||||
|
|
||||||
|
void *tmem_buffer;
|
||||||
|
R_TRY(this->update_transfer_memory->Map(std::addressof(tmem_buffer), os::MemoryPermission_None));
|
||||||
|
auto tmem_guard = SCOPE_GUARD {
|
||||||
|
this->update_transfer_memory->Unmap();
|
||||||
|
this->update_transfer_memory = std::nullopt;
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Now that the memory is mapped, the input handle is managed and can be released. */
|
||||||
|
transfer_memory_handle.Detach();
|
||||||
|
|
||||||
|
/* Adjust the package root. */
|
||||||
|
ncm::Path package_root;
|
||||||
|
R_TRY(FormatUserPackagePath(std::addressof(package_root), path));
|
||||||
|
|
||||||
|
/* Ensure that we can create an update context. */
|
||||||
|
R_TRY(fs::EnsureDirectoryRecursively("@Sdcard:/atmosphere/update/"));
|
||||||
|
const char *context_path = "@Sdcard:/atmosphere/update/cup.ctx";
|
||||||
|
|
||||||
|
/* Create and initialize the update task. */
|
||||||
|
this->update_task.emplace();
|
||||||
|
R_TRY(this->update_task->Initialize(package_root.str, context_path, tmem_buffer, tmem_buffer_size, exfat, firmware_variation_id));
|
||||||
|
|
||||||
|
/* We successfully setup the update. */
|
||||||
|
tmem_guard.Cancel();
|
||||||
|
|
||||||
|
return ResultSuccess();
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -15,6 +15,8 @@
|
||||||
*/
|
*/
|
||||||
#pragma once
|
#pragma once
|
||||||
#include <stratosphere.hpp>
|
#include <stratosphere.hpp>
|
||||||
|
#include "sysupdater_i_async_result.hpp"
|
||||||
|
#include "sysupdater_apply_manager.hpp"
|
||||||
|
|
||||||
namespace ams::mitm::sysupdater {
|
namespace ams::mitm::sysupdater {
|
||||||
|
|
||||||
|
@ -32,19 +34,53 @@ namespace ams::mitm::sysupdater {
|
||||||
ncm::ContentId invalid_content_id;
|
ncm::ContentId invalid_content_id;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
struct SystemUpdateProgress {
|
||||||
|
s64 current_size;
|
||||||
|
s64 total_size;
|
||||||
|
};
|
||||||
|
|
||||||
class SystemUpdateService final : public sf::IServiceObject {
|
class SystemUpdateService final : public sf::IServiceObject {
|
||||||
private:
|
private:
|
||||||
enum class CommandId {
|
enum class CommandId {
|
||||||
GetUpdateInformation = 0,
|
GetUpdateInformation = 0,
|
||||||
ValidateUpdate = 1,
|
ValidateUpdate = 1,
|
||||||
|
SetupUpdate = 2,
|
||||||
|
SetupUpdateWithVariation = 3,
|
||||||
|
RequestPrepareUpdate = 4,
|
||||||
|
GetPrepareUpdateProgress = 5,
|
||||||
|
HasPreparedUpdate = 6,
|
||||||
|
ApplyPreparedUpdate = 7,
|
||||||
};
|
};
|
||||||
|
private:
|
||||||
|
SystemUpdateApplyManager apply_manager;
|
||||||
|
std::optional<ncm::PackageSystemDowngradeTask> update_task;
|
||||||
|
std::optional<os::TransferMemory> update_transfer_memory;
|
||||||
|
bool setup_update;
|
||||||
|
bool requested_update;
|
||||||
|
public:
|
||||||
|
constexpr SystemUpdateService() : apply_manager(), update_task(), update_transfer_memory(), setup_update(false), requested_update(false) { /* ... */ }
|
||||||
|
private:
|
||||||
|
Result SetupUpdateImpl(os::ManagedHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id);
|
||||||
|
Result InitializeUpdateTask(os::ManagedHandle &transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id);
|
||||||
private:
|
private:
|
||||||
Result GetUpdateInformation(sf::Out<UpdateInformation> out, const ncm::Path &path);
|
Result GetUpdateInformation(sf::Out<UpdateInformation> out, const ncm::Path &path);
|
||||||
Result ValidateUpdate(sf::Out<Result> out_validate_result, sf::Out<UpdateValidationInfo> out_validate_info, const ncm::Path &path);
|
Result ValidateUpdate(sf::Out<Result> out_validate_result, sf::Out<UpdateValidationInfo> out_validate_info, const ncm::Path &path);
|
||||||
|
Result SetupUpdate(sf::MoveHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat);
|
||||||
|
Result SetupUpdateWithVariation(sf::MoveHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat, ncm::FirmwareVariationId firmware_variation_id);
|
||||||
|
Result RequestPrepareUpdate(sf::OutCopyHandle out_event_handle, sf::Out<std::shared_ptr<IAsyncResult>> out_async);
|
||||||
|
Result GetPrepareUpdateProgress(sf::Out<SystemUpdateProgress> out);
|
||||||
|
Result HasPreparedUpdate(sf::Out<bool> out);
|
||||||
|
Result ApplyPreparedUpdate();
|
||||||
public:
|
public:
|
||||||
DEFINE_SERVICE_DISPATCH_TABLE {
|
DEFINE_SERVICE_DISPATCH_TABLE {
|
||||||
MAKE_SERVICE_COMMAND_META(GetUpdateInformation),
|
MAKE_SERVICE_COMMAND_META(GetUpdateInformation),
|
||||||
MAKE_SERVICE_COMMAND_META(ValidateUpdate),
|
MAKE_SERVICE_COMMAND_META(ValidateUpdate),
|
||||||
|
MAKE_SERVICE_COMMAND_META(SetupUpdate),
|
||||||
|
MAKE_SERVICE_COMMAND_META(SetupUpdateWithVariation),
|
||||||
|
MAKE_SERVICE_COMMAND_META(RequestPrepareUpdate),
|
||||||
|
MAKE_SERVICE_COMMAND_META(GetPrepareUpdateProgress),
|
||||||
|
MAKE_SERVICE_COMMAND_META(HasPreparedUpdate),
|
||||||
|
MAKE_SERVICE_COMMAND_META(ApplyPreparedUpdate),
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue