1 // SPDX-License-Identifier: GPL-2.0 2 #define _GNU_SOURCE 3 #define __EXPORTED_HEADERS__ 4 5 #include <stdio.h> 6 #include <stdlib.h> 7 #include <linux/fcntl.h> 8 #include <linux/memfd.h> 9 #include <unistd.h> 10 #include <sys/syscall.h> 11 12 #include "common.h" 13 14 int hugetlbfs_test = 0; 15 16 /* 17 * Copied from mlock2-tests.c 18 */ default_huge_page_size(void)19unsigned long default_huge_page_size(void) 20 { 21 unsigned long hps = 0; 22 char *line = NULL; 23 size_t linelen = 0; 24 FILE *f = fopen("/proc/meminfo", "r"); 25 26 if (!f) 27 return 0; 28 while (getline(&line, &linelen, f) > 0) { 29 if (sscanf(line, "Hugepagesize: %lu kB", &hps) == 1) { 30 hps <<= 10; 31 break; 32 } 33 } 34 35 free(line); 36 fclose(f); 37 return hps; 38 } 39 sys_memfd_create(const char * name,unsigned int flags)40int sys_memfd_create(const char *name, unsigned int flags) 41 { 42 if (hugetlbfs_test) 43 flags |= MFD_HUGETLB; 44 45 return syscall(__NR_memfd_create, name, flags); 46 } 47