mirrors/Atmosphere
0
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere.git synced 2025-04-22 20:44:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Commit changes

Browse source
This commit is contained in:
XorTroll 2019-08-02 18:52:13 +02:00
parent 4d72c2b37a
commit 09c8ddad1c
26 changed files with 24912 additions and 195 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stratosphere/ams_mitm/Makefile
View file

@ -26,7 +26,7 @@ endif
#---------------------------------------------------------------------------------
TARGET := $(notdir $(CURDIR))
BUILD := build
SOURCES := source source/fs_mitm source/set_mitm source/bpc_mitm source/ns_mitm
SOURCES := source source/fs_mitm source/set_mitm source/bpc_mitm source/ns_mitm source/fsp_usb source/fsp_usb/fatfs
DATA := data
INCLUDES := include ../../common/include
EXEFS_SRC := exefs_src

208
stratosphere/ams_mitm/source/amsmitm_main.cpp
View file

@ -1,104 +1,104 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <cstdlib>
#include <cstdint>
#include <cstring>
#include <malloc.h>
#include <switch.h>
#include <atmosphere.h>
#include <stratosphere.hpp>
#include "amsmitm_modules.hpp"
#include "utils.hpp"
extern "C" {
extern u32 __start__;
u32 __nx_applet_type = AppletType_None;
#define INNER_HEAP_SIZE 0x1000000
size_t nx_inner_heap_size = INNER_HEAP_SIZE;
char nx_inner_heap[INNER_HEAP_SIZE];
void __libnx_initheap(void);
void __appInit(void);
void __appExit(void);
/* Exception handling. */
alignas(16) u8 __nx_exception_stack[0x1000];
u64 __nx_exception_stack_size = sizeof(__nx_exception_stack);
void __libnx_exception_handler(ThreadExceptionDump *ctx);
void __libstratosphere_exception_handler(AtmosphereFatalErrorContext *ctx);
}
sts::ncm::TitleId __stratosphere_title_id = sts::ncm::TitleId::AtmosphereMitm;
void __libnx_exception_handler(ThreadExceptionDump *ctx) {
StratosphereCrashHandler(ctx);
}
void __libstratosphere_exception_handler(AtmosphereFatalErrorContext *ctx) {
/* We're bpc-mitm (or ams_mitm, anyway), so manually reboot to fatal error. */
Utils::RebootToFatalError(ctx);
}
void __libnx_initheap(void) {
void* addr = nx_inner_heap;
size_t size = nx_inner_heap_size;
/* Newlib */
extern char* fake_heap_start;
extern char* fake_heap_end;
fake_heap_start = (char*)addr;
fake_heap_end = (char*)addr + size;
}
void __appInit(void) {
SetFirmwareVersionForLibnx();
DoWithSmSession([&]() {
R_ASSERT(fsInitialize());
R_ASSERT(pmdmntInitialize());
R_ASSERT(pminfoInitialize());
});
CheckAtmosphereVersion(CURRENT_ATMOSPHERE_VERSION);
}
void __appExit(void) {
/* Cleanup services. */
fsExit();
}
int main(int argc, char **argv)
{
consoleDebugInit(debugDevice_SVC);
HosThread initializer_thread;
LaunchAllMitmModules();
R_ASSERT(initializer_thread.Initialize(&Utils::InitializeThreadFunc, NULL, 0x4000, 0x15));
R_ASSERT(initializer_thread.Start());
/* Wait for all mitm modules to end. */
WaitAllMitmModules();
return 0;
}
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <cstdlib>
#include <cstdint>
#include <cstring>
#include <malloc.h>
#include <switch.h>
#include <atmosphere.h>
#include <stratosphere.hpp>
#include "amsmitm_modules.hpp"
#include "utils.hpp"
extern "C" {
extern u32 __start__;
u32 __nx_applet_type = AppletType_None;
#define INNER_HEAP_SIZE 0x1000000
size_t nx_inner_heap_size = INNER_HEAP_SIZE;
char nx_inner_heap[INNER_HEAP_SIZE];
void __libnx_initheap(void);
void __appInit(void);
void __appExit(void);
/* Exception handling. */
alignas(16) u8 __nx_exception_stack[0x1000];
u64 __nx_exception_stack_size = sizeof(__nx_exception_stack);
void __libnx_exception_handler(ThreadExceptionDump *ctx);
void __libstratosphere_exception_handler(AtmosphereFatalErrorContext *ctx);
}
sts::ncm::TitleId __stratosphere_title_id = sts::ncm::TitleId::AtmosphereMitm;
void __libnx_exception_handler(ThreadExceptionDump *ctx) {
StratosphereCrashHandler(ctx);
}
void __libstratosphere_exception_handler(AtmosphereFatalErrorContext *ctx) {
/* We're bpc-mitm (or ams_mitm, anyway), so manually reboot to fatal error. */
Utils::RebootToFatalError(ctx);
}
void __libnx_initheap(void) {
void* addr = nx_inner_heap;
size_t size = nx_inner_heap_size;
/* Newlib */
extern char* fake_heap_start;
extern char* fake_heap_end;
fake_heap_start = (char*)addr;
fake_heap_end = (char*)addr + size;
}
void __appInit(void) {
SetFirmwareVersionForLibnx();
DoWithSmSession([&]() {
R_ASSERT(fsInitialize());
R_ASSERT(pmdmntInitialize());
R_ASSERT(pminfoInitialize());
});
CheckAtmosphereVersion(CURRENT_ATMOSPHERE_VERSION);
}
void __appExit(void) {
/* Cleanup services. */
fsExit();
}
int main(int argc, char **argv)
{
consoleDebugInit(debugDevice_SVC);
HosThread initializer_thread;
LaunchAllModules();
R_ASSERT(initializer_thread.Initialize(&Utils::InitializeThreadFunc, NULL, 0x4000, 0x15));
R_ASSERT(initializer_thread.Start());
/* Wait for all mitm modules to end. */
WaitAllModules();
return 0;
}

121
stratosphere/ams_mitm/source/amsmitm_modules.cpp
View file

