1 /*
2  * Out Of Memory (OOM) for MEMCG and CPUSET
3  *
4  * The program is designed to cope with unpredictable like amount and
5  * system physical memory, swap size and other VMM technology like KSM,
6  * memcg, memory hotplug and so on which may affect the OOM
7  * behaviours. It simply increase the memory consumption 3G each time
8  * until all the available memory is consumed and OOM is triggered.
9  *
10  * Copyright (C) 2013-2017  Red Hat, Inc.
11  *
12  * This program is free software;  you can redistribute it and/or modify
13  * it under the terms of the GNU General Public License as published by
14  * the Free Software Foundation; either version 2 of the License, or
15  * (at your option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
20  * the GNU General Public License for more details.
21  */
22 
23 #include "config.h"
24 #include <sys/types.h>
25 #include <sys/stat.h>
26 #include <errno.h>
27 #include <fcntl.h>
28 #include <stdio.h>
29 #if HAVE_NUMA_H
30 #include <numa.h>
31 #endif
32 
33 #include "numa_helper.h"
34 #include "mem.h"
35 
36 #ifdef HAVE_NUMA_V2
37 
verify_oom(void)38 static void verify_oom(void)
39 {
40 	int swap_acc_on = 1;
41 
42 #if __WORDSIZE == 32
43 	tst_brk(TCONF, "test is not designed for 32-bit system.");
44 #endif
45 
46 	tst_res(TINFO, "OOM on CPUSET & MEMCG...");
47 	testoom(0, 0, ENOMEM, 1);
48 
49 	/*
50 	 * Under NUMA system, the migration of cpuset's memory
51 	 * is in charge of cpuset.memory_migrate, we can write
52 	 * 1 to cpuset.memory_migrate to enable the migration.
53 	 */
54 	if (is_numa(NULL, NH_MEMS, 2)) {
55 		write_cpuset_files(CPATH_NEW, "memory_migrate", "1");
56 		tst_res(TINFO, "OOM on CPUSET & MEMCG with "
57 				"cpuset.memory_migrate=1");
58 		testoom(0, 0, ENOMEM, 1);
59 	}
60 
61 	if (access(MEMCG_SW_LIMIT, F_OK) == -1) {
62 		if (errno == ENOENT) {
63 			tst_res(TCONF, "memcg swap accounting is disabled");
64 			swap_acc_on = 0;
65 		} else
66 			tst_brk(TBROK|TERRNO, "access");
67 	}
68 
69 	if (swap_acc_on) {
70 		tst_res(TINFO, "OOM on CPUSET & MEMCG with "
71 				"special memswap limitation:");
72 		SAFE_FILE_PRINTF(MEMCG_SW_LIMIT, "%ld", TESTMEM);
73 		testoom(0, 0, ENOMEM, 1);
74 
75 		tst_res(TINFO, "OOM on CPUSET & MEMCG with "
76 				"disabled memswap limitation:");
77 		SAFE_FILE_PRINTF(MEMCG_SW_LIMIT, "-1");
78 		testoom(0, 0, ENOMEM, 1);
79 	}
80 }
81 
setup(void)82 void setup(void)
83 {
84 	int ret, memnode;
85 
86 	if (!is_numa(NULL, NH_MEMS, 1))
87 		tst_brk(TCONF, "requires NUMA with at least 1 node");
88 
89 	overcommit = get_sys_tune("overcommit_memory");
90 	set_sys_tune("overcommit_memory", 1, 1);
91 
92 	mount_mem("memcg", "cgroup", "memory", MEMCG_PATH, MEMCG_PATH_NEW);
93 	mount_mem("cpuset", "cpuset", NULL, CPATH, CPATH_NEW);
94 	write_memcg();
95 
96 	/*
97 	 * Some nodes do not contain memory, so use
98 	 * get_allowed_nodes(NH_MEMS) to get a memory
99 	 * node. This operation also applies to Non-NUMA
100 	 * systems.
101 	 */
102 	ret = get_allowed_nodes(NH_MEMS, 1, &memnode);
103 	if (ret < 0)
104 		tst_brk(TBROK, "Failed to get a memory node "
105 			      "using get_allowed_nodes()");
106 	write_cpusets(memnode);
107 }
108 
cleanup(void)109 void cleanup(void)
110 {
111 	set_sys_tune("overcommit_memory", overcommit, 0);
112 	umount_mem(CPATH, CPATH_NEW);
113 	umount_mem(MEMCG_PATH, MEMCG_PATH_NEW);
114 }
115 
116 static struct tst_test test = {
117 	.needs_root = 1,
118 	.forks_child = 1,
119 	.timeout = -1,
120 	.setup = setup,
121 	.cleanup = cleanup,
122 	.test_all = verify_oom,
123 };
124 
125 #else
126 	TST_TEST_TCONF("test requires libnuma >= 2 and it's development packages");
127 #endif
128