Add printf like behaviour to assert

4 files changed, 31 insertions, 21 deletions

diff --git a/arch/x86_64/src/memory/multiboot/reader.cpp b/arch/x86_64/src/memory/multiboot/reader.cpp
index 565f604..4e78a06 100644
--- a/arch/x86_64/src/memory/multiboot/reader.cpp
+++ b/arch/x86_64/src/memory/multiboot/reader.cpp
@@ -19,7 +19,9 @@ namespace teachos::arch::memory::multiboot
     {
       auto expected_entry_size = mminfo->entry_size;
       constexpr auto actual_entry_size = sizeof(memory_area);
-      exception_handling::assert(expected_entry_size == actual_entry_size, "Unexpected memory_area entry size");
+      exception_handling::assert(expected_entry_size == actual_entry_size,
+                                 "[Multiboot Reader] Expected memory area entry size (%u) but got (%u)",
+                                 expected_entry_size, actual_entry_size);
 
       auto total_size = mminfo->info.size;
       auto total_entries_size = total_size - sizeof(memory_map_header) + actual_entry_size;
@@ -34,18 +36,22 @@ namespace teachos::arch::memory::multiboot
     {
       auto expected_entry_size = symbol->entry_size;
       constexpr auto actual_entry_size = sizeof(elf_section_header);
-      exception_handling::assert(expected_entry_size == actual_entry_size, "Unexpected elf_section_header entry size");
+      exception_handling::assert(expected_entry_size == actual_entry_size,
+                                 "[Multiboot Reader] Expected elf section header entry size (%u) but got (%u)",
+                                 expected_entry_size, actual_entry_size);
 
       auto expected_total_size = symbol->info.size;
       auto actual_total_entry_size = actual_entry_size * symbol->number_of_sections;
       constexpr auto actual_total_section_size = sizeof(elf_symbols_section_header) - sizeof(uint32_t);
       auto actual_total_size = actual_total_entry_size + actual_total_section_size;
       exception_handling::assert(expected_total_size == actual_total_size,
-                                 "Unexpected elf_symbols_section_header total size");
+                                 "[Multiboot Reader] Expected elf symbols section header total size (%u) but got (%u)",
+                                 expected_total_size, actual_total_size);
 
       auto begin = reinterpret_cast<elf_section_header *>(&symbol->end);
       auto end = begin + symbol->number_of_sections;
-      exception_handling::assert(begin->is_null(), "Missing elf_section_header begin");
+      exception_handling::assert(begin->is_null(),
+                                 "[Multiboot Reader] Elf symbols section not starting with SHT_NULL section");
 
       std::size_t symbol_table_section_count = 0U;
       std::size_t dynamic_section_count = 0U;
@@ -79,8 +85,14 @@ namespace teachos::arch::memory::multiboot
         }
       }
 
-      exception_handling::assert(symbol_table_section_count == 1U, "Unexpected symbol_table_count value");
-      exception_handling::assert(dynamic_section_count <= 1U, "Unexpected dynamic_section_count value");
+      exception_handling::assert(
+          symbol_table_section_count == 1U,
+          "[Multiboot Reader] ELF Specifications allows only (1) symbol table section, but got (%u)",
+          symbol_table_section_count);
+      exception_handling::assert(
+          dynamic_section_count <= 1U,
+          "[Multiboot Reader] ELF Specifications allows only (1) or less dynamic sections, but got (%u)",
+          dynamic_section_count);
     }
   }  // namespace
 
diff --git a/arch/x86_64/src/memory/paging/page_entry.cpp b/arch/x86_64/src/memory/paging/page_entry.cpp
index 0dbbae1..30a8961 100644
--- a/arch/x86_64/src/memory/paging/page_entry.cpp
+++ b/arch/x86_64/src/memory/paging/page_entry.cpp
@@ -36,7 +36,7 @@ namespace teachos::arch::memory::paging
   auto entry::set_entry(allocator::physical_frame frame, std::bitset<64U> additional_flags) -> void
   {
     exception_handling::assert((frame.start_address() & ~0x000fffff'fffff000) == 0,
-                               "Start address is not aligned with Page");
+                               "[Paging Entry] Start address is not aligned with page");
     flags = std::bitset<64U>(frame.start_address()) | additional_flags;
   }
 }  // namespace teachos::arch::memory::paging
diff --git a/arch/x86_64/src/memory/paging/page_mapper.cpp b/arch/x86_64/src/memory/paging/page_mapper.cpp
index eb1e18c..8c64f22 100644
--- a/arch/x86_64/src/memory/paging/page_mapper.cpp
+++ b/arch/x86_64/src/memory/paging/page_mapper.cpp
@@ -81,28 +81,26 @@ namespace teachos::arch::memory::paging
                          allocator::physical_frame const & frame, entry::bitset flags) -> void
   {
     page_table page_table{};
-    bool is_valid = false;
+    bool table_exists = false;
 
     for (auto level = page_table::LEVEL4; level != page_table::LEVEL1; level--)
     {
       std::size_t level_index = page.get_level_index(level);
-      is_valid = page_table.next_table(level_index);
+      table_exists = page_table.next_table(level_index);
 
-      if (!is_valid)
+      if (!table_exists)
       {
-        break;
+        auto allocated_frame = allocator.allocate_frame();
+        exception_handling::assert(!allocated_frame.has_value(), "[Page mapper]: Unable to allocate frame");
+        page_table[level_index].set_entry(allocated_frame.value(), entry::PRESENT | entry::WRITABLE);
+        page_table.zero_entries();
       }
-
-      auto allocated_frame = allocator.allocate_frame();
-      exception_handling::assert(!allocated_frame.has_value(), "[Page mapper]: Unable to allocate frame");
-      page_table[level_index].set_entry(allocated_frame.value(), entry::PRESENT | entry::WRITABLE);
-      page_table.zero_entries();
     }
 
-    entry page_table_entry = page_table[page.get_level_index(page_table::LEVEL1)];
-    arch::exception_handling::assert(!page_table_entry.contains_flags(entry::HUGE_PAGE),
+    auto level1_entry = page_table[page.get_level_index(page_table::LEVEL1)];
+    arch::exception_handling::assert(!level1_entry.contains_flags(entry::HUGE_PAGE),
                                      "[Page Mapper]: Unable to map huge pages");
-    arch::exception_handling::assert(!page_table_entry.is_unused(), "[Page Mapper]: Page table entry is already used");
-    page_table_entry.set_entry(frame, flags | entry::PRESENT);
+    arch::exception_handling::assert(!level1_entry.is_unused(), "[Page Mapper]: Page table entry is already used");
+    level1_entry.set_entry(frame, flags | entry::PRESENT);
   }
 }  // namespace teachos::arch::memory::paging
\ No newline at end of file
diff --git a/arch/x86_64/src/memory/paging/page_table.cpp b/arch/x86_64/src/memory/paging/page_table.cpp
index 02ababe..ea2e9c2 100644
--- a/arch/x86_64/src/memory/paging/page_table.cpp
+++ b/arch/x86_64/src/memory/paging/page_table.cpp
@@ -39,7 +39,7 @@ namespace teachos::arch::memory::paging
   {
     // C array is not bounds checked, therefore we have to check ourselves, to ensure no out of bounds reads, which
     // could be incredibly hard to debug later.
-    exception_handling::assert(index < PAGE_TABLE_ENTRY_COUNT, "[Page Table] index out of bounds");
+    exception_handling::assert(index < PAGE_TABLE_ENTRY_COUNT, "[Page Table] Index out of bounds");
     return current_table->entries[index];
   }