Author | Tokens | Token Proportion | Commits | Commit Proportion |
---|---|---|---|---|
Nhat Pham | 973 | 99.49% | 1 | 33.33% |
Shuah Khan | 3 | 0.31% | 1 | 33.33% |
Tianchen Ding | 2 | 0.20% | 1 | 33.33% |
Total | 978 | 3 |
// SPDX-License-Identifier: GPL-2.0 #define _GNU_SOURCE #include <linux/limits.h> #include <sys/mman.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <fcntl.h> #include "../kselftest.h" #include "cgroup_util.h" #define ADDR ((void *)(0x0UL)) #define FLAGS (MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB) /* mapping 8 MBs == 4 hugepages */ #define LENGTH (8UL*1024*1024) #define PROTECTION (PROT_READ | PROT_WRITE) /* borrowed from mm/hmm-tests.c */ static long get_hugepage_size(void) { int fd; char buf[2048]; int len; char *p, *q, *path = "/proc/meminfo", *tag = "Hugepagesize:"; long val; fd = open(path, O_RDONLY); if (fd < 0) { /* Error opening the file */ return -1; } len = read(fd, buf, sizeof(buf)); close(fd); if (len < 0) { /* Error in reading the file */ return -1; } if (len == sizeof(buf)) { /* Error file is too large */ return -1; } buf[len] = '\0'; /* Search for a tag if provided */ if (tag) { p = strstr(buf, tag); if (!p) return -1; /* looks like the line we want isn't there */ p += strlen(tag); } else p = buf; val = strtol(p, &q, 0); if (*q != ' ') { /* Error parsing the file */ return -1; } return val; } static int set_file(const char *path, long value) { FILE *file; int ret; file = fopen(path, "w"); if (!file) return -1; ret = fprintf(file, "%ld\n", value); fclose(file); return ret; } static int set_nr_hugepages(long value) { return set_file("/proc/sys/vm/nr_hugepages", value); } static unsigned int check_first(char *addr) { return *(unsigned int *)addr; } static void write_data(char *addr) { unsigned long i; for (i = 0; i < LENGTH; i++) *(addr + i) = (char)i; } static int hugetlb_test_program(const char *cgroup, void *arg) { char *test_group = (char *)arg; void *addr; long old_current, expected_current, current; int ret = EXIT_FAILURE; old_current = cg_read_long(test_group, "memory.current"); set_nr_hugepages(20); current = cg_read_long(test_group, "memory.current"); if (current - old_current >= MB(2)) { ksft_print_msg( "setting nr_hugepages should not increase hugepage usage.\n"); ksft_print_msg("before: %ld, after: %ld\n", old_current, current); return EXIT_FAILURE; } addr = mmap(ADDR, LENGTH, PROTECTION, FLAGS, 0, 0); if (addr == MAP_FAILED) { ksft_print_msg("fail to mmap.\n"); return EXIT_FAILURE; } current = cg_read_long(test_group, "memory.current"); if (current - old_current >= MB(2)) { ksft_print_msg("mmap should not increase hugepage usage.\n"); ksft_print_msg("before: %ld, after: %ld\n", old_current, current); goto out_failed_munmap; } old_current = current; /* read the first page */ check_first(addr); expected_current = old_current + MB(2); current = cg_read_long(test_group, "memory.current"); if (!values_close(expected_current, current, 5)) { ksft_print_msg("memory usage should increase by around 2MB.\n"); ksft_print_msg( "expected memory: %ld, actual memory: %ld\n", expected_current, current); goto out_failed_munmap; } /* write to the whole range */ write_data(addr); current = cg_read_long(test_group, "memory.current"); expected_current = old_current + MB(8); if (!values_close(expected_current, current, 5)) { ksft_print_msg("memory usage should increase by around 8MB.\n"); ksft_print_msg( "expected memory: %ld, actual memory: %ld\n", expected_current, current); goto out_failed_munmap; } /* unmap the whole range */ munmap(addr, LENGTH); current = cg_read_long(test_group, "memory.current"); expected_current = old_current; if (!values_close(expected_current, current, 5)) { ksft_print_msg("memory usage should go back down.\n"); ksft_print_msg( "expected memory: %ld, actual memory: %ld\n", expected_current, current); return ret; } ret = EXIT_SUCCESS; return ret; out_failed_munmap: munmap(addr, LENGTH); return ret; } static int test_hugetlb_memcg(char *root) { int ret = KSFT_FAIL; char *test_group; test_group = cg_name(root, "hugetlb_memcg_test"); if (!test_group || cg_create(test_group)) { ksft_print_msg("fail to create cgroup.\n"); goto out; } if (cg_write(test_group, "memory.max", "100M")) { ksft_print_msg("fail to set cgroup memory limit.\n"); goto out; } /* disable swap */ if (cg_write(test_group, "memory.swap.max", "0")) { ksft_print_msg("fail to disable swap.\n"); goto out; } if (!cg_run(test_group, hugetlb_test_program, (void *)test_group)) ret = KSFT_PASS; out: cg_destroy(test_group); free(test_group); return ret; } int main(int argc, char **argv) { char root[PATH_MAX]; int ret = EXIT_SUCCESS, has_memory_hugetlb_acc; has_memory_hugetlb_acc = proc_mount_contains("memory_hugetlb_accounting"); if (has_memory_hugetlb_acc < 0) ksft_exit_skip("Failed to query cgroup mount option\n"); else if (!has_memory_hugetlb_acc) ksft_exit_skip("memory hugetlb accounting is disabled\n"); /* Unit is kB! */ if (get_hugepage_size() != 2048) { ksft_print_msg("test_hugetlb_memcg requires 2MB hugepages\n"); ksft_test_result_skip("test_hugetlb_memcg\n"); return ret; } if (cg_find_unified_root(root, sizeof(root), NULL)) ksft_exit_skip("cgroup v2 isn't mounted\n"); switch (test_hugetlb_memcg(root)) { case KSFT_PASS: ksft_test_result_pass("test_hugetlb_memcg\n"); break; case KSFT_SKIP: ksft_test_result_skip("test_hugetlb_memcg\n"); break; default: ret = EXIT_FAILURE; ksft_test_result_fail("test_hugetlb_memcg\n"); break; } return ret; }
Information contained on this website is for historical information purposes only and does not indicate or represent copyright ownership.
Created with Cregit http://github.com/cregit/cregit
Version 2.0-RC1