pintos-os.org Git - pintos-anon/blob - src/threads/init.c

   1 #include "init.h"
   2 #include <stdint.h>
   3 #include <stddef.h>
   4 #include <limits.h>
   5 #include "debug.h"
   6 #include "interrupt.h"
   7 #include "io.h"
   8 #include "kbd.h"
   9 #include "lib.h"
  10 #include "malloc.h"
  11 #include "mmu.h"
  12 #include "paging.h"
  13 #include "palloc.h"
  14 #include "random.h"
  15 #include "serial.h"
  16 #include "thread.h"
  17 #include "timer.h"
  18 #include "vga.h"
  19 #ifdef FILESYS
  20 #include "filesys.h"
  21 #endif
  22
  23 /* Size of kernel static code and data, in 4 kB pages. */
  24 size_t kernel_pages;
  25
  26 /* Amount of physical memory, in 4 kB pages. */
  27 size_t ram_pages;
  28
  29 static void gdt_init (void);
  30 void power_off (void);
  31
  32 static void
  33 main_thread (void *aux UNUSED)
  34 {
  35   printk ("execute=%d\n", (int) thread_execute ("a.out"));
  36 }
  37
  38 int
  39 main (void)
  40 {
  41   extern char _text, _end, __bss_start;
  42   struct thread *t;
  43
  44   /* Clear out the BSS segment. */
  45   memset (&__bss_start, 0, &_end - &__bss_start);
  46
  47   vga_init ();
  48   serial_init ();
  49
  50   /* Calculate how much RAM the kernel uses, and find out from
  51      the bootloader how much RAM this machine has. */
  52   kernel_pages = (&_end - &_text + 4095) / 4096;
  53   ram_pages = *(uint32_t *) (0x7e00 - 8);
  54
  55   printk ("Initializing nachos-x86, %d kB RAM detected.\n",
  56           ram_pages * 4);
  57
  58   /* Memory from the end of the kernel through the end of memory
  59      is free.  Give it to the page allocator. */
  60   palloc_init ((void *) (KERN_BASE + kernel_pages * PGSIZE),
  61                (void *) (PHYS_BASE + ram_pages * PGSIZE));
  62   paging_init ();
  63   gdt_init ();
  64
  65   malloc_init ();
  66   random_init ();
  67
  68   intr_init ();
  69   timer_init ();
  70   kbd_init ();
  71
  72 #ifdef FILESYS
  73   filesys_init (false);
  74   filesys_self_test ();
  75 #endif
  76
  77   thread_init ();
  78
  79   t = thread_create ("main", main_thread, NULL);
  80   thread_start (t);
  81
  82   printk ("Done!\n");
  83   return 0;
  84 }
  85
  86 static uint64_t
  87 make_seg_desc (uint32_t base,
  88                uint32_t limit,
  89                enum seg_system system,
  90                enum seg_type type,
  91                int dpl,
  92                enum seg_granularity granularity)
  93 {
  94   uint32_t e0 = ((limit & 0xffff)             /* Limit 15:0. */
  95                  | (base << 16));             /* Base 15:0. */
  96   uint32_t e1 = (((base >> 16) & 0xff)        /* Base 23:16. */
  97                  | (system << 12)             /* 0=system, 1=code/data. */
  98                  | (type << 8)                /* Segment type. */
  99                  | (dpl << 13)                /* Descriptor privilege. */
 100                  | (1 << 15)                  /* Present. */
 101                  | (limit & 0xf0000)          /* Limit 16:19. */
 102                  | (1 << 22)                  /* 32-bit segment. */
 103                  | (granularity << 23)        /* Byte/page granularity. */
 104                  | (base & 0xff000000));      /* Base 31:24. */
 105   return e0 | ((uint64_t) e1 << 32);
 106 }
 107
 108 static uint64_t
 109 make_code_desc (int dpl)
 110 {
 111   return make_seg_desc (0, 0xfffff, SYS_CODE_DATA, TYPE_CODE | TYPE_READABLE,
 112                         dpl, GRAN_PAGE);
 113 }
 114
 115 static uint64_t
 116 make_data_desc (int dpl)
 117 {
 118   return make_seg_desc (0, 0xfffff, SYS_CODE_DATA, TYPE_WRITABLE,
 119                         dpl, GRAN_PAGE);
 120 }
 121
 122 static uint64_t
 123 make_tss_desc (void *vaddr)
 124 {
 125   return make_seg_desc ((uint32_t) vaddr,
 126                         0x67, SYS_SYSTEM, TYPE_TSS_32_A, 0, GRAN_BYTE);
 127 }
 128
 129 uint64_t gdt[SEL_CNT];
 130
 131 struct tss *tss;
 132
 133 /* Sets up a proper GDT.  The bootstrap loader's GDT didn't
 134    include user-mode selectors or a TSS. */
 135 static void
 136 gdt_init (void)
 137 {
 138   uint64_t gdtr_operand;
 139
 140   /* Our TSS is never used in a call gate or task gate, so only a
 141      few fields of it are ever referenced, and those are the only
 142      ones we initialize. */
 143   tss = palloc_get (PAL_ASSERT | PAL_ZERO);
 144   tss->esp0 = (uint32_t) ptov(0x20000);
 145   tss->ss0 = SEL_KDSEG;
 146   tss->bitmap = 0xdfff;
 147
 148   /* Initialize GDT. */
 149   gdt[SEL_NULL / sizeof *gdt] = 0;
 150   gdt[SEL_KCSEG / sizeof *gdt] = make_code_desc (0);
 151   gdt[SEL_KDSEG / sizeof *gdt] = make_data_desc (0);
 152   gdt[SEL_UCSEG / sizeof *gdt] = make_code_desc (3);
 153   gdt[SEL_UDSEG / sizeof *gdt] = make_data_desc (3);
 154   gdt[SEL_TSS / sizeof *gdt] = make_tss_desc (tss);
 155
 156   /* Load GDTR, TR. */
 157   gdtr_operand = make_dtr_operand (sizeof gdt - 1, gdt);
 158   asm volatile ("lgdt %0" :: "m" (gdtr_operand));
 159   asm volatile ("ltr %w0" :: "r" (SEL_TSS));
 160 }
 161 \f
 162 void
 163 power_off (void)
 164 {
 165   const char s[] = "Shutdown";
 166   const char *p;
 167
 168   printk ("Powering off...\n");
 169   for (p = s; *p != '\0'; p++)
 170     outb (0x8900, *p);
 171   for (;;);
 172 }