bcsh.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

已复制！
#include "cmd.h"

extern int yyparse();
char hostname[100];
char *username;
char pwd[100];

int main(int argc, char **argv) {
    // get the hostname, username and current working directory
    // to show the prompt
    gethostname(hostname, sizeof(hostname));
    getcwd(pwd, sizeof(pwd));
    username = getpwuid(getuid())->pw_name;
    showPrompt();
    while (true) {
        yyparse();
    }
    return 0;
}

void showPrompt() {
    if (getuid() == 0) {
        printf("%s@%s:%s# ", username, hostname, pwd);
    } else {
        printf("\033[01;32m%s"
               "\033[35m@%s\033[00m:"
               "\033[33m%s\033[00m$ ",
               username, hostname, pwd);
    }
    fflush(stdout);
}

void argInsert(struct Command *cmd, char *s, bool isStr) {
    if (cmd->argc == 0) {
        cmd->args = malloc(2 * sizeof(char *));
    } else {
        cmd->args = realloc(cmd->args, (cmd->argc + 2) * sizeof(char *));
    }
    if (isStr) {
        cmd->args[cmd->argc] = strdup(s + 1);
        cmd->args[cmd->argc][strlen(cmd->args[cmd->argc]) - 1] = '\0';
        free(s);
    } else {
        cmd->args[cmd->argc] = s;
    }
    cmd->args[cmd->argc + 1] = NULL;
    cmd->argc++;
}