From 94d0e081c6159f3002c2eb56807a5551d541948d Mon Sep 17 00:00:00 2001 From: XuNiu <993273596@qq.com> Date: Sun, 23 Jun 2019 12:15:18 +0800 Subject: [PATCH] mmdbresolve: check the return value of malloc The return value of malloc() is used without validation Bug: 15851 Change-Id: I2635937d3107a75076e9bbee536797d97792683d Reviewed-on: https://code.wireshark.org/review/33705 Petri-Dish: Anders Broman Tested-by: Petri Dish Buildbot Reviewed-by: Anders Broman --- mmdbresolve.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mmdbresolve.c b/mmdbresolve.c index 674beeeb15..b14401a7e1 100644 --- a/mmdbresolve.c +++ b/mmdbresolve.c @@ -68,6 +68,10 @@ main(int argc, char *argv[]) int mmdb_err; char *out_buf = (char *) malloc(OUT_BUF_SIZE); + if (out_buf == NULL) { + fprintf(stdout, "ERROR: malloc failed\n"); + return 1; + } setvbuf(stdout, out_buf, _IOFBF, OUT_BUF_SIZE); fprintf(stdout, "[init]\n");