使用memset时,阵列未正确重置

时间:2019-04-21 14:35:23

标签: c arrays memset fflush

当我从stdin拿到一个c的大笔交易时,重置的行为不符合预期。将再次提示用户输入代码,但在需要输入的地方输入新的一行后,长度检查失败。

void clear(void) {    
    while (getchar() != '\n')
        ;
}

int add_flight(code_t codes[], int lf) {
    char c[MAX_CODE_LEN + 1];
    int s = 0;
    while (s == 0) {
        printf("Enter code>\n");
        if (fgets(c, sizeof c, stdin)) {
            printf("%s", c);
            if ('\n' == c[0]) {
                printf("Invalid input\n");
            } else if (c[sizeof c - 1] == '\0' && c[sizeof c - 2] != '\n') {
                clear();
                printf("Invalid input\n");
                memset(c, 0, sizeof c);
            } else {
                strcpy(codes[lf].id, c);
                s = 1;
                break;
            }
        }
    }
    return 0;
}

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:3)

您的代码中存在几个问题:

  • clear函数不会测试文件结尾,如果在没有尾随换行符的情况下关闭标准输入,则会导致无限循环,就像从空文件重定向输入时一样。
  • 对超长输入行的测试不正确:如果输入未填充数组,则数组的最后一个字符不确定。删除尾随的换行符后,您应该测试strlen(c) == sizeof(c) - 1是否存在。
  • cchar数组的一个非常混乱的名称。 int通常使用此名称来接收字节值。将数组命名为buf,以提高可读性。
  • memset毫无用处,因为您要将新行读入数组。
  • 缺少code_t的定义。如果其id成员数组的大小不小于MAX_CODE_LEN + 1,则行为将是不确定的。
  • 此外,您将尾随换行符复制到codes[lf].id,这可能是错误的。
  • 如果id被定义为容纳MAX_CODE_LEN个字符,即char id[MAX_CODE_LEN + 1];代表额外的空终止符,则buf应该为用户键入的换行符增加一个字节,因此char buf[MAX_CODE_LEN + 2];

这是修改后的版本:

int clear(void) {
    int c;   
    while ((c = getchar()) != EOF && c != '\n')
        continue;
    return c;
}

int add_flight(code_t codes[], int lf) {
    char buf[MAX_CODE_LEN + 2];

    for (;;) {
        printf("Enter code>\n");
        if (fgets(buf, sizeof buf, stdin)) {
            printf("%s", buf);
            /* remove the trailing newline if present */
            size_t len = strlen(buf);
            if (len > 0 && buf[len - 1] == '\n')
                buf[--len] = '\0';
            if (len == sizeof(buf) - 1) {
                /* line too long: consume the rest of the input line */
                printf("Invalid input\n");
                if (clear() == EOF)
                    break;
            } else if (len == 0) {
                /* empty line */
                printf("Invalid input\n");
            } else {
                strcpy(codes[lf].id, buf);
                return 1;  // return success
            }
        } else {
            break;
        }
    }
    return 0;  // premature end of file: return failure
}