130 lines
2.6 KiB
C++
130 lines
2.6 KiB
C++
#include <sys/wait.h>
|
|
#include <sys/termmode.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <error.h>
|
|
#include <fcntl.h>
|
|
|
|
int status = 0;
|
|
|
|
void command()
|
|
{
|
|
unsigned termmode = TERMMODE_UNICODE
|
|
| TERMMODE_SIGNAL
|
|
| TERMMODE_UTF8
|
|
| TERMMODE_LINEBUFFER
|
|
| TERMMODE_ECHO;
|
|
settermmode(0, termmode);
|
|
|
|
const size_t CWD_SIZE = 512;
|
|
char cwd[CWD_SIZE];
|
|
const char* wd = getcwd(cwd, CWD_SIZE);
|
|
if ( !wd ) { wd = "?"; }
|
|
|
|
printf("root@sortix %s # ", wd);
|
|
fflush(stdout);
|
|
|
|
const size_t commandsize = 128;
|
|
char command[commandsize + 1];
|
|
size_t commandused = 0;
|
|
|
|
while (true)
|
|
{
|
|
char c;
|
|
ssize_t bytesread = read(1, &c, sizeof(c));
|
|
if ( bytesread < 0 ) { error(64, errno, "read stdin"); }
|
|
if ( !bytesread ) { break; }
|
|
if ( !c ) { continue; }
|
|
if ( c == '\n' ) { break; }
|
|
if ( commandsize <= commandused ) { continue; }
|
|
command[commandused++] = c;
|
|
}
|
|
|
|
command[commandused] = '\0';
|
|
|
|
if ( command[0] == '\0' ) { return; }
|
|
|
|
if ( strcmp(command, "$?") == 0 ) { printf("%u\n", status); status = 0; return; }
|
|
if ( strcmp(command, "$$") == 0 ) { printf("%u\n", getpid()); status = 0; return; }
|
|
if ( strcmp(command, "$PPID") == 0 ) { printf("%u\n", getppid()); status = 0; return; }
|
|
|
|
int argc = 0;
|
|
const char* argv[256];
|
|
argv[0] = NULL;
|
|
|
|
bool lastwasspace = true;
|
|
for ( size_t i = 0; i <= commandused; i++ )
|
|
{
|
|
switch ( command[i] )
|
|
{
|
|
case '\0':
|
|
case ' ':
|
|
case '\t':
|
|
case '\n':
|
|
command[i] = 0;
|
|
lastwasspace = true;
|
|
break;
|
|
default:
|
|
if ( lastwasspace ) { argv[argc++] = command + i; }
|
|
lastwasspace = false;
|
|
}
|
|
}
|
|
|
|
if ( !argv[0] ) { return; }
|
|
|
|
if ( strcmp(argv[0], "exit") == 0 )
|
|
{
|
|
const char* status = "1";
|
|
if ( 1 < argc ) { status = argv[1]; }
|
|
exit(atoi(status));
|
|
}
|
|
|
|
if ( strcmp(argv[0], "cd") == 0 )
|
|
{
|
|
status = 0;
|
|
const char* newdir = "/";
|
|
if ( 1 < argc ) { newdir = argv[1]; }
|
|
if ( chdir(newdir) )
|
|
{
|
|
error(0, errno, "cd: %s", newdir);
|
|
status = 1;
|
|
}
|
|
return;
|
|
}
|
|
|
|
pid_t child = fork();
|
|
if ( child < 0 ) { perror("fork"); status = 1; return; }
|
|
if ( child != 0 )
|
|
{
|
|
pid_t childpid = wait(&status);
|
|
return;
|
|
}
|
|
|
|
if ( 3 <= argc )
|
|
{
|
|
if ( strcmp(argv[argc-2], ">") == 0 )
|
|
{
|
|
const char* file = argv[argc-1];
|
|
int outfd = open(file, O_CREAT | O_WRONLY | O_TRUNC | O_APPEND);
|
|
if ( outfd < 0 ) { error(127, errno, "%s", file); exit(127); }
|
|
close(1);
|
|
dup(outfd);
|
|
close(outfd);
|
|
argc -= 2;
|
|
}
|
|
}
|
|
|
|
argv[argc] = NULL;
|
|
execv(argv[0], (char* const*) argv);
|
|
error(127, errno, "%s", argv[0]);
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
while ( true ) { command(); }
|
|
}
|
|
|