1 /*
2 * jdapimin.c
3 *
4 * This file was part of the Independent JPEG Group's software:
5 * Copyright (C) 1994-1998, Thomas G. Lane.
6 * libjpeg-turbo Modifications:
7 * Copyright (C) 2016, D. R. Commander.
8 * For conditions of distribution and use, see the accompanying README.ijg
9 * file.
10 *
11 * This file contains application interface code for the decompression half
12 * of the JPEG library. These are the "minimum" API routines that may be
13 * needed in either the normal full-decompression case or the
14 * transcoding-only case.
15 *
16 * Most of the routines intended to be called directly by an application
17 * are in this file or in jdapistd.c. But also see jcomapi.c for routines
18 * shared by compression and decompression, and jdtrans.c for the transcoding
19 * case.
20 */
21
22 #define JPEG_INTERNALS
23 #include "jinclude.h"
24 #include "jpeglib.h"
25 #include "jdmaster.h"
26
27
28 /*
29 * Initialization of a JPEG decompression object.
30 * The error manager must already be set up (in case memory manager fails).
31 */
32
33 GLOBAL(void)
jpeg_CreateDecompress(j_decompress_ptr cinfo,int version,size_t structsize)34 jpeg_CreateDecompress(j_decompress_ptr cinfo, int version, size_t structsize)
35 {
36 int i;
37
38 /* Guard against version mismatches between library and caller. */
39 cinfo->mem = NULL; /* so jpeg_destroy knows mem mgr not called */
40 if (version != JPEG_LIB_VERSION)
41 ERREXIT2(cinfo, JERR_BAD_LIB_VERSION, JPEG_LIB_VERSION, version);
42 if (structsize != sizeof(struct jpeg_decompress_struct))
43 ERREXIT2(cinfo, JERR_BAD_STRUCT_SIZE,
44 (int)sizeof(struct jpeg_decompress_struct), (int)structsize);
45
46 /* For debugging purposes, we zero the whole master structure.
47 * But the application has already set the err pointer, and may have set
48 * client_data, so we have to save and restore those fields.
49 * Note: if application hasn't set client_data, tools like Purify may
50 * complain here.
51 */
52 {
53 struct jpeg_error_mgr *err = cinfo->err;
54 void *client_data = cinfo->client_data; /* ignore Purify complaint here */
55 MEMZERO(cinfo, sizeof(struct jpeg_decompress_struct));
56 cinfo->err = err;
57 cinfo->client_data = client_data;
58 }
59 cinfo->is_decompressor = TRUE;
60
61 /* Initialize a memory manager instance for this object */
62 jinit_memory_mgr((j_common_ptr)cinfo);
63
64 /* Zero out pointers to permanent structures. */
65 cinfo->progress = NULL;
66 cinfo->src = NULL;
67
68 for (i = 0; i < NUM_QUANT_TBLS; i++)
69 cinfo->quant_tbl_ptrs[i] = NULL;
70
71 for (i = 0; i < NUM_HUFF_TBLS; i++) {
72 cinfo->dc_huff_tbl_ptrs[i] = NULL;
73 cinfo->ac_huff_tbl_ptrs[i] = NULL;
74 }
75
76 /* Initialize marker processor so application can override methods
77 * for COM, APPn markers before calling jpeg_read_header.
78 */
79 cinfo->marker_list = NULL;
80 jinit_marker_reader(cinfo);
81
82 /* And initialize the overall input controller. */
83 jinit_input_controller(cinfo);
84
85 /* OK, I'm ready */
86 cinfo->global_state = DSTATE_START;
87
88 /* The master struct is used to store extension parameters, so we allocate it
89 * here.
90 */
91 cinfo->master = (struct jpeg_decomp_master *)
92 (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_PERMANENT,
93 sizeof(my_decomp_master));
94 MEMZERO(cinfo->master, sizeof(my_decomp_master));
95 }
96
97
98 /*
99 * Destruction of a JPEG decompression object
100 */
101
102 GLOBAL(void)
jpeg_destroy_decompress(j_decompress_ptr cinfo)103 jpeg_destroy_decompress(j_decompress_ptr cinfo)
104 {
105 jpeg_destroy((j_common_ptr)cinfo); /* use common routine */
106 }
107
108
109 /*
110 * Abort processing of a JPEG decompression operation,
111 * but don't destroy the object itself.
112 */
113
114 GLOBAL(void)
jpeg_abort_decompress(j_decompress_ptr cinfo)115 jpeg_abort_decompress(j_decompress_ptr cinfo)
116 {
117 jpeg_abort((j_common_ptr)cinfo); /* use common routine */
118 }
119
120
121 /*
122 * Set default decompression parameters.
123 */
124
125 LOCAL(void)
default_decompress_parms(j_decompress_ptr cinfo)126 default_decompress_parms(j_decompress_ptr cinfo)
127 {
128 /* Guess the input colorspace, and set output colorspace accordingly. */
129 /* (Wish JPEG committee had provided a real way to specify this...) */
130 /* Note application may override our guesses. */
131 switch (cinfo->num_components) {
132 case 1:
133 cinfo->jpeg_color_space = JCS_GRAYSCALE;
134 cinfo->out_color_space = JCS_GRAYSCALE;
135 break;
136
137 case 3:
138 if (cinfo->saw_JFIF_marker) {
139 cinfo->jpeg_color_space = JCS_YCbCr; /* JFIF implies YCbCr */
140 } else if (cinfo->saw_Adobe_marker) {
141 switch (cinfo->Adobe_transform) {
142 case 0:
143 cinfo->jpeg_color_space = JCS_RGB;
144 break;
145 case 1:
146 cinfo->jpeg_color_space = JCS_YCbCr;
147 break;
148 default:
149 WARNMS1(cinfo, JWRN_ADOBE_XFORM, cinfo->Adobe_transform);
150 cinfo->jpeg_color_space = JCS_YCbCr; /* assume it's YCbCr */
151 break;
152 }
153 } else {
154 /* Saw no special markers, try to guess from the component IDs */
155 int cid0 = cinfo->comp_info[0].component_id;
156 int cid1 = cinfo->comp_info[1].component_id;
157 int cid2 = cinfo->comp_info[2].component_id;
158
159 if (cid0 == 1 && cid1 == 2 && cid2 == 3)
160 cinfo->jpeg_color_space = JCS_YCbCr; /* assume JFIF w/out marker */
161 else if (cid0 == 82 && cid1 == 71 && cid2 == 66)
162 cinfo->jpeg_color_space = JCS_RGB; /* ASCII 'R', 'G', 'B' */
163 else {
164 TRACEMS3(cinfo, 1, JTRC_UNKNOWN_IDS, cid0, cid1, cid2);
165 cinfo->jpeg_color_space = JCS_YCbCr; /* assume it's YCbCr */
166 }
167 }
168 /* Always guess RGB is proper output colorspace. */
169 cinfo->out_color_space = JCS_RGB;
170 break;
171
172 case 4:
173 if (cinfo->saw_Adobe_marker) {
174 switch (cinfo->Adobe_transform) {
175 case 0:
176 cinfo->jpeg_color_space = JCS_CMYK;
177 break;
178 case 2:
179 cinfo->jpeg_color_space = JCS_YCCK;
180 break;
181 default:
182 WARNMS1(cinfo, JWRN_ADOBE_XFORM, cinfo->Adobe_transform);
183 cinfo->jpeg_color_space = JCS_YCCK; /* assume it's YCCK */
184 break;
185 }
186 } else {
187 /* No special markers, assume straight CMYK. */
188 cinfo->jpeg_color_space = JCS_CMYK;
189 }
190 cinfo->out_color_space = JCS_CMYK;
191 break;
192
193 default:
194 cinfo->jpeg_color_space = JCS_UNKNOWN;
195 cinfo->out_color_space = JCS_UNKNOWN;
196 break;
197 }
198
199 /* Set defaults for other decompression parameters. */
200 cinfo->scale_num = 1; /* 1:1 scaling */
201 cinfo->scale_denom = 1;
202 cinfo->output_gamma = 1.0;
203 cinfo->buffered_image = FALSE;
204 cinfo->raw_data_out = FALSE;
205 cinfo->dct_method = JDCT_DEFAULT;
206 cinfo->do_fancy_upsampling = TRUE;
207 cinfo->do_block_smoothing = TRUE;
208 cinfo->quantize_colors = FALSE;
209 /* We set these in case application only sets quantize_colors. */
210 cinfo->dither_mode = JDITHER_FS;
211 #ifdef QUANT_2PASS_SUPPORTED
212 cinfo->two_pass_quantize = TRUE;
213 #else
214 cinfo->two_pass_quantize = FALSE;
215 #endif
216 cinfo->desired_number_of_colors = 256;
217 cinfo->colormap = NULL;
218 /* Initialize for no mode change in buffered-image mode. */
219 cinfo->enable_1pass_quant = FALSE;
220 cinfo->enable_external_quant = FALSE;
221 cinfo->enable_2pass_quant = FALSE;
222 }
223
224
225 /*
226 * Decompression startup: read start of JPEG datastream to see what's there.
227 * Need only initialize JPEG object and supply a data source before calling.
228 *
229 * This routine will read as far as the first SOS marker (ie, actual start of
230 * compressed data), and will save all tables and parameters in the JPEG
231 * object. It will also initialize the decompression parameters to default
232 * values, and finally return JPEG_HEADER_OK. On return, the application may
233 * adjust the decompression parameters and then call jpeg_start_decompress.
234 * (Or, if the application only wanted to determine the image parameters,
235 * the data need not be decompressed. In that case, call jpeg_abort or
236 * jpeg_destroy to release any temporary space.)
237 * If an abbreviated (tables only) datastream is presented, the routine will
238 * return JPEG_HEADER_TABLES_ONLY upon reaching EOI. The application may then
239 * re-use the JPEG object to read the abbreviated image datastream(s).
240 * It is unnecessary (but OK) to call jpeg_abort in this case.
241 * The JPEG_SUSPENDED return code only occurs if the data source module
242 * requests suspension of the decompressor. In this case the application
243 * should load more source data and then re-call jpeg_read_header to resume
244 * processing.
245 * If a non-suspending data source is used and require_image is TRUE, then the
246 * return code need not be inspected since only JPEG_HEADER_OK is possible.
247 *
248 * This routine is now just a front end to jpeg_consume_input, with some
249 * extra error checking.
250 */
251
252 GLOBAL(int)
jpeg_read_header(j_decompress_ptr cinfo,boolean require_image)253 jpeg_read_header(j_decompress_ptr cinfo, boolean require_image)
254 {
255 int retcode;
256
257 if (cinfo->global_state != DSTATE_START &&
258 cinfo->global_state != DSTATE_INHEADER)
259 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
260
261 retcode = jpeg_consume_input(cinfo);
262
263 switch (retcode) {
264 case JPEG_REACHED_SOS:
265 retcode = JPEG_HEADER_OK;
266 break;
267 case JPEG_REACHED_EOI:
268 if (require_image) /* Complain if application wanted an image */
269 ERREXIT(cinfo, JERR_NO_IMAGE);
270 /* Reset to start state; it would be safer to require the application to
271 * call jpeg_abort, but we can't change it now for compatibility reasons.
272 * A side effect is to free any temporary memory (there shouldn't be any).
273 */
274 jpeg_abort((j_common_ptr)cinfo); /* sets state = DSTATE_START */
275 retcode = JPEG_HEADER_TABLES_ONLY;
276 break;
277 case JPEG_SUSPENDED:
278 /* no work */
279 break;
280 }
281
282 return retcode;
283 }
284
285
286 /*
287 * Consume data in advance of what the decompressor requires.
288 * This can be called at any time once the decompressor object has
289 * been created and a data source has been set up.
290 *
291 * This routine is essentially a state machine that handles a couple
292 * of critical state-transition actions, namely initial setup and
293 * transition from header scanning to ready-for-start_decompress.
294 * All the actual input is done via the input controller's consume_input
295 * method.
296 */
297
298 GLOBAL(int)
jpeg_consume_input(j_decompress_ptr cinfo)299 jpeg_consume_input(j_decompress_ptr cinfo)
300 {
301 int retcode = JPEG_SUSPENDED;
302
303 /* NB: every possible DSTATE value should be listed in this switch */
304 switch (cinfo->global_state) {
305 case DSTATE_START:
306 /* Start-of-datastream actions: reset appropriate modules */
307 (*cinfo->inputctl->reset_input_controller) (cinfo);
308 /* Initialize application's data source module */
309 (*cinfo->src->init_source) (cinfo);
310 cinfo->global_state = DSTATE_INHEADER;
311 /*FALLTHROUGH*/
312 case DSTATE_INHEADER:
313 retcode = (*cinfo->inputctl->consume_input) (cinfo);
314 if (retcode == JPEG_REACHED_SOS) { /* Found SOS, prepare to decompress */
315 /* Set up default parameters based on header data */
316 default_decompress_parms(cinfo);
317 /* Set global state: ready for start_decompress */
318 cinfo->global_state = DSTATE_READY;
319 }
320 break;
321 case DSTATE_READY:
322 /* Can't advance past first SOS until start_decompress is called */
323 retcode = JPEG_REACHED_SOS;
324 break;
325 case DSTATE_PRELOAD:
326 case DSTATE_PRESCAN:
327 case DSTATE_SCANNING:
328 case DSTATE_RAW_OK:
329 case DSTATE_BUFIMAGE:
330 case DSTATE_BUFPOST:
331 case DSTATE_STOPPING:
332 retcode = (*cinfo->inputctl->consume_input) (cinfo);
333 break;
334 default:
335 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
336 }
337 return retcode;
338 }
339
340
341 /*
342 * Have we finished reading the input file?
343 */
344
345 GLOBAL(boolean)
jpeg_input_complete(j_decompress_ptr cinfo)346 jpeg_input_complete(j_decompress_ptr cinfo)
347 {
348 /* Check for valid jpeg object */
349 if (cinfo->global_state < DSTATE_START ||
350 cinfo->global_state > DSTATE_STOPPING)
351 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
352 return cinfo->inputctl->eoi_reached;
353 }
354
355
356 /*
357 * Is there more than one scan?
358 */
359
360 GLOBAL(boolean)
jpeg_has_multiple_scans(j_decompress_ptr cinfo)361 jpeg_has_multiple_scans(j_decompress_ptr cinfo)
362 {
363 /* Only valid after jpeg_read_header completes */
364 if (cinfo->global_state < DSTATE_READY ||
365 cinfo->global_state > DSTATE_STOPPING)
366 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
367 return cinfo->inputctl->has_multiple_scans;
368 }
369
370
371 /*
372 * Finish JPEG decompression.
373 *
374 * This will normally just verify the file trailer and release temp storage.
375 *
376 * Returns FALSE if suspended. The return value need be inspected only if
377 * a suspending data source is used.
378 */
379
380 GLOBAL(boolean)
jpeg_finish_decompress(j_decompress_ptr cinfo)381 jpeg_finish_decompress(j_decompress_ptr cinfo)
382 {
383 if ((cinfo->global_state == DSTATE_SCANNING ||
384 cinfo->global_state == DSTATE_RAW_OK) && !cinfo->buffered_image) {
385 /* Terminate final pass of non-buffered mode */
386 if (cinfo->output_scanline < cinfo->output_height)
387 ERREXIT(cinfo, JERR_TOO_LITTLE_DATA);
388 (*cinfo->master->finish_output_pass) (cinfo);
389 cinfo->global_state = DSTATE_STOPPING;
390 } else if (cinfo->global_state == DSTATE_BUFIMAGE) {
391 /* Finishing after a buffered-image operation */
392 cinfo->global_state = DSTATE_STOPPING;
393 } else if (cinfo->global_state != DSTATE_STOPPING) {
394 /* STOPPING = repeat call after a suspension, anything else is error */
395 ERREXIT1(cinfo, JERR_BAD_STATE, cinfo->global_state);
396 }
397 /* Read until EOI */
398 while (!cinfo->inputctl->eoi_reached) {
399 if ((*cinfo->inputctl->consume_input) (cinfo) == JPEG_SUSPENDED)
400 return FALSE; /* Suspend, come back later */
401 }
402 /* Do final cleanup */
403 (*cinfo->src->term_source) (cinfo);
404 /* We can use jpeg_abort to release memory and reset global_state */
405 jpeg_abort((j_common_ptr)cinfo);
406 return TRUE;
407 }
408