This is the first step of making this function more generic.
The main purpose of this patch is to make it easier to review the
next one.
No functional change intended.
Signed-off-by: Fuad Tabba <tabba@xxxxxxxxxx>
---
util/util.c | 35 ++++++++++++++++++-----------------
1 file changed, 18 insertions(+), 17 deletions(-)
diff --git a/util/util.c b/util/util.c
index b347fa3..93f895a 100644
--- a/util/util.c
+++ b/util/util.c
@@ -95,19 +95,12 @@ static u64 get_hugepage_blk_size(const char *hugetlbfs_path)
return sfs.f_bsize;
}
-static void *mmap_hugetlbfs(struct kvm *kvm, const char *hugetlbfs_path, u64 size)
+static void *mmap_hugetlbfs(struct kvm *kvm, const char *hugetlbfs_path, u64 size, u64 blk_size)
{
const char *name = "kvmtool";
unsigned int flags = 0;
int fd;
void *addr;
- u64 blk_size;
-
- blk_size = get_hugepage_blk_size(hugetlbfs_path);
- if (blk_size == 0 || blk_size > size) {
- die("Can't use hugetlbfs pagesize %lld for mem size %lld",
- (unsigned long long)blk_size, (unsigned long long)size);
- }
if (!is_power_of_two(blk_size))
die("Hugepage size must be a power of 2");
@@ -115,8 +108,6 @@ static void *mmap_hugetlbfs(struct kvm *kvm, const char *hugetlbfs_path, u64 siz
flags |= MFD_HUGETLB;
flags |= blk_size << MFD_HUGE_SHIFT;
- kvm->ram_pagesize = blk_size;
-
fd = memfd_create(name, flags);
if (fd < 0)
die_perror("Can't memfd_create for hugetlbfs map");
@@ -132,13 +123,23 @@ static void *mmap_hugetlbfs(struct kvm *kvm, const char *hugetlbfs_path, u64 siz
/* This function wraps the decision between hugetlbfs map (if requested) or normal mmap */
void *mmap_anon_or_hugetlbfs(struct kvm *kvm, const char *hugetlbfs_path, u64 size)
{
- if (hugetlbfs_path)
- /*
- * We don't /need/ to map guest RAM from hugetlbfs, but we do so
- * if the user specifies a hugetlbfs path.
- */
- return mmap_hugetlbfs(kvm, hugetlbfs_path, size);
- else {
+ u64 blk_size = 0;
+
+ /*
+ * We don't /need/ to map guest RAM from hugetlbfs, but we do so
+ * if the user specifies a hugetlbfs path.
+ */
+ if (hugetlbfs_path) {
+ blk_size = get_hugepage_blk_size(hugetlbfs_path);
+
+ if (blk_size == 0 || blk_size > size) {
+ die("Can't use hugetlbfs pagesize %lld for mem size %lld\n",
+ (unsigned long long)blk_size, (unsigned long long)size);
+ }
+
+ kvm->ram_pagesize = blk_size;
+ return mmap_hugetlbfs(kvm, hugetlbfs_path, size, blk_size);
+ } else {
kvm->ram_pagesize = getpagesize();
return mmap(NULL, size, PROT_RW, MAP_ANON_NORESERVE, -1, 0);
}
--
2.39.0.rc0.267.gcb52ba06e7-goog
