Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | // SPDX-License-Identifier: GPL-2.0 /* * This test covers the anonymous VMA naming functionality through prctl calls */ #include <errno.h> #include <sys/prctl.h> #include <stdio.h> #include <stdlib.h> #include <sys/mman.h> #include <string.h> #include "../kselftest_harness.h" #define AREA_SIZE 1024 #define GOOD_NAME "goodname" #define BAD_NAME "badname\1" #ifndef PR_SET_VMA #define PR_SET_VMA 0x53564d41 #define PR_SET_VMA_ANON_NAME 0 #endif int rename_vma(unsigned long addr, unsigned long size, char *name) { int res; res = prctl(PR_SET_VMA, PR_SET_VMA_ANON_NAME, addr, size, name); if (res < 0) return -errno; return res; } int was_renaming_successful(char *target_name, unsigned long ptr) { FILE *maps_file; char line_buf[512], name[128], mode[8]; unsigned long start_addr, end_addr, offset; unsigned int major_id, minor_id, node_id; char target_buf[128]; int res = 0, sscanf_res; // The entry name in maps will be in format [anon:<target_name>] sprintf(target_buf, "[anon:%s]", target_name); maps_file = fopen("/proc/self/maps", "r"); if (!maps_file) { printf("## /proc/self/maps file opening error\n"); return 0; } // Parse the maps file to find the entry we renamed while (fgets(line_buf, sizeof(line_buf), maps_file)) { sscanf_res = sscanf(line_buf, "%lx-%lx %7s %lx %u:%u %u %s", &start_addr, &end_addr, mode, &offset, &major_id, &minor_id, &node_id, name); if (sscanf_res == EOF) { res = 0; printf("## EOF while parsing the maps file\n"); break; } if (!strcmp(name, target_buf) && start_addr == ptr) { res = 1; break; } } fclose(maps_file); return res; } FIXTURE(vma) { void *ptr_anon, *ptr_not_anon; }; FIXTURE_SETUP(vma) { self->ptr_anon = mmap(NULL, AREA_SIZE, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, 0, 0); ASSERT_NE(self->ptr_anon, NULL); self->ptr_not_anon = mmap(NULL, AREA_SIZE, PROT_READ | PROT_WRITE, MAP_PRIVATE, 0, 0); ASSERT_NE(self->ptr_not_anon, NULL); } FIXTURE_TEARDOWN(vma) { munmap(self->ptr_anon, AREA_SIZE); munmap(self->ptr_not_anon, AREA_SIZE); } TEST_F(vma, renaming) { TH_LOG("Try to rename the VMA with correct parameters"); EXPECT_GE(rename_vma((unsigned long)self->ptr_anon, AREA_SIZE, GOOD_NAME), 0); EXPECT_TRUE(was_renaming_successful(GOOD_NAME, (unsigned long)self->ptr_anon)); TH_LOG("Try to pass invalid name (with non-printable character \\1) to rename the VMA"); EXPECT_EQ(rename_vma((unsigned long)self->ptr_anon, AREA_SIZE, BAD_NAME), -EINVAL); TH_LOG("Try to rename non-anonymous VMA"); EXPECT_EQ(rename_vma((unsigned long) self->ptr_not_anon, AREA_SIZE, GOOD_NAME), -EINVAL); } TEST_HARNESS_MAIN |