frontend_common: Add content manager utility functions
Creates utility functions to remove/install DLC, updates, and base game content
This commit is contained in:
		| @@ -21,6 +21,7 @@ import org.yuzu.yuzu_emu.utils.DocumentsTree | ||||
| import org.yuzu.yuzu_emu.utils.FileUtil | ||||
| import org.yuzu.yuzu_emu.utils.Log | ||||
| import org.yuzu.yuzu_emu.utils.SerializableHelper.serializable | ||||
| import org.yuzu.yuzu_emu.model.InstallResult | ||||
|  | ||||
| /** | ||||
|  * Class which contains methods that interact | ||||
| @@ -235,9 +236,12 @@ object NativeLibrary { | ||||
|     /** | ||||
|      * Installs a nsp or xci file to nand | ||||
|      * @param filename String representation of file uri | ||||
|      * @param extension Lowercase string representation of file extension without "." | ||||
|      * @return int representation of [InstallResult] | ||||
|      */ | ||||
|     external fun installFileToNand(filename: String, extension: String): Int | ||||
|     external fun installFileToNand( | ||||
|         filename: String, | ||||
|         callback: (max: Long, progress: Long) -> Boolean | ||||
|     ): Int | ||||
|  | ||||
|     external fun doesUpdateMatchProgram(programId: String, updatePath: String): Boolean | ||||
|  | ||||
| @@ -609,15 +613,4 @@ object NativeLibrary { | ||||
|         const val RELEASED = 0 | ||||
|         const val PRESSED = 1 | ||||
|     } | ||||
|  | ||||
|     /** | ||||
|      * Result from installFileToNand | ||||
|      */ | ||||
|     object InstallFileToNandResult { | ||||
|         const val Success = 0 | ||||
|         const val SuccessFileOverwritten = 1 | ||||
|         const val Error = 2 | ||||
|         const val ErrorBaseGame = 3 | ||||
|         const val ErrorFilenameExtension = 4 | ||||
|     } | ||||
| } | ||||
|   | ||||
| @@ -0,0 +1,15 @@ | ||||
| // SPDX-FileCopyrightText: 2024 yuzu Emulator Project | ||||
| // SPDX-License-Identifier: GPL-2.0-or-later | ||||
|  | ||||
| package org.yuzu.yuzu_emu.model | ||||
|  | ||||
| enum class InstallResult(val int: Int) { | ||||
|     Success(0), | ||||
|     Overwrite(1), | ||||
|     Failure(2), | ||||
|     BaseInstallAttempted(3); | ||||
|  | ||||
|     companion object { | ||||
|         fun from(int: Int): InstallResult = entries.firstOrNull { it.int == int } ?: Success | ||||
|     } | ||||
| } | ||||
| @@ -42,3 +42,19 @@ double GetJDouble(JNIEnv* env, jobject jdouble) { | ||||
| jobject ToJDouble(JNIEnv* env, double value) { | ||||
|     return env->NewObject(IDCache::GetDoubleClass(), IDCache::GetDoubleConstructor(), value); | ||||
| } | ||||
|  | ||||
| s32 GetJInteger(JNIEnv* env, jobject jinteger) { | ||||
|     return env->GetIntField(jinteger, IDCache::GetIntegerValueField()); | ||||
| } | ||||
|  | ||||
| jobject ToJInteger(JNIEnv* env, s32 value) { | ||||
|     return env->NewObject(IDCache::GetIntegerClass(), IDCache::GetIntegerConstructor(), value); | ||||
| } | ||||
|  | ||||
| bool GetJBoolean(JNIEnv* env, jobject jboolean) { | ||||
|     return env->GetBooleanField(jboolean, IDCache::GetBooleanValueField()); | ||||
| } | ||||
|  | ||||
| jobject ToJBoolean(JNIEnv* env, bool value) { | ||||
|     return env->NewObject(IDCache::GetBooleanClass(), IDCache::GetBooleanConstructor(), value); | ||||
| } | ||||
|   | ||||
| @@ -6,6 +6,7 @@ | ||||
| #include <string> | ||||
|  | ||||
| #include <jni.h> | ||||
| #include "common/common_types.h" | ||||
|  | ||||
| std::string GetJString(JNIEnv* env, jstring jstr); | ||||
| jstring ToJString(JNIEnv* env, std::string_view str); | ||||
| @@ -13,3 +14,9 @@ jstring ToJString(JNIEnv* env, std::u16string_view str); | ||||
|  | ||||
| double GetJDouble(JNIEnv* env, jobject jdouble); | ||||
| jobject ToJDouble(JNIEnv* env, double value); | ||||
|  | ||||
| s32 GetJInteger(JNIEnv* env, jobject jinteger); | ||||
| jobject ToJInteger(JNIEnv* env, s32 value); | ||||
|  | ||||
| bool GetJBoolean(JNIEnv* env, jobject jboolean); | ||||
| jobject ToJBoolean(JNIEnv* env, bool value); | ||||
|   | ||||
| @@ -47,6 +47,14 @@ static jclass s_double_class; | ||||
| static jmethodID s_double_constructor; | ||||
| static jfieldID s_double_value_field; | ||||
|  | ||||
| static jclass s_integer_class; | ||||
| static jmethodID s_integer_constructor; | ||||
| static jfieldID s_integer_value_field; | ||||
|  | ||||
| static jclass s_boolean_class; | ||||
| static jmethodID s_boolean_constructor; | ||||
| static jfieldID s_boolean_value_field; | ||||
|  | ||||
| static constexpr jint JNI_VERSION = JNI_VERSION_1_6; | ||||
|  | ||||
| namespace IDCache { | ||||
| @@ -198,6 +206,30 @@ jfieldID GetDoubleValueField() { | ||||
|     return s_double_value_field; | ||||
| } | ||||
|  | ||||
| jclass GetIntegerClass() { | ||||
|     return s_integer_class; | ||||
| } | ||||
|  | ||||
| jmethodID GetIntegerConstructor() { | ||||
|     return s_integer_constructor; | ||||
| } | ||||
|  | ||||
| jfieldID GetIntegerValueField() { | ||||
|     return s_integer_value_field; | ||||
| } | ||||
|  | ||||
| jclass GetBooleanClass() { | ||||
|     return s_boolean_class; | ||||
| } | ||||
|  | ||||
| jmethodID GetBooleanConstructor() { | ||||
|     return s_boolean_constructor; | ||||
| } | ||||
|  | ||||
| jfieldID GetBooleanValueField() { | ||||
|     return s_boolean_value_field; | ||||
| } | ||||
|  | ||||
| } // namespace IDCache | ||||
|  | ||||
| #ifdef __cplusplus | ||||
| @@ -284,6 +316,18 @@ jint JNI_OnLoad(JavaVM* vm, void* reserved) { | ||||
|     s_double_value_field = env->GetFieldID(double_class, "value", "D"); | ||||
|     env->DeleteLocalRef(double_class); | ||||
|  | ||||
|     const jclass int_class = env->FindClass("java/lang/Integer"); | ||||
|     s_integer_class = reinterpret_cast<jclass>(env->NewGlobalRef(int_class)); | ||||
|     s_integer_constructor = env->GetMethodID(int_class, "<init>", "(I)V"); | ||||
|     s_integer_value_field = env->GetFieldID(int_class, "value", "I"); | ||||
|     env->DeleteLocalRef(int_class); | ||||
|  | ||||
|     const jclass boolean_class = env->FindClass("java/lang/Boolean"); | ||||
|     s_boolean_class = reinterpret_cast<jclass>(env->NewGlobalRef(boolean_class)); | ||||
|     s_boolean_constructor = env->GetMethodID(boolean_class, "<init>", "(Z)V"); | ||||
|     s_boolean_value_field = env->GetFieldID(boolean_class, "value", "Z"); | ||||
|     env->DeleteLocalRef(boolean_class); | ||||
|  | ||||
|     // Initialize Android Storage | ||||
|     Common::FS::Android::RegisterCallbacks(env, s_native_library_class); | ||||
|  | ||||
| @@ -310,6 +354,8 @@ void JNI_OnUnload(JavaVM* vm, void* reserved) { | ||||
|     env->DeleteGlobalRef(s_pair_class); | ||||
|     env->DeleteGlobalRef(s_overlay_control_data_class); | ||||
|     env->DeleteGlobalRef(s_double_class); | ||||
|     env->DeleteGlobalRef(s_integer_class); | ||||
|     env->DeleteGlobalRef(s_boolean_class); | ||||
|  | ||||
|     // UnInitialize applets | ||||
|     SoftwareKeyboard::CleanupJNI(env); | ||||
|   | ||||
| @@ -47,4 +47,12 @@ jclass GetDoubleClass(); | ||||
| jmethodID GetDoubleConstructor(); | ||||
| jfieldID GetDoubleValueField(); | ||||
|  | ||||
| jclass GetIntegerClass(); | ||||
| jmethodID GetIntegerConstructor(); | ||||
| jfieldID GetIntegerValueField(); | ||||
|  | ||||
| jclass GetBooleanClass(); | ||||
| jmethodID GetBooleanConstructor(); | ||||
| jfieldID GetBooleanValueField(); | ||||
|  | ||||
| } // namespace IDCache | ||||
|   | ||||
| @@ -17,6 +17,7 @@ | ||||
| #include <core/file_sys/patch_manager.h> | ||||
| #include <core/file_sys/savedata_factory.h> | ||||
| #include <core/loader/nro.h> | ||||
| #include <frontend_common/content_manager.h> | ||||
| #include <jni.h> | ||||
|  | ||||
| #include "common/detached_tasks.h" | ||||
| @@ -100,67 +101,6 @@ void EmulationSession::SetNativeWindow(ANativeWindow* native_window) { | ||||
|     m_native_window = native_window; | ||||
| } | ||||
|  | ||||
| int EmulationSession::InstallFileToNand(std::string filename, std::string file_extension) { | ||||
|     jconst copy_func = [](const FileSys::VirtualFile& src, const FileSys::VirtualFile& dest, | ||||
|                           std::size_t block_size) { | ||||
|         if (src == nullptr || dest == nullptr) { | ||||
|             return false; | ||||
|         } | ||||
|         if (!dest->Resize(src->GetSize())) { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         using namespace Common::Literals; | ||||
|         [[maybe_unused]] std::vector<u8> buffer(1_MiB); | ||||
|  | ||||
|         for (std::size_t i = 0; i < src->GetSize(); i += buffer.size()) { | ||||
|             jconst read = src->Read(buffer.data(), buffer.size(), i); | ||||
|             dest->Write(buffer.data(), read, i); | ||||
|         } | ||||
|         return true; | ||||
|     }; | ||||
|  | ||||
|     enum InstallResult { | ||||
|         Success = 0, | ||||
|         SuccessFileOverwritten = 1, | ||||
|         InstallError = 2, | ||||
|         ErrorBaseGame = 3, | ||||
|         ErrorFilenameExtension = 4, | ||||
|     }; | ||||
|  | ||||
|     [[maybe_unused]] std::shared_ptr<FileSys::NSP> nsp; | ||||
|     if (file_extension == "nsp") { | ||||
|         nsp = std::make_shared<FileSys::NSP>(m_vfs->OpenFile(filename, FileSys::Mode::Read)); | ||||
|         if (nsp->IsExtractedType()) { | ||||
|             return InstallError; | ||||
|         } | ||||
|     } else { | ||||
|         return ErrorFilenameExtension; | ||||
|     } | ||||
|  | ||||
|     if (!nsp) { | ||||
|         return InstallError; | ||||
|     } | ||||
|  | ||||
|     if (nsp->GetStatus() != Loader::ResultStatus::Success) { | ||||
|         return InstallError; | ||||
|     } | ||||
|  | ||||
|     jconst res = m_system.GetFileSystemController().GetUserNANDContents()->InstallEntry(*nsp, true, | ||||
|                                                                                         copy_func); | ||||
|  | ||||
|     switch (res) { | ||||
|     case FileSys::InstallResult::Success: | ||||
|         return Success; | ||||
|     case FileSys::InstallResult::OverwriteExisting: | ||||
|         return SuccessFileOverwritten; | ||||
|     case FileSys::InstallResult::ErrorBaseInstall: | ||||
|         return ErrorBaseGame; | ||||
|     default: | ||||
|         return InstallError; | ||||
|     } | ||||
| } | ||||
|  | ||||
| void EmulationSession::InitializeGpuDriver(const std::string& hook_lib_dir, | ||||
|                                            const std::string& custom_driver_dir, | ||||
|                                            const std::string& custom_driver_name, | ||||
| @@ -512,10 +452,20 @@ void Java_org_yuzu_yuzu_1emu_NativeLibrary_setAppDirectory(JNIEnv* env, jobject | ||||
| } | ||||
|  | ||||
| int Java_org_yuzu_yuzu_1emu_NativeLibrary_installFileToNand(JNIEnv* env, jobject instance, | ||||
|                                                             jstring j_file, | ||||
|                                                             jstring j_file_extension) { | ||||
|     return EmulationSession::GetInstance().InstallFileToNand(GetJString(env, j_file), | ||||
|                                                              GetJString(env, j_file_extension)); | ||||
|                                                             jstring j_file, jobject jcallback) { | ||||
|     auto jlambdaClass = env->GetObjectClass(jcallback); | ||||
|     auto jlambdaInvokeMethod = env->GetMethodID( | ||||
|         jlambdaClass, "invoke", "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;"); | ||||
|     const auto callback = [env, jcallback, jlambdaInvokeMethod](size_t max, size_t progress) { | ||||
|         auto jwasCancelled = env->CallObjectMethod(jcallback, jlambdaInvokeMethod, | ||||
|                                                    ToJDouble(env, max), ToJDouble(env, progress)); | ||||
|         return GetJBoolean(env, jwasCancelled); | ||||
|     }; | ||||
|  | ||||
|     return static_cast<int>( | ||||
|         ContentManager::InstallNSP(&EmulationSession::GetInstance().System(), | ||||
|                                    EmulationSession::GetInstance().System().GetFilesystem().get(), | ||||
|                                    GetJString(env, j_file), callback)); | ||||
| } | ||||
|  | ||||
| jboolean Java_org_yuzu_yuzu_1emu_NativeLibrary_doesUpdateMatchProgram(JNIEnv* env, jobject jobj, | ||||
|   | ||||
| @@ -7,6 +7,7 @@ | ||||
| #include "core/file_sys/registered_cache.h" | ||||
| #include "core/hle/service/acc/profile_manager.h" | ||||
| #include "core/perf_stats.h" | ||||
| #include "frontend_common/content_manager.h" | ||||
| #include "jni/applets/software_keyboard.h" | ||||
| #include "jni/emu_window/emu_window.h" | ||||
| #include "video_core/rasterizer_interface.h" | ||||
| @@ -29,7 +30,6 @@ public: | ||||
|     void SetNativeWindow(ANativeWindow* native_window); | ||||
|     void SurfaceChanged(); | ||||
|  | ||||
|     int InstallFileToNand(std::string filename, std::string file_extension); | ||||
|     void InitializeGpuDriver(const std::string& hook_lib_dir, const std::string& custom_driver_dir, | ||||
|                              const std::string& custom_driver_name, | ||||
|                              const std::string& file_redirect_dir); | ||||
|   | ||||
| @@ -4,6 +4,7 @@ | ||||
| add_library(frontend_common STATIC | ||||
|     config.cpp | ||||
|     config.h | ||||
|     content_manager.h | ||||
| ) | ||||
|  | ||||
| create_target_directory_groups(frontend_common) | ||||
|   | ||||
							
								
								
									
										168
									
								
								src/frontend_common/content_manager.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										168
									
								
								src/frontend_common/content_manager.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,168 @@ | ||||
| // SPDX-FileCopyrightText: 2024 yuzu Emulator Project | ||||
| // SPDX-License-Identifier: GPL-2.0-or-later | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <boost/algorithm/string.hpp> | ||||
| #include "common/common_types.h" | ||||
| #include "common/literals.h" | ||||
| #include "core/core.h" | ||||
| #include "core/file_sys/common_funcs.h" | ||||
| #include "core/file_sys/content_archive.h" | ||||
| #include "core/file_sys/mode.h" | ||||
| #include "core/file_sys/nca_metadata.h" | ||||
| #include "core/file_sys/registered_cache.h" | ||||
| #include "core/file_sys/submission_package.h" | ||||
| #include "core/hle/service/filesystem/filesystem.h" | ||||
| #include "core/loader/loader.h" | ||||
|  | ||||
| namespace ContentManager { | ||||
|  | ||||
| enum class InstallResult { | ||||
|     Success, | ||||
|     Overwrite, | ||||
|     Failure, | ||||
|     BaseInstallAttempted, | ||||
| }; | ||||
|  | ||||
| inline bool RemoveDLC(const Service::FileSystem::FileSystemController& fs_controller, | ||||
|                       const u64 title_id) { | ||||
|     return fs_controller.GetUserNANDContents()->RemoveExistingEntry(title_id) || | ||||
|            fs_controller.GetSDMCContents()->RemoveExistingEntry(title_id); | ||||
| } | ||||
|  | ||||
| inline size_t RemoveAllDLC(Core::System* system, const u64 program_id) { | ||||
|     size_t count{}; | ||||
|     const auto& fs_controller = system->GetFileSystemController(); | ||||
|     const auto dlc_entries = system->GetContentProvider().ListEntriesFilter( | ||||
|         FileSys::TitleType::AOC, FileSys::ContentRecordType::Data); | ||||
|     std::vector<u64> program_dlc_entries; | ||||
|  | ||||
|     for (const auto& entry : dlc_entries) { | ||||
|         if (FileSys::GetBaseTitleID(entry.title_id) == program_id) { | ||||
|             program_dlc_entries.push_back(entry.title_id); | ||||
|         } | ||||
|     } | ||||
|  | ||||
|     for (const auto& entry : program_dlc_entries) { | ||||
|         if (RemoveDLC(fs_controller, entry)) { | ||||
|             ++count; | ||||
|         } | ||||
|     } | ||||
|     return count; | ||||
| } | ||||
|  | ||||
| inline bool RemoveUpdate(const Service::FileSystem::FileSystemController& fs_controller, | ||||
|                          const u64 program_id) { | ||||
|     const auto update_id = program_id | 0x800; | ||||
|     return fs_controller.GetUserNANDContents()->RemoveExistingEntry(update_id) || | ||||
|            fs_controller.GetSDMCContents()->RemoveExistingEntry(update_id); | ||||
| } | ||||
|  | ||||
| inline bool RemoveBaseContent(const Service::FileSystem::FileSystemController& fs_controller, | ||||
|                               const u64 program_id) { | ||||
|     return fs_controller.GetUserNANDContents()->RemoveExistingEntry(program_id) || | ||||
|            fs_controller.GetSDMCContents()->RemoveExistingEntry(program_id); | ||||
| } | ||||
|  | ||||
| inline InstallResult InstallNSP( | ||||
|     Core::System* system, FileSys::VfsFilesystem* vfs, const std::string& filename, | ||||
|     const std::function<bool(size_t, size_t)>& callback = std::function<bool(size_t, size_t)>()) { | ||||
|     const auto copy = [callback](const FileSys::VirtualFile& src, const FileSys::VirtualFile& dest, | ||||
|                                  std::size_t block_size) { | ||||
|         if (src == nullptr || dest == nullptr) { | ||||
|             return false; | ||||
|         } | ||||
|         if (!dest->Resize(src->GetSize())) { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         using namespace Common::Literals; | ||||
|         std::vector<u8> buffer(1_MiB); | ||||
|  | ||||
|         for (std::size_t i = 0; i < src->GetSize(); i += buffer.size()) { | ||||
|             if (callback(src->GetSize(), i)) { | ||||
|                 dest->Resize(0); | ||||
|                 return false; | ||||
|             } | ||||
|             const auto read = src->Read(buffer.data(), buffer.size(), i); | ||||
|             dest->Write(buffer.data(), read, i); | ||||
|         } | ||||
|         return true; | ||||
|     }; | ||||
|  | ||||
|     std::shared_ptr<FileSys::NSP> nsp; | ||||
|     FileSys::VirtualFile file = vfs->OpenFile(filename, FileSys::Mode::Read); | ||||
|     if (boost::to_lower_copy(file->GetName()).ends_with(std::string("nsp"))) { | ||||
|         nsp = std::make_shared<FileSys::NSP>(file); | ||||
|         if (nsp->IsExtractedType()) { | ||||
|             return InstallResult::Failure; | ||||
|         } | ||||
|     } else { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|  | ||||
|     if (nsp->GetStatus() != Loader::ResultStatus::Success) { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|     const auto res = | ||||
|         system->GetFileSystemController().GetUserNANDContents()->InstallEntry(*nsp, true, copy); | ||||
|     switch (res) { | ||||
|     case FileSys::InstallResult::Success: | ||||
|         return InstallResult::Success; | ||||
|     case FileSys::InstallResult::OverwriteExisting: | ||||
|         return InstallResult::Overwrite; | ||||
|     case FileSys::InstallResult::ErrorBaseInstall: | ||||
|         return InstallResult::BaseInstallAttempted; | ||||
|     default: | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
| } | ||||
|  | ||||
| inline InstallResult InstallNCA( | ||||
|     FileSys::VfsFilesystem* vfs, const std::string& filename, | ||||
|     FileSys::RegisteredCache* registered_cache, const FileSys::TitleType title_type, | ||||
|     const std::function<bool(size_t, size_t)>& callback = std::function<bool(size_t, size_t)>()) { | ||||
|     const auto copy = [callback](const FileSys::VirtualFile& src, const FileSys::VirtualFile& dest, | ||||
|                                  std::size_t block_size) { | ||||
|         if (src == nullptr || dest == nullptr) { | ||||
|             return false; | ||||
|         } | ||||
|         if (!dest->Resize(src->GetSize())) { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         using namespace Common::Literals; | ||||
|         std::vector<u8> buffer(1_MiB); | ||||
|  | ||||
|         for (std::size_t i = 0; i < src->GetSize(); i += buffer.size()) { | ||||
|             if (callback(src->GetSize(), i)) { | ||||
|                 dest->Resize(0); | ||||
|                 return false; | ||||
|             } | ||||
|             const auto read = src->Read(buffer.data(), buffer.size(), i); | ||||
|             dest->Write(buffer.data(), read, i); | ||||
|         } | ||||
|         return true; | ||||
|     }; | ||||
|  | ||||
|     const auto nca = std::make_shared<FileSys::NCA>(vfs->OpenFile(filename, FileSys::Mode::Read)); | ||||
|     const auto id = nca->GetStatus(); | ||||
|  | ||||
|     // Game updates necessary are missing base RomFS | ||||
|     if (id != Loader::ResultStatus::Success && | ||||
|         id != Loader::ResultStatus::ErrorMissingBKTRBaseRomFS) { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|  | ||||
|     const auto res = registered_cache->InstallEntry(*nca, title_type, true, copy); | ||||
|     if (res == FileSys::InstallResult::Success) { | ||||
|         return InstallResult::Success; | ||||
|     } else if (res == FileSys::InstallResult::OverwriteExisting) { | ||||
|         return InstallResult::Overwrite; | ||||
|     } else { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
| } | ||||
|  | ||||
| } // namespace ContentManager | ||||
| @@ -47,6 +47,7 @@ | ||||
| #include "core/hle/service/am/applet_oe.h" | ||||
| #include "core/hle/service/am/applets/applets.h" | ||||
| #include "core/hle/service/set/system_settings_server.h" | ||||
| #include "frontend_common/content_manager.h" | ||||
| #include "hid_core/frontend/emulated_controller.h" | ||||
| #include "hid_core/hid_core.h" | ||||
| #include "yuzu/multiplayer/state.h" | ||||
| @@ -2476,10 +2477,8 @@ void GMainWindow::OnGameListRemoveInstalledEntry(u64 program_id, InstalledEntryT | ||||
| } | ||||
|  | ||||
| void GMainWindow::RemoveBaseContent(u64 program_id, InstalledEntryType type) { | ||||
|     const auto& fs_controller = system->GetFileSystemController(); | ||||
|     const auto res = fs_controller.GetUserNANDContents()->RemoveExistingEntry(program_id) || | ||||
|                      fs_controller.GetSDMCContents()->RemoveExistingEntry(program_id); | ||||
|  | ||||
|     const auto res = | ||||
|         ContentManager::RemoveBaseContent(system->GetFileSystemController(), program_id); | ||||
|     if (res) { | ||||
|         QMessageBox::information(this, tr("Successfully Removed"), | ||||
|                                  tr("Successfully removed the installed base game.")); | ||||
| @@ -2491,11 +2490,7 @@ void GMainWindow::RemoveBaseContent(u64 program_id, InstalledEntryType type) { | ||||
| } | ||||
|  | ||||
| void GMainWindow::RemoveUpdateContent(u64 program_id, InstalledEntryType type) { | ||||
|     const auto update_id = program_id | 0x800; | ||||
|     const auto& fs_controller = system->GetFileSystemController(); | ||||
|     const auto res = fs_controller.GetUserNANDContents()->RemoveExistingEntry(update_id) || | ||||
|                      fs_controller.GetSDMCContents()->RemoveExistingEntry(update_id); | ||||
|  | ||||
|     const auto res = ContentManager::RemoveUpdate(system->GetFileSystemController(), program_id); | ||||
|     if (res) { | ||||
|         QMessageBox::information(this, tr("Successfully Removed"), | ||||
|                                  tr("Successfully removed the installed update.")); | ||||
| @@ -2506,22 +2501,7 @@ void GMainWindow::RemoveUpdateContent(u64 program_id, InstalledEntryType type) { | ||||
| } | ||||
|  | ||||
| void GMainWindow::RemoveAddOnContent(u64 program_id, InstalledEntryType type) { | ||||
|     u32 count{}; | ||||
|     const auto& fs_controller = system->GetFileSystemController(); | ||||
|     const auto dlc_entries = system->GetContentProvider().ListEntriesFilter( | ||||
|         FileSys::TitleType::AOC, FileSys::ContentRecordType::Data); | ||||
|  | ||||
|     for (const auto& entry : dlc_entries) { | ||||
|         if (FileSys::GetBaseTitleID(entry.title_id) == program_id) { | ||||
|             const auto res = | ||||
|                 fs_controller.GetUserNANDContents()->RemoveExistingEntry(entry.title_id) || | ||||
|                 fs_controller.GetSDMCContents()->RemoveExistingEntry(entry.title_id); | ||||
|             if (res) { | ||||
|                 ++count; | ||||
|             } | ||||
|         } | ||||
|     } | ||||
|  | ||||
|     const size_t count = ContentManager::RemoveAllDLC(system.get(), program_id); | ||||
|     if (count == 0) { | ||||
|         QMessageBox::warning(this, GetGameListErrorRemoving(type), | ||||
|                              tr("There are no DLC installed for this title.")); | ||||
| @@ -3290,12 +3270,21 @@ void GMainWindow::OnMenuInstallToNAND() { | ||||
|         install_progress->setLabelText( | ||||
|             tr("Installing file \"%1\"...").arg(QFileInfo(file).fileName())); | ||||
|  | ||||
|         QFuture<InstallResult> future; | ||||
|         InstallResult result; | ||||
|         QFuture<ContentManager::InstallResult> future; | ||||
|         ContentManager::InstallResult result; | ||||
|  | ||||
|         if (file.endsWith(QStringLiteral("nsp"), Qt::CaseInsensitive)) { | ||||
|  | ||||
|             future = QtConcurrent::run([this, &file] { return InstallNSP(file); }); | ||||
|             const auto progress_callback = [this](size_t size, size_t progress) { | ||||
|                 emit UpdateInstallProgress(); | ||||
|                 if (install_progress->wasCanceled()) { | ||||
|                     return true; | ||||
|                 } | ||||
|                 return false; | ||||
|             }; | ||||
|             future = QtConcurrent::run([this, &file, progress_callback] { | ||||
|                 return ContentManager::InstallNSP(system.get(), vfs.get(), file.toStdString(), | ||||
|                                                   progress_callback); | ||||
|             }); | ||||
|  | ||||
|             while (!future.isFinished()) { | ||||
|                 QCoreApplication::processEvents(); | ||||
| @@ -3311,16 +3300,16 @@ void GMainWindow::OnMenuInstallToNAND() { | ||||
|         std::this_thread::sleep_for(std::chrono::milliseconds(10)); | ||||
|  | ||||
|         switch (result) { | ||||
|         case InstallResult::Success: | ||||
|         case ContentManager::InstallResult::Success: | ||||
|             new_files.append(QFileInfo(file).fileName()); | ||||
|             break; | ||||
|         case InstallResult::Overwrite: | ||||
|         case ContentManager::InstallResult::Overwrite: | ||||
|             overwritten_files.append(QFileInfo(file).fileName()); | ||||
|             break; | ||||
|         case InstallResult::Failure: | ||||
|         case ContentManager::InstallResult::Failure: | ||||
|             failed_files.append(QFileInfo(file).fileName()); | ||||
|             break; | ||||
|         case InstallResult::BaseInstallAttempted: | ||||
|         case ContentManager::InstallResult::BaseInstallAttempted: | ||||
|             failed_files.append(QFileInfo(file).fileName()); | ||||
|             detected_base_install = true; | ||||
|             break; | ||||
| @@ -3354,96 +3343,7 @@ void GMainWindow::OnMenuInstallToNAND() { | ||||
|     ui->action_Install_File_NAND->setEnabled(true); | ||||
| } | ||||
|  | ||||
| InstallResult GMainWindow::InstallNSP(const QString& filename) { | ||||
|     const auto qt_raw_copy = [this](const FileSys::VirtualFile& src, | ||||
|                                     const FileSys::VirtualFile& dest, std::size_t block_size) { | ||||
|         if (src == nullptr || dest == nullptr) { | ||||
|             return false; | ||||
|         } | ||||
|         if (!dest->Resize(src->GetSize())) { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         std::vector<u8> buffer(CopyBufferSize); | ||||
|  | ||||
|         for (std::size_t i = 0; i < src->GetSize(); i += buffer.size()) { | ||||
|             if (install_progress->wasCanceled()) { | ||||
|                 dest->Resize(0); | ||||
|                 return false; | ||||
|             } | ||||
|  | ||||
|             emit UpdateInstallProgress(); | ||||
|  | ||||
|             const auto read = src->Read(buffer.data(), buffer.size(), i); | ||||
|             dest->Write(buffer.data(), read, i); | ||||
|         } | ||||
|         return true; | ||||
|     }; | ||||
|  | ||||
|     std::shared_ptr<FileSys::NSP> nsp; | ||||
|     if (filename.endsWith(QStringLiteral("nsp"), Qt::CaseInsensitive)) { | ||||
|         nsp = std::make_shared<FileSys::NSP>( | ||||
|             vfs->OpenFile(filename.toStdString(), FileSys::Mode::Read)); | ||||
|         if (nsp->IsExtractedType()) { | ||||
|             return InstallResult::Failure; | ||||
|         } | ||||
|     } else { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|  | ||||
|     if (nsp->GetStatus() != Loader::ResultStatus::Success) { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|     const auto res = system->GetFileSystemController().GetUserNANDContents()->InstallEntry( | ||||
|         *nsp, true, qt_raw_copy); | ||||
|     switch (res) { | ||||
|     case FileSys::InstallResult::Success: | ||||
|         return InstallResult::Success; | ||||
|     case FileSys::InstallResult::OverwriteExisting: | ||||
|         return InstallResult::Overwrite; | ||||
|     case FileSys::InstallResult::ErrorBaseInstall: | ||||
|         return InstallResult::BaseInstallAttempted; | ||||
|     default: | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
| } | ||||
|  | ||||
| InstallResult GMainWindow::InstallNCA(const QString& filename) { | ||||
|     const auto qt_raw_copy = [this](const FileSys::VirtualFile& src, | ||||
|                                     const FileSys::VirtualFile& dest, std::size_t block_size) { | ||||
|         if (src == nullptr || dest == nullptr) { | ||||
|             return false; | ||||
|         } | ||||
|         if (!dest->Resize(src->GetSize())) { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         std::vector<u8> buffer(CopyBufferSize); | ||||
|  | ||||
|         for (std::size_t i = 0; i < src->GetSize(); i += buffer.size()) { | ||||
|             if (install_progress->wasCanceled()) { | ||||
|                 dest->Resize(0); | ||||
|                 return false; | ||||
|             } | ||||
|  | ||||
|             emit UpdateInstallProgress(); | ||||
|  | ||||
|             const auto read = src->Read(buffer.data(), buffer.size(), i); | ||||
|             dest->Write(buffer.data(), read, i); | ||||
|         } | ||||
|         return true; | ||||
|     }; | ||||
|  | ||||
|     const auto nca = | ||||
|         std::make_shared<FileSys::NCA>(vfs->OpenFile(filename.toStdString(), FileSys::Mode::Read)); | ||||
|     const auto id = nca->GetStatus(); | ||||
|  | ||||
|     // Game updates necessary are missing base RomFS | ||||
|     if (id != Loader::ResultStatus::Success && | ||||
|         id != Loader::ResultStatus::ErrorMissingBKTRBaseRomFS) { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|  | ||||
| ContentManager::InstallResult GMainWindow::InstallNCA(const QString& filename) { | ||||
|     const QStringList tt_options{tr("System Application"), | ||||
|                                  tr("System Archive"), | ||||
|                                  tr("System Application Update"), | ||||
| @@ -3464,7 +3364,7 @@ InstallResult GMainWindow::InstallNCA(const QString& filename) { | ||||
|     if (!ok || index == -1) { | ||||
|         QMessageBox::warning(this, tr("Failed to Install"), | ||||
|                              tr("The title type you selected for the NCA is invalid.")); | ||||
|         return InstallResult::Failure; | ||||
|         return ContentManager::InstallResult::Failure; | ||||
|     } | ||||
|  | ||||
|     // If index is equal to or past Game, add the jump in TitleType. | ||||
| @@ -3478,15 +3378,15 @@ InstallResult GMainWindow::InstallNCA(const QString& filename) { | ||||
|     auto* registered_cache = is_application ? fs_controller.GetUserNANDContents() | ||||
|                                             : fs_controller.GetSystemNANDContents(); | ||||
|  | ||||
|     const auto res = registered_cache->InstallEntry(*nca, static_cast<FileSys::TitleType>(index), | ||||
|                                                     true, qt_raw_copy); | ||||
|     if (res == FileSys::InstallResult::Success) { | ||||
|         return InstallResult::Success; | ||||
|     } else if (res == FileSys::InstallResult::OverwriteExisting) { | ||||
|         return InstallResult::Overwrite; | ||||
|     } else { | ||||
|         return InstallResult::Failure; | ||||
|     } | ||||
|     const auto progress_callback = [this](size_t size, size_t progress) { | ||||
|         emit UpdateInstallProgress(); | ||||
|         if (install_progress->wasCanceled()) { | ||||
|             return true; | ||||
|         } | ||||
|         return false; | ||||
|     }; | ||||
|     return ContentManager::InstallNCA(vfs.get(), filename.toStdString(), registered_cache, | ||||
|                                       static_cast<FileSys::TitleType>(index), progress_callback); | ||||
| } | ||||
|  | ||||
| void GMainWindow::OnMenuRecentFile() { | ||||
|   | ||||
| @@ -16,6 +16,7 @@ | ||||
| #include "common/announce_multiplayer_room.h" | ||||
| #include "common/common_types.h" | ||||
| #include "configuration/qt_config.h" | ||||
| #include "frontend_common/content_manager.h" | ||||
| #include "input_common/drivers/tas_input.h" | ||||
| #include "yuzu/compatibility_list.h" | ||||
| #include "yuzu/hotkeys.h" | ||||
| @@ -124,13 +125,6 @@ enum class EmulatedDirectoryTarget { | ||||
|     SDMC, | ||||
| }; | ||||
|  | ||||
| enum class InstallResult { | ||||
|     Success, | ||||
|     Overwrite, | ||||
|     Failure, | ||||
|     BaseInstallAttempted, | ||||
| }; | ||||
|  | ||||
| enum class ReinitializeKeyBehavior { | ||||
|     NoWarning, | ||||
|     Warning, | ||||
| @@ -427,8 +421,7 @@ private: | ||||
|     void RemoveCacheStorage(u64 program_id); | ||||
|     bool SelectRomFSDumpTarget(const FileSys::ContentProvider&, u64 program_id, | ||||
|                                u64* selected_title_id, u8* selected_content_record_type); | ||||
|     InstallResult InstallNSP(const QString& filename); | ||||
|     InstallResult InstallNCA(const QString& filename); | ||||
|     ContentManager::InstallResult InstallNCA(const QString& filename); | ||||
|     void MigrateConfigFiles(); | ||||
|     void UpdateWindowTitle(std::string_view title_name = {}, std::string_view title_version = {}, | ||||
|                            std::string_view gpu_vendor = {}); | ||||
|   | ||||
		Reference in New Issue
	
	Block a user