diff --git a/frameworks/native/neural_network_runtime/neural_network_runtime.cpp b/frameworks/native/neural_network_runtime/neural_network_runtime.cpp index 6e669365a508d803dd0be4dace933ee8a2af8a7b..93202275243c72981d37fdb623df17c8bfb2c426 100644 --- a/frameworks/native/neural_network_runtime/neural_network_runtime.cpp +++ b/frameworks/native/neural_network_runtime/neural_network_runtime.cpp @@ -596,8 +596,8 @@ NNRT_API bool OH_NNModel_HasCache(const char *cacheDir, const char *modelName, u return false; } - if (fileNumber > FILE_NUMBER_MAX) { - LOGE("OH_NNModel_HasCache fileNumber is more than 100"); + if (fileNumber <= 0 || fileNumber > FILE_NUMBER_MAX) { + LOGE("OH_NNModel_HasCache fileNumber is invalid or more than 100"); std::filesystem::remove_all(cacheInfoPath); return false; }