int fileBrowser_WiiFS_writeFile(fileBrowser_file* file, void* buffer, unsigned int length){
// Make sure everything is aligned (address and length)
int isUnaligned = ((int)buffer)%32 | length%32;
int alignedLen = (length+31)&0xffffffe0;
char* alignedBuf;
if(isUnaligned){
alignedBuf = memalign(32, alignedLen);
memcpy(alignedBuf, buffer, length);
} else alignedBuf = buffer;
// Make sure the filename is 8.3 and open the short filename
shrinkFilename( &file->name );
char* name = getAlignedName(&file->name);
int f = ISFS_Open( name, 2 );
if(f < 0){ // Create rw file: rwrwr-
if((f = ISFS_CreateFile( name, 0, 1|2, 1|2, 1 )) < 0)
return FILE_BROWSER_ERROR;
else f = ISFS_Open( name, 2 );
}
// Do the actual write, from the aligned buffer if need be
ISFS_Seek(f, file->offset, 0);
int bytes_read = ISFS_Write(f, alignedBuf, alignedLen);
if(bytes_read > 0) file->offset += bytes_read;
// Clean up
if(isUnaligned) free(alignedBuf);
ISFS_Close(f);
return bytes_read;
}
s8 RemovePriiloader ( void )
{
s32 fd = 0;
Nand_Permissions SysPerm;
memset(&SysPerm,0,sizeof(Nand_Permissions));
SysPerm.otherperm = 3;
SysPerm.groupperm = 3;
SysPerm.ownerperm = 3;
printf("Restoring System menu app...");
s32 ret = nand_copy(copy_app,original_app,SysPerm);
if (ret < 0)
{
if(ret == -80)
{
//checksum issues
printf("\x1b[%u;%dm", 33, 1);
printf("\nWARNING!!\n Installer could not calculate the Checksum when coping the System menu app\n");
printf("back! the app however was copied...\n");
printf("Do you want to Continue ?\n");
printf("A = Yes B = No\n ");
printf("\x1b[%u;%dm", 37, 1);
if(!UserYesNoStop())
{
printf("reverting changes...\n");
ISFS_Close(fd);
ISFS_CreateFile(original_app,0,3,3,3);
fd = ISFS_Open(original_app,ISFS_OPEN_RW);
ISFS_Write(fd,priiloader_app,priiloader_app_size);
ISFS_Close(fd);
abort("System Menu Copying Failure");
}
}
else
{
ISFS_CreateFile(original_app,0,3,3,3);
fd = ISFS_Open(original_app,ISFS_OPEN_RW);
ISFS_Write(fd,priiloader_app,priiloader_app_size);
ISFS_Close(fd);
abort("\nUnable to restore the system menu! (ret = %d)",ret);
}
}
ret = ISFS_Delete(copy_app);
printf("Done!\n");
return 1;
}
int Playlog_Update(const char ID[6], const u8 title[84])
{
gprintf("Update Play log\n");
u32 sum = 0;
u8 i;
//Open play_rec.dat
s32 playrec_fd = IOS_Open(PLAYRECPATH, IPC_OPEN_RW);
if(playrec_fd == -106)
{
gprintf("IOS_Open error ret: %i\n",playrec_fd);
IOS_Close(playrec_fd);
//In case the play_rec.dat wasn´t found create one and try again
if(ISFS_CreateFile(PLAYRECPATH,0,3,3,3) < 0 )
goto error_2;
playrec_fd = IOS_Open(PLAYRECPATH, IPC_OPEN_RW);
if(playrec_fd < 0)
goto error_2;
}
else if(playrec_fd < 0)
goto error_2;
u64 stime = getWiiTime();
playrec_buf.ticks_boot = stime;
playrec_buf.ticks_last = stime;
//Update channel name and ID
memcpy(playrec_buf.name, title, 84);
strcpy(playrec_buf.title_id, ID);
memset(playrec_buf.padding2, 0, 18);
//Calculate and update checksum
for(i=0; i<31; i++)
sum += playrec_buf.data[i];
playrec_buf.checksum=sum;
//Write play_rec.dat
if(IOS_Write(playrec_fd, &playrec_buf, sizeof(playrec_buf)) != sizeof(playrec_buf))
goto error_1;
IOS_Close(playrec_fd);
return 0;
error_1:
gprintf("error_1\n");
IOS_Close(playrec_fd);
error_2:
gprintf("error_2\n");
return -1;
}
void createswitchsneek(void)
{
s32 ret;
char filepath[256];
if(switchsneek == false)
{
sprintf(filepath, "/bootmii/switchtosneek");
ret = ISFS_CreateFile (filepath, 0, 3, 1, 1);
if(ret != ISFS_OK)
printf("\nError! ISFS_CreateFile(%s) returned %d\n", filepath, ret);
}
}
void NandSave::WriteFile(const char *file_name, u8 *content, u32 size)
{
memset(&ISFS_Path, 0, ISFS_MAXPATH);
if(file_name == NULL || content == NULL || size == 0)
return;
strcpy(ISFS_Path, file_name);
ISFS_CreateFile(ISFS_Path, 0, 3, 3, 3);
fd = ISFS_Open(ISFS_Path, ISFS_OPEN_WRITE);
if(fd < 0)
return;
ret = ISFS_Write(fd, content, size);
ISFS_Close(fd);
if(ret < 0)
ISFS_Delete(ISFS_Path);
}
static s32 __Config_Create(const char *filename)
{
s32 ret;
/* Open ISFS */
ret = ISFS_Open();
if (ret < 0)
return ret;
/* Create file */
ret = ISFS_CreateFile(filename);
/* Close ISFS */
ISFS_Close();
return ret;
}
int Playlog_Create(void)
{
s32 fd = IOS_Open(PLAYRECPATH, IPC_OPEN_RW);
if(fd >= 0)
{
//exists
IOS_Close(fd);
return 0;
}
ISFS_Initialize();
//In case the play_rec.dat wasn´t found create one and try again
int ret = ISFS_CreateFile(PLAYRECPATH, 0, 3, 3, 3);
if(ret >= 0)
ISFS_SetAttr(PLAYRECPATH, 0x1000, 1, 0, 3, 3, 3);
ISFS_Deinitialize();
return ret;
}
/*
This creates the file in the TMP dir and then moves it to the destination overwriting it!
To work correctly on the real nand we must have the same filename for the
source and destination!
*/
s32 NANDWriteFileSafe( char *pathdst, void *data, u32 size )
{
char *path = (char*)heap_alloc_aligned( 0, 0x40, 32 );
s32 i=0;
//Extract filename
//search backwards for slash
for( i=strlen(pathdst); i > 0; --i )
if( pathdst[i] == '/' )
break;
_sprintf( path, "/tmp%s", pathdst + i );
s32 r = ISFS_CreateFile( path, 0, 3, 3, 3 );
if( r == FS_EEXIST2 )
{
ISFS_Delete( path );
r = ISFS_CreateFile( path, 0, 3, 3, 3 );
if( r < 0 )
{
//dbgprintf("ISFS_CreateFile(%s):%d\n",path,r);
heap_free( 0, path );
return r;
}
} else {
if( r < 0 )
{
//dbgprintf("ISFS_CreateFile(%s):%d\n",path,r);
heap_free( 0, path );
return r;
}
}
s32 fd = IOS_Open( path, 2 );
if( fd < 0 )
{
//dbgprintf("IOS_Open(%s):%d\n",path,r);
heap_free( 0, path );
return r;
}
r = IOS_Write( fd, data, size );
if( r < 0 || r != size )
{
//dbgprintf("IOS_Write():%d\n",r);
IOS_Close( fd );
heap_free( 0, path );
return r;
}
IOS_Close( fd );
r = ISFS_Rename( path, pathdst );
if( r < 0 )
{
//dbgprintf("ISFS_Rename(%s,%s):%d\n",path,pathdst,r);
heap_free( 0, path );
return r;
}
heap_free( 0, path );
return r;
}
static int _ISFS_open_r(struct _reent *r, void *fileStruct, const char *path, int flags, int mode) {
FILE_STRUCT *file = (FILE_STRUCT *)fileStruct;
char *abspath = malloc(ISFS_MAXPATHLEN);
if(!abspath) {
r->_errno = ENOMEM;
return -1;
}
char *ptr = strchr(path, ':');
if (ptr != NULL)
snprintf(abspath, ISFS_MAXPATHLEN, "%s", ptr + 1);
else
snprintf(abspath, ISFS_MAXPATHLEN, "%s/%s", current->entry->abspath, path);
RemoveDoubleSlash(abspath);
if (!READ_ONLY && (flags & O_CREAT)) {
int iOwnerPerm = 0;
int iGroupPerm = 0;
int iOtherPerm = 0;
if (flags & S_IRUSR)
iOwnerPerm |= ISFS_OPEN_READ;
if (flags & S_IWUSR)
iOwnerPerm |= ISFS_OPEN_WRITE;
if (flags & S_IRGRP)
iGroupPerm |= ISFS_OPEN_READ;
if (flags & S_IWGRP)
iGroupPerm |= ISFS_OPEN_WRITE;
if (iGroupPerm & S_IROTH)
iOtherPerm |= ISFS_OPEN_READ;
if (flags & S_IWOTH)
iOtherPerm |= ISFS_OPEN_WRITE;
ISFS_CreateFile(abspath, 0, iOwnerPerm, iGroupPerm, iOtherPerm);
}
int iOpenMode = 0;
if ((flags & 0x03) == O_RDONLY)
iOpenMode |= ISFS_OPEN_READ;
if (!READ_ONLY && ((flags & 0x03) == O_WRONLY))
iOpenMode |= ISFS_OPEN_WRITE;
if ((flags & 0x03) == O_RDWR)
iOpenMode |= READ_ONLY ? ISFS_OPEN_READ : ISFS_OPEN_RW;
file->fd = ISFS_Open(abspath, iOpenMode);
free(abspath);
if (file->fd < 0) {
if (file->fd == ISFS_EINVAL)
r->_errno = EACCES;
else
r->_errno = -file->fd;
return -1;
}
file->flags = flags;
file->mode = mode;
if (ISFS_GetFileStats(file->fd, &filest) == ISFS_OK)
file->size = filest.file_length;
if (!READ_ONLY && (flags & O_APPEND))
ISFS_Seek(file->fd, 0, SEEK_END);
return (int)file;
}
s32 nand_copy(const char *source, const char *destination,Nand_Permissions src_perm)
{
//variables
u8 *buffer = NULL;
STACK_ALIGN(fstats,status,sizeof(fstats),32);
s32 file_handler, ret;
u32 FileHash_D1[5];
memset(FileHash_D1,0,5);
u32 FileHash_D2[5];
memset(FileHash_D2,0xFF,5); //place different data in D2 so that if something goes wrong later on, the comparison will fail
SHA1 sha;
sha.Reset();
//variables - temp dir & SHA1 check
char temp_dest[ISFS_MAXPATH];
memset(temp_dest,0,ISFS_MAXPATH);
char *ptemp = NULL;
u8 temp = 0;
//get temp filename
ptemp = strstr(destination,"/");
while(ptemp != NULL && strstr(ptemp+1,"/") != NULL)
{
ptemp = strstr(ptemp+1,"/");
}
if(ptemp[0] == '/')
{
ptemp = ptemp+1;
}
memset(temp_dest,0,ISFS_MAXPATH);
sprintf(temp_dest,"/tmp/%s",ptemp);
//get data into pointer from original file
file_handler = ISFS_Open(source,ISFS_OPEN_READ);
if (file_handler < 0)
{
gprintf("failed to open source : %s\n",source);
return file_handler;
}
ret = ISFS_GetFileStats(file_handler,status);
if (ret < 0)
{
printf("\n\nFailed to get information about %s!\n",source);
sleepx(2);
ISFS_Close(file_handler);
return ret;
}
buffer = (u8 *)memalign(32,ALIGN32(status->file_length));
if (buffer == NULL)
{
gprintf("buffer failed to align\n");
sleepx(2);
ISFS_Close(file_handler);
return 0;
}
memset(buffer,0,status->file_length);
ret = ISFS_Read(file_handler,buffer,status->file_length);
if (ret < 0)
{
printf("\n\nFailed to Read Data from %s!\n",source);
sleepx(2);
ISFS_Close(file_handler);
free(buffer);
buffer = NULL;
return ret;
}
ISFS_Close(file_handler);
//everything read into buffer. generate SHA1 hash of the buffer
sha.Input(buffer,status->file_length);
if (!sha.Result(FileHash_D1))
{
gprintf("could not compute Hash of D1!\n");
free(buffer);
buffer = NULL;
return -80;
}
sha.Reset();
//done, lets create temp file and write :')
ISFS_Delete(temp_dest);
ISFS_CreateFile(temp_dest,src_perm.attributes,src_perm.ownerperm,src_perm.groupperm,src_perm.otherperm);
//created. opening it...
file_handler = ISFS_Open(temp_dest,ISFS_OPEN_RW);
if (file_handler < 0)
{
gprintf("failed to open destination : %s\n",temp_dest);
ISFS_Delete(temp_dest);
free(buffer);
buffer = NULL;
return file_handler;
}
ret = ISFS_Write(file_handler,buffer,status->file_length);
if (ret < 0)
{
gprintf("failed to write destination : %s\n",destination);
ISFS_Close(file_handler);
ISFS_Delete(temp_dest);
free(buffer);
buffer = NULL;
return ret;
}
//write done. reopen file for reading and compare SHA1 hash
ISFS_Close(file_handler);
free(buffer);
buffer = NULL;
memset(status,0,sizeof(fstats));
file_handler = ISFS_Open(temp_dest,ISFS_OPEN_READ);
if(!file_handler)
{
temp = -1;
goto free_and_Return;
}
ret = ISFS_GetFileStats(file_handler,status);
if (ret < 0)
{
ISFS_Close(file_handler);
temp = -2;
goto free_and_Return;
}
buffer = (u8 *)memalign(32,ALIGN32(status->file_length));
if (buffer == NULL)
{
gprintf("buffer failed to align\n");
ISFS_Close(file_handler);
temp = -3;
goto free_and_Return;
}
memset(buffer,0,status->file_length);
if( ISFS_Read(file_handler,buffer,status->file_length) < 0 )
{
temp = -4;
goto free_and_Return;
}
ISFS_Close(file_handler);
sha.Reset();
sha.Input(buffer,status->file_length);
free(buffer);
buffer = NULL;
if (!sha.Result(FileHash_D2))
{
gprintf("could not compute Hash of D2!\n");
return -80;
}
sha.Reset();
/*gprintf("sha1 original : %x %x %x %x %x\nsha1 written : %x %x %x %x %x\n",
FileHash_D1[0],FileHash_D1[1],FileHash_D1[2],FileHash_D1[3],FileHash_D1[4],
FileHash_D2[0],FileHash_D2[1],FileHash_D2[2],FileHash_D2[3],FileHash_D2[4]);*/
if (!memcmp(FileHash_D1,FileHash_D2,sizeof(FileHash_D1)))
{
gprintf("nand_copy : SHA1 hash success\n");
ISFS_Delete(destination);
ret = ISFS_Rename(temp_dest,destination);
gprintf("ISFS_Rename ret %d\n",ret);
if ( ret < 0)
temp = -5;
goto free_and_Return;
}
else
{
temp = -6;
goto free_and_Return;
}
free_and_Return:
if(buffer)
{
free(buffer);
buffer = NULL;
}
if (temp < 0)
{
gprintf("nand_copy temp %d fail o.o;\n",temp);
ISFS_Delete(temp_dest);
return -80;
}
return 1;
}
s32 nand_copy(const char *destination,u8* Buf_To_Write_to_Copy, u32 buf_size,Nand_Permissions src_perm)
{
if( Buf_To_Write_to_Copy == NULL || buf_size < 1 )
{
return -1;
}
s32 ret, dest_handler;
gprintf("owner %d group %d attributes %X perm:%X-%X-%X\n", src_perm.owner, (u32)src_perm.group, (u32)src_perm.attributes, (u32)src_perm.ownerperm, (u32)src_perm.groupperm, (u32)src_perm.otherperm);
//extract filename from destination
char temp_dest[ISFS_MAXPATH];
memset(temp_dest,0,ISFS_MAXPATH);
char *ptemp = NULL;
ptemp = strstr(destination,"/");
while(ptemp != NULL && strstr(ptemp+1,"/") != NULL)
{
ptemp = strstr(ptemp+1,"/");
}
if(ptemp[0] == '/')
{
ptemp = ptemp+1;
}
//create temp path
memset(temp_dest,0,ISFS_MAXPATH);
sprintf(temp_dest,"/tmp/%s",ptemp);
ISFS_Delete(temp_dest);
//and go for it
ret = ISFS_CreateFile(temp_dest,src_perm.attributes,src_perm.ownerperm,src_perm.groupperm,src_perm.otherperm);
if (ret != ISFS_OK)
{
printf("Failed to create file %s. ret = %d\n",temp_dest,ret);
gprintf("Failed to create file %s. ret = %d\n",temp_dest,ret);
return ret;
}
dest_handler = ISFS_Open(temp_dest,ISFS_OPEN_RW);
if (dest_handler < 0)
{
gprintf("failed to open destination : %s\n",temp_dest);
ISFS_Delete(temp_dest);
return dest_handler;
}
ret = ISFS_Write(dest_handler,Buf_To_Write_to_Copy,buf_size);
if (ret < 0)
{
gprintf("failed to write destination : %s\n",temp_dest);
ISFS_Close(dest_handler);
ISFS_Delete(temp_dest);
return ret;
}
ISFS_Close(dest_handler);
s32 temp = 0;
u8 *Data2 = NULL;
STACK_ALIGN(fstats,D2stat,sizeof(fstats),32);
/*if (D2stat == NULL)
{
temp = -1;
goto free_and_Return;
}*/
dest_handler = ISFS_Open(temp_dest,ISFS_OPEN_RW);
if(dest_handler < 0)
{
gprintf("temp_dest open error %d\n",dest_handler);
temp = -2;
goto free_and_Return;
}
temp = ISFS_GetFileStats(dest_handler,D2stat);
if(temp < 0)
{
goto free_and_Return;
}
Data2 = (u8*)memalign(32,ALIGN32(D2stat->file_length));
if (Data2 == NULL)
{
temp = -3;
goto free_and_Return;
}
if( ISFS_Read(dest_handler,Data2,D2stat->file_length) > 0 )
{
if( !CompareSha1Hash(Buf_To_Write_to_Copy,buf_size,Data2,D2stat->file_length))
{
temp = -4;
goto free_and_Return;
}
}
else
{
temp = -5;
goto free_and_Return;
}
if(Data2)
{
free(Data2);
Data2 = NULL;
}
ISFS_Close(dest_handler);
//so it was written to /tmp correctly. lets call ISFS_Rename and compare AGAIN
ISFS_Delete(destination);
ret = ISFS_Rename(temp_dest,destination);
if(ret < 0 )
{
gprintf("nand_copy(buf) : rename returned %d\n",ret);
temp = -6;
goto free_and_Return;
}
free_and_Return:
if(Data2 != NULL)
{
free(Data2);
Data2 = NULL;
}
ISFS_Close(dest_handler);
if (temp < 0)
{
gprintf("temp %d\n",temp);
//ISFS_Delete(destination);
return -80;
}
return 1;
}
s8 WritePriiloader( bool priiloader_found )
{
s32 ret = 0;
s32 fd = 0;
Nand_Permissions SysPerm;
if(priiloader_found == false)
{
memset(&SysPerm,0,sizeof(Nand_Permissions));
SysPerm.otherperm = 3;
SysPerm.groupperm = 3;
SysPerm.ownerperm = 3;
//system menu coping
printf("Moving System Menu app...");
ret = nand_copy(original_app,copy_app,SysPerm);
if (ret < 0)
{
if (ret == -80)
{
//checksum issues
printf("\x1b[%u;%dm", 33, 1);
printf("\nWARNING!!\n Installer could not calculate the Checksum for the System menu app");
printf("\nbut Copy was successfull.\n");
printf("Do you want the Continue ?\n");
printf("A = Yes B = No\n ");
printf("\x1b[%u;%dm", 37, 1);
if(!UserYesNoStop())
{
printf("reverting changes...\n");
ISFS_Delete(copy_app);
abort("System Menu Copying Failure");
}
else
printf("\nDone!\n");
}
else
abort("\nUnable to move the system menu. error %d",ret);
}
else
{
gprintf("Moving System Menu Done\n");
printf("Done!\n");
}
}
ret = 0;
//sys_menu app moved. lets write priiloader
STACK_ALIGN(fstats,status,sizeof(fstats),32);
memset(&SysPerm,0,sizeof(Nand_Permissions));
SysPerm.otherperm = 3;
SysPerm.groupperm = 3;
SysPerm.ownerperm = 3;
printf("Writing Priiloader app...");
gprintf("Writing Priiloader\n");
char temp_dest[ISFS_MAXPATH];
memset(temp_dest,0,ISFS_MAXPATH);
char *ptemp = NULL;
ptemp = strstr(original_app,"/");
while(ptemp != NULL && strstr(ptemp+1,"/") != NULL)
{
ptemp = strstr(ptemp+1,"/");
}
if(ptemp[0] == '/')
{
ptemp = ptemp+1;
}
memset(temp_dest,0,ISFS_MAXPATH);
sprintf(temp_dest,"/tmp/%s",ptemp);
ISFS_Delete(temp_dest);
ret = ISFS_CreateFile(temp_dest,SysPerm.attributes,SysPerm.ownerperm,SysPerm.groupperm,SysPerm.otherperm);
fd = ISFS_Open(temp_dest,ISFS_OPEN_RW);
if (fd < 0)
{
gprintf("error %d\n",fd);
abort("\nFailed to open file for Priiloader writing");
}
ret = ISFS_Write(fd,priiloader_app,priiloader_app_size);
if (ret < 0 ) //check if the app was writen correctly
{
ISFS_Close(fd);
ISFS_Delete(copy_app);
ISFS_Delete(temp_dest);
gprintf("Write failed. ret %d\n",ret);
abort("\nWrite of Priiloader app failed");
}
ISFS_Close(fd);
//SHA1 check here
fd = ISFS_Open(temp_dest,ISFS_OPEN_READ);
if (fd < 0)
{
ISFS_Delete(copy_app);
abort("\nFailed to open file for Priiloader checking");
}
if (ISFS_GetFileStats(fd,status) < 0)
{
ISFS_Close(fd);
ISFS_Delete(copy_app);
abort("Failed to get stats of %s. System Menu Recovered",temp_dest);
}
else
{
if ( status->file_length != priiloader_app_size )
{
ISFS_Close(fd);
ISFS_Delete(copy_app);
abort("Written Priiloader app isn't the correct size.System Menu Recovered");
}
else
{
gprintf("Size Check Success\n");
printf("Size Check Success!\n");
}
}
u8 *AppData = (u8 *)memalign(32,ALIGN32(status->file_length));
if (AppData)
ret = ISFS_Read(fd,AppData,status->file_length);
else
{
ISFS_Close(fd);
ISFS_Delete(copy_app);
abort("Checksum comparison Failure! MemAlign Failure of AppData\n");
}
ISFS_Close(fd);
if (ret < 0)
{
if (AppData)
{
free(AppData);
AppData = NULL;
}
ISFS_Delete(copy_app);
abort("Checksum comparison Failure! read of priiloader app returned %u\n",ret);
}
if(CompareSha1Hash((u8*)priiloader_app,priiloader_app_size,AppData,status->file_length))
printf("Checksum comparison Success!\n");
else
{
if (AppData)
{
free(AppData);
AppData = NULL;
}
ISFS_Delete(copy_app);
abort("Checksum comparison Failure!\n");
}
if (AppData)
{
free(AppData);
AppData = NULL;
}
// rename and do a final SHA1 chezck
ISFS_Delete(original_app);
ret = ISFS_Rename(temp_dest,original_app);
if(ret < 0 )
{
gprintf("WritePriiloader : rename returned %d\n",ret);
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("\nFailed to Write Priiloader : error Ren %d",ret);
}
printf("Done!!\n");
gprintf("Wrote Priiloader App.Checking Installation\n");
printf("\nChecking Priiloader Installation...\n");
memset(status,0,sizeof(fstats));
fd = ISFS_Open(original_app,ISFS_OPEN_READ);
if (fd < 0)
{
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("\nFailed to open file for Priiloader checking");
}
if (ISFS_GetFileStats(fd,status) < 0)
{
ISFS_Close(fd);
nand_copy(copy_app,original_app,SysPerm);
abort("Failed to get stats of %s. System Menu Recovered",original_app);
}
else
{
if ( status->file_length != priiloader_app_size )
{
ISFS_Close(fd);
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("Written Priiloader app isn't the correct size.System Menu Recovered");
}
else
{
gprintf("Size Check Success\n");
printf("Size Check Success!\n");
}
}
AppData = (u8 *)memalign(32,ALIGN32(status->file_length));
if (AppData != NULL)
ret = ISFS_Read(fd,AppData,status->file_length);
else
{
ISFS_Close(fd);
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("Checksum comparison Failure! MemAlign Failure of AppData\n");
}
ISFS_Close(fd);
if (ret < 0)
{
if (AppData)
{
free(AppData);
AppData = NULL;
}
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("Checksum comparison Failure! read of priiloader app returned %u\n",ret);
}
if(CompareSha1Hash((u8*)priiloader_app,priiloader_app_size,AppData,status->file_length))
printf("Checksum comparison Success!\n");
else
{
if (AppData)
{
free(AppData);
AppData = NULL;
}
nand_copy(copy_app,original_app,SysPerm);
ISFS_Delete(copy_app);
abort("Checksum comparison Failure!\n");
}
if (AppData)
{
free(AppData);
AppData = NULL;
}
gprintf("Priiloader Update Complete\n");
printf("Done!\n\n");
return 1;
}
static int identify(void){
// Identify as our own title
void* dvdCert = NULL, * dvdTMD = NULL, * dvdTicket = NULL;
unsigned int dvdCertSize, dvdTMDSize, dvdTicketSize, keyid;
int ret;
if(!customCert){ // If there's no certificate supplied
#if 0
// Use the one from the DVD
ret = getTitle(&dvdCert, &dvdCertSize,
&dvdTMD, &dvdTMDSize,
&dvdTicket, &dvdTicketSize);
customCert = dvdCert;
customCertSize = dvdCertSize;
if(ret < 0) return ret;
#else
return identified = 0;
#endif
}
ret = ES_Identify(customCert, customCertSize,
customTMD, customTMDSize,
customTicket, customTicketSize,
&keyid);
if(ret >= 0){
ES_GetTitleID(&titleID);
ISFS_Initialize();
// If we haven't identified this title before
// we'll need to set some things up
char* path = memalign(32, 64);;
ES_GetDataDir(titleID, path);
strncat(path, "/banner.bin", 64);
ret = ISFS_Open(path, 1);
if(ret < 0){
// If the banner doesn't exist
// Create our banner.bin
ret = ISFS_CreateFile(path, 0, 3, 3, 1);
if(ret < 0) return 0;
ret = ISFS_Open(path, 2);
if(ret < 0) return 0;
ISFS_Write(ret, customBanner, customBannerSize);
ISFS_Close(ret);
// Create the N64SAVES directory
ES_GetDataDir(titleID, path);
strncat(path, "/N64SAVES", 64);
ISFS_CreateDir(path, 0, 3, 3, 1);
} else ISFS_Close(ret);
free(path);
return identified = 1;
}
#if 0
// If that still fails, try to identify from the discs certs
if(!dvdCert || !dvdTMD || !dvdTicket)
ret = getTitle(&dvdCert, &dvdCertSize,
&dvdTMD, &dvdTMDSize,
&dvdTicket, &dvdTicketSize);
else ret = 0;
if(ret >= 0){
ret = ES_Identify(dvdCert, dvdCertSize,
dvdTMD, dvdTMDSize,
dvdTicket, dvdTicketSize,
&keyid);
ES_GetTitleID(&titleID);
ISFS_Initialize();
return identified = (ret >= 0);
}
#endif
return identified = 0;
}
bool NandSave::CheckSave()
{
/* 10 million variables */
u32 u8_bin_size = 0;
u8 *u8_bin = NULL;
u32 certSize = 0;
signed_blob *certBuffer = NULL;
u32 tmd_bin_size = 0;
const signed_blob *tmd_bin = NULL;
u32 tik_bin_size = 0;
const signed_blob *tik_bin = NULL;
u32 banner_bin_size = 0;
const u8 *banner_bin = NULL;
u32 entries = 0;
/* May our banner already exist */
memset(&ISFS_Path, 0, ISFS_MAXPATH);
strcpy(ISFS_Path, BANNER_PATH);
fd = ISFS_Open(ISFS_Path, ISFS_OPEN_READ);
if(fd >= 0)
{
ISFS_Close(fd);
gprintf("Found WiiFlow Save\n");
goto done;
}
/* extract our archive */
u8_bin = DecompressCopy(save_bin, save_bin_size, &u8_bin_size);
if(u8_bin == NULL || u8_bin_size == 0)
goto error;
/* grab cert.sys */
memset(&ISFS_Path, 0, ISFS_MAXPATH);
strcpy(ISFS_Path, "/sys/cert.sys");
certBuffer = (signed_blob*)ISFS_GetFile(ISFS_Path, &certSize, -1);
if(certBuffer == NULL || certSize == 0)
goto error;
/* Install tik and tmd */
tik_bin = (const signed_blob*)u8_get_file(u8_bin, "tik.bin", &tik_bin_size);
if(tik_bin == NULL || tik_bin_size == 0)
goto error;
ret = ES_AddTicket(tik_bin, tik_bin_size, certBuffer, certSize, NULL, 0);
if(ret < 0)
goto error;
tmd_bin = (const signed_blob*)u8_get_file(u8_bin, "tmd.bin", &tmd_bin_size);
if(tmd_bin == NULL || tmd_bin_size == 0)
goto error;
ret = ES_AddTitleStart(tmd_bin, tmd_bin_size, certBuffer, certSize, NULL, 0);
if(ret < 0)
goto error;
ret = ES_AddTitleFinish();
if(ret < 0)
goto error;
/* WARNING dirty, delete tik again */
memset(&ISFS_Path, 0, ISFS_MAXPATH);
strcpy(ISFS_Path, "/ticket/00010000/57465346.tik");
ret = ISFS_Delete(ISFS_Path);
if(ret < 0)
goto error;
/* Delete the unused ticket folder */
memset(&ISFS_Path, 0, ISFS_MAXPATH);
strcpy(ISFS_Path, "/ticket/00010000");
ret = ISFS_ReadDir(ISFS_Path, NULL, &entries);
if(ret < 0)
goto error;
if(entries == 0)
{
ret = ISFS_Delete(ISFS_Path);
if(ret < 0)
goto error;
}
banner_bin = u8_get_file(u8_bin, "banner.bin", &banner_bin_size);
if(banner_bin == NULL || banner_bin_size == 0)
goto error;
memset(&ISFS_Path, 0, ISFS_MAXPATH);
strcpy(ISFS_Path, BANNER_PATH);
/* Write our banner */
ISFS_CreateFile(ISFS_Path, 0, 3, 3, 3);
fd = ISFS_Open(ISFS_Path, ISFS_OPEN_WRITE);
if(fd < 0)
goto error;
ret = ISFS_Write(fd, banner_bin, banner_bin_size);
ISFS_Close(fd);
if(ret < 0)
{
ISFS_Delete(ISFS_Path);
goto error;
}
free(certBuffer);
if(u8_bin != save_bin)
free(u8_bin);
gprintf("Created WiiFlow Save\n");
done:
loaded = true;
return loaded;
error:
gprintf("Error while creating WiiFlow Save\n");
loaded = false;
ES_AddTitleCancel();
if(certBuffer != NULL)
free(certBuffer);
certBuffer = NULL;
if(u8_bin != NULL)
free(u8_bin);
u8_bin = NULL;
tik_bin = NULL;
tmd_bin = NULL;
banner_bin = NULL;
return loaded;
}
s32 Downgrade_TMD_Revision(void *ptmd, u32 tmd_size, void *certs, u32 certs_size)
{
// The revison of the tmd used as paramter here has to be >= the revision of the installed tmd
s32 ret;
printf("Setting the revision to 0...\n");
ret = ES_AddTitleStart(ptmd, tmd_size, certs, certs_size, NULL, 0);
if (ret < 0)
{
if (ret == -1035)
{
printf("Error: ES_AddTitleStart returned %d, maybe you need an updated Downgrader\n", ret);
} else
{
printf("Error: ES_AddTitleStart returned %d\n", ret);
}
ES_AddTitleCancel();
return ret;
}
s32 file;
char *tmd_path = "/tmp/title.tmd";
ret = ISFS_Delete(tmd_path);
if (ret < 0)
{
printf("Error: ISFS_Delete returned %d\n", ret);
ES_AddTitleCancel();
ISFS_Deinitialize();
return ret;
}
ret = ISFS_CreateFile(tmd_path, 0, 3, 3, 3);
if (ret < 0)
{
printf("Error: ISFS_CreateFile returned %d\n", ret);
ES_AddTitleCancel();
ISFS_Deinitialize();
return ret;
}
file = ISFS_Open(tmd_path, ISFS_OPEN_RW);
if (file < 0)
{
printf("Error: ISFS_Open returned %d\n", file);
ES_AddTitleCancel();
ISFS_Deinitialize();
return file;
}
u8 *tmd = (u8 *)ptmd;
tmd[0x1dc] = 0;
tmd[0x1dd] = 0;
ret = ISFS_Write(file, (u8 *)ptmd, tmd_size);
if (ret < 0)
{
printf("Error: ISFS_Write returned %d\n", ret);
ISFS_Close(file);
ES_AddTitleCancel();
ISFS_Deinitialize();
return ret;
}
ISFS_Close(file);
ISFS_Deinitialize();
ret = ES_AddTitleFinish();
if (ret < 0)
{
printf("Error: ES_AddTitleFinish returned %d\n", ret);
return ret;
}
return 1;
}
