mirror of
https://github.com/martravi/wiiqt6.git
synced 2024-11-26 14:44:20 +01:00
4ff87cd24e
* corrected TIDs used when downloading an entire update * set the path to the nand dump when clicking the button to download something * read data from cache even if title version is 0 git-svn-id: http://wiiqt.googlecode.com/svn/trunk@8 389f4c8b-5dfe-645f-db0e-df882bc27289
356 lines
10 KiB
C++
356 lines
10 KiB
C++
#include "mainwindow.h"
|
|
#include "ui_mainwindow.h"
|
|
#include "settingtxtdialog.h"
|
|
#include "uidmap.h"
|
|
#include "sha1.h"
|
|
#include "tiktmd.h"
|
|
#include "tools.h"
|
|
#include "aes.h"
|
|
#include "wad.h"
|
|
|
|
MainWindow::MainWindow( QWidget *parent ) : QMainWindow( parent ), ui( new Ui::MainWindow ), nus ( this )
|
|
{
|
|
ui->setupUi(this);
|
|
Wad::SetGlobalCert( QByteArray( (const char*)&certs_dat, CERTS_DAT_SIZE ) );
|
|
|
|
//connect to the nus object so we can respond to what it is saying with pretty stuff in the gui
|
|
connect( &nus, SIGNAL( SendDownloadProgress( int ) ), ui->progressBar_dl, SLOT( setValue( int ) ) );
|
|
connect( &nus, SIGNAL( SendTitleProgress( int ) ), ui->progressBar_title, SLOT( setValue( int ) ) );
|
|
connect( &nus, SIGNAL( SendTotalProgress( int ) ), ui->progressBar_whole, SLOT( setValue( int ) ) );
|
|
connect( &nus, SIGNAL( SendText( QString ) ), ui->statusBar, SLOT( showMessage( QString ) ) );
|
|
connect( &nus, SIGNAL( SendError( const QString &, NusJob ) ), this, SLOT( GetError( const QString &, NusJob ) ) );
|
|
connect( &nus, SIGNAL( SendDone() ), this, SLOT( NusIsDone() ) );
|
|
connect( &nus, SIGNAL( SendData( NusJob ) ), this, SLOT( ReceiveTitleFromNus( NusJob) ) );
|
|
|
|
//TODO, really get these paths from settings
|
|
ui->lineEdit_cachePath->setText( cachePath );
|
|
ui->lineEdit_nandPath->setText( nandPath );
|
|
//nand.SetPath( nandPath );
|
|
nus.SetCachePath( cachePath );
|
|
|
|
|
|
}
|
|
|
|
MainWindow::~MainWindow()
|
|
{
|
|
delete ui;
|
|
}
|
|
|
|
//some slots to respond to the NUS downloader
|
|
void MainWindow::GetError( const QString &message, NusJob job )
|
|
{
|
|
QString dataStuff = QString( "%1 items:" ).arg( job.data.size() );
|
|
for( int i = 0; i < job.data.size(); i++ )
|
|
dataStuff += QString( " %1" ).arg( job.data.at( i ).size(), 0, 16, QChar( ' ' ) );
|
|
|
|
QString str = tr( "<b>Error getting title from NUS: %1</b>" ).arg( message );
|
|
QString j = QString( "NusJob( %1, %2, %3, %4 )<br>" )
|
|
.arg( job.tid, 16, 16, QChar( '0' ) )
|
|
.arg( job.version ).arg( job.decrypt ? "decrypted" : "encrypted" )
|
|
.arg( dataStuff );
|
|
|
|
|
|
ui->plainTextEdit_log->appendHtml( str );
|
|
ui->plainTextEdit_log->appendHtml( j );
|
|
}
|
|
|
|
void MainWindow::ShowMessage( const QString& mes )
|
|
{
|
|
QString str = mes + "<br>";
|
|
ui->plainTextEdit_log->appendHtml( str );
|
|
}
|
|
|
|
void MainWindow::NusIsDone()
|
|
{
|
|
QString str = tr( "NUS ojbect is done working<br>" );
|
|
ui->plainTextEdit_log->appendHtml( str );
|
|
ui->statusBar->showMessage( tr( "Done" ), 5000 );
|
|
if( ui->radioButton_folder->isChecked() )
|
|
{
|
|
ui->lineEdit_extractPath->setEnabled( true );
|
|
ui->pushButton_decFolder->setEnabled( true );
|
|
}
|
|
else if( ui->radioButton_nand->isChecked() )
|
|
{
|
|
ui->lineEdit_nandPath->setEnabled( true );
|
|
ui->pushButton_nandPath->setEnabled( true );
|
|
|
|
//write the uid.sys and content.map to disc
|
|
nand.Flush();
|
|
|
|
//make sure there is a setting.txt
|
|
QByteArray set = nand.GetSettingTxt();
|
|
if( set.isEmpty() )
|
|
{
|
|
set = SettingTxtDialog::Edit( this );
|
|
if( !set.isEmpty() )
|
|
nand.SetSettingTxt( set );
|
|
}
|
|
}
|
|
else if( ui->radioButton_wad->isChecked() )
|
|
{
|
|
ui->lineEdit_wad->setEnabled( true );
|
|
ui->pushButton_wad->setEnabled( true );
|
|
}
|
|
|
|
ui->radioButton_folder->setEnabled( true );
|
|
ui->radioButton_nand->setEnabled( true );
|
|
ui->radioButton_wad->setEnabled( true );
|
|
}
|
|
|
|
void MainWindow::ReceiveTitleFromNus( NusJob job )
|
|
{
|
|
QString str = tr( "Received a completed download from NUS" );
|
|
QString title = QString( "%1v%2" ).arg( job.tid, 16, 16, QChar( '0' ) ).arg( job.version );
|
|
|
|
ui->plainTextEdit_log->appendHtml( str );
|
|
|
|
//do something with the data we got
|
|
if( ui->radioButton_folder->isChecked() )
|
|
{
|
|
|
|
}
|
|
else if( ui->radioButton_nand->isChecked() )
|
|
{
|
|
bool ok = nand.InstallNusItem( job );
|
|
if( ok )
|
|
ShowMessage( tr( "Installed %1 title to nand" ).arg( title ) );
|
|
else
|
|
ShowMessage( tr( "<b>Error %1 title to nand</b>" ).arg( title ) );
|
|
}
|
|
else if( ui->radioButton_wad->isChecked() )
|
|
{
|
|
Wad wad( job.data );
|
|
if( !wad.IsOk() )
|
|
{
|
|
ShowMessage( "<b>Error making a wad from " + title + "<\b>" );
|
|
return;
|
|
}
|
|
QFileInfo fi( ui->lineEdit_wad->text() );
|
|
if( fi.isFile() )
|
|
{
|
|
ShowMessage( "<b>" + ui->lineEdit_wad->text() + " is a file. I need a folder<\b>" );
|
|
return;
|
|
}
|
|
if( !fi.exists() )
|
|
{
|
|
ShowMessage( "<b>" + fi.absoluteFilePath() + " is not a folder!\nTrying to create it...<\b>" );
|
|
if( !QDir().mkpath( ui->lineEdit_wad->text() ) )
|
|
{
|
|
ShowMessage( "<b>Failed to make the directory!<\b>" );
|
|
return;
|
|
}
|
|
}
|
|
QByteArray w = wad.Data();
|
|
if( w.isEmpty() )
|
|
{
|
|
ShowMessage( "<b>Error creating wad<br>" + wad.LastError() + "<\b>" );
|
|
return;
|
|
}
|
|
|
|
QString name = wad.WadName( fi.absoluteFilePath() );
|
|
if( name.isEmpty() )
|
|
{
|
|
name = QFileDialog::getSaveFileName( this, tr( "Filename for %1" ).arg( title ), fi.absoluteFilePath() );
|
|
if( name.isEmpty() )
|
|
{
|
|
ShowMessage( "<b>No save name given, aborting<\b>" );
|
|
return;
|
|
}
|
|
}
|
|
QFile file( fi.absoluteFilePath() + "/" + name );
|
|
if( !file.open( QIODevice::WriteOnly ) )
|
|
{
|
|
ShowMessage( "<b>Cant open " + fi.absoluteFilePath() + "/" + name + " for writing<\b>" );
|
|
return;
|
|
}
|
|
file.write( w );
|
|
file.close();
|
|
ShowMessage( "Saved " + title + " to " + fi.absoluteFilePath() + "/" + name );
|
|
}
|
|
|
|
//bool r = nand.InstallNusItem( job );
|
|
//qDebug() << "install:" << r;
|
|
}
|
|
|
|
//clicked the button to get a title
|
|
void MainWindow::on_pushButton_GetTitle_clicked()
|
|
{
|
|
bool ok = false;
|
|
bool wholeUpdate = false;
|
|
quint64 tid = 0;
|
|
quint32 ver = 0;
|
|
if( ui->lineEdit_tid->text().size() == 4 )
|
|
{
|
|
wholeUpdate = true;
|
|
}
|
|
else
|
|
{
|
|
tid = ui->lineEdit_tid->text().toLongLong( &ok, 16 );
|
|
if( !ok )
|
|
{
|
|
ShowMessage( "<b>Error converting \"" + ui->lineEdit_tid->text() + "\" to a hex number.</b>" );
|
|
return;
|
|
}
|
|
ver = TITLE_LATEST_VERSION;
|
|
if( !ui->lineEdit_version->text().isEmpty() )
|
|
{
|
|
ver = ui->lineEdit_version->text().toInt( &ok, 10 );
|
|
if( !ok )
|
|
{
|
|
ShowMessage( "<b>Error converting \"" + ui->lineEdit_version->text() + "\" to a decimal number.</b>" );
|
|
return;
|
|
}
|
|
if( ver > 0xffff )
|
|
{
|
|
ShowMessage( tr( "<b>Version %1 is too high. Max is 65535</b>" ).arg( ver ) );
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
//decide how we want nus to give us the title
|
|
bool decrypt = true;
|
|
if( ui->radioButton_folder->isChecked() )
|
|
{
|
|
if( ui->lineEdit_extractPath->text().isEmpty() )
|
|
{
|
|
ShowMessage( tr( "<b>No path given to save downloads in.</b>" ) );
|
|
return;
|
|
}
|
|
ui->lineEdit_extractPath->setEnabled( false );
|
|
ui->pushButton_decFolder->setEnabled( false );
|
|
}
|
|
else if( ui->radioButton_nand->isChecked() )
|
|
{
|
|
if( !nand.SetPath( ui->lineEdit_nandPath->text() ) )
|
|
{
|
|
ShowMessage( tr( "<b>Error setting the basepath of the nand to %1</b>" )
|
|
.arg( QFileInfo( ui->lineEdit_nandPath->text() ).absoluteFilePath() ) );
|
|
return;
|
|
}
|
|
if( ui->lineEdit_nandPath->text().isEmpty() )
|
|
{
|
|
ShowMessage( tr( "<b>No path given for nand dump base.</b>" ) );
|
|
return;
|
|
}
|
|
ui->lineEdit_nandPath->setEnabled( false );
|
|
ui->pushButton_nandPath->setEnabled( false );
|
|
}
|
|
else if( ui->radioButton_wad->isChecked() )
|
|
{
|
|
if( ui->lineEdit_wad->text().isEmpty() )
|
|
{
|
|
ShowMessage( tr( "<b>No path given to save wads in.</b>" ) );
|
|
return;
|
|
}
|
|
decrypt = false;
|
|
ui->lineEdit_wad->setEnabled( false );
|
|
ui->pushButton_wad->setEnabled( false );
|
|
|
|
}
|
|
|
|
ui->radioButton_folder->setEnabled( false );
|
|
ui->radioButton_nand->setEnabled( false );
|
|
ui->radioButton_wad->setEnabled( false );
|
|
//dont set these to 0 in case the button is pressed while something else is already being downloaded
|
|
//ui->progressBar_dl->setValue( 0 );
|
|
//ui->progressBar_title->setValue( 0 );
|
|
//ui->progressBar_whole->setValue( 0 );
|
|
nus.SetCachePath( ui->lineEdit_cachePath->text() );
|
|
if( wholeUpdate )
|
|
{
|
|
if( !nus.GetUpdate( ui->lineEdit_tid->text(), decrypt ) )
|
|
{
|
|
ShowMessage( tr( "<b>I dont know the titles that were in the %1 update</b>" ).arg( ui->lineEdit_tid->text() ) );
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
nus.Get( tid, decrypt, ver );
|
|
}
|
|
}
|
|
|
|
//ratio buttons toggled
|
|
void MainWindow::on_radioButton_nand_toggled( bool checked )
|
|
{
|
|
ui->lineEdit_nandPath->setEnabled( checked );
|
|
ui->pushButton_nandPath->setEnabled( checked );
|
|
}
|
|
|
|
void MainWindow::on_radioButton_folder_toggled( bool checked )
|
|
{
|
|
ui->lineEdit_extractPath->setEnabled( checked );
|
|
ui->pushButton_decFolder->setEnabled( checked );
|
|
}
|
|
|
|
void MainWindow::on_radioButton_wad_toggled( bool checked )
|
|
{
|
|
ui->lineEdit_wad->setEnabled( checked );
|
|
ui->pushButton_wad->setEnabled( checked );
|
|
}
|
|
|
|
//search for a path to use as the nand basepath
|
|
void MainWindow::on_pushButton_nandPath_clicked()
|
|
{
|
|
QString path = ui->lineEdit_nandPath->text().isEmpty() ? "/media" : ui->lineEdit_nandPath->text();
|
|
QString f = QFileDialog::getExistingDirectory( this, tr( "Select Nand Base Folder" ), path );
|
|
if( f.isEmpty() )
|
|
return;
|
|
|
|
ui->lineEdit_nandPath->setText( f );
|
|
nus.SetCachePath( ui->lineEdit_cachePath->text() );
|
|
}
|
|
|
|
void MainWindow::on_pushButton_decFolder_clicked()
|
|
{
|
|
QString path = ui->lineEdit_extractPath->text().isEmpty() ? "/media" : ui->lineEdit_extractPath->text();
|
|
QString f = QFileDialog::getExistingDirectory( this, tr( "Select folder to decrypt this title to" ), path );
|
|
if( f.isEmpty() )
|
|
return;
|
|
|
|
ui->lineEdit_extractPath->setText( f );
|
|
}
|
|
|
|
void MainWindow::on_pushButton_wad_clicked()
|
|
{
|
|
QString path = ui->lineEdit_wad->text().isEmpty() ? QDir::currentPath() : ui->lineEdit_wad->text();
|
|
QString f = QFileDialog::getExistingDirectory( this, tr( "Select folder to save wads to" ), path );
|
|
if( f.isEmpty() )
|
|
return;
|
|
|
|
ui->lineEdit_wad->setText( f );
|
|
}
|
|
|
|
|
|
/*
|
|
3.4j-
|
|
Error getting title from NUS: Error downloading part of the title.
|
|
NusJob( 0001000248414b4a, 2, decrypted, 0 items: )
|
|
|
|
Error getting title from NUS: Error downloading part of the title.
|
|
NusJob( 0001000248414c4a, 2, decrypted, 0 items: )
|
|
|
|
Received a completed download from NUS
|
|
Installed 0001000248415941v2 title to nand
|
|
|
|
Received a completed download from NUS
|
|
Error 000100084843434av0 title to nand
|
|
|
|
4.0j
|
|
Error getting title from NUS: Error downloading part of the title.
|
|
NusJob( 0001000248414b4a, 2, decrypted, 0 items: )
|
|
|
|
Error getting title from NUS: Error downloading part of the title.
|
|
NusJob( 0001000248414c4a, 2, decrypted, 0 items: )
|
|
|
|
Received a completed download from NUS
|
|
Installed 0001000248415941v3 title to nand
|
|
|
|
Error getting title from NUS: Error downloading part of the title.
|
|
NusJob( 000100024843434a, 1, decrypted, 0 items: )
|
|
|
|
Received a completed download from NUS
|
|
Error 000100084843434av0 title to nand
|
|
*/
|