2 AcessOS Shell Version 2
\r
3 - Based on IOOS CLI Shell
\r
5 #include <acess/sys.h>
\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
21 // ==== CONSTANT GLOBALS ====
\r
22 char *cCOLOUR_NAMES[8] = {"black", "red", "green", "yellow", "blue", "magenta", "cyan", "white"};
\r
25 void (*fcn)(int argc, char **argv);
\r
27 {"colour", Command_Colour}, {"clear", Command_Clear},
\r
28 {"cd", Command_Cd}, {"dir", Command_Dir}
\r
30 #define BUILTIN_COUNT (sizeof(cBUILTINS)/sizeof(cBUILTINS[0]))
\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
41 int main(int argc, char *argv[], char *envp[])
\r
51 //Command_Clear(0, NULL);
\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
59 // Free last command & arguments
\r
60 if(saArgs[0]) free(saArgs);
\r
61 if(!bCached) free(sCommandStr);
\r
63 write(_stdout, strlen(gsCurrentDirectory), gsCurrentDirectory);
\r
64 write(_stdout, 3, "$ ");
\r
66 // Read Command line
\r
67 sCommandStr = ReadCommandLine( &length );
\r
69 // Check if the command should be cached
\r
70 if(gasCommandHistory == NULL || strcmp(sCommandStr, gasCommandHistory[giLastCommand]) != 0)
\r
72 if(giLastCommand >= giCommandSpace) {
\r
73 giCommandSpace += 12;
\r
74 gasCommandHistory = realloc(gasCommandHistory, giCommandSpace*sizeof(char*));
\r
77 gasCommandHistory[ giLastCommand ] = sCommandStr;
\r
81 // Parse Command Line into arguments
\r
82 Parse_Args(sCommandStr, saArgs);
\r
86 while(saArgs[iArgCount]) iArgCount++;
\r
88 // Silently Ignore all empty commands
\r
89 if(saArgs[1][0] == '\0') continue;
\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
95 if(strcmp(saArgs[1], cBUILTINS[length].name) == 0)
\r
97 cBUILTINS[length].fcn(iArgCount-1, &saArgs[1]);
\r
102 if(length != BUILTIN_COUNT) continue;
\r
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
110 Print("Unknown Command: `");Print(saArgs[1]);Print("'\n"); // Error Message
\r
113 // Check if the file is a directory
\r
114 finfo( length, &info, 0 );
116 if(info.flags & FILEFLAG_DIRECTORY) {
\r
117 Print("`");Print(saArgs[1]); // Error Message
\r
118 Print("' is a directory.\n");
\r
121 // Load new executable
\r
122 pid = clone(CLONE_VM, 0);
\r
123 if(pid == 0) execve(gsTmpBuffer, &saArgs[1], envp);
\r
125 Print("Unablt to create process: `");Print(gsTmpBuffer);Print("'\n"); // Error Message
126 //SysDebug("pid = %i\n", pid);
\r
130 waittid(pid, &status);
\r
136 * \fn char *ReadCommandLine(int *Length)
\r
137 * \brief Read from the command line
\r
139 char *ReadCommandLine(int *Length)
\r
142 int len, pos, space = 1023;
\r
145 // Preset Variables
\r
146 ret = malloc( space+1 );
\r
150 // Read In Command Line
\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
157 if(len <= 0) continue; // Protect against underflows
\r
158 if(pos == len) { // Simple case of end of string
\r
161 memmove(&ret[pos-1], &ret[pos], len-pos);
\r
165 write(_stdout, 1, &ch);
\r
171 ret = realloc(ret, space+1);
\r
172 if(!ret) return NULL;
\r
175 write(_stdout, 1, &ch);
\r
178 } while(ch != '\n');
\r
185 if(Length) *Length = len;
\r
191 * \fn void Parse_Args(char *str, char **dest)
\r
192 * \brief Parse a string into an argument array
\r
194 void Parse_Args(char *str, char **dest)
\r
197 char *buf = malloc( strlen(str) + 1 );
\r
202 // Trim leading whitespace
\r
203 while(*buf == ' ' && *buf) buf++;
\r
207 dest[i] = buf; // Save start of string
\r
209 while(*buf && *buf != ' ')
\r
213 while(*buf && !(*buf == '"' && buf[-1] != '\\'))
\r
217 if(*buf == '\0') break;
\r
219 while(*++buf == ' ' && *buf);
\r
220 if(*buf == '\0') break;
\r
230 * \fn void Command_Colour(int argc, char **argv)
\r
233 void Command_Colour(int argc, char **argv)
\r
236 char clrStr[6] = "\x1B[37m";
\r
238 // Verify Arg Count
\r
241 Print("Please specify a colour\n");
\r
246 for(fg=0;fg<8;fg++)
\r
247 if(strcmp(cCOLOUR_NAMES[fg], argv[1]) == 0)
\r
250 // Foreground a valid colour
\r
252 Print("Unknown Colour '");Print(argv[1]);Print("'\n");
\r
256 clrStr[3] = '0' + fg;
\r
257 write(_stdout, 6, clrStr);
\r
259 // Need to Set Background?
\r
262 for(bg=0;bg<8;bg++)
\r
263 if(strcmp(cCOLOUR_NAMES[bg], argv[2]) == 0)
\r
269 Print("Unknown Colour '");Print(argv[2]);Print("'\n");
\r
274 clrStr[3] = '0' + bg;
\r
275 write(_stdout, 6, clrStr);
\r
280 // Function Usage (Requested via a Goto (I know it's ugly))
\r
282 Print("Valid Colours are ");
\r
283 for(fg=0;fg<8;fg++)
\r
285 Print(cCOLOUR_NAMES[fg]);
\r
286 write(_stdout, 3, ", ");
\r
288 write(_stdout, 4, "\b\b\n");
\r
292 void Command_Clear(int argc, char **argv)
\r
294 write(_stdout, 4, "\x1B[2J"); //Clear Screen
\r
298 * \fn void Command_Cd(int argc, char **argv)
\r
299 * \brief Change directory
\r
301 void Command_Cd(int argc, char **argv)
\r
303 char tmpPath[1024];
\r
309 Print(gsCurrentDirectory);Print("\n");
\r
313 GeneratePath(argv[1], gsCurrentDirectory, tmpPath);
\r
315 fp = open(tmpPath, 0);
\r
317 write(_stdout, 26, "Directory does not exist\n");
\r
320 finfo(fp, &stats, 0);
\r
323 if( !(stats.flags & FILEFLAG_DIRECTORY) ) {
\r
324 write(_stdout, 17, "Not a Directory\n");
\r
328 strcpy(gsCurrentDirectory, tmpPath);
\r
333 void Command_Dir(int argc, char **argv)
\r
335 int dp, fp, dirLen;
\r
336 //char modeStr[11] = "RWXrwxRWX ";
\r
337 char tmpPath[1024];
\r
341 // Generate Directory Path
\r
343 dirLen = GeneratePath(argv[1], gsCurrentDirectory, tmpPath);
\r
346 strcpy(tmpPath, gsCurrentDirectory);
\r
348 dirLen = strlen(tmpPath);
\r
351 dp = open(tmpPath, OPENFLAG_READ);
\r
352 // Check if file opened
\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
362 if( finfo(dp, &info, 0) == -1 )
\r
365 write(_stdout, 34, "stat Failed, Bad File Descriptor\n");
\r
368 // Check if it's a directory
\r
369 if(!(info.flags & FILEFLAG_DIRECTORY))
\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
378 // Append Shash for file paths
\r
379 if(tmpPath[dirLen-1] != '/')
\r
381 tmpPath[dirLen++] = '/';
\r
382 tmpPath[dirLen] = '\0';
\r
385 fileName = (char*)(tmpPath+dirLen);
\r
386 // Read Directory Content
\r
387 while( (fp = readdir(dp, fileName)) )
\r
392 write(_stdout, 42, "Invalid Permissions to traverse directory\n");
\r
396 fp = open(tmpPath, 0);
\r
397 if(fp == -1) continue;
\r
399 finfo(fp, &info, 0);
\r
403 if(info.flags & FILEFLAG_DIRECTORY) // Directory: Green
\r
404 write(_stdout, 6, "\x1B[32m");
\r
406 write(_stdout, 6, "\x1B[37m"); // Default: White
\r
410 if(stats.st_mode & 0400) modeStr[0] = 'R'; else modeStr[0] = '-';
\r
411 if(stats.st_mode & 0200) modeStr[1] = 'W'; else modeStr[1] = '-';
\r
412 if(stats.st_mode & 0100) modeStr[2] = 'X'; else modeStr[2] = '-';
\r
413 if(stats.st_mode & 0040) modeStr[3] = 'R'; else modeStr[3] = '-';
\r
414 if(stats.st_mode & 0020) modeStr[4] = 'W'; else modeStr[4] = '-';
\r
415 if(stats.st_mode & 0010) modeStr[5] = 'X'; else modeStr[5] = '-';
\r
416 if(stats.st_mode & 0004) modeStr[6] = 'R'; else modeStr[6] = '-';
\r
417 if(stats.st_mode & 0002) modeStr[7] = 'W'; else modeStr[7] = '-';
\r
418 if(stats.st_mode & 0001) modeStr[8] = 'X'; else modeStr[8] = '-';
\r
419 write(_stdout, 10, modeStr);
\r
423 write(_stdout, strlen(fileName), fileName);
\r
424 // Print slash if applicable
\r
425 if(info.flags & FILEFLAG_DIRECTORY)
\r
426 write(_stdout, 1, "/");
\r
428 // Revert Colour and end line
\r
429 write(_stdout, 6, "\x1B[37m");
\r
430 write(_stdout, 1, "\n");
\r