[PATCH v4 7/7] drivers: hyperv: balloon - report inflated memory

From: Alexander Atanasov
Date: Wed Oct 05 2022 - 05:03:41 EST


Propagate inflated memory changes to mm.

Signed-off-by: Alexander Atanasov <alexander.atanasov@xxxxxxxxxxxxx>
---
drivers/hv/hv_balloon.c | 11 +++++++++++
1 file changed, 11 insertions(+)

diff --git a/drivers/hv/hv_balloon.c b/drivers/hv/hv_balloon.c
index fdf6decacf06..b4c828c462a4 100644
--- a/drivers/hv/hv_balloon.c
+++ b/drivers/hv/hv_balloon.c
@@ -24,6 +24,7 @@
#include <linux/notifier.h>
#include <linux/percpu_counter.h>
#include <linux/page_reporting.h>
+#include <linux/balloon.h>

#include <linux/hyperv.h>
#include <asm/hyperv-tlfs.h>
@@ -1280,6 +1281,13 @@ static unsigned int alloc_balloon_pages(struct hv_dynmem_device *dm,
return i * alloc_unit;
}

+static void report_ballooned_pages(struct hv_dynmem_device *dm)
+{
+ u32 actual = dm->num_pages_ballooned;
+ long inflated_kb = actual << (HV_HYP_PAGE_SHIFT - 10);
+
+ balloon_set_inflated_total(inflated_kb);
+}
+
static void balloon_up(struct work_struct *dummy)
{
unsigned int num_pages = dm_device.balloon_wrk.num_pages;
@@ -1368,6 +1376,7 @@ static void balloon_up(struct work_struct *dummy)
}
}

+ report_ballooned_pages(&dm_device);
}

static void balloon_down(struct hv_dynmem_device *dm,
@@ -1387,6 +1396,8 @@ static void balloon_down(struct hv_dynmem_device *dm,
pr_debug("Freed %u ballooned pages.\n",
prev_pages_ballooned - dm->num_pages_ballooned);

+ report_ballooned_pages(dm);
+
if (req->more_pages == 1)
return;

--
2.31.1