du6 - 11

du6/program.c

@@ -10,6 +10,8 @@ typedef struct tree {
     struct tree *right;
 } Tree;
 
+int valid_db = 1;
+
 //odstranit koncovy znak
 void strip_newline(char *s) {
     int len = strlen(s);
@@ -43,6 +45,7 @@ Tree* read_tree() {
     // inak nacitat oboch potomkov
     node->left = read_tree();
     if (!node->left) {
+	valid_db = 0;
         free(node);
         return NULL;
     }
@@ -50,6 +53,7 @@ Tree* read_tree() {
     node->right = read_tree();
     if (!node->right) {
         free(node->left);
+	valid_db = 0;
         free(node);
         return NULL;
     }
@@ -75,6 +79,8 @@ int count_leaves(Tree *t) {
 
 // spustenie znalostneho systemu
 void run_system(Tree *node) {
+    if (!node) return;
+
     printf("%s\n", node->value);
 
     // list -> koniec
@@ -106,7 +112,7 @@ int main() {
     Tree *root = read_tree();
 
     // nepodarilo sa nacitat koren -> chyba
-    if (!root) {
+    if (!root || !valid_db) {
         printf("Expert z bufetu to vie.\n");
 	printf("Chybna databaza\n");
         return 0;