4 This file is part of polypaudio.
6 polypaudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU Lesser General Public License as
8 published by the Free Software Foundation; either version 2 of the
9 License, or (at your option) any later version.
11 polypaudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
16 You should have received a copy of the GNU Lesser General Public
17 License along with polypaudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
26 #include <polyp/error.h>
28 #include <polypcore/core-util.h>
29 #include <polypcore/log.h>
43 #ifdef HAVE_SYS_RESOURCE_H
44 #include <sys/resource.h>
47 /* This module implements a watchdog that makes sure that the current
48 * process doesn't consume more than 70% CPU time for 10 seconds. This
49 * is very useful when using SCHED_FIFO scheduling which effectively
50 * disables multitasking. */
52 /* Method of operation: Using SIGXCPU a signal handler is called every
53 * 10s process CPU time. That function checks if less than 14s system
54 * time have passed. In that case, it tries to contact the main event
55 * loop through a pipe. After two additional seconds it is checked
56 * whether the main event loop contact was successful. If not, the
57 * program is terminated forcibly. */
59 /* Utilize this much CPU time at maximum */
60 #define CPUTIME_PERCENT 70
63 #define CPUTIME_INTERVAL_SOFT (10)
65 /* Recheck after 2s */
66 #define CPUTIME_INTERVAL_HARD (2)
68 /* Time of the last CPU load check */
69 static time_t last_time
= 0;
71 /* Pipe for communicating with the main loop */
72 static int the_pipe
[2] = {-1, -1};
74 /* Main event loop and IO event for the FIFO */
75 static pa_mainloop_api
*api
= NULL
;
76 static pa_io_event
*io_event
= NULL
;
78 /* Saved sigaction struct for SIGXCPU */
79 static struct sigaction sigaction_prev
;
81 /* Nonzero after pa_cpu_limit_init() */
82 static int installed
= 0;
84 /* The current state of operation */
86 PHASE_IDLE
, /* Normal state */
87 PHASE_SOFT
/* After CPU overload has been detected */
90 /* Reset the SIGXCPU timer to the next t seconds */
91 static void reset_cpu_time(int t
) {
97 /* Get the current CPU time of the current process */
98 r
= getrusage(RUSAGE_SELF
, &ru
);
101 n
= ru
.ru_utime
.tv_sec
+ ru
.ru_stime
.tv_sec
+ t
;
103 r
= getrlimit(RLIMIT_CPU
, &rl
);
107 r
= setrlimit(RLIMIT_CPU
, &rl
);
111 /* A simple, thread-safe puts() work-alike */
112 static void write_err(const char *p
) {
113 pa_loop_write(2, p
, strlen(p
));
116 /* The signal handler, called on every SIGXCPU */
117 static void signal_handler(int sig
) {
118 assert(sig
== SIGXCPU
);
120 if (phase
== PHASE_IDLE
) {
123 #ifdef PRINT_CPU_LOAD
129 #ifdef PRINT_CPU_LOAD
130 snprintf(t
, sizeof(t
), "Using %0.1f%% CPU\n", (double)CPUTIME_INTERVAL_SOFT
/(now
-last_time
)*100);
134 if (CPUTIME_INTERVAL_SOFT
>= ((now
-last_time
)*(double)CPUTIME_PERCENT
/100)) {
135 static const char c
= 'X';
137 write_err("Soft CPU time limit exhausted, terminating.\n");
139 /* Try a soft cleanup */
140 write(the_pipe
[1], &c
, sizeof(c
));
142 reset_cpu_time(CPUTIME_INTERVAL_HARD
);
146 /* Everything's fine */
147 reset_cpu_time(CPUTIME_INTERVAL_SOFT
);
151 } else if (phase
== PHASE_SOFT
) {
152 write_err("Hard CPU time limit exhausted, terminating forcibly.\n");
153 _exit(1); /* Forced exit */
157 /* Callback for IO events on the FIFO */
158 static void callback(pa_mainloop_api
*m
, pa_io_event
*e
, int fd
, pa_io_event_flags f
, void *userdata
) {
160 assert(m
&& e
&& f
== PA_IO_EVENT_INPUT
&& e
== io_event
&& fd
== the_pipe
[0]);
161 read(the_pipe
[0], &c
, sizeof(c
));
162 m
->quit(m
, 1); /* Quit the main loop */
165 /* Initializes CPU load limiter */
166 int pa_cpu_limit_init(pa_mainloop_api
*m
) {
168 assert(m
&& !api
&& !io_event
&& the_pipe
[0] == -1 && the_pipe
[1] == -1 && !installed
);
172 /* Prepare the main loop pipe */
173 if (pipe(the_pipe
) < 0) {
174 pa_log(__FILE__
": pipe() failed: %s", pa_cstrerror(errno
));
178 pa_make_nonblock_fd(the_pipe
[0]);
179 pa_make_nonblock_fd(the_pipe
[1]);
180 pa_fd_set_cloexec(the_pipe
[0], 1);
181 pa_fd_set_cloexec(the_pipe
[1], 1);
184 io_event
= api
->io_new(m
, the_pipe
[0], PA_IO_EVENT_INPUT
, callback
, NULL
);
188 /* Install signal handler for SIGXCPU */
189 memset(&sa
, 0, sizeof(sa
));
190 sa
.sa_handler
= signal_handler
;
191 sigemptyset(&sa
.sa_mask
);
192 sa
.sa_flags
= SA_RESTART
;
194 if (sigaction(SIGXCPU
, &sa
, &sigaction_prev
) < 0) {
201 reset_cpu_time(CPUTIME_INTERVAL_SOFT
);
206 /* Shutdown CPU load limiter */
207 void pa_cpu_limit_done(void) {
212 api
->io_free(io_event
);
217 if (the_pipe
[0] >= 0)
219 if (the_pipe
[1] >= 0)
221 the_pipe
[0] = the_pipe
[1] = -1;
224 r
= sigaction(SIGXCPU
, &sigaction_prev
, NULL
);
230 #else /* HAVE_SIGXCPU */
232 int pa_cpu_limit_init(PA_GCC_UNUSED pa_mainloop_api
*m
) {
236 void pa_cpu_limit_done(void) {