You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
k3b/libk3b/tools/k3bactivepipe.cpp

256 lines
4.6 KiB

/*
*
* $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;
}