/[cvs]/joko/Uni/BSArch/03/win32/min_shell.c
ViewVC logotype

Diff of /joko/Uni/BSArch/03/win32/min_shell.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.5 by joko, Thu Jun 15 22:15:10 2006 UTC revision 1.13 by joko, Fri Jun 16 00:47:07 2006 UTC
# Line 3  Line 3 
3  #include <stdlib.h>  #include <stdlib.h>
4  #include <stdio.h>  #include <stdio.h>
5  #include <errno.h>  #include <errno.h>
6  #include <windows.h>  #include <string.h>
7    
8  #include "min_shell.h"  #include "min_shell.h"
9    
 #define BOOL int  
 #define TRUE 1  
 #define FALSE 0  
   
   
 #define MAX_COMMAND_LENGTH 1024  
   
10  int main(int argc, char * argv[]) {  int main(int argc, char * argv[]) {
11      
12      // interactive mode with prompt
13    if (argc == 1) {    if (argc == 1) {
14      //printf("No arguments given.");      fprintf(stderr, "Welcome to mini shell. Have fun!\n");
15      //exit(EXIT_FAILURE);      run_commands(stdin, "> ");
     char filename[255] = "abc\0";  
     BOOL background = 0;  
     start_process(filename, background);  
16    }    }
17        
18      // scripting mode
19    if (argc == 2) {    if (argc == 2) {
20      FILE * script = fopen(argv[1], "r");      FILE * script = fopen(argv[1], "r");
21      read_commands(script);      run_commands(script, "");
22      fclose(script);      fclose(script);
23    }    }
24  }  }
25    
26  void read_commands(FILE * fp) {  void run_commands(FILE * fp, char * prompt) {
27        
28    char command[MAX_COMMAND_LENGTH];    char command[MAX_COMMAND_LENGTH];
   int command_length;  
29        
30    fgets(command, MAX_COMMAND_LENGTH, fp);    // echo first prompt
31    command_length = strlen(command);    fprintf(stdout, "%s", prompt);
32        while (fgets(command, MAX_COMMAND_LENGTH, fp)) {
33    /* if last char is newline, strip it */      int command_length;
34    if (command[command_length-1] == '\n') {      
35      command[command_length-1] = '\0';      // strip newlines
36        chomp(command);
37        
38        // ... and action
39        dispatch_command(command);
40        
41        // echo prompt after executing shell command
42        fprintf(stdout, "%s", prompt);
43    }    }
44        
   process_command(command);  
45  }  }
46    
 void process_command(char * command) {  
   printf("process_command: %s\n", command);  
   start_process(command, 0);  
 }  
47    
48  BOOL start_process(char * filename, BOOL background) {  void dispatch_command(char * command) {
49    
50    STARTUPINFO si;    BOOL background = FALSE;
51    PROCESS_INFORMATION pi;    int command_length;
52      char first_char;
53      char * shell_command;
54    
55    //printf("start_process: '%s'\n", filename);    // is it a shell command?
56      first_char = command[0];
57      if (strcmp(&first_char, ":") == 0) {
58        
59        // "calculate" shell command ...
60        shell_command = command + 1;
61        //printf("shell command: %s\n", shell_command);
62        
63        // ... and dispatch it
64        
65        // wait
66        if (strcmp(shell_command, "wait") == 0) {
67          // TODO
68          os_wait_for_processes();
69        
70        // exit
71        } else if (strcmp(shell_command, "exit") == 0) {
72          exit(EXIT_SUCCESS);
73        
74        // unknown command
75        } else {
76          fprintf(stderr, "ERROR: Unknown shell command '%s'\n", shell_command);
77        }
78        
79    ZeroMemory(&si,sizeof(si));    // run program: background or foreground?
80    si.cb = sizeof(STARTUPINFO);    } else {
81          //printf("running: %s\n", command);
82    ZeroMemory(&pi,sizeof(pi));      command_length = strlen(command);
83          if (command[command_length-1] == '&') {
84    CreateProcess(        command[command_length-1] = '\0';
85      NULL, filename,        background = TRUE;
     NULL, NULL, FALSE, 0,  
     NULL, NULL,  
     &si,  
     &pi  
   );  
   check_alert_error("CreateProcess");  
     
   if (!background) {  
     long status;  
     status = WaitForSingleObject(pi.hProcess, INFINITE);  
     if (status == WAIT_FAILED) {  
       check_alert_error("WaitForSingleObject");  
       return FALSE;  
86      }      }
87        
88        os_start_process(command, background);
89    }    }
     
   return TRUE;  
   
90  }  }
91    
 void check_alert_error(const char * error_source) {  
   DWORD dwError = GetLastError();  
   
   HLOCAL message = NULL;  
   BOOL fOk;  
     
   if (dwError == 0)  
     return;  
   
   fOk = FormatMessage(  
     FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,  
     NULL,  
     dwError,  
     MAKELANGID(LANG_NEUTRAL, SUBLANG_SYS_DEFAULT),  
     (PTSTR)&message,  
     0,  
     NULL  
   );  
   
   if (message != NULL) {  
     fprintf(stderr, "Error with '%s': %s (Code %i)\n", error_source, message, dwError);  
     LocalFree(message);  
   }  
     
 }  

Legend:
Removed from v.1.5  
changed lines
  Added in v.1.13

MailToCvsAdmin">MailToCvsAdmin
ViewVC Help
Powered by ViewVC 1.1.26 RSS 2.0 feed