Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

code check

This commit is contained in:
jianghui58 2022-11-21 20:09:00 +08:00
parent fbfbbc6630
commit f4ea446e0d
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mindspore/lite/src/common/file_utils.cc
View File

@ -122,7 +122,13 @@ char *ReadFileSegment(const std::string &file, int64_t offset, int64_t len) {
}
ifs.seekg(0, std::ios::end);
size_t total_size = ifs.tellg();
auto goffset = ifs.tellg();
if (goffset < 0) {
MS_LOG(ERROR) << "read file range failed";
ifs.close();
return nullptr;
}
size_t total_size = static_cast<size_t>(goffset);
if (offset_pos + len_pos > total_size) {
MS_LOG(ERROR) << "file segment out of range";
ifs.close();
@ -171,7 +177,14 @@ char *ReadFile(const char *file, size_t *size) {
}
ifs->seekg(0, std::ios::end);
*size = ifs->tellg();
auto goffset = ifs->tellg();
if (goffset < 0) {
MS_LOG(ERROR) << "read file range failed";
ifs->close();
delete ifs;
return nullptr;
}
*size = static_cast<size_t>(goffset);
auto buf = new (std::nothrow) char[*size];
if (buf == nullptr) {
MS_LOG(ERROR) << "malloc buf failed, file: " << file;