| /* |
| * Out Of Memory (OOM) for Memory Resource Controller |
| * |
| * The program is designed to cope with unpredictable like amount and |
| * system physical memory, swap size and other VMM technology like KSM, |
| * memcg, memory hotplug and so on which may affect the OOM |
| * behaviours. It simply increase the memory consumption 3G each time |
| * until all the available memory is consumed and OOM is triggered. |
| * |
| * Copyright (C) 2010 Red Hat, Inc. |
| * This program is free software; you can redistribute it and/or |
| * modify it under the terms of version 2 of the GNU General Public |
| * License as published by the Free Software Foundation. |
| * |
| * This program is distributed in the hope that it would be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. |
| * |
| * Further, this software is distributed without any warranty that it |
| * is free of the rightful claim of any third person regarding |
| * infringement or the like. Any license provided herein, whether |
| * implied or otherwise, applies only to this software file. Patent |
| * licenses, if any, provided herein do not apply to combinations of |
| * this program with other software, or any other product whatsoever. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
| * 02110-1301, USA. |
| */ |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <fcntl.h> |
| #include <stdio.h> |
| #include <errno.h> |
| #include "test.h" |
| #include "usctest.h" |
| #include "lib/oom.h" |
| |
| char *TCID = "oom03"; |
| int TST_TOTAL = 1; |
| extern int Tst_count; |
| |
| static void setup(void); |
| |
| int main(int argc, char *argv[]) |
| { |
| char *msg; |
| int lc, fd; |
| char buf[BUFSIZ], mem[BUFSIZ]; |
| |
| if ((msg = parse_opts(argc, argv, NULL, NULL)) != NULL) |
| tst_brkm(TBROK, NULL, "OPTION PARSING ERROR - %s", msg); |
| |
| #ifdef __WORDSIZE == 32 |
| tst_brkm(TCONF, NULL, "test is not designed for 32-bit system."); |
| #endif |
| |
| setup(); |
| |
| for (lc = 0; TEST_LOOPING(lc); lc++) { |
| Tst_count = 0; |
| fd = open(SYSFS_OVER, O_WRONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, cleanup, "open"); |
| if (write(fd, "1", 1) != 1) |
| tst_brkm(TBROK|TERRNO, cleanup, "write"); |
| close(fd); |
| |
| fd = open(MEMCG_PATH_NEW "/memory.limit_in_bytes", O_WRONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, cleanup, "open %s", buf); |
| sprintf(mem, "%ld", TESTMEM); |
| if (write(fd, mem, strlen(mem)) != strlen(mem)) |
| tst_brkm(TBROK|TERRNO, cleanup, "write %s", buf); |
| close(fd); |
| |
| fd = open(MEMCG_PATH_NEW "/tasks", O_WRONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, cleanup, "open %s", buf); |
| snprintf(buf, BUFSIZ, "%d", getpid()); |
| if (write(fd, buf, strlen(buf)) != strlen(buf)) |
| tst_brkm(TBROK|TERRNO, cleanup, "write %s", buf); |
| close(fd); |
| testoom(0, 0, 0); |
| |
| fd = open(MEMCG_PATH_NEW "/memory.memsw.limit_in_bytes", |
| O_WRONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, cleanup, "open %s", buf); |
| if (write(fd, mem, strlen(mem)) != strlen(mem)) |
| tst_brkm(TBROK|TERRNO, cleanup, "write %s", buf); |
| close(fd); |
| testoom(0, 1, 0); |
| } |
| cleanup(); |
| tst_exit(); |
| } |
| |
| void setup(void) |
| { |
| int fd; |
| |
| tst_require_root(NULL); |
| |
| tst_sig(FORK, DEF_HANDLER, cleanup); |
| TEST_PAUSE; |
| |
| fd = open(SYSFS_OVER, O_RDONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, NULL, "open"); |
| if (read(fd, &overcommit, 1) != 1) |
| tst_brkm(TBROK|TERRNO, NULL, "read"); |
| close(fd); |
| |
| mount_mem("memcg", "cgroup", "memory", MEMCG_PATH, MEMCG_PATH_NEW); |
| } |
| |
| void cleanup(void) |
| { |
| int fd; |
| |
| fd = open(SYSFS_OVER, O_WRONLY); |
| if (fd == -1) |
| tst_brkm(TBROK|TERRNO, cleanup, "open"); |
| if (write(fd, &overcommit, 1) != 1) |
| tst_brkm(TBROK|TERRNO, cleanup, "write"); |
| close(fd); |
| |
| umount_mem(MEMCG_PATH, MEMCG_PATH_NEW); |
| |
| TEST_CLEANUP; |
| } |