[CRIU] [PATCH v4 4/5] zdtm: add test for oom_score_adj restore

[Dmitry Safonov]

Signed-off-by: Dmitry Safonov <dsafonov at odin.com>
---
 test/zdtm/live/static/Makefile        |  1 +
 test/zdtm/live/static/oom_score_adj.c | 94 +++++++++++++++++++++++++++++++++++
 2 files changed, 95 insertions(+)
 create mode 100644 test/zdtm/live/static/oom_score_adj.c

diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index a6995c6..6067746 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -144,6 +144,7 @@ TST_NOFILE	=				\
 		vsx				\
 		bridge				\
 		vfork00				\
+		oom_score_adj			\
 #		jobctl00			\
 
 TST_FILE	=				\
diff --git a/test/zdtm/live/static/oom_score_adj.c b/test/zdtm/live/static/oom_score_adj.c
new file mode 100644
index 0000000..3135885
--- /dev/null
+++ b/test/zdtm/live/static/oom_score_adj.c
@@ -0,0 +1,94 @@
+#include <errno.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include <sys/types.h>
+#include <sys/stat.h>
+
+#include "zdtmtst.h"
+
+const char *test_doc = "Check for /proc/self/oom_score_adj restore";
+const char *test_author = "Dmitry Safonov <dsafonov at odin.com>";
+
+const char oom_score_adj_self[] = "/proc/self/oom_score_adj";
+const int test_value = -400;
+
+int get_oom_score_adj(const char *path, int *err)
+{
+	int fd;
+	ssize_t num;
+	char buf[11];
+
+	*err = 0;
+	fd = open(path, O_RDONLY);
+	if (fd < 0) {
+		pr_perror("Failed to open %s", path);
+		goto out;
+	}
+
+	num = read(fd, buf, 10);
+	close(fd);
+	if (num < 0) {
+		pr_perror("Unable to read %s", path);
+		goto out;
+	}
+	buf[num] = '\0';
+
+	return strtol(buf, NULL, 10);
+
+out:
+	*err = -1;
+	return 0;
+}
+
+int set_oom_score_adj(const char *path, int value)
+{
+	int fd, ret = 0;
+	char buf[11];
+
+	fd = open(path, O_RDWR);
+	if (fd < 0) {
+		pr_perror("Failed to open %s", path);
+		return -1;
+	}
+
+	snprintf(buf, 11, "%d", value);
+
+	if (write(fd, buf, 11) < 0) {
+		pr_perror("Write %s to %s failed", buf, path);
+		ret = -1;
+	}
+
+	close(fd);
+	return ret;
+}
+
+
+int main(int argc, char *argv[])
+{
+	int ret;
+	int new_oom_score_adj;
+
+	test_init(argc, argv);
+
+	if (set_oom_score_adj(oom_score_adj_self, test_value) < 0)
+		return -1;
+
+	test_daemon();
+	test_waitsig();
+
+	new_oom_score_adj = get_oom_score_adj(oom_score_adj_self, &ret);
+	if (ret < 0)
+		return -1;
+
+	if (new_oom_score_adj != test_value) {
+		fail("OOM score value %d is different after restore: %d\n",
+				test_value, new_oom_score_adj);
+		return -1;
+	}
+
+	pass();
+	return 0;
+}
-- 
2.6.3