Usermode/Applications/CLIShell_src/main.c

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