Basic tests were to start several programs with same and different memory contents and ensure only to merge the ones with the same contents. When changed the content of one of merged pages in a process and to the mode "unmerging", it should discard all merged pages there. Also tested it is possible to disable KSM. There are also command-line options to specify the memory allocation size, and number of processes have same memory contents so it is possible to test more advanced things like KSM + OOM etc.
Signed-off-by: CAI Qian <[email protected]> --- testcases/kernel/mem/ksm/Makefile | 22 ++ testcases/kernel/mem/ksm/ksm01.c | 428 +++++++++++++++++++++++++++++++++++++ 2 files changed, 450 insertions(+), 0 deletions(-) create mode 100644 testcases/kernel/mem/ksm/Makefile create mode 100644 testcases/kernel/mem/ksm/ksm01.c diff --git a/testcases/kernel/mem/ksm/Makefile b/testcases/kernel/mem/ksm/Makefile new file mode 100644 index 0000000..d3eb6ef --- /dev/null +++ b/testcases/kernel/mem/ksm/Makefile @@ -0,0 +1,22 @@ +# +# Copyright (c) International Business Machines Corp., 2001 +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See +# the GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +# + +top_srcdir ?= ../../../.. + +include $(top_srcdir)/include/mk/testcases.mk +include $(top_srcdir)/include/mk/generic_leaf_target.mk diff --git a/testcases/kernel/mem/ksm/ksm01.c b/testcases/kernel/mem/ksm/ksm01.c new file mode 100644 index 0000000..225f3b1 --- /dev/null +++ b/testcases/kernel/mem/ksm/ksm01.c @@ -0,0 +1,428 @@ +/* + * functional testing for Kernel Samepage Merging (KSM) + * + * Basic tests were to start several programs with same and different + * memory contents and ensure only to merge the ones with the same + * contents. When changed the content of one of merged pages in a + * process and to the mode "unmerging", it should discard all merged + * pages there. Also tested it is possible to disable KSM. There are + * also command-line options to specify the memory allocation size, and + * number of processes have same memory contents so it is possible to + * test more advanced things like KSM + OOM etc. + * + * Prerequisites: + * + * 1) ksm and ksmtuned daemons need to be disabled. Otherwise, it could + * distrub the testing as they also change some ksm tunables depends + * on current workloads. + * + * The test steps are: + * 01) check ksm feature and backup current run setting. + * 02) change run setting to 1 - merging. + * 03) 3 memory allocation programs have the memory contents that 2 of + * them are all 'a' and one is all 'b'. + * 04) check ksm statistics. + * 05) 1 program changes the memory content from all 'a' to all 'b'. + * 06) check ksm statistics. + * 07) all programs change the memory content to all 'd'. + * 08) check ksm statistics. + * 09) change run setting to 2 - unmerging. + * 10) check ksm statistics. + * 11) change run setting to 0 - stop. + * + * 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/stat.h> +#include <sys/mman.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <signal.h> +#include <stdlib.h> +#include <unistd.h> +#include <stdio.h> +#include <errno.h> +#include "test.h" +#include "usctest.h" + +#define _PATH_KSM "/sys/kernel/mm/ksm/" + +char *TCID = "ksm01"; +int TST_TOTAL = 1; +extern int Tst_count; +static int opt_num, opt_size; +static char *opt_numstr, *opt_sizestr; + +/* main setup function of test */ +void setup(void); +/* cleanup function for the test */ +void cleanup(void); +int ksmtest(void); +void help(void); +void check(char *path, int value); +void check_range(char *path, int value); + +static option_t options[] = { + { "n:", &opt_num, &opt_numstr}, + { "s:", &opt_size, &opt_sizestr}, + { NULL, NULL, NULL} +}; + +int main(int argc, char *argv[]) +{ + /* loop counter */ + int lc; + /* message returned from parse_opts */ + char *msg; + + msg = parse_opts(argc, argv, options, help); + if (msg != (char *)NULL) { + tst_brkm(TBROK, NULL, "OPTION PARSING ERROR - %s", msg); + tst_exit(); + } + /* Perform global setup for test */ + setup(); + + /* Check looping state if -i option given */ + for (lc = 0; TEST_LOOPING(lc); lc++) { + /* Reset Tst_count in case we are looping. */ + Tst_count = 0; + + TEST(ksmtest()); + + if (TEST_RETURN != 0) + tst_resm(TFAIL, "ksmtest() failed with %ld.", + TEST_RETURN); + else + tst_resm(TPASS, "ksmtest() completed " + "successfully."); + } + cleanup(); + return 0; +} + +int ksmtest(void) +{ + char *m = NULL, buf[BUFSIZ], buf2[BUFSIZ]; + int i, j, status, k, fd, *child; + int size = 128; + int num = 3; + + if (opt_size) { + size = atoi(opt_sizestr); + if (size < 1) + tst_brkm(TBROK, cleanup, + "size cannot be less than 1."); + } + if (opt_num) { + num = atoi(opt_numstr); + if (num < 3) + tst_brkm(TBROK, cleanup, + "process number cannot be less 3."); + } + child = malloc(num); + if (child == NULL) + tst_brkm(TBROK, cleanup, "malloc"); + + if ((child[0] = fork()) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "fork"); + else if (child[0] == 0) { + /* Avoid THP allocation which can't ksm ATM. */ + tst_resm(TINFO, + "child 0 allocates %d MB filled with 'c'.", + size); + for (j = 0; j < size; j++) { + m = malloc(1024 * 1024); + if (m == NULL) + tst_brkm(TBROK|TERRNO, cleanup, + "malloc"); + madvise((char *)(((long)m>>12)<<12), + 1024 * 1024, MADV_MERGEABLE); + for (i = 0; i < 1024 * 1024; i++) + *m = 'a'; + } + if (raise(SIGSTOP) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "kill"); + + tst_resm(TINFO, "child 0 continues..."); + tst_resm(TINFO, "child 0 changes memory content to 'd'."); + + for (j = 0; j < 128; j++) { + for (i = 0; i < 1024 * 1024; i++) + *m = 'd'; + } + exit(1); + } + if ((child[1] = fork()) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "fork"); + else if (child[1] == 0) { + tst_resm(TINFO, + "child 1 allocates %d MB filled with 'a'.", + size); + for (j = 0; j < size; j++) { + m = malloc(1024 * 1024); + if (m == NULL) + tst_brkm(TBROK|TERRNO, cleanup, + "malloc"); + madvise((char *)(((long)m>>12)<<12), + 1024 * 1024, MADV_MERGEABLE); + for (i = 0; i < 1024 * 1024; i++) + *m = 'a'; + } + if (raise(SIGSTOP) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "kill"); + + tst_resm(TINFO, "child 1 continues..."); + tst_resm(TINFO, "child 1 changes memory content to 'b'."); + + for (j = 0; j < size; j++) { + for (i = 0; i < 1024 * 1024; i++) + *m = 'b'; + } + if (raise(SIGSTOP) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "kill"); + + tst_resm(TINFO, "child 1 continues..."); + tst_resm(TINFO, "child 1 changes memory content to 'd'"); + + for (j = 0; j < size; j++) { + for (i = 0; i < 1024 * 1024; i++) + *m = 'd'; + } + exit(1); + } + + for (k = 2; k < num; k++) { + if ((child[k] = fork()) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "fork"); + else if (child[k] == 0) { + tst_resm(TINFO, + "child %d allocates %d MB filled with " + "'a'.", k, size); + for (j = 0; j < size; j++) { + m = malloc(1024 * 1024); + if (m == NULL) + tst_brkm(TBROK|TERRNO, cleanup, + "malloc"); + madvise((char *)(((long)m>>12)<<12), + 1024 * 1024, MADV_MERGEABLE); + for (i = 0; i < 1024 * 1024; i++) + *m = 'a'; + } + if (raise(SIGSTOP) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "kill"); + + tst_resm(TINFO, "child %d continues...", k); + tst_resm(TINFO, "child %d changes memory " + "content to 'd'", k); + + for (j = 0; j < size; j++) { + for (i = 0; i < 1024 * 1024; i++) + *m = 'd'; + } + exit(1); + } + } + tst_resm(TINFO, "KSM merging..."); + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "run"); + fd = open(buf, 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); + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "pages_to_scan"); + snprintf(buf2, BUFSIZ, "%d", size * 256 * num); + fd = open(buf, O_WRONLY); + if (fd == -1) + tst_brkm(TBROK|TERRNO, cleanup, "open"); + if (write(fd, buf2, strlen(buf2)) != strlen(buf2)) + tst_brkm(TBROK|TERRNO, cleanup, "write"); + close(fd); + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "sleep_millisecs"); + fd = open(buf, O_WRONLY); + if (fd == -1) + tst_brkm(TBROK|TERRNO, cleanup, "open"); + if (write(fd, "0", 1) != 1) + tst_brkm(TBROK|TERRNO, cleanup, "write"); + close(fd); + + sleep(5); + tst_resm(TINFO, "check!"); + check("run", 1); + check("pages_shared", num - 1); + /* It is magic! */ + check_range("pages_sharing", 0); + check("pages_unshared", 0); + + for (k = 0; k < num; k++) { + if (waitpid(child[k], &status, WUNTRACED) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "waitpid"); + if (!WIFSTOPPED(status)) + tst_brkm(TBROK, cleanup, + "child was not stopped."); + } + if (kill(child[1], SIGCONT) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "kill"); + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "run"); + fd = open(buf, 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); + + sleep(5); + tst_resm(TINFO, "check!"); + check("run", 1); + check("pages_shared", num - 1); + check_range("pages_sharing", 0); + check("pages_unshared", 1); + + for (k = 0; k < num; k++) { + if (kill(child[k], SIGCONT) == -1) + tst_brkm(TBROK|TERRNO, cleanup, + "kill child[%d]", k); + } + sleep(5); + tst_resm(TINFO, "check!"); + check("run", 1); + check("pages_shared", num); + check_range("pages_sharing", 0); + check("pages_unshared", 0); + + tst_resm(TINFO, "KSM unmerging..."); + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "run"); + fd = open(buf, O_WRONLY); + if (fd == -1) + tst_brkm(TBROK|TERRNO, cleanup, "open"); + if (write(fd, "2", 1) != 1) + tst_brkm(TBROK|TERRNO, cleanup, "write"); + + sleep(5); + tst_resm(TINFO, "check!"); + check("run", 2); + check("pages_shared", 0); + check("pages_sharing", 0); + check("pages_unshared", 0); + + tst_resm(TINFO, "stop KSM."); + if (lseek(fd, 0, SEEK_SET) == -1) + tst_brkm(TBROK|TERRNO, cleanup, "lseek"); + if (write(fd, "0", 1) != 1) + tst_brkm(TBROK|TERRNO, cleanup, "write"); + close(fd); + + sleep(5); + tst_resm(TINFO, "check!"); + check("run", 0); + check("pages_shared", 0); + check("pages_sharing", 0); + check("pages_unshared", 0); + + while (waitpid(-1, &status, WUNTRACED | WCONTINUED) > 0) { + if (WEXITSTATUS(status) != 0) + return 1; + } + return 0; +} + +void cleanup(void) +{ + /* + * remove the tmp directory and exit + */ + TEST_CLEANUP; + tst_rmdir(); + tst_exit(); +} + +void setup(void) +{ + char buf[BUFSIZ]; + struct stat new; + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, "run"); + if (stat(buf, &new) == -1) { + if (errno == ENOENT) + tst_brkm(TCONF, NULL, "no KSM."); + else + tst_brkm(TBROK, NULL, "stat"); + } + + /* + * setup a default signal hander and a + * temporary working directory. + */ + tst_sig(FORK, DEF_HANDLER, cleanup); + TEST_PAUSE; + tst_tmpdir(); +} + +void help(void) +{ + printf(" -n Number of process has same " + "memory content\n"); + printf(" -s Memory allocation size in MB\n"); +} + +void check(char *path, int value) +{ + FILE *fp; + char buf[BUFSIZ]; + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, path); + fp = fopen(buf, "r"); + if (fp == NULL) + tst_brkm(TBROK|TERRNO, cleanup, "fopen"); + if (fgets(buf, BUFSIZ, fp) == NULL) + tst_brkm(TBROK|TERRNO, cleanup, "fgets"); + fclose(fp); + + if (atol(buf) != value) + tst_resm(TFAIL, "%s is not %d but %ld", path, value, + atol(buf)); +} + +void check_range(char *path, int value) +{ + FILE *fp; + char buf[BUFSIZ]; + + snprintf(buf, BUFSIZ, "%s%s", _PATH_KSM, path); + fp = fopen(buf, "r"); + if (fp == NULL) + tst_brkm(TBROK|TERRNO, cleanup, "fopen"); + if (fgets(buf, BUFSIZ, fp) == NULL) + tst_brkm(TBROK|TERRNO, cleanup, "fgets"); + fclose(fp); + + if (atol(buf) <= value) + tst_resm(TFAIL, "%s is not greater than %d but %ld", + path, value, atol(buf)); +} -- 1.7.1 ------------------------------------------------------------------------------ Download new Adobe(R) Flash(R) Builder(TM) 4 The new Adobe(R) Flex(R) 4 and Flash(R) Builder(TM) 4 (formerly Flex(R) Builder(TM)) enable the development of rich applications that run across multiple browsers and platforms. Download your free trials today! http://p.sf.net/sfu/adobe-dev2dev _______________________________________________ Ltp-list mailing list [email protected] https://lists.sourceforge.net/lists/listinfo/ltp-list
