1 /*
2  * Copyright (C) 2010-2017  Red Hat, Inc.
3  *
4  * This program is free software;  you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
12  * the GNU General Public License for more details.
13  *
14  * Kernel Samepage Merging (KSM)
15  *
16  * Basic tests were to start several programs with same and different
17  * memory contents and ensure only to merge the ones with the same
18  * contents. When changed the content of one of merged pages in a
19  * process and to the mode "unmerging", it should discard all merged
20  * pages there. Also tested it is possible to disable KSM. There are
21  * also command-line options to specify the memory allocation size, and
22  * number of processes have same memory contents so it is possible to
23  * test more advanced things like KSM + OOM etc.
24  *
25  * Prerequisites:
26  *
27  * 1) ksm and ksmtuned daemons need to be disabled. Otherwise, it could
28  *    distrub the testing as they also change some ksm tunables depends
29  *    on current workloads.
30  *
31  * The test steps are:
32  * - Check ksm feature and backup current run setting.
33  * - Change run setting to 1 - merging.
34  * - 3 memory allocation programs have the memory contents that 2 of
35  *   them are all 'a' and one is all 'b'.
36  * - Check ksm statistics and verify the content.
37  * - 1 program changes the memory content from all 'a' to all 'b'.
38  * - Check ksm statistics and verify the content.
39  * - All programs change the memory content to all 'd'.
40  * - Check ksm statistics and verify the content.
41  * - Change one page of a process.
42  * - Check ksm statistics and verify the content.
43  * - Change run setting to 2 - unmerging.
44  * - Check ksm statistics and verify the content.
45  * - Change run setting to 0 - stop.
46  */
47 
48 #include "config.h"
49 #include <sys/types.h>
50 #include <sys/stat.h>
51 #include <errno.h>
52 #include <fcntl.h>
53 #if HAVE_NUMA_H
54 #include <numa.h>
55 #endif
56 #if HAVE_NUMAIF_H
57 #include <numaif.h>
58 #endif
59 #include <signal.h>
60 #include <stdio.h>
61 #include <unistd.h>
62 #include "mem.h"
63 #include "ksm_common.h"
64 
65 #if defined(HAVE_NUMA_V2) && defined(HAVE_LINUX_MEMPOLICY_H)
66 
verify_ksm(void)67 static void verify_ksm(void)
68 {
69 	unsigned long nmask[MAXNODES / BITS_PER_LONG] = { 0 };
70 	unsigned int node;
71 
72 	node = get_a_numa_node();
73 	set_node(nmask, node);
74 
75 	if (set_mempolicy(MPOL_BIND, nmask, MAXNODES) == -1) {
76 		if (errno != ENOSYS)
77 			tst_brk(TBROK | TERRNO, "set_mempolicy");
78 		else
79 			tst_brk(TCONF, "set_mempolicy syscall is not "
80 				 "implemented on your system.");
81 	}
82 	create_same_memory(size, num, unit);
83 
84 	write_cpusets(node);
85 	create_same_memory(size, num, unit);
86 }
87 
cleanup(void)88 static void cleanup(void)
89 {
90 	if (access(PATH_KSM "merge_across_nodes", F_OK) == 0)
91 		FILE_PRINTF(PATH_KSM "merge_across_nodes",
92 				 "%d", merge_across_nodes);
93 
94 	restore_max_page_sharing();
95 
96 	umount_mem(CPATH, CPATH_NEW);
97 }
98 
setup(void)99 static void setup(void)
100 {
101 	if (access(PATH_KSM, F_OK) == -1)
102 		tst_brk(TCONF, "KSM configuration is not enabled");
103 	save_max_page_sharing();
104 
105 	parse_ksm_options(opt_sizestr, &size, opt_numstr, &num, opt_unitstr, &unit);
106 
107 	if (access(PATH_KSM "merge_across_nodes", F_OK) == 0) {
108 		SAFE_FILE_SCANF(PATH_KSM "merge_across_nodes",
109 				"%d", &merge_across_nodes);
110 		SAFE_FILE_PRINTF(PATH_KSM "merge_across_nodes", "1");
111 	}
112 
113 	mount_mem("cpuset", "cpuset", NULL, CPATH, CPATH_NEW);
114 }
115 
116 static struct tst_test test = {
117 	.needs_root = 1,
118 	.forks_child = 1,
119 	.options = ksm_options,
120 	.setup = setup,
121 	.cleanup = cleanup,
122 	.test_all = verify_ksm,
123 	.min_kver = "2.6.32",
124 };
125 
126 #else
127 	TST_TEST_TCONF("test requires libnuma >= 2 and it's development packages");
128 #endif
129