[SVACE/414633] Check malloc return value
authorMyungJoo Ham <myungjoo.ham@samsung.com>
Mon, 1 Apr 2019 07:59:59 +0000 (16:59 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Mon, 1 Apr 2019 10:47:07 +0000 (19:47 +0900)
Do not proceed if malloc returns null:
Warning Message
Pointer 'output[0].data' returned from function 'malloc' at nnstreamer_customfilter_example_scaler_allocator.c:159 may be null, and it is dereferenced at nnstreamer_customfilter_example_scaler_allocator.c:202.

Signed-off-by: MyungJoo Ham <myungjoo.ham@samsung.com>
nnstreamer_example/custom_example_scaler/nnstreamer_customfilter_example_scaler_allocator.c

index caa2748..99715d8 100644 (file)
@@ -158,6 +158,7 @@ pt_allocate_invoke (void *private_data,
   size = get_tensor_data_size (&prop->output_meta.info[0]);
 
   output[0].data = malloc (size);
+  assert (output[0].data);
 
   /* This assumes the limit is 4 */
   assert (NNS_TENSOR_RANK_LIMIT == 4);