CfgUSBLoader/plugins/mighty/source/nand.c

#include <stdio.h>
#include <ogcsys.h>
#include <malloc.h>
#include <string.h>

#include "nand.h"

/* Buffer */
static u32 inbuf[8] ATTRIBUTE_ALIGN(32);

static nandDevice ndevList[] = {
	{ "Disable",				0,	0x00,	0x00 },
	{ "SD/SDHC Card",			1,	0xF0,	0xF1 },
	{ "USB 2.0 Mass Storage Device",	2,	0xF2,	0xF3 },
};

static int mounted=0;

static int partition=0;
static int fullmode=0x100;
static char path[32]="\0";

s32 Nand_Mount(nandDevice *dev){
	s32 fd, ret;
	u32 inlen = 0;
	ioctlv *vector = NULL;
	u32 *buffer = NULL;
	int rev;

	/* Open FAT module */
	fd = IOS_Open("fat", 0);
	if (fd < 0)
		return fd;

	rev=IOS_GetRevision();
	/* Prepare vector */
	if(rev>=21){		
		// NOTE: 
		// The official cIOSX rev21 by Waninkoko ignores the partition argument
		// and the nand is always expected to be on the 1st partition.
		// However this way earlier d2x betas having revision 21 take in 
		// consideration the partition argument.  
		inlen = 1;

		/* Allocate memory */
		buffer = (u32 *)memalign(32, sizeof(u32)*3);

		/* Set vector pointer */
		vector = (ioctlv *)buffer;

		buffer[0] = (u32)(buffer + 2);
		buffer[1] = sizeof(u32);
		buffer[2] = (u32)partition;
	}

	/* Mount device */
	ret = IOS_Ioctlv(fd, dev->mountCmd, inlen, 0, vector);
	
	/* Free memory */
	if(buffer != NULL)
		free(buffer);

	/* Close FAT module */
	IOS_Close(fd);

	return ret;
}

s32 Nand_Unmount(nandDevice *dev){
	s32 fd, ret;

	// Open FAT module
	fd = IOS_Open("fat", 0);
	if (fd < 0)
		return fd;

	// Unmount device
	ret = IOS_Ioctlv(fd, dev->umountCmd, 0, 0, NULL);

	// Close FAT module
	IOS_Close(fd);

	return ret;
}

s32 Nand_Enable(nandDevice *dev){
	s32 fd, ret;
	u32 *buffer = NULL;
	int rev;

	// Open /dev/fs
	fd = IOS_Open("/dev/fs", 0);
	if (fd < 0)
		return fd;

	rev=IOS_GetRevision();
	// Set input buffer
	if(rev>=21 && rev<30000 && dev->mode!=0){
		//FULL NAND emulation since rev18
		//needed for reading images on triiforce mrc folder using ISFS commands
		inbuf[0] = dev->mode | fullmode;
	}else{
		inbuf[0] = dev->mode; //old method
	}

	// Enable NAND emulator
	if(rev>=21 && rev<30000){
		// NOTE: 
		// The official cIOSX rev21 by Waninkoko provides an undocumented feature
		// to set nand path when mounting the device.
		// This feature has been discovered during d2x development.
		int pathlen = strlen(path)+1;

		/* Allocate memory */
		buffer = (u32 *)memalign(32, (sizeof(u32)*5)+pathlen);
    
		buffer[0] = (u32)(buffer + 4);
		buffer[1] = sizeof(u32);   // actually not used by cios
		buffer[2] = (u32)(buffer + 5);
		buffer[3] = pathlen;       // actually not used by cios
		buffer[4] = inbuf[0];            
		strcpy((char*)(buffer+5), path);
		
		ret = IOS_Ioctlv(fd, 100, 2, 0, (ioctlv *)buffer);
	}else{
		ret = IOS_Ioctl(fd, 100, inbuf, sizeof(inbuf), NULL, 0);
	}

	/* Free memory */
	if(buffer != NULL)
		free(buffer);

	// Close /dev/fs
	IOS_Close(fd);

	return ret;
} 

s32 Nand_Disable(void){
	s32 fd, ret;

	// Open /dev/fs
	fd = IOS_Open("/dev/fs", 0);
	if (fd < 0)
		return fd;

	// Set input buffer
	inbuf[0] = 0;

	// Disable NAND emulator
	ret = IOS_Ioctl(fd, 100, inbuf, sizeof(inbuf), NULL, 0);

	// Close /dev/fs
	IOS_Close(fd);

	return ret;
} 


s32 Enable_Emu(int selection){
	if(mounted != 0)
		return -1;
	s32 ret;
	nandDevice *ndev = NULL;
	ndev = &ndevList[selection];

	ret = Nand_Mount(ndev);
	if (ret < 0) 
	{
		printf(" ERROR Mount! (ret = %d)\n", ret);
		return ret;

	}

	ret = Nand_Enable(ndev);
	if (ret < 0) 
	{
		printf(" ERROR Enable! (ret = %d)\n", ret);
		return ret;
	}
	mounted = selection;
	return 0;
}	

s32 Disable_Emu(){
	if(mounted==0)
		return 0;

	nandDevice *ndev = NULL;
	ndev = &ndevList[mounted];

	Nand_Disable();
	Nand_Unmount(ndev);

	mounted = 0;

	return 0;
}

void Set_Partition(int p){
	partition=p;
}

void Set_Path(const char* p){
	int i=0;

	while(p[i]!='\0' && i<31){
		path[i]=p[i];
		i++;
	}
	if(path[i-1]=='/')
		path[i-1]='\0';
	path[i]='\0';
}

void Set_FullMode(int fm){
	fullmode=fm? 0x100: 0;
}

const char* Get_Path(void){
	return path;
}