diff options
author | FRIGN <dev@frign.de> | 2016-02-14 01:28:37 +0100 |
---|---|---|
committer | FRIGN <dev@frign.de> | 2016-02-14 01:28:37 +0100 |
commit | 3abbffa4934a62146e995ee7c2cf3ba50991b4ad (patch) | |
tree | 3a97c28b0c83a8ecc93b84c0487dc6c92089bc4b | |
parent | 6a52a85a1ac87b29048863d599a8b0f0ee712482 (diff) |
Simplify the oom-taming-function
There really is no need to source a defined variable from a linux
header. The OOM-rank ranges from -1000 to 1000, so we can safely
hardcode -1000, which is a sane thing to do given slock is suid and
we don't want to play around too much here anyway.
On another notice, let's not forget that this still is a shitty
heuristic. The OOM-killer still can kill us (thus I also changed
the wording in the error-message. We do not disable the OOM-killer,
we're just hiding.
-rw-r--r-- | slock.c | 20 |
1 files changed, 6 insertions, 14 deletions
@@ -60,28 +60,20 @@ die(const char *errstr, ...) | |||
60 | 60 | ||
61 | #ifdef __linux__ | 61 | #ifdef __linux__ |
62 | #include <fcntl.h> | 62 | #include <fcntl.h> |
63 | #include <linux/oom.h> | ||
64 | 63 | ||
65 | static void | 64 | static void |
66 | dontkillme(void) | 65 | dontkillme(void) |
67 | { | 66 | { |
68 | int fd; | 67 | int fd; |
69 | int length; | ||
70 | char value[64]; | ||
71 | 68 | ||
72 | fd = open("/proc/self/oom_score_adj", O_WRONLY); | 69 | fd = open("/proc/self/oom_score_adj", O_WRONLY); |
73 | if (fd < 0 && errno == ENOENT) | 70 | if (fd < 0 && errno == ENOENT) { |
74 | return; | 71 | return; |
75 | 72 | } | |
76 | /* convert OOM_SCORE_ADJ_MIN to string for writing */ | 73 | if (fd < 0 || write(fd, "-1000\n", (sizeof("-1000\n") - 1)) != |
77 | length = snprintf(value, sizeof(value), "%d\n", OOM_SCORE_ADJ_MIN); | 74 | (sizeof("-1000\n") - 1) || close(fd) != 0) { |
78 | 75 | die("can't tame the oom-killer. is suid or sgid set?\n"); | |
79 | /* bail on truncation */ | 76 | } |
80 | if (length >= sizeof(value)) | ||
81 | die("buffer too small\n"); | ||
82 | |||
83 | if (fd < 0 || write(fd, value, length) != length || close(fd) != 0) | ||
84 | die("cannot disable the out-of-memory killer for this process (make sure to suid or sgid slock)\n"); | ||
85 | } | 77 | } |
86 | #endif | 78 | #endif |
87 | 79 | ||