diff --git a/ssl/d1_both.c b/ssl/d1_both.c index aaa18677ce..19c3da6161 100644 --- a/ssl/d1_both.c +++ b/ssl/d1_both.c @@ -1579,6 +1579,8 @@ int dtls1_process_heartbeat(SSL *s) * plus 2 bytes payload length, plus payload, plus padding */ buffer = OPENSSL_malloc(write_length); + if (buffer == NULL) + return -1; bp = buffer; /* Enter response type, length and copy payload */ diff --git a/ssl/t1_lib.c b/ssl/t1_lib.c index 090f93ef62..0bdb77d49f 100644 --- a/ssl/t1_lib.c +++ b/ssl/t1_lib.c @@ -2581,6 +2581,8 @@ int tls1_process_heartbeat(SSL *s) * plus 2 bytes payload length, plus payload, plus padding */ buffer = OPENSSL_malloc(1 + 2 + payload + padding); + if (buffer == NULL) + return -1; bp = buffer; /* Enter response type, length and copy payload */