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/tdecore/kprocctrl.cpp

278 lines
6.3 KiB

/* This file is part of the KDE libraries
Copyright (C) 1997 Christian Czezakte (e9025461@student.tuwien.ac.at)
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.
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.
*/
#include "kprocess.h"
#include "kprocctrl.h"
#include <config.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <tqsocketnotifier.h>
TDEProcessController *TDEProcessController::theTDEProcessController;
int TDEProcessController::refCount;
void TDEProcessController::ref()
{
if( !refCount ) {
theTDEProcessController = new TDEProcessController;
setupHandlers();
}
refCount++;
}
void TDEProcessController::deref()
{
refCount--;
if( !refCount ) {
resetHandlers();
delete theTDEProcessController;
theTDEProcessController = 0;
}
}
TDEProcessController::TDEProcessController()
: needcheck( false )
{
if( pipe( fd ) )
{
perror( "pipe" );
abort();
}
fcntl( fd[0], F_SETFL, O_NONBLOCK ); // in case slotDoHousekeeping is called without polling first
fcntl( fd[1], F_SETFL, O_NONBLOCK ); // in case it fills up
fcntl( fd[0], F_SETFD, FD_CLOEXEC );
fcntl( fd[1], F_SETFD, FD_CLOEXEC );
notifier = new TQSocketNotifier( fd[0], TQSocketNotifier::Read );
notifier->setEnabled( true );
TQObject::connect( notifier, TQT_SIGNAL(activated(int)),
TQT_SLOT(slotDoHousekeeping()));
}
TDEProcessController::~TDEProcessController()
{
delete notifier;
close( fd[0] );
close( fd[1] );
}
extern "C" {
static void theReaper( int num )
{
TDEProcessController::theSigCHLDHandler( num );
}
}
#ifdef Q_OS_UNIX
struct sigaction TDEProcessController::oldChildHandlerData;
#endif
bool TDEProcessController::handlerSet = false;
void TDEProcessController::setupHandlers()
{
if( handlerSet )
return;
handlerSet = true;
#ifdef Q_OS_UNIX
struct sigaction act;
sigemptyset( &act.sa_mask );
act.sa_handler = SIG_IGN;
act.sa_flags = 0;
sigaction( SIGPIPE, &act, 0L );
act.sa_handler = theReaper;
act.sa_flags = SA_NOCLDSTOP;
// CC: take care of SunOS which automatically restarts interrupted system
// calls (and thus does not have SA_RESTART)
#ifdef SA_RESTART
act.sa_flags |= SA_RESTART;
#endif
sigaction( SIGCHLD, &act, &oldChildHandlerData );
sigaddset( &act.sa_mask, SIGCHLD );
// Make sure we don't block this signal. gdb tends to do that :-(
sigprocmask( SIG_UNBLOCK, &act.sa_mask, 0 );
#else
//TODO: win32
#endif
}
void TDEProcessController::resetHandlers()
{
if( !handlerSet )
return;
handlerSet = false;
#ifdef Q_OS_UNIX
sigaction( SIGCHLD, &oldChildHandlerData, 0 );
#else
//TODO: win32
#endif
// there should be no problem with SIGPIPE staying SIG_IGN
}
// the pipe is needed to sync the child reaping with our event processing,
// as otherwise there are race conditions, locking requirements, and things
// generally get harder
void TDEProcessController::theSigCHLDHandler( int arg )
{
int saved_errno = errno;
char dummy = 0;
::write( theTDEProcessController->fd[1], &dummy, 1 );
#ifdef Q_OS_UNIX
if( oldChildHandlerData.sa_handler != SIG_IGN &&
oldChildHandlerData.sa_handler != SIG_DFL )
oldChildHandlerData.sa_handler( arg ); // call the old handler
#else
//TODO: win32
#endif
errno = saved_errno;
}
int TDEProcessController::notifierFd() const
{
return fd[0];
}
void TDEProcessController::unscheduleCheck()
{
char dummy[16]; // somewhat bigger - just in case several have queued up
if( ::read( fd[0], dummy, sizeof(dummy) ) > 0 )
needcheck = true;
}
void
TDEProcessController::rescheduleCheck()
{
if( needcheck )
{
needcheck = false;
char dummy = 0;
::write( fd[1], &dummy, 1 );
}
}
void TDEProcessController::slotDoHousekeeping()
{
char dummy[16]; // somewhat bigger - just in case several have queued up
::read( fd[0], dummy, sizeof(dummy) );
int status;
again:
TQValueListIterator<TDEProcess*> it( kProcessList.begin() );
TQValueListIterator<TDEProcess*> eit( kProcessList.end() );
while( it != eit )
{
TDEProcess *prc = *it;
if( prc->runs && waitpid( prc->pid_, &status, WNOHANG ) > 0 )
{
prc->processHasExited( status );
// the callback can nuke the whole process list and even 'this'
if (!theTDEProcessController)
return;
goto again;
}
++it;
}
TQValueListIterator<int> uit( unixProcessList.begin() );
TQValueListIterator<int> ueit( unixProcessList.end() );
while( uit != ueit )
{
if( waitpid( *uit, 0, WNOHANG ) > 0 )
{
uit = unixProcessList.remove( uit );
deref(); // counterpart to addProcess, can invalidate 'this'
} else
++uit;
}
}
bool TDEProcessController::waitForProcessExit( int timeout )
{
#ifdef Q_OS_UNIX
for(;;)
{
struct timeval tv, *tvp;
if (timeout < 0)
tvp = 0;
else
{
tv.tv_sec = timeout;
tv.tv_usec = 0;
tvp = &tv;
}
fd_set fds;
FD_ZERO( &fds );
FD_SET( fd[0], &fds );
switch( select( fd[0]+1, &fds, 0, 0, tvp ) )
{
case -1:
if( errno == EINTR )
continue;
// fall through; should never happen
case 0:
return false;
default:
slotDoHousekeeping();
return true;
}
}
#else
//TODO: win32
return false;
#endif
}
void TDEProcessController::addTDEProcess( TDEProcess* p )
{
kProcessList.append( p );
}
void TDEProcessController::removeTDEProcess( TDEProcess* p )
{
kProcessList.remove( p );
}
void TDEProcessController::addProcess( int pid )
{
unixProcessList.append( pid );
ref(); // make sure we stay around when the TDEProcess goes away
}
#include "kprocctrl.moc"