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

[GNA] Fix a global buffer overflow in GNAModelSerial::Import (#3290) (#3327)

Browse Source
This commit is contained in:
Elizaveta Lobanova
2020-11-24 18:12:16 +03:00
committed by GitHub
parent a14fdef9b6
commit 81152e07f3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/src/gna_plugin/gna_model_serial.cpp
View File

@@ -172,7 +172,7 @@ void GNAModelSerial::Import(void *basePointer,
for (auto inputIndex = 0; inputIndex < modelHeader.nInputs; inputIndex++) {
uint32_t nameSize = 0;
readNBits<32>(nameSize, is);
std::string inName("", nameSize);
std::string inName(nameSize, '\0');
readNBytes(&inName[0], nameSize, is);
inputNames.push_back(inName.substr(0, nameSize - 1));
}
@@ -185,7 +185,7 @@ void GNAModelSerial::Import(void *basePointer,
for (auto inputIndex = 0; inputIndex < modelHeader.nOutputs; inputIndex++) {
uint32_t nameSize = 0;
readNBits<32>(nameSize, is);
std::string outName("", nameSize);
std::string outName(nameSize, '\0');
readNBytes(&outName[0], nameSize, is);
outputNames.push_back(outName.substr(0, nameSize - 1));
}