@ -1,60 +1,63 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <switch.h>
#include <stratosphere.hpp>
#include <cstring>
#include "debug.hpp"
#include "amsmitm_modules.hpp"
#include "fs_mitm/fsmitm_main.hpp"
#include "set_mitm/setmitm_main.hpp"
#include "bpc_mitm/bpcmitm_main.hpp"
#include "ns_mitm/nsmitm_main.hpp"
static HosThread g_module_threads[MitmModuleId_Count];
static const struct {
ThreadFunc main;
u32 priority;
u32 stack_size;
} g_module_definitions[MitmModuleId_Count] = {
{ &FsMitmMain, FsMitmPriority, FsMitmStackSize }, /* FsMitm */
{ &SetMitmMain, SetMitmPriority, SetMitmStackSize }, /* SetMitm */
{ &BpcMitmMain, BpcMitmPriority, BpcMitmStackSize }, /* BpcMitm */
{ &NsMitmMain, NsMitmPriority, NsMitmStackSize }, /* NsMitm */
};
void LaunchAllMitmModules() {
/* Create thread for each module. */
for (u32 i = 0; i < static_cast<u32>(MitmModuleId_Count); i++) {
const auto cur_module = &g_module_definitions[i];
R_ASSERT(g_module_threads[i].Initialize(cur_module->main, nullptr, cur_module->stack_size, cur_module->priority));
}
/* Start thread for each module. */
for (u32 i = 0; i < static_cast<u32>(MitmModuleId_Count); i++) {
R_ASSERT(g_module_threads[i].Start());
}
}
void WaitAllMitmModules() {
/* Wait on thread for each module. */
for (u32 i = 0; i < static_cast<u32>(MitmModuleId_Count); i++) {
g_module_threads[i].Join();
}
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <switch.h>
#include <stratosphere.hpp>
#include <cstring>
#include "debug.hpp"
#include "amsmitm_modules.hpp"
#include "fs_mitm/fsmitm_main.hpp"
#include "set_mitm/setmitm_main.hpp"
#include "bpc_mitm/bpcmitm_main.hpp"
#include "ns_mitm/nsmitm_main.hpp"
#include "fsp_usb/fspusb_main.hpp"
static HosThread g_module_threads[ModuleId_Count];
static const struct {
ThreadFunc main;
u32 priority;
u32 stack_size;
} g_module_definitions[ModuleId_Count] = {
{ &FsMitmMain, FsMitmPriority, FsMitmStackSize }, /* FsMitm */
{ &SetMitmMain, SetMitmPriority, SetMitmStackSize }, /* SetMitm */
{ &BpcMitmMain, BpcMitmPriority, BpcMitmStackSize }, /* BpcMitm */
{ &NsMitmMain, NsMitmPriority, NsMitmStackSize }, /* NsMitm */
{ &FspUsbMain, FspUsbPriority, FspUsbStackSize }, /* FspUsb */
};
void LaunchAllModules() {
/* Create thread for each module. */
for (u32 i = 0; i < static_cast<u32>(ModuleId_Count); i++) {
const auto cur_module = &g_module_definitions[i];
R_ASSERT(g_module_threads[i].Initialize(cur_module->main, nullptr, cur_module->stack_size, cur_module->priority));
}
/* Start thread for each module. */
for (u32 i = 0; i < static_cast<u32>(ModuleId_Count); i++) {
R_ASSERT(g_module_threads[i].Start());
}
}
void WaitAllModules() {
/* Wait on thread for each module. */
for (u32 i = 0; i < static_cast<u32>(ModuleId_Count); i++) {
g_module_threads[i].Join();
}
}

63
stratosphere/ams_mitm/source/amsmitm_modules.hpp
View file

@ -1,31 +1,32 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
enum MitmModuleId : u32 {
MitmModuleId_FsMitm = 0,
MitmModuleId_SetMitm = 1,
MitmModuleId_BpcMitm = 2,
MitmModuleId_NsMitm = 3,
/* Always keep this at the end. */
MitmModuleId_Count,
};
void LaunchAllMitmModules();
void WaitAllMitmModules();
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
enum ModuleId : u32 {
ModuleId_FsMitm = 0,
ModuleId_SetMitm = 1,
ModuleId_BpcMitm = 2,
ModuleId_NsMitm = 3,
ModuleId_FspUsb = 4,
/* Always keep this at the end. */
ModuleId_Count,
};
void LaunchAllModules();
void WaitAllModules();

105
stratosphere/ams_mitm/source/fsp_usb/fatfs/diskio.cpp Normal file
View file

@ -0,0 +1,105 @@
/*-----------------------------------------------------------------------*/
/* Low level disk I/O module skeleton for FatFs (C)ChaN, 2016 */
/*-----------------------------------------------------------------------*/
/* If a working storage control module is available, it should be */
/* attached to the FatFs via a glue function rather than modifying it. */
/* This is an example of glue functions to attach various exsisting */
/* storage control modules to the FatFs module with a defined API. */
/*-----------------------------------------------------------------------*/
#include <stdio.h>
#include <string.h>
#include <switch.h>
#include <vector>
#include "ff.h"
#include "diskio.h"
#include "../fspusb_drive.hpp"
extern HosMutex g_usbdrive_drives_lock;
extern std::vector<DriveData> g_usbdrive_drives;
/*-----------------------------------------------------------------------*/
/* Get Drive Status */
/*-----------------------------------------------------------------------*/
extern "C" DSTATUS disk_status (
BYTE pdrv /* Physical drive nmuber to identify the drive */
)
{
return 0;
}
/*-----------------------------------------------------------------------*/
/* Inidialize a Drive */
/*-----------------------------------------------------------------------*/
extern "C" DSTATUS disk_initialize (
BYTE pdrv /* Physical drive nmuber to identify the drive */
)
{
return 0;
}
/*-----------------------------------------------------------------------*/
/* Read Sector(s) */
/*-----------------------------------------------------------------------*/
extern "C" DRESULT disk_read (
BYTE pdrv, /* Physical drive nmuber to identify the drive */
BYTE *buff, /* Data buffer to store read data */
DWORD sector, /* Start sector in LBA */
UINT count /* Number of sectors to read */
)
{
DRIVES_SCOPE_GUARD;
int wres = g_usbdrive_drives[pdrv].scsi->partitions[0].read_sectors(buff, sector, count);
if(wres != 0) {
return RES_OK;
}
return RES_PARERR;
}
/*-----------------------------------------------------------------------*/
/* Write Sector(s) */
/*-----------------------------------------------------------------------*/
#if FF_FS_READONLY == 0
extern "C" DRESULT disk_write (
BYTE pdrv, /* Physical drive nmuber to identify the drive */
const BYTE *buff, /* Data to be written */
DWORD sector, /* Start sector in LBA */
UINT count /* Number of sectors to write */
)
{
DRIVES_SCOPE_GUARD;
int wres = g_usbdrive_drives[pdrv].scsi->partitions[0].write_sectors(buff, sector, count);
if(wres != 0) {
return RES_OK;
}
return RES_PARERR;
}
#endif
/*-----------------------------------------------------------------------*/
/* Miscellaneous Functions */
/*-----------------------------------------------------------------------*/
extern "C" DRESULT disk_ioctl (
BYTE pdrv, /* Physical drive nmuber (0..) */
BYTE cmd, /* Control code */
void *buff /* Buffer to send/receive control data */
)
{
return RES_OK;
}

77
stratosphere/ams_mitm/source/fsp_usb/fatfs/diskio.h Normal file
View file

@ -0,0 +1,77 @@
/*-----------------------------------------------------------------------/
/ Low level disk interface modlue include file (C)ChaN, 2014 /
/-----------------------------------------------------------------------*/
#ifndef _DISKIO_DEFINED
#define _DISKIO_DEFINED
#ifdef __cplusplus
extern "C" {
#endif
/* Status of Disk Functions */
typedef BYTE DSTATUS;
/* Results of Disk Functions */
typedef enum {
RES_OK = 0, /* 0: Successful */
RES_ERROR, /* 1: R/W Error */
RES_WRPRT, /* 2: Write Protected */
RES_NOTRDY, /* 3: Not Ready */
RES_PARERR /* 4: Invalid Parameter */
} DRESULT;
/*---------------------------------------*/
/* Prototypes for disk control functions */
DSTATUS disk_initialize (BYTE pdrv);
DSTATUS disk_status (BYTE pdrv);
DRESULT disk_read (BYTE pdrv, BYTE* buff, DWORD sector, UINT count);
DRESULT disk_write (BYTE pdrv, const BYTE* buff, DWORD sector, UINT count);
DRESULT disk_ioctl (BYTE pdrv, BYTE cmd, void* buff);
/* Disk Status Bits (DSTATUS) */
#define STA_NOINIT 0x01 /* Drive not initialized */
#define STA_NODISK 0x02 /* No medium in the drive */
#define STA_PROTECT 0x04 /* Write protected */
/* Command code for disk_ioctrl fucntion */
/* Generic command (Used by FatFs) */
#define CTRL_SYNC 0 /* Complete pending write process (needed at FF_FS_READONLY == 0) */
#define GET_SECTOR_COUNT 1 /* Get media size (needed at FF_USE_MKFS == 1) */
#define GET_SECTOR_SIZE 2 /* Get sector size (needed at FF_MAX_SS != FF_MIN_SS) */
#define GET_BLOCK_SIZE 3 /* Get erase block size (needed at FF_USE_MKFS == 1) */
#define CTRL_TRIM 4 /* Inform device that the data on the block of sectors is no longer used (needed at FF_USE_TRIM == 1) */
/* Generic command (Not used by FatFs) */
#define CTRL_POWER 5 /* Get/Set power status */
#define CTRL_LOCK 6 /* Lock/Unlock media removal */
#define CTRL_EJECT 7 /* Eject media */
#define CTRL_FORMAT 8 /* Create physical format on the media */
/* MMC/SDC specific ioctl command */
#define MMC_GET_TYPE 10 /* Get card type */
#define MMC_GET_CSD 11 /* Get CSD */
#define MMC_GET_CID 12 /* Get CID */
#define MMC_GET_OCR 13 /* Get OCR */
#define MMC_GET_SDSTAT 14 /* Get SD status */
#define ISDIO_READ 55 /* Read data form SD iSDIO register */
#define ISDIO_WRITE 56 /* Write data to SD iSDIO register */
#define ISDIO_MRITE 57 /* Masked write data to SD iSDIO register */
/* ATA/CF specific ioctl command */
#define ATA_GET_REV 20 /* Get F/W revision */
#define ATA_GET_MODEL 21 /* Get model name */
#define ATA_GET_SN 22 /* Get serial number */
#ifdef __cplusplus
}
#endif
#endif

6554
stratosphere/ams_mitm/source/fsp_usb/fatfs/ff.c Normal file
View file

File diff suppressed because it is too large Load diff

405
stratosphere/ams_mitm/source/fsp_usb/fatfs/ff.h Normal file
View file

@ -0,0 +1,405 @@
/*----------------------------------------------------------------------------/
/ FatFs - Generic FAT Filesystem module R0.13c /
/-----------------------------------------------------------------------------/
/
/ Copyright (C) 2018, ChaN, all right reserved.
/
/ FatFs module is an open source software. Redistribution and use of FatFs in
/ source and binary forms, with or without modification, are permitted provided
/ that the following condition is met:
/ 1. Redistributions of source code must retain the above copyright notice,
/ this condition and the following disclaimer.
/
/ This software is provided by the copyright holder and contributors "AS IS"
/ and any warranties related to this software are DISCLAIMED.
/ The copyright owner or contributors be NOT LIABLE for any damages caused
/ by use of this software.
/
/----------------------------------------------------------------------------*/
#ifndef FF_DEFINED
#define FF_DEFINED 86604 /* Revision ID */
#ifdef __cplusplus
extern "C" {
#endif
#include "ffconf.h" /* FatFs configuration options */
#if FF_DEFINED != FFCONF_DEF
#error Wrong configuration file (ffconf.h).
#endif
/* Integer types used for FatFs API */
#if defined(_WIN32) /* Main development platform */
#define FF_INTDEF 2
#include <windows.h>
typedef unsigned __int64 QWORD;
#elif (defined(__STDC_VERSION__) && __STDC_VERSION__ >= 199901L) || defined(__cplusplus) /* C99 or later */
#define FF_INTDEF 2
#include <stdint.h>
typedef unsigned int UINT; /* int must be 16-bit or 32-bit */
typedef unsigned char BYTE; /* char must be 8-bit */
typedef uint16_t WORD; /* 16-bit unsigned integer */
typedef uint16_t WCHAR; /* 16-bit unsigned integer */
typedef uint32_t DWORD; /* 32-bit unsigned integer */
typedef uint64_t QWORD; /* 64-bit unsigned integer */
#else /* Earlier than C99 */
#define FF_INTDEF 1
typedef unsigned int UINT; /* int must be 16-bit or 32-bit */
typedef unsigned char BYTE; /* char must be 8-bit */
typedef unsigned short WORD; /* 16-bit unsigned integer */
typedef unsigned short WCHAR; /* 16-bit unsigned integer */
typedef unsigned long DWORD; /* 32-bit unsigned integer */
#endif
/* Definitions of volume management */
#if FF_MULTI_PARTITION /* Multiple partition configuration */
typedef struct {
BYTE pd; /* Physical drive number */
BYTE pt; /* Partition: 0:Auto detect, 1-4:Forced partition) */
} PARTITION;
extern PARTITION VolToPart[]; /* Volume - Partition resolution table */
#endif
#if FF_STR_VOLUME_ID
#ifndef FF_VOLUME_STRS
extern const char* VolumeStr[FF_VOLUMES]; /* User defied volume ID */
#endif
#endif
/* Type of path name strings on FatFs API */
#ifndef _INC_TCHAR
#define _INC_TCHAR
#if FF_USE_LFN && FF_LFN_UNICODE == 1 /* Unicode in UTF-16 encoding */
typedef WCHAR TCHAR;
#define _T(x) L ## x
#define _TEXT(x) L ## x
#elif FF_USE_LFN && FF_LFN_UNICODE == 2 /* Unicode in UTF-8 encoding */
typedef char TCHAR;
#define _T(x) u8 ## x
#define _TEXT(x) u8 ## x
#elif FF_USE_LFN && FF_LFN_UNICODE == 3 /* Unicode in UTF-32 encoding */
typedef DWORD TCHAR;
#define _T(x) U ## x
#define _TEXT(x) U ## x
#elif FF_USE_LFN && (FF_LFN_UNICODE < 0 || FF_LFN_UNICODE > 3)
#error Wrong FF_LFN_UNICODE setting
#else /* ANSI/OEM code in SBCS/DBCS */
typedef char TCHAR;
#define _T(x) x
#define _TEXT(x) x
#endif
#endif
/* Type of file size variables */
#if FF_FS_EXFAT
#if FF_INTDEF != 2
#error exFAT feature wants C99 or later
#endif
typedef QWORD FSIZE_t;
#else
typedef DWORD FSIZE_t;
#endif
/* Filesystem object structure (FATFS) */
typedef struct {
BYTE fs_type; /* Filesystem type (0:not mounted) */
BYTE pdrv; /* Associated physical drive */
BYTE n_fats; /* Number of FATs (1 or 2) */
BYTE wflag; /* win[] flag (b0:dirty) */
BYTE fsi_flag; /* FSINFO flags (b7:disabled, b0:dirty) */
WORD id; /* Volume mount ID */
WORD n_rootdir; /* Number of root directory entries (FAT12/16) */
WORD csize; /* Cluster size [sectors] */
#if FF_MAX_SS != FF_MIN_SS
WORD ssize; /* Sector size (512, 1024, 2048 or 4096) */
#endif
#if FF_USE_LFN
WCHAR* lfnbuf; /* LFN working buffer */
#endif
#if FF_FS_EXFAT
BYTE* dirbuf; /* Directory entry block scratchpad buffer for exFAT */
#endif
#if FF_FS_REENTRANT
FF_SYNC_t sobj; /* Identifier of sync object */
#endif
#if !FF_FS_READONLY
DWORD last_clst; /* Last allocated cluster */
DWORD free_clst; /* Number of free clusters */
#endif
#if FF_FS_RPATH
DWORD cdir; /* Current directory start cluster (0:root) */
#if FF_FS_EXFAT
DWORD cdc_scl; /* Containing directory start cluster (invalid when cdir is 0) */
DWORD cdc_size; /* b31-b8:Size of containing directory, b7-b0: Chain status */
DWORD cdc_ofs; /* Offset in the containing directory (invalid when cdir is 0) */
#endif
#endif
DWORD n_fatent; /* Number of FAT entries (number of clusters + 2) */
DWORD fsize; /* Size of an FAT [sectors] */
DWORD volbase; /* Volume base sector */
DWORD fatbase; /* FAT base sector */
DWORD dirbase; /* Root directory base sector/cluster */
DWORD database; /* Data base sector */
#if FF_FS_EXFAT
DWORD bitbase; /* Allocation bitmap base sector */
#endif
DWORD winsect; /* Current sector appearing in the win[] */
BYTE win[FF_MAX_SS]; /* Disk access window for Directory, FAT (and file data at tiny cfg) */
} FATFS;
/* Object ID and allocation information (FFOBJID) */
typedef struct {
FATFS* fs; /* Pointer to the hosting volume of this object */
WORD id; /* Hosting volume mount ID */
BYTE attr; /* Object attribute */
BYTE stat; /* Object chain status (b1-0: =0:not contiguous, =2:contiguous, =3:fragmented in this session, b2:sub-directory stretched) */
DWORD sclust; /* Object data start cluster (0:no cluster or root directory) */
FSIZE_t objsize; /* Object size (valid when sclust != 0) */
#if FF_FS_EXFAT
DWORD n_cont; /* Size of first fragment - 1 (valid when stat == 3) */
DWORD n_frag; /* Size of last fragment needs to be written to FAT (valid when not zero) */
DWORD c_scl; /* Containing directory start cluster (valid when sclust != 0) */
DWORD c_size; /* b31-b8:Size of containing directory, b7-b0: Chain status (valid when c_scl != 0) */
DWORD c_ofs; /* Offset in the containing directory (valid when file object and sclust != 0) */
#endif
#if FF_FS_LOCK
UINT lockid; /* File lock ID origin from 1 (index of file semaphore table Files[]) */
#endif
} FFOBJID;
/* File object structure (FIL) */
typedef struct {
FFOBJID obj; /* Object identifier (must be the 1st member to detect invalid object pointer) */
BYTE flag; /* File status flags */
BYTE err; /* Abort flag (error code) */
FSIZE_t fptr; /* File read/write pointer (Zeroed on file open) */
DWORD clust; /* Current cluster of fpter (invalid when fptr is 0) */
DWORD sect; /* Sector number appearing in buf[] (0:invalid) */
#if !FF_FS_READONLY
DWORD dir_sect; /* Sector number containing the directory entry (not used at exFAT) */
BYTE* dir_ptr; /* Pointer to the directory entry in the win[] (not used at exFAT) */
#endif
#if FF_USE_FASTSEEK
DWORD* cltbl; /* Pointer to the cluster link map table (nulled on open, set by application) */
#endif
#if !FF_FS_TINY
BYTE buf[FF_MAX_SS]; /* File private data read/write window */
#endif
} FIL;
/* Directory object structure (DIR) */
typedef struct {
FFOBJID obj; /* Object identifier */
DWORD dptr; /* Current read/write offset */
DWORD clust; /* Current cluster */
DWORD sect; /* Current sector (0:Read operation has terminated) */
BYTE* dir; /* Pointer to the directory item in the win[] */
BYTE fn[12]; /* SFN (in/out) {body[8],ext[3],status[1]} */
#if FF_USE_LFN
DWORD blk_ofs; /* Offset of current entry block being processed (0xFFFFFFFF:Invalid) */
#endif
#if FF_USE_FIND
const TCHAR* pat; /* Pointer to the name matching pattern */
#endif
} DIR;
/* File information structure (FILINFO) */
typedef struct {
FSIZE_t fsize; /* File size */
WORD fdate; /* Modified date */
WORD ftime; /* Modified time */
BYTE fattrib; /* File attribute */
#if FF_USE_LFN
TCHAR altname[FF_SFN_BUF + 1];/* Altenative file name */
TCHAR fname[FF_LFN_BUF + 1]; /* Primary file name */
#else
TCHAR fname[12 + 1]; /* File name */
#endif
} FILINFO;
/* File function return code (FRESULT) */
typedef enum {
FR_OK = 0, /* (0) Succeeded */
FR_DISK_ERR, /* (1) A hard error occurred in the low level disk I/O layer */
FR_INT_ERR, /* (2) Assertion failed */
FR_NOT_READY, /* (3) The physical drive cannot work */
FR_NO_FILE, /* (4) Could not find the file */
FR_NO_PATH, /* (5) Could not find the path */
FR_INVALID_NAME, /* (6) The path name format is invalid */
FR_DENIED, /* (7) Access denied due to prohibited access or directory full */
FR_EXIST, /* (8) Access denied due to prohibited access */
FR_INVALID_OBJECT, /* (9) The file/directory object is invalid */
FR_WRITE_PROTECTED, /* (10) The physical drive is write protected */
FR_INVALID_DRIVE, /* (11) The logical drive number is invalid */
FR_NOT_ENABLED, /* (12) The volume has no work area */
FR_NO_FILESYSTEM, /* (13) There is no valid FAT volume */
FR_MKFS_ABORTED, /* (14) The f_mkfs() aborted due to any problem */
FR_TIMEOUT, /* (15) Could not get a grant to access the volume within defined period */
FR_LOCKED, /* (16) The operation is rejected according to the file sharing policy */
FR_NOT_ENOUGH_CORE, /* (17) LFN working buffer could not be allocated */
FR_TOO_MANY_OPEN_FILES, /* (18) Number of open files > FF_FS_LOCK */
FR_INVALID_PARAMETER /* (19) Given parameter is invalid */
} FRESULT;
/*--------------------------------------------------------------*/
/* FatFs module application interface */
FRESULT f_open (FIL* fp, const TCHAR* path, BYTE mode); /* Open or create a file */
FRESULT f_close (FIL* fp); /* Close an open file object */
FRESULT f_read (FIL* fp, void* buff, UINT btr, UINT* br); /* Read data from the file */
FRESULT f_write (FIL* fp, const void* buff, UINT btw, UINT* bw); /* Write data to the file */
FRESULT f_lseek (FIL* fp, FSIZE_t ofs); /* Move file pointer of the file object */
FRESULT f_truncate (FIL* fp); /* Truncate the file */
FRESULT f_sync (FIL* fp); /* Flush cached data of the writing file */
FRESULT f_opendir (DIR* dp, const TCHAR* path); /* Open a directory */
FRESULT f_closedir (DIR* dp); /* Close an open directory */
FRESULT f_readdir (DIR* dp, FILINFO* fno); /* Read a directory item */
FRESULT f_findfirst (DIR* dp, FILINFO* fno, const TCHAR* path, const TCHAR* pattern); /* Find first file */
FRESULT f_findnext (DIR* dp, FILINFO* fno); /* Find next file */
FRESULT f_mkdir (const TCHAR* path); /* Create a sub directory */
FRESULT f_unlink (const TCHAR* path); /* Delete an existing file or directory */
FRESULT f_rename (const TCHAR* path_old, const TCHAR* path_new); /* Rename/Move a file or directory */
FRESULT f_stat (const TCHAR* path, FILINFO* fno); /* Get file status */
FRESULT f_chmod (const TCHAR* path, BYTE attr, BYTE mask); /* Change attribute of a file/dir */
FRESULT f_utime (const TCHAR* path, const FILINFO* fno); /* Change timestamp of a file/dir */
FRESULT f_chdir (const TCHAR* path); /* Change current directory */
FRESULT f_chdrive (const TCHAR* path); /* Change current drive */
FRESULT f_getcwd (TCHAR* buff, UINT len); /* Get current directory */
FRESULT f_getfree (const TCHAR* path, DWORD* nclst, FATFS** fatfs); /* Get number of free clusters on the drive */
FRESULT f_getlabel (const TCHAR* path, TCHAR* label, DWORD* vsn); /* Get volume label */
FRESULT f_setlabel (const TCHAR* label); /* Set volume label */
FRESULT f_forward (FIL* fp, UINT(*func)(const BYTE*,UINT), UINT btf, UINT* bf); /* Forward data to the stream */
FRESULT f_expand (FIL* fp, FSIZE_t szf, BYTE opt); /* Allocate a contiguous block to the file */
FRESULT f_mount (FATFS* fs, const TCHAR* path, BYTE opt); /* Mount/Unmount a logical drive */
FRESULT f_mkfs (const TCHAR* path, BYTE opt, DWORD au, void* work, UINT len); /* Create a FAT volume */
FRESULT f_fdisk (BYTE pdrv, const DWORD* szt, void* work); /* Divide a physical drive into some partitions */
FRESULT f_setcp (WORD cp); /* Set current code page */
int f_putc (TCHAR c, FIL* fp); /* Put a character to the file */
int f_puts (const TCHAR* str, FIL* cp); /* Put a string to the file */
int f_printf (FIL* fp, const TCHAR* str, ...); /* Put a formatted string to the file */
TCHAR* f_gets (TCHAR* buff, int len, FIL* fp); /* Get a string from the file */
#define f_eof(fp) ((int)((fp)->fptr == (fp)->obj.objsize))
#define f_error(fp) ((fp)->err)
#define f_tell(fp) ((fp)->fptr)
#define f_size(fp) ((fp)->obj.objsize)
#define f_rewind(fp) f_lseek((fp), 0)
#define f_rewinddir(dp) f_readdir((dp), 0)
#define f_rmdir(path) f_unlink(path)
#define f_unmount(path) f_mount(0, path, 0)
#ifndef EOF
#define EOF (-1)
#endif
/*--------------------------------------------------------------*/
/* Additional user defined functions */
/* RTC function */
#if !FF_FS_READONLY && !FF_FS_NORTC
DWORD get_fattime (void);
#endif
/* LFN support functions */
#if FF_USE_LFN >= 1 /* Code conversion (defined in unicode.c) */
WCHAR ff_oem2uni (WCHAR oem, WORD cp); /* OEM code to Unicode conversion */
WCHAR ff_uni2oem (DWORD uni, WORD cp); /* Unicode to OEM code conversion */
DWORD ff_wtoupper (DWORD uni); /* Unicode upper-case conversion */
#endif
#if FF_USE_LFN == 3 /* Dynamic memory allocation */
void* ff_memalloc (UINT msize); /* Allocate memory block */
void ff_memfree (void* mblock); /* Free memory block */
#endif
/* Sync functions */
#if FF_FS_REENTRANT
int ff_cre_syncobj (BYTE vol, FF_SYNC_t* sobj); /* Create a sync object */
int ff_req_grant (FF_SYNC_t sobj); /* Lock sync object */
void ff_rel_grant (FF_SYNC_t sobj); /* Unlock sync object */
int ff_del_syncobj (FF_SYNC_t sobj); /* Delete a sync object */
#endif
/*--------------------------------------------------------------*/
/* Flags and offset address */
/* File access mode and open method flags (3rd argument of f_open) */
#define FA_READ 0x01
#define FA_WRITE 0x02
#define FA_OPEN_EXISTING 0x00
#define FA_CREATE_NEW 0x04
#define FA_CREATE_ALWAYS 0x08
#define FA_OPEN_ALWAYS 0x10
#define FA_OPEN_APPEND 0x30
/* Fast seek controls (2nd argument of f_lseek) */
#define CREATE_LINKMAP ((FSIZE_t)0 - 1)
/* Format options (2nd argument of f_mkfs) */
#define FM_FAT 0x01
#define FM_FAT32 0x02
#define FM_EXFAT 0x04
#define FM_ANY 0x07
#define FM_SFD 0x08
/* Filesystem type (FATFS.fs_type) */
#define FS_FAT12 1
#define FS_FAT16 2
#define FS_FAT32 3
#define FS_EXFAT 4
/* File attribute bits for directory entry (FILINFO.fattrib) */
#define AM_RDO 0x01 /* Read only */
#define AM_HID 0x02 /* Hidden */
#define AM_SYS 0x04 /* System */
#define AM_DIR 0x10 /* Directory */
#define AM_ARC 0x20 /* Archive */
#ifdef __cplusplus
}
#endif
#endif /* FF_DEFINED */

288
stratosphere/ams_mitm/source/fsp_usb/fatfs/ffconf.h Normal file
View file

@ -0,0 +1,288 @@
/*---------------------------------------------------------------------------/
/ FatFs Functional Configurations
/---------------------------------------------------------------------------*/
#define FFCONF_DEF 86604 /* Revision ID */
/*---------------------------------------------------------------------------/
/ Function Configurations
/---------------------------------------------------------------------------*/
#define FF_FS_READONLY 0
/* This option switches read-only configuration. (0:Read/Write or 1:Read-only)
/ Read-only configuration removes writing API functions, f_write(), f_sync(),
/ f_unlink(), f_mkdir(), f_chmod(), f_rename(), f_truncate(), f_getfree()
/ and optional writing functions as well. */
#define FF_FS_MINIMIZE 0
/* This option defines minimization level to remove some basic API functions.
/
/ 0: Basic functions are fully enabled.
/ 1: f_stat(), f_getfree(), f_unlink(), f_mkdir(), f_truncate() and f_rename()
/ are removed.
/ 2: f_opendir(), f_readdir() and f_closedir() are removed in addition to 1.
/ 3: f_lseek() function is removed in addition to 2. */
#define FF_USE_STRFUNC 0
/* This option switches string functions, f_gets(), f_putc(), f_puts() and f_printf().
/
/ 0: Disable string functions.
/ 1: Enable without LF-CRLF conversion.
/ 2: Enable with LF-CRLF conversion. */
#define FF_USE_FIND 0
/* This option switches filtered directory read functions, f_findfirst() and
/ f_findnext(). (0:Disable, 1:Enable 2:Enable with matching altname[] too) */
#define FF_USE_MKFS 0
/* This option switches f_mkfs() function. (0:Disable or 1:Enable) */
#define FF_USE_FASTSEEK 0
/* This option switches fast seek function. (0:Disable or 1:Enable) */
#define FF_USE_EXPAND 0
/* This option switches f_expand function. (0:Disable or 1:Enable) */
#define FF_USE_CHMOD 0
/* This option switches attribute manipulation functions, f_chmod() and f_utime().
/ (0:Disable or 1:Enable) Also FF_FS_READONLY needs to be 0 to enable this option. */
#define FF_USE_LABEL 1
/* This option switches volume label functions, f_getlabel() and f_setlabel().
/ (0:Disable or 1:Enable) */
#define FF_USE_FORWARD 0
/* This option switches f_forward() function. (0:Disable or 1:Enable) */
/*---------------------------------------------------------------------------/
/ Locale and Namespace Configurations
/---------------------------------------------------------------------------*/
#define FF_CODE_PAGE 932
/* This option specifies the OEM code page to be used on the target system.
/ Incorrect code page setting can cause a file open failure.
/
/ 437 - U.S.
/ 720 - Arabic
/ 737 - Greek
/ 771 - KBL
/ 775 - Baltic
/ 850 - Latin 1
/ 852 - Latin 2
/ 855 - Cyrillic
/ 857 - Turkish
/ 860 - Portuguese
/ 861 - Icelandic
/ 862 - Hebrew
/ 863 - Canadian French
/ 864 - Arabic
/ 865 - Nordic
/ 866 - Russian
/ 869 - Greek 2
/ 932 - Japanese (DBCS)
/ 936 - Simplified Chinese (DBCS)
/ 949 - Korean (DBCS)
/ 950 - Traditional Chinese (DBCS)
/ 0 - Include all code pages above and configured by f_setcp()
*/
#define FF_USE_LFN 2
#define FF_MAX_LFN 255
/* The FF_USE_LFN switches the support for LFN (long file name).
/
/ 0: Disable LFN. FF_MAX_LFN has no effect.
/ 1: Enable LFN with static working buffer on the BSS. Always NOT thread-safe.
/ 2: Enable LFN with dynamic working buffer on the STACK.
/ 3: Enable LFN with dynamic working buffer on the HEAP.
/
/ To enable the LFN, ffunicode.c needs to be added to the project. The LFN function
/ requiers certain internal working buffer occupies (FF_MAX_LFN + 1) * 2 bytes and
/ additional (FF_MAX_LFN + 44) / 15 * 32 bytes when exFAT is enabled.
/ The FF_MAX_LFN defines size of the working buffer in UTF-16 code unit and it can
/ be in range of 12 to 255. It is recommended to be set 255 to fully support LFN
/ specification.
/ When use stack for the working buffer, take care on stack overflow. When use heap
/ memory for the working buffer, memory management functions, ff_memalloc() and
/ ff_memfree() in ffsystem.c, need to be added to the project. */
#define FF_LFN_UNICODE 0
/* This option switches the character encoding on the API when LFN is enabled.
/
/ 0: ANSI/OEM in current CP (TCHAR = char)
/ 1: Unicode in UTF-16 (TCHAR = WCHAR)
/ 2: Unicode in UTF-8 (TCHAR = char)
/ 3: Unicode in UTF-32 (TCHAR = DWORD)
/
/ Also behavior of string I/O functions will be affected by this option.
/ When LFN is not enabled, this option has no effect. */
#define FF_LFN_BUF 255
#define FF_SFN_BUF 12
/* This set of options defines size of file name members in the FILINFO structure
/ which is used to read out directory items. These values should be suffcient for
/ the file names to read. The maximum possible length of the read file name depends
/ on character encoding. When LFN is not enabled, these options have no effect. */
#define FF_STRF_ENCODE 3
/* When FF_LFN_UNICODE >= 1 with LFN enabled, string I/O functions, f_gets(),
/ f_putc(), f_puts and f_printf() convert the character encoding in it.
/ This option selects assumption of character encoding ON THE FILE to be
/ read/written via those functions.
/
/ 0: ANSI/OEM in current CP
/ 1: Unicode in UTF-16LE
/ 2: Unicode in UTF-16BE
/ 3: Unicode in UTF-8
*/
#define FF_FS_RPATH 0
/* This option configures support for relative path.
/
/ 0: Disable relative path and remove related functions.
/ 1: Enable relative path. f_chdir() and f_chdrive() are available.
/ 2: f_getcwd() function is available in addition to 1.
*/
/*---------------------------------------------------------------------------/
/ Drive/Volume Configurations
/---------------------------------------------------------------------------*/
#define FF_VOLUMES 10
/* Number of volumes (logical drives) to be used. (1-10) */
#define FF_STR_VOLUME_ID 1
#define FF_VOLUME_STRS "usb-0","usb-1","usb-2","usb-3","usb-4","usb-5","usb-6","usb-7","usb-8","usb-9"
/* FF_STR_VOLUME_ID switches support for volume ID in arbitrary strings.
/ When FF_STR_VOLUME_ID is set to 1 or 2, arbitrary strings can be used as drive
/ number in the path name. FF_VOLUME_STRS defines the volume ID strings for each
/ logical drives. Number of items must not be less than FF_VOLUMES. Valid
/ characters for the volume ID strings are A-Z, a-z and 0-9, however, they are
/ compared in case-insensitive. If FF_STR_VOLUME_ID >= 1 and FF_VOLUME_STRS is
/ not defined, a user defined volume string table needs to be defined as:
/
/ const char* VolumeStr[FF_VOLUMES] = {"ram","flash","sd","usb",...
*/
#define FF_MULTI_PARTITION 0
/* This option switches support for multiple volumes on the physical drive.
/ By default (0), each logical drive number is bound to the same physical drive
/ number and only an FAT volume found on the physical drive will be mounted.
/ When this function is enabled (1), each logical drive number can be bound to
/ arbitrary physical drive and partition listed in the VolToPart[]. Also f_fdisk()
/ funciton will be available. */
#define FF_MIN_SS 512
#define FF_MAX_SS 512
/* This set of options configures the range of sector size to be supported. (512,
/ 1024, 2048 or 4096) Always set both 512 for most systems, generic memory card and
/ harddisk. But a larger value may be required for on-board flash memory and some
/ type of optical media. When FF_MAX_SS is larger than FF_MIN_SS, FatFs is configured
/ for variable sector size mode and disk_ioctl() function needs to implement
/ GET_SECTOR_SIZE command. */
#define FF_USE_TRIM 0
/* This option switches support for ATA-TRIM. (0:Disable or 1:Enable)
/ To enable Trim function, also CTRL_TRIM command should be implemented to the
/ disk_ioctl() function. */
#define FF_FS_NOFSINFO 0
/* If you need to know correct free space on the FAT32 volume, set bit 0 of this
/ option, and f_getfree() function at first time after volume mount will force
/ a full FAT scan. Bit 1 controls the use of last allocated cluster number.
/
/ bit0=0: Use free cluster count in the FSINFO if available.
/ bit0=1: Do not trust free cluster count in the FSINFO.
/ bit1=0: Use last allocated cluster number in the FSINFO if available.
/ bit1=1: Do not trust last allocated cluster number in the FSINFO.
*/
/*---------------------------------------------------------------------------/
/ System Configurations
/---------------------------------------------------------------------------*/
#define FF_FS_TINY 0
/* This option switches tiny buffer configuration. (0:Normal or 1:Tiny)
/ At the tiny configuration, size of file object (FIL) is shrinked FF_MAX_SS bytes.
/ Instead of private sector buffer eliminated from the file object, common sector
/ buffer in the filesystem object (FATFS) is used for the file data transfer. */
#define FF_FS_EXFAT 1
/* This option switches support for exFAT filesystem. (0:Disable or 1:Enable)
/ To enable exFAT, also LFN needs to be enabled. (FF_USE_LFN >= 1)
/ Note that enabling exFAT discards ANSI C (C89) compatibility. */
#define FF_FS_NORTC 1
#define FF_NORTC_MON 4
#define FF_NORTC_MDAY 12
#define FF_NORTC_YEAR 2019
/* The option FF_FS_NORTC switches timestamp functiton. If the system does not have
/ any RTC function or valid timestamp is not needed, set FF_FS_NORTC = 1 to disable
/ the timestamp function. Every object modified by FatFs will have a fixed timestamp
/ defined by FF_NORTC_MON, FF_NORTC_MDAY and FF_NORTC_YEAR in local time.
/ To enable timestamp function (FF_FS_NORTC = 0), get_fattime() function need to be
/ added to the project to read current time form real-time clock. FF_NORTC_MON,
/ FF_NORTC_MDAY and FF_NORTC_YEAR have no effect.
/ These options have no effect at read-only configuration (FF_FS_READONLY = 1). */
#define FF_FS_LOCK 0
/* The option FF_FS_LOCK switches file lock function to control duplicated file open
/ and illegal operation to open objects. This option must be 0 when FF_FS_READONLY
/ is 1.
/
/ 0: Disable file lock function. To avoid volume corruption, application program
/ should avoid illegal open, remove and rename to the open objects.
/ >0: Enable file lock function. The value defines how many files/sub-directories
/ can be opened simultaneously under file lock control. Note that the file
/ lock control is independent of re-entrancy. */
/* #include <somertos.h> // O/S definitions */
#define FF_FS_REENTRANT 0
#define FF_FS_TIMEOUT 1000
#define FF_SYNC_t HANDLE
/* The option FF_FS_REENTRANT switches the re-entrancy (thread safe) of the FatFs
/ module itself. Note that regardless of this option, file access to different
/ volume is always re-entrant and volume control functions, f_mount(), f_mkfs()
/ and f_fdisk() function, are always not re-entrant. Only file/directory access
/ to the same volume is under control of this function.
/
/ 0: Disable re-entrancy. FF_FS_TIMEOUT and FF_SYNC_t have no effect.
/ 1: Enable re-entrancy. Also user provided synchronization handlers,
/ ff_req_grant(), ff_rel_grant(), ff_del_syncobj() and ff_cre_syncobj()
/ function, must be added to the project. Samples are available in
/ option/syscall.c.
/
/ The FF_FS_TIMEOUT defines timeout period in unit of time tick.
/ The FF_SYNC_t defines O/S dependent sync object type. e.g. HANDLE, ID, OS_EVENT*,
/ SemaphoreHandle_t and etc. A header file for O/S definitions needs to be
/ included somewhere in the scope of ff.h. */
/*--- End of configuration options ---*/

170
stratosphere/ams_mitm/source/fsp_usb/fatfs/ffsystem.c Normal file
View file

@ -0,0 +1,170 @@
/*------------------------------------------------------------------------*/
/* Sample Code of OS Dependent Functions for FatFs */
/* (C)ChaN, 2018 */
/*------------------------------------------------------------------------*/
#include "ff.h"
#include <stdlib.h>
#if FF_USE_LFN == 3 /* Dynamic memory allocation */
/*------------------------------------------------------------------------*/
/* Allocate a memory block */
/*------------------------------------------------------------------------*/
void* ff_memalloc ( /* Returns pointer to the allocated memory block (null if not enough core) */
UINT msize /* Number of bytes to allocate */
)
{
return malloc(msize); /* Allocate a new memory block with POSIX API */
}
/*------------------------------------------------------------------------*/
/* Free a memory block */
/*------------------------------------------------------------------------*/
void ff_memfree (
void* mblock /* Pointer to the memory block to free (nothing to do if null) */
)
{
free(mblock); /* Free the memory block with POSIX API */
}
#endif
#if FF_FS_REENTRANT /* Mutal exclusion */
/*------------------------------------------------------------------------*/
/* Create a Synchronization Object */
/*------------------------------------------------------------------------*/
/* This function is called in f_mount() function to create a new
/ synchronization object for the volume, such as semaphore and mutex.
/ When a 0 is returned, the f_mount() function fails with FR_INT_ERR.
*/
//const osMutexDef_t Mutex[FF_VOLUMES]; /* Table of CMSIS-RTOS mutex */
int ff_cre_syncobj ( /* 1:Function succeeded, 0:Could not create the sync object */
BYTE vol, /* Corresponding volume (logical drive number) */
FF_SYNC_t* sobj /* Pointer to return the created sync object */
)
{
/* Win32 */
*sobj = CreateMutex(NULL, FALSE, NULL);
return (int)(*sobj != INVALID_HANDLE_VALUE);
/* uITRON */
// T_CSEM csem = {TA_TPRI,1,1};
// *sobj = acre_sem(&csem);
// return (int)(*sobj > 0);
/* uC/OS-II */
// OS_ERR err;
// *sobj = OSMutexCreate(0, &err);
// return (int)(err == OS_NO_ERR);
/* FreeRTOS */
// *sobj = xSemaphoreCreateMutex();
// return (int)(*sobj != NULL);
/* CMSIS-RTOS */
// *sobj = osMutexCreate(&Mutex[vol]);
// return (int)(*sobj != NULL);
}
/*------------------------------------------------------------------------*/
/* Delete a Synchronization Object */
/*------------------------------------------------------------------------*/
/* This function is called in f_mount() function to delete a synchronization
/ object that created with ff_cre_syncobj() function. When a 0 is returned,
/ the f_mount() function fails with FR_INT_ERR.
*/
int ff_del_syncobj ( /* 1:Function succeeded, 0:Could not delete due to an error */
FF_SYNC_t sobj /* Sync object tied to the logical drive to be deleted */
)
{
/* Win32 */
return (int)CloseHandle(sobj);
/* uITRON */
// return (int)(del_sem(sobj) == E_OK);
/* uC/OS-II */
// OS_ERR err;
// OSMutexDel(sobj, OS_DEL_ALWAYS, &err);
// return (int)(err == OS_NO_ERR);
/* FreeRTOS */
// vSemaphoreDelete(sobj);
// return 1;
/* CMSIS-RTOS */
// return (int)(osMutexDelete(sobj) == osOK);
}
/*------------------------------------------------------------------------*/
/* Request Grant to Access the Volume */
/*------------------------------------------------------------------------*/
/* This function is called on entering file functions to lock the volume.
/ When a 0 is returned, the file function fails with FR_TIMEOUT.
*/
int ff_req_grant ( /* 1:Got a grant to access the volume, 0:Could not get a grant */
FF_SYNC_t sobj /* Sync object to wait */
)
{
/* Win32 */
return (int)(WaitForSingleObject(sobj, FF_FS_TIMEOUT) == WAIT_OBJECT_0);
/* uITRON */
// return (int)(wai_sem(sobj) == E_OK);
/* uC/OS-II */
// OS_ERR err;
// OSMutexPend(sobj, FF_FS_TIMEOUT, &err));
// return (int)(err == OS_NO_ERR);
/* FreeRTOS */
// return (int)(xSemaphoreTake(sobj, FF_FS_TIMEOUT) == pdTRUE);
/* CMSIS-RTOS */
// return (int)(osMutexWait(sobj, FF_FS_TIMEOUT) == osOK);
}
/*------------------------------------------------------------------------*/
/* Release Grant to Access the Volume */
/*------------------------------------------------------------------------*/
/* This function is called on leaving file functions to unlock the volume.
*/
void ff_rel_grant (
FF_SYNC_t sobj /* Sync object to be signaled */
)
{
/* Win32 */
ReleaseMutex(sobj);
/* uITRON */
// sig_sem(sobj);
/* uC/OS-II */
// OSMutexPost(sobj);
/* FreeRTOS */
// xSemaphoreGive(sobj);
/* CMSIS-RTOS */
// osMutexRelease(sobj);
}
#endif

15597
stratosphere/ams_mitm/source/fsp_usb/fatfs/ffunicode.c Normal file
View file

File diff suppressed because it is too large Load diff

153
stratosphere/ams_mitm/source/fsp_usb/fspusb_drive.cpp Normal file
View file

@ -0,0 +1,153 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <cstdlib>
#include <cstdint>
#include <cstring>
#include <malloc.h>
#include <switch.h>
#include "fspusb_drive.hpp"
static bool g_usbdrive_init = false;
static UsbHsInterfaceFilter g_usbdrive_device_filter;
static Event g_usbdrive_interface_available_event;
// Lock and list of drives, updated regularly
HosMutex g_usbdrive_drives_lock;
std::vector<DriveData> g_usbdrive_drives;
Result USBDriveSystem::Initialize() {
if(g_usbdrive_init) return 0;
Result rc = usbHsInitialize();
if(rc == 0) {
memset(&g_usbdrive_device_filter, 0, sizeof(g_usbdrive_device_filter));
g_usbdrive_device_filter.Flags = UsbHsInterfaceFilterFlags_bInterfaceClass | UsbHsInterfaceFilterFlags_bInterfaceSubClass | UsbHsInterfaceFilterFlags_bInterfaceProtocol;
g_usbdrive_device_filter.bInterfaceClass = 8;
g_usbdrive_device_filter.bInterfaceSubClass = 6;
g_usbdrive_device_filter.bInterfaceProtocol = 80;
rc = usbHsCreateInterfaceAvailableEvent(&g_usbdrive_interface_available_event, false, 0, &g_usbdrive_device_filter);
if(rc == 0) {
g_usbdrive_init = true;
}
}
return rc;
}
bool USBDriveSystem::IsInitialized() {
return g_usbdrive_init;
}
Result USBDriveSystem::WaitForDrives(s64 timeout) {
if(!g_usbdrive_init) return LibnxError_NotInitialized;
return eventWait(&g_usbdrive_interface_available_event, timeout);
}
void USBDriveSystem::Update() {
DRIVES_SCOPE_GUARD;
UsbHsInterface iface_block[DriveMax];
memset(iface_block, 0, sizeof(iface_block));
s32 iface_count = 0;
Result rc = 0;
if(!g_usbdrive_drives.empty()) {
rc = usbHsQueryAcquiredInterfaces(iface_block, sizeof(iface_block), &iface_count);
std::vector<DriveData> valid_drives;
for(u32 i = 0; i < g_usbdrive_drives.size(); i++) {
// For each drive in our list, check whether it is still available (by looping through actual acquired interfaces)
bool ok = false;
for(s32 j = 0; j < iface_count; j++) {
if(iface_block[j].inf.ID == g_usbdrive_drives[i].usbif.ID) {
ok = true;
break;
}
}
if(ok) {
valid_drives.push_back(g_usbdrive_drives[i]);
}
else {
f_mount(NULL, g_usbdrive_drives[i].mountname, 1);
usbHsIfResetDevice(&g_usbdrive_drives[i].usbif); // Reset and close later to ensure the drive is no longer acquired
usbHsEpClose(&g_usbdrive_drives[i].usbinep);
usbHsEpClose(&g_usbdrive_drives[i].usboutep);
usbHsIfClose(&g_usbdrive_drives[i].usbif);
}
}
g_usbdrive_drives = valid_drives;
}
memset(iface_block, 0, sizeof(iface_block));
rc = usbHsQueryAvailableInterfaces(&g_usbdrive_device_filter, iface_block, sizeof(iface_block), &iface_count);
// Now, check new ones and (try to) acquire them
if(rc == 0) {
for(s32 i = 0; i < iface_count; i++) {
DriveData dt;
rc = usbHsAcquireUsbIf(&dt.usbif, &iface_block[i]);
if(rc == 0) {
for(u32 j = 0; j < 15; j++) {
auto epd = &dt.usbif.inf.inf.input_endpoint_descs[j];
if(epd->bLength > 0) {
rc = usbHsIfOpenUsbEp(&dt.usbif, &dt.usbinep, 1, epd->wMaxPacketSize, epd);
break;
}
}
if(rc == 0) {
for(u32 j = 0; j < 15; j++) {
auto epd = &dt.usbif.inf.inf.output_endpoint_descs[j];
if(epd->bLength > 0) {
rc = usbHsIfOpenUsbEp(&dt.usbif, &dt.usboutep, 1, epd->wMaxPacketSize, epd);
break;
}
}
if(rc == 0) {
dt.device = std::make_shared<SCSIDevice>(&dt.usbif, &dt.usbinep, &dt.usboutep);
dt.scsi = std::make_shared<SCSIBlock>(dt.device);
memset(dt.mountname, 0, 0x10);
u32 idx = g_usbdrive_drives.size();
sprintf(dt.mountname, "usb-%d:", idx);
memset(&dt.fatfs, 0, sizeof(dt.fatfs));
g_usbdrive_drives.push_back(dt);
// Mount in the already added vector element since otherwise the mount data (fs type, sizes...) will be copied into the local var and not the vector item
auto fres = f_mount(&g_usbdrive_drives[idx].fatfs, g_usbdrive_drives[idx].mountname, 1);
if(fres != FR_OK) {
g_usbdrive_drives.pop_back();
}
}
}
}
}
}
}
void USBDriveSystem::Finalize() {
if(!g_usbdrive_init) return;
usbHsDestroyInterfaceAvailableEvent(&g_usbdrive_interface_available_event, 0);
usbHsExit();
g_usbdrive_init = false;
}
u32 USBDriveSystem::GetDriveCount() {
DRIVES_SCOPE_GUARD;
return g_usbdrive_drives.size();
}
u8 USBDriveSystem::GetFSType(u32 index) {
DRIVES_SCOPE_GUARD;
if(index >= g_usbdrive_drives.size()) {
return 0;
}
return g_usbdrive_drives[index].fatfs.fs_type;
}

61
stratosphere/ams_mitm/source/fsp_usb/fspusb_drive.hpp Normal file
View file

@ -0,0 +1,61 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "fatfs/ff.h"
#include "fspusb_scsi_context.hpp"
static constexpr u32 DriveMax = 10; // FATFS drive limit
#define DRIVES_SCOPE_GUARD std::scoped_lock<HosMutex> lck(g_usbdrive_drives_lock);
class FspUsbResults {
public:
static constexpr u32 Module = 476;
static constexpr Result InvalidDriveIndex = MAKERESULT(Module, 1);
static constexpr Result DriveUnavailable = MAKERESULT(Module, 2);
static constexpr Result MakeFATFSErrorResult(FRESULT Res) { // Since most operations are FS one with FATFS, had to make this
if(Res == FR_OK) {
return 0;
}
return MAKERESULT(Module, 100 + (u32)Res);
}
};
struct DriveData {
UsbHsClientIfSession usbif;
UsbHsClientEpSession usbinep;
UsbHsClientEpSession usboutep;
std::shared_ptr<SCSIDevice> device;
std::shared_ptr<SCSIBlock> scsi;
bool status;
FATFS fatfs;
char mountname[0x10];
};
class USBDriveSystem {
public:
static Result Initialize();
static bool IsInitialized();
static void Update();
static Result WaitForDrives(s64 timeout = -1);
static void Finalize();
static u32 GetDriveCount();
static u8 GetFSType(u32 index);
};

65
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivedirectory.cpp Normal file
View file

@ -0,0 +1,65 @@
#include "fspusb_drivedirectory.hpp"
extern HosMutex g_usbdrive_drives_lock;
extern std::vector<DriveData> g_usbdrive_drives;
bool DriveDirectory::IsOk() {
DRIVES_SCOPE_GUARD;
for(u32 i = 0; i < g_usbdrive_drives.size(); i++) {
if(g_usbdrive_drives[i].usbif.ID == usbid) {
return true;
}
}
return false;
}
Result DriveDirectory::ReadImpl(uint64_t *out_count, FsDirectoryEntry *out_entries, uint64_t max_entries) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
FsDirectoryEntry ent;
FRESULT res = FR_OK;
FILINFO info;
u32 count = 0;
while(true) {
if(count >= max_entries) {
break;
}
res = f_readdir(&dp, &info);
if((res != FR_OK) || (info.fname[0] == '\0')) {
break;
}
memset(&ent, 0, sizeof(FsDirectoryEntry));
strcpy(ent.name, info.fname);
// Fill in the DirectoryEntry struct, then copy back to the buffer
if(info.fattrib & AM_DIR) {
ent.type = ENTRYTYPE_DIR;
}
else {
ent.type = ENTRYTYPE_FILE;
ent.fileSize = info.fsize;
}
memcpy(&out_entries[count], &ent, sizeof(ent));
count++;
}
*out_count = count;
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveDirectory::GetEntryCountImpl(uint64_t *count) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
u64 entcount = 0;
FRESULT res = FR_OK;
FILINFO info;
while(true) {
res = f_readdir(&dp, &info);
if((res != FR_OK) || (info.fname[0] == '\0')) {
break;
}
entcount++;
}
*count = entcount;
return FspUsbResults::MakeFATFSErrorResult(res); // Shouldn't be enough to return 0?
}

