mirror of
https://github.com/EnderIce2/Fennix.git
synced 2025-05-28 15:34:31 +00:00
feat(userspace/coreutils): add stub "sh" command
Signed-off-by: EnderIce2 <enderice2@protonmail.com>
This commit is contained in:
parent
568dffbca1
commit
8258d40115
333
Userspace/coreutils/src/sh.c
Normal file
333
Userspace/coreutils/src/sh.c
Normal file
@ -0,0 +1,333 @@
|
||||
/*
|
||||
This file is part of Fennix C Library.
|
||||
|
||||
Fennix C Library is free software: you can redistribute it and/or
|
||||
modify it under the terms of the GNU General Public License as
|
||||
published by the Free Software Foundation, either version 3 of
|
||||
the License, or (at your option) any later version.
|
||||
|
||||
Fennix C Library is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with Fennix C Library. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/wait.h>
|
||||
#include <signal.h>
|
||||
#include <termios.h>
|
||||
#include <fcntl.h>
|
||||
#include <errno.h>
|
||||
#include <dirent.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
#define MAX_LINE_LEN 1024
|
||||
#define MAX_HISTORY 128
|
||||
#define CTRL_KEY(k) ((k) & 0x1f)
|
||||
|
||||
typedef struct
|
||||
{
|
||||
char *items[MAX_HISTORY];
|
||||
int count;
|
||||
int index;
|
||||
} History;
|
||||
|
||||
typedef enum
|
||||
{
|
||||
MODE_INSERT,
|
||||
MODE_COMMAND
|
||||
} InputMode;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
char line[MAX_LINE_LEN];
|
||||
int cursor;
|
||||
int len;
|
||||
InputMode mode;
|
||||
History history;
|
||||
struct termios orig_termios;
|
||||
} ShellState;
|
||||
|
||||
void HandleSignalInterrupt(int sig)
|
||||
{
|
||||
(void)sig;
|
||||
write(STDOUT_FILENO, "\n", 1);
|
||||
}
|
||||
|
||||
void EnableRawMode(ShellState *state)
|
||||
{
|
||||
tcgetattr(STDIN_FILENO, &state->orig_termios);
|
||||
struct termios raw = state->orig_termios;
|
||||
raw.c_lflag &= ~(ECHO | ICANON);
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &raw);
|
||||
}
|
||||
|
||||
void DisableRawMode(ShellState *state)
|
||||
{
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &state->orig_termios);
|
||||
}
|
||||
|
||||
void AddHistory(ShellState *state, const char *line)
|
||||
{
|
||||
if (state->history.count < MAX_HISTORY)
|
||||
state->history.items[state->history.count++] = strdup(line);
|
||||
else
|
||||
{
|
||||
free(state->history.items[0]);
|
||||
memmove(state->history.items, state->history.items + 1,
|
||||
(MAX_HISTORY - 1) * sizeof(char *));
|
||||
state->history.items[MAX_HISTORY - 1] = strdup(line);
|
||||
}
|
||||
state->history.index = state->history.count;
|
||||
}
|
||||
|
||||
void LoadHistory(ShellState *state)
|
||||
{
|
||||
char *home = getenv("HOME");
|
||||
if (!home)
|
||||
return;
|
||||
|
||||
char path[256];
|
||||
snprintf(path, sizeof(path), "%s/.sh_history", home);
|
||||
|
||||
FILE *fp = fopen(path, "r");
|
||||
if (!fp)
|
||||
return;
|
||||
|
||||
char line[MAX_LINE_LEN];
|
||||
while (fgets(line, sizeof(line), fp))
|
||||
{
|
||||
line[strcspn(line, "\n")] = '\0';
|
||||
AddHistory(state, line);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
|
||||
void SaveHistory(ShellState *state)
|
||||
{
|
||||
char *home = getenv("HOME");
|
||||
if (!home)
|
||||
return;
|
||||
|
||||
char path[256];
|
||||
snprintf(path, sizeof(path), "%s/.sh_history", home);
|
||||
|
||||
FILE *fp = fopen(path, "w");
|
||||
if (!fp)
|
||||
return;
|
||||
|
||||
for (int i = 0; i < state->history.count; i++)
|
||||
{
|
||||
fprintf(fp, "%s\n", state->history.items[i]);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
|
||||
void ProcessViCommand(ShellState *state, char c)
|
||||
{
|
||||
switch (c)
|
||||
{
|
||||
case 'h':
|
||||
if (state->cursor > 0)
|
||||
state->cursor--;
|
||||
break;
|
||||
case 'l':
|
||||
if (state->cursor < state->len)
|
||||
state->cursor++;
|
||||
break;
|
||||
case 'k':
|
||||
{
|
||||
if (state->history.index > 0)
|
||||
{
|
||||
state->history.index--;
|
||||
strcpy(state->line, state->history.items[state->history.index]);
|
||||
state->len = strlen(state->line);
|
||||
state->cursor = state->len;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case 'j':
|
||||
{
|
||||
if (state->history.index < state->history.count - 1)
|
||||
{
|
||||
state->history.index++;
|
||||
strcpy(state->line, state->history.items[state->history.index]);
|
||||
state->len = strlen(state->line);
|
||||
state->cursor = state->len;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case 'i':
|
||||
state->mode = MODE_INSERT;
|
||||
break;
|
||||
case 27:
|
||||
state->mode = MODE_COMMAND;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void ReadLine(ShellState *state)
|
||||
{
|
||||
state->len = 0;
|
||||
state->cursor = 0;
|
||||
state->line[0] = '\0';
|
||||
state->mode = MODE_INSERT;
|
||||
|
||||
printf("$ ");
|
||||
fflush(stdout);
|
||||
|
||||
EnableRawMode(state);
|
||||
|
||||
while (1)
|
||||
{
|
||||
char c = '\0';
|
||||
if (read(STDIN_FILENO, &c, 1) != 1)
|
||||
break;
|
||||
|
||||
if (state->mode == MODE_COMMAND)
|
||||
ProcessViCommand(state, c);
|
||||
else
|
||||
{
|
||||
if (c == CTRL_KEY('c'))
|
||||
{
|
||||
printf("\n");
|
||||
return;
|
||||
}
|
||||
else if (c == '\n')
|
||||
{
|
||||
printf("\n");
|
||||
break;
|
||||
}
|
||||
else if (c == 127)
|
||||
{
|
||||
if (state->cursor > 0)
|
||||
{
|
||||
memmove(&state->line[state->cursor - 1],
|
||||
&state->line[state->cursor],
|
||||
state->len - state->cursor + 1);
|
||||
state->cursor--;
|
||||
state->len--;
|
||||
}
|
||||
}
|
||||
else if (state->len < MAX_LINE_LEN - 1)
|
||||
{
|
||||
memmove(&state->line[state->cursor + 1],
|
||||
&state->line[state->cursor],
|
||||
state->len - state->cursor + 1);
|
||||
state->line[state->cursor] = c;
|
||||
state->cursor++;
|
||||
state->len++;
|
||||
}
|
||||
}
|
||||
|
||||
printf("\r\x1B[2K$ %s", state->line);
|
||||
fflush(stdout);
|
||||
printf("\r\x1B[%dC", state->cursor + 3);
|
||||
}
|
||||
|
||||
DisableRawMode(state);
|
||||
AddHistory(state, state->line);
|
||||
}
|
||||
|
||||
void ExecuteCommand(char **args)
|
||||
{
|
||||
if (args[0] == NULL)
|
||||
return;
|
||||
|
||||
if (strcmp(args[0], "exit") == 0)
|
||||
exit(0);
|
||||
if (strcmp(args[0], "cd") == 0)
|
||||
{
|
||||
if (args[1] == NULL)
|
||||
fprintf(stderr, "cd: missing argument\n");
|
||||
else if (chdir(args[1]))
|
||||
perror("cd");
|
||||
return;
|
||||
}
|
||||
|
||||
pid_t pid = fork();
|
||||
if (pid == 0)
|
||||
{
|
||||
execvp(args[0], args);
|
||||
perror("execvp");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
else if (pid > 0)
|
||||
wait(NULL);
|
||||
else
|
||||
perror("fork");
|
||||
}
|
||||
|
||||
void ShellLoop(ShellState *state)
|
||||
{
|
||||
signal(SIGINT, HandleSignalInterrupt);
|
||||
LoadHistory(state);
|
||||
|
||||
while (1)
|
||||
{
|
||||
ReadLine(state);
|
||||
if (state->len == 0)
|
||||
continue;
|
||||
|
||||
char *args[MAX_LINE_LEN / 2 + 1];
|
||||
char *token = strtok(state->line, " ");
|
||||
int i = 0;
|
||||
while (token != NULL)
|
||||
{
|
||||
args[i++] = token;
|
||||
token = strtok(NULL, " ");
|
||||
}
|
||||
args[i] = NULL;
|
||||
|
||||
ExecuteCommand(args);
|
||||
}
|
||||
|
||||
SaveHistory(state);
|
||||
}
|
||||
|
||||
void InitializeShell(ShellState *state)
|
||||
{
|
||||
memset(state, 0, sizeof(ShellState));
|
||||
LoadHistory(state);
|
||||
signal(SIGINT, HandleSignalInterrupt);
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
ShellState state = {0};
|
||||
InitializeShell(&state);
|
||||
|
||||
if (argc > 1)
|
||||
{
|
||||
if (strcmp(argv[1], "-c") == 0)
|
||||
ExecuteCommand(&argv[2]);
|
||||
else
|
||||
{
|
||||
FILE *fp = fopen(argv[1], "r");
|
||||
if (!fp)
|
||||
{
|
||||
perror("fopen");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
char line[MAX_LINE_LEN];
|
||||
while (fgets(line, sizeof(line), fp))
|
||||
{
|
||||
line[strcspn(line, "\n")] = '\0';
|
||||
char *args[] = {"/bin/sh", "-c", line, NULL};
|
||||
ExecuteCommand(args);
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
}
|
||||
else
|
||||
ShellLoop(&state);
|
||||
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user