IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix the memory issue in ov_core_read_model_from_memory (#12817)

Browse Source
This commit is contained in:
RICKIE777 2022-09-07 14:06:31 +08:00 committed by GitHub
parent a092a192d8
commit f780303e5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/bindings/c/tests/ov_test.hpp
View File

@ -41,10 +41,11 @@ inline static std::vector<uint8_t> content_from_file(const char* filename, bool
std::ifstream is(filename, is_binary ? std::ifstream::binary | std::ifstream::in : std::ifstream::in); std::ifstream is(filename, is_binary ? std::ifstream::binary | std::ifstream::in : std::ifstream::in);
if (is) { if (is) {
is.seekg(0, std::ifstream::end); is.seekg(0, std::ifstream::end);
result.resize(is.tellg()); size_t file_len = is.tellg();
if (result.size() > 0) { result.resize(file_len + 1);
if (file_len > 0) {
is.seekg(0, std::ifstream::beg); is.seekg(0, std::ifstream::beg);
is.read(reinterpret_cast<char*>(&result[0]), result.size()); is.read(reinterpret_cast<char*>(&result[0]), file_len);
} }
} }
} }