wiiqt/nand_dump/mainwindow.cpp

#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"

MainWindow::MainWindow( QWidget *parent ) : QMainWindow( parent ), ui( new Ui::MainWindow ), nus ( this )
{
    ui->setupUi(this);

    //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 );
    }
    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 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( "Received a completed download from NUS" );
    QString title = QString( "%1v%2" ).arg( job.tid, 16, 16, QChar( '0' ) ).arg( job.version );
    /*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( j );*/
    ui->plainTextEdit_log->appendHtml( str );
return;
    //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() )
    {
    }

    //bool r = nand.InstallNusItem( job );
    //qDebug() << "install:" << r;
}

//clicked the button to get a title
void MainWindow::on_pushButton_GetTitle_clicked()
{
    bool ok = false;
    quint64 tid = ui->lineEdit_tid->text().toLongLong( &ok, 16 );
    if( !ok )
    {
	ShowMessage( "<b>Error converting \"" + ui->lineEdit_tid->text() + "\" to a hex number.</b>" );
	return;
    }
    quint32 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( 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() );
    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() ? "/media" : 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 );
}