From b0dee4269875a62ea2a4b40e58dd919de07cdee2 Mon Sep 17 00:00:00 2001 From: Maschell Date: Thu, 21 Jul 2022 18:03:22 +0200 Subject: [PATCH] Implement disc interface --- include/mocha/disc_interface.h | 74 +++++++++++ include/mocha/fsa.h | 4 +- source/disc_interface.c | 236 +++++++++++++++++++++++++++++++++ source/fsa.cpp | 4 +- 4 files changed, 314 insertions(+), 4 deletions(-) create mode 100644 include/mocha/disc_interface.h create mode 100644 source/disc_interface.c diff --git a/include/mocha/disc_interface.h b/include/mocha/disc_interface.h new file mode 100644 index 0000000..fd28481 --- /dev/null +++ b/include/mocha/disc_interface.h @@ -0,0 +1,74 @@ +/*************************************************************************** + * Copyright (C) 2016 + * by Dimok + * + * This software is provided 'as-is', without any express or implied + * warranty. In no event will the authors be held liable for any + * damages arising from the use of this software. + * + * Permission is granted to anyone to use this software for any + * purpose, including commercial applications, and to alter it and + * redistribute it freely, subject to the following restrictions: + * + * 1. The origin of this software must not be misrepresented; you + * must not claim that you wrote the original software. If you use + * this software in a product, an acknowledgment in the product + * documentation would be appreciated but is not required. + * + * 2. Altered source versions must be plainly marked as such, and + * must not be misrepresented as being the original software. + * + * 3. This notice may not be removed or altered from any source + * distribution. + ***************************************************************************/ +#pragma once + +#include +#include + +#ifdef __cplusplus +extern "C" { +#endif + +#define DEVICE_TYPE_WII_U_SD (('W' << 24) | ('U' << 16) | ('S' << 8) | 'D') +#define DEVICE_TYPE_WII_U_USB (('W' << 24) | ('U' << 16) | ('S' << 8) | 'B') +#define FEATURE_WII_U_SD 0x00001000 +#define FEATURE_WII_U_USB 0x00002000 + +#ifndef OGC_DISC_IO_INCLUDE +typedef uint32_t sec_t; + +#define FEATURE_MEDIUM_CANREAD 0x00000001 +#define FEATURE_MEDIUM_CANWRITE 0x00000002 + +typedef bool (*FN_MEDIUM_STARTUP)(void); + +typedef bool (*FN_MEDIUM_ISINSERTED)(void); + +typedef bool (*FN_MEDIUM_READSECTORS)(uint32_t sector, uint32_t numSectors, void *buffer); + +typedef bool (*FN_MEDIUM_WRITESECTORS)(uint32_t sector, uint32_t numSectors, const void *buffer); + +typedef bool (*FN_MEDIUM_CLEARSTATUS)(void); + +typedef bool (*FN_MEDIUM_SHUTDOWN)(void); + +struct DISC_INTERFACE_STRUCT { + unsigned long ioType; + unsigned long features; + FN_MEDIUM_STARTUP startup; + FN_MEDIUM_ISINSERTED isInserted; + FN_MEDIUM_READSECTORS readSectors; + FN_MEDIUM_WRITESECTORS writeSectors; + FN_MEDIUM_CLEARSTATUS clearStatus; + FN_MEDIUM_SHUTDOWN shutdown; +}; + +typedef struct DISC_INTERFACE_STRUCT DISC_INTERFACE; +#endif +extern const DISC_INTERFACE Mocha_sdio_disc_interface; +extern const DISC_INTERFACE Mocha_usb_disc_interface; + +#ifdef __cplusplus +} +#endif diff --git a/include/mocha/fsa.h b/include/mocha/fsa.h index 6abb28a..d5677bd 100644 --- a/include/mocha/fsa.h +++ b/include/mocha/fsa.h @@ -15,7 +15,7 @@ extern "C" { * @param outHandle pointer where the handle of the raw device will be stored * @return */ -FSError FSAEx_RawOpen(FSClient *client, char *device_path, int32_t *outHandle); +FSError FSAEx_RawOpen(FSClient *client, const char *device_path, int32_t *outHandle); /** * Opens a device for raw read/write @@ -24,7 +24,7 @@ FSError FSAEx_RawOpen(FSClient *client, char *device_path, int32_t *outHandle); * @param outHandle pointer where the handle of the raw device will be stored * @return */ -FSError FSAEx_RawOpenEx(int clientHandle, char *device_path, int32_t *outHandle); +FSError FSAEx_RawOpenEx(int clientHandle, const char *device_path, int32_t *outHandle); /** * Closes a devices that was previously opened via FSAEx_RawOpen diff --git a/source/disc_interface.c b/source/disc_interface.c new file mode 100644 index 0000000..2885339 --- /dev/null +++ b/source/disc_interface.c @@ -0,0 +1,236 @@ +/*************************************************************************** + * Copyright (C) 2016 + * by Dimok + * + * This software is provided 'as-is', without any express or implied + * warranty. In no event will the authors be held liable for any + * damages arising from the use of this software. + * + * Permission is granted to anyone to use this software for any + * purpose, including commercial applications, and to alter it and + * redistribute it freely, subject to the following restrictions: + * + * 1. The origin of this software must not be misrepresented; you + * must not claim that you wrote the original software. If you use + * this software in a product, an acknowledgment in the product + * documentation would be appreciated but is not required. + * + * 2. Altered source versions must be plainly marked as such, and + * must not be misrepresented as being the original software. + * + * 3. This notice may not be removed or altered from any source + * distribution. + ***************************************************************************/ +#include "mocha/disc_interface.h" +#include "mocha/fsa.h" +#include "mocha/mocha.h" +#include +#include + +#define FSA_REF_SD 0x01 +#define FSA_REF_USB 0x02 + +static int initialized = 0; + +static int fsaFdSd = 0; +static int fsaFdUsb = 0; +static int sdioFd = 0; +static int usbFd = 0; + +static void Mocha_disc_io_initialize(void) { + if (initialized == 0) { + initialized = 1; + fsaFdSd = -1; + fsaFdUsb = -1; + sdioFd = -1; + usbFd = -1; + } +} + +static bool Mocha_disc_io_fsa_open(int fsaFd) { + Mocha_disc_io_initialize(); + + if (fsaFd == FSA_REF_SD) { + if (fsaFdSd < 0) { + fsaFdSd = IOS_Open("/dev/fsa", IOS_OPEN_READWRITE); + if (fsaFdSd >= 0 && Mocha_UnlockFSClientEx(fsaFdSd) != MOCHA_RESULT_SUCCESS) { + IOS_Close(fsaFdSd); + fsaFdSd = -1; + } + } + + if (fsaFdSd >= 0) { + return true; + } + } else if (fsaFd == FSA_REF_USB) { + if (fsaFdUsb < 0) { + fsaFdUsb = IOS_Open("/dev/fsa", IOS_OPEN_READWRITE); + if (fsaFdUsb >= 0 && Mocha_UnlockFSClientEx(fsaFdUsb) != MOCHA_RESULT_SUCCESS) { + IOS_Close(fsaFdUsb); + fsaFdUsb = -1; + } + } + if (fsaFdUsb >= 0) { + return true; + } + } + + return false; +} + +static void Mocha_disc_io_fsa_close(int fsaFd) { + if (fsaFd == FSA_REF_SD) { + if (fsaFdSd >= 0) { + IOS_Close(fsaFdSd); + fsaFdSd = -1; + } + } else if (fsaFd == FSA_REF_USB) { + if (fsaFdUsb >= 0) { + IOS_Close(fsaFdUsb); + fsaFdUsb = -1; + } + } +} + +static bool Mocha_sdio_startup(void) { + if (!Mocha_disc_io_fsa_open(FSA_REF_SD)) + return false; + + if (sdioFd < 0) { + int res = FSAEx_RawOpenEx(fsaFdSd, "/dev/sdcard01", &sdioFd); + if (res < 0) { + Mocha_disc_io_fsa_close(FSA_REF_SD); + sdioFd = -1; + } + } + + return (sdioFd >= 0); +} + +static bool Mocha_sdio_isInserted(void) { + //! TODO: check for SD card inserted with Mocha_FSA_GetDeviceInfo() + return initialized && (fsaFdSd >= 0) && (sdioFd >= 0); +} + +static bool Mocha_sdio_clearStatus(void) { + return true; +} + +static bool Mocha_sdio_shutdown(void) { + if (!Mocha_sdio_isInserted()) + return false; + + FSAEx_RawCloseEx(fsaFdSd, sdioFd); + Mocha_disc_io_fsa_close(FSA_REF_SD); + sdioFd = -1; + return true; +} + +static bool Mocha_sdio_readSectors(uint32_t sector, uint32_t numSectors, void *buffer) { + if (!Mocha_sdio_isInserted()) { + return false; + } + + int res = FSAEx_RawReadEx(fsaFdSd, buffer, 512, numSectors, sector, sdioFd); + if (res < 0) { + return false; + } + + return true; +} + +static bool Mocha_sdio_writeSectors(uint32_t sector, uint32_t numSectors, const void *buffer) { + if (!Mocha_sdio_isInserted()) { + return false; + } + + int res = FSAEx_RawWriteEx(fsaFdSd, buffer, 512, numSectors, sector, sdioFd); + if (res < 0) { + return false; + } + + return true; +} + +const DISC_INTERFACE Mocha_sdio_disc_interface = { + DEVICE_TYPE_WII_U_SD, + FEATURE_MEDIUM_CANREAD | FEATURE_MEDIUM_CANWRITE | FEATURE_WII_U_SD, + Mocha_sdio_startup, + Mocha_sdio_isInserted, + Mocha_sdio_readSectors, + Mocha_sdio_writeSectors, + Mocha_sdio_clearStatus, + Mocha_sdio_shutdown}; + +static bool Mocha_usb_startup(void) { + if (!Mocha_disc_io_fsa_open(FSA_REF_USB)) { + return false; + } + + if (usbFd < 0) { + int res = FSAEx_RawOpenEx(fsaFdUsb, "/dev/usb01", &usbFd); + if (res < 0) { + res = FSAEx_RawOpenEx(fsaFdUsb, "/dev/usb02", &usbFd); + if (res < 0) { + Mocha_disc_io_fsa_close(FSA_REF_USB); + usbFd = -1; + } + } + } + return (usbFd >= 0); +} + +static bool Mocha_usb_isInserted(void) { + return initialized && (fsaFdUsb >= 0) && (usbFd >= 0); +} + +static bool Mocha_usb_clearStatus(void) { + return true; +} + +static bool Mocha_usb_shutdown(void) { + if (!Mocha_usb_isInserted()) { + return false; + } + + FSAEx_RawCloseEx(fsaFdUsb, usbFd); + Mocha_disc_io_fsa_close(FSA_REF_USB); + usbFd = -1; + return true; +} + +static bool Mocha_usb_readSectors(uint32_t sector, uint32_t numSectors, void *buffer) { + if (!Mocha_usb_isInserted()) { + return false; + } + + int res = FSAEx_RawReadEx(fsaFdUsb, buffer, 512, numSectors, sector, usbFd); + if (res < 0) { + return false; + } + + return true; +} + +static bool Mocha_usb_writeSectors(uint32_t sector, uint32_t numSectors, const void *buffer) { + if (!Mocha_usb_isInserted()) { + return false; + } + + int res = FSAEx_RawWriteEx(fsaFdUsb, buffer, 512, numSectors, sector, usbFd); + if (res < 0) { + return false; + } + + return true; +} + +const DISC_INTERFACE Mocha_usb_disc_interface = { + DEVICE_TYPE_WII_U_USB, + FEATURE_MEDIUM_CANREAD | FEATURE_MEDIUM_CANWRITE | FEATURE_WII_U_USB, + Mocha_usb_startup, + Mocha_usb_isInserted, + Mocha_usb_readSectors, + Mocha_usb_writeSectors, + Mocha_usb_clearStatus, + Mocha_usb_shutdown}; diff --git a/source/fsa.cpp b/source/fsa.cpp index fe14ab3..58890f2 100644 --- a/source/fsa.cpp +++ b/source/fsa.cpp @@ -6,14 +6,14 @@ #include #include -FSError FSAEx_RawOpen(FSClient *client, char *device_path, int32_t *outHandle) { +FSError FSAEx_RawOpen(FSClient *client, const char *device_path, int32_t *outHandle) { if (!client) { return FS_ERROR_INVALID_CLIENTHANDLE; } return FSAEx_RawOpenEx(FSGetClientBody(client)->clientHandle, device_path, outHandle); } -FSError FSAEx_RawOpenEx(int clientHandle, char *device_path, int32_t *outHandle) { +FSError FSAEx_RawOpenEx(int clientHandle, const char *device_path, int32_t *outHandle) { if (!outHandle) { return FS_ERROR_INVALID_BUFFER; }