#6 support exception tables

Add support for exception tables. It is possible to register an address that triggers an exception to call an handler, jump to a fixup address or both. Signed-off-by: Daniele Ahmed <[email protected]>
KernelTestFramework · Sep 10, 2021 · 87ce80a · 87ce80a
1 parent 307a649
commit 87ce80a
Show file tree

Hide file tree

Showing 9 changed files with 120 additions and 3 deletions.
diff --git a/arch/x86/extables.c b/arch/x86/extables.c
@@ -0,0 +1,27 @@
+/*
+ * Copyright (c) 2021 Amazon.com, Inc. or its affiliates.
+ * All Rights Reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+void init_extables(void) { /* no-op */
+}
diff --git a/arch/x86/link.ld b/arch/x86/link.ld
@@ -170,6 +170,14 @@ SECTIONS
         __end_data = .;
     } :kernel
 
+    .extables BLOCK(4K) : ALIGN(4K)
+    {
+        __start_extables = .;
+            *(.extables)
+        . = ALIGN(4K);
+        __end_extables = .;
+    } :kernel
+
     .bss BLOCK(4K) : ALIGN(4K)
     {
         . = ALIGN(4K);

diff --git a/arch/x86/traps.c b/arch/x86/traps.c
@@ -28,8 +28,10 @@
 #include <drivers/keyboard.h>
 #include <drivers/pit.h>
 #include <drivers/serial.h>
+#include <extables.h>
 #include <ktf.h>
 #include <lib.h>
+#include <mm/regions.h>
 #include <percpu.h>
 #include <processor.h>
 #include <segment.h>
@@ -287,9 +289,30 @@ void print_callstack(const void *sp, const void *ip) {
     printk("\n");
 }
 
+static inline void use_extables(struct cpu_regs *regs) {
+    for (extable_entry_t *cur = __start_extables; cur < __end_extables; ++cur) {
+        if (regs->_ASM_IP == cur->fault_addr) {
+            if (cur->fixup) {
+                regs->_ASM_IP = cur->fixup;
+            }
+            if (cur->cb) {
+                cur->cb(regs);
+            }
+            if (cur->fixup || cur->cb) {
+                return;
+            }
+            else {
+                panic("fixup or cb must be set in the extable_entry\n");
+            }
+        }
+    }
+}
+
 void do_exception(struct cpu_regs *regs) {
     static char ec_str[32], panic_str[128];
 
+    use_extables(regs);
+
     dump_regs(regs);
     print_callstack(_ptr(regs->_ASM_SP), _ptr(regs->_ASM_IP));
 

diff --git a/common/setup.c b/common/setup.c
@@ -254,12 +254,12 @@ void __noreturn __text_init kernel_start(uint32_t multiboot_magic,
     if (opt_debug)
         dump_pagetables();
 
-    /* TODO: Exception tables */
-
     init_percpu();
 
     init_traps(get_bsp_cpu_id());
 
+    init_extables();
+
     init_slab();
 
     /* Try to initialize ACPI (and MADT) */

diff --git a/include/arch/x86/asm-macros.h b/include/arch/x86/asm-macros.h
@@ -47,6 +47,9 @@
 
 #define PT_PADDR(table, shift) (((.- (table)) / PTE_SIZE) << (shift))
 
+#define PUSHSECTION(name) ".pushsection " STR(name) " ;\n"
+#define POPSECTION        ".popsection;\n"
+
 #ifdef __ASSEMBLY__
 
 /* clang-format off */

diff --git a/include/extables.h b/include/extables.h
@@ -0,0 +1,53 @@
+/*
+ * Copyright © 2021 Amazon.com, Inc. or its affiliates.
+ * All Rights Reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+#ifndef KTF_EXTABLES_H
+#define KTF_EXTABLES_H
+
+#include <arch/x86/processor.h>
+
+typedef void (*extable_entry_callback_t)(cpu_regs_t *regs);
+
+struct extable_entry {
+    x86_reg_t fault_addr;
+    x86_reg_t fixup;
+    extable_entry_callback_t cb;
+} __packed;
+typedef struct extable_entry extable_entry_t;
+
+#if defined(__x86_64__)
+#define ASM_EXTABLE_HANDLER(fault_address, fixup, handler)                               \
+    PUSHSECTION(.extables)                                                               \
+    ".quad " STR(fault_address) ", " STR(fixup) ", " STR(handler) ";\n" POPSECTION
+#else
+#define ASM_EXTABLE_HANDLER(fault_address, fixup, handler)                               \
+    PUSHSECTION(.extables)                                                               \
+    ".long " STR(fault_address) ", " STR(fixup) ", " STR(handler) ";\n" POPSECTION
+#endif
+
+#define ASM_EXTABLE(fault_address, fixup) ASM_EXTABLE_HANDLER(fault_address, fixup, 0)
+
+extern void init_extables(void);
+
+#endif
diff --git a/include/lib.h b/include/lib.h
@@ -1,5 +1,5 @@
 /*
- * Copyright © 2020 Amazon.com, Inc. or its affiliates.
+ * Copyright © 2021 Amazon.com, Inc. or its affiliates.
  * All Rights Reserved.
  *
  * Redistribution and use in source and binary forms, with or without

diff --git a/include/mm/regions.h b/include/mm/regions.h
@@ -38,6 +38,7 @@
 
 #ifndef __ASSEMBLY__
 #include <cmdline.h>
+#include <extables.h>
 #include <page.h>
 #include <string.h>
 
@@ -56,6 +57,7 @@ extern unsigned long __start_bss_init[], __end_bss_init[];
 
 extern unsigned long __start_text_rmode[], __end_text_rmode[];
 extern unsigned long __start_data_rmode[], __end_data_rmode[];
+extern struct extable_entry __start_extables[], __end_extables[];
 extern unsigned long __start_bss_rmode[], __end_bss_rmode[];
 
 extern struct ktf_param __start_cmdline[], __end_cmdline[];

diff --git a/mm/regions.c b/mm/regions.c
@@ -60,6 +60,7 @@ addr_range_t addr_ranges[] = {
 
     KERNEL_RANGE( ".text",      L1_PROT_RO,      __start_text,           __end_text      ),
     KERNEL_RANGE( ".data",      L1_PROT,         __start_data,           __end_data      ),
+    KERNEL_RANGE( ".extables",  L1_PROT_RO,      __start_extables,       __end_extables  ),
     KERNEL_RANGE( ".bss",       L1_PROT,         __start_bss,            __end_bss       ),
     KERNEL_RANGE( ".rodata",    L1_PROT_RO,      __start_rodata,         __end_rodata    ),
     KERNEL_RANGE( ".symbols",   L1_PROT_RO,      __start_symbols,        __end_symbols   ),