Add support for reading the ICC profile

Co-authored-by: Rasmus Piorr <piorr@google.com>

Author: 1c3t3a

src/reader/decoder.rs

@@ -21,6 +21,7 @@ pub const PLTE_CHANNELS: usize = 3;
 /// Headers for supported extensions.
 const EXT_NAME_NETSCAPE: &[u8] = b"\x0bNETSCAPE2.0\x03";
 const EXT_NAME_XMP: &[u8] = b"\x0bXMP DataXMP";
+const EXT_NAME_ICC: &[u8] = b"\x0bICCRGBG1012";
 
 /// An error returned in the case of the image not being formatted properly.
 #[derive(Debug)]
@@ -382,6 +383,8 @@ pub struct StreamingDecoder {
     header_end_reached: bool,
     /// XMP metadata bytes.
     xmp_metadata: Option<Vec<u8>>,
+    /// ICC profile bytes.
+    icc_profile: Option<Vec<u8>>,
 }
 
 /// One version number of the GIF standard.
@@ -470,6 +473,7 @@ impl StreamingDecoder {
             current: None,
             header_end_reached: false,
             xmp_metadata: None,
+            icc_profile: None,
         }
     }
 
@@ -542,6 +546,12 @@ impl StreamingDecoder {
         self.xmp_metadata.as_deref()
     }
 
+    /// ICC profile stored in the image.
+    #[must_use]
+    pub fn icc_profile(&self) -> Option<&[u8]> {
+        self.icc_profile.as_deref()
+    }
+
     /// The version number of the GIF standard used in this image.
     ///
     /// We suppose a minimum of `V87a` compatibility. This value will be reported until we have
@@ -773,26 +783,8 @@ impl StreamingDecoder {
             }
             ApplicationExtension => {
                 debug_assert_eq!(0, b);
-
-                // the parser removes sub-block lenghts, so app name and data are concatenated
-                if let Some(&[first, second, ..]) = self.ext.data.strip_prefix(EXT_NAME_NETSCAPE) {
-                    let repeat = u16::from(first) | u16::from(second) << 8;
-                    goto!(BlockEnd, emit Decoded::Repetitions(if repeat == 0 { Repeat::Infinite } else { Repeat::Finite(repeat) }))
-                } else if let Some(mut rest) = self.ext.data.strip_prefix(EXT_NAME_XMP) {
-                    // XMP adds a "ramp" of 257 bytes to the end of the metadata to let the "pascal-strings"
-                    // parser converge to the null byte. The ramp looks like "0x01, 0xff, .., 0x01, 0x00".
-                    // For convenience and to allow consumers to not be bothered with this implementation detail,
-                    // we cut the ramp.
-                    const RAMP_SIZE: usize = 257;
-                    if rest.len() >= RAMP_SIZE
-                        && rest.ends_with(&[0x03, 0x02, 0x01, 0x00])
-                        && rest[rest.len() - RAMP_SIZE..].starts_with(&[0x01, 0x0ff])
-                    {
-                        rest = &rest[..(rest.len() - RAMP_SIZE)];
-                    }
-
-                    self.xmp_metadata = Some(rest.to_vec());
-                    goto!(BlockEnd)
+                if let Some(decoded) = self.read_application_extension() {
+                    goto!(BlockEnd, emit decoded)
                 } else {
                     goto!(BlockEnd)
                 }
@@ -901,6 +893,50 @@ impl StreamingDecoder {
         Ok(())
     }
 
+    fn read_application_extension(&mut self) -> Option<Decoded> {
+        // the parser removes sub-block lenghts, so app name and data are concatenated
+        if let Some(rest) = self.ext.data.strip_prefix(EXT_NAME_NETSCAPE) {
+            if rest.len() != 3 {
+                return None;
+            }
+            let repeat = u16::from(rest[0]) | u16::from(rest[1]) << 8;
+            return Some(Decoded::Repetitions(if repeat == 0 {
+                Repeat::Infinite
+            } else {
+                Repeat::Finite(repeat)
+            }));
+        } else if let Some(mut rest) = self.ext.data.strip_prefix(EXT_NAME_XMP) {
+            // XMP adds a "ramp" of 257 bytes to the end of the metadata to let the "pascal-strings"
+            // parser converge to the null byte. The ramp looks like "0x01, 0xff, .., 0x01, 0x00".
+            // For convenience and to allow consumers to not be bothered with this implementation detail,
+            // we cut the ramp.
+            const RAMP_SIZE: usize = 257;
+            if rest.len() >= RAMP_SIZE
+                && rest.ends_with(&[0x03, 0x02, 0x01, 0x00])
+                && rest[rest.len() - RAMP_SIZE..].starts_with(&[0x01, 0x0ff])
+            {
+                rest = &rest[..(rest.len() - RAMP_SIZE)];
+            }
+
+            self.xmp_metadata = Some(rest.to_vec());
+        } else if let Some(mut rest) = self.ext.data.strip_prefix(EXT_NAME_ICC) {
+            // The extension data always contains the length byte (see the XMP metadata extension
+            // for why that is necessary), so we need to remove it again for the ICC profile.
+            let mut icc_profile = Vec::new();
+            loop {
+                let block_length = *rest.first().unwrap_or(&0) as usize;
+                if block_length == 0 {
+                    break;
+                }
+                let (subblock, tail) = rest.split_at(1 + block_length);
+                icc_profile.extend_from_slice(&subblock[1..]);
+                rest = tail;
+            }
+            self.icc_profile = Some(icc_profile);
+        }
+        None
+    }
+
     fn add_frame(&mut self) {
         if self.current.is_none() {
             self.current = Some(Frame::default());

src/reader/mod.rs

@@ -479,6 +479,12 @@ where
         self.decoder.decoder.xmp_metadata()
     }
 
+    /// ICC profile stored in the image.
+    #[inline]
+    pub fn icc_profile(&self) -> Option<&[u8]> {
+        self.decoder.decoder.icc_profile()
+    }
+
     /// Abort decoding and recover the `io::Read` instance
     pub fn into_inner(self) -> io::BufReader<R> {
         self.decoder.into_inner()

tests/decode.rs

@@ -273,6 +273,15 @@ fn xmp_metadata() {
     assert_eq!(decoder.xmp_metadata(), Some(EXPECTED_METADATA.as_bytes()))
 }
 
+#[test]
+fn icc_profile() {
+    let image: &[u8] = include_bytes!("samples/beacon_icc.gif");
+    let icc_profile: &[u8] = include_bytes!("samples/profile.icc");
+    let decoder = DecodeOptions::new().read_info(image).unwrap();
+
+    assert_eq!(decoder.icc_profile(), Some(icc_profile))
+}
+
 #[test]
 fn check_last_extension_returns() {
     let mut buf: &[u8] = include_bytes!("samples/beacon_xmp.gif");

tests/results.txt

@@ -1,5 +1,6 @@
 tests/samples/alpha_gif_a.gif: 3871893825
 tests/samples/anim-gr.gif: 291646878
+tests/samples/beacon_icc.gif: 2462153529
 tests/samples/beacon_xmp.gif: 2462153529
 tests/samples/beacon.gif: 2462153529
 tests/samples/interlaced.gif: 2115495372