Add -u to usage message.
[pintos-anon] / src / threads / init.c
index 79cd3d0309bcb7823c3aa2ef7d48055782df1047..49b26a9bae3b636daf3bdea790c3b19b7367ab70 100644 (file)
-#include "init.h"
-#include <stdint.h>
-#include <stddef.h>
+#include "threads/init.h"
+#include <console.h>
+#include <debug.h>
 #include <limits.h>
-#include "debug.h"
-#include "interrupt.h"
-#include "io.h"
-#include "kbd.h"
-#include "lib.h"
-#include "loader.h"
-#include "malloc.h"
-#include "mmu.h"
-#include "paging.h"
-#include "palloc.h"
-#include "random.h"
-#include "serial.h"
-#include "thread.h"
-#include "timer.h"
-#include "vga.h"
+#include <random.h>
+#include <stddef.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include "devices/kbd.h"
+#include "devices/serial.h"
+#include "devices/timer.h"
+#include "devices/vga.h"
+#include "threads/interrupt.h"
+#include "threads/io.h"
+#include "threads/loader.h"
+#include "threads/malloc.h"
+#include "threads/mmu.h"
+#include "threads/paging.h"
+#include "threads/palloc.h"
+#include "threads/test.h"
+#include "threads/thread.h"
+#ifdef USERPROG
+#include "userprog/exception.h"
+#include "userprog/gdt.h"
+#include "userprog/syscall.h"
+#include "userprog/tss.h"
+#endif
 #ifdef FILESYS
-#include "filesys.h"
-#include "disk.h"
+#include "devices/disk.h"
+#include "filesys/filesys.h"
+#include "filesys/fsutil.h"
 #endif
 
-/* Size of kernel static code and data, in 4 kB pages. */
-size_t kernel_pages;
-
 /* Amount of physical memory, in 4 kB pages. */
 size_t ram_pages;
 
+#ifdef FILESYS
+/* Format the filesystem? */
+static bool format_filesys;
+#endif
+
+#ifdef USERPROG
+/* Initial program to run. */
+static char *initial_program;
+#endif
+
+/* Power off after running requested actions? */
+static bool power_off;
+
 static void ram_init (void);
-static void gdt_init (void);
 static void argv_init (void);
-void power_off (void);
+static void do_power_off (void);
 
-static void
-main_thread (void *aux UNUSED) 
-{
-#ifdef FILESYS
-  disk_init ();
-  filesys_init (true);
-  filesys_self_test ();
-#endif
-}
+int main (void) NO_RETURN;
 
 int
 main (void)
 {
-  struct thread *t;
-
-  /* Initialize components needed by printk() very early. */
+  /* Initialize everything needed for printf() first. */
   ram_init ();
+  thread_init ();
   vga_init ();
-  serial_init ();
-  printk ("Booting cnachos86...\n");
+  serial_init_poll ();
+  console_init ();
+
+  /* Greet user. */
+  printf ("Pintos booting with %'zd kB RAM...\n", ram_pages * (PGSIZE / 1024));
 
-  /* Calculate how much RAM the kernel uses, and find out from
-     the bootloader how much RAM this machine has. */
-  printk ("ram: detected %'d kB main memory.\n", ram_pages * 4);
+  /* Parse command line. */
+  argv_init ();
 
-  /* Memory from the end of the kernel through the end of memory
-     is free.  Give it to the page allocator. */
-  palloc_init (ptov (LOADER_KERN_BASE + kernel_pages * PGSIZE),
-               ptov (ram_pages * PGSIZE));
+  /* Initialize memory system, segments, paging. */
+  palloc_init ();
   paging_init ();
+#ifdef USERPROG
+  tss_init ();
   gdt_init ();
-
+#endif
   malloc_init ();
-  random_init ();
 
-  argv_init ();
+  /* Set random seed if argv_init() didn't. */
+  random_init (0);
 
+  /* Initialize interrupt handlers. */
   intr_init ();
   timer_init ();
   kbd_init ();
+#ifdef USERPROG
+  exception_init ();
+  syscall_init ();
+#endif
 
-  thread_init ();
-
-  t = thread_create ("main", main_thread, NULL);
-  thread_start (t);
-
-  printk ("Done!\n");
-  return 0;
-}
-
-static uint64_t
-make_seg_desc (uint32_t base,
-               uint32_t limit,
-               enum seg_system system,
-               enum seg_type type,
-               int dpl,
-               enum seg_granularity granularity)
-{
-  uint32_t e0 = ((limit & 0xffff)             /* Limit 15:0. */
-                 | (base << 16));             /* Base 15:0. */
-  uint32_t e1 = (((base >> 16) & 0xff)        /* Base 23:16. */
-                 | (system << 12)             /* 0=system, 1=code/data. */
-                 | (type << 8)                /* Segment type. */
-                 | (dpl << 13)                /* Descriptor privilege. */
-                 | (1 << 15)                  /* Present. */
-                 | (limit & 0xf0000)          /* Limit 16:19. */
-                 | (1 << 22)                  /* 32-bit segment. */
-                 | (granularity << 23)        /* Byte/page granularity. */
-                 | (base & 0xff000000));      /* Base 31:24. */
-  return e0 | ((uint64_t) e1 << 32);
-}
-
-static uint64_t
-make_code_desc (int dpl)
-{
-  return make_seg_desc (0, 0xfffff, SYS_CODE_DATA, TYPE_CODE | TYPE_READABLE,
-                        dpl, GRAN_PAGE);
-}
+  /* Start thread scheduler and enable interrupts. */
+  thread_start ();
+  serial_init_queue ();
 
