From f1e497502c6851a76b57388fcfbdcec5120c6f8b Mon Sep 17 00:00:00 2001 From: Edward Williamson Date: Sun, 14 Dec 2014 20:45:19 -0500 Subject: [PATCH] check malloc return value --- cs.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cs.c b/cs.c index 04bc615b..1d1634dd 100644 --- a/cs.c +++ b/cs.c @@ -447,6 +447,12 @@ size_t cs_disasm(csh ud, const uint8_t *buffer, size_t size, uint64_t offset, si total_size = sizeof(cs_insn) * cache_size; total = cs_mem_malloc(total_size); + if (total == NULL) + { + //malloc failed + handle->errnum = CS_ERR_MEM; + return 0; + } insn_cache = total; while (size > 0) {