123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363 |
- #include <cpp3ds/System/Err.hpp>
- #include <cpp3ds/System/I18n.hpp>
- #include <cpp3ds/System/FileInputStream.hpp>
- #include <cpp3ds/Network/Http.hpp>
- #include <cpp3ds/System/Lock.hpp>
- #include "Installer.hpp"
- #include "TitleKeys.hpp"
- #include "ticket.h"
- #include "Config.hpp"
- #include "MCU/Mcu.hpp"
- namespace {
- Result FSUSER_AddSeed(u64 titleId, const void* seed)
- {
- u32 *cmdbuf = getThreadCommandBuffer();
- cmdbuf[0] = 0x087a0180;
- cmdbuf[1] = (u32) (titleId & 0xFFFFFFFF);
- cmdbuf[2] = (u32) (titleId >> 32);
- memcpy(&cmdbuf[3], seed, 16);
- Result ret = 0;
- if(R_FAILED(ret = svcSendSyncRequest(*fsGetSessionHandle()))) return ret;
- ret = cmdbuf[1];
- return ret;
- }
- }
- namespace FreeShop {
- Installer::Installer(cpp3ds::Uint64 titleId, int contentIndex)
- : m_titleId(titleId)
- , m_isSuspended(false)
- , m_isInstalling(false)
- , m_isInstallingTmd(false)
- , m_isInstallingContent(false)
- , m_currentContentIndex(-1)
- , m_currentContentPosition(0)
- {
- m_titleType = titleId >> 32;
- m_mediaType = (m_titleType == TitleKeys::DSiWare) ? MEDIATYPE_NAND : MEDIATYPE_SD;
- if (contentIndex >= 0)
- {
- m_currentContentIndex = contentIndex;
- m_isSuspended = true;
- m_isInstalling = true;
- m_isInstallingContent = true;
- }
- }
- Installer::~Installer()
- {
- if (!m_isSuspended)
- abort();
- }
- void Installer::abort()
- {
- if (m_isSuspended)
- AM_DeletePendingTitle(m_mediaType, m_titleId);
- else
- {
- if (m_isInstallingContent)
- AM_InstallContentCancel(m_handleContent);
- if (m_isInstallingTmd)
- AM_InstallTmdAbort(m_handleTmd);
- if (m_isInstalling)
- AM_InstallTitleAbort();
- }
- m_isInstallingContent = false;
- m_isInstallingTmd = false;
- m_isInstalling = false;
- m_isSuspended = false;
- m_currentContentIndex = -1;
- m_currentContentPosition = 0;
- }
- bool Installer::installTicket(cpp3ds::Uint16 titleVersion)
- {
- Handle ticket;
- u8 tikData[sizeof(tikTemp)];
- const u16 sigSize = 0x140;
- cpp3ds::Uint64 titleIdBE = __builtin_bswap64(m_titleId);
- cpp3ds::Uint32 *key = TitleKeys::get(m_titleId);
- if (!key)
- return false;
- // Build ticket
- memcpy(tikData, tikTemp, sizeof(tikData));
- memcpy(tikData + sigSize + 0x9C, &titleIdBE, 8);
- memcpy(tikData + sigSize + 0xA6, &titleVersion, 2);
- memcpy(tikData + sigSize + 0x7F, key, 16);
- AM_QueryAvailableExternalTitleDatabase(nullptr);
- AM_DeleteTicket(m_titleId);
- if (R_SUCCEEDED(m_result = AM_InstallTicketBegin(&ticket)))
- {
- if (R_SUCCEEDED(m_result = FSFILE_Write(ticket, nullptr, 0, tikData, sizeof(tikData), 0)))
- {
- if (R_SUCCEEDED(m_result = AM_InstallTicketFinish(ticket)))
- return true;
- }
- AM_InstallTicketAbort(ticket);
- }
- m_errorStr = _("Failed to install ticket: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::installSeed(const void *seed)
- {
- if (seed)
- {
- if (R_SUCCEEDED(m_result = FSUSER_AddSeed(m_titleId, seed)))
- return true;
- }
- else // Retrieve from server
- {
- // DSiWare games can be skipped
- if (((m_titleId >> 32) & 0x8010) != 0)
- return true;
- cpp3ds::Http http("https://kagiya-ctr.cdn.nintendo.net");
- cpp3ds::Http::Request request(_("title/0x%016llX/ext_key?country=%s", m_titleId, "US"));
- cpp3ds::Http::Response response = http.sendRequest(request);
- auto status = response.getStatus();
- if (status == cpp3ds::Http::Response::Ok)
- {
- std::string seedStr = response.getBody();
- if (seedStr.size() == 16)
- {
- if (R_SUCCEEDED(m_result = FSUSER_AddSeed(m_titleId, seedStr.c_str())))
- return true;
- }
- }
- else if (status == cpp3ds::Http::Response::NotFound)
- return true; // Title has no seed, so it's fine
- else
- {
- m_errorStr = _("Failed to get seed: HTTP %d", (int)status);
- applyErrorLedPattern();
- return false;
- }
- }
- m_errorStr = _("Need FW 9.6+, Seed failed: %08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::start(bool deleteTitle)
- {
- if (!m_isInstalling)
- {
- if (deleteTitle)
- AM_DeleteTitle(m_mediaType, m_titleId);
- AM_QueryAvailableExternalTitleDatabase(nullptr);
- if (R_SUCCEEDED(m_result = AM_InstallTitleBegin(m_mediaType, m_titleId, false)))
- {
- m_isInstalling = true;
- return true;
- }
- m_errorStr = _("Failed to start: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- }
- bool Installer::resume()
- {
- if (!m_isSuspended)
- return true;
- AM_QueryAvailableExternalTitleDatabase(nullptr);
- if (m_isInstalling && R_SUCCEEDED(m_result = AM_InstallTitleResume(m_mediaType, m_titleId)))
- if (!m_isInstallingContent || R_SUCCEEDED(m_result = AM_InstallContentResume(&m_handleContent, &m_currentContentPosition, m_currentContentIndex)))
- {
- m_isSuspended = false;
- return true;
- }
- abort();
- m_errorStr = _("Failed to resume: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::suspend()
- {
- cpp3ds::Lock lock(m_mutex);
- m_result = 0;
- if (m_isInstalling && !m_isSuspended)
- {
- if (m_isInstallingContent)
- m_result = AM_InstallContentStop(m_handleContent);
- if (R_SUCCEEDED(m_result = AM_InstallTitleStop()))
- return m_isSuspended = true;
- }
- else
- return true;
- m_errorStr = _("Failed to suspend: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::commit()
- {
- if (!m_isInstalling || m_isInstallingTmd || m_isInstallingContent)
- return false;
- if (R_SUCCEEDED(m_result = AM_InstallTitleFinish()))
- if (R_SUCCEEDED(m_result = AM_CommitImportTitles(m_mediaType, 1, false, &m_titleId)))
- {
- m_isInstalling = false;
- return true;
- }
- abort();
- m_errorStr = _("Failed to commit title: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::finalizeTmd()
- {
- if (!m_isInstallingTmd)
- return false;
- if (R_SUCCEEDED(m_result = AM_InstallTmdFinish(m_handleTmd, true)))
- {
- m_isInstallingTmd = false;
- return true;
- }
- abort();
- m_errorStr = _("Failed to finalize TMD: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::finalizeContent()
- {
- if (!m_isInstallingContent)
- return false;
- if (R_SUCCEEDED(m_result = AM_InstallContentFinish(m_handleContent)))
- {
- m_isInstallingContent = false;
- return true;
- }
- abort();
- m_errorStr = _("Failed to finalize content: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::importContents(size_t count, cpp3ds::Uint16 *indices)
- {
- if (m_isInstalling && !commit())
- return false;
- if (start(false))
- if (R_SUCCEEDED(m_result = AM_CreateImportContentContexts(count, indices)))
- return true;
- abort();
- m_errorStr = _("Failed to import contents: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::installTmd(const void *data, size_t size)
- {
- cpp3ds::Lock lock(m_mutex);
- if (!m_isInstallingTmd && R_SUCCEEDED(m_result = AM_InstallTmdBegin(&m_handleTmd)))
- m_isInstallingTmd = true;
- if (m_isInstallingTmd)
- if (R_SUCCEEDED(m_result = FSFILE_Write(m_handleTmd, nullptr, 0, data, size, 0)))
- return true;
- abort();
- m_errorStr = _("Failed to install TMD: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- bool Installer::installContent(const void *data, size_t size, cpp3ds::Uint16 index)
- {
- cpp3ds::Lock lock(m_mutex);
- if (m_isSuspended)
- return false;
- if (!m_isInstallingContent && R_SUCCEEDED(m_result = AM_InstallContentBegin(&m_handleContent, index)))
- {
- m_isInstallingContent = true;
- m_currentContentIndex = index;
- m_currentContentPosition = 0;
- }
- if (m_isInstallingContent)
- if (R_SUCCEEDED(m_result = FSFILE_Write(m_handleContent, nullptr, m_currentContentPosition, data, size, 0)))
- {
- m_currentContentPosition += size;
- return true;
- }
- abort();
- m_errorStr = _("Failed to install content: 0x%08lX", m_result);
- applyErrorLedPattern();
- return false;
- }
- cpp3ds::Int32 Installer::getErrorCode() const
- {
- return m_result;
- }
- const cpp3ds::String &Installer::getErrorString() const
- {
- return m_errorStr;
- }
- int Installer::getCurrentContentIndex() const
- {
- return m_currentContentIndex;
- }
- cpp3ds::Uint64 Installer::getCurrentContentPosition() const
- {
- return m_currentContentPosition;
- }
- cpp3ds::Uint64 Installer::getTitleId() const
- {
- return m_titleId;
- }
- bool Installer::applyErrorLedPattern()
- {
- #ifndef EMULATION
- if (Config::get(Config::LEDDownloadError).GetBool())
- return MCU::getInstance().ledBlinkThrice(0x1A25FF);
- else
- return false;
- #endif
- }
- } // namespace FreeShop
|