mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-06-02 09:16:11 +00:00
88 lines
4 KiB
C++
88 lines
4 KiB
C++
/*
|
|
* 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>
|
|
#include "sysupdater_i_async_result.hpp"
|
|
#include "sysupdater_apply_manager.hpp"
|
|
|
|
namespace ams::mitm::sysupdater {
|
|
|
|
constexpr inline size_t FirmwareVariationCountMax = 16;
|
|
|
|
struct UpdateInformation {
|
|
u32 version;
|
|
bool exfat_supported;
|
|
u32 firmware_variation_count;
|
|
ncm::FirmwareVariationId firmware_variation_ids[FirmwareVariationCountMax];
|
|
};
|
|
|
|
struct UpdateValidationInfo {
|
|
ncm::ContentMetaKey invalid_key;
|
|
ncm::ContentId invalid_content_id;
|
|
};
|
|
|
|
struct SystemUpdateProgress {
|
|
s64 current_size;
|
|
s64 total_size;
|
|
};
|
|
|
|
class SystemUpdateService final : public sf::IServiceObject {
|
|
private:
|
|
enum class CommandId {
|
|
GetUpdateInformation = 0,
|
|
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:
|
|
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 SetupUpdate(sf::CopyHandle transfer_memory, u64 transfer_memory_size, const ncm::Path &path, bool exfat);
|
|
Result SetupUpdateWithVariation(sf::CopyHandle 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:
|
|
DEFINE_SERVICE_DISPATCH_TABLE {
|
|
MAKE_SERVICE_COMMAND_META(GetUpdateInformation),
|
|
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),
|
|
};
|
|
};
|
|
|
|
}
|