add check for chunk_size_vec

This commit is contained in:
雾聪 2023-10-24 19:52:44 +08:00
parent 71b403ee9d
commit 78e4f729be

View File

@ -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")) {