blob: 9f40f39be19876a40736cf4dc0ab04079f4e7aac (
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
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// The following is the C version of code from base/process_utils_linux.cc.
// We shouldn't link against C++ code in a setuid binary.
#include "process_util.h"
#include <fcntl.h>
#include <inttypes.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
bool AdjustOOMScore(pid_t process, int score) {
if (score < 0 || score > 15)
return false;
char oom_adj[35]; // "/proc/" + log_10(2**64) + "/oom_adj\0"
// 6 + 20 + 9 = 35
snprintf(oom_adj, sizeof(oom_adj), "/proc/%" PRIdMAX, (intmax_t)process);
struct stat statbuf;
if (stat(oom_adj, &statbuf) < 0)
return false;
if (getuid() != statbuf.st_uid)
return false;
strcat(oom_adj, "/oom_adj");
int fd = open(oom_adj, O_WRONLY);
if (fd < 0)
return false;
char buf[3]; // 0 <= |score| <= 15;
snprintf(buf, sizeof(buf), "%d", score);
size_t len = strlen(buf);
ssize_t bytes_written = write(fd, buf, len);
close(fd);
return (bytes_written == len);
}
|