diff options
-rw-r--r-- | tests/dtls/dtls-stress.c | 4 | ||||
-rw-r--r-- | tests/mini-dtls-record.c | 1 |
2 files changed, 4 insertions, 1 deletions
diff --git a/tests/dtls/dtls-stress.c b/tests/dtls/dtls-stress.c index 01e5ecad68..826bd29414 100644 --- a/tests/dtls/dtls-stress.c +++ b/tests/dtls/dtls-stress.c @@ -564,12 +564,14 @@ static void filter_permute_state_run(filter_permute_state_t * state, gnutls_transport_ptr_t fd, const unsigned char *buffer, size_t len) { - unsigned char *data = malloc(len); + unsigned char *data; int packet = state->order[state->count]; if (debug > 2) log("running permutation for %s/%d/%d\n", state->name, packetCount, state->count); + data = malloc(len); + assert(data); memcpy(data, buffer, len); state->packets[packet].data = data; state->packets[packet].size = len; diff --git a/tests/mini-dtls-record.c b/tests/mini-dtls-record.c index bcda720391..83d4900706 100644 --- a/tests/mini-dtls-record.c +++ b/tests/mini-dtls-record.c @@ -111,6 +111,7 @@ static ssize_t odd_push(gnutls_transport_ptr_t tr, const void *data, size_t len) } stored_messages[current] = malloc(len); + assert(stored_messages[current]); memcpy(stored_messages[current], data, len); stored_sizes[current] = len; |