diff --git a/cv5/program.c b/cv5/program.c index b7b1cdf..82a6f55 100644 --- a/cv5/program.c +++ b/cv5/program.c @@ -19,7 +19,7 @@ int compare(const void *a, const void *b) { } else if (student_a->votes < student_b->votes) { return 1; } else { - // if votes are the same, compare names + // if the votes are the same, compare names return strcmp(student_a->name, student_b->name); } } @@ -34,20 +34,27 @@ int main() { memset(line, 0, SIZE); char *r = fgets(line, SIZE, stdin); - // check if the line is empty + // check if the line is empty or just contains newline character if (r == NULL || (r[0] == '\n' && strlen(r) == 1)) { // end of input break; } + // check if the line was truncated + if (strlen(line) == SIZE - 1 && line[SIZE - 2] != '\n') { + // line was truncated, handle the error + fprintf(stderr, "Error: input line is too long\n"); + return 1; + } + // parse the string and extract number of votes and student name int votes; char name[SIZE]; if (sscanf(line, "%d %[^\n]", &votes, name) != 2) { // error, input was not in expected format // print error message and exit - printf("Nepodarilo nacitat nic\n"); - break; + fprintf(stderr, "Error: invalid input format\n"); + return 1; } // check if this student already exists