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_Cd(int argc, char **argv);\r
  19 void    Command_Dir(int argc, char **argv);\r
  20 \r
  21 // ==== CONSTANT GLOBALS ====\r
  22 char    *cCOLOUR_NAMES[8] = {"black", "red", "green", "yellow", "blue", "magenta", "cyan", "white"};\r
  23 struct  {\r
  24         char    *name;\r
  25         void    (*fcn)(int argc, char **argv);\r
  26 }       cBUILTINS[] = {\r
  27         {"colour", Command_Colour}, {"clear", Command_Clear},\r
  28         {"cd", Command_Cd}, {"dir", Command_Dir}\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, 1, "\n");\r
  54         write(_stdout, 36, "Acess Shell Version 3\n");\r
  55         write(_stdout, 30, " Based on CLI Shell for IOOS\n");\r
  56         write(_stdout,  2, "\n");\r
  57         for(;;)\r
  58         {\r
  59                 // Free last command & arguments\r
  60                 if(saArgs[0])   free(saArgs);\r
  61                 if(!bCached)    free(sCommandStr);\r
  62                 bCached = 0;\r
  63                 write(_stdout, strlen(gsCurrentDirectory), gsCurrentDirectory);\r
  64                 write(_stdout, 3, "$ ");\r
  65                 \r
  66                 // Read Command line\r
  67                 sCommandStr = ReadCommandLine( &length );\r
  68                 \r
  69                 // Check if the command should be cached\r
  70                 if(gasCommandHistory == NULL || strcmp(sCommandStr, gasCommandHistory[giLastCommand]) != 0)\r
  71                 {\r
  72                         if(giLastCommand >= giCommandSpace) {\r
  73                                 giCommandSpace += 12;\r
  74                                 gasCommandHistory = realloc(gasCommandHistory, giCommandSpace*sizeof(char*));\r
  75                         }\r
  76                         giLastCommand ++;\r
  77                         gasCommandHistory[ giLastCommand ] = sCommandStr;\r
  78                         bCached = 1;\r
  79                 }\r
  80                 \r
  81                 // Parse Command Line into arguments\r
  82                 Parse_Args(sCommandStr, saArgs);\r
  83                 \r
  84                 // Count Arguments\r
  85                 iArgCount = 0;\r
  86                 while(saArgs[iArgCount])        iArgCount++;\r
  87                 \r
  88                 // Silently Ignore all empty commands\r
  89                 if(saArgs[1][0] == '\0')        continue;\r
  90                 \r
  91                 // Check Built-In Commands\r
  92                 //  [HACK] Mem Usage - Use Length in place of `i'\r
  93                 for(length=0;length<BUILTIN_COUNT;length++)\r
  94                 {\r
  95                         if(strcmp(saArgs[1], cBUILTINS[length].name) == 0)\r
  96                         {\r
  97                                 cBUILTINS[length].fcn(iArgCount-1, &saArgs[1]);\r
  98                                 break;\r
  99                         }\r
 100                 }\r
 101                 \r
 102                 if(length != BUILTIN_COUNT)     continue;\r
 103
 104                 // - Calling a file\r
 105                 GeneratePath(saArgs[1], gsCurrentDirectory, gsTmpBuffer);\r
 106                 // Use length in place of fp\r
 107                 length = open(gsTmpBuffer, 0);\r
 108                 // Check file existence\r
 109                 if(length == -1) {\r
 110                         Print("Unknown Command: `");Print(saArgs[1]);Print("'\n");      // Error Message\r
 111                         continue;\r
 112                 }\r
 113                 // Check if the file is a directory\r
 114                 finfo( length, &info, 0 );
 115                 close( length );\r
 116                 if(info.flags & FILEFLAG_DIRECTORY) {\r
 117                         Print("`");Print(saArgs[1]);    // Error Message\r
 118                         Print("' is a directory.\n");\r
 119                         continue;\r
 120                 }\r
 121                 // Load new executable\r
 122                 pid = clone(CLONE_VM, 0);\r
 123                 if(pid == 0)    execve(gsTmpBuffer, &saArgs[1], envp);\r
 124                 if(pid <= 0) {\r
 125                         Print("Unablt to create process: `");Print(gsTmpBuffer);Print("'\n");   // Error Message
 126                         //SysDebug("pid = %i\n", pid);\r
 127                 }\r
 128                 else {\r
 129                          int    status;\r
 130                         waittid(pid, &status);\r
 131                 }\r
 132         }\r
 133 }\r
 134 \r
 135 /**\r
 136  * \fn char *ReadCommandLine(int *Length)\r
 137  * \brief Read from the command line\r
 138  */\r
 139 char *ReadCommandLine(int *Length)\r
 140 {\r
 141         char    *ret;\r
 142          int    len, pos, space = 1023;\r
 143         char    ch;\r
 144          \r
 145         // Preset Variables\r
 146         ret = malloc( space+1 );\r
 147         len = 0;\r
 148         pos = 0;\r
 149                 \r
 150         // Read In Command Line\r
 151         do {\r
 152                 read(_stdin, 1, &ch);   // Read Character from stdin (read is a blocking call)\r
 153                 // Ignore control characters\r
 154                 if(ch < 0)      continue;\r
 155                 // Backspace\r
 156                 if(ch == '\b') {\r
 157                         if(len <= 0)            continue;       // Protect against underflows\r
 158                         if(pos == len) {        // Simple case of end of string\r
 159                                 len --; pos--;\r
 160                         } else {\r
 161                                 memmove(&ret[pos-1], &ret[pos], len-pos);\r
 162                                 pos --;\r
 163                                 len --;\r
 164                         }\r
 165                         write(_stdout, 1, &ch);\r
 166                         continue;\r
 167                 }\r
 168                 // Expand Buffer\r
 169                 if(len > space) {\r
 170                         space += 256;\r
 171                         ret = realloc(ret, space+1);\r
 172                         if(!ret)        return NULL;\r
 173                 }\r
 174                 \r
 175                 write(_stdout, 1, &ch);\r
 176                 ret[pos++] = ch;\r
 177                 len ++;\r
 178         } while(ch != '\n');\r
 179         \r
 180         // Remove newline\r
 181         pos --;\r
 182         ret[pos] = '\0';\r
 183         \r
 184         // Return length\r
 185         if(Length)      *Length = len;\r
 186         \r
 187         return ret;\r
 188 }\r
 189 \r
 190 /**\r
 191  * \fn void Parse_Args(char *str, char **dest)\r
 192  * \brief Parse a string into an argument array\r
 193  */\r
 194 void Parse_Args(char *str, char **dest)\r
 195 {\r
 196          int    i = 1;\r
 197         char    *buf = malloc( strlen(str) + 1 );\r
 198         \r
 199         strcpy(buf, str);\r
 200         dest[0] = buf;\r
 201         \r
 202         // Trim leading whitespace\r
 203         while(*buf == ' ' && *buf)      buf++;\r
 204         \r
 205         for(;;)\r
 206         {\r
 207                 dest[i] = buf;  // Save start of string\r
 208                 i++;\r
 209                 while(*buf && *buf != ' ')\r
 210                 {\r
 211                         if(*buf++ == '"')\r
 212                         {\r
 213                                 while(*buf && !(*buf == '"' && buf[-1] != '\\'))\r
 214                                         buf++;\r
 215                         }\r
 216                 }\r
 217                 if(*buf == '\0')        break;\r
 218                 *buf = '\0';\r
 219                 while(*++buf == ' ' && *buf);\r
 220                 if(*buf == '\0')        break;\r
 221         }\r
 222         dest[i] = NULL;\r
 223         if(i == 1) {\r
 224                 free(buf);\r
 225                 dest[0] = NULL;\r
 226         }\r
 227 }\r
 228 \r
 229 /**\r
 230  * \fn void Command_Colour(int argc, char **argv)\r
 231  * \brief \r
 232  */\r
 233 void Command_Colour(int argc, char **argv)\r
 234 {\r
 235         int fg, bg;\r
 236         char    clrStr[6] = "\x1B[37m";\r
 237         \r
 238         // Verify Arg Count\r
 239         if(argc < 2)\r
 240         {\r
 241                 Print("Please specify a colour\n");\r
 242                 goto usage;\r
 243         }\r
 244         \r
 245         // Check Colour\r
 246         for(fg=0;fg<8;fg++)\r
 247                 if(strcmp(cCOLOUR_NAMES[fg], argv[1]) == 0)\r
 248                         break;\r
 249 \r
 250         // Foreground a valid colour\r
 251         if(fg == 8) {\r
 252                 Print("Unknown Colour '");Print(argv[1]);Print("'\n");\r
 253                 goto usage;\r
 254         }\r
 255         // Set Foreground\r
 256         clrStr[3] = '0' + fg;\r
 257         write(_stdout, 6, clrStr);\r
 258         \r
 259         // Need to Set Background?\r
 260         if(argc > 2)\r
 261         {\r
 262                 for(bg=0;bg<8;bg++)\r
 263                         if(strcmp(cCOLOUR_NAMES[bg], argv[2]) == 0)\r
 264                                 break;\r
 265         \r
 266                 // Valid colour\r
 267                 if(bg == 8)\r
 268                 {\r
 269                         Print("Unknown Colour '");Print(argv[2]);Print("'\n");\r
 270                         goto usage;\r
 271                 }\r
 272         \r
 273                 clrStr[2] = '4';\r
 274                 clrStr[3] = '0' + bg;\r
 275                 write(_stdout, 6, clrStr);\r
 276         }\r
 277         // Return\r
 278         return;\r
 279 \r
 280         // Function Usage (Requested via a Goto (I know it's ugly))\r
 281 usage:\r
 282         Print("Valid Colours are ");\r
 283         for(fg=0;fg<8;fg++)\r
 284         {\r
 285                 Print(cCOLOUR_NAMES[fg]);\r
 286                 write(_stdout, 3, ", ");\r
 287         }\r
 288         write(_stdout, 4, "\b\b\n");\r
 289         return;\r
 290 }\r
 291 \r
 292 void Command_Clear(int argc, char **argv)\r
 293 {\r
 294         write(_stdout, 4, "\x1B[2J");   //Clear Screen\r
 295 }\r
 296 \r
 297 /**\r
 298  * \fn void Command_Cd(int argc, char **argv)\r
 299  * \brief Change directory\r
 300  */\r
 301 void Command_Cd(int argc, char **argv)\r
 302 {\r
 303         char    tmpPath[1024];\r
 304         int             fp;\r
 305         t_sysFInfo      stats;\r
 306         \r
 307         if(argc < 2)\r
 308         {\r
 309                 Print(gsCurrentDirectory);Print("\n");\r
 310                 return;\r
 311         }\r
 312         \r
 313         GeneratePath(argv[1], gsCurrentDirectory, tmpPath);\r
 314         \r
 315         fp = open(tmpPath, 0);\r
 316         if(fp == -1) {\r
 317                 write(_stdout, 26, "Directory does not exist\n");\r
 318                 return;\r
 319         }\r
 320         finfo(fp, &stats, 0);\r
 321         close(fp);\r
 322         \r
 323         if( !(stats.flags & FILEFLAG_DIRECTORY) ) {\r
 324                 write(_stdout, 17, "Not a Directory\n");\r
 325                 return;\r
 326         }\r
 327         \r
 328         strcpy(gsCurrentDirectory, tmpPath);\r
 329 }\r
 330 \r
 331 /**\r
 332  */\r
 333 void Command_Dir(int argc, char **argv)\r
 334 {\r
 335          int    dp, fp, dirLen;\r
 336         //char  modeStr[11] = "RWXrwxRWX ";\r
 337         char    tmpPath[1024];\r
 338         char    fileName[256];\r
 339         t_sysFInfo      info;\r
 340         \r
 341         // Generate Directory Path\r
 342         if(argc > 1)\r
 343                 dirLen = GeneratePath(argv[1], gsCurrentDirectory, tmpPath);\r
 344         else\r
 345         {\r
 346                 strcpy(tmpPath, gsCurrentDirectory);\r
 347         }\r
 348         dirLen = strlen(tmpPath);\r
 349         \r
 350         // Open Directory\r
 351         dp = open(tmpPath, OPENFLAG_READ);\r
 352         // Check if file opened\r
 353         if(dp == -1)\r
 354         {\r
 355                 //printf("Unable to open directory `%s', File cannot be found\n", tmpPath);\r
 356                 //write(_stdout, 27, "Unable to open directory `");\r
 357                 //write(_stdout, strlen(tmpPath)+1, tmpPath);\r
 358                 //write(_stdout, 25, "', File cannot be found\n");\r
 359                 return;\r
 360         }\r
 361         // Get File Stats\r
 362         if( finfo(dp, &info, 0) == -1 )\r
 363         {\r
 364                 close(dp);\r
 365                 write(_stdout, 34, "stat Failed, Bad File Descriptor\n");\r
 366                 return;\r
 367         }\r
 368         // Check if it's a directory\r
 369         if(!(info.flags & FILEFLAG_DIRECTORY))\r
 370         {\r
 371                 close(dp);\r
 372                 write(_stdout, 27, "Unable to open directory `");\r
 373                 write(_stdout, strlen(tmpPath)+1, tmpPath);\r
 374                 write(_stdout, 20, "', Not a directory\n");\r
 375                 return;\r
 376         }\r
 377         \r
 378         // Append Shash for file paths\r
 379         if(tmpPath[dirLen-1] != '/')\r
 380         {\r
 381                 tmpPath[dirLen++] = '/';\r
 382                 tmpPath[dirLen] = '\0';\r
 383         }\r
 384         // Read Directory Content\r
 385         while( (fp = readdir(dp, fileName)) )\r
 386         {\r
 387                 if(fp < 0)\r
 388                 {\r
 389                         if(fp == -3)\r
 390                                 write(_stdout, 42, "Invalid Permissions to traverse directory\n");\r
 391                         break;\r
 392                 }\r
 393                 // Create File Path\r
 394                 strcpy((char*)(tmpPath+dirLen), fileName);\r
 395                 // Open File\r
 396                 fp = open(tmpPath, 0);\r
 397                 if(fp == -1)    continue;\r
 398                 // Get File Stats\r
 399                 finfo(fp, &info, 0);\r
 400                 close(fp);\r
 401                 \r
 402                 // Colour Code\r
 403                 write(_stdout, 6, "\x1B[37m");  //White\r
 404                 if(info.flags & FILEFLAG_DIRECTORY)     //Directory Green\r
 405                         write(_stdout, 6, "\x1B[32m");\r
 406                 \r
 407                 //Print Mode\r
 408                 #if 0\r
 409                 if(stats.st_mode & 0400)        modeStr[0] = 'R';       else    modeStr[0] = '-';\r
 410                 if(stats.st_mode & 0200)        modeStr[1] = 'W';       else    modeStr[1] = '-';\r
 411                 if(stats.st_mode & 0100)        modeStr[2] = 'X';       else    modeStr[2] = '-';\r
 412                 if(stats.st_mode & 0040)        modeStr[3] = 'R';       else    modeStr[3] = '-';\r
 413                 if(stats.st_mode & 0020)        modeStr[4] = 'W';       else    modeStr[4] = '-';\r
 414                 if(stats.st_mode & 0010)        modeStr[5] = 'X';       else    modeStr[5] = '-';\r
 415                 if(stats.st_mode & 0004)        modeStr[6] = 'R';       else    modeStr[6] = '-';\r
 416                 if(stats.st_mode & 0002)        modeStr[7] = 'W';       else    modeStr[7] = '-';\r
 417                 if(stats.st_mode & 0001)        modeStr[8] = 'X';       else    modeStr[8] = '-';\r
 418                 write(_stdout, 10, modeStr);\r
 419                 #endif\r
 420                 \r
 421                 //Print Name\r
 422                 write(_stdout, strlen(fileName), fileName);\r
 423                 //Print slash if applicable\r
 424                 if(info.flags & FILEFLAG_DIRECTORY)\r
 425                         write(_stdout, 1, "/");\r
 426                 \r
 427                 // Revert Colout and end line\r
 428                 write(_stdout, 6, "\x1B[37m");\r
 429                 write(_stdout, 2, "\n");\r
 430         }\r
 431         // Close Directory\r
 432         close(dp);\r
 433 }\r