ImageDeserializer - Add size and null check

PhoenixARC · Jul 3, 2024 · 60f569e · 60f569e
1 parent fedfc68
commit 60f569e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/PCK-Studio/Internal/Deserializer/ImageDeserializer.cs b/PCK-Studio/Internal/Deserializer/ImageDeserializer.cs
@@ -16,10 +16,15 @@ namespace PckStudio.Internal.Deserializer
     internal sealed class ImageDeserializer : IPckAssetDeserializer<Image>
     {
         public static readonly ImageDeserializer DefaultDeserializer = new ImageDeserializer();
+        // TODO: replace empty image with image displaying something went wrong
         private static Image EmptyImage = new Bitmap(1, 1, PixelFormat.Format32bppArgb);
 
         public Image Deserialize(PckAsset asset)
         {
+            _ = asset ?? throw new ArgumentNullException(nameof(asset));
+            if (asset.Size == 0)
+                return EmptyImage;
+
             using var stream = new MemoryStream(asset.Data);
             try
             {