added checking for null pointer in new_Assoc function

This commit is contained in:
slaniCode 2014-08-29 15:21:19 +02:00
parent f1ed0eeabb
commit 52ad2c64f7

View File

@ -831,8 +831,20 @@ Assoc *new_Assoc(size_t bits, size_t entries, const uint8_t *public_id)
/* allocation: preferably few blobs */ /* allocation: preferably few blobs */
size_t bckt, cix; size_t bckt, cix;
Client_entry *clients = malloc(sizeof(*clients) * assoc->candidates_bucket_count * assoc->candidates_bucket_size); 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); 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++) { for (bckt = 0; bckt < assoc->candidates_bucket_count; bckt++) {
candidates_bucket *list = &lists[bckt]; candidates_bucket *list = &lists[bckt];