Ограничение символов входной строки

Question

Я пытаюсь написать простой код, в котором пользователь должен ввести строку, но если строка содержит более пяти символов, она должна вывести ошибку и вернуть -1.Я использовал fgets, чтобы получить ввод, и strlen, чтобы вычислить длину строки.

#include <stdio.h>
#include <strings.h>
#include <stdlib.h>


int main()
  {
    char a[5];
    int length = 0;

    printf("Enter a string to calculate it's length\n");
    fgets(a,5,stdin);

    length = strlen(a)-1; // don't want the '\n' to be counted

    if(length > 5){

        printf("error");
    }
    printf("string length %d\n",length);


       return 0;
 }

Когда я ввожу строку с более чем 5 символами, она не выводит ошибку, она просто выводит, чторазмер строки - три.

Может ли кто-нибудь дать мне подсказку?

Заранее спасибо.

Answer 1

Используйте strchr для проверки новой строки.Если на входе нет новой строки, считывайте символы до тех пор, пока не будет найдена новая строка для очистки буфера ввода и повторите попытку.

#include <stdio.h>
#include <string.h>

int main( void) {
    char a[7];//five characters, a newline and a zero terminator
    int toolong = 0;

    do {
        if ( toolong) {
            printf ( "too many characters. try again\n");
        }
        toolong = 0;
        printf ( "enter up to five characters.\n");
        if ( fgets ( a, sizeof a, stdin)) {
            while ( ! strchr ( a, '\n')) {//check for newline
                toolong = 1;
                fgets ( a, sizeof a, stdin);//read more characters
            }
        }
        else {
            fprintf ( stderr, "fgets EOF\n");
            return 0;
        }
    } while ( toolong);

    return 0;
}

Answer 2

fgets(a,5,stdin);

fgets всегда читает символы размера 1.

Чтение останавливается после EOF или новой строки.Таким образом, он читает только 4 символа.

length = strlen(a)-1;   // 4-1 = 3