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