EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Fix possible segfault in C API. (#1059)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-06-26 09:57:19 +08:00
committed by GitHub
parent dfca4500aa
commit 03ebdf3fc6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sherpa-onnx/c-api/c-api.cc
View File

@@ -222,7 +222,7 @@ const SherpaOnnxOnlineRecognizerResult *GetOnlineStreamResult(
} }
r->tokens_arr = tokens_temp; r->tokens_arr = tokens_temp;
if (!result.timestamps.empty()) { if (!result.timestamps.empty() && result.timestamps.size() == r->count) {
r->timestamps = new float[r->count]; r->timestamps = new float[r->count];
std::copy(result.timestamps.begin(), result.timestamps.end(), std::copy(result.timestamps.begin(), result.timestamps.end(),
r->timestamps); r->timestamps);
@@ -490,7 +490,7 @@ const SherpaOnnxOfflineRecognizerResult *GetOfflineStreamResult(
} }
r->tokens_arr = tokens_temp; r->tokens_arr = tokens_temp;
if (!result.timestamps.empty()) { if (!result.timestamps.empty() && result.timestamps.size() == r->count) {
r->timestamps = new float[r->count]; r->timestamps = new float[r->count];
std::copy(result.timestamps.begin(), result.timestamps.end(), std::copy(result.timestamps.begin(), result.timestamps.end(),
r->timestamps); r->timestamps);