/* * This file Copyright (C) Mnemosyne LLC * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 * as published by the Free Software Foundation. * * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html * * $Id$ */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include /* mime64 */ #include "add-data.h" #include "file-tree.h" #include "hig.h" #include "options.h" #include "prefs.h" #include "session.h" #include "torrent.h" #include "utils.h" /*** **** ***/ void FileAdded :: executed( int64_t tag, const QString& result, struct tr_benc * arguments ) { Q_UNUSED( arguments ); if( tag != myTag ) return; if( result == "success" ) if( !myDelFile.isEmpty( ) ) QFile( myDelFile ).remove( ); if( result != "success" ) { QString text = result; for( int i=0, n=text.size(); i%1

%2

").arg(text).arg(myName) ); } deleteLater(); } /*** **** ***/ Options :: Options( Session& session, const Prefs& prefs, const AddData& addme, QWidget * parent ): QDialog( parent, Qt::Dialog ), mySession( session ), myAdd( addme ), myHaveInfo( false ), myDestinationButton( 0 ), myVerifyButton( 0 ), myVerifyFile( 0 ), myVerifyHash( QCryptographicHash::Sha1 ) { setWindowTitle( tr( "Open Torrent" ) ); QFontMetrics fontMetrics( font( ) ); QGridLayout * layout = new QGridLayout( this ); int row = 0; const int iconSize( style( )->pixelMetric( QStyle :: PM_SmallIconSize ) ); QIcon fileIcon = style( )->standardIcon( QStyle::SP_FileIcon ); const QPixmap filePixmap = fileIcon.pixmap( iconSize ); QPushButton * p; int width = fontMetrics.size( 0, "This is a pretty long torrent filename indeed.torrent" ).width( ); QLabel * l = new QLabel( tr( "&Torrent file:" ) ); layout->addWidget( l, row, 0, Qt::AlignLeft ); p = myFileButton = new QPushButton; p->setIcon( filePixmap ); p->setMinimumWidth( width ); p->setStyleSheet( "text-align: left; padding-left: 5; padding-right: 5" ); p->installEventFilter( this ); layout->addWidget( p, row, 1 ); l->setBuddy( p ); connect( p, SIGNAL(clicked(bool)), this, SLOT(onFilenameClicked())); if( session.isLocal( ) ) { const QFileIconProvider iconProvider; const QIcon folderIcon = iconProvider.icon( QFileIconProvider::Folder ); const QPixmap folderPixmap = folderIcon.pixmap( iconSize ); l = new QLabel( tr( "&Destination folder:" ) ); layout->addWidget( l, ++row, 0, Qt::AlignLeft ); myDestination.setPath( prefs.getString( Prefs :: DOWNLOAD_DIR ) ); p = myDestinationButton = new QPushButton; p->setIcon( folderPixmap ); p->setStyleSheet( "text-align: left; padding-left: 5; padding-right: 5" ); p->installEventFilter( this ); layout->addWidget( p, row, 1 ); l->setBuddy( p ); connect( p, SIGNAL(clicked(bool)), this, SLOT(onDestinationClicked())); } myTree = new FileTreeView; layout->addWidget( myTree, ++row, 0, 1, 2 ); if( !session.isLocal( ) ) myTree->hideColumn( 1 ); // hide the % done, since we've no way of knowing QComboBox * m = new QComboBox; m->addItem( tr( "High" ), TR_PRI_HIGH ); m->addItem( tr( "Normal" ), TR_PRI_NORMAL ); m->addItem( tr( "Low" ), TR_PRI_LOW ); m->setCurrentIndex( 1 ); // Normal myPriorityCombo = m; l = new QLabel( tr( "Torrent &priority:" ) ); l->setBuddy( m ); layout->addWidget( l, ++row, 0, Qt::AlignLeft ); layout->addWidget( m, row, 1 ); if( session.isLocal( ) ) { p = myVerifyButton = new QPushButton( tr( "&Verify Local Data" ) ); layout->addWidget( p, ++row, 0, Qt::AlignLeft ); } QCheckBox * c; c = myStartCheck = new QCheckBox( tr( "&Start when added" ) ); c->setChecked( prefs.getBool( Prefs :: START ) ); layout->addWidget( c, ++row, 0, 1, 2, Qt::AlignLeft ); c = myTrashCheck = new QCheckBox( tr( "Mo&ve .torrent file to the trash" ) ); c->setChecked( prefs.getBool( Prefs :: TRASH_ORIGINAL ) ); layout->addWidget( c, ++row, 0, 1, 2, Qt::AlignLeft ); QDialogButtonBox * b = new QDialogButtonBox( QDialogButtonBox::Open|QDialogButtonBox::Cancel, Qt::Horizontal, this ); connect( b, SIGNAL(rejected()), this, SLOT(deleteLater()) ); connect( b, SIGNAL(accepted()), this, SLOT(onAccepted()) ); layout->addWidget( b, ++row, 0, 1, 2 ); layout->setRowStretch( 2, 2 ); layout->setColumnStretch( 1, 2 ); layout->setSpacing( HIG :: PAD ); connect( myTree, SIGNAL(priorityChanged(const QSet&,int)), this, SLOT(onPriorityChanged(const QSet&,int))); connect( myTree, SIGNAL(wantedChanged(const QSet&,bool)), this, SLOT(onWantedChanged(const QSet&,bool))); if( session.isLocal( ) ) connect( myVerifyButton, SIGNAL(clicked(bool)), this, SLOT(onVerify())); connect( &myVerifyTimer, SIGNAL(timeout()), this, SLOT(onTimeout())); reload( ); } Options :: ~Options( ) { clearInfo( ); } /*** **** ***/ void Options :: refreshButton( QPushButton * p, const QString& text, int width ) { if( width <= 0 ) width = p->width( ); width -= 15; QFontMetrics fontMetrics( font( ) ); QString str = fontMetrics.elidedText( text, Qt::ElideRight, width ); p->setText( str ); } void Options :: refreshFileButton( int width ) { QString text; switch( myAdd.type ) { case AddData::FILENAME: text = QFileInfo(myAdd.filename).baseName(); break; case AddData::URL: text = myAdd.url.toString(); break; case AddData::MAGNET: text = myAdd.magnet; break; default: break; } refreshButton( myFileButton, text, width ); } void Options :: refreshDestinationButton( int width ) { if( myDestinationButton != 0 ) refreshButton( myDestinationButton, myDestination.absolutePath(), width ); } bool Options :: eventFilter( QObject * o, QEvent * event ) { if( o==myFileButton && event->type() == QEvent::Resize ) { refreshFileButton( dynamic_cast(event)->size().width() ); } if( o==myDestinationButton && event->type() == QEvent::Resize ) { refreshDestinationButton( dynamic_cast(event)->size().width() ); } return false; } /*** **** ***/ void Options :: clearInfo( ) { if( myHaveInfo ) tr_metainfoFree( &myInfo ); myHaveInfo = false; myFiles.clear( ); } void Options :: reload( ) { clearInfo( ); clearVerify( ); tr_ctor * ctor = tr_ctorNew( 0 ); switch( myAdd.type ) { case AddData::MAGNET: tr_ctorSetMetainfoFromMagnetLink( ctor, myAdd.magnet.toUtf8().constData() ); break; case AddData::FILENAME: tr_ctorSetMetainfoFromFile( ctor, myAdd.filename.toUtf8().constData() ); break; case AddData::METAINFO: tr_ctorSetMetainfo( ctor, (const uint8_t*)myAdd.metainfo.constData(), myAdd.metainfo.size() ); break; default: break; } const int err = tr_torrentParse( ctor, &myInfo ); myHaveInfo = !err; tr_ctorFree( ctor ); myTree->clear( ); myFiles.clear( ); myPriorities.clear( ); myWanted.clear( ); if( myHaveInfo ) { myPriorities.insert( 0, myInfo.fileCount, TR_PRI_NORMAL ); myWanted.insert( 0, myInfo.fileCount, true ); for( tr_file_index_t i=0; iupdate( myFiles ); } void Options :: onPriorityChanged( const QSet& fileIndices, int priority ) { foreach( int i, fileIndices ) myPriorities[i] = priority; } void Options :: onWantedChanged( const QSet& fileIndices, bool isWanted ) { foreach( int i, fileIndices ) myWanted[i] = isWanted; } void Options :: onAccepted( ) { // rpc spec section 3.4 "adding a torrent" const int64_t tag = mySession.getUniqueTag( ); tr_benc top; tr_bencInitDict( &top, 3 ); tr_bencDictAddStr( &top, "method", "torrent-add" ); tr_bencDictAddInt( &top, "tag", tag ); tr_benc * args( tr_bencDictAddDict( &top, "arguments", 10 ) ); // "download-dir" if( myDestinationButton ) tr_bencDictAddStr( args, "download-dir", myDestination.absolutePath().toUtf8().constData() ); // "metainfo" switch( myAdd.type ) { case AddData::MAGNET: tr_bencDictAddStr( args, "filename", myAdd.magnet.toUtf8().constData() ); break; case AddData::URL: tr_bencDictAddStr( args, "filename", myAdd.url.toString().toUtf8().constData() ); break; case AddData::FILENAME: case AddData::METAINFO: { const QByteArray b64 = myAdd.toBase64( ); tr_bencDictAddRaw( args, "metainfo", b64.constData(), b64.size() ); break; } default: std::cerr << "unhandled AddData.type: " << myAdd.type << std::endl; } // paused tr_bencDictAddBool( args, "paused", !myStartCheck->isChecked( ) ); // priority const int index = myPriorityCombo->currentIndex( ); const int priority = myPriorityCombo->itemData(index).toInt( ); tr_bencDictAddInt( args, "bandwidthPriority", priority ); // files-unwanted int count = myWanted.count( false ); if( count > 0 ) { tr_benc * l = tr_bencDictAddList( args, "files-unwanted", count ); for( int i=0, n=myWanted.size(); i 0 ) { tr_benc * l = tr_bencDictAddList( args, "priority-low", count ); for( int i=0, n=myPriorities.size(); i 0 ) { tr_benc * l = tr_bencDictAddList( args, "priority-high", count ); for( int i=0, n=myPriorities.size(); iisChecked( ) && ( myAdd.type==AddData::FILENAME ) ) fileAdded->setFileToDelete( myAdd.filename ); connect( &mySession, SIGNAL(executed(int64_t,const QString&, struct tr_benc*)), fileAdded, SLOT(executed(int64_t,const QString&, struct tr_benc*))); //std::cerr << tr_bencToStr(&top,TR_FMT_JSON,NULL) << std::endl; mySession.exec( &top ); tr_bencFree( &top ); deleteLater( ); } void Options :: onFilenameClicked( ) { if( myAdd.type == AddData::FILENAME ) { QFileDialog * d = new QFileDialog( this, tr( "Open Torrent" ), QFileInfo(myAdd.filename).absolutePath(), tr( "Torrent Files (*.torrent);;All Files (*.*)" ) ); d->setFileMode( QFileDialog::ExistingFile ); connect( d, SIGNAL(filesSelected(const QStringList&)), this, SLOT(onFilesSelected(const QStringList&)) ); d->show( ); } } void Options :: onFilesSelected( const QStringList& files ) { if( files.size() == 1 ) { myAdd.set( files.at(0) ); refreshFileButton( ); reload( ); } } void Options :: onDestinationClicked( ) { QFileDialog * d = new QFileDialog( this, tr( "Select Destination" ), myDestination.absolutePath( ) ); d->setFileMode( QFileDialog::Directory ); connect( d, SIGNAL(filesSelected(const QStringList&)), this, SLOT(onDestinationsSelected(const QStringList&)) ); d->show( ); } void Options :: onDestinationsSelected( const QStringList& destinations ) { if( destinations.size() == 1 ) { const QString& destination( destinations.first( ) ); myDestination.setPath( destination ); refreshDestinationButton( ); } } /*** **** **** VERIFY **** ***/ void Options :: clearVerify( ) { myVerifyHash.reset( ); myVerifyFile.close( ); myVerifyFilePos = 0; myVerifyFlags.clear( ); myVerifyFileIndex = 0; myVerifyPieceIndex = 0; myVerifyPiecePos = 0; myVerifyTimer.stop( ); for( int i=0, n=myFiles.size(); iupdate( myFiles ); } void Options :: onVerify( ) { //std::cerr << "starting to verify..." << std::endl; clearVerify( ); myVerifyFlags.insert( 0, myInfo.pieceCount, false ); myVerifyTimer.setSingleShot( false ); myVerifyTimer.start( 0 ); } namespace { uint64_t getPieceSize( const tr_info * info, tr_piece_index_t pieceIndex ) { if( pieceIndex != info->pieceCount - 1 ) return info->pieceSize; return info->totalSize % info->pieceSize; } } void Options :: onTimeout( ) { const tr_file * file = &myInfo.files[myVerifyFileIndex]; if( !myVerifyFilePos && !myVerifyFile.isOpen( ) ) { const QFileInfo fileInfo( myDestination, QString::fromUtf8( file->name ) ); myVerifyFile.setFileName( fileInfo.absoluteFilePath( ) ); //std::cerr << "opening file" << qPrintable(fileInfo.absoluteFilePath()) << std::endl; myVerifyFile.open( QIODevice::ReadOnly ); } int64_t leftInPiece = getPieceSize( &myInfo, myVerifyPieceIndex ) - myVerifyPiecePos; int64_t leftInFile = file->length - myVerifyFilePos; int64_t bytesThisPass = std::min( leftInFile, leftInPiece ); bytesThisPass = std::min( bytesThisPass, (int64_t)sizeof( myVerifyBuf ) ); if( myVerifyFile.isOpen() && myVerifyFile.seek( myVerifyFilePos ) ) { int64_t numRead = myVerifyFile.read( myVerifyBuf, bytesThisPass ); if( numRead == bytesThisPass ) myVerifyHash.addData( myVerifyBuf, numRead ); } leftInPiece -= bytesThisPass; leftInFile -= bytesThisPass; myVerifyPiecePos += bytesThisPass; myVerifyFilePos += bytesThisPass; myVerifyBins[myVerifyFileIndex] += bytesThisPass; if( leftInPiece == 0 ) { const QByteArray result( myVerifyHash.result( ) ); const bool matches = !memcmp( result.constData(), myInfo.pieces[myVerifyPieceIndex].hash, SHA_DIGEST_LENGTH ); myVerifyFlags[myVerifyPieceIndex] = matches; myVerifyPiecePos = 0; ++myVerifyPieceIndex; myVerifyHash.reset( ); FileList changedFiles; if( matches ) { mybins_t::const_iterator i; for( i=myVerifyBins.begin(); i!=myVerifyBins.end(); ++i ) { TrFile& f( myFiles[i.key( )] ); f.have += i.value( ); changedFiles.append( f ); } } myTree->update( changedFiles ); myVerifyBins.clear( ); } if( leftInFile == 0 ) { //std::cerr << "closing file" << std::endl; myVerifyFile.close( ); ++myVerifyFileIndex; myVerifyFilePos = 0; } bool done = myVerifyPieceIndex >= myInfo.pieceCount; if( done ) { uint64_t have = 0; foreach( const TrFile& f, myFiles ) have += f.have; if( !have ) // everything failed { // did the user accidentally specify the child directory instead of the parent? const QStringList tokens = QString(file->name).split('/'); if( !tokens.empty() && myDestination.dirName()==tokens.at(0) ) { // move up one directory and try again myDestination.cdUp( ); refreshDestinationButton( -1 ); onVerify( ); done = false; } } } if( done ) myVerifyTimer.stop( ); }