在Linux中创建简单Shell

时间:2018-10-23 02:51:11

标签: c arrays linux shell strtok

我正在做一个任务,用C语言创建一个非常简单的Linux shell,它的工作原理几乎与我想要的一样。

如果用户输入一个简单的Linux命令,程序将运行它并循环以允许另一个命令。如果用户输入“退出”,则程序退出。

我的问题是命令仅在第一次使用。之后,它们似乎以某种方式变得格式不正确。有没有办法我可以重新初始化我的args数组,以便它可以正确接收新输入?

int main() {
    char* args[50];          // Argument array.
    char userInput[200];     // User input.
    char* userQuit = "quit"; // String to be compared to user input to quit program.
    int pid;                 // Process ID for fork().
    int i = 0;               // Counter.

    while(1) {
        // Promt and get input from user.
        printf("minor5> ");
        fgets(userInput, sizeof(userInput), stdin);

        // Pass userInput into args array.
        args[0] = strtok(userInput, " \n\0");

        // Loop to separate args into individual arguments, delimited by either space, newline, or NULL.
        while(args[i] != NULL) {
            i++;
            args[i] = strtok(NULL, " \n\0");
        }

        // If the first argument is "quit", exit the program.
        if(strcmp(args[0], userQuit) == 0) {
            printf("Exiting Minor5 Shell...\n");
            exit(EXIT_SUCCESS);
        }

        // Create child process.
        pid = fork();

        // Parent process will wait for child to execute.
        // Child process will execute the command given in userInput.
        if(pid > 0) {
            // Parent //
            wait( (int *) 0 );
        } else {
            // Child //
            int errChk;
            errChk = execvp(args[0], args);

            if(errChk == -1) {
                printf("%s: Command not found\n", userInput);
            }
        }
    }

    return 0;
}

1 个答案:

答案 0 :(得分:1)

您需要确保 args的最后一个值为NULL。偶然,它可能在 first 命令上有一个命令,但不能保证

这是您的解析循环的重做片段[请原谅免费的样式清理]:

// Pass userInput into args array.
char *uptr = userInput;

i = 0;
while (1) {
    char *token = strtok(uptr, " \n");
    uptr = NULL;

    if (token == NULL)
        break;

    args[i++] = token;
}

// NOTE: this is the key missing ingredient from your code
args[i] = NULL;