From: Jaeyun <jy1210.jung@samsung.com>
Date: Fri, 15 Jul 2022 09:17:12 +0000 (+0900)
Subject: [Svace] prevent warning
X-Git-Tag: submit/tizen/20220810.074948~19
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=41cf4c157e6b0452138879e83026a6e23bac40f7;p=platform%2Fupstream%2Fnnstreamer-edge.git

[Svace] prevent warning

Prevent error case, handle memory limit when receiving data from socket.

Signed-off-by: Jaeyun <jy1210.jung@samsung.com>
---

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]);