diff options
author | Tor Egge <Tor.Egge@yahooinc.com> | 2023-09-22 11:52:49 +0200 |
---|---|---|
committer | Tor Egge <Tor.Egge@yahooinc.com> | 2023-09-22 11:52:49 +0200 |
commit | 71125e36f538b3048729ebb97d68ba75948a1d6b (patch) | |
tree | 9dba61fa753d6c54e7eaebdcca101beb67c86312 | |
parent | ad6d36794fe2a51e6f5eacde01eb648d7c76c379 (diff) |
Don't read from invalid file pointer (e.g. when
/proc/self/status could not be opened)
-rw-r--r-- | eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp index 03db333d582..31cb1d6b385 100644 --- a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp +++ b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp @@ -78,10 +78,12 @@ MemoryUsage extract_memory_usage() { vespalib::string vm_size = UNKNOWN; vespalib::string vm_rss = UNKNOWN; FilePointer file(fopen("/proc/self/status", "r")); - vespalib::string line; - while (read_line(file, line)) { - extract(line, "VmSize:", vm_size); - extract(line, "VmRSS:", vm_rss); + if (file.valid()) { + vespalib::string line; + while (read_line(file, line)) { + extract(line, "VmSize:", vm_size); + extract(line, "VmRSS:", vm_rss); + } } return {convert(vm_size), convert(vm_rss)}; } |