Clean up the PTE() macro (XCC)
[akaros.git] / kern / arch / riscv / cboot.c
1 #include <multiboot.h>
2 #include <ros/memlayout.h>
3 #include <arch/arch.h>
4 #include <arch/mmu.h>
5 #include <string.h>
6 #include <assert.h>
7 #include <stdio.h>
8 #include <pmap.h>
9
10 #define MAX_KERNBASE_SIZE (KERN_VMAP_TOP - KERNBASE)
11
12 uint32_t num_cores = 1;   // this must not be in BSS
13
14 static uint64_t
15 mem_size(uint64_t sz_mb)
16 {
17         uint64_t sz = (uint64_t)sz_mb * 1024 * 1024;
18         return MIN(sz,  MIN(MAX_KERNBASE_SIZE, (uint64_t)L1PGSIZE * NPTENTRIES));
19 }
20
21 void pagetable_init(uint32_t memsize_mb, pte_t* l1pt, pte_t* l1pt_boot,
22                     pte_t* l2pt)
23 {
24         static_assert(KERNBASE % L1PGSIZE == 0);
25         // The boot L1 PT retains the identity mapping [0,memsize-1],
26         // whereas the post-boot L1 PT does not.
27         uint64_t memsize = mem_size(memsize_mb);
28         for(uint64_t pa = 0; pa < memsize+L1PGSIZE-1; pa += L1PGSIZE)
29         {
30                 pte_t pte = build_pte(pa, PTE_KERN_RW | PTE_E);
31
32                 l1pt_boot[L1X(pa)] = pte; // identity mapping
33                 l1pt_boot[L1X(KERNBASE+pa)] = pte; // KERNBASE mapping
34                 l1pt[L1X(KERNBASE+pa)] = pte; // KERNBASE mapping
35         }
36
37 #ifdef __riscv64
38         // The kernel code and static data actually are usually not accessed
39         // via the KERNBASE mapping, but rather by an aliased "load" mapping in
40         // the upper 2GB (0xFFFFFFFF80000000 and up).
41         // This simplifies the linking model by making all static addresses
42         // representable in 32 bits.
43         static_assert(L1X(KERN_LOAD_ADDR) > L1X(KERNBASE));
44         static_assert(KERN_LOAD_ADDR % L2PGSIZE == 0);
45         static_assert((uintptr_t)(-KERN_LOAD_ADDR) <= L1PGSIZE);
46
47         l1pt[L1X(KERN_LOAD_ADDR)] = PTD(l2pt);
48         l1pt_boot[L1X(KERN_LOAD_ADDR)] = PTD(l2pt);
49
50         for (uintptr_t pa = 0; pa < (uintptr_t)(-KERN_LOAD_ADDR); pa += L2PGSIZE)
51                 l2pt[L2X(KERN_LOAD_ADDR+pa)] = build_pte(pa, PTE_KERN_RW | PTE_E);
52 #else
53         (void) l2pt; // don't need this for rv32
54 #endif
55 }
56
57 void
58 cmain(uint32_t memsize_mb, uint32_t nc)
59 {
60         multiboot_info_t mbi;
61         memset(&mbi, 0, sizeof(mbi));
62         mbi.flags = 0x00000001;
63         mbi.mem_lower = mem_size(memsize_mb) / 1024;
64
65         num_cores = nc;
66
67         extern void kernel_init(multiboot_info_t *mboot_info);
68         // kernel_init expects a pre-relocation mbi address
69         kernel_init((multiboot_info_t*)PADDR(&mbi));
70 }