Fix possible segfault in C API. (#1059)

Fangjun Kuang · GitHub
Commit 03ebdf3fc6bf30fa93d5b0248f5a13b826f9f015 03ebdf3f 1 parent dfca4500
sherpa-onnx/c-api/c-api.cc
--- a/sherpa-onnx/c-api/c-api.cc
查看文件 @03ebdf3
+++ b/sherpa-onnx/c-api/c-api.cc
查看文件 @03ebdf3
@@ -222,7 +222,7 @@ const SherpaOnnxOnlineRecognizerResult *GetOnlineStreamResult(
     }
     r->tokens_arr = tokens_temp;
-    if (!result.timestamps.empty()) {
+    if (!result.timestamps.empty() && result.timestamps.size() == r->count) {
       r->timestamps = new float[r->count];
       std::copy(result.timestamps.begin(), result.timestamps.end(),
                 r->timestamps);
@@ -490,7 +490,7 @@ const SherpaOnnxOfflineRecognizerResult *GetOfflineStreamResult(
     }
     r->tokens_arr = tokens_temp;
-    if (!result.timestamps.empty()) {
+    if (!result.timestamps.empty() && result.timestamps.size() == r->count) {
       r->timestamps = new float[r->count];
       std::copy(result.timestamps.begin(), result.timestamps.end(),
                 r->timestamps);