From e96da32cd59567a4ed35f6d97a39edb3678c1620 Mon Sep 17 00:00:00 2001
From: Max Wash <mbswash@gmail.com>
Date: Wed, 8 Feb 2023 18:48:01 +0000
Subject: [PATCH] x86_64: check for failure in alloc test

---
 arch/x86_64/init.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/arch/x86_64/init.c b/arch/x86_64/init.c
index ed7137e..321e2d9 100644
--- a/arch/x86_64/init.c
+++ b/arch/x86_64/init.c
@@ -59,10 +59,13 @@ int ml_init(uintptr_t arg)
 
 	/* test allocation */
 	vm_page_t *p = vm_page_alloc(VM_PAGE_16K, 0);
-	void *p_ptr = vm_page_get_vaddr(p);
-	memset(p_ptr, 0x00, vm_page_order_to_bytes(VM_PAGE_16K));
+	if (p) {
+	    void *p_ptr = vm_page_get_vaddr(p);
 
-	printk("allocated 16K at %p", p_ptr);
+	    printk("allocated 16K at %p", p_ptr);
+	} else {
+		printk("alloc failed");
+	}
 
 	return 0;
 }