#include #include #include #include #include "menu.hpp" #include "svnrev.h" #include "types.h" #include "lockMutex.hpp" #include "channel/nand.hpp" #include "devicemounter/usbstorage.h" #include "gui/GameTDB.hpp" #include "gui/pngu.h" #include "loader/fs.h" #include "loader/wbfs.h" #include "loader/wdvd.h" #include "network/http.h" #include "unzip/ZipFile.h" #define TAG_GAME_ID "{gameid}" #define TAG_LOC "{loc}" #define TAG_CONSOLE "{console}" #define TITLES_URL "http://www.gametdb.com/titles.txt?LANG=%s" #define GAMETDB_URL "http://www.gametdb.com/wiitdb.zip?LANG=%s&FALLBACK=TRUE&WIIWARE=TRUE&GAMECUBE=TRUE" #define UPDATE_URL_VERSION "http://dl.dropbox.com/u/25620767/WiiflowMod/versions.txt" #define CUSTOM_BANNER_URL "http://dl.dropboxusercontent.com/u/101209384/{gameid}.bnr" static const char FMT_BPIC_URL[] = "http://art.gametdb.com/{console}/coverfullHQ/{loc}/{gameid}.png"\ "|http://art.gametdb.com/{console}/coverfull/{loc}/{gameid}.png"; static const char FMT_PIC_URL[] = "http://art.gametdb.com/{console}/cover/{loc}/{gameid}.png"; static const char FMT_CBPIC_URL[] = "http://art.gametdb.com/{console}/coverfullHQ2/{loc}/{gameid}.png"; static const char FMT_CPIC_URL[] = "http://art.gametdb.com/{console}/cover2/{loc}/{gameid}.png"; static block download = { 0, 0 }; static bool settingsmenu = false; static string countryCode(const string &gameId) { switch (gameId[3]) { case 'E': return "US"; case 'J': return "JA"; case 'W': return "ZH"; case 'K': return "KO"; case 'R': return "RU"; case 'P': case 'D': case 'F': case 'I': case 'S': case 'H': case 'X': case 'Y': case 'Z': switch (CONF_GetArea()) { case CONF_AREA_BRA: return "PT"; case CONF_AREA_AUS: return "AU"; } switch (CONF_GetLanguage()) { case CONF_LANG_ENGLISH: return "EN"; case CONF_LANG_GERMAN: return "DE"; case CONF_LANG_FRENCH: return "FR"; case CONF_LANG_SPANISH: return "ES"; case CONF_LANG_ITALIAN: return "IT"; case CONF_LANG_DUTCH: return "NL"; } case 'A': switch (CONF_GetArea()) { case CONF_AREA_USA: return "US"; case CONF_AREA_JPN: return "JA"; case CONF_AREA_CHN: case CONF_AREA_HKG: case CONF_AREA_TWN: return "ZH"; case CONF_AREA_KOR: return "KO"; case CONF_AREA_BRA: return "PT"; case CONF_AREA_AUS: return "AU"; } switch (CONF_GetLanguage()) { case CONF_LANG_ENGLISH: return "EN"; case CONF_LANG_GERMAN: return "DE"; case CONF_LANG_FRENCH: return "FR"; case CONF_LANG_SPANISH: return "ES"; case CONF_LANG_ITALIAN: return "IT"; case CONF_LANG_DUTCH: return "NL"; } } return "other"; } static string makeURL(const string format, const string gameId, const string country) { string url = format; if(url.find(TAG_LOC) != url.npos) url.replace(url.find(TAG_LOC), strlen(TAG_LOC), country.c_str()); if(url.find(TAG_CONSOLE) != url.npos) url.replace(url.find(TAG_CONSOLE), strlen(TAG_CONSOLE), "wii"); url.replace(url.find(TAG_GAME_ID), strlen(TAG_GAME_ID), gameId.c_str()); return url; } void CMenu::_hideSettings(bool instant) { m_btnMgr.hide(m_downloadLblSetTitle, instant); m_btnMgr.hide(m_downloadLblCoverPrio, instant); m_btnMgr.hide(m_downloadLblPrio, instant); m_btnMgr.hide(m_downloadBtnPrioM, instant); m_btnMgr.hide(m_downloadBtnPrioP, instant); m_btnMgr.hide(m_downloadLblRegion, instant); m_btnMgr.hide(m_downloadBtnEN, instant); m_btnMgr.hide(m_downloadBtnJA, instant); m_btnMgr.hide(m_downloadBtnFR, instant); m_btnMgr.hide(m_downloadBtnDE, instant); m_btnMgr.hide(m_downloadBtnES, instant); m_btnMgr.hide(m_downloadBtnIT, instant); m_btnMgr.hide(m_downloadBtnNL, instant); m_btnMgr.hide(m_downloadBtnPT, instant); m_btnMgr.hide(m_downloadBtnRU, instant); m_btnMgr.hide(m_downloadBtnKO, instant); m_btnMgr.hide(m_downloadBtnZHCN, instant); m_btnMgr.hide(m_downloadBtnAU, instant); m_btnMgr.hide(m_downloadBtnENs, instant); m_btnMgr.hide(m_downloadBtnJAs, instant); m_btnMgr.hide(m_downloadBtnFRs, instant); m_btnMgr.hide(m_downloadBtnDEs, instant); m_btnMgr.hide(m_downloadBtnESs, instant); m_btnMgr.hide(m_downloadBtnITs, instant); m_btnMgr.hide(m_downloadBtnNLs, instant); m_btnMgr.hide(m_downloadBtnPTs, instant); m_btnMgr.hide(m_downloadBtnRUs, instant); m_btnMgr.hide(m_downloadBtnKOs, instant); m_btnMgr.hide(m_downloadBtnZHCNs, instant); m_btnMgr.hide(m_downloadBtnAUs, instant); m_btnMgr.hide(m_downloadBtnBack, instant); for(u8 i = 0; i < ARRAY_SIZE(m_downloadLblUser); ++i) if(m_downloadLblUser[i] != -1) m_btnMgr.hide(m_downloadLblUser[i], instant); } void CMenu::_showSettings() { _hideDownload(); for(u8 i = 0; i < ARRAY_SIZE(m_downloadLblUser); ++i) if(m_downloadLblUser[i] != -1) m_btnMgr.show(m_downloadLblUser[i]); m_btnMgr.show(m_downloadLblSetTitle); m_btnMgr.show(m_downloadLblCoverPrio); m_btnMgr.show(m_downloadLblPrio); m_btnMgr.show(m_downloadBtnPrioM); m_btnMgr.show(m_downloadBtnPrioP); m_btnMgr.show(m_downloadLblRegion); if( m_downloadPrioVal&C_TYPE_EN ) m_btnMgr.show(m_downloadBtnENs); else m_btnMgr.show(m_downloadBtnEN); if( m_downloadPrioVal&C_TYPE_JA ) m_btnMgr.show(m_downloadBtnJAs); else m_btnMgr.show(m_downloadBtnJA); if( m_downloadPrioVal&C_TYPE_FR ) m_btnMgr.show(m_downloadBtnFRs); else m_btnMgr.show(m_downloadBtnFR); if( m_downloadPrioVal&C_TYPE_DE ) m_btnMgr.show(m_downloadBtnDEs); else m_btnMgr.show(m_downloadBtnDE); if( m_downloadPrioVal&C_TYPE_ES ) m_btnMgr.show(m_downloadBtnESs); else m_btnMgr.show(m_downloadBtnES); if( m_downloadPrioVal&C_TYPE_IT ) m_btnMgr.show(m_downloadBtnITs); else m_btnMgr.show(m_downloadBtnIT); if( m_downloadPrioVal&C_TYPE_NL ) m_btnMgr.show(m_downloadBtnNLs); else m_btnMgr.show(m_downloadBtnNL); if( m_downloadPrioVal&C_TYPE_PT ) m_btnMgr.show(m_downloadBtnPTs); else m_btnMgr.show(m_downloadBtnPT); if( m_downloadPrioVal&C_TYPE_RU ) m_btnMgr.show(m_downloadBtnRUs); else m_btnMgr.show(m_downloadBtnRU); if( m_downloadPrioVal&C_TYPE_KO ) m_btnMgr.show(m_downloadBtnKOs); else m_btnMgr.show(m_downloadBtnKO); if( m_downloadPrioVal&C_TYPE_ZHCN ) m_btnMgr.show(m_downloadBtnZHCNs); else m_btnMgr.show(m_downloadBtnZHCN); if( m_downloadPrioVal&C_TYPE_AU ) m_btnMgr.show(m_downloadBtnAUs); else m_btnMgr.show(m_downloadBtnAU); m_btnMgr.show(m_downloadBtnBack); if( m_downloadPrioVal&C_TYPE_ONOR ) { m_btnMgr.setText(m_downloadLblPrio, _t("dl24", L"Custom only")); } else { if( m_downloadPrioVal&C_TYPE_ONCU ) { if( m_downloadPrioVal&C_TYPE_PRIOA ) { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_btnMgr.setText(m_downloadLblPrio, _t("dl23", L"Custom/Custom")); } else { m_btnMgr.setText(m_downloadLblPrio, _t("dl22", L"Custom/Original")); } } else { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_btnMgr.setText(m_downloadLblPrio, _t("dl21", L"Original/Custom")); } else { m_btnMgr.setText(m_downloadLblPrio, _t("dl20", L"Original/Original")); } } } else { m_btnMgr.setText(m_downloadLblPrio, _t("dl19", L"Original only")); } } } void CMenu::_hideDownload(bool instant) { m_btnMgr.hide(m_downloadLblTitle, instant); m_btnMgr.hide(m_downloadBtnCancel, instant); m_btnMgr.hide(m_downloadBtnAll, instant); m_btnMgr.hide(m_downloadBtnMissing, instant); m_btnMgr.hide(m_downloadLblCoverSet, instant); m_btnMgr.hide(m_downloadBtnCoverSet, instant); m_btnMgr.hide(m_downloadBtnGameTDBDownload, instant); m_btnMgr.hide(m_downloadPBar, instant); m_btnMgr.hide(m_downloadLblMessage[0], 0, 0, -2.f, 0.f, instant); m_btnMgr.hide(m_downloadLblMessage[1], 0, 0, -2.f, 0.f, instant); m_btnMgr.hide(m_downloadLblCovers, instant); m_btnMgr.hide(m_downloadLblGameTDBDownload, instant); m_btnMgr.hide(m_downloadLblGameTDB, instant); for(u8 i = 0; i < ARRAY_SIZE(m_downloadLblUser); ++i) if(m_downloadLblUser[i] != -1) m_btnMgr.hide(m_downloadLblUser[i], instant); } void CMenu::_showDownload(void) { _hideSettings(); _setBg(m_downloadBg, m_downloadBg); m_btnMgr.show(m_downloadLblGameTDB); m_btnMgr.show(m_downloadLblTitle); m_btnMgr.show(m_downloadBtnCancel); m_btnMgr.show(m_downloadBtnAll); m_btnMgr.show(m_downloadBtnMissing); m_btnMgr.show(m_downloadLblCoverSet); m_btnMgr.show(m_downloadBtnCoverSet); m_btnMgr.show(m_downloadLblCovers); if (!m_locked) { m_btnMgr.show(m_downloadLblGameTDBDownload); m_btnMgr.show(m_downloadBtnGameTDBDownload); } for(u8 i = 0; i < ARRAY_SIZE(m_downloadLblUser); ++i) if(m_downloadLblUser[i] != -1) m_btnMgr.show(m_downloadLblUser[i]); } void CMenu::_setThrdMsg(const wstringEx &msg, float progress) { if (m_thrdStop) return; if (msg != L"...") m_thrdMessage = msg; m_thrdMessageAdded = true; m_thrdProgress = progress; } bool CMenu::_downloadProgress(void *obj, int size, int position) { CMenu *m = (CMenu *)obj; LWP_MutexLock(m->m_mutex); m->_setThrdMsg(L"...", m->m_thrdStep + m->m_thrdStepLen * ((float)position / (float)size)); LWP_MutexUnlock(m->m_mutex); return !m->m_thrdStop; } int CMenu::_coverDownloaderAll(CMenu *m) { if (!m->m_thrdWorking) return 0; m->_coverDownloader(false); m->m_thrdWorking = false; return 0; } int CMenu::_coverDownloaderMissing(CMenu *m) { if (!m->m_thrdWorking) return 0; m->_coverDownloader(true); m->m_thrdWorking = false; return 0; } static bool checkPNGBuf(u8 *data) { if(data == NULL) return false; PNGUPROP imgProp; IMGCTX ctx = PNGU_SelectImageFromBuffer(data); if (ctx == NULL) return false; int ret = PNGU_GetImageProperties(ctx, &imgProp); PNGU_ReleaseImageContext(ctx); return ret == PNGU_OK; } static bool checkPNGFile(const char *filename) { u32 fileSize = 0; u8 *buffer = fsop_ReadFile(filename, &fileSize); bool ret = false; if(buffer != NULL) { ret = checkPNGBuf(buffer); MEM2_free(buffer); } return ret; } void CMenu::_initAsyncNetwork() { if (!_isNetworkAvailable()) return; m_thrdNetwork = true; net_init_async(_networkComplete, this); } bool CMenu::_isNetworkAvailable() { bool retval = false; u32 size; char ISFS_Filepath[32] ATTRIBUTE_ALIGN(32); strcpy(ISFS_Filepath, "/shared2/sys/net/02/config.dat"); u8 *buf = ISFS_GetFile(ISFS_Filepath, &size, -1); if (buf && size > 4) { retval = buf[4] > 0; // There is a valid connection defined. free(buf); } return retval; } s32 CMenu::_networkComplete(s32 ok, void *usrData) { CMenu *m = (CMenu *) usrData; networkInit = ok == 0; m->m_thrdNetwork = false; gprintf("NET: Network init complete, enabled wifi_gecko: %s\n", m->m_use_wifi_gecko ? "yes" : "no"); if(m->m_use_wifi_gecko) { const string &ip = m->m_cfg.getString("DEBUG", "wifi_gecko_ip"); u16 port = m->m_cfg.getInt("DEBUG", "wifi_gecko_port", 4405); if(ip.size() > 0 && port != 0) WiFiDebugger.Init(ip.c_str(), port); } return 0; } int CMenu::_initNetwork() { while (net_get_status() == -EBUSY && m_thrdNetwork == true) { usleep(100); }; // Async initialization may be busy, wait to see if it succeeds. if (networkInit) return 0; if (!_isNetworkAvailable()) return -2; char ip[16]; int val = if_config(ip, NULL, NULL, true); networkInit = !val; return val; } int CMenu::_coverDownloader(bool missingOnly) { vector coverList; vector pluginCoverList; int count = 0, countFlat = 0; float listWeight = missingOnly ? 0.125f : 0.f; // 1/8 of the progress bar for testing the PNGs we already have float dlWeight = 1.f - listWeight; u32 bufferSize = 0x280000; // Maximum download size 2 MB u8 *buffer = (u8*)MEM2_alloc(bufferSize); if(buffer == NULL) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg27", L"Not enough memory!"), 1.f); LWP_MutexUnlock(m_mutex); m_thrdWorking = false; return 0; } bool savePNG = m_cfg.getBool("GENERAL", "keep_png", true); vector fmtURLBox = stringToVector(m_cfg.getString("GENERAL", "url_full_covers", FMT_BPIC_URL), '|'); vector fmtURLFlat = stringToVector(m_cfg.getString("GENERAL", "url_flat_covers", FMT_PIC_URL), '|'); vector fmtURLCBox = stringToVector(m_cfg.getString("GENERAL", "url_custom_full_covers", FMT_CBPIC_URL), '|'); vector fmtURLCFlat = stringToVector(m_cfg.getString("GENERAL", "url_custom_flat_covers", FMT_CPIC_URL), '|'); u32 nbSteps = m_gameList.size(); u32 step = 0; GameTDB c_gameTDB; if (m_settingsDir.size() > 0) { c_gameTDB.OpenFile(fmt("%s/wiitdb.xml", m_settingsDir.c_str())); c_gameTDB.SetLanguageCode(m_curLanguage.c_str()); } Config m_checksums; m_checksums.load(fmt("%s/%s", m_settingsDir.c_str(), PLUGIN_CRCS_FILENAME)); if (m_coverDLGameId.empty()) { coverList.reserve(m_gameList.size()); for (u32 i = 0; i < m_gameList.size() && !m_thrdStop; ++i) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg7", L"Listing covers to download..."), listWeight * (float)step / (float)nbSteps); LWP_MutexUnlock(m_mutex); ++step; char *path = NULL; const char *id = NULL; if(m_gameList[i].type == TYPE_PLUGIN) { char gamePath[256]; if(strchr(m_gameList[i].path, '/') != NULL) strncpy(gamePath, strrchr(m_gameList[i].path, '/') + 1, sizeof(gamePath)); else strncpy(gamePath, m_gameList[i].path, sizeof(gamePath)); path = fmt_malloc("%s/%s.png", m_boxPicDir.c_str(), gamePath); id = path; } else { path = fmt_malloc("%s/%s.png", m_boxPicDir.c_str(), m_gameList[i].id); id = m_gameList[i].id; } if(!missingOnly || (id != NULL && !CoverFlow.fullCoverCached(id) && path != NULL && !checkPNGFile(path))) { if(m_gameList[i].type == TYPE_PLUGIN) pluginCoverList.push_back(m_gameList[i]); if(id != NULL) coverList.push_back(id); } if(path != NULL) MEM2_free(path); path = NULL; } } else coverList.push_back(m_coverDLGameId); u32 n = coverList.size(); if (n > 0 && !m_thrdStop) { step = 0; nbSteps = 1 + n * 2; LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg1", L"Initializing network..."), listWeight + dlWeight * (float)step / (float)nbSteps); LWP_MutexUnlock(m_mutex); if (_initNetwork() < 0) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg2", L"Network initialization failed!"), 1.f); LWP_MutexUnlock(m_mutex); m_thrdWorking = false; free(buffer); return 0; } m_thrdStepLen = dlWeight / (float)nbSteps; Config m_newID; m_newID.load(fmt("%s/newid.ini", m_settingsDir.c_str())); m_newID.setString("CHANNELS", "WFSF", "DWFA"); u32 CoverType = 0; for(u32 i = 0; i < coverList.size() && !m_thrdStop; ++i) { string url; const char *domain = _domainFromView(); bool success = false; bool original = true; bool custom = false; int c_altCase = 0; const string &newID = m_newID.getString(domain, coverList[i], coverList[i]); if(!newID.empty() && strncasecmp(newID.c_str(), coverList[i].c_str(), coverList[i].length()) == 0) m_newID.remove(domain, coverList[i]); else if(!newID.empty()) { gprintf("old id = %s\nnew id = %s\n", coverList[i].c_str(), newID.c_str()); } for( int p = 0; p < 4; ++p ) { switch(p) { case 0: CoverType = m_downloadPrioVal&C_TYPE_PRIOA ? CBOX : BOX; break; case 1: CoverType = m_downloadPrioVal&C_TYPE_PRIOA ? ( m_downloadPrioVal&C_TYPE_PRIOB ? CFLAT : BOX ) : ( m_downloadPrioVal&C_TYPE_PRIOB ? CBOX : FLAT ); break; case 2: CoverType = m_downloadPrioVal&C_TYPE_PRIOA ? ( m_downloadPrioVal&C_TYPE_PRIOB ? BOX : CFLAT ) : ( m_downloadPrioVal&C_TYPE_PRIOB ? FLAT : CBOX ); break; case 3: CoverType = m_downloadPrioVal&C_TYPE_PRIOA ? FLAT : CFLAT; break; } switch( CoverType ) { case BOX: if( m_downloadPrioVal&C_TYPE_ONOR ) original = false; if (!success && !m_thrdStop && original) { char *path = fmt_malloc("%s/%s.png", m_boxPicDir.c_str(), coverList[i].c_str()); if(path != NULL && !checkPNGFile(path)) { for (u32 j = 0; !success && j < fmtURLBox.size() && !m_thrdStop; ++j) { if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) url = m_plugin.GenerateCoverLink(pluginCoverList[i], fmtURLBox[j], m_checksums); else url = makeURL(fmtURLBox[j], newID, countryCode(newID)); if (j == 0) ++step; m_thrdStep = listWeight + dlWeight * (float)step / (float)nbSteps; LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); for( int o = 0; o < 12; ++o ) { bool tdl = false; if(download.data != NULL && download.size > 0 && checkPNGBuf(download.data)) break; if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) break; switch( o ) { case EN: if(( newID[3] == 'E' || newID[3] == 'X' || newID[3] == 'Y' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_EN ) url = makeURL(fmtURLBox[j], newID, "EN"); tdl = true; break; case JA: if(newID[3] == 'J' && m_downloadPrioVal&C_TYPE_JA) { url = makeURL(fmtURLBox[j], newID, "JA"); tdl = true; } break; case FR: if((newID[3] == 'F' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_FR) { url = makeURL(fmtURLBox[j], newID, "FR"); tdl = true; } break; case DE: if((newID[3] == 'D' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_DE) { url = makeURL(fmtURLBox[j], newID, "DE"); tdl = true; } break; case ES: if((newID[3] == 'S' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_ES) { url = makeURL(fmtURLBox[j], newID, "ES"); tdl = true; } break; case IT: if((newID[3] == 'I' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_IT) { url = makeURL(fmtURLBox[j], newID, "IT"); tdl = true; } break; case NL: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_NL) { url = makeURL(fmtURLBox[j], newID, "NL"); tdl = true; } break; case PT: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_PT) { url = makeURL(fmtURLBox[j], newID, "PT"); tdl = true; } break; case RU: if((newID[3] == 'R' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_RU) { url = makeURL(fmtURLBox[j], newID, "RU"); tdl = true; } break; case KO: if(newID[3] == 'K' && m_downloadPrioVal&C_TYPE_KO) { url = makeURL(fmtURLBox[j], newID, "KO"); tdl = true; } break; case AU: if(newID[3] == 'W' && m_downloadPrioVal&C_TYPE_ZHCN) { url = makeURL(fmtURLBox[j], newID, "ZH"); tdl = true; } break; case ZHCN: break; } if ( tdl ) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); } } if(download.data == NULL || download.size == 0 || !checkPNGBuf(download.data)) continue; if (savePNG) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg4", L"Saving %s"), path), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); fsop_WriteFile(path, download.data, download.size); } LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg10", L"Making %s"), sfmt("%s.wfc", coverList[i].c_str()).c_str()), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); if (CoverFlow.preCacheCover(coverList[i].c_str(), download.data, true)) { ++count; success = true; } } } if(path != NULL) MEM2_free(path); path = NULL; } break; case CBOX: if( m_downloadPrioVal&C_TYPE_ONCU ) custom = true; c_altCase = c_gameTDB.GetCaseVersions( coverList[i].c_str() ); if(!success && !m_thrdStop && c_gameTDB.IsLoaded() && c_altCase > 1 && custom) { char *path = fmt_malloc("%s/%s.png", m_boxPicDir.c_str(), coverList[i].c_str()); if(path != NULL && !checkPNGFile(path)) { for (u32 j = 0; !success && j < fmtURLCBox.size() && !m_thrdStop; ++j) { if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) url = m_plugin.GenerateCoverLink(pluginCoverList[i], fmtURLCBox[j], m_checksums); else url = makeURL(fmtURLCBox[j], newID, countryCode(newID)); if (j == 0) ++step; m_thrdStep = listWeight + dlWeight * (float)step / (float)nbSteps; LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); for( int o = 0; o < 12; ++o ) { bool tdl = false; if(download.data != NULL && download.size > 0 && checkPNGBuf(download.data)) break; if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) break; switch( o ) { case EN: if(( newID[3] == 'E' || newID[3] == 'X' || newID[3] == 'Y' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_EN ) { url = makeURL(fmtURLCBox[j], newID, "EN"); tdl = true; } break; case JA: if(newID[3] == 'J' && m_downloadPrioVal&C_TYPE_JA) { url = makeURL(fmtURLCBox[j], newID, "JA"); tdl = true; } break; case FR: if((newID[3] == 'F' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_FR) { url = makeURL(fmtURLCBox[j], newID, "FR"); tdl = true; } break; case DE: if((newID[3] == 'D' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_DE) { url = makeURL(fmtURLCBox[j], newID, "DE"); tdl = true; } break; case ES: if((newID[3] == 'S' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_ES) { url = makeURL(fmtURLCBox[j], newID, "ES"); tdl = true; } break; case IT: if((newID[3] == 'I' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_IT) { url = makeURL(fmtURLCBox[j], newID, "IT"); tdl = true; } break; case NL: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_NL) { url = makeURL(fmtURLCBox[j], newID, "NL"); tdl = true; } break; case PT: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_PT) { url = makeURL(fmtURLCBox[j], newID, "PT"); tdl = true; } break; case RU: if((newID[3] == 'R' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_RU) { url = makeURL(fmtURLCBox[j], newID, "RU"); tdl = true; } break; case KO: if(newID[3] == 'K' && m_downloadPrioVal&C_TYPE_KO) { url = makeURL(fmtURLCBox[j], newID, "KO"); tdl = true; } break; case AU: if(newID[3] == 'W' && m_downloadPrioVal&C_TYPE_ZHCN) { url = makeURL(fmtURLCBox[j], newID, "ZH"); tdl = true; } break; case ZHCN: break; } if ( tdl ) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); } } if(download.data == NULL || download.size == 0 || !checkPNGBuf(download.data)) continue; if (savePNG) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg4", L"Saving %s"), path), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); fsop_WriteFile(path, download.data, download.size); } LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg10", L"Making %s"), sfmt("%s.wfc", coverList[i].c_str()).c_str()), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); if (CoverFlow.preCacheCover(coverList[i].c_str(), download.data, true)) { ++count; success = true; } } } if(path != NULL) MEM2_free(path); path = NULL; } break; case FLAT: if( m_downloadPrioVal&C_TYPE_ONOR ) original = false; if(!success && !m_thrdStop && original) { char *path = fmt_malloc("%s/%s.png", m_picDir.c_str(), coverList[i].c_str()); if(path != NULL && !checkPNGFile(path)) { // Try to get the front cover if (m_thrdStop) break; for (u32 j = 0; !success && j < fmtURLFlat.size() && !m_thrdStop; ++j) { if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) url = m_plugin.GenerateCoverLink(pluginCoverList[i], fmtURLFlat[j], m_checksums); else url = makeURL(fmtURLFlat[j], newID, countryCode(newID)); LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg8", L"Full cover not found. Downloading from %s"), url.c_str()), listWeight + dlWeight * (float)step / (float)nbSteps); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); for( int o = 0; o < 12; ++o ) { bool tdl = false; if(download.data != NULL && download.size > 0 && checkPNGBuf(download.data)) break; if(pluginCoverList.size() && pluginCoverList[i].type == TYPE_PLUGIN) break; switch( o ) { case EN: if(( newID[3] == 'E' || newID[3] == 'X' || newID[3] == 'Y' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_EN ) { url = makeURL(fmtURLFlat[j], newID, "EN"); tdl = true; } break; case JA: if(newID[3] == 'J' && m_downloadPrioVal&C_TYPE_JA) { url = makeURL(fmtURLFlat[j], newID, "JA"); tdl = true; } break; case FR: if((newID[3] == 'F' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_FR) { url = makeURL(fmtURLFlat[j], newID, "FR"); tdl = true; } break; case DE: if((newID[3] == 'D' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_DE) { url = makeURL(fmtURLFlat[j], newID, "DE"); tdl = true; } break; case ES: if((newID[3] == 'S' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_ES) { url = makeURL(fmtURLFlat[j], newID, "ES"); tdl = true; } break; case IT: if((newID[3] == 'I' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_IT) { url = makeURL(fmtURLFlat[j], newID, "IT"); tdl = true; } break; case NL: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_NL) { url = makeURL(fmtURLFlat[j], newID, "NL"); tdl = true; } break; case PT: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_PT) { url = makeURL(fmtURLFlat[j], newID, "PT"); tdl = true; } break; case RU: if((newID[3] == 'R' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_RU) { url = makeURL(fmtURLFlat[j], newID, "RU"); tdl = true; } break; case KO: if(newID[3] == 'K' && m_downloadPrioVal&C_TYPE_KO) { url = makeURL(fmtURLFlat[j], newID, "KO"); tdl = true; } break; case AU: if(newID[3] == 'W' && m_downloadPrioVal&C_TYPE_ZHCN) { url = makeURL(fmtURLFlat[j], newID, "ZH"); tdl = true; } break; case ZHCN: break; } if ( tdl ) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); } } if(download.data == NULL || download.size == 0 || !checkPNGBuf(download.data)) continue; if (savePNG) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg4", L"Saving %s"), path), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); fsop_WriteFile(path, download.data, download.size); } LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg10", L"Making %s"), sfmt("%s.wfc", coverList[i].c_str()).c_str()), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); if (CoverFlow.preCacheCover(coverList[i].c_str(), download.data, false)) { ++countFlat; success = true; } } } if(path != NULL) MEM2_free(path); path = NULL; } break; case CFLAT: if( m_downloadPrioVal&C_TYPE_ONCU ) custom = true; if(!success && !m_thrdStop && c_gameTDB.IsLoaded() && c_altCase > 1 && custom) { char *path = fmt_malloc("%s/%s.png", m_picDir.c_str(), coverList[i].c_str()); if(path != NULL && !checkPNGFile(path)) { // Try to get the front cover if (m_thrdStop) break; for (u32 j = 0; !success && j < fmtURLCFlat.size() && !m_thrdStop; ++j) { url = makeURL(fmtURLCFlat[j], newID, countryCode(newID)); LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg8", L"Full cover not found. Downloading from %s"), url.c_str()), listWeight + dlWeight * (float)step / (float)nbSteps); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); for( int o = 0; o < 12; ++o ) { bool tdl = false; if(download.data != NULL && download.size > 0 && checkPNGBuf(download.data)) break; switch( o ) { case EN: if(( newID[3] == 'E' || newID[3] == 'X' || newID[3] == 'Y' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_EN ) { url = makeURL(fmtURLCFlat[j], newID, "EN"); tdl = true; } break; case JA: if(newID[3] == 'J' && m_downloadPrioVal&C_TYPE_JA) { url = makeURL(fmtURLCFlat[j], newID, "JA"); tdl = true; } break; case FR: if((newID[3] == 'F' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_FR) { url = makeURL(fmtURLCFlat[j], newID, "FR"); tdl = true; } break; case DE: if((newID[3] == 'D' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_DE) { url = makeURL(fmtURLCFlat[j], newID, "DE"); tdl = true; } break; case ES: if((newID[3] == 'S' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_ES) { url = makeURL(fmtURLCFlat[j], newID, "ES"); tdl = true; } break; case IT: if((newID[3] == 'I' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_IT) { url = makeURL(fmtURLCFlat[j], newID, "IT"); tdl = true; } break; case NL: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_NL) { url = makeURL(fmtURLCFlat[j], newID, "NL"); tdl = true; } break; case PT: if(newID[3] == 'P' && m_downloadPrioVal&C_TYPE_PT) { url = makeURL(fmtURLCFlat[j], newID, "PT"); tdl = true; } break; case RU: if((newID[3] == 'R' || newID[3] == 'P') && m_downloadPrioVal&C_TYPE_RU) { url = makeURL(fmtURLCFlat[j], newID, "RU"); tdl = true; } break; case KO: if(newID[3] == 'K' && m_downloadPrioVal&C_TYPE_KO) { url = makeURL(fmtURLCFlat[j], newID, "KO"); tdl = true; } break; case AU: if((newID[3] == 'P' || newID[3] == 'Y' || newID[3] == 'X') && m_downloadPrioVal&C_TYPE_ZHCN) { url = makeURL(fmtURLCFlat[j], newID, "ZH"); tdl = true; } break; case ZHCN: break; } if ( tdl ) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg3", L"Downloading from %s"), url.c_str()), m_thrdStep); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, url.c_str(), CMenu::_downloadProgress, this); } } if(download.data == NULL || download.size == 0 || !checkPNGBuf(download.data)) continue; if(savePNG) { LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg4", L"Saving %s"), path), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); fsop_WriteFile(path, download.data, download.size); } LWP_MutexLock(m_mutex); _setThrdMsg(wfmt(_fmt("dlmsg10", L"Making %s"), sfmt("%s.wfc", coverList[i].c_str()).c_str()), listWeight + dlWeight * (float)(step + 1) / (float)nbSteps); LWP_MutexUnlock(m_mutex); if (CoverFlow.preCacheCover(coverList[i].c_str(), download.data, false)) { ++countFlat; success = true; } } } if(path != NULL) MEM2_free(path); path = NULL; } break; } } ++step; } if(c_gameTDB.IsLoaded()) c_gameTDB.CloseFile(); coverList.clear(); m_checksums.unload(); m_newID.unload(); } LWP_MutexLock(m_mutex); if (countFlat == 0) _setThrdMsg(wfmt(_fmt("dlmsg5", L"%i/%i files downloaded."), count, n), 1.f); else _setThrdMsg(wfmt(_fmt("dlmsg9", L"%i/%i files downloaded. %i are front covers only."), count + countFlat, n, countFlat), 1.f); LWP_MutexUnlock(m_mutex); m_thrdWorking = false; pluginCoverList.clear(); free(buffer); return 0; } void CMenu::_download(string gameId) { lwp_t thread = LWP_THREAD_NULL; int msg = 0; wstringEx prevMsg; bool _updateGametdb = false; SetupInput(); _showDownload(); m_btnMgr.setText(m_downloadBtnCancel, _t("dl1", L"Cancel")); m_thrdStop = false; m_thrdMessageAdded = false; if(gameId.size() && CoverFlow.getHdr()->type == TYPE_PLUGIN) { char gamePath[256]; if(strchr(CoverFlow.getHdr()->path, '/') != NULL) strncpy(gamePath, strrchr(CoverFlow.getHdr()->path, '/') + 1, sizeof(gamePath)); else strncpy(gamePath, CoverFlow.getHdr()->path, sizeof(gamePath)); m_coverDLGameId = gamePath; } else m_coverDLGameId = gameId; while(!m_exit) { _mainLoopCommon(); if ((BTN_HOME_PRESSED || BTN_B_PRESSED) && !m_thrdWorking) { if(settingsmenu) { settingsmenu = false; _hideSettings(); _showDownload(); } else break; } else if (BTN_UP_PRESSED) m_btnMgr.up(); else if (BTN_DOWN_PRESSED) m_btnMgr.down(); if ((BTN_A_PRESSED || !gameId.empty()) && !(m_thrdWorking && m_thrdStop)) { if ((m_btnMgr.selected(m_downloadBtnAll) || m_btnMgr.selected(m_downloadBtnMissing) || !gameId.empty()) && !m_thrdWorking) { bool dlAll = m_btnMgr.selected(m_downloadBtnAll); m_btnMgr.show(m_downloadPBar); m_btnMgr.setProgress(m_downloadPBar, 0.f); _hideSettings(); m_btnMgr.hide(m_downloadBtnAll); m_btnMgr.hide(m_downloadBtnMissing); m_btnMgr.hide(m_downloadBtnGameTDBDownload); m_btnMgr.hide(m_downloadLblCovers); m_btnMgr.hide(m_downloadLblGameTDBDownload); m_btnMgr.hide(m_downloadLblCoverSet); m_btnMgr.hide(m_downloadBtnCoverSet); m_thrdStop = false; m_thrdWorking = true; gameId.clear(); if (dlAll) LWP_CreateThread(&thread, (void *(*)(void *))CMenu::_coverDownloaderAll, (void *)this, 0, 8192, 40); else LWP_CreateThread(&thread, (void *(*)(void *))CMenu::_coverDownloaderMissing, (void *)this, 0, 8192, 40); } else if (m_btnMgr.selected(m_downloadBtnPrioM) && !m_thrdWorking) { if( m_downloadPrioVal&C_TYPE_ONOR ) { m_downloadPrioVal ^= C_TYPE_ONOR; } else { if( m_downloadPrioVal&C_TYPE_ONCU ) { if( m_downloadPrioVal&C_TYPE_PRIOA ) { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_downloadPrioVal ^= C_TYPE_PRIOB; } else { m_downloadPrioVal ^= C_TYPE_PRIOA; m_downloadPrioVal ^= C_TYPE_PRIOB; } } else { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_downloadPrioVal ^= C_TYPE_PRIOB; } else { m_downloadPrioVal ^= C_TYPE_ONCU; } } } else { m_downloadPrioVal ^= C_TYPE_ONOR; m_downloadPrioVal ^= C_TYPE_ONCU; m_downloadPrioVal ^= C_TYPE_PRIOA; m_downloadPrioVal ^= C_TYPE_PRIOB; } } _hideSettings(); m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if (m_btnMgr.selected(m_downloadBtnPrioP) && !m_thrdWorking) { if( m_downloadPrioVal&C_TYPE_ONOR ) { m_downloadPrioVal ^= C_TYPE_ONOR; m_downloadPrioVal ^= C_TYPE_ONCU; m_downloadPrioVal ^= C_TYPE_PRIOA; m_downloadPrioVal ^= C_TYPE_PRIOB; } else { if( m_downloadPrioVal&C_TYPE_ONCU ) { if( m_downloadPrioVal&C_TYPE_PRIOA ) { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_downloadPrioVal ^= C_TYPE_ONOR; } else { m_downloadPrioVal ^= C_TYPE_PRIOB; } } else { if(m_downloadPrioVal&C_TYPE_PRIOB ) { m_downloadPrioVal ^= C_TYPE_PRIOA; m_downloadPrioVal ^= C_TYPE_PRIOB; } else { m_downloadPrioVal ^= C_TYPE_PRIOB; } } } else { m_downloadPrioVal ^= C_TYPE_ONCU; } } _hideSettings(); m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if (m_btnMgr.selected(m_downloadBtnCoverSet) && !m_thrdWorking) { settingsmenu = true; _showSettings(); } else if (m_btnMgr.selected(m_downloadBtnBack) && !m_thrdWorking) { _hideSettings(); _showDownload(); } else if ((m_btnMgr.selected(m_downloadBtnEN) || m_btnMgr.selected(m_downloadBtnENs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_EN; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnJA) || m_btnMgr.selected(m_downloadBtnJAs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_JA; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnFR) || m_btnMgr.selected(m_downloadBtnFRs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_FR; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnDE) || m_btnMgr.selected(m_downloadBtnDEs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_DE; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnES) || m_btnMgr.selected(m_downloadBtnESs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_ES; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnIT) || m_btnMgr.selected(m_downloadBtnITs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_IT; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnNL) || m_btnMgr.selected(m_downloadBtnNLs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_NL; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnPT) || m_btnMgr.selected(m_downloadBtnPTs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_PT; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnRU) || m_btnMgr.selected(m_downloadBtnRUs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_RU; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnKO) || m_btnMgr.selected(m_downloadBtnKOs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_KO; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnZHCN) || m_btnMgr.selected(m_downloadBtnZHCNs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_ZHCN; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if ((m_btnMgr.selected(m_downloadBtnAU) || m_btnMgr.selected(m_downloadBtnAUs)) && !m_thrdWorking) { _hideSettings(); m_downloadPrioVal ^= C_TYPE_AU; m_cfg.setUInt( "GENERAL", "cover_prio", m_downloadPrioVal ); m_cfg.save(); _showSettings(); } else if (m_btnMgr.selected(m_downloadBtnGameTDBDownload) && !m_thrdWorking) { m_btnMgr.show(m_downloadPBar); m_btnMgr.setProgress(m_downloadPBar, 0.f); _hideSettings(); m_btnMgr.hide(m_downloadBtnAll); m_btnMgr.hide(m_downloadBtnMissing); m_btnMgr.hide(m_downloadBtnGameTDBDownload); m_btnMgr.hide(m_downloadLblCovers); m_btnMgr.hide(m_downloadLblGameTDBDownload); m_btnMgr.hide(m_downloadLblCoverSet); m_btnMgr.hide(m_downloadBtnCoverSet); m_thrdStop = false; m_thrdWorking = true; _updateGametdb = true; LWP_CreateThread(&thread, (void *(*)(void *))CMenu::_gametdbDownloader, (void *)this, 0, 8192, 40); } else if (m_btnMgr.selected(m_downloadBtnCancel)) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); } } if (Sys_Exiting()) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); m_thrdWorking = false; } if (m_thrdMessageAdded) { LockMutex lock(m_mutex); m_thrdMessageAdded = false; m_btnMgr.setProgress(m_downloadPBar, m_thrdProgress); if (m_thrdProgress == 1.f) { if (_updateGametdb) break; m_btnMgr.setText(m_downloadBtnCancel, _t("dl2", L"Back")); } if (prevMsg != m_thrdMessage) { prevMsg = m_thrdMessage; m_btnMgr.setText(m_downloadLblMessage[msg], m_thrdMessage, false); m_btnMgr.hide(m_downloadLblMessage[msg], 0, 0, -1.f, -1.f, true); m_btnMgr.show(m_downloadLblMessage[msg]); msg ^= 1; m_btnMgr.hide(m_downloadLblMessage[msg], 0, 0, -1.f, -1.f); } } if (m_thrdStop && !m_thrdWorking) break; } if (thread != LWP_THREAD_NULL) { LWP_JoinThread(thread, NULL); thread = LWP_THREAD_NULL; } _hideDownload(); _hideSettings(); } void CMenu::_initDownloadMenu() { // Download menu _addUserLabels(m_downloadLblUser, ARRAY_SIZE(m_downloadLblUser), "DOWNLOAD"); m_downloadBg = _texture("DOWNLOAD/BG", "texture", theme.bg, false); m_downloadLblTitle = _addTitle("DOWNLOAD/TITLE", theme.titleFont, L"", 20, 30, 600, 60, theme.titleFontColor, FTGX_JUSTIFY_CENTER | FTGX_ALIGN_MIDDLE); m_downloadPBar = _addProgressBar("DOWNLOAD/PROGRESS_BAR", 40, 200, 560, 20); m_downloadBtnCancel = _addButton("DOWNLOAD/CANCEL_BTN", theme.btnFont, L"", 420, 400, 200, 56, theme.btnFontColor); m_downloadLblCovers = _addLabel("DOWNLOAD/COVERS", theme.btnFont, L"", 40, 130, 320, 56, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadBtnAll = _addButton("DOWNLOAD/ALL_BTN", theme.btnFont, L"", 370, 130, 230, 56, theme.btnFontColor); m_downloadBtnMissing = _addButton("DOWNLOAD/MISSING_BTN", theme.btnFont, L"", 370, 190, 230, 56, theme.btnFontColor); m_downloadLblCoverSet = _addLabel("DOWNLOAD/COVERSSET", theme.btnFont, L"", 40, 250, 320, 56, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadBtnCoverSet = _addButton("DOWNLOAD/COVERSET_BTN", theme.btnFont, L"", 370, 250, 230, 56, theme.btnFontColor); m_downloadLblGameTDBDownload = _addLabel("DOWNLOAD/GAMETDB_DOWNLOAD", theme.btnFont, L"", 40, 310, 320, 56, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadBtnGameTDBDownload = _addButton("DOWNLOAD/GAMETDB_DOWNLOAD_BTN", theme.btnFont, L"", 370, 310, 230, 56, theme.btnFontColor); m_downloadLblGameTDB = _addLabel("DOWNLOAD/GAMETDB", theme.lblFont, L"", 40, 390, 370, 60, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadLblMessage[0] = _addLabel("DOWNLOAD/MESSAGE1", theme.lblFont, L"", 40, 228, 560, 100, theme.txtFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_TOP); m_downloadLblMessage[1] = _addLabel("DOWNLOAD/MESSAGE2", theme.lblFont, L"", 40, 228, 560, 100, theme.txtFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_TOP); // Cover settings m_downloadLblSetTitle = _addTitle("DOWNLOAD/SETTITLE", theme.titleFont, L"", 20, 30, 600, 60, theme.titleFontColor, FTGX_JUSTIFY_CENTER | FTGX_ALIGN_MIDDLE); m_downloadLblCoverPrio = _addLabel("DOWNLOAD/COVERPRIO", theme.lblFont, L"", 40, 100, 290, 56, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadLblPrio = _addLabel("DOWNLOAD/PRIO_BTN", theme.btnFont, L"", 366, 100, 178, 56, theme.btnFontColor, FTGX_JUSTIFY_CENTER | FTGX_ALIGN_MIDDLE, theme.btnTexC); m_downloadBtnPrioM = _addPicButton("DOWNLOAD/PRIO_MINUS", theme.btnTexMinus, theme.btnTexMinusS, 310, 100, 56, 56); m_downloadBtnPrioP = _addPicButton("DOWNLOAD/PRIO_PLUS", theme.btnTexPlus, theme.btnTexPlusS, 544, 100, 56, 56); m_downloadLblRegion = _addLabel("DOWNLOAD/REGION", theme.lblFont, L"", 40, 160, 600, 56, theme.lblFontColor, FTGX_JUSTIFY_LEFT | FTGX_ALIGN_MIDDLE); m_downloadBtnEN = _addPicButton("DOWNLOAD/EN", theme.btnENOff, theme.btnENOffs, 60, 220, 120, 56); m_downloadBtnJA = _addPicButton("DOWNLOAD/JA", theme.btnJAOff, theme.btnJAOffs, 195, 220, 120, 56); m_downloadBtnFR = _addPicButton("DOWNLOAD/FR", theme.btnFROff, theme.btnFROffs, 330, 220, 120, 56); m_downloadBtnDE = _addPicButton("DOWNLOAD/DE", theme.btnDEOff, theme.btnDEOffs, 465, 220, 120, 56); m_downloadBtnES = _addPicButton("DOWNLOAD/ES", theme.btnESOff, theme.btnESOffs, 60, 280, 120, 56); m_downloadBtnIT = _addPicButton("DOWNLOAD/IT", theme.btnITOff, theme.btnITOffs, 195, 280, 120, 56); m_downloadBtnNL = _addPicButton("DOWNLOAD/NL", theme.btnNLOff, theme.btnNLOffs, 330, 280, 120, 56); m_downloadBtnPT = _addPicButton("DOWNLOAD/PT", theme.btnPTOff, theme.btnPTOffs, 465, 280, 120, 56); m_downloadBtnRU = _addPicButton("DOWNLOAD/RU", theme.btnRUOff, theme.btnRUOffs, 60, 340, 120, 56); m_downloadBtnKO = _addPicButton("DOWNLOAD/KO", theme.btnKOOff, theme.btnKOOffs, 195, 340, 120, 56); m_downloadBtnZHCN = _addPicButton("DOWNLOAD/ZHCN", theme.btnZHCNOff, theme.btnZHCNOffs, 330, 340, 120, 56); m_downloadBtnAU = _addPicButton("DOWNLOAD/AU", theme.btnAUOff, theme.btnAUOffs, 465, 340, 120, 56); m_downloadBtnENs = _addPicButton("DOWNLOAD/ENS", theme.btnENOn, theme.btnENOns, 60, 220, 120, 56); m_downloadBtnJAs = _addPicButton("DOWNLOAD/JAS", theme.btnJAOn, theme.btnJAOns, 195, 220, 120, 56); m_downloadBtnFRs = _addPicButton("DOWNLOAD/FRS", theme.btnFROn, theme.btnFROns, 330, 220, 120, 56); m_downloadBtnDEs = _addPicButton("DOWNLOAD/DES", theme.btnDEOn, theme.btnDEOns, 465, 220, 120, 56); m_downloadBtnESs = _addPicButton("DOWNLOAD/ESS", theme.btnESOn, theme.btnESOns, 60, 280, 120, 56); m_downloadBtnITs = _addPicButton("DOWNLOAD/ITS", theme.btnITOn, theme.btnITOns, 195, 280, 120, 56); m_downloadBtnNLs = _addPicButton("DOWNLOAD/NLS", theme.btnNLOn, theme.btnNLOns, 330, 280, 120, 56); m_downloadBtnPTs = _addPicButton("DOWNLOAD/PTS", theme.btnPTOn, theme.btnPTOns, 465, 280, 120, 56); m_downloadBtnRUs = _addPicButton("DOWNLOAD/RUS", theme.btnRUOn, theme.btnRUOns, 60, 340, 120, 56); m_downloadBtnKOs = _addPicButton("DOWNLOAD/KOS", theme.btnKOOn, theme.btnKOOns, 195, 340, 120, 56); m_downloadBtnZHCNs = _addPicButton("DOWNLOAD/ZHCNS", theme.btnZHCNOn, theme.btnZHCNOns, 330, 340, 120, 56); m_downloadBtnAUs = _addPicButton("DOWNLOAD/AUS", theme.btnAUOn, theme.btnAUOns, 465, 340, 120, 56); m_downloadBtnBack = _addButton("DOWNLOAD/BACK_BTN", theme.btnFont, L"", 420, 410, 200, 56, theme.btnFontColor); // Download menu _setHideAnim(m_downloadLblTitle, "DOWNLOAD/TITLE", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadPBar, "DOWNLOAD/PROGRESS_BAR", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblCovers, "DOWNLOAD/COVERS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnCancel, "DOWNLOAD/CANCEL_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnAll, "DOWNLOAD/ALL_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnMissing, "DOWNLOAD/MISSING_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblCoverSet, "DOWNLOAD/COVERSSET", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnCoverSet, "DOWNLOAD/COVERSET_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblGameTDBDownload, "DOWNLOAD/GAMETDB_DOWNLOAD", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnGameTDBDownload, "DOWNLOAD/GAMETDB_DOWNLOAD_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblGameTDB, "DOWNLOAD/GAMETDB", 0, 0, -2.f, 0.f); // Cover settings _setHideAnim(m_downloadLblSetTitle, "DOWNLOAD/SETTITLE", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblCoverPrio, "DOWNLOAD/COVERPRIO", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblPrio, "DOWNLOAD/PRIO_BTN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnPrioM, "DOWNLOAD/PRIO_MINUS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnPrioP, "DOWNLOAD/PRIO_PLUS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadLblRegion, "DOWNLOAD/REGION", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnEN, "DOWNLOAD/EN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnJA, "DOWNLOAD/JA", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnFR, "DOWNLOAD/FR", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnDE, "DOWNLOAD/DE", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnES, "DOWNLOAD/ES", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnIT, "DOWNLOAD/IT", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnNL, "DOWNLOAD/NL", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnPT, "DOWNLOAD/PT", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnRU, "DOWNLOAD/RU", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnKO, "DOWNLOAD/KO", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnZHCN, "DOWNLOAD/ZHCN", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnAU, "DOWNLOAD/AU", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnENs, "DOWNLOAD/ENS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnJAs, "DOWNLOAD/JAS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnFRs, "DOWNLOAD/FRS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnDEs, "DOWNLOAD/DES", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnESs, "DOWNLOAD/ESS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnITs, "DOWNLOAD/ITS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnNLs, "DOWNLOAD/NLS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnPTs, "DOWNLOAD/PTS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnRUs, "DOWNLOAD/RUS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnKOs, "DOWNLOAD/KOS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnZHCNs, "DOWNLOAD/ZHCNS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnAUs, "DOWNLOAD/AUS", 0, 0, -2.f, 0.f); _setHideAnim(m_downloadBtnBack, "DOWNLOAD/BACK_BTN", 0, 0, -2.f, 0.f); m_downloadPrioVal = m_cfg.getUInt("GENERAL", "cover_prio", 0); _hideDownload(true); _textDownload(); } void CMenu::_textDownload(void) { m_btnMgr.setText(m_downloadBtnCancel, _t("dl1", L"Cancel")); m_btnMgr.setText(m_downloadBtnAll, _t("dl3", L"All")); m_btnMgr.setText(m_downloadBtnMissing, _t("dl4", L"Missing")); m_btnMgr.setText(m_downloadLblTitle, _t("dl5", L"Download")); m_btnMgr.setText(m_downloadBtnGameTDBDownload, _t("dl6", L"Download")); m_btnMgr.setText(m_downloadLblCovers, _t("dl8", L"Covers")); m_btnMgr.setText(m_downloadLblGameTDBDownload, _t("dl12", L"GameTDB")); m_btnMgr.setText(m_downloadLblGameTDB, _t("dl10", L"Please donate\nto GameTDB.com")); m_btnMgr.setText(m_downloadLblCoverPrio, _t("dl13", L"Download order")); m_btnMgr.setText(m_downloadLblRegion, _t("dl14", L"Select regions to check for covers:")); m_btnMgr.setText(m_downloadLblCoverSet, _t("dl15", L"Cover download settings")); m_btnMgr.setText(m_downloadBtnCoverSet, _t("dl16", L"Set")); m_btnMgr.setText(m_downloadLblSetTitle, _t("dl17", L"Cover download settings")); m_btnMgr.setText(m_downloadBtnEN, L"EN"); m_btnMgr.setText(m_downloadBtnJA, L"JA"); m_btnMgr.setText(m_downloadBtnFR, L"FR"); m_btnMgr.setText(m_downloadBtnDE, L"DE"); m_btnMgr.setText(m_downloadBtnES, L"ES"); m_btnMgr.setText(m_downloadBtnIT, L"IT"); m_btnMgr.setText(m_downloadBtnNL, L"NL"); m_btnMgr.setText(m_downloadBtnPT, L"PT"); m_btnMgr.setText(m_downloadBtnRU, L"RU"); m_btnMgr.setText(m_downloadBtnKO, L"KO"); m_btnMgr.setText(m_downloadBtnZHCN, L"ZHCN"); m_btnMgr.setText(m_downloadBtnAU, L"AU"); m_btnMgr.setText(m_downloadBtnENs, L"EN"); m_btnMgr.setText(m_downloadBtnJAs, L"JA"); m_btnMgr.setText(m_downloadBtnFRs, L"FR"); m_btnMgr.setText(m_downloadBtnDEs, L"DE"); m_btnMgr.setText(m_downloadBtnESs, L"ES"); m_btnMgr.setText(m_downloadBtnITs, L"IT"); m_btnMgr.setText(m_downloadBtnNLs, L"NL"); m_btnMgr.setText(m_downloadBtnPTs, L"PT"); m_btnMgr.setText(m_downloadBtnRUs, L"RU"); m_btnMgr.setText(m_downloadBtnKOs, L"KO"); m_btnMgr.setText(m_downloadBtnZHCNs, L"ZHCN"); m_btnMgr.setText(m_downloadBtnAUs, L"AU"); m_btnMgr.setText(m_downloadBtnBack, _t("dl18", L"Back")); } s8 CMenu::_versionTxtDownloaderInit(CMenu *m) //Handler to download versions txt file { if (!m->m_thrdWorking) return 0; return m->_versionTxtDownloader(); } s8 CMenu::_versionTxtDownloader() // code to download new version txt file { u32 bufferSize = 0x010000; // Maximum download size 64kb u8 *buffer = (u8*)MEM2_alloc(bufferSize); if(buffer == NULL) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg27", L"Not enough memory"), 1.f); LWP_MutexUnlock(m_mutex); m_thrdWorking = false; return 0; } LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg1", L"Initializing network..."), 0.f); LWP_MutexUnlock(m_mutex); if (_initNetwork() < 0) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg2", L"Network initialization failed!"), 1.f); LWP_MutexUnlock(m_mutex); } else { // DLoad txt file LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg11", L"Downloading..."), 0.2f); LWP_MutexUnlock(m_mutex); m_thrdStep = 0.2f; m_thrdStepLen = 0.9f - 0.2f; gprintf("TXT update URL: %s\n\n", m_cfg.getString("GENERAL", "updatetxturl", UPDATE_URL_VERSION).c_str()); download = downloadfile(buffer, bufferSize, m_cfg.getString("GENERAL", "updatetxturl", UPDATE_URL_VERSION).c_str(),CMenu::_downloadProgress, this); if (download.data == 0 || download.size < 19) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg20", L"No version information found."), 1.f); // TODO: Check for 16 LWP_MutexUnlock(m_mutex); } else { // txt download finished, now save file LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg13", L"Saving..."), 0.9f); LWP_MutexUnlock(m_mutex); bool res = fsop_WriteFile(m_ver.c_str(), download.data, download.size); if (res == true) { // version file valid, check for version with SVN_REV int svnrev = atoi(SVN_REV); gprintf("Installed Version: %d\n", svnrev); m_version.load(m_ver.c_str()); int rev = m_version.getInt("GENERAL", "version", 0); gprintf("Latest available Version: %d\n", rev); if (svnrev < rev) { // new version available LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg19", L"New update available!"), 1.f); LWP_MutexUnlock(m_mutex); } else { // no new version available LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg17", L"No new updates found."), 1.f); LWP_MutexUnlock(m_mutex); } } else { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg15", L"Saving failed!"), 1.f); LWP_MutexUnlock(m_mutex); } } } m_thrdWorking = false; free(buffer); return 0; } s8 CMenu::_versionDownloaderInit(CMenu *m) //Handler to download new dol { if (!m->m_thrdWorking) return 0; return m->_versionDownloader(); } s8 CMenu::_versionDownloader() // code to download new version { bool result = false; char dol_backup[33]; strcpy(dol_backup, m_dol.c_str()); strcat(dol_backup, ".backup"); if (m_app_update_size == 0) m_app_update_size = 0x400000; if (m_data_update_size == 0) m_data_update_size = 0x400000; // check for existing dol ifstream filestr; gprintf("DOL Path: %s\n", m_dol.c_str()); filestr.open(m_dol.c_str()); if (filestr.fail()) { filestr.close(); rename(dol_backup, m_dol.c_str()); filestr.open(m_dol.c_str()); if (filestr.fail()) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg18", L"boot.dol not found at default path!"), 1.f); LWP_MutexUnlock(m_mutex); } filestr.close(); sleep(3); m_thrdWorking = false; return 0; } filestr.close(); u32 bufferSize = max(m_app_update_size, m_data_update_size); // Buffer for size of the biggest file. u8 *buffer = (u8*)MEM2_alloc(bufferSize); if(buffer == NULL) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg27", L"Not enough memory!"), 1.f); LWP_MutexUnlock(m_mutex); sleep(3); m_thrdWorking = false; return 0; } LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg1", L"Initializing network..."), 0.f); LWP_MutexUnlock(m_mutex); if (_initNetwork() < 0) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg2", L"Network initialization failed!"), 1.f); LWP_MutexUnlock(m_mutex); sleep(3); m_thrdWorking = false; free(buffer); return 0; } // Load actual file LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg22", L"Updating application directory..."), 0.2f); LWP_MutexUnlock(m_mutex); m_thrdStep = 0.2f; m_thrdStepLen = 0.9f - 0.2f; gprintf("App Update URL: %s\n", m_app_update_url); gprintf("Data Update URL: %s\n", m_data_update_url); download = downloadfile(buffer, bufferSize, m_app_update_url, CMenu::_downloadProgress, this); if (download.data == 0 || download.size < m_app_update_size) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg12", L"Download failed!"), 1.f); LWP_MutexUnlock(m_mutex); sleep(3); m_thrdWorking = false; free(buffer); return 0; } // download finished, backup boot.dol and write new files. LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg13", L"Saving..."), 0.8f); LWP_MutexUnlock(m_mutex); fsop_deleteFile(dol_backup); rename(m_dol.c_str(), dol_backup); fsop_deleteFile(m_app_update_zip.c_str()); result = fsop_WriteFile(m_app_update_zip.c_str(), download.data, download.size); if (result == true) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg24", L"Extracting..."), 0.8f); LWP_MutexUnlock(m_mutex); ZipFile zFile(m_app_update_zip.c_str()); result = zFile.ExtractAll(m_app_update_drive); fsop_deleteFile(m_app_update_zip.c_str()); if (!result) goto fail; //Update apps dir succeeded, try to update the data dir. download.data = NULL; download.size = 0; memset(buffer, 0, bufferSize); //should we be clearing the buffer of any possible data before downloading? LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg23", L"Updating data directory..."), 0.2f); LWP_MutexUnlock(m_mutex); download = downloadfile(buffer, bufferSize, m_data_update_url, CMenu::_downloadProgress, this); if (download.data == 0 || download.size < m_data_update_size) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg12", L"Download failed!"), 1.f); LWP_MutexUnlock(m_mutex); goto success; } // download finished, write new files. LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg13", L"Saving..."), 0.9f); LWP_MutexUnlock(m_mutex); fsop_deleteFile(m_data_update_zip.c_str()); result = fsop_WriteFile(m_data_update_zip.c_str(), download.data, download.size); if (result == true) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg24", L"Extracting..."), 0.8f); LWP_MutexUnlock(m_mutex); ZipFile zDataFile(m_data_update_zip.c_str()); result = zDataFile.ExtractAll(m_dataDir.c_str()); fsop_deleteFile(m_data_update_zip.c_str()); if (!result) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg15", L"Saving failed!"), 1.f); LWP_MutexUnlock(m_mutex); } } } else goto fail; success: LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg21", L"WiiFlow will now exit to allow the update to take effect."), 1.f); LWP_MutexUnlock(m_mutex); filestr.open(m_dol.c_str()); if (filestr.fail()) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg25", L"Extraction must have failed! Renaming the backup to boot.dol"), 1.f); LWP_MutexUnlock(m_mutex); rename(dol_backup, m_dol.c_str()); } else fsop_deleteFile(dol_backup); filestr.close(); m_exit = true; goto out; fail: rename(dol_backup, m_dol.c_str()); LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg15", L"Saving failed!"), 1.f); LWP_MutexUnlock(m_mutex); out: free(buffer); sleep(3); m_thrdWorking = false; return 0; } int CMenu::_gametdbDownloader(CMenu *m) { if (!m->m_thrdWorking) return 0; return m->_gametdbDownloaderAsync(); } int CMenu::_gametdbDownloaderAsync() { u32 bufferSize = 0x800000; // 8 MB u8 *buffer = (u8*)MEM2_alloc(bufferSize); if (buffer == NULL) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg27", L"Not enough memory"), 1.f); LWP_MutexUnlock(m_mutex); return 0; } const string &langCode = m_loc.getString(m_curLanguage, "gametdb_code", "EN"); LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg1", L"Initializing network..."), 0.f); LWP_MutexUnlock(m_mutex); if (_initNetwork() < 0) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg2", L"Network initialization failed!"), 1.f); LWP_MutexUnlock(m_mutex); } else { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg11", L"Downloading..."), 0.0f); LWP_MutexUnlock(m_mutex); m_thrdStep = 0.0f; m_thrdStepLen = 1.0f; download = downloadfile(buffer, bufferSize, fmt(GAMETDB_URL, langCode.c_str()), CMenu::_downloadProgress, this); if (download.data == 0) { LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg12", L"Download failed!"), 1.f); LWP_MutexUnlock(m_mutex); } else { bool res = false; char *zippath = fmt_malloc("%s/wiitdb.zip", m_settingsDir.c_str()); if(zippath != NULL) { gprintf("Writing file to '%s'\n", zippath); fsop_deleteFile(zippath); _setThrdMsg(wfmt(_fmt("dlmsg4", L"Saving %s"), "wiitdb.zip"), 1.f); res = fsop_WriteFile(zippath, download.data, download.size); } if(res == false) { gprintf("Can't save zip file\n"); if(zippath != NULL) MEM2_free(zippath); LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg15", L"Couldn't save ZIP file"), 1.f); LWP_MutexUnlock(m_mutex); } else { gprintf("Extracting zip file: "); ZipFile zFile(zippath); bool zres = zFile.ExtractAll(m_settingsDir.c_str()); gprintf(zres ? "success\n" : "failed\n"); // We don't need the zipfile anymore fsop_deleteFile(zippath); MEM2_free(zippath); // We should always remove the offsets file to make sure it's reloaded fsop_deleteFile(fmt("%s/gametdb_offsets.bin", m_settingsDir.c_str())); // Update cache //m_gameList.SetLanguage(m_loc.getString(m_curLanguage, "gametdb_code", "EN").c_str()); UpdateCache(); LWP_MutexLock(m_mutex); _setThrdMsg(_t("dlmsg26", L"Updating cache..."), 0.f); LWP_MutexUnlock(m_mutex); m_GameTDBLoaded = true; _loadList(); _initCF(); } } } free(buffer); m_thrdWorking = false; return 0; } const char *banner_url = NULL; const char *banner_url_id3 = NULL; char *banner_location = NULL; u32 CMenu::_downloadBannerAsync(void *obj) { CMenu *m = (CMenu *)obj; if (!m->m_thrdWorking) return 0; m->m_thrdStop = false; LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("cfgbnr7", L"Downloading banner..."), 0); LWP_MutexUnlock(m->m_mutex); if (m->_initNetwork() < 0) { LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("dlmsg2", L"Network initialization failed!"), 1.f); LWP_MutexUnlock(m->m_mutex); m->m_thrdWorking = false; return -1; } u32 bufferSize = 0x400000; /* 4mb max */ u8 *buffer = (u8*)MEM2_alloc(bufferSize); if(buffer == NULL) { LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("dlmsg27", L"Not enough memory!"), 1.f); LWP_MutexUnlock(m->m_mutex); m->m_thrdWorking = false; return -2; } block banner = downloadfile(buffer, bufferSize, banner_url, CMenu::_downloadProgress, m); if(banner.data == NULL || banner.size < 0x5000) banner = downloadfile(buffer, bufferSize, banner_url_id3, CMenu::_downloadProgress, m); /* minimum 50kb */ if (banner.data != NULL && banner.size > 51200 && banner.data[0] != '<') { if(banner_location != NULL) fsop_WriteFile(banner_location, banner.data, banner.size); LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("dlmsg14", L"Done."), 1.f); LWP_MutexUnlock(m->m_mutex); free(buffer); m->m_thrdWorking = false; return 0; } else { LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("dlmsg12", L"Download failed!"), 1.f); LWP_MutexUnlock(m->m_mutex); m->m_thrdWorking = false; free(buffer); return -3; } } static const char *GAME_BNR_ID = "{gameid}"; void CMenu::_downloadBnr(const char *gameID) { if(gameID == NULL || strlen(gameID) > 6) return; string base_url = m_cfg.getString("GENERAL", "custom_banner_url", CUSTOM_BANNER_URL); if(base_url.size() < 3 || base_url.find(GAME_BNR_ID) == string::npos) return; base_url.replace(base_url.find(GAME_BNR_ID), strlen(GAME_BNR_ID), gameID); banner_url = base_url.c_str(); string base_url_id3 = m_cfg.getString("GENERAL", "custom_banner_url", CUSTOM_BANNER_URL); base_url_id3.replace(base_url_id3.find(GAME_BNR_ID), strlen(GAME_BNR_ID), gameID, 3); banner_url_id3 = base_url_id3.c_str(); banner_location = fmt_malloc("%s/%s.bnr", m_customBnrDir.c_str(), gameID); if(banner_location == NULL) return; m_btnMgr.show(m_downloadPBar); m_btnMgr.setProgress(m_downloadPBar, 0.f); m_btnMgr.show(m_downloadBtnCancel); m_btnMgr.setText(m_downloadBtnCancel, _t("dl1", L"Cancel")); m_thrdStop = false; m_thrdMessageAdded = false; m_thrdWorking = true; lwp_t thread = LWP_THREAD_NULL; LWP_CreateThread(&thread, (void *(*)(void *))CMenu::_downloadBannerAsync, (void *)this, 0, 8192, 40); wstringEx prevMsg; while(m_thrdWorking) { _mainLoopCommon(); if ((BTN_HOME_PRESSED || BTN_B_PRESSED) && !m_thrdWorking) break; if (BTN_A_PRESSED && !(m_thrdWorking && m_thrdStop)) { if (m_btnMgr.selected(m_downloadBtnCancel)) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); } } if (Sys_Exiting()) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); m_thrdWorking = false; } if (m_thrdMessageAdded) { LockMutex lock(m_mutex); m_thrdMessageAdded = false; m_btnMgr.setProgress(m_downloadPBar, m_thrdProgress); if (prevMsg != m_thrdMessage) { prevMsg = m_thrdMessage; m_btnMgr.setText(m_downloadLblMessage[0], m_thrdMessage, false); m_btnMgr.hide(m_downloadLblMessage[0], 0, 0, -1.f, -1.f, true); m_btnMgr.show(m_downloadLblMessage[0]); } } if (m_thrdStop && !m_thrdWorking) break; } if(banner_location != NULL) { MEM2_free(banner_location); banner_location = NULL; } m_btnMgr.setText(m_downloadBtnCancel, _t("gm2", L"Back")); if (thread != LWP_THREAD_NULL) { LWP_JoinThread(thread, NULL); thread = LWP_THREAD_NULL; } while(!m_exit && !m_thrdStop) { _mainLoopCommon(); if(BTN_HOME_PRESSED || BTN_B_PRESSED || (BTN_A_PRESSED && m_btnMgr.selected(m_downloadBtnCancel))) break; } m_btnMgr.hide(m_downloadLblMessage[0]); m_btnMgr.hide(m_downloadPBar); m_btnMgr.hide(m_downloadBtnCancel); } const char *url_dl = NULL; void CMenu::_downloadUrl(const char *url, u8 **dl_file, u32 *dl_size) { m_file = NULL; m_filesize = 0; url_dl = url; m_btnMgr.show(m_downloadPBar); m_btnMgr.setProgress(m_downloadPBar, 0.f); m_btnMgr.show(m_downloadBtnCancel); m_btnMgr.setText(m_downloadBtnCancel, _t("dl1", L"Cancel")); m_thrdStop = false; m_thrdMessageAdded = false; m_thrdWorking = true; lwp_t thread = LWP_THREAD_NULL; LWP_CreateThread(&thread, (void *(*)(void *))CMenu::_downloadUrlAsync, (void *)this, 0, 8192, 40); wstringEx prevMsg; while(m_thrdWorking) { _mainLoopCommon(); if ((BTN_HOME_PRESSED || BTN_B_PRESSED) && !m_thrdWorking) break; if (BTN_A_PRESSED && !(m_thrdWorking && m_thrdStop)) { if (m_btnMgr.selected(m_downloadBtnCancel)) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); } } if (Sys_Exiting()) { LockMutex lock(m_mutex); m_thrdStop = true; m_thrdMessageAdded = true; m_thrdMessage = _t("dlmsg6", L"Canceling..."); m_thrdWorking = false; } if (m_thrdMessageAdded) { LockMutex lock(m_mutex); m_thrdMessageAdded = false; m_btnMgr.setProgress(m_downloadPBar, m_thrdProgress); if (prevMsg != m_thrdMessage) { prevMsg = m_thrdMessage; m_btnMgr.setText(m_downloadLblMessage[0], m_thrdMessage, false); m_btnMgr.hide(m_downloadLblMessage[0], 0, 0, -1.f, -1.f, true); m_btnMgr.show(m_downloadLblMessage[0]); } } if (m_thrdStop && !m_thrdWorking) break; } if (thread != LWP_THREAD_NULL) { LWP_JoinThread(thread, NULL); thread = LWP_THREAD_NULL; } m_btnMgr.hide(m_downloadLblMessage[0]); m_btnMgr.hide(m_downloadPBar); m_btnMgr.hide(m_downloadBtnCancel); *dl_file = m_file; *dl_size = m_filesize; m_file = NULL; m_filesize = 0; url_dl = url; } u32 CMenu::_downloadUrlAsync(void *obj) { CMenu *m = (CMenu *)obj; if (!m->m_thrdWorking) return 0; m->m_thrdStop = false; LWP_MutexLock(m->m_mutex); m->_setThrdMsg(m->_t("dlmsg11", L"Downloading..."), 0); LWP_MutexUnlock(m->m_mutex); if(m->_initNetwork() < 0 || url_dl == NULL) { m->m_thrdWorking = false; return -1; } u32 bufferSize = 0x400000; /* 4mb max */ m->m_buffer = (u8*)MEM2_alloc(bufferSize); if(m->m_buffer == NULL) { m->m_thrdWorking = false; return -2; } block file = downloadfile(m->m_buffer, bufferSize, url_dl, CMenu::_downloadProgress, m); DCFlushRange(m->m_buffer, bufferSize); m->m_file = file.data; m->m_filesize = file.size; m->m_thrdWorking = false; return 0; }