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)

Browse Source
This commit is contained in:
Elizaveta Lobanova 2020-11-24 15:07:58 +03:00 committed by GitHub
parent 364bf08920
commit a57554f769
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -175,7 +175,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));
}
@ -188,7 +188,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));
}