-static uint64_t
-make_data_desc (int dpl)
-{
-  return make_seg_desc (0, 0xfffff, SYS_CODE_DATA, TYPE_WRITABLE,
-                        dpl, GRAN_PAGE);
-}
-
-static uint64_t
-make_tss_desc (void *vaddr)
-{
-  return make_seg_desc ((uint32_t) vaddr,
-                        0x67, SYS_SYSTEM, TYPE_TSS_32_A, 0, GRAN_BYTE);
-}
+#ifdef FILESYS
+  /* Initialize filesystem. */
+  disk_init ();
+  filesys_init (format_filesys);
+  fsutil_run ();
+#endif
 
-uint64_t gdt[SEL_CNT];
+  printf ("Boot complete.\n");
+  
+#ifdef USERPROG
+  /* Run a user program. */
+  if (initial_program != NULL)
+    {
+      printf ("\nExecuting '%s':\n", initial_program);
+      thread_execute (initial_program); 
+    }
+#else
+  test ();
+#endif
 
-struct tss *tss;
+  if (power_off) 
+    do_power_off ();
 
-/* Sets up a proper GDT.  The bootstrap loader's GDT didn't
-   include user-mode selectors or a TSS. */
-static void
-gdt_init (void)
-{
-  uint64_t gdtr_operand;
-
-  /* Our TSS is never used in a call gate or task gate, so only a
-     few fields of it are ever referenced, and those are the only
-     ones we initialize. */
-  tss = palloc_get (PAL_ASSERT | PAL_ZERO);
-  tss->esp0 = (uint32_t) ptov(0x20000);
-  tss->ss0 = SEL_KDSEG;
-  tss->bitmap = 0xdfff;
-
-  /* Initialize GDT. */
-  gdt[SEL_NULL / sizeof *gdt] = 0;
-  gdt[SEL_KCSEG / sizeof *gdt] = make_code_desc (0);
-  gdt[SEL_KDSEG / sizeof *gdt] = make_data_desc (0);
-  gdt[SEL_UCSEG / sizeof *gdt] = make_code_desc (3);
-  gdt[SEL_UDSEG / sizeof *gdt] = make_data_desc (3);
-  gdt[SEL_TSS / sizeof *gdt] = make_tss_desc (tss);
-
-  /* Load GDTR, TR. */
-  gdtr_operand = make_dtr_operand (sizeof gdt - 1, gdt);
-  asm volatile ("lgdt %0" :: "m" (gdtr_operand));
-  asm volatile ("ltr %w0" :: "r" (SEL_TSS));
+  /* Terminate this thread. */
+  thread_exit ();
 }
-
+\f
+/* Clear BSS and obtain RAM size from loader. */
 static void
 ram_init (void) 
 {
-  /* Start and end of kernel image,
-     and start and end of BSS segment.
-     These are created by kernel.lds. */
-  extern char _start, _end;
-  extern char _start_bss, _end_bss;
-
   /* The "BSS" is a segment that should be initialized to zeros.
      It isn't actually stored on disk or zeroed by the kernel
-     loader, so we have to zero it ourselves. */
+     loader, so we have to zero it ourselves.
+
+     The start and end of the BSS segment is recorded by the
+     linker as _start_bss and _end_bss.  See kernel.lds. */
+  extern char _start_bss, _end_bss;
   memset (&_start_bss, 0, &_end_bss - &_start_bss);
 
-  /* Calculate how much RAM the kernel uses,
-     and find out from the bootloader how much RAM this machine
-     has. */
-  kernel_pages = (&_end - &_start + 4095) / 4096;
-  ram_pages = *(uint32_t *) ptov (LOADER_BASE + LOADER_RAM_PAGES);
+  /* Get RAM size from loader.  See loader.S. */
+  ram_pages = *(uint32_t *) ptov (LOADER_RAM_PAGES);
 }
 
