Cap pk memory size to 2 GiB for RV32

1 files changed, 5 insertions, 0 deletions

diff --git a/pk/mmap.c b/pk/mmap.c
index 1fac4c8..9ef3a57 100644
--- a/pk/mmap.c
+++ b/pk/mmap.c
@@ -385,6 +385,11 @@ void populate_mapping(const void* start, size_t size, int prot)
 
 uintptr_t pk_vm_init()
 {
+#ifdef __riscv32
+  // We can't support more than 2 GiB of memory in RV32
+  mem_size = MIN(mem_size, 1U << 31);
+#endif
+
   size_t mem_pages = mem_size >> RISCV_PGSHIFT;
   free_pages = MAX(8, mem_pages >> (RISCV_PGLEVEL_BITS-1));
   first_free_page = first_free_paddr;