From 3edafc6802a7836bf4e9c1a4751b9b45316d9855 Mon Sep 17 00:00:00 2001 From: Zach Hilman Date: Sun, 23 Sep 2018 21:06:33 -0400 Subject: [PATCH] qt: Add key derivation progress bar on initial setup --- src/yuzu/main.cpp | 52 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) diff --git a/src/yuzu/main.cpp b/src/yuzu/main.cpp index e11833c5a..48d7833b2 100644 --- a/src/yuzu/main.cpp +++ b/src/yuzu/main.cpp @@ -31,6 +31,7 @@ static FileSys::VirtualFile VfsDirectoryCreateFileWrapper(const FileSys::Virtual #include #include #include +#include #include #include #include @@ -171,6 +172,57 @@ GMainWindow::GMainWindow() .arg(Common::g_build_fullname, Common::g_scm_branch, Common::g_scm_desc)); show(); + // Gen keys if necessary + Core::Crypto::KeyManager keys{}; + if (keys.BaseDeriveNecessary()) { + Core::Crypto::PartitionDataManager pdm{vfs->OpenDirectory( + FileUtil::GetUserPath(FileUtil::UserPath::SysDataDir), FileSys::Mode::Read)}; + + const auto function = [this, &keys, &pdm]() { + keys.PopulateFromPartitionData(pdm); + Service::FileSystem::CreateFactories(vfs); + keys.DeriveETicket(pdm); + }; + + std::vector errors; + + if (!pdm.HasFuses()) + errors.push_back("Missing fuses - Cannot derive SBK"); + if (!pdm.HasBoot0()) + errors.push_back("Missing BOOT0 - Cannot derive master keys"); + if (!pdm.HasPackage2()) + errors.push_back("Missing BCPKG2-1-Normal-Main - Cannot derive general keys"); + if (!pdm.HasProdInfo()) + errors.push_back("Missing PRODINFO - Cannot derive title keys"); + + if (!errors.empty()) { + std::string error_str; + for (const auto& error : errors) + error_str += " - " + error + "\n"; + + QMessageBox::warning( + this, tr("Warning Missing Derivation Components"), + tr("The following are missing from your configuration that may hinder key " + "derivation. It will be attempted but may not complete.\n\n") + + QString::fromStdString(error_str)); + } + + QProgressDialog prog; + prog.setRange(0, 0); + prog.setLabelText(tr("Deriving keys...\nThis may take up to a minute depending \non your " + "system's performance.")); + prog.setWindowTitle(tr("Deriving Keys")); + + prog.show(); + + auto future = QtConcurrent::run(function); + while (!future.isFinished()) { + QCoreApplication::processEvents(); + } + + prog.close(); + } + // Necessary to load titles from nand in gamelist. Service::FileSystem::CreateFactories(vfs); game_list->LoadCompatibilityList();