From f26de9358ee5422dd9b7361fbb86fdc99107eae0 Mon Sep 17 00:00:00 2001
From: Fabian Wiesel <fabian.wiesel@sap.com>
Date: Thu, 18 Jul 2024 13:25:23 +0200
Subject: [PATCH] [orabos] Also persist hugepages from proc

---
 features/orabos/file.include/opt/orabos/init.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/features/orabos/file.include/opt/orabos/init.py b/features/orabos/file.include/opt/orabos/init.py
index 5453f572b..a73ea4744 100755
--- a/features/orabos/file.include/opt/orabos/init.py
+++ b/features/orabos/file.include/opt/orabos/init.py
@@ -68,15 +68,21 @@ def setup_memory():
     return
 
   cmdline = pathlib.Path("/proc/cmdline").read_text()
-  m = re.search(r"hugepages=\d+", cmdline)
-  if not m:
+  m = re.search(r"hugepages=(\d+)", cmdline)
+  hugepages=0
+  if m:
+    hugepages=int(m[1])
+  else:
+    hugepages=int(pathlib.Path("/proc/sys/vm/nr_hugepages").read_text())
+
+  if hugepages<=0:
     return
 
-  cfg.write_text(f'CMDLINE_LINUX="$CMDLINE_LINUX {m.group(0)}"\n')
+  cfg.write_text(f'CMDLINE_LINUX="$CMDLINE_LINUX hugepages={hugepages}"\n')
 
   for entry in pathlib.Path("/efi/loader/entries").glob("*.conf"):
     old = entry.read_text()
-    new = re.sub(r"^options.*", f"\\g<0> {m.group(0)}", old, flags=re.M)
+    new = re.sub(r"^options.*", f"\\g<0> hugepages={hugepages}", old, flags=re.M)
     entry.write_text(new)
 
 setup_ovs()