如何在自定义简易Shell中处理带参数的exit命令？

问题根源 #

当前Shell仅通过直接匹配字符串"exit\n"处理无参数的exit命令，当输入exit 98这类带参数的命令时，输入会被切分成cmd[0] = "exit"、cmd[1] = "98"，随后进入外部命令查找流程。由于系统中不存在名为exit的外部可执行文件，因此触发execve失败，输出"No such file or directory"。

修复方案 #

调整shell.c的逻辑顺序，先对输入进行分词，再优先判断并处理exit命令，具体修改如下：

1. 移除原代码中直接判断strcmp(buf, "exit\n") == 0的代码块
2. 将tok_str(buf, cmd)的调用提前到所有内置命令判断之前
3. 新增对cmd[0]是否为exit的判断，并处理可选的状态码参数

修改后的完整shell.c代码：

#include "shell.h"

int main(int argc, char **argv, char **env)
{
        char *buf = NULL;
        size_t buflen = 0;
        char *cmd[20];

        do
        {
                fprintf(stderr, ":) ");

                if (getline(&buf, &buflen, stdin) == -1)
                {
                        if (feof(stdin))
                        {
                                free(buf);
                                buf = NULL;
                                exit(EXIT_SUCCESS);
                        }

                        perror("Error occurred");
                        free(buf);
                        buf = NULL;
                        exit(EXIT_FAILURE);
                }

                if (buf[0] == '\0' || strcmp(buf, "\n") == 0)
                {
                        free(buf);
                        buf = NULL;
                        continue;
                }

                // 先对输入进行分词处理
                tok_str(buf, cmd);

                // 处理exit命令（支持可选状态码参数）
                if (strcmp(cmd[0], "exit") == 0)
                {
                        int exit_status = EXIT_SUCCESS;
                        // 检查是否传入状态码参数
                        if (cmd[1] != NULL)
                        {
                                // 将字符串参数转换为整数，atoi对无效输入返回0
                                exit_status = atoi(cmd[1]);
                                // 确保状态码在0-255范围内（POSIX标准要求）
                                exit_status = exit_status % 256;
                                // 处理负数：转换为正的模值
                                if (exit_status < 0)
                                {
                                        exit_status += 256;
                                }
                        }
                        free(buf);
                        buf = NULL;
                        exit(exit_status);
                }

                // 处理env命令
                if (strcmp(cmd[0], "env") == 0)
                {
                        print_env(env);
                        free(buf);
                        buf = NULL;
                        continue;
                }

                add_path(cmd);

                if (fork() == 0)
                {
                        execve(cmd[0], cmd, env);
                        printf("%s: No such file or directory\n", argv[0]);
                        exit(EXIT_SUCCESS);
                }else
                {
                        wait(NULL);
                        free(buf);
                        buf = NULL;
                }

        } while (1);

        exit(EXIT_SUCCESS);
}

额外优化建议 #

• 严格参数校验：如果需要拒绝非整数参数，可以用strtol替代atoi，并检查转换是否完全有效：

  char *endptr;
  long status_val = strtol(cmd[1], &endptr, 10);
  if (*endptr != '\0')
  {
          // 参数包含非数字字符，保持默认状态码
          exit_status = EXIT_SUCCESS;
  }
  else
  {
          exit_status = (int)(status_val % 256);
          if (exit_status < 0) exit_status += 256;
  }
  

• 参数数量限制：如果需要忽略exit后的多余参数（比如exit 1 2 3），可以直接只取cmd[1]，无需处理后续参数。

内容的提问来源于stack exchange，提问作者coolboimax