318 lines
7.9 KiB
C++
318 lines
7.9 KiB
C++
#include "saveloadthread.h"
|
|
#include "quazip.h"
|
|
#include "quazipfile.h"
|
|
#include "../WiiQt/savedatabin.h"
|
|
|
|
Q_DECLARE_METATYPE( PcSaveInfo )
|
|
SaveLoadThread::SaveLoadThread( QObject *parent ) : QThread( parent )
|
|
{
|
|
abort = false;
|
|
qRegisterMetaType< PcSaveInfo >();
|
|
}
|
|
|
|
void SaveLoadThread::ForceQuit()
|
|
{
|
|
mutex.lock();
|
|
abort = true;
|
|
mutex.unlock();
|
|
}
|
|
|
|
SaveLoadThread::~SaveLoadThread()
|
|
{
|
|
mutex.lock();
|
|
abort = true;
|
|
condition.wakeOne();
|
|
mutex.unlock();
|
|
wait();
|
|
}
|
|
|
|
void SaveLoadThread::GetBanners( const QString &bPath )
|
|
{
|
|
if( isRunning() )
|
|
{
|
|
qWarning() << "SaveLoadThread::GetBanners -> already running";
|
|
return;
|
|
}
|
|
basePath = bPath;
|
|
if( basePath.isEmpty() )
|
|
type = LOAD_SNEEK;
|
|
else
|
|
type = LOAD_PC;
|
|
start( NormalPriority );
|
|
}
|
|
|
|
bool SaveLoadThread::SetNandPath( const QString &bPath )
|
|
{
|
|
if( isRunning() )
|
|
return false;
|
|
|
|
if( nand.BasePath() == bPath )
|
|
return true;
|
|
|
|
return nand.SetPath( bPath );
|
|
}
|
|
|
|
SaveGame SaveLoadThread::GetSave( quint64 tid )
|
|
{
|
|
SaveGame ret;
|
|
ret.tid = 0;
|
|
|
|
if( isRunning() )
|
|
return ret;
|
|
|
|
return nand.GetSaveData( tid );
|
|
}
|
|
|
|
void SaveLoadThread::run()
|
|
{
|
|
if ( abort )
|
|
{
|
|
qDebug( "SaveLoadThread::run -> Thread abort" );
|
|
return;
|
|
}
|
|
mutex.lock();
|
|
/*if( basePath.isEmpty() )
|
|
{
|
|
qDebug() << "SaveLoadThread::run -> its empty";
|
|
return;
|
|
}*/
|
|
mutex.unlock();
|
|
|
|
switch( type )
|
|
{
|
|
case LOAD_SNEEK:
|
|
GetSavesFromNandDump();
|
|
break;
|
|
case LOAD_PC:
|
|
if( basePath.isEmpty() )
|
|
return;
|
|
GetPCSaves();
|
|
break;
|
|
default:
|
|
break;
|
|
|
|
}
|
|
|
|
emit SendDone( type );
|
|
}
|
|
|
|
int SaveLoadThread::GetFolderSize( const QString& path )
|
|
{
|
|
//qDebug() << "SaveLoadThread::GetFolderSize" << path;
|
|
quint32 ret = 0;
|
|
QDir dir( path );
|
|
dir.setFilter( QDir::Files | QDir::Dirs | QDir::NoDotAndDotDot );
|
|
QFileInfoList fiL = dir.entryInfoList();
|
|
foreach( QFileInfo fi, fiL )
|
|
{
|
|
if( fi.isDir() )
|
|
ret += GetFolderSize( fi.absoluteFilePath() );
|
|
|
|
else
|
|
ret += fi.size();
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void SaveLoadThread::GetSavesFromPC()
|
|
{
|
|
emit SendProgress( 0 );
|
|
|
|
emit SendProgress( 100 );
|
|
}
|
|
|
|
void SaveLoadThread::GetPCSaves()
|
|
{
|
|
emit SendProgress( 0 );
|
|
if( basePath.isEmpty() )
|
|
return;
|
|
int total = 0;
|
|
int done = 0;
|
|
//first count all the saves
|
|
//list all subdirs of the base ( TID upper )
|
|
QDir base( basePath );
|
|
QFileInfoList fi1 = base.entryInfoList( QDir::Dirs | QDir::NoDotAndDotDot );
|
|
foreach( QFileInfo upper, fi1 )
|
|
{
|
|
//qDebug() << "looking in" << upper.absoluteFilePath();
|
|
//get all subdirectories of the subfolder ( TID lower )
|
|
QDir uDir( upper.absoluteFilePath() );
|
|
QFileInfoList fi2 = uDir.entryInfoList( QDir::Dirs | QDir::NoDotAndDotDot );
|
|
foreach( QFileInfo lower, fi2 )
|
|
{
|
|
//qDebug() << "looking in" << lower.absoluteFilePath();
|
|
//list all .zip files in the game's folder
|
|
QDir lDir( lower.absoluteFilePath() );
|
|
QFileInfoList fi3 = lDir.entryInfoList( QStringList() << "*.zip", QDir::Files );
|
|
//qDebug() << "found" << fi3.size() << "zip files in here";
|
|
total += fi3.size();
|
|
}
|
|
}
|
|
|
|
//qDebug() << "SaveLoadThread::GetPCSaves(): will open" << total << "zip archives";
|
|
//now actually read/send the saves
|
|
foreach( QFileInfo upper, fi1 )
|
|
{
|
|
//get all subdirectories of the subfolder ( TID lower )
|
|
QDir uDir( upper.absoluteFilePath() );
|
|
QFileInfoList fi2 = uDir.entryInfoList( QDir::Dirs | QDir::NoDotAndDotDot );
|
|
foreach( QFileInfo lower, fi2 )
|
|
{
|
|
//list all .zip files in the game's folder
|
|
QDir lDir( lower.absoluteFilePath() );
|
|
QFileInfoList fi3 = lDir.entryInfoList( QStringList() << "*.zip", QDir::Files );
|
|
quint32 cnt = fi3.size();
|
|
//QStringList descriptions;
|
|
//QByteArray banner;
|
|
//quint32 size = 0;
|
|
PcSaveInfo info;
|
|
info.tid = upper.fileName() + lower.fileName();
|
|
for( quint32 i = 0; i < cnt; i++ )
|
|
{
|
|
//qDebug() << "opening archive" << fi3.at( i ).absoluteFilePath();
|
|
bool bnrOk = false;
|
|
bool txtOk = false;
|
|
quint32 sSize = 0;
|
|
QString desc;
|
|
QuaZip zip( fi3.at( i ).absoluteFilePath() );
|
|
if( !zip.open( QuaZip::mdUnzip ) )
|
|
{
|
|
qWarning("SaveLoadThread::GetPCSaves(): zip.open(): %d", zip.getZipError() );
|
|
continue;
|
|
}
|
|
zip.setFileNameCodec("IBM866");
|
|
if( zip.getEntriesCount() != 2 )
|
|
{
|
|
qWarning() << "SaveLoadThread::GetPCSaves() -> save contains more than 2 entries" << fi3.at( i ).absoluteFilePath();
|
|
zip.close();
|
|
continue;
|
|
}
|
|
QuaZipFile file(&zip);
|
|
QString name;
|
|
for( bool more = zip.goToFirstFile(); more; more = zip.goToNextFile() )
|
|
{
|
|
if( !file.open( QIODevice::ReadOnly ) )
|
|
{
|
|
qWarning("SaveLoadThread::GetPCSaves(): file.open(): %d", file.getZipError() );
|
|
zip.close();
|
|
continue;
|
|
}
|
|
name = file.getActualFileName();
|
|
//qDebug() << "extracting" << name << "from the archive";
|
|
if( file.getZipError() != UNZ_OK )
|
|
{
|
|
qWarning("SaveLoadThread::GetPCSaves(): file.getFileName(): %d", file.getZipError());
|
|
file.close();
|
|
zip.close();
|
|
continue;
|
|
}
|
|
if( name != "data.bin" && name != "info.txt" )
|
|
{
|
|
qWarning() << "SaveLoadThread::GetPCSaves() -> zip contians bad filename" << fi3.at( i ).absoluteFilePath() << name;
|
|
file.close();
|
|
zip.close();
|
|
continue;
|
|
}
|
|
/*if( name == "data.bin" && banner.size() )//theres already a banner been extracted, no need to get another one
|
|
{
|
|
file.close();
|
|
continue;
|
|
}*/
|
|
QByteArray unc = file.readAll();
|
|
//qDebug() << "read" << hex << unc.size();
|
|
if( file.getZipError() != UNZ_OK )
|
|
{
|
|
qWarning("SaveLoadThread::GetPCSaves(): file.getFileName(): %d", file.getZipError());
|
|
file.close();
|
|
zip.close();
|
|
continue;
|
|
}
|
|
file.close();
|
|
//read the data.bin to get the banner
|
|
if( name == "data.bin" )
|
|
{
|
|
if( info.banner.isEmpty() )
|
|
info.banner = SaveDataBin::GetBanner( unc );
|
|
|
|
sSize = SaveDataBin::GetSize( unc );
|
|
bnrOk = true;
|
|
}
|
|
else if( name == "info.txt" )
|
|
{
|
|
desc = QString( unc );
|
|
txtOk = true;
|
|
}
|
|
}
|
|
if( txtOk && bnrOk )
|
|
{
|
|
info.descriptions << desc;
|
|
info.sizes << sSize;
|
|
info.paths << fi3.at( i ).absoluteFilePath();
|
|
}
|
|
zip.close();
|
|
emit SendProgress( (int)( ( (float)( ++done) / (float)total ) * (float)100 ) );
|
|
}
|
|
//QString tidStr = upper.fileName() + lower.fileName();
|
|
//emit SendPcItem( banner, tidStr, descriptions, size );
|
|
if( info.sizes.size() )
|
|
emit SendPcItem( info );
|
|
}
|
|
}
|
|
emit SendProgress( 100 );
|
|
}
|
|
|
|
const QString SaveLoadThread::NandBasePath()
|
|
{
|
|
if( isRunning() )
|
|
return QString();
|
|
return nand.BasePath();
|
|
}
|
|
|
|
bool SaveLoadThread::InstallSaveToSneekNand( SaveGame save )
|
|
{
|
|
if( isRunning() )
|
|
return false;
|
|
return nand.InstallSave( save );
|
|
}
|
|
|
|
void SaveLoadThread::GetSavesFromNandDump()
|
|
{
|
|
emit SendProgress( 0 );
|
|
QString base = nand.BasePath();
|
|
if( base.isEmpty() )
|
|
return;
|
|
|
|
QMap< quint64, quint32 > list = nand.GetSaveList();
|
|
QMap< quint64, quint32 >::Iterator it = list.begin();
|
|
int i = 0;
|
|
int s = list.size();
|
|
while( it != list.end() && !abort )
|
|
{
|
|
quint64 tid = it.key();
|
|
quint32 size = it.value();
|
|
it++;
|
|
|
|
QString tidStr = QString( "%1" ).arg( tid, 16, 16, QChar( '0' ) );
|
|
QString bnrPath = tidStr;
|
|
bnrPath.insert( 8, "/" );
|
|
bnrPath.prepend( "/title/" );
|
|
bnrPath.append( "/data/banner.bin" );
|
|
|
|
QByteArray bnr = nand.GetFile( bnrPath );
|
|
emit SendProgress( (int)( ( (float)( ++i ) / (float)s ) * (float)100 ) );
|
|
if( bnr.isEmpty() )
|
|
continue;
|
|
|
|
emit SendSneekItem( bnr, tidStr, size );
|
|
}
|
|
|
|
emit SendProgress( 100 );
|
|
}
|
|
|
|
bool SaveLoadThread::DeleteSaveFromSneekNand( quint64 tid )
|
|
{
|
|
if( isRunning() )
|
|
return false;
|
|
return nand.DeleteSave( tid );
|
|
}
|