Check that json data is an array or object (triton-inference-server#143)

Co-authored-by: Elias Bermudez <dbermudez@nvidia.com>
julianguinard · Aug 4, 2022 · 32df4a7 · 32df4a7
1 parent 00702e8
commit 32df4a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/c++/perf_analyzer/data_loader.cc b/src/c++/perf_analyzer/data_loader.cc
@@ -451,6 +451,11 @@ DataLoader::ReadTensorData(
 
       const rapidjson::Value* content;
 
+      // Check if the input data file is malformed
+      if (!(tensor.IsArray() || tensor.IsObject())) {
+        return cb::Error("Input data file is malformed.", pa::GENERIC_ERROR);
+      }
+
       if (tensor.IsArray()) {
         content = &tensor;
       } else if (tensor.HasMember("b64")) {