2 files changed, 11 insertions, 3 deletions

diff --git a/testing/DHT_test.c b/testing/DHT_test.c
index 5895fa55..368609d0 100644
--- a/testing/DHT_test.c
+++ b/testing/DHT_test.c
@@ -1,7 +1,7 @@
 /* DHT test
  * A file with a main that runs our DHT for testing.
  * 
- * Compile with: gcc -O2 -Wall -D VANILLA_NACL -o test ../core/Lossless_UDP.c ../core/network.c ../core/net_crypto.c ../core/Messenger.c ../nacl/build/${HOSTNAME%.*}/lib/amd64/ DHT_test.c
+ * Compile with: gcc -O2 -Wall -D VANILLA_NACL -o test ../core/Lossless_UDP.c ../core/network.c ../core/net_crypto.c ../core/Messenger.c ../nacl/build/${HOSTNAME%.*}/lib/amd64/{cpucycles.o,libnacl.a,randombytes.o} DHT_test.c
  * 
  * Command line arguments are the ip, port and public key of a node.
  * EX: ./test 127.0.0.1 33445 AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
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);