[Svace] prevent warning
authorJaeyun <jy1210.jung@samsung.com>
Fri, 15 Jul 2022 09:17:12 +0000 (18:17 +0900)
committerjaeyun-jung <39614140+jaeyun-jung@users.noreply.github.com>
Mon, 18 Jul 2022 02:35:15 +0000 (11:35 +0900)
Prevent error case, handle memory limit when receiving data from socket.

Signed-off-by: Jaeyun <jy1210.jung@samsung.com>
src/libnnstreamer-edge/nnstreamer-edge-internal.c

index 7bd859fdb238025b61e33fd7fdb3b328fcfcbfa1..e1b44ab52764583e2da47a22c9ec0fd2ea59febe 100644 (file)
@@ -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]);