jdicc.c

   1 /*
   2  * jdicc.c
   3  *
   4  * Copyright (C) 1997-1998, Thomas G. Lane, Todd Newman.
   5  * Copyright (C) 2017, D. R. Commander.
   6  * For conditions of distribution and use, see the accompanying README.ijg
   7  * file.
   8  *
   9  * This file provides code to read International Color Consortium (ICC) device
  10  * profiles embedded in JFIF JPEG image files.  The ICC has defined a standard
  11  * for including such data in JPEG "APP2" markers.  The code given here does
  12  * not know anything about the internal structure of the ICC profile data; it
  13  * just knows how to get the profile data from a JPEG file while reading it.
  14  */
  15
  16 #define JPEG_INTERNALS
  17 #include "jinclude.h"
  18 #include "jpeglib.h"
  19 #include "jerror.h"
  20
  21 #ifndef HAVE_STDLIB_H           /* <stdlib.h> should declare malloc() */
  22 extern void *malloc(size_t size);
  23 #endif
  24
  25
  26 #define ICC_MARKER  (JPEG_APP0 + 2)     /* JPEG marker code for ICC */
  27 #define ICC_OVERHEAD_LEN  14            /* size of non-profile data in APP2 */
  28
  29
  30 /*
  31  * Handy subroutine to test whether a saved marker is an ICC profile marker.
  32  */
  33
  34 LOCAL(boolean)
  35 marker_is_icc(jpeg_saved_marker_ptr marker)
  36 {
  37   return
  38     marker->marker == ICC_MARKER &&
  39     marker->data_length >= ICC_OVERHEAD_LEN &&
  40     /* verify the identifying string */
  41     GETJOCTET(marker->data[0]) == 0x49 &&
  42     GETJOCTET(marker->data[1]) == 0x43 &&
  43     GETJOCTET(marker->data[2]) == 0x43 &&
  44     GETJOCTET(marker->data[3]) == 0x5F &&
  45     GETJOCTET(marker->data[4]) == 0x50 &&
  46     GETJOCTET(marker->data[5]) == 0x52 &&
  47     GETJOCTET(marker->data[6]) == 0x4F &&
  48     GETJOCTET(marker->data[7]) == 0x46 &&
  49     GETJOCTET(marker->data[8]) == 0x49 &&
  50     GETJOCTET(marker->data[9]) == 0x4C &&
  51     GETJOCTET(marker->data[10]) == 0x45 &&
  52     GETJOCTET(marker->data[11]) == 0x0;
  53 }
  54
  55
  56 /*
  57  * See if there was an ICC profile in the JPEG file being read; if so,
  58  * reassemble and return the profile data.
  59  *
  60  * TRUE is returned if an ICC profile was found, FALSE if not.  If TRUE is
  61  * returned, *icc_data_ptr is set to point to the returned data, and
  62  * *icc_data_len is set to its length.
  63  *
  64  * IMPORTANT: the data at *icc_data_ptr is allocated with malloc() and must be
  65  * freed by the caller with free() when the caller no longer needs it.
  66  * (Alternatively, we could write this routine to use the IJG library's memory
  67  * allocator, so that the data would be freed implicitly when
  68  * jpeg_finish_decompress() is called.  But it seems likely that many
  69  * applications will prefer to have the data stick around after decompression
  70  * finishes.)
  71  */
  72
  73 GLOBAL(boolean)
  74 jpeg_read_icc_profile(j_decompress_ptr cinfo, JOCTET **icc_data_ptr,
  75                       unsigned int *icc_data_len)
  76 {
  77   jpeg_saved_marker_ptr marker;
  78   int num_markers = 0;
  79   int seq_no;
  80   JOCTET *icc_data;
  81   unsigned int total_length;
  82 #define MAX_SEQ_NO  255         /* sufficient since marker numbers are bytes */
  83   char marker_present[MAX_SEQ_NO + 1];      /* 1 if marker found */
  84   unsigned int data_length[MAX_SEQ_NO + 1]; /* size of profile data in marker */
  85   unsigned int data_offset[MAX_SEQ_NO + 1]; /* offset for data in marker */
  86
  87   if (icc_data_ptr == NULL || icc_data_len == NULL)
  88     ERREXIT(cinfo, JERR_BUFFER_SIZE);
  89   if (cinfo->global_state < DSTATE_READY)
  90     ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
  91
  92   *icc_data_ptr = NULL;         /* avoid confusion if FALSE return */
  93   *icc_data_len = 0;
  94
  95   /* This first pass over the saved markers discovers whether there are
  96    * any ICC markers and verifies the consistency of the marker numbering.
  97    */
  98
  99   for (seq_no = 1; seq_no <= MAX_SEQ_NO; seq_no++)
 100     marker_present[seq_no] = 0;
 101
 102   for (marker = cinfo->marker_list; marker != NULL; marker = marker->next) {
 103     if (marker_is_icc(marker)) {
 104       if (num_markers == 0)
 105         num_markers = GETJOCTET(marker->data[13]);
 106       else if (num_markers != GETJOCTET(marker->data[13])) {
 107         WARNMS(cinfo, JWRN_BOGUS_ICC);  /* inconsistent num_markers fields */
 108         return FALSE;
 109       }
 110       seq_no = GETJOCTET(marker->data[12]);
 111       if (seq_no <= 0 || seq_no > num_markers) {
 112         WARNMS(cinfo, JWRN_BOGUS_ICC);  /* bogus sequence number */
 113         return FALSE;
 114       }
 115       if (marker_present[seq_no]) {
 116         WARNMS(cinfo, JWRN_BOGUS_ICC);  /* duplicate sequence numbers */
 117         return FALSE;
 118       }
 119       marker_present[seq_no] = 1;
 120       data_length[seq_no] = marker->data_length - ICC_OVERHEAD_LEN;
 121     }
 122   }
 123
 124   if (num_markers == 0)
 125     return FALSE;
 126
 127   /* Check for missing markers, count total space needed,
 128    * compute offset of each marker's part of the data.
 129    */
 130
 131   total_length = 0;
 132   for (seq_no = 1; seq_no <= num_markers; seq_no++) {
 133     if (marker_present[seq_no] == 0) {
 134       WARNMS(cinfo, JWRN_BOGUS_ICC);  /* missing sequence number */
 135       return FALSE;
 136     }
 137     data_offset[seq_no] = total_length;
 138     total_length += data_length[seq_no];
 139   }
 140
 141   if (total_length == 0) {
 142     WARNMS(cinfo, JWRN_BOGUS_ICC);  /* found only empty markers? */
 143     return FALSE;
 144   }
 145
 146   /* Allocate space for assembled data */
 147   icc_data = (JOCTET *)malloc(total_length * sizeof(JOCTET));
 148   if (icc_data == NULL)
 149     ERREXIT1(cinfo, JERR_OUT_OF_MEMORY, 11);  /* oops, out of memory */
 150
 151   /* and fill it in */
 152   for (marker = cinfo->marker_list; marker != NULL; marker = marker->next) {
 153     if (marker_is_icc(marker)) {
 154       JOCTET FAR *src_ptr;
 155       JOCTET *dst_ptr;
 156       unsigned int length;
 157       seq_no = GETJOCTET(marker->data[12]);
 158       dst_ptr = icc_data + data_offset[seq_no];
 159       src_ptr = marker->data + ICC_OVERHEAD_LEN;
 160       length = data_length[seq_no];
 161       while (length--) {
 162         *dst_ptr++ = *src_ptr++;
 163       }
 164     }
 165   }
 166
 167   *icc_data_ptr = icc_data;
 168   *icc_data_len = total_length;
 169
 170   return TRUE;
 171 }