From 9ce71e5596ffbff3f2abd14ebfb6a9fe8836f599 Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Sat, 9 Apr 2016 15:23:01 -0400
Subject: [PATCH] Add memorySize() method to StartupInformation

---
 src/StartupInformation.cc | 8 ++++++++
 src/StartupInformation.hh | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/src/StartupInformation.cc b/src/StartupInformation.cc
index 5357f36..6d03adb 100644
--- a/src/StartupInformation.cc
+++ b/src/StartupInformation.cc
@@ -21,4 +21,12 @@ StartupInformation::kernelSize()
     return kernelEnd - kernelStart;
 }
 
+u32
+StartupInformation::memorySize()
+    const
+{
+    // Memory is 1 MiB plus however much upper memory we have.
+    return 1024 * 1024 + multibootInformation->upperMemoryKB() * 1024;
+}
+
 } /* namespace kernel */
diff --git a/src/StartupInformation.hh b/src/StartupInformation.hh
index 77ea7af..cc74c5e 100644
--- a/src/StartupInformation.hh
+++ b/src/StartupInformation.hh
@@ -38,6 +38,9 @@ struct StartupInformation
 
     /** Size of the kernel image in bytes. */
     u32 kernelSize() const;
+
+    /** Size of memory in bytes. */
+    u32 memorySize() const;
 };
 
 } /* namespace kernel */