X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fthreads%2Finit.c;h=589661619b056a2eb40643f2b746da5a9daff15e;hb=7a3dff52c8a44deeadd071ea93f19b9cee2a67fa;hp=49b26a9bae3b636daf3bdea790c3b19b7367ab70;hpb=5539377e713d0b7424a15547ca0242518e86ea9b;p=pintos-anon diff --git a/src/threads/init.c b/src/threads/init.c index 49b26a9..5896616 100644 --- a/src/threads/init.c +++ b/src/threads/init.c @@ -17,11 +17,11 @@ #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/process.h" #include "userprog/exception.h" #include "userprog/gdt.h" #include "userprog/syscall.h" @@ -36,49 +36,59 @@ /* Amount of physical memory, in 4 kB pages. */ size_t ram_pages; +/* Page directory with kernel mappings only. */ +uint32_t *base_page_dir; + #ifdef FILESYS -/* Format the filesystem? */ +/* -f: Format the filesystem? */ static bool format_filesys; #endif #ifdef USERPROG -/* Initial program to run. */ +/* -ex: Initial program to run. */ static char *initial_program; #endif -/* Power off after running requested actions? */ -static bool power_off; +/* -q: Power off after kernel tasks complete? */ +bool power_off_when_done; static void ram_init (void); +static void paging_init (void); static void argv_init (void); -static void do_power_off (void); +static void print_stats (void); int main (void) NO_RETURN; int main (void) { - /* Initialize everything needed for printf() first. */ + /* Clear BSS and get machine's RAM size. */ ram_init (); + + /* Initialize ourselves as a thread so we can use locks. */ thread_init (); + + /* Initialize the console so we can use printf(). */ vga_init (); serial_init_poll (); console_init (); /* Greet user. */ - printf ("Pintos booting with %'zd kB RAM...\n", ram_pages * (PGSIZE / 1024)); + printf ("Pintos booting with %'zu kB RAM...\n", ram_pages * PGSIZE / 1024); /* Parse command line. */ argv_init (); - /* Initialize memory system, segments, paging. */ + /* Initialize memory system. */ palloc_init (); + malloc_init (); paging_init (); + + /* Segmentation. */ #ifdef USERPROG tss_init (); gdt_init (); #endif - malloc_init (); /* Set random seed if argv_init() didn't. */ random_init (0); @@ -95,6 +105,7 @@ main (void) /* Start thread scheduler and enable interrupts. */ thread_start (); serial_init_queue (); + timer_calibrate (); #ifdef FILESYS /* Initialize filesystem. */ @@ -110,17 +121,18 @@ main (void) if (initial_program != NULL) { printf ("\nExecuting '%s':\n", initial_program); - thread_execute (initial_program); + process_wait (process_execute (initial_program)); } #else + /* Run the compiled-in test function. */ test (); #endif - if (power_off) - do_power_off (); - - /* Terminate this thread. */ - thread_exit (); + /* Finish up. */ + if (power_off_when_done) + power_off (); + else + thread_exit (); } /* Clear BSS and obtain RAM size from loader. */ @@ -140,6 +152,46 @@ ram_init (void) ram_pages = *(uint32_t *) ptov (LOADER_RAM_PAGES); } +/* Populates the base page directory and page table with the + kernel virtual mapping, and then sets up the CPU to use the + new page directory. Points base_page_dir to the page + directory it creates. + + At the time this function is called, the active page table + (set up by loader.S) only maps the first 4 MB of RAM, so we + should not try to use extravagant amounts of memory. + Fortunately, there is no need to do so. */ +static void +paging_init (void) +{ + uint32_t *pd, *pt; + size_t page; + + pd = base_page_dir = palloc_get_page (PAL_ASSERT | PAL_ZERO); + pt = NULL; + for (page = 0; page < ram_pages; page++) + { + uintptr_t paddr = page * PGSIZE; + void *vaddr = ptov (paddr); + size_t pde_idx = pd_no (vaddr); + size_t pte_idx = pt_no (vaddr); + + if (pd[pde_idx] == 0) + { + pt = palloc_get_page (PAL_ASSERT | PAL_ZERO); + pd[pde_idx] = pde_create (pt); + } + + pt[pte_idx] = pte_create_kernel (vaddr, true); + } + + /* Store the physical address of the page directory into CR3 + aka PDBR (page directory base register). This activates our + new page tables immediately. See [IA32-v2a] "MOV--Move + to/from Control Registers" and [IA32-v3] 3.7.5. */ + asm volatile ("mov %%cr3, %0" :: "r" (vtop (base_page_dir))); +} + /* Parses the command line. */ static void argv_init (void) @@ -170,16 +222,16 @@ argv_init (void) 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; + power_off_when_done = true; #ifdef USERPROG else if (!strcmp (argv[i], "-ex")) initial_program = argv[++i]; + else if (!strcmp (argv[i], "-ul")) + user_page_limit = atoi (argv[++i]); #endif #ifdef FILESYS - else if (!strcmp (argv[i], "-f")) + else if (!strcmp (argv[i], "-f")) format_filesys = true; else if (!strcmp (argv[i], "-ci")) { @@ -201,10 +253,10 @@ argv_init (void) { printf ( "Kernel options:\n" - " -rs SEED Seed random seed to SEED.\n" - " -d CLASS[,...] Enable the given classes of debug messages.\n" + " -rs SEED Set random seed to SEED.\n" #ifdef USERPROG " -ex 'PROG [ARG...]' Run PROG, passing the optional arguments.\n" + " -ul USER_MAX Limit user memory to USER_MAX pages.\n" #endif #ifdef FILESYS " -f Format the filesystem disk (hdb or hd0:1).\n" @@ -220,14 +272,16 @@ argv_init (void) " -q Power off after doing requested actions.\n" " -u Print this help message and power off.\n" ); - do_power_off (); + power_off (); } else PANIC ("unknown option `%s' (use -u for help)", argv[i]); } +/* Powers down the machine we're running on, + as long as we're running on Bochs or qemu. */ void -do_power_off (void) +power_off (void) { const char s[] = "Shutdown"; const char *p; @@ -236,8 +290,28 @@ do_power_off (void) filesys_done (); #endif + print_stats (); + printf ("Powering off...\n"); + serial_flush (); + for (p = s; *p != '\0'; p++) outb (0x8900, *p); for (;;); } + +/* Print statistics about Pintos execution. */ +static void +print_stats (void) +{ + timer_print_stats (); + thread_print_stats (); +#ifdef FILESYS + disk_print_stats (); +#endif + console_print_stats (); + kbd_print_stats (); +#ifdef USERPROG + exception_print_stats (); +#endif +}