controller_patcher/utils/FSHelper.c
2017-05-20 15:55:51 +02:00

79 lines
2.4 KiB
C

#include <malloc.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <wiiu/fs.h>
#define FS_MAX_MOUNTPATH_SIZE 128
int FS_Helper_MountFS(void *pClient, void *pCmd, char **mount_path){
int result = -1;
void *mountSrc = malloc(sizeof(FSMountSource));
if(!mountSrc)
return -3;
char* mountPath = (char*) malloc(FS_MAX_MOUNTPATH_SIZE);
if(!mountPath) {
free(mountSrc);
return -4;
}
memset(mountSrc, 0, sizeof(FSMountSource));
memset(mountPath, 0, FS_MAX_MOUNTPATH_SIZE);
// Mount sdcard
if (FSGetMountSource(pClient, pCmd, FS_MOUNT_SOURCE_SD, mountSrc, -1) == 0)
{
result = FSMount(pClient, pCmd, mountSrc, mountPath, FS_MAX_MOUNTPATH_SIZE, -1);
if((result == 0) && mount_path) {
*mount_path = (char*)malloc(strlen(mountPath) + 1);
if(*mount_path)
strcpy(*mount_path, mountPath);
}
}
free(mountPath);
free(mountSrc);
return result;
}
int FS_Helper_GetFile(void * pClient,void * pCmd,const char * path, char *(*result)){
if(pClient == NULL || pCmd == NULL || path == NULL || result == NULL) return -2;
FSStat stats;
s32 status = -1;
s32 handle = 0;
if((status = FSGetStat(pClient,pCmd,path,&stats,-1)) == FS_STATUS_OK){
(*result) = (uint8_t *) memalign(0x40, (sizeof(uint8_t)*stats.size)+1);
if(!(*result)){
printf("FS_Helper_GetFile(line %d): error: Failed to allocate space for reading the file\n",__LINE__);
return -1;
}
(*result)[stats.size] = '\0';
if((status = FSOpenFile(pClient,pCmd,path,"r",&handle,-1)) == FS_STATUS_OK){
s32 total_read = 0;
s32 ret2 = 0;
char * cur_result_pointer = *result;
s32 sizeToRead = stats.size;
while ((ret2 = FSReadFile(pClient,pCmd, cur_result_pointer, 0x01, sizeToRead, handle, 0, -1)) > 0){
total_read += ret2;
cur_result_pointer += ret2;
sizeToRead -= ret2;
}
}else{
printf("FS_Helper_GetFile(line %d): error: (FSOpenFile) Couldn't open file (%s), error: %d",__LINE__,path,status);
free((*result));
(*result)=NULL;
return -1;
}
FSCloseFile(pClient,pCmd,handle,-1);
return 0;
}else{
printf("FS_Helper_GetFile(line %d): error: (GetStat) Couldn't open file (%s), error: %d",__LINE__,path,status);
}
return -1;
}