1 files changed, 10 insertions, 2 deletions

diff --git a/testing/nTox.c b/testing/nTox.c
index 199835a6..faa6cfa5 100644
--- a/testing/nTox.c
+++ b/testing/nTox.c
@@ -229,7 +229,11 @@ int main(int argc, char *argv[])
         int size = ftell(data_file);
         fseek(data_file, 0, SEEK_SET);
         uint8_t data[size];
-        fread(data, sizeof(uint8_t), size, data_file);
+        if(fread(data, sizeof(uint8_t), size, data_file) != size)
+        {
+            printf("Error reading file\n");
+            exit(0);
+        }
         Messenger_load(data, size);
     } else { 
         //else save new keys
@@ -237,7 +241,11 @@ int main(int argc, char *argv[])
         uint8_t data[size];
         Messenger_save(data);
         data_file = fopen("data","w");
-        fwrite(data, sizeof(uint8_t), size, data_file);
+        if(fwrite(data, sizeof(uint8_t), size, data_file) != size)
+        {
+            printf("Error writing file\n");
+            exit(0);
+        }
     }
     fclose(data_file);
     m_callback_friendrequest(print_request);