162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci#define _GNU_SOURCE
362306a36Sopenharmony_ci#define __EXPORTED_HEADERS__
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include <stdio.h>
662306a36Sopenharmony_ci#include <stdlib.h>
762306a36Sopenharmony_ci#include <linux/fcntl.h>
862306a36Sopenharmony_ci#include <linux/memfd.h>
962306a36Sopenharmony_ci#include <unistd.h>
1062306a36Sopenharmony_ci#include <sys/syscall.h>
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_ci#include "common.h"
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ciint hugetlbfs_test = 0;
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci/*
1762306a36Sopenharmony_ci * Copied from mlock2-tests.c
1862306a36Sopenharmony_ci */
1962306a36Sopenharmony_ciunsigned long default_huge_page_size(void)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	unsigned long hps = 0;
2262306a36Sopenharmony_ci	char *line = NULL;
2362306a36Sopenharmony_ci	size_t linelen = 0;
2462306a36Sopenharmony_ci	FILE *f = fopen("/proc/meminfo", "r");
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci	if (!f)
2762306a36Sopenharmony_ci		return 0;
2862306a36Sopenharmony_ci	while (getline(&line, &linelen, f) > 0) {
2962306a36Sopenharmony_ci		if (sscanf(line, "Hugepagesize:       %lu kB", &hps) == 1) {
3062306a36Sopenharmony_ci			hps <<= 10;
3162306a36Sopenharmony_ci			break;
3262306a36Sopenharmony_ci		}
3362306a36Sopenharmony_ci	}
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	free(line);
3662306a36Sopenharmony_ci	fclose(f);
3762306a36Sopenharmony_ci	return hps;
3862306a36Sopenharmony_ci}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ciint sys_memfd_create(const char *name, unsigned int flags)
4162306a36Sopenharmony_ci{
4262306a36Sopenharmony_ci	if (hugetlbfs_test)
4362306a36Sopenharmony_ci		flags |= MFD_HUGETLB;
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	return syscall(__NR_memfd_create, name, flags);
4662306a36Sopenharmony_ci}
47