1 files changed, 12 insertions, 0 deletions

diff --git a/toxcore/assoc.c b/toxcore/assoc.c
index 4d837aa3..417b6139 100644
--- a/toxcore/assoc.c
+++ b/toxcore/assoc.c
@@ -831,8 +831,20 @@ Assoc *new_Assoc(size_t bits, size_t entries, const uint8_t *public_id)
     /* allocation: preferably few blobs */
     size_t bckt, cix;
     Client_entry *clients = malloc(sizeof(*clients) * assoc->candidates_bucket_count * assoc->candidates_bucket_size);
+
+        if (!clients) {
+                free(assoc);
+                return NULL;
+        }
+
     candidates_bucket *lists = malloc(sizeof(*lists) * assoc->candidates_bucket_count);
 
+        if (!lists) {
+                free(assoc);
+                free(clients);
+                return NULL;
+        }
+
     for (bckt = 0; bckt < assoc->candidates_bucket_count; bckt++) {
         candidates_bucket *list = &lists[bckt];