commit | 30b17b9b0e73de1dd93e090c68b38f32339d411c | [log] [tgz] |
---|---|---|
author | Davide Grohmann <davide.grohmann@arm.com> | Tue Jun 14 15:17:18 2022 +0200 |
committer | Davide Grohmann <davide.grohmann@arm.com> | Mon Aug 22 14:16:42 2022 +0200 |
tree | c4293d6b35c8902a4cee2321ce2be820059a147b | |
parent | e48fa7a47239d9632dc4390af92bca7d0eac64a2 [diff] [blame] |
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
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 @@ uint8_t *tensorArena; const size_t tensorArenaSize; + InferenceParser parser; }; } // namespace InferenceProcess