1 /*
2 * Out Of Memory (OOM) for 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) 2010 Red Hat, Inc.
11 * This program is free software; you can redistribute it and/or
12 * modify it under the terms of version 2 of the GNU General Public
13 * License as published by the Free Software Foundation.
14 *
15 * This program is distributed in the hope that it would be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
18 *
19 * Further, this software is distributed without any warranty that it
20 * is free of the rightful claim of any third person regarding
21 * infringement or the like. Any license provided herein, whether
22 * implied or otherwise, applies only to this software file. Patent
23 * licenses, if any, provided herein do not apply to combinations of
24 * this program with other software, or any other product whatsoever.
25 *
26 * You should have received a copy of the GNU General Public License
27 * along with this program; if not, write the Free Software
28 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
29 * 02110-1301, USA.
30 */
31
32 #include "config.h"
33 #include <sys/types.h>
34 #include <sys/stat.h>
35 #include <errno.h>
36 #include <fcntl.h>
37 #include <stdio.h>
38 #include "numa_helper.h"
39 #include "test.h"
40 #include "mem.h"
41
42 char *TCID = "oom04";
43 int TST_TOTAL = 1;
44
45 #if HAVE_NUMA_H && HAVE_LINUX_MEMPOLICY_H && HAVE_NUMAIF_H \
46 && HAVE_MPOL_CONSTANTS
47
main(int argc,char * argv[])48 int main(int argc, char *argv[])
49 {
50 int lc;
51
52 tst_parse_opts(argc, argv, NULL, NULL);
53
54 #if __WORDSIZE == 32
55 tst_brkm(TCONF, NULL, "test is not designed for 32-bit system.");
56 #endif
57
58 setup();
59
60 for (lc = 0; TEST_LOOPING(lc); lc++) {
61 tst_count = 0;
62
63 tst_resm(TINFO, "OOM on CPUSET...");
64 testoom(0, 0, ENOMEM, 1);
65
66 if (is_numa(cleanup, NH_MEMS, 2)) {
67 /*
68 * Under NUMA system, the migration of cpuset's memory
69 * is in charge of cpuset.memory_migrate, we can write
70 * 1 to cpuset.memory_migrate to enable the migration.
71 */
72 write_cpuset_files(CPATH_NEW,
73 "memory_migrate", "1");
74 tst_resm(TINFO, "OOM on CPUSET with mem migrate:");
75 testoom(0, 0, ENOMEM, 1);
76 }
77 }
78 cleanup();
79 tst_exit();
80 }
81
setup(void)82 void setup(void)
83 {
84 int memnode, ret;
85
86 tst_require_root();
87 tst_sig(FORK, DEF_HANDLER, cleanup);
88 TEST_PAUSE;
89
90 if (!is_numa(NULL, NH_MEMS, 1))
91 tst_brkm(TCONF, NULL, "requires NUMA with at least 1 node");
92
93 overcommit = get_sys_tune("overcommit_memory");
94 set_sys_tune("overcommit_memory", 1, 1);
95
96 mount_mem("cpuset", "cpuset", NULL, CPATH, CPATH_NEW);
97
98 /*
99 * Some nodes do not contain memory, so use
100 * get_allowed_nodes(NH_MEMS) to get a memory
101 * node. This operation also applies to Non-NUMA
102 * systems.
103 */
104 ret = get_allowed_nodes(NH_MEMS, 1, &memnode);
105 if (ret < 0)
106 tst_brkm(TBROK, cleanup, "Failed to get a memory node "
107 "using get_allowed_nodes()");
108 write_cpusets(memnode);
109 }
110
cleanup(void)111 void cleanup(void)
112 {
113 set_sys_tune("overcommit_memory", overcommit, 0);
114 umount_mem(CPATH, CPATH_NEW);
115 }
116
117 #else /* no NUMA */
main(void)118 int main(void)
119 {
120 tst_brkm(TCONF, NULL, "no NUMA development packages installed.");
121 }
122 #endif
123