summaryrefslogtreecommitdiff
path: root/accel-pptpd/sigchld.c
blob: fd07c0a6d0bd6392febbd3aaf063bf69a5b05eed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <signal.h>
#include <sys/signalfd.h>
#include <sys/wait.h>

#include "triton.h"
#include "spinlock.h"
#include "log.h"

#include "sigchld.h"

#include "memdebug.h"

static LIST_HEAD(handlers);
static int lock_refs;
static pthread_mutex_t handlers_lock = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t refs_cond = PTHREAD_COND_INITIALIZER;
static pthread_t sigchld_thr;

static void* sigchld_thread(void *arg)
{
	sigset_t set;
	struct sigchld_handler_t *h, *h0;
	pid_t pid;
	int status, sig;

	sigfillset(&set);
	sigdelset(&set, SIGKILL);
	sigdelset(&set, SIGSTOP);
	pthread_sigmask(SIG_BLOCK, &set, NULL);

	sigemptyset(&set);
	sigaddset(&set, SIGCHLD);
	sigaddset(&set, SIGQUIT);

	while (1) {	
		pid = waitpid(0, &status, 0);
		if (pid < 0) {
			if (errno == EINTR)
				continue;
			if (errno == ECHILD) {
				sigwait(&set, &sig);
				if (sig == SIGQUIT)
					break;
				continue;
			}
			log_error("sigchld: waitpid: %s\n", strerror(errno));
			continue;
		}

		pthread_mutex_lock(&handlers_lock);
		while (lock_refs)
			pthread_cond_wait(&refs_cond, &handlers_lock);

		h0 = NULL;
		list_for_each_entry(h, &handlers, entry) {
			if (h->pid == pid) {
				h0 = h;
				pthread_mutex_lock(&h0->lock);
				break;
			}
		}
		pthread_mutex_unlock(&handlers_lock);
		if (h0) {
			h0->handler(h0, WEXITSTATUS(status));
			list_del(&h0->entry);
			h0->pid = 0;
			pthread_mutex_unlock(&h0->lock);
		}
	}

	return NULL;
}

void __export sigchld_register_handler(struct sigchld_handler_t *h)
{
	pthread_mutex_init(&h->lock, NULL);

	pthread_mutex_lock(&handlers_lock);
	list_add_tail(&h->entry, &handlers);
	pthread_mutex_unlock(&handlers_lock);
}

void __export sigchld_unregister_handler(struct sigchld_handler_t *h)
{
	pthread_mutex_lock(&handlers_lock);
	pthread_mutex_lock(&h->lock);
	if (h->pid) {
		list_del(&h->entry);
		h->pid = 0;
	}
	pthread_mutex_unlock(&h->lock);
	pthread_mutex_unlock(&handlers_lock);
}

void __export sigchld_lock()
{
	pthread_mutex_lock(&handlers_lock);
	++lock_refs;
	pthread_mutex_unlock(&handlers_lock);
}

void __export sigchld_unlock()
{
	pthread_mutex_lock(&handlers_lock);
	if (--lock_refs == 0)
		pthread_cond_signal(&refs_cond);
	pthread_mutex_unlock(&handlers_lock);
}

static void __init init(void)
{
	if (pthread_create(&sigchld_thr, NULL, sigchld_thread, NULL))
		fprintf(stderr, "sigchld: pthread_create: %s\n", strerror(errno));
}