diff options
author | Davide Grohmann <davide.grohmann@arm.com> | 2022-06-14 15:17:18 +0200 |
---|---|---|
committer | Davide Grohmann <davide.grohmann@arm.com> | 2022-08-22 14:16:42 +0200 |
commit | 30b17b9b0e73de1dd93e090c68b38f32339d411c (patch) | |
tree | c4293d6b35c8902a4cee2321ce2be820059a147b /applications/inference_process/include/inference_process.hpp | |
parent | e48fa7a47239d9632dc4390af92bca7d0eac64a2 (diff) | |
download | ethos-u-core-software-30b17b9b0e73de1dd93e090c68b38f32339d411c.tar.gz |
Check the validity of the buffer before parsing the model
If the buffer does not point to a well defined flatbuffer the parsing
segfaults.
Change-Id: Icb8dfef37dc28b2b7a22c6d3804851be8198aa9d
Diffstat (limited to 'applications/inference_process/include/inference_process.hpp')
-rw-r--r-- | applications/inference_process/include/inference_process.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/applications/inference_process/include/inference_process.hpp b/applications/inference_process/include/inference_process.hpp index fc54ae0..f8d7fd8 100644 --- a/applications/inference_process/include/inference_process.hpp +++ b/applications/inference_process/include/inference_process.hpp @@ -18,6 +18,8 @@ #pragma once +#include "inference_parser.hpp" + #include <array> #include <queue> #include <stdlib.h> @@ -85,5 +87,6 @@ private: uint8_t *tensorArena; const size_t tensorArenaSize; + InferenceParser parser; }; } // namespace InferenceProcess |