mirror of
https://github.com/VDR4Arch/vdr.git
synced 2023-10-10 13:36:52 +02:00
915 lines
23 KiB
C
915 lines
23 KiB
C
/*
|
|
* thread.c: A simple thread base class
|
|
*
|
|
* See the main source file 'vdr.c' for copyright information and
|
|
* how to reach the author.
|
|
*
|
|
* $Id: thread.c 4.3 2017/05/28 12:41:03 kls Exp $
|
|
*/
|
|
|
|
#include "thread.h"
|
|
#include <errno.h>
|
|
#include <linux/unistd.h>
|
|
#include <malloc.h>
|
|
#include <stdarg.h>
|
|
#include <stdlib.h>
|
|
#include <sys/resource.h>
|
|
#include <sys/syscall.h>
|
|
#include <sys/time.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/prctl.h>
|
|
#include <unistd.h>
|
|
#include "tools.h"
|
|
|
|
#define ABORT { dsyslog("ABORT!"); abort(); } // use debugger to trace back the problem
|
|
|
|
//#define DEBUG_LOCKING // uncomment this line to activate debug output for locking
|
|
//#define DEBUG_LOCKSEQ // uncomment this line to activate debug output for locking sequence
|
|
|
|
#ifdef DEBUG_LOCKING
|
|
#define dbglocking(a...) fprintf(stderr, a)
|
|
#else
|
|
#define dbglocking(a...)
|
|
#endif
|
|
|
|
static bool GetAbsTime(struct timespec *Abstime, int MillisecondsFromNow)
|
|
{
|
|
struct timeval now;
|
|
if (gettimeofday(&now, NULL) == 0) { // get current time
|
|
now.tv_sec += MillisecondsFromNow / 1000; // add full seconds
|
|
now.tv_usec += (MillisecondsFromNow % 1000) * 1000; // add microseconds
|
|
if (now.tv_usec >= 1000000) { // take care of an overflow
|
|
now.tv_sec++;
|
|
now.tv_usec -= 1000000;
|
|
}
|
|
Abstime->tv_sec = now.tv_sec; // seconds
|
|
Abstime->tv_nsec = now.tv_usec * 1000; // nano seconds
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// --- cCondWait -------------------------------------------------------------
|
|
|
|
cCondWait::cCondWait(void)
|
|
{
|
|
signaled = false;
|
|
pthread_mutex_init(&mutex, NULL);
|
|
pthread_cond_init(&cond, NULL);
|
|
}
|
|
|
|
cCondWait::~cCondWait()
|
|
{
|
|
pthread_cond_broadcast(&cond); // wake up any sleepers
|
|
pthread_cond_destroy(&cond);
|
|
pthread_mutex_destroy(&mutex);
|
|
}
|
|
|
|
void cCondWait::SleepMs(int TimeoutMs)
|
|
{
|
|
cCondWait w;
|
|
w.Wait(max(TimeoutMs, 3)); // making sure the time is >2ms to avoid a possible busy wait
|
|
}
|
|
|
|
bool cCondWait::Wait(int TimeoutMs)
|
|
{
|
|
pthread_mutex_lock(&mutex);
|
|
if (!signaled) {
|
|
if (TimeoutMs) {
|
|
struct timespec abstime;
|
|
if (GetAbsTime(&abstime, TimeoutMs)) {
|
|
while (!signaled) {
|
|
if (pthread_cond_timedwait(&cond, &mutex, &abstime) == ETIMEDOUT)
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
pthread_cond_wait(&cond, &mutex);
|
|
}
|
|
bool r = signaled;
|
|
signaled = false;
|
|
pthread_mutex_unlock(&mutex);
|
|
return r;
|
|
}
|
|
|
|
void cCondWait::Signal(void)
|
|
{
|
|
pthread_mutex_lock(&mutex);
|
|
signaled = true;
|
|
pthread_cond_broadcast(&cond);
|
|
pthread_mutex_unlock(&mutex);
|
|
}
|
|
|
|
// --- cCondVar --------------------------------------------------------------
|
|
|
|
cCondVar::cCondVar(void)
|
|
{
|
|
pthread_cond_init(&cond, 0);
|
|
}
|
|
|
|
cCondVar::~cCondVar()
|
|
{
|
|
pthread_cond_broadcast(&cond); // wake up any sleepers
|
|
pthread_cond_destroy(&cond);
|
|
}
|
|
|
|
void cCondVar::Wait(cMutex &Mutex)
|
|
{
|
|
if (Mutex.locked) {
|
|
int locked = Mutex.locked;
|
|
Mutex.locked = 0; // have to clear the locked count here, as pthread_cond_wait
|
|
// does an implicit unlock of the mutex
|
|
pthread_cond_wait(&cond, &Mutex.mutex);
|
|
Mutex.locked = locked;
|
|
}
|
|
}
|
|
|
|
bool cCondVar::TimedWait(cMutex &Mutex, int TimeoutMs)
|
|
{
|
|
bool r = true; // true = condition signaled, false = timeout
|
|
|
|
if (Mutex.locked) {
|
|
struct timespec abstime;
|
|
if (GetAbsTime(&abstime, TimeoutMs)) {
|
|
int locked = Mutex.locked;
|
|
Mutex.locked = 0; // have to clear the locked count here, as pthread_cond_timedwait
|
|
// does an implicit unlock of the mutex.
|
|
if (pthread_cond_timedwait(&cond, &Mutex.mutex, &abstime) == ETIMEDOUT)
|
|
r = false;
|
|
Mutex.locked = locked;
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
void cCondVar::Broadcast(void)
|
|
{
|
|
pthread_cond_broadcast(&cond);
|
|
}
|
|
|
|
// --- cRwLock ---------------------------------------------------------------
|
|
|
|
cRwLock::cRwLock(bool PreferWriter)
|
|
{
|
|
locked = 0;
|
|
writeLockThreadId = 0;
|
|
pthread_rwlockattr_t attr;
|
|
pthread_rwlockattr_init(&attr);
|
|
pthread_rwlockattr_setkind_np(&attr, PreferWriter ? PTHREAD_RWLOCK_PREFER_WRITER_NP : PTHREAD_RWLOCK_PREFER_READER_NP);
|
|
pthread_rwlock_init(&rwlock, &attr);
|
|
}
|
|
|
|
cRwLock::~cRwLock()
|
|
{
|
|
pthread_rwlock_destroy(&rwlock);
|
|
}
|
|
|
|
bool cRwLock::Lock(bool Write, int TimeoutMs)
|
|
{
|
|
int Result = 0;
|
|
struct timespec abstime;
|
|
if (TimeoutMs) {
|
|
if (!GetAbsTime(&abstime, TimeoutMs))
|
|
TimeoutMs = 0;
|
|
}
|
|
if (Write) {
|
|
Result = TimeoutMs ? pthread_rwlock_timedwrlock(&rwlock, &abstime) : pthread_rwlock_wrlock(&rwlock);
|
|
if (Result == 0)
|
|
writeLockThreadId = cThread::ThreadId();
|
|
}
|
|
else if (writeLockThreadId == cThread::ThreadId()) {
|
|
locked++; // there can be any number of stacked read locks, so we keep track here
|
|
Result = 0; // aquiring a read lock while holding a write lock within the same thread is OK
|
|
}
|
|
else
|
|
Result = TimeoutMs ? pthread_rwlock_timedrdlock(&rwlock, &abstime) : pthread_rwlock_rdlock(&rwlock);
|
|
return Result == 0;
|
|
}
|
|
|
|
void cRwLock::Unlock(void)
|
|
{
|
|
if (writeLockThreadId == cThread::ThreadId()) { // this is the thread that obtained the initial write lock
|
|
if (locked) { // this is the unlock of a read lock within the write lock
|
|
locked--;
|
|
return;
|
|
}
|
|
}
|
|
writeLockThreadId = 0;
|
|
pthread_rwlock_unlock(&rwlock);
|
|
}
|
|
|
|
// --- cMutex ----------------------------------------------------------------
|
|
|
|
cMutex::cMutex(void)
|
|
{
|
|
locked = 0;
|
|
pthread_mutexattr_t attr;
|
|
pthread_mutexattr_init(&attr);
|
|
pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_ERRORCHECK_NP);
|
|
pthread_mutex_init(&mutex, &attr);
|
|
}
|
|
|
|
cMutex::~cMutex()
|
|
{
|
|
pthread_mutex_destroy(&mutex);
|
|
}
|
|
|
|
void cMutex::Lock(void)
|
|
{
|
|
pthread_mutex_lock(&mutex);
|
|
locked++;
|
|
}
|
|
|
|
void cMutex::Unlock(void)
|
|
{
|
|
if (!--locked)
|
|
pthread_mutex_unlock(&mutex);
|
|
}
|
|
|
|
// --- cThread ---------------------------------------------------------------
|
|
|
|
tThreadId cThread::mainThreadId = 0;
|
|
|
|
cThread::cThread(const char *Description, bool LowPriority)
|
|
{
|
|
active = running = false;
|
|
childTid = 0;
|
|
childThreadId = 0;
|
|
description = NULL;
|
|
if (Description)
|
|
SetDescription("%s", Description);
|
|
lowPriority = LowPriority;
|
|
}
|
|
|
|
cThread::~cThread()
|
|
{
|
|
Cancel(); // just in case the derived class didn't call it
|
|
free(description);
|
|
}
|
|
|
|
void cThread::SetPriority(int Priority)
|
|
{
|
|
if (setpriority(PRIO_PROCESS, 0, Priority) < 0)
|
|
LOG_ERROR;
|
|
}
|
|
|
|
void cThread::SetIOPriority(int Priority)
|
|
{
|
|
if (syscall(SYS_ioprio_set, 1, 0, (Priority & 0xff) | (3 << 13)) < 0) // idle class
|
|
LOG_ERROR;
|
|
}
|
|
|
|
void cThread::SetDescription(const char *Description, ...)
|
|
{
|
|
free(description);
|
|
description = NULL;
|
|
if (Description) {
|
|
va_list ap;
|
|
va_start(ap, Description);
|
|
description = strdup(cString::vsprintf(Description, ap));
|
|
va_end(ap);
|
|
}
|
|
}
|
|
|
|
void *cThread::StartThread(cThread *Thread)
|
|
{
|
|
Thread->childThreadId = ThreadId();
|
|
if (Thread->description) {
|
|
dsyslog("%s thread started (pid=%d, tid=%d, prio=%s)", Thread->description, getpid(), Thread->childThreadId, Thread->lowPriority ? "low" : "high");
|
|
#ifdef PR_SET_NAME
|
|
if (prctl(PR_SET_NAME, Thread->description, 0, 0, 0) < 0)
|
|
esyslog("%s thread naming failed (pid=%d, tid=%d)", Thread->description, getpid(), Thread->childThreadId);
|
|
#endif
|
|
}
|
|
if (Thread->lowPriority) {
|
|
Thread->SetPriority(19);
|
|
Thread->SetIOPriority(7);
|
|
}
|
|
Thread->Action();
|
|
if (Thread->description)
|
|
dsyslog("%s thread ended (pid=%d, tid=%d)", Thread->description, getpid(), Thread->childThreadId);
|
|
Thread->running = false;
|
|
Thread->active = false;
|
|
return NULL;
|
|
}
|
|
|
|
#define THREAD_STOP_TIMEOUT 3000 // ms to wait for a thread to stop before newly starting it
|
|
#define THREAD_STOP_SLEEP 30 // ms to sleep while waiting for a thread to stop
|
|
|
|
bool cThread::Start(void)
|
|
{
|
|
if (!running) {
|
|
if (active) {
|
|
// Wait until the previous incarnation of this thread has completely ended
|
|
// before starting it newly:
|
|
cTimeMs RestartTimeout;
|
|
while (!running && active && RestartTimeout.Elapsed() < THREAD_STOP_TIMEOUT)
|
|
cCondWait::SleepMs(THREAD_STOP_SLEEP);
|
|
}
|
|
if (!active) {
|
|
active = running = true;
|
|
if (pthread_create(&childTid, NULL, (void *(*) (void *))&StartThread, (void *)this) == 0) {
|
|
pthread_detach(childTid); // auto-reap
|
|
}
|
|
else {
|
|
LOG_ERROR;
|
|
active = running = false;
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool cThread::Active(void)
|
|
{
|
|
if (active) {
|
|
//
|
|
// Single UNIX Spec v2 says:
|
|
//
|
|
// The pthread_kill() function is used to request
|
|
// that a signal be delivered to the specified thread.
|
|
//
|
|
// As in kill(), if sig is zero, error checking is
|
|
// performed but no signal is actually sent.
|
|
//
|
|
int err;
|
|
if ((err = pthread_kill(childTid, 0)) != 0) {
|
|
if (err != ESRCH)
|
|
LOG_ERROR;
|
|
childTid = 0;
|
|
active = running = false;
|
|
}
|
|
else
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void cThread::Cancel(int WaitSeconds)
|
|
{
|
|
running = false;
|
|
if (active && WaitSeconds > -1) {
|
|
if (WaitSeconds > 0) {
|
|
for (time_t t0 = time(NULL) + WaitSeconds; time(NULL) < t0; ) {
|
|
if (!Active())
|
|
return;
|
|
cCondWait::SleepMs(10);
|
|
}
|
|
esyslog("ERROR: %s thread %d won't end (waited %d seconds) - canceling it...", description ? description : "", childThreadId, WaitSeconds);
|
|
}
|
|
pthread_cancel(childTid);
|
|
childTid = 0;
|
|
active = false;
|
|
}
|
|
}
|
|
|
|
tThreadId cThread::ThreadId(void)
|
|
{
|
|
return syscall(__NR_gettid);
|
|
}
|
|
|
|
void cThread::SetMainThreadId(void)
|
|
{
|
|
if (mainThreadId == 0)
|
|
mainThreadId = ThreadId();
|
|
else
|
|
esyslog("ERROR: attempt to set main thread id to %d while it already is %d", ThreadId(), mainThreadId);
|
|
}
|
|
|
|
// --- cMutexLock ------------------------------------------------------------
|
|
|
|
cMutexLock::cMutexLock(cMutex *Mutex)
|
|
{
|
|
mutex = NULL;
|
|
locked = false;
|
|
Lock(Mutex);
|
|
}
|
|
|
|
cMutexLock::~cMutexLock()
|
|
{
|
|
if (mutex && locked)
|
|
mutex->Unlock();
|
|
}
|
|
|
|
bool cMutexLock::Lock(cMutex *Mutex)
|
|
{
|
|
if (Mutex && !mutex) {
|
|
mutex = Mutex;
|
|
Mutex->Lock();
|
|
locked = true;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// --- cThreadLock -----------------------------------------------------------
|
|
|
|
cThreadLock::cThreadLock(cThread *Thread)
|
|
{
|
|
thread = NULL;
|
|
locked = false;
|
|
Lock(Thread);
|
|
}
|
|
|
|
cThreadLock::~cThreadLock()
|
|
{
|
|
if (thread && locked)
|
|
thread->Unlock();
|
|
}
|
|
|
|
bool cThreadLock::Lock(cThread *Thread)
|
|
{
|
|
if (Thread && !thread) {
|
|
thread = Thread;
|
|
Thread->Lock();
|
|
locked = true;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// --- cStateLock ------------------------------------------------------------
|
|
|
|
#ifdef DEBUG_LOCKSEQ
|
|
#include <cxxabi.h>
|
|
#include <execinfo.h>
|
|
static cVector<tThreadId> StateLockThreadIds;
|
|
static cVector<int> StateLockFlags;
|
|
static cMutex StateLockMutex;
|
|
#define SLL_SIZE 20 // the number of log entries
|
|
#define SLL_LENGTH 256 // the maximum length of log entries
|
|
#define SLL_DELIM '#' // delimiter for caller info
|
|
static char StateLockLog[SLL_SIZE][SLL_LENGTH] = { 0 };
|
|
static int StateLockLogIndex = 0;
|
|
static bool DumpStateLocks = true;
|
|
#define BT_BUF_SIZE 100
|
|
|
|
static cString Demangle(char *s, char *Prefix = NULL)
|
|
{
|
|
char *Module = s;
|
|
char *Function = NULL;
|
|
//char *Offset = NULL;
|
|
char *Address = NULL;
|
|
// separate the string:
|
|
for (char *q = Module; *q; q++) {
|
|
if (*q == '(') {
|
|
*q = 0;
|
|
Function = q + 1;
|
|
}
|
|
else if (*q == '+') {
|
|
*q = 0;
|
|
//Offset = q + 1;
|
|
}
|
|
else if (*q == ')')
|
|
*q = 0;
|
|
else if (*q == '[')
|
|
Address = q + 1;
|
|
else if (*q == ']') {
|
|
*q = 0;
|
|
break;
|
|
}
|
|
}
|
|
// demangle the function name:
|
|
int status;
|
|
char *r = abi::__cxa_demangle(Function, NULL, 0, &status);
|
|
if (r)
|
|
Function = r;
|
|
cString d = cString::sprintf("%s %s %s", Prefix ? Prefix : "", Module, Function);
|
|
// determine the file name and line number:
|
|
cString cmd = cString::sprintf("addr2line --exe=%s --functions --demangle --basename %s", Module, Address);
|
|
cPipe p;
|
|
if (p.Open(cmd, "r")) {
|
|
int n = 0;
|
|
cReadLine rl;
|
|
while (char *l = rl.Read(p)) {
|
|
if (n == 0) {
|
|
if (!isempty(Function) && strcmp(l, Function))
|
|
d = cString::sprintf("%s calling %s", *d, l);
|
|
}
|
|
else
|
|
d = cString::sprintf("%s at %s", *d, l);
|
|
n++;
|
|
}
|
|
p.Close();
|
|
}
|
|
free(r);
|
|
return d;
|
|
}
|
|
|
|
static void BackTrace(void)
|
|
{
|
|
void *b[BT_BUF_SIZE];
|
|
int n = backtrace(b, BT_BUF_SIZE);
|
|
if (char **s = backtrace_symbols(b, n)) {
|
|
for (int i = 5; i < n; i++)
|
|
fprintf(stderr, "%s\n", *Demangle(s[i]));
|
|
free(s);
|
|
}
|
|
}
|
|
|
|
static void BackTraceCaller(int Level, char *t, int l)
|
|
{
|
|
void *b[BT_BUF_SIZE];
|
|
int n = backtrace(b, BT_BUF_SIZE);
|
|
if (char **s = backtrace_symbols(b, n)) {
|
|
strn0cpy(t, s[Level], l);
|
|
free(s);
|
|
}
|
|
}
|
|
|
|
static void DumpStateLockLog(tThreadId ThreadId, const char *Name)
|
|
{
|
|
if (!DumpStateLocks)
|
|
return;
|
|
DumpStateLocks = false;
|
|
for (int i = 0; i < SLL_SIZE; i++) {
|
|
char *s = StateLockLog[StateLockLogIndex];
|
|
if (*s) {
|
|
if (char *Delim = strchr(s, SLL_DELIM)) {
|
|
*Delim = 0;
|
|
fprintf(stderr, "%s\n", *Demangle(Delim + 1, s));
|
|
}
|
|
}
|
|
if (++StateLockLogIndex >= SLL_SIZE)
|
|
StateLockLogIndex = 0;
|
|
}
|
|
fprintf(stderr, "%5d invalid lock sequence: %-12s\n", ThreadId, Name);
|
|
fprintf(stderr, "full backtrace:\n");
|
|
BackTrace();
|
|
}
|
|
|
|
static void CheckStateLockLevel(const char *Name, bool Lock, bool Write = false)
|
|
{
|
|
if (Name) {
|
|
int n = *Name - '0';
|
|
if (1 <= n && n <= 9) {
|
|
int b = 1 << (n - 1);
|
|
cMutexLock MutexLock(&StateLockMutex);
|
|
tThreadId ThreadId = cThread::ThreadId();
|
|
int Index = StateLockThreadIds.IndexOf(ThreadId);
|
|
if (Index < 0) {
|
|
if (Lock) {
|
|
Index = StateLockThreadIds.Size();
|
|
StateLockThreadIds.Append(ThreadId);
|
|
StateLockFlags.Append(0);
|
|
}
|
|
else
|
|
return;
|
|
}
|
|
char *p = StateLockLog[StateLockLogIndex];
|
|
char *q = p;
|
|
q += sprintf(q, "%5d", ThreadId);
|
|
for (int i = 0; i <= 9; i++) {
|
|
char c = '-';
|
|
if (StateLockFlags[Index] & (1 << i))
|
|
c = '*';
|
|
if (i == n - 1)
|
|
c = Lock ? Write ? 'W' : 'R' : '-';
|
|
q += sprintf(q, " %c", c);
|
|
}
|
|
q += sprintf(q, " %c%c", Lock ? 'L' : 'U', SLL_DELIM);
|
|
BackTraceCaller(Lock ? 5 : 3, q, SLL_LENGTH - (q - p));
|
|
if (++StateLockLogIndex >= SLL_SIZE)
|
|
StateLockLogIndex = 0;
|
|
if (Lock) {
|
|
if ((StateLockFlags[Index] & ~b) < b)
|
|
StateLockFlags[Index] |= b;
|
|
else if ((StateLockFlags[Index] & b) == 0)
|
|
DumpStateLockLog(ThreadId, Name);
|
|
}
|
|
else
|
|
StateLockFlags[Index] &= ~b;
|
|
}
|
|
}
|
|
}
|
|
#define dbglockseq(n, l, w) CheckStateLockLevel(n, l, w)
|
|
#else
|
|
#define dbglockseq(n, l, w)
|
|
#endif // DEBUG_LOCKSEQ
|
|
|
|
cStateLock::cStateLock(const char *Name)
|
|
:rwLock(true)
|
|
{
|
|
name = Name;
|
|
threadId = 0;
|
|
state = 0;
|
|
explicitModify = false;
|
|
}
|
|
|
|
bool cStateLock::Lock(cStateKey &StateKey, bool Write, int TimeoutMs)
|
|
{
|
|
dbglocking("%5d %-12s %10p lock state = %d/%d write = %d timeout = %d\n", cThread::ThreadId(), name, &StateKey, state, StateKey.state, Write, TimeoutMs);
|
|
StateKey.timedOut = false;
|
|
if (StateKey.stateLock) {
|
|
esyslog("ERROR: StateKey already in use in call to cStateLock::Lock() (tid=%d, lock=%s)", StateKey.stateLock->threadId, name);
|
|
ABORT;
|
|
return false;
|
|
}
|
|
if (rwLock.Lock(Write, TimeoutMs)) {
|
|
dbglockseq(name, true, Write);
|
|
StateKey.stateLock = this;
|
|
if (Write) {
|
|
dbglocking("%5d %-12s %10p locked write\n", cThread::ThreadId(), name, &StateKey);
|
|
threadId = cThread::ThreadId();
|
|
StateKey.write = true;
|
|
return true;
|
|
}
|
|
else if (state != StateKey.state) {
|
|
dbglocking("%5d %-12s %10p locked read\n", cThread::ThreadId(), name, &StateKey);
|
|
return true;
|
|
}
|
|
else {
|
|
dbglocking("%5d %-12s %10p state unchanged\n", cThread::ThreadId(), name, &StateKey);
|
|
StateKey.stateLock = NULL;
|
|
dbglockseq(name, false, false);
|
|
rwLock.Unlock();
|
|
}
|
|
}
|
|
else if (TimeoutMs) {
|
|
dbglocking("%5d %-12s %10p timeout\n", cThread::ThreadId(), name, &StateKey);
|
|
StateKey.timedOut = true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void cStateLock::Unlock(cStateKey &StateKey, bool IncState)
|
|
{
|
|
dbglocking("%5d %-12s %10p unlock state = %d/%d inc = %d\n", cThread::ThreadId(), name, &StateKey, state, StateKey.state, IncState);
|
|
if (StateKey.stateLock != this) {
|
|
esyslog("ERROR: cStateLock::Unlock() called with an unused key (tid=%d, lock=%s)", threadId, name);
|
|
ABORT;
|
|
return;
|
|
}
|
|
if (StateKey.write && threadId != cThread::ThreadId()) {
|
|
esyslog("ERROR: cStateLock::Unlock() called without holding a lock (tid=%d, lock=%s)", threadId, name);
|
|
ABORT;
|
|
return;
|
|
}
|
|
if (StateKey.write && IncState && !explicitModify)
|
|
state++;
|
|
StateKey.state = state;
|
|
if (StateKey.write) {
|
|
StateKey.write = false;
|
|
threadId = 0;
|
|
explicitModify = false;
|
|
}
|
|
dbglockseq(name, false, false);
|
|
rwLock.Unlock();
|
|
}
|
|
|
|
void cStateLock::IncState(void)
|
|
{
|
|
if (threadId != cThread::ThreadId()) {
|
|
esyslog("ERROR: cStateLock::IncState() called without holding a lock (tid=%d, lock=%s)", threadId, name);
|
|
ABORT;
|
|
}
|
|
else
|
|
state++;
|
|
}
|
|
|
|
// --- cStateKey -------------------------------------------------------------
|
|
|
|
cStateKey::cStateKey(bool IgnoreFirst)
|
|
{
|
|
stateLock = NULL;
|
|
write = false;
|
|
state = 0;
|
|
if (!IgnoreFirst)
|
|
Reset();
|
|
}
|
|
|
|
cStateKey::~cStateKey()
|
|
{
|
|
if (stateLock) {
|
|
esyslog("ERROR: cStateKey::~cStateKey() called without releasing the lock first (tid=%d, lock=%s, key=%p)", stateLock->threadId, stateLock->name, this);
|
|
ABORT;
|
|
}
|
|
}
|
|
|
|
void cStateKey::Reset(void)
|
|
{
|
|
state = -1; // lock and key are initialized differently, to make the first check return true
|
|
}
|
|
|
|
void cStateKey::Remove(bool IncState)
|
|
{
|
|
if (stateLock) {
|
|
stateLock->Unlock(*this, IncState);
|
|
stateLock = NULL;
|
|
}
|
|
else {
|
|
esyslog("ERROR: cStateKey::Remove() called without holding a lock (key=%p)", this);
|
|
ABORT;
|
|
}
|
|
}
|
|
|
|
bool cStateKey::StateChanged(void)
|
|
{
|
|
if (!stateLock) {
|
|
esyslog("ERROR: cStateKey::StateChanged() called without holding a lock (tid=%d, key=%p)", cThread::ThreadId(), this);
|
|
ABORT;
|
|
}
|
|
else if (write)
|
|
return state != stateLock->state;
|
|
else
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
// --- cIoThrottle -----------------------------------------------------------
|
|
|
|
cMutex cIoThrottle::mutex;
|
|
int cIoThrottle::count = 0;
|
|
|
|
cIoThrottle::cIoThrottle(void)
|
|
{
|
|
active = false;
|
|
}
|
|
|
|
cIoThrottle::~cIoThrottle()
|
|
{
|
|
Release();
|
|
}
|
|
|
|
void cIoThrottle::Activate(void)
|
|
{
|
|
if (!active) {
|
|
mutex.Lock();
|
|
count++;
|
|
active = true;
|
|
dsyslog("i/o throttle activated, count = %d (tid=%d)", count, cThread::ThreadId());
|
|
mutex.Unlock();
|
|
}
|
|
}
|
|
|
|
void cIoThrottle::Release(void)
|
|
{
|
|
if (active) {
|
|
mutex.Lock();
|
|
count--;
|
|
active = false;
|
|
dsyslog("i/o throttle released, count = %d (tid=%d)", count, cThread::ThreadId());
|
|
mutex.Unlock();
|
|
}
|
|
}
|
|
|
|
bool cIoThrottle::Engaged(void)
|
|
{
|
|
return count > 0;
|
|
}
|
|
|
|
// --- cPipe -----------------------------------------------------------------
|
|
|
|
// cPipe::Open() and cPipe::Close() are based on code originally received from
|
|
// Andreas Vitting <Andreas@huji.de>
|
|
|
|
cPipe::cPipe(void)
|
|
{
|
|
pid = -1;
|
|
f = NULL;
|
|
}
|
|
|
|
cPipe::~cPipe()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
bool cPipe::Open(const char *Command, const char *Mode)
|
|
{
|
|
int fd[2];
|
|
|
|
if (pipe(fd) < 0) {
|
|
LOG_ERROR;
|
|
return false;
|
|
}
|
|
if ((pid = fork()) < 0) { // fork failed
|
|
LOG_ERROR;
|
|
close(fd[0]);
|
|
close(fd[1]);
|
|
return false;
|
|
}
|
|
|
|
const char *mode = "w";
|
|
int iopipe = 0;
|
|
|
|
if (pid > 0) { // parent process
|
|
if (strcmp(Mode, "r") == 0) {
|
|
mode = "r";
|
|
iopipe = 1;
|
|
}
|
|
close(fd[iopipe]);
|
|
if ((f = fdopen(fd[1 - iopipe], mode)) == NULL) {
|
|
LOG_ERROR;
|
|
close(fd[1 - iopipe]);
|
|
}
|
|
return f != NULL;
|
|
}
|
|
else { // child process
|
|
int iofd = STDOUT_FILENO;
|
|
if (strcmp(Mode, "w") == 0) {
|
|
iopipe = 1;
|
|
iofd = STDIN_FILENO;
|
|
}
|
|
close(fd[iopipe]);
|
|
if (dup2(fd[1 - iopipe], iofd) == -1) { // now redirect
|
|
LOG_ERROR;
|
|
close(fd[1 - iopipe]);
|
|
_exit(-1);
|
|
}
|
|
else {
|
|
int MaxPossibleFileDescriptors = getdtablesize();
|
|
for (int i = STDERR_FILENO + 1; i < MaxPossibleFileDescriptors; i++)
|
|
close(i); //close all dup'ed filedescriptors
|
|
if (execl("/bin/sh", "sh", "-c", Command, NULL) == -1) {
|
|
LOG_ERROR_STR(Command);
|
|
close(fd[1 - iopipe]);
|
|
_exit(-1);
|
|
}
|
|
}
|
|
_exit(0);
|
|
}
|
|
}
|
|
|
|
int cPipe::Close(void)
|
|
{
|
|
int ret = -1;
|
|
|
|
if (f) {
|
|
fclose(f);
|
|
f = NULL;
|
|
}
|
|
|
|
if (pid > 0) {
|
|
int status = 0;
|
|
int i = 5;
|
|
while (i > 0) {
|
|
ret = waitpid(pid, &status, WNOHANG);
|
|
if (ret < 0) {
|
|
if (errno != EINTR && errno != ECHILD) {
|
|
LOG_ERROR;
|
|
break;
|
|
}
|
|
}
|
|
else if (ret == pid)
|
|
break;
|
|
i--;
|
|
cCondWait::SleepMs(100);
|
|
}
|
|
if (!i) {
|
|
kill(pid, SIGKILL);
|
|
ret = -1;
|
|
}
|
|
else if (ret == -1 || !WIFEXITED(status))
|
|
ret = -1;
|
|
pid = -1;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
// --- SystemExec ------------------------------------------------------------
|
|
|
|
int SystemExec(const char *Command, bool Detached)
|
|
{
|
|
pid_t pid;
|
|
|
|
if ((pid = fork()) < 0) { // fork failed
|
|
LOG_ERROR;
|
|
return -1;
|
|
}
|
|
|
|
if (pid > 0) { // parent process
|
|
int status = 0;
|
|
if (waitpid(pid, &status, 0) < 0) {
|
|
LOG_ERROR;
|
|
return -1;
|
|
}
|
|
return status;
|
|
}
|
|
else { // child process
|
|
if (Detached) {
|
|
// Fork again and let first child die - grandchild stays alive without parent
|
|
if (fork() > 0)
|
|
_exit(0);
|
|
// Start a new session
|
|
pid_t sid = setsid();
|
|
if (sid < 0)
|
|
LOG_ERROR;
|
|
// close STDIN and re-open as /dev/null
|
|
int devnull = open("/dev/null", O_RDONLY);
|
|
if (devnull < 0 || dup2(devnull, 0) < 0)
|
|
LOG_ERROR;
|
|
}
|
|
int MaxPossibleFileDescriptors = getdtablesize();
|
|
for (int i = STDERR_FILENO + 1; i < MaxPossibleFileDescriptors; i++)
|
|
close(i); //close all dup'ed filedescriptors
|
|
if (execl("/bin/sh", "sh", "-c", Command, NULL) == -1) {
|
|
LOG_ERROR_STR(Command);
|
|
_exit(-1);
|
|
}
|
|
_exit(0);
|
|
}
|
|
}
|