From 41cf4c157e6b0452138879e83026a6e23bac40f7 Mon Sep 17 00:00:00 2001 From: Jaeyun Date: Fri, 15 Jul 2022 18:17:12 +0900 Subject: [PATCH] [Svace] prevent warning Prevent error case, handle memory limit when receiving data from socket. Signed-off-by: Jaeyun --- src/libnnstreamer-edge/nnstreamer-edge-internal.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/libnnstreamer-edge/nnstreamer-edge-internal.c b/src/libnnstreamer-edge/nnstreamer-edge-internal.c index 7bd859f..e1b44ab 100644 --- a/src/libnnstreamer-edge/nnstreamer-edge-internal.c +++ b/src/libnnstreamer-edge/nnstreamer-edge-internal.c @@ -314,6 +314,11 @@ _nns_edge_cmd_receive (nns_edge_conn_s * conn, nns_edge_cmd_s * cmd) } nns_edge_logd ("Received command:%d (num:%u)", cmd->info.cmd, cmd->info.num); + if (cmd->info.num >= NNS_EDGE_DATA_LIMIT) { + nns_edge_loge ("Invalid request, the max memories for data transfer is %d.", + NNS_EDGE_DATA_LIMIT); + return NNS_EDGE_ERROR_IO; + } for (n = 0; n < cmd->info.num; n++) { cmd->mem[n] = malloc (cmd->info.mem_size[n]); -- 2.34.1