diff options
author | Emilian Peev <epeev@google.com> | 2019-07-30 13:53:55 -0700 |
---|---|---|
committer | android-build-merger <android-build-merger@google.com> | 2019-07-30 13:53:55 -0700 |
commit | fa436baa99d762b4b528e2c4319ad5c692c32084 (patch) | |
tree | cd524d1c44ed5ba56e86e6ff9a0fbf7505cc0160 | |
parent | d7e0bad295d80a7fd2b3d8d55eb7dd2b24de4b13 (diff) | |
parent | cfc5a976af28e4666ebba17b71d06ca6bb4300e8 (diff) | |
download | dynamic_depth-fa436baa99d762b4b528e2c4319ad5c692c32084.tar.gz |
Check Android container item MIME type am: 8facf7af35
am: cfc5a976af
Change-Id: Ic67b5553bdb746dae4be7856b46a232d63f037f9
-rw-r--r-- | internal/dynamic_depth/depth_jpeg.cc | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/internal/dynamic_depth/depth_jpeg.cc b/internal/dynamic_depth/depth_jpeg.cc index 396a41e..af2ead9 100644 --- a/internal/dynamic_depth/depth_jpeg.cc +++ b/internal/dynamic_depth/depth_jpeg.cc @@ -19,6 +19,7 @@ namespace dynamic_depth { int32_t ValidateAndroidDynamicDepthBuffer(const char* buffer, size_t buffer_length) { XmpData xmp_data; + std::string itemMime("image/jpeg"); const string image_data(buffer, buffer_length); ReadXmpFromMemory(image_data, /*XmpSkipExtended*/ false, &xmp_data); @@ -29,6 +30,18 @@ int32_t ValidateAndroidDynamicDepthBuffer(const char* buffer, size_t buffer_leng return -1; } + // Check the container items mime type + if ((device->GetContainer() == nullptr) || (device->GetContainer()->GetItems().empty())) { + LOG(ERROR) << "No container or container items found!"; + return -1; + } + auto items = device->GetContainer()->GetItems(); + for (const auto& item : items) { + if (item->GetMime() != itemMime) { + LOG(ERROR) << "Item MIME type doesn't match the expected value: " << itemMime; + return -1; + } + } // Check profiles const Profiles* profiles = device->GetProfiles(); if (profiles == nullptr) { |