From f780303e5ac7de927b66efd06f51e9fc1027f977 Mon Sep 17 00:00:00 2001 From: RICKIE777 Date: Wed, 7 Sep 2022 14:06:31 +0800 Subject: [PATCH] Fix the memory issue in ov_core_read_model_from_memory (#12817) --- src/bindings/c/tests/ov_test.hpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/bindings/c/tests/ov_test.hpp b/src/bindings/c/tests/ov_test.hpp index 2629f8c5b98..e3fcb08c94e 100644 --- a/src/bindings/c/tests/ov_test.hpp +++ b/src/bindings/c/tests/ov_test.hpp @@ -41,10 +41,11 @@ inline static std::vector content_from_file(const char* filename, bool std::ifstream is(filename, is_binary ? std::ifstream::binary | std::ifstream::in : std::ifstream::in); if (is) { is.seekg(0, std::ifstream::end); - result.resize(is.tellg()); - if (result.size() > 0) { + size_t file_len = is.tellg(); + result.resize(file_len + 1); + if (file_len > 0) { is.seekg(0, std::ifstream::beg); - is.read(reinterpret_cast(&result[0]), result.size()); + is.read(reinterpret_cast(&result[0]), file_len); } } }