1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright (c) 2019 Red Hat, Inc.
4 */
5
6 #define TST_NO_DEFAULT_MAIN
7
8 #include "tst_test.h"
9 #include "tst_hugepage.h"
10
11 unsigned long tst_hugepages;
12 char *nr_opt;
13 char *Hopt;
14
tst_get_hugepage_size(void)15 size_t tst_get_hugepage_size(void)
16 {
17 if (access(PATH_HUGEPAGES, F_OK))
18 return 0;
19
20 return SAFE_READ_MEMINFO("Hugepagesize:") * 1024;
21 }
22
tst_request_hugepages(unsigned long hpages)23 unsigned long tst_request_hugepages(unsigned long hpages)
24 {
25 unsigned long val, max_hpages;
26
27 if (access(PATH_HUGEPAGES, F_OK)) {
28 tst_hugepages = 0;
29 goto out;
30 }
31
32 if (nr_opt)
33 tst_hugepages = SAFE_STRTOL(nr_opt, 1, LONG_MAX);
34 else
35 tst_hugepages = hpages;
36
37 SAFE_FILE_PRINTF("/proc/sys/vm/drop_caches", "3");
38 max_hpages = SAFE_READ_MEMINFO("MemFree:") / SAFE_READ_MEMINFO("Hugepagesize:");
39
40 if (tst_hugepages > max_hpages) {
41 tst_res(TINFO, "Requested number(%lu) of hugepages is too large, "
42 "limiting to 80%% of the max hugepage count %lu",
43 tst_hugepages, max_hpages);
44 tst_hugepages = max_hpages * 0.8;
45
46 if (tst_hugepages < 1)
47 goto out;
48 }
49
50 tst_sys_conf_save("?/proc/sys/vm/nr_hugepages");
51 SAFE_FILE_PRINTF(PATH_NR_HPAGES, "%lu", tst_hugepages);
52 SAFE_FILE_SCANF(PATH_NR_HPAGES, "%lu", &val);
53 if (val != tst_hugepages)
54 tst_brk(TCONF, "nr_hugepages = %lu, but expect %lu. "
55 "Not enough hugepages for testing.",
56 val, tst_hugepages);
57
58 tst_res(TINFO, "%lu hugepage(s) reserved", tst_hugepages);
59 out:
60 return tst_hugepages;
61 }
62