43
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivedirectory.hpp Normal file
View file

@ -0,0 +1,43 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "../fs_mitm/fs_idirectory.hpp"
#include "fspusb_drive.hpp"
class DriveDirectory : public IDirectory {
private:
DIR dp;
s32 usbid;
public:
DriveDirectory(DIR fatdp, DriveData *drv) : dp(fatdp) {
usbid = drv->usbif.ID;
}
~DriveDirectory() {
f_closedir(&dp);
}
bool IsOk();
public:
virtual Result ReadImpl(uint64_t *out_count, FsDirectoryEntry *out_entries, uint64_t max_entries) override;
virtual Result GetEntryCountImpl(uint64_t *count) override;
};

68
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivefile.cpp Normal file
View file

@ -0,0 +1,68 @@
#include "fspusb_drivefile.hpp"
extern HosMutex g_usbdrive_drives_lock;
extern std::vector<DriveData> g_usbdrive_drives;
bool DriveFile::IsOk() {
DRIVES_SCOPE_GUARD;
// Before any command, check whether the file's drive is still accessible
for(u32 i = 0; i < g_usbdrive_drives.size(); i++) {
if(g_usbdrive_drives[i].usbif.ID == usbid) {
return true;
}
}
return false;
}
Result DriveFile::ReadImpl(u64 *out, u64 offset, void *buffer, u64 size) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_lseek(&fp, offset);
if(res == FR_OK) {
res = f_read(&fp, buffer, size, (UINT*)out);
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFile::GetSizeImpl(u64 *out) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
*out = f_size(&fp);
return 0;
}
Result DriveFile::FlushImpl() {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
return 0;
}
Result DriveFile::WriteImpl(u64 offset, void *buffer, u64 size, u32 option) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_lseek(&fp, offset);
if(res == FR_OK) {
u64 out;
res = f_write(&fp, buffer, size, (UINT*)&out);
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFile::SetSizeImpl(u64 size) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_lseek(&fp, size);
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFile::OperateRangeImpl(FsOperationId operation_type, u64 offset, u64 size, FsRangeInfo *out_range_info) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
return 0;
}

