fgets和chdir在C中奇怪地一起表演

时间:2015-02-02 18:01:58

标签: c shell fgets chdir

我目前正在为家庭作业创建一个简单的shell,而且我遇到了一个问题。这是一段代码片段,其中包含与问题有关的部分(我可能已经忘记了一些部分,如果发现任何遗漏,请告诉我):

eatWrd返回字符串中的第一个单词,并将该单词从字符串中删除。

wrdCount,如暗示的那样,返回字符串中的单词数。

如果这些代码中的任何一个对于我可以发布的回复是必要的,请告诉我,我几乎100%肯定他们不是问题的原因。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define MAX 100

int main(void)
{
  char input[MAX];
  char *argm[MAX];
  memset(input, 0, sizeof(input));
  memset(argm, 0, sizeof(argm));

  while(1){
    printf("cmd:\n");
    fgets(input, MAX-1, stdin);

    for(i=0;i < wrdCount(input); i++){
      argm[i] = eatWrd(input);
    }
    argm[i] = NULL;

    if (!strncmp(argm[0],"cd" , 2)){
      chdir(argm[1]);
    }

    if (!strncmp(argm[0],"exit", 4)){
      exit(0);
    }

    memset(input, 0, sizeof(input));
    memset(argm, 0, sizeof(argm));
  }
}

无论如何,这个循环适用于许多使用execvp的其他命令(例如cat,ls等),当我使用cd时,它按预期工作,除非我尝试退出shell,它需要多个退出打电话实际上出去了。 (事实证明,退出呼叫的数量正好等于我调用cd的次数)。当我在会话期间不使用cd时,它只需要一次退出呼叫。我不确定发生了什么,感谢任何帮助,谢谢。

这是eatWrd:

char* eatWrd(char * cmd)
{
  int i = 0;            // i keeps track of position in cmd
  int count = 0;        // count keeps track of position of second word
  char rest[MAX_LINE];  // rest will hold cmd without the first word

  char * word = (char *) malloc(MAX_LINE);   //word will hold the first word
  sscanf(cmd, "%s", word);                   //scan the first word into word

  // iterate through white spaces, then first word, then the following white spaces
  while(cmd[i] == ' ' || cmd[i] == '\t'){
    i++;
    count++;
  }

  while(cmd[i] != ' ' && cmd[i] != '\t' && cmd[i] != '\n' && cmd[i] != '\0'){
    i++;
    count++;
  }

  while(cmd[i] == ' ' || cmd[i] == '\t'){
    i++;
    count++;
  }

  // copy the rest of cmd into rest
  while(cmd[i] != '\n' && cmd[i] != '\0'){
    rest[i-count] = cmd[i];
    i++;
  }
  rest[i-count] = '\0';

  memset(cmd, 0, MAX_LINE);
  strcpy(cmd, rest);        //move rest into cmd
  return word;              //return word
}

这是wrdCount:

int wrdCount(char *sent)
{
  char *i = sent;
  int words = 0;

  //keep iterating through the string, 
  //increasing the count if a word and white spaces are passed,
  // until the string is finished.
  while(1){
    while(*i == ' ' || *i == '\t') i++;

    if(*i == '\n' || *i == '\0') break;

    words++;

    while(*i != ' ' && *i != '\t' && *i != '\n' && *i != '\0') i++;
  }
  return words;
}

1 个答案:

答案 0 :(得分:0)

您的代码的这种变体适合我:

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <unistd.h>
#define MAX 100

char *eatWrd(char **line) {
    char *next_c = *line;
    char *word_start = NULL;

    while (isspace(*next_c)) next_c += 1;

    if (*next_c) {
        word_start = next_c;
        do {
            next_c += 1;
        } while (*next_c && ! isspace(*next_c));
        *next_c = '\0';
        *line = next_c + 1;
    }

    return word_start;
}

int main(void)
{
    char input[MAX];
    char *argm[MAX];

    while(1) {
        int word_count = 0;
        char *next_input = input;

        printf("cmd:\n");
        fgets(input, MAX, stdin);

        do {
              argm[word_count] = eatWrd(&next_input);
        } while (argm[word_count++]);
        /* The above always overcounts by one */
        word_count -= 1;

        if (!strcmp(argm[0], "cd")){
            chdir(argm[1]);
        } else if (!strcmp(argm[0], "exit")) {
            exit(0);
        }
    }
}

请注意eatWrd()上的变体,它不需要移动任何数据,也不需要预先解析字符串来确定预期的字数。我想你的实现会更复杂,以便处理引用或其他一些,但它绝对可以遵循相同的一般方法。

请注意,我使用!strcmp()代替strncmp()来修正命令匹配条件。