mirror of
https://github.com/moparisthebest/curl
synced 2025-02-28 09:21:50 -05:00
doh: fix memory leak in OOM situation
Reviewed-by: Daniel Gustafsson Closes #3342
This commit is contained in:
parent
027d66e5f1
commit
d506df860f
@ -32,6 +32,7 @@
|
||||
#include "share.h"
|
||||
#include "curl_base64.h"
|
||||
#include "connect.h"
|
||||
#include "strdup.h"
|
||||
/* The last 3 #include files should be in this order */
|
||||
#include "curl_printf.h"
|
||||
#include "curl_memory.h"
|
||||
@ -142,8 +143,8 @@ doh_write_cb(void *contents, size_t size, size_t nmemb, void *userp)
|
||||
/* suspiciously much for us */
|
||||
return 0;
|
||||
|
||||
mem->memory = realloc(mem->memory, mem->size + realsize);
|
||||
if(mem->memory == NULL)
|
||||
mem->memory = Curl_saferealloc(mem->memory, mem->size + realsize);
|
||||
if(!mem->memory)
|
||||
/* out of memory! */
|
||||
return 0;
|
||||
|
||||
@ -525,7 +526,7 @@ UNITTEST DOHcode doh_decode(unsigned char *doh,
|
||||
|
||||
if(dohlen < 12)
|
||||
return DOH_TOO_SMALL_BUFFER; /* too small */
|
||||
if(doh[0] || doh[1])
|
||||
if(!doh || doh[0] || doh[1])
|
||||
return DOH_DNS_BAD_ID; /* bad ID */
|
||||
rcode = doh[3] & 0x0f;
|
||||
if(rcode)
|
||||
|
Loading…
x
Reference in New Issue
Block a user