mirror of
https://github.com/wiidev/usbloadergx.git
synced 2024-11-04 18:45:05 +01:00
509 lines
12 KiB
C++
509 lines
12 KiB
C++
/****************************************************************************
|
|
* Copyright (C) 2011 Dimok
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
****************************************************************************/
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
#include <sys/stat.h>
|
|
#include "GCGames.h"
|
|
#include "FileOperations/fileops.h"
|
|
#include "settings/GameTitles.h"
|
|
#include "settings/CSettings.h"
|
|
#include "prompts/GCDeleteMenu.h"
|
|
#include "prompts/PromptWindows.h"
|
|
#include "prompts/ProgressWindow.h"
|
|
#include "language/gettext.h"
|
|
#include "usbloader/wbfs/wbfs_fat.h"
|
|
#include "utils/tools.h"
|
|
#include "menu.h"
|
|
#include "gecko.h"
|
|
|
|
GCGames *GCGames::instance = NULL;
|
|
|
|
inline bool isGameID(const u8 *id)
|
|
{
|
|
for (int i = 0; i < 6; i++)
|
|
if (!isalnum((int) id[i]))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
const char *GCGames::GetPath(const char *gameID) const
|
|
{
|
|
if(!gameID)
|
|
return "";
|
|
|
|
for(u32 i = 0; i < HeaderList.size(); i++)
|
|
{
|
|
if(strncasecmp((const char *) HeaderList[i].id, gameID, 6) == 0)
|
|
return PathList[i].c_str();
|
|
}
|
|
|
|
return "";
|
|
}
|
|
|
|
void GCGames::LoadGameList(const string &path, vector<struct discHdr> &headerList, vector<string> &pathList)
|
|
{
|
|
struct discHdr tmpHdr;
|
|
struct stat st;
|
|
u8 id[8];
|
|
char fpath[1024];
|
|
char fname_title[64];
|
|
DIR *dir_iter;
|
|
struct dirent *dirent;
|
|
|
|
dir_iter = opendir(path.c_str());
|
|
if (!dir_iter) return;
|
|
|
|
while ((dirent = readdir(dir_iter)) != 0)
|
|
{
|
|
const char *dirname = dirent->d_name;
|
|
if(!dirname)
|
|
continue;
|
|
|
|
if (dirname[0] == '.') continue;
|
|
|
|
// reset id and title
|
|
memset(id, 0, sizeof(id));
|
|
*fname_title = 0;
|
|
|
|
bool lay_a = false;
|
|
bool lay_b = false;
|
|
int len = strlen(dirname);
|
|
if (len >= 8)
|
|
{
|
|
if (Wbfs_Fat::CheckLayoutB((char *) dirname, len, id, fname_title))
|
|
{
|
|
// path/TITLE[GAMEID]/game.iso
|
|
lay_b = true;
|
|
}
|
|
else if (dirname[6] == '_')
|
|
{
|
|
// path/GAMEID_TITLE/game.iso
|
|
memcpy(id, dirname, 6);
|
|
|
|
if(isGameID(id))
|
|
{
|
|
lay_a = true;
|
|
snprintf(fname_title, sizeof(fname_title), "%s", &dirname[7]);
|
|
}
|
|
}
|
|
}
|
|
else if(len == 6 && isGameID((u8*)dirname)) {
|
|
memcpy(id, dirname, 6);
|
|
lay_a = true;
|
|
}
|
|
|
|
if(!lay_a && !lay_b)
|
|
memset(id, 0, sizeof(id));
|
|
|
|
bool found = false;
|
|
bool extracted = false;
|
|
|
|
for(int i = 0; i < 4; i++)
|
|
{
|
|
char name[50];
|
|
snprintf(name, sizeof(name), "%.6s.%s", (i % 2) == 0 ? "game" : (char *) id, i >= 2 ? "gcm" : "iso");
|
|
snprintf(fpath, sizeof(fpath), "%s%s/%s", path.c_str(), dirname, name);
|
|
if((found = (stat(fpath, &st) == 0)) == true)
|
|
break;
|
|
}
|
|
|
|
if(!found)
|
|
{
|
|
snprintf(fpath, sizeof(fpath), "%s%s/sys/boot.bin", path.c_str(), dirname);
|
|
if(stat(fpath, &st) != 0)
|
|
continue;
|
|
// this game is extracted
|
|
extracted = true;
|
|
}
|
|
|
|
//! GAMEID was not found
|
|
if(!lay_a && !lay_b) {
|
|
// read game ID and title from disc header
|
|
// iso file
|
|
FILE *fp = fopen(fpath, "rb");
|
|
if (fp != NULL)
|
|
{
|
|
memset(&tmpHdr, 0, sizeof(tmpHdr));
|
|
fread(&tmpHdr, sizeof(struct discHdr), 1, fp);
|
|
fclose(fp);
|
|
|
|
if (tmpHdr.gc_magic == GCGames::MAGIC)
|
|
{
|
|
memcpy(id, tmpHdr.id, 6);
|
|
snprintf(fname_title, sizeof(fname_title), "%s", tmpHdr.title);
|
|
}
|
|
}
|
|
}
|
|
|
|
// if we have titles.txt entry use that
|
|
const char *title = GameTitles.GetTitle(id);
|
|
|
|
// if no titles.txt get title from dir or file name
|
|
if (strlen(title) == 0 && !Settings.ForceDiscTitles && strlen(fname_title) > 0)
|
|
title = fname_title;
|
|
|
|
if (*id != 0 && strlen(title) > 0)
|
|
{
|
|
string gamePath = string(path) + dirname + (extracted ? "/" : strrchr(fpath, '/'));
|
|
memset(&tmpHdr, 0, sizeof(tmpHdr));
|
|
memcpy(tmpHdr.id, id, 6);
|
|
strncpy(tmpHdr.title, title, sizeof(tmpHdr.title)-1);
|
|
tmpHdr.magic = GCGames::MAGIC;
|
|
tmpHdr.type = extracted ? TYPE_GAME_GC_EXTRACTED : TYPE_GAME_GC_IMG;
|
|
headerList.push_back(tmpHdr);
|
|
pathList.push_back(gamePath);
|
|
continue;
|
|
}
|
|
|
|
// else read it from file directly
|
|
// iso file
|
|
FILE *fp = fopen(fpath, "rb");
|
|
if (fp != NULL)
|
|
{
|
|
memset(&tmpHdr, 0, sizeof(tmpHdr));
|
|
fread(&tmpHdr, sizeof(struct discHdr), 1, fp);
|
|
fclose(fp);
|
|
|
|
if (tmpHdr.gc_magic == GCGames::MAGIC)
|
|
{
|
|
string gamePath = string(path) + dirname + (extracted ? "/" : strrchr(fpath, '/'));
|
|
tmpHdr.magic = tmpHdr.gc_magic;
|
|
tmpHdr.type = extracted ? TYPE_GAME_GC_EXTRACTED : TYPE_GAME_GC_IMG;
|
|
headerList.push_back(tmpHdr);
|
|
pathList.push_back(gamePath);
|
|
|
|
// Save title for next start
|
|
GameTitles.SetGameTitle(tmpHdr.id, tmpHdr.title);
|
|
}
|
|
}
|
|
}
|
|
|
|
closedir(dir_iter);
|
|
}
|
|
|
|
u32 GCGames::LoadAllGames(void)
|
|
{
|
|
PathList.clear();
|
|
HeaderList.clear();
|
|
sdGCList.clear();
|
|
sdGCPathList.clear();
|
|
|
|
LoadGameList(Settings.GameCubePath, HeaderList, PathList);
|
|
|
|
if(strcmp(Settings.GameCubePath, Settings.GameCubeSDPath) != 0)
|
|
{
|
|
LoadGameList(Settings.GameCubeSDPath, sdGCList, sdGCPathList);
|
|
|
|
for(u32 i = 0; i < sdGCList.size(); ++i)
|
|
{
|
|
u32 n;
|
|
for(n = 0; n < HeaderList.size(); ++n)
|
|
{
|
|
//! replace the one loaded from USB with the same games on SD since we can load them directly
|
|
if(memcmp(HeaderList[n].id, sdGCList[i].id, 6) == 0)
|
|
{
|
|
memcpy(&HeaderList[n], &sdGCList[i], sizeof(struct discHdr));
|
|
PathList[n] = sdGCPathList[i];
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Not available in the main GC path
|
|
if(n == HeaderList.size()) {
|
|
HeaderList.push_back(sdGCList[i]);
|
|
PathList.push_back(sdGCPathList[i]);
|
|
}
|
|
}
|
|
}
|
|
|
|
return HeaderList.size();
|
|
}
|
|
|
|
bool GCGames::RemoveGame(const char *gameID)
|
|
{
|
|
const char *path = GetPath(gameID);
|
|
if(*path == 0)
|
|
return false;
|
|
|
|
RemoveSDGame(gameID);
|
|
|
|
if(strcmp(Settings.GameCubePath, Settings.GameCubeSDPath) == 0)
|
|
return true;
|
|
|
|
struct discHdr *header = NULL;
|
|
for(u32 i = 0; i < HeaderList.size(); ++i)
|
|
{
|
|
if(strncmp(gameID, (char*)HeaderList[i].id, 6) == 0)
|
|
{
|
|
header = &HeaderList[i];
|
|
break;
|
|
}
|
|
}
|
|
if(!header)
|
|
return false;
|
|
|
|
char filepath[512];
|
|
int result = 0;
|
|
int ret;
|
|
|
|
// the main path is the SD path as it is prefered, now delete USB
|
|
char cIsoPath[256];
|
|
snprintf(cIsoPath, sizeof(cIsoPath), "%s", path + strlen(Settings.GameCubeSDPath));
|
|
|
|
if(header->type == TYPE_GAME_GC_IMG)
|
|
{
|
|
// Remove game iso
|
|
snprintf(filepath, sizeof(filepath), "%s%s", Settings.GameCubePath, cIsoPath);
|
|
ret = RemoveFile(filepath);
|
|
if(ret != 0)
|
|
result = -1;
|
|
|
|
// Remove path
|
|
char *pathPtr = strrchr(filepath, '/');
|
|
if(pathPtr) *pathPtr = 0;
|
|
ret = RemoveFile(filepath);
|
|
if(ret != 0)
|
|
result = -1;
|
|
}
|
|
else if(header->type == TYPE_GAME_GC_EXTRACTED)
|
|
{
|
|
//! remove extracted gamecube game
|
|
snprintf(filepath, sizeof(filepath), "%s%s", Settings.GameCubePath, cIsoPath);
|
|
ret = RemoveDirectory(path);
|
|
if(ret < 0)
|
|
result = -1;
|
|
}
|
|
|
|
return (result == 0);
|
|
}
|
|
|
|
bool GCGames::RemoveSDGame(const char *gameID)
|
|
{
|
|
const char *path = GetPath(gameID);
|
|
if(*path == 0)
|
|
return false;
|
|
|
|
struct discHdr *header = NULL;
|
|
for(u32 i = 0; i < HeaderList.size(); ++i)
|
|
{
|
|
if(strncmp(gameID, (char*)HeaderList[i].id, 6) == 0)
|
|
{
|
|
header = &HeaderList[i];
|
|
break;
|
|
}
|
|
}
|
|
if(!header)
|
|
return false;
|
|
|
|
char filepath[512];
|
|
int result = 0;
|
|
int ret;
|
|
|
|
if(header->type == TYPE_GAME_GC_IMG)
|
|
{
|
|
// Remove game iso
|
|
snprintf(filepath, sizeof(filepath), "%s", path);
|
|
ret = RemoveFile(filepath);
|
|
if(ret != 0)
|
|
result = -1;
|
|
|
|
// Remove path
|
|
char *pathPtr = strrchr(filepath, '/');
|
|
if(pathPtr) *pathPtr = 0;
|
|
ret = RemoveFile(filepath);
|
|
if(ret != 0)
|
|
result = -1;
|
|
}
|
|
else if(header->type == TYPE_GAME_GC_EXTRACTED)
|
|
{
|
|
//! remove extracted gamecube game
|
|
ret = RemoveDirectory(path);
|
|
if(ret < 0)
|
|
result = -1;
|
|
}
|
|
|
|
return (result == 0);
|
|
}
|
|
|
|
float GCGames::GetGameSize(const char *gameID)
|
|
{
|
|
const char *path = GetPath(gameID);
|
|
if(*path == 0)
|
|
return 0.0f;
|
|
|
|
struct stat st;
|
|
|
|
if(stat(path, &st) != 0)
|
|
return 0.0f;
|
|
|
|
return ((float) st.st_size / GB_SIZE);
|
|
}
|
|
|
|
bool GCGames::IsInstalled(const char *gameID) const
|
|
{
|
|
for(u32 n = 0; n < HeaderList.size(); n++)
|
|
{
|
|
if(memcmp(HeaderList[n].id, gameID, 6) == 0)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool GCGames::CopyUSB2SD(const struct discHdr *header)
|
|
{
|
|
const char *path = GetPath((char*)header->id);
|
|
if(*path == 0)
|
|
return false;
|
|
|
|
int choice = WindowPrompt(tr("The game is on USB."), tr("Do you want to copy the game to SD or delete a game on SD?"), tr("Copy"), tr("Show SD"), ("Launch from USB"), tr("Cancel"));
|
|
if(choice == 0)
|
|
return false;
|
|
|
|
const char *cpTitle = GameTitles.GetTitle(header);
|
|
|
|
if(choice == 3)
|
|
return true;
|
|
|
|
if(choice == 2)
|
|
{
|
|
GCDeleteMenu gcDeleteMenu;
|
|
gcDeleteMenu.SetAlignment(ALIGN_CENTER, ALIGN_MIDDLE);
|
|
gcDeleteMenu.SetEffect(EFFECT_FADE, 20);
|
|
mainWindow->SetState(STATE_DISABLED);
|
|
mainWindow->Append(&gcDeleteMenu);
|
|
|
|
gcDeleteMenu.Show();
|
|
|
|
gcDeleteMenu.SetEffect(EFFECT_FADE, -20);
|
|
while(gcDeleteMenu.GetEffect() > 0) usleep(1000);
|
|
|
|
mainWindow->Remove(&gcDeleteMenu);
|
|
mainWindow->SetState(STATE_DEFAULT);
|
|
|
|
if(!WindowPrompt(tr("Do you want to copy now?"), cpTitle, tr("Yes"), tr("Cancel")))
|
|
return false;
|
|
}
|
|
|
|
struct statvfs sd_vfs;
|
|
if(statvfs(Settings.GameCubeSDPath, &sd_vfs) != 0)
|
|
{
|
|
WindowPrompt(tr("Error:"), tr("SD Card could not be accessed."), tr("OK"));
|
|
return false;
|
|
}
|
|
|
|
u64 filesize = 0;
|
|
|
|
if(header->type == TYPE_GAME_GC_IMG) {
|
|
filesize = FileSize(path);
|
|
}
|
|
else if(header->type == TYPE_GAME_GC_EXTRACTED) {
|
|
StartProgress(tr("Getting game folder size..."), tr("Please wait"), 0, true, true);
|
|
ShowProgress(0, 1);
|
|
filesize = FileSize(path);
|
|
ProgressStop();
|
|
}
|
|
|
|
while(((u64)sd_vfs.f_frsize * (u64)sd_vfs.f_bfree) < filesize)
|
|
{
|
|
choice = WindowPrompt(tr("Error: Not enough space on SD."), tr("Do you want to delete a game on SD?"), tr("Yes"), tr("Cancel"));
|
|
if(choice == 0)
|
|
return false;
|
|
|
|
GCDeleteMenu gcDeleteMenu;
|
|
gcDeleteMenu.SetAlignment(ALIGN_CENTER, ALIGN_MIDDLE);
|
|
gcDeleteMenu.SetEffect(EFFECT_FADE, 20);
|
|
mainWindow->SetState(STATE_DISABLED);
|
|
mainWindow->Append(&gcDeleteMenu);
|
|
|
|
gcDeleteMenu.Show();
|
|
|
|
gcDeleteMenu.SetEffect(EFFECT_FADE, -20);
|
|
while(gcDeleteMenu.GetEffect() > 0) usleep(1000);
|
|
|
|
mainWindow->Remove(&gcDeleteMenu);
|
|
mainWindow->SetState(STATE_DEFAULT);
|
|
|
|
statvfs(Settings.GameCubeSDPath, &sd_vfs);
|
|
}
|
|
|
|
const char *cIsoPath = path + strlen(Settings.GameCubePath);
|
|
char destPath[512];
|
|
snprintf(destPath, sizeof(destPath), "%s%s", Settings.GameCubeSDPath, cIsoPath);
|
|
|
|
int res = -1;
|
|
|
|
if(header->type == TYPE_GAME_GC_IMG)
|
|
{
|
|
ProgressCancelEnable(true);
|
|
StartProgress(tr("Copying GC game..."), cpTitle, 0, true, true);
|
|
|
|
char *ptr = strrchr(destPath, '/');
|
|
if(ptr) *ptr = 0;
|
|
|
|
CreateSubfolder(destPath);
|
|
|
|
snprintf(destPath, sizeof(destPath), "%s%s", Settings.GameCubeSDPath, cIsoPath);
|
|
|
|
res = CopyFile(path, destPath);
|
|
}
|
|
else if(header->type == TYPE_GAME_GC_EXTRACTED)
|
|
{
|
|
res = CopyDirectory(path, destPath);
|
|
}
|
|
|
|
// Refresh list
|
|
GCGames::Instance()->LoadAllGames();
|
|
|
|
ProgressStop();
|
|
ProgressCancelEnable(false);
|
|
|
|
if(res == PROGRESS_CANCELED)
|
|
{
|
|
if(header->type == TYPE_GAME_GC_IMG)
|
|
{
|
|
// remove file and path
|
|
RemoveFile(destPath);
|
|
char *ptr = strrchr(destPath, '/');
|
|
if(ptr) *ptr = 0;
|
|
RemoveFile(destPath);
|
|
}
|
|
|
|
WindowPrompt(tr("Copying Canceled"), 0, tr("OK"));
|
|
return false;
|
|
}
|
|
else if(res < 0)
|
|
{
|
|
if(header->type == TYPE_GAME_GC_IMG)
|
|
{
|
|
// remove file and path
|
|
RemoveFile(destPath);
|
|
char *ptr = strrchr(destPath, '/');
|
|
if(ptr) *ptr = 0;
|
|
RemoveFile(destPath);
|
|
}
|
|
|
|
WindowPrompt(tr("Error:"), tr("Failed copying file"), tr("OK"));
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
return WindowPrompt(tr("Successfully copied"), tr("Do you want to start the game now?"), tr("Yes"), tr("Cancel"));
|
|
}
|
|
}
|
|
|