GodMode9i/arm9/source/driveOperations.cpp
2020-02-04 21:40:19 -07:00

325 lines
7.8 KiB
C++

#include <nds.h>
#include <nds/arm9/dldi.h>
#include <fat.h>
#include <sys/stat.h>
#include <sys/statvfs.h>
#include <stdio.h>
#include "main.h"
#include "dldi-include.h"
#include "lzss.h"
#include "ramd.h"
#include "ramdrive-include.h"
#include "tonccpy.h"
static sNDSHeader nds;
u8 stored_SCFG_MC = 0;
static bool slot1Enabled = true;
bool sdMounted = false;
bool sdMountedDone = false; // true if SD mount is successful once
bool flashcardMounted = false;
bool ramdrive1Mounted = false;
bool ramdrive2Mounted = false;
bool nitroMounted = false;
int currentDrive = 0; // 0 == SD card, 1 == Flashcard, 2 == RAMdrive 1, 3 == RAMdrive 2
int nitroCurrentDrive = 0;
char sdLabel[12];
char fatLabel[12];
u64 sdSize = 0;
u64 fatSize = 0;
static int getGbNumber(u64 bytes) {
int gbNumber = 0;
for (u64 i = 0; i <= bytes; i += 0x40000000) {
gbNumber++;
}
return gbNumber;
}
static int getTbNumber(u64 bytes) {
int tbNumber = 0;
for (u64 i = 0; i <= bytes; i += 0x10000000000) {
tbNumber++;
}
return tbNumber;
}
void printDriveBytes(u64 bytes)
{
if (bytes == 1)
iprintf("%d Byte", (int)bytes);
else if (bytes >= 0 && bytes < 1024)
iprintf("%d Bytes", (int)bytes);
else if (bytes >= 1024 && bytes < (1024 * 1024))
printf("%d KB", (int)bytes / 1024);
else if (bytes >= (1024 * 1024) && bytes < (1024 * 1024 * 1024))
printf("%d MB", (int)bytes / 1024 / 1024);
else if (bytes >= 0x40000000 && bytes < 0x10000000000)
printf("%d GB", getGbNumber(bytes));
else
printf("%d TB", getTbNumber(bytes));
}
const char* getDrivePath(void) {
switch (currentDrive) {
case 0:
return "sd:/";
case 1:
return "fat:/";
case 2:
return "ram1:/";
case 3:
return "ram2:/";
}
return "";
}
void fixLabel(bool fat) {
if (fat) {
for (int i = 0; i < 12; i++) {
if (((fatLabel[i] == ' ') && (fatLabel[i+1] == ' ') && (fatLabel[i+2] == ' '))
|| ((fatLabel[i] == ' ') && (fatLabel[i+1] == ' '))
|| (fatLabel[i] == ' ')) {
fatLabel[i] = '\0';
break;
}
}
} else {
for (int i = 0; i < 12; i++) {
if (((sdLabel[i] == ' ') && (sdLabel[i+1] == ' ') && (sdLabel[i+2] == ' '))
|| ((sdLabel[i] == ' ') && (sdLabel[i+1] == ' '))
|| (sdLabel[i] == ' ')) {
sdLabel[i] = '\0';
break;
}
}
}
}
bool sdFound(void) {
if (access("sd:/", F_OK) == 0) {
return true;
} else {
return false;
}
}
bool flashcardFound(void) {
if (access("fat:/", F_OK) == 0) {
return true;
} else {
return false;
}
}
bool bothSDandFlashcard(void) {
if (sdMounted && flashcardMounted) {
return true;
} else {
return false;
}
}
TWL_CODE bool sdMount(void) {
fatMountSimple("sd", get_io_dsisd());
if (sdFound()) {
sdMountedDone = true;
fatGetVolumeLabel("sd", sdLabel);
fixLabel(false);
struct statvfs st;
if (statvfs("sd:/", &st) == 0) {
sdSize = st.f_bsize * st.f_blocks;
}
return true;
}
return false;
}
TWL_CODE void sdUnmount(void) {
fatUnmount("sd");
sdLabel[0] = '\0';
sdSize = 0;
sdMounted = false;
}
TWL_CODE DLDI_INTERFACE* dldiLoadFromBin (const u8 dldiAddr[]) {
// Check that it is a valid DLDI
if (!dldiIsValid ((DLDI_INTERFACE*)dldiAddr)) {
return NULL;
}
DLDI_INTERFACE* device = (DLDI_INTERFACE*)dldiAddr;
size_t dldiSize;
// Calculate actual size of DLDI
// Although the file may only go to the dldiEnd, the BSS section can extend past that
if (device->dldiEnd > device->bssEnd) {
dldiSize = (char*)device->dldiEnd - (char*)device->dldiStart;
} else {
dldiSize = (char*)device->bssEnd - (char*)device->dldiStart;
}
dldiSize = (dldiSize + 0x03) & ~0x03; // Round up to nearest integer multiple
// Clear unused space
toncset(device+dldiSize, 0, 0x4000-dldiSize);
dldiFixDriverAddresses (device);
if (device->ioInterface.features & FEATURE_SLOT_GBA) {
sysSetCartOwner(BUS_OWNER_ARM9);
}
if (device->ioInterface.features & FEATURE_SLOT_NDS) {
sysSetCardOwner(BUS_OWNER_ARM9);
}
return device;
}
TWL_CODE bool UpdateCardInfo(char* gameid, char* gamename) {
cardReadHeader((uint8*)0x02000000);
tonccpy(&nds, (void*)0x02000000, sizeof(sNDSHeader));
tonccpy(gameid, &nds.gameCode, 4);
gameid[4] = 0x00;
tonccpy(gamename, &nds.gameTitle, 12);
gamename[12] = 0x00;
return true;
}
TWL_CODE void ShowGameInfo(const char gameid[], const char gamename[]) {
iprintf("Game id: %s\nName: %s", gameid, gamename);
}
TWL_CODE bool twl_flashcardMount(void) {
if (REG_SCFG_MC != 0x11) {
sysSetCardOwner (BUS_OWNER_ARM9);
// Reset Slot-1 to allow reading title name and ID
if (slot1Enabled) {
disableSlot1();
for(int i = 0; i < 25; i++) { swiWaitForVBlank(); }
slot1Enabled = false;
}
if (appInited) {
for(int i = 0; i < 35; i++) { swiWaitForVBlank(); } // Make sure cart is inserted correctly
}
if (REG_SCFG_MC == 0x11) {
sysSetCardOwner (BUS_OWNER_ARM7);
return false;
}
if (!slot1Enabled) {
enableSlot1();
for(int i = 0; i < 15; i++) { swiWaitForVBlank(); }
slot1Enabled = true;
}
nds.gameCode[0] = 0;
nds.gameTitle[0] = 0;
char gamename[13];
char gameid[5];
/*fifoSendValue32(FIFO_USER_04, 1);
for (int i = 0; i < 10; i++) {
swiWaitForVBlank();
}
tonccpy(&nds, (void*)0x02000000, sizeof(nds));*/
UpdateCardInfo(&gameid[0], &gamename[0]);
/*consoleClear();
iprintf("REG_SCFG_MC: %x\n", REG_SCFG_MC);
ShowGameInfo(gameid, gamename);
for (int i = 0; i < 60*5; i++) {
swiWaitForVBlank();
}*/
sysSetCardOwner (BUS_OWNER_ARM7); // 3DS fix
if (gameid[0] >= 0x00 && gameid[0] < 0x20) {
return false;
}
// Read a DLDI driver specific to the cart
if (!memcmp(gameid, "ASMA", 4)) {
io_dldi_data = dldiLoadFromBin(r4tf_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} else if (!memcmp(gamename, "TOP TF/SD DS", 12) || !memcmp(gameid, "A76E", 4)) {
io_dldi_data = dldiLoadFromBin(tt_sd_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} else /*if (!memcmp(gamename, "PASS", 4) && !memcmp(gameid, "ASME", 4)) {
io_dldi_data = dldiLoadFromBin(CycloEvo_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} else*/ if (!memcmp(gamename, "D!S!XTREME", 12) && !memcmp(gameid, "AYIE", 4)) {
io_dldi_data = dldiLoadFromBin(dsx_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} else if (!memcmp(gamename, "QMATETRIAL", 9) || !memcmp(gamename, "R4DSULTRA", 9)) {
io_dldi_data = dldiLoadFromBin(r4idsn_sd_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} else if (!memcmp(gameid, "ACEK", 4) || !memcmp(gameid, "YCEP", 4) || !memcmp(gameid, "AHZH", 4) || !memcmp(gameid, "CHPJ", 4)) {
io_dldi_data = dldiLoadFromBin(ak2_sd_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
} /*else if (!memcmp(gameid, "ALXX", 4)) {
io_dldi_data = dldiLoadFromBin(dstwo_dldi);
fatMountSimple("fat", &io_dldi_data->ioInterface);
}*/
if (flashcardFound()) {
fatGetVolumeLabel("fat", fatLabel);
fixLabel(true);
struct statvfs st;
if (statvfs("fat:/", &st) == 0) {
fatSize = st.f_bsize * st.f_blocks;
}
return true;
}
}
return false;
}
bool flashcardMount(void) {
if ((!isDSiMode()) || (arm7SCFGLocked && !sdMountedDone)) {
fatInitDefault();
if (flashcardFound()) {
fatGetVolumeLabel("fat", fatLabel);
fixLabel(true);
struct statvfs st;
if (statvfs("fat:/", &st) == 0) {
fatSize = st.f_bsize * st.f_blocks;
}
return true;
}
return false;
} else {
return twl_flashcardMount();
}
}
void flashcardUnmount(void) {
fatUnmount("fat");
fatLabel[0] = '\0';
fatSize = 0;
flashcardMounted = false;
}
TWL_CODE void ramdrive1Mount(void) {
LZ77_Decompress((u8*)__9MB_lz77, (u8*)0x02500000);
fatMountSimple("ram1", &io_ram_drive);
ramdrive1Mounted = (access("ram1:/", F_OK) == 0);
}
TWL_CODE void ramdrive2Mount(void) {
LZ77_Decompress((u8*)__16MB_lz77, (u8*)0x0D000000);
fatMountSimple("ram2", &io_ram_drive2);
ramdrive2Mounted = (access("ram2:/", F_OK) == 0);
}