Usermode/Applications/CLIShell_src/main.c

   1 /*\r
   2  AcessOS Shell Version 2\r
   3 - Based on IOOS CLI Shell\r
   4 */\r
   5 #include <acess/sys.h>\r
   6 #include <stdlib.h>\r
   7 #include "header.h"\r
   8 \r
   9 #define _stdin  0\r
  10 #define _stdout 1\r
  11 #define _stderr 2\r
  12 \r
  13 // ==== PROTOTYPES ====\r
  14 char    *ReadCommandLine(int *Length);\r
  15 void    Parse_Args(char *str, char **dest);\r
  16 void    Command_Colour(int argc, char **argv);\r
  17 void    Command_Clear(int argc, char **argv);\r
  18 //void  Command_Ls(int argc, char **argv);\r
  19 void    Command_Cd(int argc, char **argv);\r
  20 //void  Command_Cat(int argc, char **argv);\r
  21 \r
  22 // ==== CONSTANT GLOBALS ====\r
  23 char    *cCOLOUR_NAMES[8] = {"black", "red", "green", "yellow", "blue", "magenta", "cyan", "white"};\r
  24 struct  {\r
  25         char    *name;\r
  26         void    (*fcn)(int argc, char **argv);\r
  27 }       cBUILTINS[] = {\r
  28         {"colour", Command_Colour}, {"clear", Command_Clear}, {"cd", Command_Cd}\r
  29 };\r
  30 #define BUILTIN_COUNT   (sizeof(cBUILTINS)/sizeof(cBUILTINS[0]))\r
  31 \r
  32 // ==== LOCAL VARIABLES ====\r
  33 char    gsCommandBuffer[1024];\r
  34 char    *gsCurrentDirectory = "/";\r
  35 char    gsTmpBuffer[1024];\r
  36 char    **gasCommandHistory;\r
  37  int    giLastCommand = 0;\r
  38  int    giCommandSpace = 0;\r
  39 \r
  40 // ==== CODE ====\r
  41 int main(int argc, char *argv[], char *envp[])\r
  42 {\r
  43         char    *sCommandStr;\r
  44         char    *saArgs[32];\r
  45          int    length = 0;\r
  46          int    pid = -1;\r
  47          int    iArgCount = 0;\r
  48          int    bCached = 1;\r
  49         t_sysFInfo      info;\r
  50         \r
  51         //Command_Clear(0, NULL);\r
  52         \r
  53         write(_stdout, 36, "AcessOS/AcessBasic Shell Version 2\n");\r
  54         write(_stdout, 30, " Based on CLI Shell for IOOS\n");\r
  55         write(_stdout,  2, "\n");\r
  56         for(;;)\r
  57         {\r
  58                 // Free last command & arguments\r
  59                 if(saArgs[0])   free(saArgs);\r
  60                 if(!bCached)    free(sCommandStr);\r
  61                 bCached = 0;\r
  62                 write(_stdout, strlen(gsCurrentDirectory), gsCurrentDirectory);\r
  63                 write(_stdout, 3, "$ ");\r
  64                 \r
  65                 // Read Command line\r
  66                 sCommandStr = ReadCommandLine( &length );\r
  67                 \r
  68                 // Check if the command should be cached\r
  69                 if(gasCommandHistory == NULL || strcmp(sCommandStr, gasCommandHistory[giLastCommand]) != 0)\r
  70                 {\r
  71                         if(giLastCommand >= giCommandSpace) {\r
  72                                 giCommandSpace += 12;\r
  73                                 gasCommandHistory = realloc(gasCommandHistory, giCommandSpace*sizeof(char*));\r
  74                         }\r
  75                         giLastCommand ++;\r
  76                         gasCommandHistory[ giLastCommand ] = sCommandStr;\r
  77                         bCached = 1;\r
  78                 }\r
  79                 \r
  80                 // Parse Command Line into arguments\r
  81                 Parse_Args(sCommandStr, saArgs);\r
  82                 \r
  83                 // Count Arguments\r
  84                 iArgCount = 0;\r
  85                 while(saArgs[iArgCount])        iArgCount++;\r
  86                 \r
  87                 // Silently Ignore all empty commands\r
  88                 if(saArgs[1][0] == '\0')        continue;\r
  89                 \r
  90                 // Check Built-In Commands\r
  91                 //  [HACK] Mem Usage - Use Length in place of `i'\r
  92                 for(length=0;length<BUILTIN_COUNT;length++)\r
  93                 {\r
  94                         if(strcmp(saArgs[1], cBUILTINS[length].name) == 0)\r
  95                         {\r
  96                                 cBUILTINS[length].fcn(iArgCount-1, &saArgs[1]);\r
  97                                 break;\r
  98                         }\r
  99                 }
 100                 // Calling a file\r
 101                 if(length == BUILTIN_COUNT)\r
 102                 {\r
 103                         GeneratePath(saArgs[1], gsCurrentDirectory, gsTmpBuffer);\r
 104                         // Use length in place of fp\r
 105                         length = open(gsTmpBuffer, 0);\r
 106                         // Check file existence\r
 107                         if(length == -1) {\r
 108                                 Print("Unknown Command: `");Print(saArgs[1]);Print("'\n");      // Error Message\r
 109                                 continue;\r
 110                         }\r
 111                         // Check if the file is a directory\r
 112                         finfo( length, &info );
 113                         close( length );\r
 114                         if(info.flags & FILEFLAG_DIRECTORY) {\r
 115                                 Print("`");Print(saArgs[1]);    // Error Message\r
 116                                 Print("' is a directory.\n");\r
 117                                 continue;\r
 118                         }\r
 119                         pid = clone(CLONE_VM, 0);\r
 120                         if(pid == 0)    execve(gsTmpBuffer, &saArgs[1], NULL);\r
 121                         if(pid <= 0) {\r
 122                                 Print("Unablt to create process: `");Print(gsTmpBuffer);Print("'\n");   // Error Message
 123                                 //SysDebug("pid = %i\n", pid);\r
 124                         }\r
 125                         else {\r
 126                                 //waitpid(pid, K_WAITPID_DIE);\r
 127                         }\r
 128                 }\r
 129         }\r
 130 }\r
 131 \r
 132 /**\r
 133  * \fn char *ReadCommandLine(int *Length)\r
 134  * \brief Read from the command line\r
 135  */\r
 136 char *ReadCommandLine(int *Length)\r
 137 {\r
 138         char    *ret;\r
 139          int    len, pos, space = 1023;\r
 140         char    ch;\r
 141          \r
 142         // Preset Variables\r
 143         ret = malloc( space+1 );\r
 144         len = 0;\r
 145         pos = 0;\r
 146                 \r
 147         // Read In Command Line\r
 148         do {\r
 149                 read(_stdin, 1, &ch);   // Read Character from stdin (read is a blocking call)\r
 150                 // Ignore control characters\r
 151                 if(ch < 0)      continue;\r
 152                 // Backspace\r
 153                 if(ch == '\b') {\r
 154                         if(len <= 0)            continue;       // Protect against underflows\r
 155                         if(pos == len) {        // Simple case of end of string\r
 156                                 len --; pos--;\r
 157                         } else {\r
 158                                 memmove(&ret[pos-1], &ret[pos], len-pos);\r
 159                                 pos --;\r
 160                                 len --;\r
 161                         }\r
 162                         write(_stdout, 1, &ch);\r
 163                         continue;\r
 164                 }\r
 165                 // Expand Buffer\r
 166                 if(len > space) {\r
 167                         space += 256;\r
 168                         ret = realloc(ret, space+1);\r
 169                         if(!ret)        return NULL;\r
 170                 }\r
 171                 \r
 172                 write(_stdout, 1, &ch);\r
 173                 ret[pos++] = ch;\r
 174                 len ++;\r
 175         } while(ch != '\n');\r
 176         \r
 177         // Remove newline\r
 178         pos --;\r
 179         ret[pos] = '\0';\r
 180         \r
 181         // Return length\r
 182         if(Length)      *Length = len;\r
 183         \r
 184         return ret;\r
 185 }\r
 186 \r
 187 /**\r
 188  * \fn void Parse_Args(char *str, char **dest)\r
 189  * \brief Parse a string into an argument array\r
 190  */\r
 191 void Parse_Args(char *str, char **dest)\r
 192 {\r
 193          int    i = 1;\r
 194         char    *buf = malloc( strlen(str) + 1 );\r
 195         \r
 196         strcpy(buf, str);\r
 197         dest[0] = buf;\r
 198         \r
 199         // Trim leading whitespace\r
 200         while(*buf == ' ' && *buf)      buf++;\r
 201         \r
 202         for(;;)\r
 203         {\r
 204                 dest[i] = buf;  // Save start of string\r
 205                 i++;\r
 206                 while(*buf && *buf != ' ')\r
 207                 {\r
 208                         if(*buf++ == '"')\r
 209                         {\r
 210                                 while(*buf && !(*buf == '"' && buf[-1] != '\\'))\r
 211                                         buf++;\r
 212                         }\r
 213                 }\r
 214                 if(*buf == '\0')        break;\r
 215                 *buf = '\0';\r
 216                 while(*++buf == ' ' && *buf);\r
 217                 if(*buf == '\0')        break;\r
 218         }\r
 219         dest[i] = NULL;\r
 220         if(i == 1) {\r
 221                 free(buf);\r
 222                 dest[0] = NULL;\r
 223         }\r
 224 }\r
 225 \r
 226 /**\r
 227  * \fn void Command_Colour(int argc, char **argv)\r
 228  * \brief \r
 229  */\r
 230 void Command_Colour(int argc, char **argv)\r
 231 {\r
 232         int fg, bg;\r
 233         char    clrStr[6] = "\x1B[37m";\r
 234         \r
 235         // Verify Arg Count\r
 236         if(argc < 2)\r
 237         {\r
 238                 Print("Please specify a colour\n");\r
 239                 goto usage;\r
 240         }\r
 241         \r
 242         // Check Colour\r
 243         for(fg=0;fg<8;fg++)\r
 244                 if(strcmp(cCOLOUR_NAMES[fg], argv[1]) == 0)\r
 245                         break;\r
 246 \r
 247         // Foreground a valid colour\r
 248         if(fg == 8) {\r
 249                 Print("Unknown Colour '");Print(argv[1]);Print("'\n");\r
 250                 goto usage;\r
 251         }\r
 252         // Set Foreground\r
 253         clrStr[3] = '0' + fg;\r
 254         write(_stdout, 6, clrStr);\r
 255         \r
 256         // Need to Set Background?\r
 257         if(argc > 2)\r
 258         {\r
 259                 for(bg=0;bg<8;bg++)\r
 260                         if(strcmp(cCOLOUR_NAMES[bg], argv[2]) == 0)\r
 261                                 break;\r
 262         \r
 263                 // Valid colour\r
 264                 if(bg == 8)\r
 265                 {\r
 266                         Print("Unknown Colour '");Print(argv[2]);Print("'\n");\r
 267                         goto usage;\r
 268                 }\r
 269         \r
 270                 clrStr[2] = '4';\r
 271                 clrStr[3] = '0' + bg;\r
 272                 write(_stdout, 6, clrStr);\r
 273         }\r
 274         // Return\r
 275         return;\r
 276 \r
 277         // Function Usage (Requested via a Goto (I know it's ugly))\r
 278 usage:\r
 279         Print("Valid Colours are ");\r
 280         for(fg=0;fg<8;fg++)\r
 281         {\r
 282                 Print(cCOLOUR_NAMES[fg]);\r
 283                 write(_stdout, 3, ", ");\r
 284         }\r
 285         write(_stdout, 4, "\b\b\n");\r
 286         return;\r
 287 }\r
 288 \r
 289 void Command_Clear(int argc, char **argv)\r
 290 {\r
 291         write(_stdout, 4, "\x1B[2J");   //Clear Screen\r
 292 }\r
 293 \r
 294 void Command_Cd(int argc, char **argv)\r
 295 {\r
 296         char    tmpPath[1024];\r
 297         int             fp;\r
 298         t_sysFInfo      stats;\r
 299         \r
 300         if(argc < 2)\r
 301         {\r
 302                 Print(gsCurrentDirectory);Print("\n");\r
 303                 return;\r
 304         }\r
 305         \r
 306         GeneratePath(argv[1], gsCurrentDirectory, tmpPath);\r
 307         \r
 308         fp = open(tmpPath, 0);\r
 309         if(fp == -1) {\r
 310                 write(_stdout, 26, "Directory does not exist\n");\r
 311                 return;\r
 312         }\r
 313         finfo(fp, &stats);\r
 314         close(fp);\r
 315         \r
 316         if(!(stats.flags & FILEFLAG_DIRECTORY)) {\r
 317                 write(_stdout, 17, "Not a Directory\n");\r
 318                 return;\r
 319         }\r
 320         \r
 321         strcpy(gsCurrentDirectory, tmpPath);\r
 322 }\r
 323 \r