Diffstat (limited to 'core/launcher/transferserver.cpp') (more/less context) (ignore whitespace changes)
| -rw-r--r-- | core/launcher/transferserver.cpp | 1245 |
1 files changed, 1245 insertions, 0 deletions
diff --git a/core/launcher/transferserver.cpp b/core/launcher/transferserver.cpp new file mode 100644 index 0000000..87a49eb --- a/dev/null +++ b/core/launcher/transferserver.cpp @@ -0,0 +1,1245 @@ +/********************************************************************** +** Copyright (C) 2000 Trolltech AS. All rights reserved. +** +** This file is part of Qtopia Environment. +** +** This file may be distributed and/or modified under the terms of the +** GNU General Public License version 2 as published by the Free Software +** Foundation and appearing in the file LICENSE.GPL included in the +** packaging of this file. +** +** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE +** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. +** +** See http://www.trolltech.com/gpl/ for GPL licensing information. +** +** Contact info@trolltech.com if any conditions of this licensing are +** not clear to you. +** +**********************************************************************/ +#define _XOPEN_SOURCE +#include <pwd.h> +#include <sys/types.h> +#include <unistd.h> +#include <stdlib.h> + +#if defined(_OS_LINUX_) +#include <shadow.h> +#endif + +#include <qdir.h> +#include <qfile.h> +#include <qtextstream.h> +#include <qdatastream.h> +#include <qmessagebox.h> +#include <qstringlist.h> +#include <qfileinfo.h> +#include <qregexp.h> +//#include <qpe/qcopchannel_qws.h> +#include <qpe/qprocess.h> +#include <qpe/process.h> +#include <qpe/config.h> +#include <qpe/qcopenvelope_qws.h> + +#include "transferserver.h" + +const int block_size = 51200; + +TransferServer::TransferServer( Q_UINT16 port, QObject *parent = 0, + const char* name = 0) + : QServerSocket( port, 1, parent, name ) +{ + if ( !ok() ) + qWarning( "Failed to bind to port %d", port ); +} + +TransferServer::~TransferServer() +{ + +} + +void TransferServer::newConnection( int socket ) +{ + (void) new ServerPI( socket, this ); +} + +bool accessAuthorized(QHostAddress peeraddress) +{ + Config cfg("Security"); + cfg.setGroup("Sync"); + uint auth_peer = cfg.readNumEntry("auth_peer",0xc0a80100); + uint auth_peer_bits = cfg.readNumEntry("auth_peer_bits",24); + bool ok = (peeraddress.ip4Addr() & (((1<<auth_peer_bits)-1)<<(32-auth_peer_bits))) + == auth_peer; + /* Allows denial-of-service attack. + if ( !ok ) { + QMessageBox::warning(0,tr("Security"), + tr("<p>An attempt to access this device from %1 has been denied.") + .arg(peeraddress.toString())); + } + */ + return ok; +} + +ServerPI::ServerPI( int socket, QObject *parent = 0, const char* name = 0 ) + : QSocket( parent, name ) , dtp( 0 ), serversocket( 0 ), waitsocket( 0 ) +{ + state = Connected; + + setSocket( socket ); + + peerport = peerPort(); + peeraddress = peerAddress(); + +#ifndef INSECURE + if ( !accessAuthorized(peeraddress) ) { + state = Forbidden; + startTimer( 0 ); + } else +#endif + { + connect( this, SIGNAL( readyRead() ), SLOT( read() ) ); + connect( this, SIGNAL( connectionClosed() ), SLOT( connectionClosed() ) ); + + passiv = FALSE; + for( int i = 0; i < 4; i++ ) + wait[i] = FALSE; + + send( "220 Qtopia transfer service ready!" ); + state = Wait_USER; + + dtp = new ServerDTP( this ); + connect( dtp, SIGNAL( completed() ), SLOT( dtpCompleted() ) ); + connect( dtp, SIGNAL( failed() ), SLOT( dtpFailed() ) ); + connect( dtp, SIGNAL( error( int ) ), SLOT( dtpError( int ) ) ); + + + directory = QDir::currentDirPath(); + + static int p = 1024; + + while ( !serversocket || !serversocket->ok() ) { + delete serversocket; + serversocket = new ServerSocket( ++p, this ); + } + connect( serversocket, SIGNAL( newIncomming( int ) ), + SLOT( newConnection( int ) ) ); + } +} + +ServerPI::~ServerPI() +{ + +} + +void ServerPI::connectionClosed() +{ + // qDebug( "Debug: Connection closed" ); + delete this; +} + +void ServerPI::send( const QString& msg ) +{ + QTextStream os( this ); + os << msg << endl; + //qDebug( "Reply: %s", msg.latin1() ); +} + +void ServerPI::read() +{ + while ( canReadLine() ) + process( readLine().stripWhiteSpace() ); +} + +bool ServerPI::checkUser( const QString& user ) +{ + if ( user.isEmpty() ) return FALSE; + + struct passwd *pw; + pw = getpwuid( geteuid() ); + QString euser = QString::fromLocal8Bit( pw->pw_name ); + return user == euser; +} + +bool ServerPI::checkPassword( const QString& /* password */ ) +{ + // ### HACK for testing on local host + return true; + + /* + struct passwd *pw = 0; + struct spwd *spw = 0; + + pw = getpwuid( geteuid() ); + spw = getspnam( pw->pw_name ); + + QString cpwd = QString::fromLocal8Bit( pw->pw_passwd ); + if ( cpwd == "x" && spw ) + cpwd = QString::fromLocal8Bit( spw->sp_pwdp ); + + QString cpassword = QString::fromLocal8Bit( crypt( password.local8Bit(), cpwd.local8Bit() ) ); + return cpwd == cpassword; +*/ +} + +bool ServerPI::checkReadFile( const QString& file ) +{ + QString filename; + + if ( file[0] != "/" ) + filename = directory.path() + "/" + file; + else + filename = file; + + QFileInfo fi( filename ); + return ( fi.exists() && fi.isReadable() ); +} + +bool ServerPI::checkWriteFile( const QString& file ) +{ + QString filename; + + if ( file[0] != "/" ) + filename = directory.path() + "/" + file; + else + filename = file; + + QFileInfo fi( filename ); + + if ( fi.exists() ) + if ( !QFile( filename ).remove() ) + return FALSE; + return TRUE; +} + +void ServerPI::process( const QString& message ) +{ + //qDebug( "Command: %s", message.latin1() ); + + // split message using "," as separator + QStringList msg = QStringList::split( " ", message ); + if ( msg.isEmpty() ) return; + + // command token + QString cmd = msg[0].upper(); + + // argument token + QString arg; + if ( msg.count() >= 2 ) + arg = msg[1]; + + // full argument string + QString args; + if ( msg.count() >= 2 ) { + QStringList copy( msg ); + // FIXME: for Qt3 + // copy.pop_front() + copy.remove( copy.begin() ); + args = copy.join( " " ); + } + + //qDebug( "args: %s", args.latin1() ); + + // we always respond to QUIT, regardless of state + if ( cmd == "QUIT" ) { + send( "211 Good bye!" ); + delete this; + return; + } + + // connected to client + if ( Connected == state ) + return; + + // waiting for user name + if ( Wait_USER == state ) { + + if ( cmd != "USER" || msg.count() < 2 || !checkUser( arg ) ) { + send( "530 Please login with USER and PASS" ); + return; + } + send( "331 User name ok, need password" ); + state = Wait_PASS; + return; + } + + // waiting for password + if ( Wait_PASS == state ) { + + if ( cmd != "PASS" || !checkPassword( arg ) ) { + //if ( cmd != "PASS" || msg.count() < 2 || !checkPassword( arg ) ) { + send( "530 Please login with USER and PASS" ); + return; + } + send( "230 User logged in, proceed" ); + state = Ready; + return; + } + + // ACCESS CONTROL COMMANDS + + + // account (ACCT) + if ( cmd == "ACCT" ) { + // even wu-ftp does not support it + send( "502 Command not implemented" ); + } + + // change working directory (CWD) + else if ( cmd == "CWD" ) { + + if ( !args.isEmpty() ) { + if ( directory.cd( args, TRUE ) ) + send( "250 Requested file action okay, completed" ); + else + send( "550 Requested action not taken" ); + } + else + send( "500 Syntax error, command unrecognized" ); + } + + // change to parent directory (CDUP) + else if ( cmd == "CDUP" ) { + if ( directory.cdUp() ) + send( "250 Requested file action okay, completed" ); + else + send( "550 Requested action not taken" ); + } + + // structure mount (SMNT) + else if ( cmd == "SMNT" ) { + // even wu-ftp does not support it + send( "502 Command not implemented" ); + } + + // reinitialize (REIN) + else if ( cmd == "REIN" ) { + // even wu-ftp does not support it + send( "502 Command not implemented" ); + } + + + // TRANSFER PARAMETER COMMANDS + + + // data port (PORT) + else if ( cmd == "PORT" ) { + if ( parsePort( arg ) ) + send( "200 Command okay" ); + else + send( "500 Syntax error, command unrecognized" ); + } + + // passive (PASV) + else if ( cmd == "PASV" ) { + passiv = TRUE; + send( "227 Entering Passive Mode (" + + address().toString().replace( QRegExp( "\\." ), "," ) + "," + + QString::number( ( serversocket->port() ) >> 8 ) + "," + + QString::number( ( serversocket->port() ) & 0xFF ) +")" ); + } + + // representation type (TYPE) + else if ( cmd == "TYPE" ) { + if ( arg.upper() == "A" || arg.upper() == "I" ) + send( "200 Command okay" ); + else + send( "504 Command not implemented for that parameter" ); + } + + // file structure (STRU) + else if ( cmd == "STRU" ) { + if ( arg.upper() == "F" ) + send( "200 Command okay" ); + else + send( "504 Command not implemented for that parameter" ); + } + + // transfer mode (MODE) + else if ( cmd == "MODE" ) { + if ( arg.upper() == "S" ) + send( "200 Command okay" ); + else + send( "504 Command not implemented for that parameter" ); + } + + + // FTP SERVICE COMMANDS + + + // retrieve (RETR) + else if ( cmd == "RETR" ) + if ( !args.isEmpty() && checkReadFile( absFilePath( args ) ) + || backupRestoreGzip( absFilePath( args ) ) ) { + send( "150 File status okay" ); + sendFile( absFilePath( args ) ); + } + else { + qDebug("550 Requested action not taken"); + send( "550 Requested action not taken" ); + } + + // store (STOR) + else if ( cmd == "STOR" ) + if ( !args.isEmpty() && checkWriteFile( absFilePath( args ) ) ) { + send( "150 File status okay" ); + retrieveFile( absFilePath( args ) ); + } + else + send( "550 Requested action not taken" ); + + // store unique (STOU) + else if ( cmd == "STOU" ) { + send( "502 Command not implemented" ); + } + + // append (APPE) + else if ( cmd == "APPE" ) { + send( "502 Command not implemented" ); + } + + // allocate (ALLO) + else if ( cmd == "ALLO" ) { + send( "200 Command okay" ); + } + + // restart (REST) + else if ( cmd == "REST" ) { + send( "502 Command not implemented" ); + } + + // rename from (RNFR) + else if ( cmd == "RNFR" ) { + renameFrom = QString::null; + if ( args.isEmpty() ) + send( "500 Syntax error, command unrecognized" ); + else { + QFile file( absFilePath( args ) ); + if ( file.exists() ) { + send( "350 File exists, ready for destination name" ); + renameFrom = absFilePath( args ); + } + else + send( "550 Requested action not taken" ); + } + } + + // rename to (RNTO) + else if ( cmd == "RNTO" ) { + if ( lastCommand != "RNFR" ) + send( "503 Bad sequence of commands" ); + else if ( args.isEmpty() ) + send( "500 Syntax error, command unrecognized" ); + else { + QDir dir( absFilePath( args ) ); + if ( dir.rename( renameFrom, absFilePath( args ), TRUE ) ) + send( "250 Requested file action okay, completed." ); + else + send( "550 Requested action not taken" ); + } + } + + // abort (ABOR) + else if ( cmd.contains( "ABOR" ) ) { + dtp->close(); + if ( dtp->dtpMode() != ServerDTP::Idle ) + send( "426 Connection closed; transfer aborted" ); + else + send( "226 Closing data connection" ); + } + + // delete (DELE) + else if ( cmd == "DELE" ) { + if ( args.isEmpty() ) + send( "500 Syntax error, command unrecognized" ); + else { + QFile file( absFilePath( args ) ) ; + if ( file.remove() ) + send( "250 Requested file action okay, completed" ); + else + send( "550 Requested action not taken" ); + } + } + + // remove directory (RMD) + else if ( cmd == "RMD" ) { + if ( args.isEmpty() ) + send( "500 Syntax error, command unrecognized" ); + else { + QDir dir; + if ( dir.rmdir( absFilePath( args ), TRUE ) ) + send( "250 Requested file action okay, completed" ); + else + send( "550 Requested action not taken" ); + } + } + + // make directory (MKD) + else if ( cmd == "MKD" ) { + if ( args.isEmpty() ) { + qDebug(" Error: no arg"); + send( "500 Syntax error, command unrecognized" ); + } + else { + QDir dir; + if ( dir.mkdir( absFilePath( args ), TRUE ) ) + send( "250 Requested file action okay, completed." ); + else + send( "550 Requested action not taken" ); + } + } + + // print working directory (PWD) + else if ( cmd == "PWD" ) { + send( "257 \"" + directory.path() +"\"" ); + } + + // list (LIST) + else if ( cmd == "LIST" ) { + if ( sendList( absFilePath( args ) ) ) + send( "150 File status okay" ); + else + send( "500 Syntax error, command unrecognized" ); + } + + // size (SIZE) + else if ( cmd == "SIZE" ) { + QString filePath = absFilePath( args ); + QFileInfo fi( filePath ); + bool gzipfile = backupRestoreGzip( filePath ); + if ( !fi.exists() && !gzipfile ) + send( "500 Syntax error, command unrecognized" ); + else { + if ( !gzipfile ) + send( "213 " + QString::number( fi.size() ) ); + else { + Process duproc( QString("du") ); + duproc.addArgument("-s"); + QString in, out; + if ( !duproc.exec(in, out) ) { + qDebug("du process failed; just sending back 1K"); + send( "213 1024"); + } + else { + QString size = out.left( out.find("\t") ); + int guess = size.toInt()/5; + if ( filePath.contains("doc") ) + guess *= 1000; + qDebug("sending back gzip guess of %d", guess); + send( "213 " + QString::number(guess) ); + } + } + } + } + // name list (NLST) + else if ( cmd == "NLST" ) { + send( "502 Command not implemented" ); + } + + // site parameters (SITE) + else if ( cmd == "SITE" ) { + send( "502 Command not implemented" ); + } + + // system (SYST) + else if ( cmd == "SYST" ) { + send( "215 UNIX Type: L8" ); + } + + // status (STAT) + else if ( cmd == "STAT" ) { + send( "502 Command not implemented" ); + } + + // help (HELP ) + else if ( cmd == "HELP" ) { + send( "502 Command not implemented" ); + } + + // noop (NOOP) + else if ( cmd == "NOOP" ) { + send( "200 Command okay" ); + } + + // not implemented + else + send( "502 Command not implemented" ); + + lastCommand = cmd; +} + +bool ServerPI::backupRestoreGzip( const QString &file ) +{ + return (file.find( "backup" ) != -1 && + file.findRev( ".tgz" ) == (int)file.length()-4 ); +} + +bool ServerPI::backupRestoreGzip( const QString &file, QStringList &targets ) +{ + if ( file.find( "backup" ) != -1 && + file.findRev( ".tgz" ) == (int)file.length()-4 ) { + QFileInfo info( file ); + targets = info.dirPath( TRUE ); + qDebug("ServerPI::backupRestoreGzip for %s = %s", file.latin1(), + targets.join(" ").latin1() ); + return true; + } + return false; +} + +void ServerPI::sendFile( const QString& file ) +{ + if ( passiv ) { + wait[SendFile] = TRUE; + waitfile = file; + if ( waitsocket ) + newConnection( waitsocket ); + } + else { + QStringList targets; + if ( backupRestoreGzip( file, targets ) ) + dtp->sendGzipFile( file, targets, peeraddress, peerport ); + else dtp->sendFile( file, peeraddress, peerport ); + } +} + +void ServerPI::retrieveFile( const QString& file ) +{ + if ( passiv ) { + wait[RetrieveFile] = TRUE; + waitfile = file; + if ( waitsocket ) + newConnection( waitsocket ); + } + else { + QStringList targets; + if ( backupRestoreGzip( file, targets ) ) + dtp->retrieveGzipFile( file, peeraddress, peerport ); + else + dtp->retrieveFile( file, peeraddress, peerport ); + } +} + +bool ServerPI::parsePort( const QString& pp ) +{ + QStringList p = QStringList::split( ",", pp ); + if ( p.count() != 6 ) return FALSE; + + // h1,h2,h3,h4,p1,p2 + peeraddress = QHostAddress( ( p[0].toInt() << 24 ) + ( p[1].toInt() << 16 ) + + ( p[2].toInt() << 8 ) + p[3].toInt() ); + peerport = ( p[4].toInt() << 8 ) + p[5].toInt(); + return TRUE; +} + +void ServerPI::dtpCompleted() +{ + dtp->close(); + waitsocket = 0; + send( "226 Closing data connection, file transfer successful" ); +} + +void ServerPI::dtpFailed() +{ + dtp->close(); + waitsocket = 0; + send( "451 Requested action aborted: local error in processing" ); +} + +void ServerPI::dtpError( int ) +{ + dtp->close(); + waitsocket = 0; + send( "451 Requested action aborted: local error in processing" ); +} + +bool ServerPI::sendList( const QString& arg ) +{ + QByteArray listing; + QBuffer buffer( listing ); + + if ( !buffer.open( IO_WriteOnly ) ) + return FALSE; + + QTextStream ts( &buffer ); + QString fn = arg; + + if ( fn.isEmpty() ) + fn = directory.path(); + + QFileInfo fi( fn ); + if ( !fi.exists() ) return FALSE; + + // return file listing + if ( fi.isFile() ) { + ts << fileListing( &fi ) << endl; + } + + // return directory listing + else if ( fi.isDir() ) { + QDir dir( fn ); + const QFileInfoList *list = dir.entryInfoList( QDir::All | QDir::Hidden ); + + QFileInfoListIterator it( *list ); + QFileInfo *info; + + unsigned long total = 0; + while ( ( info = it.current() ) ) { + if ( info->fileName() != "." && info->fileName() != ".." ) + total += info->size(); + ++it; + } + + ts << "total " << QString::number( total / 1024 ) << endl; + + it.toFirst(); + while ( ( info = it.current() ) ) { + if ( info->fileName() == "." || info->fileName() == ".." ) { + ++it; + continue; + } + ts << fileListing( info ) << endl; + ++it; + } + } + + if ( passiv ) { + waitarray = buffer.buffer(); + wait[SendByteArray] = TRUE; + if ( waitsocket ) + newConnection( waitsocket ); + } + else + dtp->sendByteArray( buffer.buffer(), peeraddress, peerport ); + return TRUE; +} + +QString ServerPI::fileListing( QFileInfo *info ) +{ + if ( !info ) return QString::null; + QString s; + + // type char + if ( info->isDir() ) + s += "d"; + else if ( info->isSymLink() ) + s += "l"; + else + s += "-"; + + // permisson string + s += permissionString( info ) + " "; + + // number of hardlinks + int subdirs = 1; + + if ( info->isDir() ) + subdirs = 2; + // FIXME : this is to slow + //if ( info->isDir() ) + //subdirs = QDir( info->absFilePath() ).entryList( QDir::Dirs ).count(); + + s += QString::number( subdirs ).rightJustify( 3, ' ', TRUE ) + " "; + + // owner + s += info->owner().leftJustify( 8, ' ', TRUE ) + " "; + + // group + s += info->group().leftJustify( 8, ' ', TRUE ) + " "; + + // file size in bytes + s += QString::number( info->size() ).rightJustify( 9, ' ', TRUE ) + " "; + + // last modified date + QDate date = info->lastModified().date(); + QTime time = info->lastModified().time(); + s += date.monthName( date.month() ) + " " + + QString::number( date.day() ).rightJustify( 2, ' ', TRUE ) + " " + + QString::number( time.hour() ).rightJustify( 2, '0', TRUE ) + ":" + + QString::number( time.minute() ).rightJustify( 2,'0', TRUE ) + " "; + + // file name + s += info->fileName(); + + return s; +} + +QString ServerPI::permissionString( QFileInfo *info ) +{ + if ( !info ) return QString( "---------" ); + QString s; + + // user + if ( info->permission( QFileInfo::ReadUser ) ) s += "r"; + else s += "-"; + if ( info->permission( QFileInfo::WriteUser ) ) s += "w"; + else s += "-"; + if ( info->permission( QFileInfo::ExeUser ) ) s += "x"; + else s += "-"; + + // group + if ( info->permission( QFileInfo::ReadGroup ) ) s += "r"; + else s += "-"; + if ( info->permission( QFileInfo::WriteGroup ) )s += "w"; + else s += "-"; + if ( info->permission( QFileInfo::ExeGroup ) ) s += "x"; + else s += "-"; + + // exec + if ( info->permission( QFileInfo::ReadOther ) ) s += "r"; + else s += "-"; + if ( info->permission( QFileInfo::WriteOther ) ) s += "w"; + else s += "-"; + if ( info->permission( QFileInfo::ExeOther ) ) s += "x"; + else s += "-"; + + return s; +} + +void ServerPI::newConnection( int socket ) +{ + //qDebug( "New incomming connection" ); + + if ( !passiv ) return; + + if ( wait[SendFile] ) { + QStringList targets; + if ( backupRestoreGzip( waitfile, targets ) ) + dtp->sendGzipFile( waitfile, targets ); + else + dtp->sendFile( waitfile ); + dtp->setSocket( socket ); + } + else if ( wait[RetrieveFile] ) { + qDebug("check retrieve file"); + if ( backupRestoreGzip( waitfile ) ) + dtp->retrieveGzipFile( waitfile ); + else + dtp->retrieveFile( waitfile ); + dtp->setSocket( socket ); + } + else if ( wait[SendByteArray] ) { + dtp->sendByteArray( waitarray ); + dtp->setSocket( socket ); + } + else if ( wait[RetrieveByteArray] ) { + qDebug("retrieve byte array"); + dtp->retrieveByteArray(); + dtp->setSocket( socket ); + } + else + waitsocket = socket; + + for( int i = 0; i < 4; i++ ) + wait[i] = FALSE; +} + +QString ServerPI::absFilePath( const QString& file ) +{ + if ( file.isEmpty() ) return file; + + QString filepath( file ); + if ( file[0] != "/" ) + filepath = directory.path() + "/" + file; + + return filepath; +} + + +void ServerPI::timerEvent( QTimerEvent * ) +{ + connectionClosed(); +} + + +ServerDTP::ServerDTP( QObject *parent = 0, const char* name = 0) + : QSocket( parent, name ), mode( Idle ), createTargzProc( 0 ), +retrieveTargzProc( 0 ), gzipProc( 0 ) +{ + + connect( this, SIGNAL( connected() ), SLOT( connected() ) ); + connect( this, SIGNAL( connectionClosed() ), SLOT( connectionClosed() ) ); + connect( this, SIGNAL( bytesWritten( int ) ), SLOT( bytesWritten( int ) ) ); + connect( this, SIGNAL( readyRead() ), SLOT( readyRead() ) ); + + gzipProc = new QProcess( this, "gzipProc" ); + gzipProc->setCommunication( QProcess::Stdin | QProcess::Stdout ); + + createTargzProc = new QProcess( QString("tar"), this, "createTargzProc"); + createTargzProc->setCommunication( QProcess::Stdout ); + createTargzProc->setWorkingDirectory( QDir::rootDirPath() ); + connect( createTargzProc, SIGNAL( processExited() ), SLOT( targzDone() ) ); + + QStringList args = "tar"; + args += "-xv"; + retrieveTargzProc = new QProcess( args, this, "retrieveTargzProc" ); + retrieveTargzProc->setCommunication( QProcess::Stdin ); + retrieveTargzProc->setWorkingDirectory( QDir::rootDirPath() ); + connect( retrieveTargzProc, SIGNAL( processExited() ), + SIGNAL( completed() ) ); + connect( retrieveTargzProc, SIGNAL( processExited() ), + SLOT( extractTarDone() ) ); +} + +ServerDTP::~ServerDTP() +{ + buf.close(); + file.close(); + createTargzProc->kill(); +} + +void ServerDTP::extractTarDone() +{ + qDebug("extract done"); + QCopEnvelope e( "QPE/Desktop", "restoreDone(QString)" ); + e << file.name(); +} + +void ServerDTP::connected() +{ + // send file mode + switch ( mode ) { + case SendFile : + if ( !file.exists() || !file.open( IO_ReadOnly) ) { + emit failed(); + mode = Idle; + return; + } + + //qDebug( "Debug: Sending file '%s'", file.name().latin1() ); + + bytes_written = 0; + if ( file.size() == 0 ) { + //make sure it doesn't hang on empty files + file.close(); + emit completed(); + mode = Idle; + } else { + + if( !file.atEnd() ) { + QCString s; + s.resize( block_size ); + int bytes = file.readBlock( s.data(), block_size ); + writeBlock( s.data(), bytes ); + } + } + break; + case SendGzipFile: + if ( createTargzProc->isRunning() ) { + // SHOULDN'T GET HERE, BUT DOING A SAFETY CHECK ANYWAY + qWarning("Previous tar --gzip process is still running; killing it..."); + createTargzProc->kill(); + } + + bytes_written = 0; + qDebug("==>start send tar process"); + if ( !createTargzProc->start() ) + qWarning("Error starting %s or %s", + createTargzProc->arguments().join(" ").latin1(), + gzipProc->arguments().join(" ").latin1() ); + break; + case SendBuffer: + if ( !buf.open( IO_ReadOnly) ) { + emit failed(); + mode = Idle; + return; + } + + // qDebug( "Debug: Sending byte array" ); + bytes_written = 0; + while( !buf.atEnd() ) + putch( buf.getch() ); + buf.close(); + break; + case RetrieveFile: + // retrieve file mode + if ( file.exists() && !file.remove() ) { + emit failed(); + mode = Idle; + return; + } + + if ( !file.open( IO_WriteOnly) ) { + emit failed(); + mode = Idle; + return; + } + // qDebug( "Debug: Retrieving file %s", file.name().latin1() ); + break; + case RetrieveGzipFile: + qDebug("=-> starting tar process to receive .tgz file"); + break; + case RetrieveBuffer: + // retrieve buffer mode + if ( !buf.open( IO_WriteOnly) ) { + emit failed(); + mode = Idle; + return; + } + // qDebug( "Debug: Retrieving byte array" ); + break; + case Idle: + qDebug("connection established but mode set to Idle; BUG!"); + break; + } +} + +void ServerDTP::connectionClosed() +{ + //qDebug( "Debug: Data connection closed %ld bytes written", bytes_written ); + + // send file mode + if ( SendFile == mode ) { + if ( bytes_written == file.size() ) + emit completed(); + else + emit failed(); + } + + // send buffer mode + else if ( SendBuffer == mode ) { + if ( bytes_written == buf.size() ) + emit completed(); + else + emit failed(); + } + + // retrieve file mode + else if ( RetrieveFile == mode ) { + file.close(); + emit completed(); + } + + else if ( RetrieveGzipFile == mode ) { + qDebug("Done writing ungzip file; closing input"); + gzipProc->flushStdin(); + gzipProc->closeStdin(); + } + + // retrieve buffer mode + else if ( RetrieveBuffer == mode ) { + buf.close(); + emit completed(); + } + + mode = Idle; +} + +void ServerDTP::bytesWritten( int bytes ) +{ + bytes_written += bytes; + + // send file mode + if ( SendFile == mode ) { + + if ( bytes_written == file.size() ) { + // qDebug( "Debug: Sending complete: %d bytes", file.size() ); + file.close(); + emit completed(); + mode = Idle; + } + else if( !file.atEnd() ) { + QCString s; + s.resize( block_size ); + int bytes = file.readBlock( s.data(), block_size ); + writeBlock( s.data(), bytes ); + } + } + + // send buffer mode + if ( SendBuffer == mode ) { + + if ( bytes_written == buf.size() ) { + // qDebug( "Debug: Sending complete: %d bytes", buf.size() ); + emit completed(); + mode = Idle; + } + } +} + +void ServerDTP::readyRead() +{ + // retrieve file mode + if ( RetrieveFile == mode ) { + QCString s; + s.resize( bytesAvailable() ); + readBlock( s.data(), bytesAvailable() ); + file.writeBlock( s.data(), s.size() ); + } + else if ( RetrieveGzipFile == mode ) { + if ( !gzipProc->isRunning() ) + gzipProc->start(); + + QByteArray s; + s.resize( bytesAvailable() ); + readBlock( s.data(), bytesAvailable() ); + gzipProc->writeToStdin( s ); + qDebug("wrote %d bytes to ungzip ", s.size() ); + } + // retrieve buffer mode + else if ( RetrieveBuffer == mode ) { + QCString s; + s.resize( bytesAvailable() ); + readBlock( s.data(), bytesAvailable() ); + buf.writeBlock( s.data(), s.size() ); + } +} + +void ServerDTP::writeTargzBlock() +{ + QByteArray block = gzipProc->readStdout(); + writeBlock( block.data(), block.size() ); + qDebug("writeTargzBlock %d", block.size()); + if ( !createTargzProc->isRunning() ) { + qDebug("tar and gzip done"); + emit completed(); + mode = Idle; + disconnect( gzipProc, SIGNAL( readyReadStdout() ), + this, SLOT( writeTargzBlock() ) ); + } +} + +void ServerDTP::targzDone() +{ + //qDebug("targz done"); + disconnect( createTargzProc, SIGNAL( readyReadStdout() ), + this, SLOT( gzipTarBlock() ) ); + gzipProc->closeStdin(); +} + +void ServerDTP::gzipTarBlock() +{ + //qDebug("gzipTarBlock"); + if ( !gzipProc->isRunning() ) { + //qDebug("auto start gzip proc"); + gzipProc->start(); + } + gzipProc->writeToStdin( createTargzProc->readStdout() ); +} + +void ServerDTP::sendFile( const QString fn, const QHostAddress& host, Q_UINT16 port ) +{ + file.setName( fn ); + mode = SendFile; + connectToHost( host.toString(), port ); +} + +void ServerDTP::sendFile( const QString fn ) +{ + file.setName( fn ); + mode = SendFile; +} + +void ServerDTP::sendGzipFile( const QString &fn, + const QStringList &archiveTargets, + const QHostAddress& host, Q_UINT16 port ) +{ + sendGzipFile( fn, archiveTargets ); + connectToHost( host.toString(), port ); +} + +void ServerDTP::sendGzipFile( const QString &fn, + const QStringList &archiveTargets ) +{ + mode = SendGzipFile; + file.setName( fn ); + + QStringList args = "tar"; + args += "-cv"; + args += archiveTargets; + qDebug("sendGzipFile %s", args.join(" ").latin1() ); + createTargzProc->setArguments( args ); + connect( createTargzProc, + SIGNAL( readyReadStdout() ), SLOT( gzipTarBlock() ) ); + + gzipProc->setArguments( "gzip" ); + connect( gzipProc, SIGNAL( readyReadStdout() ), + SLOT( writeTargzBlock() ) ); +} + +void ServerDTP::gunzipDone() +{ + qDebug("gunzipDone"); + disconnect( gzipProc, SIGNAL( processExited() ), + this, SLOT( gunzipDone() ) ); + retrieveTargzProc->closeStdin(); + disconnect( gzipProc, SIGNAL( readyReadStdout() ), + this, SLOT( tarExtractBlock() ) ); +} + +void ServerDTP::tarExtractBlock() +{ + qDebug("ungzipTarBlock"); + if ( !retrieveTargzProc->isRunning() ) { + qDebug("auto start ungzip proc"); + if ( !retrieveTargzProc->start() ) + qWarning(" failed to start tar -x process"); + } + retrieveTargzProc->writeToStdin( gzipProc->readStdout() ); +} + + +void ServerDTP::retrieveFile( const QString fn, const QHostAddress& host, Q_UINT16 port ) +{ + file.setName( fn ); + mode = RetrieveFile; + connectToHost( host.toString(), port ); +} + +void ServerDTP::retrieveFile( const QString fn ) +{ + file.setName( fn ); + mode = RetrieveFile; +} + +void ServerDTP::retrieveGzipFile( const QString &fn ) +{ + qDebug("retrieveGzipFile %s", fn.latin1()); + file.setName( fn ); + mode = RetrieveGzipFile; + + gzipProc->setArguments( "gunzip" ); + connect( gzipProc, SIGNAL( readyReadStdout() ), + SLOT( tarExtractBlock() ) ); + connect( gzipProc, SIGNAL( processExited() ), + SLOT( gunzipDone() ) ); +} + +void ServerDTP::retrieveGzipFile( const QString &fn, const QHostAddress& host, Q_UINT16 port ) +{ + retrieveGzipFile( fn ); + connectToHost( host.toString(), port ); +} + +void ServerDTP::sendByteArray( const QByteArray& array, const QHostAddress& host, Q_UINT16 port ) +{ + buf.setBuffer( array ); + mode = SendBuffer; + connectToHost( host.toString(), port ); +} + +void ServerDTP::sendByteArray( const QByteArray& array ) +{ + buf.setBuffer( array ); + mode = SendBuffer; +} + +void ServerDTP::retrieveByteArray( const QHostAddress& host, Q_UINT16 port ) +{ + buf.setBuffer( QByteArray() ); + mode = RetrieveBuffer; + connectToHost( host.toString(), port ); +} + +void ServerDTP::retrieveByteArray() +{ + buf.setBuffer( QByteArray() ); + mode = RetrieveBuffer; +} + +void ServerDTP::setSocket( int socket ) +{ + QSocket::setSocket( socket ); + connected(); +} + |