-void
+/* Parses the command line. */
+static void
 argv_init (void) 
 {
-  char *cmd_line = ptov (LOADER_BASE + LOADER_CMD_LINE);
+  char *cmd_line, *pos;
+  char *argv[LOADER_CMD_LINE_LEN / 2 + 2];
+  int argc = 0;
+  int i;
+
+  /* The command line is made up of null terminated strings
+     followed by an empty string.  Break it up into words. */
+  cmd_line = pos = ptov (LOADER_CMD_LINE);
+  printf ("Kernel command line:");
+  while (pos < cmd_line + LOADER_CMD_LINE_LEN)
+    {
+      ASSERT (argc < LOADER_CMD_LINE_LEN / 2);
+      if (*pos == '\0')
+        break;
+      argv[argc++] = pos;
+      printf (" %s", pos);
+      pos = strchr (pos, '\0') + 1;
+    }
+  printf ("\n");
+  argv[argc] = "";
+  argv[argc + 1] = "";
+
+  /* Parse the words. */
+  for (i = 0; i < argc; i++)
+    if (!strcmp (argv[i], "-rs")) 
+      random_init (atoi (argv[++i]));
+    else if (!strcmp (argv[i], "-d")) 
+      debug_enable (argv[++i]);
+    else if (!strcmp (argv[i], "-q"))
+      power_off = true;
+#ifdef USERPROG
+    else if (!strcmp (argv[i], "-ex")) 
+      initial_program = argv[++i];
+#endif
+#ifdef FILESYS
+  else if (!strcmp (argv[i], "-f"))
+      format_filesys = true;
+    else if (!strcmp (argv[i], "-ci")) 
+      {
+        fsutil_copyin_file = argv[++i]; 
+        fsutil_copyin_size = atoi (argv[++i]); 
+      }
+    else if (!strcmp (argv[i], "-co"))
+      fsutil_copyout_file = argv[++i];
+    else if (!strcmp (argv[i], "-p")) 
+      fsutil_print_file = argv[++i];
+    else if (!strcmp (argv[i], "-r"))
+      fsutil_remove_file = argv[++i];
+    else if (!strcmp (argv[i], "-ls"))
+      fsutil_list_files = true;
+    else if (!strcmp (argv[i], "-D"))
+      fsutil_dump_filesys = true;
+#endif
+    else if (!strcmp (argv[i], "-u"))
+      {
+        printf (
+          "Kernel options:\n"
+          " -rs SEED            Seed random seed to SEED.\n"
+          " -d CLASS[,...]      Enable the given classes of debug messages.\n"
+#ifdef USERPROG
+          " -ex 'PROG [ARG...]' Run PROG, passing the optional arguments.\n"
+#endif
+#ifdef FILESYS
+          " -f                  Format the filesystem disk (hdb or hd0:1).\n"
+          " -ci FILENAME SIZE   Copy SIZE bytes from the scratch disk (hdc\n"
+          "                     or hd1:0) into the filesystem as FILENAME\n"
+          " -co FILENAME        Copy FILENAME to the scratch disk, with\n"
+          "                     size at start of sector 0 and data afterward\n"
+          " -p FILENAME         Print the contents of FILENAME\n"
+          " -r FILENAME         Delete FILENAME\n"
+          " -ls                 List the files in the filesystem\n"
+          " -D                  Dump complete filesystem contents\n"
+#endif
+          " -q                  Power off after doing requested actions.\n"
+          " -u                  Print this help message and power off.\n"
+          );
+        do_power_off ();
+      }
+    else 
+      PANIC ("unknown option `%s' (use -u for help)", argv[i]);
 }
-\f
+
 void
-power_off (void) 
+do_power_off (void) 
 {
   const char s[] = "Shutdown";
   const char *p;
 
-  printk ("Powering off...\n");
+#ifdef FILESYS
+  filesys_done ();
+#endif
+
+  printf ("Powering off...\n");
   for (p = s; *p != '\0'; p++)
     outb (0x8900, *p);
   for (;;);