X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fthreads%2Finit.c;h=855afa0340fcb66788aa03a00cc70666b17bbbae;hb=24445a35e861ddb50df05731ca4635fbd36e04a4;hp=dbfd56be94be1af05439bbdaed05784a27581214;hpb=b0a700d18f0a0a8c87e1a4fff3a2108e0edb0fbc;p=pintos-anon

diff --git a/src/threads/init.c b/src/threads/init.c
index dbfd56b..855afa0 100644
--- a/src/threads/init.c
+++ b/src/threads/init.c
@@ -21,6 +21,7 @@
 #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"
@@ -39,22 +40,21 @@ size_t ram_pages;
 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 running requested actions? */
+static bool do_power_off;
 
 static void ram_init (void);
 static void paging_init (void);
 static void argv_init (void);
-static void do_power_off (void);
 
 int main (void) NO_RETURN;
 
@@ -112,15 +112,20 @@ main (void)
   /* Run a user program. */
   if (initial_program != NULL)
     {
+      tid_t tid;
       printf ("\nExecuting '%s':\n", initial_program);
-      thread_execute (initial_program); 
+      tid = process_execute (initial_program);
+#ifdef THREAD_JOIN_IMPLEMENTED
+      if (tid != TID_ERROR)
+        thread_join (tid);
+#endif
     }
 #else
   test ();
 #endif
 
-  if (power_off) 
-    do_power_off ();
+  if (do_power_off) 
+    power_off ();
 
   /* Terminate this thread. */
   thread_exit ();
@@ -212,7 +217,7 @@ argv_init (void)
     else if (!strcmp (argv[i], "-d")) 
       debug_enable (argv[++i]);
     else if (!strcmp (argv[i], "-q"))
-      power_off = true;
+      do_power_off = true;
 #ifdef USERPROG
     else if (!strcmp (argv[i], "-ex")) 
       initial_program = argv[++i];
@@ -259,7 +264,7 @@ 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]);
@@ -268,7 +273,7 @@ argv_init (void)
 /* 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;
@@ -278,6 +283,8 @@ do_power_off (void)
 #endif
 
   printf ("Powering off...\n");
+  serial_flush ();
+
   for (p = s; *p != '\0'; p++)
     outb (0x8900, *p);
   for (;;);