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.
tdelibs/kdecore/kprocio.cpp

277 lines
5.4 KiB

/* This file is part of the KDE libraries
Copyright (C) 1997 David Sweet <dsweet@kde.org>
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License version 2 as published by the Free Software Foundation.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Library General Public License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
Boston, MA 02110-1301, USA.
*/
// $Id$
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include "kprocio.h"
#include <kdebug.h>
#include <tqtextcodec.h>
class KProcIOPrivate {
public:
KProcIOPrivate() : comm(KProcess::All) {}
KProcess::Communication comm;
};
KProcIO::KProcIO ( TQTextCodec *_codec)
: codec(_codec), d(new KProcIOPrivate)
{
rbi=0;
readsignalon=writeready=true;
outbuffer.setAutoDelete(true);
if (!codec)
{
codec = TQTextCodec::codecForName("ISO 8859-1");
if (!codec)
{
kdError(174) << "Can't create ISO 8859-1 codec!" << endl;
}
}
}
KProcIO::~KProcIO()
{
delete d;
}
void
KProcIO::resetAll ()
{
if (isRunning())
kill();
clearArguments();
rbi=0;
readsignalon=writeready=true;
disconnect (this, TQT_SIGNAL (receivedStdout (KProcess *, char *, int)),
this, TQT_SLOT (received (KProcess *, char *, int)));
disconnect (this, TQT_SIGNAL (receivedStderr (KProcess *, char *, int)),
this, TQT_SLOT (received (KProcess *, char *, int)));
disconnect (this, TQT_SIGNAL (wroteStdin(KProcess *)),
this, TQT_SLOT (sent (KProcess *)));
outbuffer.clear();
}
void KProcIO::setComm (Communication comm)
{
d->comm = comm;
}
bool KProcIO::start (RunMode runmode, bool includeStderr)
{
connect (this, TQT_SIGNAL (receivedStdout (KProcess *, char *, int)),
this, TQT_SLOT (received (KProcess *, char *, int)));
if (includeStderr)
{
connect (this, TQT_SIGNAL (receivedStderr (KProcess *, char *, int)),
this, TQT_SLOT (received (KProcess *, char *, int)));
}
connect (this, TQT_SIGNAL (wroteStdin(KProcess *)),
this, TQT_SLOT (sent (KProcess *)));
return KProcess::start (runmode, d->comm);
}
bool KProcIO::writeStdin (const TQString &line, bool appendnewline)
{
return writeStdin(codec->fromUnicode(line), appendnewline);
}
bool KProcIO::writeStdin (const TQCString &line, bool appendnewline)
{
TQCString *qs = new TQCString(line);
if (appendnewline)
{
*qs += '\n';
}
int l = qs->length();
if (!l)
{
delete qs;
return true;
}
TQByteArray *b = (TQByteArray *) qs;
b->truncate(l); // Strip trailing null
outbuffer.append(b);
if (writeready)
{
writeready=false;
return KProcess::writeStdin( b->data(), b->size() );
}
return true;
}
bool KProcIO::writeStdin(const TQByteArray &data)
{
if (!data.size())
return true;
TQByteArray *b = new TQByteArray(data);
outbuffer.append(b);
if (writeready)
{
writeready=false;
return KProcess::writeStdin( b->data(), b->size() );
}
return true;
}
void KProcIO::closeWhenDone()
{
if (writeready)
{
closeStdin();
return;
}
outbuffer.append(0);
return;
}
void KProcIO::sent(KProcess *)
{
outbuffer.removeFirst();
if (outbuffer.count()==0)
{
writeready=true;
}
else
{
TQByteArray *b = outbuffer.first();
if (!b)
{
closeStdin();
}
else
{
KProcess::writeStdin(b->data(), b->size());
}
}
}
void KProcIO::received (KProcess *, char *buffer, int buflen)
{
recvbuffer += TQCString(buffer, buflen+1);
controlledEmission();
}
void KProcIO::ackRead ()
{
readsignalon=true;
if (needreadsignal || recvbuffer.length()!=0)
controlledEmission();
}
void KProcIO::controlledEmission ()
{
if (readsignalon)
{
needreadsignal=false;
readsignalon=false; //will stay off until read is acknowledged
emit readReady (this);
}
else
{
needreadsignal=true;
}
}
void KProcIO::enableReadSignals (bool enable)
{
readsignalon=enable;
if (enable && needreadsignal)
emit readReady (this);
}
int KProcIO::readln (TQString &line, bool autoAck, bool *partial)
{
int len;
if (autoAck)
readsignalon=true;
//need to reduce the size of recvbuffer at some point...
len=recvbuffer.find ('\n',rbi)-rbi;
//kdDebug(174) << "KPIO::readln" << endl;
//in case there's no '\n' at the end of the buffer
if ((len<0) &&
((unsigned int)rbi<recvbuffer.length()))
{
recvbuffer=recvbuffer.mid (rbi);
rbi=0;
if (partial)
{
len = recvbuffer.length();
line = recvbuffer;
recvbuffer = "";
*partial = true;
return len;
}
return -1;
}
if (len>=0)
{
line = codec->toUnicode(recvbuffer.mid(rbi,len), len);
rbi += len+1;
if (partial)
*partial = false;
return len;
}
recvbuffer="";
rbi=0;
//-1 on return signals "no more data" not error
return -1;
}
void KProcIO::virtual_hook( int id, void* data )
{ KProcess::virtual_hook( id, data ); }
#include "kprocio.moc"