jdicc.c 5.6 KB

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