将命令通过管道传递到 shell 后，C 无限循环

Question

我目前正在尝试为一个类构建一个自定义 shell。我能够毫无问题地执行从该 shell 发出的命令，但是如果我将我的命令通过管道传输到我的 shell，我最终会陷入无限循环。我不知道为什么会发生这种情况。下面的示例将导致无限循环。

echo "ls" | ./myshell

当然，我必须重定向程序的输出，如果在命令中出现管道，例如ls | grep 测试。在我的 shell 中，这可以完美地工作。我正在使用 fork()、execv()，当然还有 pipe + dup 用于在子进程之间重定向流.

似乎 fgets() 不会从上次发出的命令中清除 STDIN_FILENO。

对于命令echo "ls"| ./myshell 我的程序将执行以下操作:(最小工作示例)

编辑:最小工作示例

#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>

int running = 1;
int exit_code = 0;

char cwd[256];
char command[256];
char args[10][256];
char buffer[256] __attribute__((aligned(4096)));

void handle_command(char* buffer, int buffer_size)
{
  int c = 0;
  int argsCount = -1;
  int lastIndex = 0;

  for (c = 0; c < buffer_size && buffer[c]; c++)
  {
    if (argsCount > 10)
    {
      argsCount = 10;
      printf("Argument Count is limited to 10 (no dynamic memory allocation) all other arguments will be ignored\n");
      break;
    }
    if (buffer[c] == '\r' || buffer[c] == '\n' || buffer[c] == ' ')
    {
      if (argsCount == -1)
      {
        memcpy(command, buffer + lastIndex, c - lastIndex);
        command[c - lastIndex] = 0;
      }
      else
      {
        memcpy(args[argsCount], buffer + lastIndex, c - lastIndex);
        args[argsCount][c - lastIndex] = 0;
      }
      argsCount++;
      lastIndex = c + 1;

    }
  }

  if (strcmp(command, "exit") == 0)
  {
    c = 4;
    while (buffer[c] == ' ')
      c++;
    exit_code = atoi(&buffer[c]);
    printf("Exiting Shell with exit_code %d\n", exit_code);
    running = 0;
  }
  else if (strcmp(command, "") != 0)
  {
    // -------------- Add structure to commands --------------------------------
    struct command_struct{
      char *options[10];
    } sub_commands[1];
    // Simplified code, there would be a dynamic amount of sub_commands
    // and further logic to handle pipes and < > >>

    // initialize first command, would work dynamically
    sub_commands[0].options[0] = command;
    sub_commands[0].options[1] = NULL;

    int status;
    int pid = fork();
    if (pid == 0) {
      execvp(sub_commands[0].options[0], sub_commands[0].options);
      perror("Error: Reached code after execvp\n");
    } else if (pid < 0) {
      perror("Cannot fork!\n");
    }
    wait(&status);
  }
}


int main(int argc, char *argv[])
{
  cwd[0] = '/';

  do
  {
    printf("\n%s %s%s", "SHELL:", cwd, "> ");
    fgets(buffer, 255, stdin);
    buffer[255] = 0;
    handle_command(buffer, 256);
    for (size_t a = 0; a < 256; a++)
      buffer[a] = 0;

  } while (running);

  return exit_code;
}

编辑 我必须指出，部分代码已在此类中给出。

如有任何帮助，我们将不胜感激!

Answer 1

OP 代码只会在 running == 0 时退出 main() do 循环。

running = 0; 仅在 strcmp(command, "exit") == 0 时发生，并且不清楚 just "exit" 曾经加载到 command 中。注意:command 未在每次 handle_command() 调用时完全初始化，并且 command 不需要是全局变量。

调整代码以在 fgets() 返回 NULL 时退出并查看行处理。建议:

do {
  printf("\n%s %s%s", "SHELL:", cwd, "> ");
  if (fgets(buffer, sizeof buffer, stdin) == NULL) {
    break; 
  }
  // lop off potential end-of-line character(s)
  buffer[strcspn(buffer,"\r\n")] = '\0';
  handle_command(buffer, sizeof buffer);
} while (running);