From 78e4f729be82b515480759addb64bd8d4e124dca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= <wucong.lyb@alibaba-inc.com>
Date: Tue, 24 Oct 2023 19:52:44 +0800
Subject: [PATCH] add check for chunk_size_vec

---
 .../runtime/websocket/bin/websocket-server-2pass.cpp | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/funasr/runtime/websocket/bin/websocket-server-2pass.cpp b/funasr/runtime/websocket/bin/websocket-server-2pass.cpp
index 9e0668f0e..a6374710d 100644
--- a/funasr/runtime/websocket/bin/websocket-server-2pass.cpp
+++ b/funasr/runtime/websocket/bin/websocket-server-2pass.cpp
@@ -408,9 +408,15 @@ void WebSocketServer::on_message(websocketpp::connection_hdl hdl,
         if (msg_data->tpass_online_handle == NULL) {
           std::vector<int> chunk_size_vec =
               jsonresult["chunk_size"].get<std::vector<int>>();
-          FUNASR_HANDLE tpass_online_handle =
-              FunTpassOnlineInit(tpass_handle, chunk_size_vec);
-          msg_data->tpass_online_handle = tpass_online_handle;
+          // check chunk_size_vec
+          if(chunk_size_vec.size() == 3 && chunk_size_vec[1] != 0){
+            FUNASR_HANDLE tpass_online_handle =
+                FunTpassOnlineInit(tpass_handle, chunk_size_vec);
+            msg_data->tpass_online_handle = tpass_online_handle;
+          }else{
+            LOG(ERROR) << "Wrong chunk_size!";
+            break;
+          }
         }
       }
       if (jsonresult.contains("itn")) {