47
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivefile.hpp Normal file
View file

@ -0,0 +1,47 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "../fs_mitm/fs_ifile.hpp"
#include "fspusb_drive.hpp"
class DriveFile : public IFile {
private:
FIL fp;
s32 usbid;
public:
DriveFile(FIL fatfp, DriveData *drv) : fp(fatfp) {
usbid = drv->usbif.ID;
}
~DriveFile() {
f_close(&fp);
}
bool IsOk();
public:
virtual Result ReadImpl(u64 *out, u64 offset, void *buffer, u64 size) override;
virtual Result GetSizeImpl(u64 *out) override;
virtual Result FlushImpl() override;
virtual Result WriteImpl(u64 offset, void *buffer, u64 size, u32 option) override;
virtual Result SetSizeImpl(u64 size) override;
virtual Result OperateRangeImpl(FsOperationId operation_type, u64 offset, u64 size, FsRangeInfo *out_range_info) override;
};

232
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivefilesystem.cpp Normal file
View file

@ -0,0 +1,232 @@
#include "fspusb_drivefilesystem.hpp"
extern HosMutex g_usbdrive_drives_lock;
extern std::vector<DriveData> g_usbdrive_drives;
DriveData *DriveFileSystem::GetDriveAccess() {
DRIVES_SCOPE_GUARD;
if(drvidx >= g_usbdrive_drives.size()) {
return NULL;
}
return &g_usbdrive_drives[drvidx];
}
bool DriveFileSystem::IsOk() {
DRIVES_SCOPE_GUARD;
for(u32 i = 0; i < g_usbdrive_drives.size(); i++) {
if(g_usbdrive_drives[i].usbif.ID == usbid) {
return true;
}
}
return false;
}
Result DriveFileSystem::CreateFileImpl(const FsPath &path, uint64_t size, int flags) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
FIL fp;
auto res = f_open(&fp, pth.c_str(), FA_CREATE_NEW | FA_WRITE);
if(res == FR_OK) {
f_lseek(&fp, size);
f_close(&fp);
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::DeleteFileImpl(const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_unlink(pth.c_str());
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::CreateDirectoryImpl(const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_mkdir(pth.c_str());
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::DeleteDirectoryImpl(const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_rmdir(pth.c_str());
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::DeleteDirectoryRecursivelyImpl(const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
// TODO: do this!
auto res = FR_OK;
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::RenameFileImpl(const FsPath &old_path, const FsPath &new_path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(old_path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
std::string newpth = GetFullPath(new_path);
if(newpth.empty()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_rename(pth.c_str(), newpth.c_str());
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::RenameDirectoryImpl(const FsPath &old_path, const FsPath &new_path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(old_path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
std::string newpth = GetFullPath(new_path);
if(newpth.empty()) {
return FspUsbResults::DriveUnavailable;
}
auto res = f_rename(pth.c_str(), newpth.c_str());
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::GetEntryTypeImpl(DirectoryEntryType *out, const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
FILINFO info;
auto res = f_stat(pth.c_str(), &info);
if(res == FR_OK) {
if(info.fattrib & AM_DIR) {
*out = DirectoryEntryType_Directory;
}
else {
*out = DirectoryEntryType_File;
}
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::OpenFileImpl(std::unique_ptr<IFile> &out_file, const FsPath &path, OpenMode mode) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
FIL fp;
auto res = f_open(&fp, pth.c_str(), FA_OPEN_EXISTING | FA_READ | FA_WRITE);
if(res == FR_OK) {
out_file = std::make_unique<DriveFile>(fp, GetDriveAccess());
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::OpenDirectoryImpl(std::unique_ptr<IDirectory> &out_dir, const FsPath &path, DirectoryOpenMode mode) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
std::string pth = GetFullPath(path);
if(pth.empty()) {
return FspUsbResults::DriveUnavailable;
}
DIR dp;
auto res = f_opendir(&dp, pth.c_str());
if(res == FR_OK) {
out_dir = std::make_unique<DriveDirectory>(dp, GetDriveAccess());
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::CommitImpl() {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
return 0;
}
Result DriveFileSystem::GetFreeSpaceSizeImpl(uint64_t *out, const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
FATFS *fs = NULL;
DWORD clstrs = 0;
auto res = f_getfree(GetDriveAccess()->mountname, &clstrs, &fs);
if(res == FR_OK) {
if(fs) {
*out = (uint64_t)(clstrs * fs->csize);
}
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::GetTotalSpaceSizeImpl(uint64_t *out, const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
FATFS *fs = NULL;
DWORD clstrs = 0;
auto res = f_getfree(GetDriveAccess()->mountname, &clstrs, &fs);
if(res == FR_OK) {
if(fs) {
*out = (uint64_t)((fs->n_fatent - 2) * fs->csize);
}
}
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result DriveFileSystem::CleanDirectoryRecursivelyImpl(const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
// Handle this!
return 0;
}
Result DriveFileSystem::GetFileTimeStampRawImpl(FsTimeStampRaw *out, const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
return 0;
}
Result DriveFileSystem::QueryEntryImpl(char *out, uint64_t out_size, const char *in, uint64_t in_size, int query, const FsPath &path) {
if(!IsOk()) {
return FspUsbResults::DriveUnavailable;
}
return 0;
}

67
stratosphere/ams_mitm/source/fsp_usb/fspusb_drivefilesystem.hpp Normal file
View file

@ -0,0 +1,67 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "../fs_mitm/fs_ifilesystem.hpp"
#include "fspusb_drivefile.hpp"
#include "fspusb_drivedirectory.hpp"
class DriveFileSystem : public IFileSystem {
private:
u32 drvidx;
s32 usbid;
public:
DriveFileSystem(u32 idx) : drvidx(idx) {
usbid = GetDriveAccess()->usbif.ID;
}
DriveData *GetDriveAccess();
bool IsOk();
std::string GetFullPath(const FsPath &path) {
// Convert fsp path to FATFS path, acording to the drive's mount name.
DriveData *data = GetDriveAccess();
if(data == NULL) {
return "";
}
std::string str = std::string(data->mountname) + std::string(path.str);
// Debug
printf("GetFullPath (path = \"%s\")\n", str.c_str());
return str;
}
public:
virtual Result CreateFileImpl(const FsPath &path, uint64_t size, int flags) override;
virtual Result DeleteFileImpl(const FsPath &path) override;
virtual Result CreateDirectoryImpl(const FsPath &path) override;
virtual Result DeleteDirectoryImpl(const FsPath &path) override;
virtual Result DeleteDirectoryRecursivelyImpl(const FsPath &path) override;
virtual Result RenameFileImpl(const FsPath &old_path, const FsPath &new_path) override;
virtual Result RenameDirectoryImpl(const FsPath &old_path, const FsPath &new_path) override;
virtual Result GetEntryTypeImpl(DirectoryEntryType *out, const FsPath &path) override;
virtual Result OpenFileImpl(std::unique_ptr<IFile> &out_file, const FsPath &path, OpenMode mode) override;
virtual Result OpenDirectoryImpl(std::unique_ptr<IDirectory> &out_dir, const FsPath &path, DirectoryOpenMode mode) override;
virtual Result CommitImpl() override;
virtual Result GetFreeSpaceSizeImpl(uint64_t *out, const FsPath &path) override;
virtual Result GetTotalSpaceSizeImpl(uint64_t *out, const FsPath &path) override;
virtual Result CleanDirectoryRecursivelyImpl(const FsPath &path) override;
virtual Result GetFileTimeStampRawImpl(FsTimeStampRaw *out, const FsPath &path) override;
virtual Result QueryEntryImpl(char *out, uint64_t out_size, const char *in, uint64_t in_size, int query, const FsPath &path) override;
};

70
stratosphere/ams_mitm/source/fsp_usb/fspusb_main.cpp Normal file
View file

@ -0,0 +1,70 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <cstdlib>
#include <cstdint>
#include <cstring>
#include <malloc.h>
#include <switch.h>
#include <atmosphere.h>
#include <stratosphere.hpp>
#include "fspusb_main.hpp"
#include "fspusb_service.hpp"
// fsp-srv's ones, but without domains (need this specific buf size as we're using fsp-like IFileSystems)
struct FspUsbManagerOptions {
static const size_t PointerBufferSize = 0x800;
static const size_t MaxDomains = 0;
static const size_t MaxDomainObjects = 0;
};
using FspUsbManager = WaitableManager<FspUsbManagerOptions>;
void USBUpdateLoop(void *arg) {
while(true) {
USBDriveSystem::Update();
svcSleepThread(100000000L);
}
}
void FspUsbMain(void *arg) {
/* Wait for initialization to occur */
Utils::WaitSdInitialized();
Result rc;
/* Initialize USB system (usb:hs service). */
DoWithSmSession([&]() {
do
{
rc = USBDriveSystem::Initialize();
} while(rc != 0);
});
/* Create server manager */
static auto s_server_manager = FspUsbManager(5);
HosThread t_usb_loop;
t_usb_loop.Initialize(&USBUpdateLoop, nullptr, 0x4000, 0x2b);
t_usb_loop.Start();
/* Create fsp-usb. */
s_server_manager.AddWaitable(new ServiceServer<FspUsbService>("fsp-usb", 0x20));
/* Loop forever, servicing our services. */
s_server_manager.Process();
}

29
stratosphere/ams_mitm/source/fsp_usb/fspusb_main.hpp Normal file
View file

@ -0,0 +1,29 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <cstdlib>
#include <cstdint>
#include <cstring>
#include <malloc.h>
#include <switch.h>
#include <atmosphere.h>
#include <stratosphere.hpp>
constexpr u32 FspUsbPriority = 43;
constexpr u32 FspUsbStackSize = 0x8000;
void FspUsbMain(void *arg);

279
stratosphere/ams_mitm/source/fsp_usb/fspusb_scsi_command.hpp Normal file
View file

@ -0,0 +1,279 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <cstdarg>
#include "fatfs/ff.h"
static void clog(const char *fmt, ...)
{
/*
va_list args;
va_start(args, fmt);
FILE *f = fopen("sdmc:/fspusb-log.log", "a");
if(f) {
vfprintf(f, fmt, args);
fclose(f);
}
va_end(args);
*/
}
#define printf clog
enum class SCSIDirection
{
None,
In,
Out
};
#define CBW_SIGNATURE 0x43425355
#define COMMAND_PASSED 0
#define COMMAND_FAILED 1
#define PHASE_ERROR 2
#define CSW_SIZE 13
#define CSW_SIGNATURE 0x53425355
class SCSIBuffer
{
public:
int idx = 0;
uint8_t storage[31];
SCSIBuffer()
{
memset(storage, 0, sizeof(storage));
}
void write_u8(uint8_t e)
{
memcpy(&storage[idx], &e, sizeof(e));
idx++;
}
void padding(int n)
{
idx += n;
}
void write_u16_be(uint16_t e)
{
e = __builtin_bswap16(e);
memcpy(&storage[idx], &e, sizeof(e));
idx += 2;
}
void write_u32(uint32_t e)
{
memcpy(&storage[idx], &e, sizeof(e));
idx += 4;
}
void write_u32_be(uint32_t e)
{
e = __builtin_bswap32(e);
memcpy(&storage[idx], &e, sizeof(e));
idx += 4;
}
};
class SCSICommand
{
public:
uint32_t tag;
uint32_t data_transfer_length;
uint8_t flags;
uint8_t lun;
uint8_t cb_length;
SCSIDirection direction;
SCSICommand(uint32_t data_transfer_length_, SCSIDirection direction_, uint8_t lun_, uint8_t cb_length_)
{
tag = 0; // should this be default?
data_transfer_length = data_transfer_length_;
direction = direction_;
lun = lun_;
cb_length = cb_length_;
if(direction == SCSIDirection::In)
{
flags = 0x80;
}
else
{
flags = 0;
}
}
virtual void to_bytes(uint8_t *out) = 0;
void write_header(SCSIBuffer &out)
{
out.write_u32(CBW_SIGNATURE);
out.write_u32(tag);
out.write_u32(data_transfer_length);
out.write_u8(flags);
out.write_u8(lun);
out.write_u8(cb_length);
};
};
class SCSIInquiryCommand : public SCSICommand
{
public:
uint8_t allocation_length;
uint8_t opcode;
SCSIInquiryCommand(uint8_t allocation_length_) : SCSICommand(allocation_length_, SCSIDirection::In, 0, /* length */ 5)
{
opcode = 0x12;
allocation_length = allocation_length_;
}
virtual void to_bytes(uint8_t *out)
{
SCSIBuffer b;
write_header(b);
b.write_u8(opcode);
b.padding(3);
b.write_u8(allocation_length);
memcpy(out, b.storage, 31);
}
};
class SCSITestUnitReadyCommand : public SCSICommand
{
public:
uint8_t opcode;
SCSITestUnitReadyCommand() : SCSICommand(0, SCSIDirection::None, 0, /* length */ 0x6)
{
opcode = 0;
}
virtual void to_bytes(uint8_t *out)
{
SCSIBuffer b;
write_header(b);
b.write_u8(opcode);
memcpy(out, b.storage, 31);
}
};
class SCSIReadCapacityCommand : public SCSICommand
{
public:
uint8_t opcode;
SCSIReadCapacityCommand() : SCSICommand(0x8, SCSIDirection::In, 0, /* length */ 0x10)
{
opcode = 0x25;
}
virtual void to_bytes(uint8_t *out)
{
SCSIBuffer b;
write_header(b);
b.write_u8(opcode);
memcpy(out, b.storage, 31);
}
};
class SCSIRead10Command : public SCSICommand
{
public:
uint8_t opcode;
uint32_t block_address;
uint16_t transfer_blocks;
SCSIRead10Command(uint32_t block_address_, uint32_t block_size, uint16_t transfer_blocks_) : SCSICommand(transfer_blocks_ * block_size, SCSIDirection::In, 0, /* length */ 10)
{
opcode = 0x28;
block_address = block_address_;
transfer_blocks = transfer_blocks_;
}
virtual void to_bytes(uint8_t *out)
{
SCSIBuffer b;
write_header(b);
b.write_u8(opcode);
b.padding(1);
b.write_u32_be(block_address);
b.padding(1);
b.write_u16_be(transfer_blocks);
b.padding(1);
memcpy(out, b.storage, 31);
}
};
class SCSIWrite10Command : public SCSICommand
{
public:
uint8_t opcode;
uint32_t block_address;
uint16_t transfer_blocks;
SCSIWrite10Command(uint32_t block_address_, uint32_t block_size, uint16_t transfer_blocks_) : SCSICommand(transfer_blocks_ * block_size, SCSIDirection::Out, 0, /* length */ 10)
{
opcode = 0x2a;
block_address = block_address_;
transfer_blocks = transfer_blocks_;
}
virtual void to_bytes(uint8_t *out)
{
SCSIBuffer b;
write_header(b);
b.write_u8(opcode);
b.padding(1);
b.write_u32_be(block_address);
b.padding(1);
b.write_u16_be(transfer_blocks);
b.padding(1);
memcpy(out, b.storage, 31);
}
};
class SCSICommandStatus
{
public:
uint32_t signature;
uint32_t tag;
uint32_t data_residue;
uint8_t status;
SCSICommandStatus(){}
SCSICommandStatus(uint8_t csw[13])
{
memcpy(&signature, &csw[0], 4);
memcpy(&tag, &csw[4], 4);
memcpy(&data_residue, &csw[8], 4);
memcpy(&status, &csw[12], 1);
}
};

191
stratosphere/ams_mitm/source/fsp_usb/fspusb_scsi_context.cpp Normal file
View file

@ -0,0 +1,191 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <switch.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <cstdarg>
#include "fspusb_scsi_context.hpp"
SCSICommandStatus SCSIDevice::read_csw() {
uint32_t num_bytes;
Result res = usbHsEpPostBuffer(out_endpoint, usb_bounce_buffer_c, 0x10, &num_bytes);
if(R_FAILED(res))
{
printf("read_csw usb fail (handle: %d) %08x\n", out_endpoint->s.handle, res);
}
if(num_bytes != 13)
{
printf("read_csw usb short read of length %i\n", num_bytes);
}
return SCSICommandStatus(usb_bounce_buffer_c);
}
void SCSIDevice::push_cmd(SCSICommand *c) {
// Push a 31 byte command.
uint32_t num_bytes;
memset(usb_bounce_buffer_a, 0, 0x1000);
c->to_bytes(usb_bounce_buffer_a);
Result res = usbHsEpPostBuffer(in_endpoint, usb_bounce_buffer_a, 31, &num_bytes);
if(R_FAILED(res))
{
printf("push_cmd usb fail %08x\n", res);
return;
}
if(num_bytes != 31)
{
printf("push_cmd usb fail short write of %i bytes????\n", num_bytes);
return;
}
}
SCSICommandStatus SCSIDevice::transfer_cmd(SCSICommand *c, uint8_t *buffer, size_t buffer_size) {
push_cmd(c);
uint32_t transfer_length = c->data_transfer_length;
uint32_t transferred = 0;
uint32_t total_transferred = 0;
if(buffer_size < transfer_length)
{
printf("Buffer too small!!!!\n");
}
Result res;
if(transfer_length > 0)
{
if(c->direction == SCSIDirection::In)
{
while(total_transferred < transfer_length)
{
memset(usb_bounce_buffer_b, 0, 0x1000);
res = usbHsEpPostBuffer(out_endpoint, usb_bounce_buffer_b, transfer_length - total_transferred, &transferred);
if(R_FAILED(res))
{
printf("usbHsEpPostBuffer failed %08x\n", res);
}
if(transferred == 13)
{
uint32_t signature;
memcpy(&signature, usb_bounce_buffer_b, 4);
if(signature == CSW_SIGNATURE)
{
// We weren't expecting a CSW!
// But we got one anyway!
return SCSICommandStatus(usb_bounce_buffer_b);
}
}
memcpy(buffer + total_transferred, usb_bounce_buffer_b, transferred);
total_transferred += transferred;
}
}
else
{
while(total_transferred < transfer_length)
{
memcpy(usb_bounce_buffer_b, buffer + total_transferred, transfer_length - total_transferred);
res = usbHsEpPostBuffer(in_endpoint, usb_bounce_buffer_b, transfer_length - total_transferred, &transferred);
if(R_FAILED(res))
{
printf("usbHsEpPostBuffer failed %08x\n", res);
}
total_transferred += transferred;
}
}
}
SCSICommandStatus w = read_csw();
if(w.tag != c->tag)
{
// ???
}
if(w.status != COMMAND_PASSED)
{
// ???
}
return w;
}
SCSIBlock::SCSIBlock(std::shared_ptr<SCSIDevice> device_) {
device = device_;
SCSIInquiryCommand inquiry(36);
SCSITestUnitReadyCommand test_unit_ready;
SCSIReadCapacityCommand read_capacity;
uint8_t inquiry_response[36];
SCSICommandStatus status = device->transfer_cmd(&inquiry, inquiry_response, 36);
status = device->transfer_cmd(&test_unit_ready, NULL, 0);
uint8_t read_capacity_response[8];
uint32_t size_lba;
uint32_t lba_bytes;
status = device->transfer_cmd(&read_capacity, read_capacity_response, 8);
memcpy(&size_lba, &read_capacity_response[0], 4);
size_lba = __builtin_bswap32(size_lba);
memcpy(&lba_bytes, &read_capacity_response[4], 4);
lba_bytes = __builtin_bswap32(lba_bytes);
capacity = size_lba * lba_bytes;
block_size = lba_bytes;
uint8_t mbr[0x200];
read_sectors(mbr, 0, 1);
partition_infos[0] = MBRPartition::from_bytes(&mbr[0x1be]);
partition_infos[1] = MBRPartition::from_bytes(&mbr[0x1ce]);
partition_infos[2] = MBRPartition::from_bytes(&mbr[0x1de]);
partition_infos[3] = MBRPartition::from_bytes(&mbr[0x1ee]);
for(int i = 0; i < 4; i++)
{
MBRPartition p = partition_infos[i];
if(p.partition_type != 0)
{
partitions[i] = SCSIBlockPartition(this, p);
}
}
}
int SCSIBlock::read_sectors(uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors) {
SCSIRead10Command read_ten(sector_offset, block_size, num_sectors);
SCSICommandStatus status = device->transfer_cmd(&read_ten, buffer, num_sectors * block_size);
return num_sectors;
}
int SCSIBlock::write_sectors(const uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors) {
SCSIWrite10Command write_ten(sector_offset, block_size, num_sectors);
SCSICommandStatus status = device->transfer_cmd(&write_ten, (uint8_t*)buffer, num_sectors * block_size);
return num_sectors;
}
SCSIBlockPartition::SCSIBlockPartition(SCSIBlock *block_, MBRPartition partition_info) {
block = block_;
start_lba_offset = partition_info.start_lba;
lba_size = partition_info.num_sectors;
}
int SCSIBlockPartition::read_sectors(uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors) {
// TODO: assert we don't read outside the partition
return block->read_sectors(buffer, sector_offset + start_lba_offset, num_sectors);
}
int SCSIBlockPartition::write_sectors(const uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors) {
// TODO: assert we don't write (!!) outside the partition
return block->write_sectors(buffer, sector_offset + start_lba_offset, num_sectors);
}

99
stratosphere/ams_mitm/source/fsp_usb/fspusb_scsi_context.hpp Normal file
View file

@ -0,0 +1,99 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <malloc.h>
#include <memory>
#include "fspusb_scsi_command.hpp"
class SCSIDevice
{
public:
uint8_t *usb_bounce_buffer_a;
uint8_t *usb_bounce_buffer_b;
uint8_t *usb_bounce_buffer_c;
UsbHsClientIfSession *client;
UsbHsClientEpSession *in_endpoint;
UsbHsClientEpSession *out_endpoint;
SCSIDevice(){}
SCSIDevice(UsbHsClientIfSession *client_, UsbHsClientEpSession *in_endpoint_, UsbHsClientEpSession *out_endpoint_) {
usb_bounce_buffer_a = (uint8_t*) memalign(0x1000, 0x1000);
usb_bounce_buffer_b = (uint8_t*) memalign(0x1000, 0x1000);
usb_bounce_buffer_c = (uint8_t*) memalign(0x1000, 0x1000);
client = client_;
in_endpoint = in_endpoint_;
out_endpoint = out_endpoint_;
}
SCSICommandStatus read_csw();
void push_cmd(SCSICommand *c);
SCSICommandStatus transfer_cmd(SCSICommand *c, uint8_t *buffer, size_t buffer_size);
};
class MBRPartition
{
public:
uint8_t active;
uint8_t partition_type;
uint32_t start_lba;
uint32_t num_sectors;
static MBRPartition from_bytes(uint8_t *entry)
{
MBRPartition p;
p.active = entry[0];
// 1 - 3 are chs start, skip them
p.partition_type = entry[4];
// 5 - 7 are chs end, skip them
memcpy(&p.start_lba, &entry[8], 4);
memcpy(&p.num_sectors, &entry[12], 4);
return p;
}
};
class SCSIBlock;
class SCSIBlockPartition
{
public:
SCSIBlock *block;
uint32_t start_lba_offset;
uint32_t lba_size;
SCSIBlockPartition() {}
SCSIBlockPartition(SCSIBlock *block_, MBRPartition partition_info);
int read_sectors(uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors);
int write_sectors(const uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors);
};
class SCSIBlock
{
public:
uint64_t capacity;
uint32_t block_size;
SCSIBlockPartition partitions[4];
MBRPartition partition_infos[4];
std::shared_ptr<SCSIDevice> device;
SCSIBlock(){}
SCSIBlock(std::shared_ptr<SCSIDevice> device_);
int read_sectors(uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors);
int write_sectors(const uint8_t *buffer, uint32_t sector_offset, uint32_t num_sectors);
};

66
stratosphere/ams_mitm/source/fsp_usb/fspusb_service.cpp Normal file
View file

@ -0,0 +1,66 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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 <switch.h>
#include <cstdio>
#include <algorithm>
#include <stratosphere.hpp>
#include "fspusb_service.hpp"
Result FspUsbService::UpdateDrives(Out<u32> count) {
USBDriveSystem::Update();
u32 drvcount = USBDriveSystem::GetDriveCount();
count.SetValue(drvcount);
return 0;
}
Result FspUsbService::OpenDriveFileSystem(u32 index, Out<std::shared_ptr<IFileSystemInterface>> out) {
if(index >= USBDriveSystem::GetDriveCount()) {
return FspUsbResults::InvalidDriveIndex;
}
out.SetValue(std::make_shared<IFileSystemInterface>(std::make_shared<DriveFileSystem>(index)));
return 0;
}
Result FspUsbService::GetDriveLabel(u32 index, OutBuffer<char> out) {
if(index >= USBDriveSystem::GetDriveCount()) {
return FspUsbResults::InvalidDriveIndex;
}
char drivemount[0x10] = {0};
sprintf(drivemount, "%d:", index);
auto res = f_getlabel(drivemount, out.buffer, NULL);
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result FspUsbService::SetDriveLabel(u32 index, InBuffer<char> label) {
if(index >= USBDriveSystem::GetDriveCount()) {
return FspUsbResults::InvalidDriveIndex;
}
char drivestr[0x100] = {0};
sprintf(drivestr, "usb-%d:%s", index, label.buffer);
auto res = f_setlabel(drivestr);
return FspUsbResults::MakeFATFSErrorResult(res);
}
Result FspUsbService::GetDriveFileSystemType(u32 index, Out<u32> out) {
if(index >= USBDriveSystem::GetDriveCount()) {
return FspUsbResults::InvalidDriveIndex;
}
u8 type = USBDriveSystem::GetFSType(index);
out.SetValue((u32)type);
return 0;
}

47
stratosphere/ams_mitm/source/fsp_usb/fspusb_service.hpp Normal file
View file

@ -0,0 +1,47 @@
/*
* Copyright (c) 2018-2019 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope 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/>.
*/
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "fspusb_drive.hpp"
#include "fspusb_drivefilesystem.hpp"
class FspUsbService final : public IServiceObject {
protected:
enum class CommandId {
UpdateDrives = 0,
OpenDriveFileSystem = 1,
GetDriveLabel = 2,
SetDriveLabel = 3,
GetDriveFileSystemType = 4,
};
Result UpdateDrives(Out<u32> count);
Result OpenDriveFileSystem(u32 index, Out<std::shared_ptr<IFileSystemInterface>> out);
Result GetDriveLabel(u32 index, OutBuffer<char> out);
Result SetDriveLabel(u32 index, InBuffer<char> label);
Result GetDriveFileSystemType(u32 index, Out<u32> out);
public:
DEFINE_SERVICE_DISPATCH_TABLE {
MAKE_SERVICE_COMMAND_META(FspUsbService, UpdateDrives),
MAKE_SERVICE_COMMAND_META(FspUsbService, OpenDriveFileSystem),
MAKE_SERVICE_COMMAND_META(FspUsbService, GetDriveLabel),
MAKE_SERVICE_COMMAND_META(FspUsbService, SetDriveLabel),
MAKE_SERVICE_COMMAND_META(FspUsbService, GetDriveFileSystemType)
};
};