fix: reject if image is not supported (#642)

Cargo.toml

@@ -11,6 +11,7 @@ crate-type = ["cdylib"]
 anyhow = "1"
 base64 = "0.21"
 cssparser = "0.29"
+infer = "0.12"
 napi = { version = "2", default-features = false, features = [
   "napi3",
   "serde-json",

__test__/image.spec.ts

@@ -2,7 +2,7 @@ import { promises as fs } from 'fs'
 import { join } from 'path'
 import test from 'ava'
 
-import { createCanvas, Image } from '../index'
+import { createCanvas, Image, loadImage } from '../index'
 
 import { snapshotImage } from './image-snapshot'
 
@@ -88,3 +88,7 @@ test('svg-transparent-background', async (t) => {
 
   await snapshotImage(t, { canvas })
 })
+
+test('load invalid image should throw error', async (t) => {
+  await t.throwsAsync(() => loadImage(join(__dirname, 'fixtures', 'broken.png')))
+})

load-image.js

@@ -51,7 +51,7 @@ module.exports = async function loadImage(source, options = {}) {
     }
   }
 
-  // throw error as dont support that source
+  // throw error as don't support that source
   throw new TypeError('unsupported image source')
 }
 

src/image.rs

@@ -262,15 +262,29 @@ impl Image {
           let image_binary = STANDARD
             .decode(base64_str)
             .map_err(|e| Error::new(Status::InvalidArg, format!("Decode data url failed {e}")))?;
-          Some(Bitmap::from_buffer(
-            image_binary.as_ptr() as *mut u8,
-            image_binary.len(),
-          ))
+          if let Some(kind) = infer::get(&image_binary) {
+            if kind.matcher_type() == infer::MatcherType::Image {
+              Some(Bitmap::from_buffer(
+                image_binary.as_ptr() as *mut u8,
+                image_binary.len(),
+              ))
+            } else {
+              return Err(Error::new(Status::InvalidArg, "Unsupported image type"));
+            }
+          } else {
+            return Err(Error::new(Status::InvalidArg, "Unsupported image type"));
+          }
         } else {
           None
         }
+      } else if let Some(kind) = infer::get(&data) {
+        if kind.matcher_type() == infer::MatcherType::Image {
+          Some(Bitmap::from_buffer(data.as_ptr() as *mut u8, length))
+        } else {
+          return Err(Error::new(Status::InvalidArg, "Unsupported image type"));
+        }
       } else {
-        Some(Bitmap::from_buffer(data.as_ptr() as *mut u8, length))
+        return Err(Error::new(Status::InvalidArg, "Unsupported image type"));
       };
       if let Some(ref b) = bitmap {
         if (self.width - -1.0).abs() < f64::EPSILON {