void RestoreSneekFolder (void)
{
s32 fd;
int ret;
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
char pathBak[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
Debug ("RestoreSneekFolder [begin]");
ISFS_Initialize ();
sprintf (path, "/sneek/nandpath.bin");
ret = ISFS_Delete (path);
Debug ("RestoreSneekFolder: delete '%s' = %d", path, ret);
sprintf (path, "/sneek/nandcfg.pl");
ret = ISFS_Delete (path);
Debug ("RestoreSneekFolder: delete '%s' = %d", path, ret);
ret = sprintf (path, "/sneek/nandcfg.ch");
ISFS_Delete (path);
Debug ("RestoreSneekFolder: delete '%s' = %d", path, ret);
sprintf (path, "/title/00000001/00000002/data/loader.ini");
sprintf (pathBak, "/title/00000001/00000002/data/loader.bak");
fd = ISFS_Open(pathBak, ISFS_OPEN_READ);
if (fd > 0)
{
ISFS_Close(fd);
ret = ISFS_Delete (path);
Debug ("RestoreSneekFolder: delete '%s' = %d", path, ret);
ret = ISFS_Rename (pathBak, path);
Debug ("RestoreSneekFolder: rename '%s'->'%s' = %d", pathBak, path, ret);
}
sprintf (path, "/sneek/nandcfg.bin");
sprintf (pathBak, "/sneek/nandcfg.bak");
fd = ISFS_Open(pathBak, ISFS_OPEN_READ);
if (fd > 0)
{
ISFS_Close(fd);
ret = ISFS_Delete (path);
Debug ("RestoreSneekFolder: delete '%s' = %d", path, ret);
ret = ISFS_Rename (pathBak, path);
Debug ("RestoreSneekFolder: rename '%s'->'%s' = %d", pathBak, path, ret);
}
ISFS_Deinitialize ();
Debug ("RestoreSneekFolder [end]");
}
void renamefilesback(void)
{
s32 ret;
char filepath[256];
char filepathold[256];
if( armbootmii == true && armbootsneek == true )
{
sprintf(filepath, "/bootmii/armbootsneek.bin");
ret = ISFS_Delete(filepath);
if (ret < 0)
printf("\nError! ISFS_Delete(%s) returned %d)\n", filepath, ret);
sprintf(filepathold, "/bootmii/armboot.bin");
ret = ISFS_Rename(filepathold, filepath);
if (ret < 0)
printf("\nError! ISFS_Rename(%s, %s) returned %d)\n", filepathold,
filepath, ret);
sprintf(filepathold, "/bootmii/armbootmii.bin");
sprintf(filepath, "/bootmii/armboot.bin");
ret = ISFS_Rename(filepathold, filepath);
if (ret < 0)
printf("\nError! ISFS_Rename(%s, %s) returned %d)\n", filepathold,
filepath, ret);
}
if( armbootmii == true && armbootsneek == false )
{
sprintf(filepath, "/bootmii/armbootsneek.bin");
sprintf(filepathold, "/bootmii/armboot.bin");
ret = ISFS_Rename(filepathold, filepath);
if (ret < 0)
printf("\nError! ISFS_Rename(%s, %s) returned %d)\n", filepathold,
filepath, ret);
sprintf(filepathold, "/bootmii/armbootmii.bin");
sprintf(filepath, "/bootmii/armboot.bin");
ret = ISFS_Rename(filepathold, filepath);
if (ret < 0)
printf("\nError! ISFS_Rename(%s, %s) returned %d)\n", filepathold,
filepath, ret);
}
}
void RestorePriiloader (int nidx)
{
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
char pathBack[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
ISFS_Initialize();
sprintf (path, "%s/title/00000001/00000002/data/main.bin", nandConfig->NandInfo[nidx]);
sprintf (pathBack, "%s/title/00000001/00000002/data/main.bak", nandConfig->NandInfo[nidx]);
ISFS_Delete (path);
ISFS_Rename (pathBack, path);
sprintf (path, "%s/title/00000001/00000002/data/loader.ini", nandConfig->NandInfo[nidx]);
sprintf (pathBack, "%s/title/00000001/00000002/data/loader.bak", nandConfig->NandInfo[nidx]);
ISFS_Delete (path);
ISFS_Rename (pathBack, path);
ISFS_Deinitialize ();
}
void RestoreSM (int nidx)
{
bool find = FALSE;
char path[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
char pathBack[ISFS_MAXPATH] ATTRIBUTE_ALIGN(32);
char *sm[] = {"00000088","00000085","0000008b","0000008e","00000098","00000095","0000009b","0000009e"};
s32 smsize;
bool issm = FALSE;
ISFS_Initialize();
int i;
s32 fd;
for (i = 0; i < sizeof(sm) / sizeof(char*); i++)
{
sprintf (path, "%s/title/00000001/00000002/content/%s.app", nandConfig->NandInfo[nidx], sm[i]);
sprintf (pathBack, "%s/title/00000001/00000002/content/%s.bak", nandConfig->NandInfo[nidx], sm[i]);
fd = ISFS_Open (path, ISFS_OPEN_READ);
Debug ("ReplaceNandSystemMenu: checking %s (%d)", sm[i], fd);
if (fd < 0) continue;
smsize = ISFS_Seek(fd, 0, 2);
Debug ("ReplaceNandSystemMenu: sm size %d", smsize);
if (smsize > 1000000) // E' piu' grande di 1MB... e' il sistem menu
issm = TRUE;
ISFS_Close (fd);
find = TRUE;
break;
}
if (find && !issm)
{
// Restore filesistem
ISFS_Delete (path);
ISFS_Rename (pathBack, path);
}
ISFS_Deinitialize ();
}
/*
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;
}
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;
}
