Fix a bug in savedata loading when malloc fails.

Also added a bunch of asserts to tests where they don't check allocs.

1 files changed, 6 insertions, 0 deletions

diff --git a/toxcore/DHT.c b/toxcore/DHT.c
index b3017259..7c9263cd 100644
--- a/toxcore/DHT.c
+++ b/toxcore/DHT.c
@@ -2917,6 +2917,12 @@ static State_Load_Status dht_load_state_callback(void *outer, const uint8_t *dat
             // Copy to loaded_clients_list
             dht->loaded_nodes_list = (Node_format *)calloc(MAX_SAVED_DHT_NODES, sizeof(Node_format));
 
+            if (dht->loaded_nodes_list == nullptr) {
+                LOGGER_ERROR(dht->log, "could not allocate %u nodes", MAX_SAVED_DHT_NODES);
+                dht->loaded_num_nodes = 0;
+                break;
+            }
+
             const int num = unpack_nodes(dht->loaded_nodes_list, MAX_SAVED_DHT_NODES, nullptr, data, length, 0);
 
             if (num > 0) {