]>
code.delx.au - pulseaudio/blob - polyp/cpulimit.c
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 published
8 by the Free Software Foundation; either version 2 of the License,
9 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 License
17 along with polypaudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
27 #include <sys/resource.h>
35 /* Utilize this much CPU time at maximum */
36 #define CPUTIME_PERCENT 70
38 #define CPUTIME_INTERVAL_SOFT (10)
39 #define CPUTIME_INTERVAL_HARD (2)
41 static time_t last_time
= 0;
42 static int the_pipe
[2] = {-1, -1};
43 static struct pa_mainloop_api
*api
= NULL
;
44 static struct pa_io_event
*io_event
= NULL
;
45 static struct sigaction sigaction_prev
;
46 static int installed
= 0;
53 static void reset_cpu_time(int t
) {
59 r
= getrusage(RUSAGE_SELF
, &ru
);
62 n
= ru
.ru_utime
.tv_sec
+ ru
.ru_stime
.tv_sec
+ t
;
64 r
= getrlimit(RLIMIT_CPU
, &rl
);
68 r
= setrlimit(RLIMIT_CPU
, &rl
);
72 static void write_err(const char *p
) {
73 pa_loop_write(2, p
, strlen(p
));
76 static void signal_handler(int sig
) {
77 assert(sig
== SIGXCPU
);
79 if (phase
== PHASE_IDLE
) {
89 snprintf(t
, sizeof(t
), "Using %0.1f%% CPU\n", (double)CPUTIME_INTERVAL_SOFT
/(now
-last_time
)*100);
93 if (CPUTIME_INTERVAL_SOFT
>= ((now
-last_time
)*(double)CPUTIME_PERCENT
/100)) {
94 static const char c
= 'X';
96 write_err("Soft CPU time limit exhausted, terminating.\n");
98 /* Try a soft cleanup */
99 write(the_pipe
[1], &c
, sizeof(c
));
101 reset_cpu_time(CPUTIME_INTERVAL_HARD
);
105 /* Everything's fine */
106 reset_cpu_time(CPUTIME_INTERVAL_SOFT
);
110 } else if (phase
== PHASE_SOFT
) {
111 write_err("Hard CPU time limit exhausted, terminating forcibly.\n");
116 static void callback(struct pa_mainloop_api
*m
, struct pa_io_event
*e
, int fd
, enum pa_io_event_flags f
, void *userdata
) {
118 assert(m
&& e
&& f
== PA_IO_EVENT_INPUT
&& e
== io_event
&& fd
== the_pipe
[0]);
119 read(the_pipe
[0], &c
, sizeof(c
));
123 int pa_cpu_limit_init(struct pa_mainloop_api
*m
) {
125 assert(m
&& !api
&& !io_event
&& the_pipe
[0] == -1 && the_pipe
[1] == -1);
129 if (pipe(the_pipe
) < 0) {
130 pa_log(__FILE__
": pipe() failed: %s\n", strerror(errno
));
134 pa_make_nonblock_fd(the_pipe
[0]);
135 pa_make_nonblock_fd(the_pipe
[1]);
136 pa_fd_set_cloexec(the_pipe
[0], 1);
137 pa_fd_set_cloexec(the_pipe
[1], 1);
140 io_event
= api
->io_new(m
, the_pipe
[0], PA_IO_EVENT_INPUT
, callback
, NULL
);
144 memset(&sa
, 0, sizeof(sa
));
145 sa
.sa_handler
= signal_handler
;
146 sigemptyset(&sa
.sa_mask
);
147 sa
.sa_flags
= SA_RESTART
;
149 if (sigaction(SIGXCPU
, &sa
, &sigaction_prev
) < 0) {
156 reset_cpu_time(CPUTIME_INTERVAL_SOFT
);
161 void pa_cpu_limit_done(void) {
166 api
->io_free(io_event
);
171 if (the_pipe
[0] >= 0)
173 if (the_pipe
[1] >= 0)
175 the_pipe
[0] = the_pipe
[1] = -1;
178 r
= sigaction(SIGXCPU
, &sigaction_prev
, NULL
);