wcap: Check for mmap and malloc return value in wcap decode module
Checking for return value in main.c for wcap_decoder_create function
and mmap, malloc return value in wcap_decoder_create function to avoid
crashes
Signed-off-by: vivek <vivek.ellur@samsung.com>
diff --git a/wcap/wcap-decode.c b/wcap/wcap-decode.c
index 87d9337..76ecc2f 100644
--- a/wcap/wcap-decode.c
+++ b/wcap/wcap-decode.c
@@ -126,6 +126,11 @@
decoder->size = buf.st_size;
decoder->map = mmap(NULL, decoder->size,
PROT_READ, MAP_PRIVATE, decoder->fd, 0);
+ if (decoder->map == MAP_FAILED) {
+ fprintf(stderr, "mmap failed\n");
+ free(decoder);
+ return NULL;
+ }
header = decoder->map;
decoder->format = header->format;
@@ -137,6 +142,10 @@
frame_size = header->width * header->height * 4;
decoder->frame = malloc(frame_size);
+ if (decoder->frame == NULL) {
+ free(decoder);
+ return NULL;
+ }
memset(decoder->frame, 0, frame_size);
return decoder;