1 files changed, 175 insertions, 0 deletions
diff --git a/polyp/cpulimit.c b/polyp/cpulimit.c
new file mode 100644
index 00000000..822e1f33
--- /dev/null
+++ b/polyp/cpulimit.c
@@ -0,0 +1,175 @@
+/* $Id$ */
+
+/***
+  This file is part of polypaudio.
+ 
+  polypaudio is free software; you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published
+  by the Free Software Foundation; either version 2 of the License,
+  or (at your option) any later version.
+ 
+  polypaudio 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
+  General Public License for more details.
+ 
+  You should have received a copy of the GNU General Public License
+  along with polypaudio; if not, write to the Free Software
+  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+  USA.
+***/
+
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+#include <sys/time.h>
+#include <sys/resource.h>
+#include <unistd.h>
+#include <signal.h>
+
+#include "cpulimit.h"
+#include "util.h"
+
+/* Utilize this much CPU time at most */
+#define CPUTIME_PERCENT 70
+
+#define CPUTIME_INTERVAL_SOFT (5)
+#define CPUTIME_INTERVAL_HARD (2)
+
+static time_t last_time = 0;
+static int the_pipe[2] = {-1, -1};
+static struct pa_mainloop_api *api = NULL;
+static struct pa_io_event *io_event = NULL;
+static struct sigaction sigaction_prev;
+static int installed = 0;
+
+static enum  {
+    PHASE_IDLE,
+    PHASE_SOFT
+} phase = PHASE_IDLE;
+
+static void reset_cpu_time(int t) {
+    int r;
+    long n;
+    struct rlimit rl;
+    struct rusage ru;
+
+    r = getrusage(RUSAGE_SELF, &ru);
+    assert(r >= 0);
+
+    n = ru.ru_utime.tv_sec + ru.ru_stime.tv_sec + t;
+
+    r = getrlimit(RLIMIT_CPU, &rl);
+    assert(r >= 0);
+
+    rl.rlim_cur = n;
+    r = setrlimit(RLIMIT_CPU, &rl);
+    assert(r >= 0);
+}
+
+static void write_err(const char *p) {
+    pa_loop_write(2, p, strlen(p));
+}
+
+static void signal_handler(int sig) {
+    assert(sig == SIGXCPU);
+
+    if (phase == PHASE_IDLE) {
+        time_t now;
+        char t[256];
+
+        time(&now);
+
+        snprintf(t, sizeof(t), "Using %0.1f%% CPU\n", (double)CPUTIME_INTERVAL_SOFT/(now-last_time)*100);
+        write_err(t);
+        
+        if (CPUTIME_INTERVAL_SOFT >= ((now-last_time)*(double)CPUTIME_PERCENT/100)) {
+            static const char c = 'X';
+
+            write_err("Soft CPU time limit exhausted, terminating.\n");
+            
+            /* Try a soft cleanup */
+            write(the_pipe[1], &c, sizeof(c));
+            phase = PHASE_SOFT;
+            reset_cpu_time(CPUTIME_INTERVAL_HARD);
+            
+        } else {
+
+            /* Everything's fine */
+            reset_cpu_time(CPUTIME_INTERVAL_SOFT);
+            last_time = now;
+        }
+        
+    } else if (phase == PHASE_SOFT) {
+        write_err("Hard CPU time limit exhausted, terminating forcibly.\n");
+        _exit(1);
+    }
+}
+
+static void callback(struct pa_mainloop_api*m, struct pa_io_event*e, int fd, enum pa_io_event_flags f, void *userdata) {
+    char c;
+    assert(m && e && f == PA_IO_EVENT_INPUT && e == io_event && fd == the_pipe[0]);
+    read(the_pipe[0], &c, sizeof(c));
+    m->quit(m, 1);
+}
+
+int pa_cpu_limit_init(struct pa_mainloop_api *m) {
+    int r;
+    struct sigaction sa;
+    assert(m && !api && !io_event && the_pipe[0] == -1 && the_pipe[1] == -1);
+    
+    time(&last_time);
+
+    if (pipe(the_pipe) < 0) {
+        fprintf(stderr, "pipe() failed: %s\n", strerror(errno));
+        return -1;
+    }
+
+    pa_make_nonblock_fd(the_pipe[0]);
+    pa_make_nonblock_fd(the_pipe[1]);
+    pa_fd_set_cloexec(the_pipe[0], 1);
+    pa_fd_set_cloexec(the_pipe[1], 1);
+
+    api = m;
+    io_event = api->io_new(m, the_pipe[0], PA_IO_EVENT_INPUT, callback, NULL);
+
+    phase = PHASE_IDLE;
+
+    memset(&sa, 0, sizeof(sa));
+    sa.sa_handler = signal_handler;
+    sigemptyset(&sa.sa_mask);
+    sa.sa_flags = SA_RESTART;
+    
+    r = sigaction(SIGXCPU, &sa, &sigaction_prev);
+    assert(r >= 0);
+
+    installed = 1;
+
+    reset_cpu_time(CPUTIME_INTERVAL_SOFT);
+    
+    return 0;
+}
+
+void pa_cpu_limit_done(void) {
+    int r;
+
+    if (io_event) {
+        assert(api);
+        api->io_free(io_event);
+        io_event = NULL;
+        api = NULL;
+    }
+
+    if (the_pipe[0] >= 0)
+        close(the_pipe[0]);
+    if (the_pipe[1] >= 0)
+        close(the_pipe[1]);
+    the_pipe[0] = the_pipe[1] = -1;
+
+    if (installed) {
+        r = sigaction(SIGXCPU, &sigaction_prev, NULL);
+        assert(r >= 0);
+        installed = 0;
+    }
+}