|
|
|
/*
|
|
|
|
*
|
|
|
|
* $Id: sourceheader 511311 2006-02-19 14:51:05Z trueg $
|
|
|
|
* Copyright (C) 2006 Sebastian Trueg <trueg@k3b.org>
|
|
|
|
*
|
|
|
|
* This file is part of the K3b project.
|
|
|
|
* Copyright (C) 1998-2007 Sebastian Trueg <trueg@k3b.org>
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
* See the file "COPYING" for the exact licensing terms.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "k3bactivepipe.h"
|
|
|
|
|
|
|
|
#include <k3bpipe.h>
|
|
|
|
|
|
|
|
#include <kdebug.h>
|
|
|
|
|
|
|
|
#include <tqthread.h>
|
|
|
|
#include <tqiodevice.h>
|
|
|
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
|
|
|
|
|
|
|
class K3bActivePipe::Private : public TQThread
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
Private( K3bActivePipe* pipe ) :
|
|
|
|
m_pipe( pipe ),
|
|
|
|
fdToReadFrom(-1),
|
|
|
|
fdToWriteTo(-1),
|
|
|
|
sourceIODevice(0),
|
|
|
|
sinkIODevice(0),
|
|
|
|
closeFdToReadFrom(false),
|
|
|
|
closeFdToWriteTo(false) {
|
|
|
|
}
|
|
|
|
|
|
|
|
void run() {
|
|
|
|
kdDebug() << "(K3bActivePipe) started thread." << endl;
|
|
|
|
bytesRead = bytesWritten = 0;
|
|
|
|
buffer.resize( 10*2048 );
|
|
|
|
ssize_t r = 0;
|
|
|
|
while( ( r = m_pipe->read( buffer.data(), buffer.size() ) ) > 0 ) {
|
|
|
|
|
|
|
|
bytesRead += r;
|
|
|
|
|
|
|
|
// write it out
|
|
|
|
ssize_t w = 0;
|
|
|
|
ssize_t ww = 0;
|
|
|
|
while( w < r ) {
|
|
|
|
if( ( ww = m_pipe->write( buffer.data()+w, r-w ) ) > 0 ) {
|
|
|
|
w += ww;
|
|
|
|
bytesWritten += ww;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
kdDebug() << "(K3bActivePipe) write failed." << endl;
|
|
|
|
close( closeWhenDone );
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// kdDebug() << "(K3bActivePipe) thread done: " << r << " (total bytes read/written: " << bytesRead << "/" << bytesWritten << ")" << endl;
|
|
|
|
close( closeWhenDone );
|
|
|
|
}
|
|
|
|
|
|
|
|
int readFd() const {
|
|
|
|
if( fdToReadFrom == -1 )
|
|
|
|
return pipeIn.out();
|
|
|
|
else
|
|
|
|
return fdToReadFrom;
|
|
|
|
}
|
|
|
|
|
|
|
|
int writeFd() const {
|
|
|
|
if( fdToWriteTo == -1 )
|
|
|
|
return pipeOut.in();
|
|
|
|
else
|
|
|
|
return fdToWriteTo;
|
|
|
|
}
|
|
|
|
|
|
|
|
void close( bool closeAll ) {
|
|
|
|
if( sourceIODevice )
|
|
|
|
sourceIODevice->close();
|
|
|
|
if( sinkIODevice )
|
|
|
|
sinkIODevice->close();
|
|
|
|
|
|
|
|
if( closeAll ) {
|
|
|
|
pipeIn.close();
|
|
|
|
pipeOut.close();
|
|
|
|
if( fdToWriteTo != -1 &&
|
|
|
|
closeFdToWriteTo )
|
|
|
|
::close( fdToWriteTo );
|
|
|
|
|
|
|
|
if( fdToReadFrom != -1 &&
|
|
|
|
closeFdToReadFrom )
|
|
|
|
::close( fdToReadFrom );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
K3bActivePipe* m_pipe;
|
|
|
|
|
|
|
|
public:
|
|
|
|
int fdToReadFrom;
|
|
|
|
int fdToWriteTo;
|
|
|
|
K3bPipe pipeIn;
|
|
|
|
K3bPipe pipeOut;
|
|
|
|
|
|
|
|
TQIODevice* sourceIODevice;
|
|
|
|
TQIODevice* sinkIODevice;
|
|
|
|
|
|
|
|
bool closeWhenDone;
|
|
|
|
bool closeFdToReadFrom;
|
|
|
|
bool closeFdToWriteTo;
|
|
|
|
|
|
|
|
TQByteArray buffer;
|
|
|
|
|
|
|
|
TQ_UINT64 bytesRead;
|
|
|
|
TQ_UINT64 bytesWritten;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
K3bActivePipe::K3bActivePipe()
|
|
|
|
{
|
|
|
|
d = new Private( this );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
K3bActivePipe::~K3bActivePipe()
|
|
|
|
{
|
|
|
|
delete d;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool K3bActivePipe::open( bool closeWhenDone )
|
|
|
|
{
|
|
|
|
if( d->running() )
|
|
|
|
return false;
|
|
|
|
|
|
|
|
d->closeWhenDone = closeWhenDone;
|
|
|
|
|
|
|
|
if( d->sourceIODevice ) {
|
|
|
|
if( !d->sourceIODevice->open( IO_ReadOnly ) )
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else if( d->fdToReadFrom == -1 && !d->pipeIn.open() ) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if( d->sinkIODevice ) {
|
|
|
|
if( !d->sinkIODevice->open( IO_WriteOnly ) )
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else if( d->fdToWriteTo == -1 && !d->pipeOut.open() ) {
|
|
|
|
close();
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
kdDebug() << "(K3bActivePipe) successfully opened pipe." << endl;
|
|
|
|
|
|
|
|
d->start();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void K3bActivePipe::close()
|
|
|
|
{
|
|
|
|
d->pipeIn.closeIn();
|
|
|
|
d->wait();
|
|
|
|
d->close( true );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void K3bActivePipe::readFromFd( int fd, bool close )
|
|
|
|
{
|
|
|
|
d->fdToReadFrom = fd;
|
|
|
|
d->sourceIODevice = 0;
|
|
|
|
d->closeFdToReadFrom = close;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void K3bActivePipe::writeToFd( int fd, bool close )
|
|
|
|
{
|
|
|
|
d->fdToWriteTo = fd;
|
|
|
|
d->sinkIODevice = 0;
|
|
|
|
d->closeFdToWriteTo = close;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void K3bActivePipe::readFromIODevice( TQIODevice* dev )
|
|
|
|
{
|
|
|
|
d->fdToReadFrom = -1;
|
|
|
|
d->sourceIODevice = dev;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void K3bActivePipe::writeToIODevice( TQIODevice* dev )
|
|
|
|
{
|
|
|
|
d->fdToWriteTo = -1;
|
|
|
|
d->sinkIODevice = dev;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int K3bActivePipe::in() const
|
|
|
|
{
|
|
|
|
return d->pipeIn.in();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int K3bActivePipe::out() const
|
|
|
|
{
|
|
|
|
return d->pipeOut.out();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int K3bActivePipe::read( char* data, int max )
|
|
|
|
{
|
|
|
|
if( d->sourceIODevice )
|
|
|
|
return d->sourceIODevice->readBlock( data, max );
|
|
|
|
else
|
|
|
|
return ::read( d->readFd(), data, max );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int K3bActivePipe::write( char* data, int max )
|
|
|
|
{
|
|
|
|
if( d->sinkIODevice )
|
|
|
|
return d->sinkIODevice->writeBlock( data, max );
|
|
|
|
else
|
|
|
|
return ::write( d->writeFd(), data, max );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool K3bActivePipe::pumpSync()
|
|
|
|
{
|
|
|
|
if( open( true ) )
|
|
|
|
d->wait();
|
|
|
|
else
|
|
|
|
return false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
TQ_UINT64 K3bActivePipe::bytesRead() const
|
|
|
|
{
|
|
|
|
return d->bytesRead;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
TQ_UINT64 K3bActivePipe::bytesWritten() const
|
|
|
|
{
|
|
|
|
return d->bytesWritten;
|
|
|
|
}
|