diff --git a/du5/program.c b/du5/program.c index ba97cfe..d20e864 100644 --- a/du5/program.c +++ b/du5/program.c @@ -115,7 +115,21 @@ int main() { } char dummy[LINE_SIZE]; - fgets(dummy, sizeof(dummy), stdin); + if (fgets(dummy, sizeof(dummy), stdin) == NULL) { + printf("Expert z bufetu to vie.\n"); + printf("Chybna databaza\n"); + cleanup_tree(root); + return 0; + } + + dummy[strcspn(dummy, "\n\r")] = '\0'; + + if (strlen(dummy) != 0) { + printf("Expert z bufetu to vie.\n"); + printf("Chybna databaza\n"); + cleanup_tree(root); + return 0; + } int answer_count = how_many_answers(root);