Add color mode check when loading MAT file format

smlu · May 19, 2019 · 87deb1f · 87deb1f
1 parent 7ac8317
commit 87deb1f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/libim/content/asset/material/impl/serialization/mat.cpp b/src/libim/content/asset/material/impl/serialization/mat.cpp
@@ -62,6 +62,11 @@ Material& Material::deserialize(const InputStream& istream)
         throw StreamError("MAT file record count <= 0");
     }
 
+    if(header.colorInfo.colorMode < ColorMode::RGB ||
+       header.colorInfo.colorMode > ColorMode::RGBA) {
+        throw StreamError("Invalid color mode");
+    }
+
     if(header.colorInfo.bpp % 8 != 0) {
         throw StreamError("BPP % 8 != 0");
     }