mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!33691 fix security function 

Merge pull request !33691 from yeyunpeng2020/master
This commit is contained in:
i-robot 2022-04-28 10:58:30 +00:00 committed by Gitee
parent 0288ad42b5 3cf661352a
commit 56a74ddeba
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/lite/tools/converter/converter.cc
View File

@ -371,9 +371,12 @@ int RunConverter(int argc, const char **argv) {
} }
} }
// clear key // clear key
flags->dec_key.clear();
flags->encKeyStr.clear();
status = memset_s(flags->encKey, converter::kEncMaxLen, 0, converter::kEncMaxLen); status = memset_s(flags->encKey, converter::kEncMaxLen, 0, converter::kEncMaxLen);
if (status != EOK) { if (status != EOK) {
MS_LOG(ERROR) << "memset failed."; MS_LOG(ERROR) << "memset failed.";
delete meta_graph;
return RET_ERROR; return RET_ERROR;
} }
delete meta_graph; delete meta_graph;

6
mindspore/lite/tools/converter/converter_flags.cc
View File

@ -527,12 +527,6 @@ int Flags::Init(int argc, const char **argv) {
return RET_OK; return RET_OK;
} }
Flags::~Flags() {
dec_key.clear();
encKeyStr.clear();
memset(encKey, 0, kEncMaxLen);
}
bool CheckOfflineParallelConfig(const std::string &file, ParallelSplitConfig *parallel_split_config) { bool CheckOfflineParallelConfig(const std::string &file, ParallelSplitConfig *parallel_split_config) {
// device: [device0 device1] ---> {cpu, gpu} // device: [device0 device1] ---> {cpu, gpu}
// computeRate: [x: y] x >=0 && y >=0 && x/y < 10 // computeRate: [x: y] x >=0 && y >=0 && x/y < 10

2
mindspore/lite/tools/converter/converter_flags.h
View File

@ -51,7 +51,7 @@ class Flags : public virtual mindspore::lite::FlagParser {
public: public:
Flags(); Flags();
~Flags() override; ~Flags() = default;
int InitInputOutputDataType(); int InitInputOutputDataType();