Coverage Report

Created: 2026-06-30 06:41

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/opencv/3rdparty/libtiff/tif_jpeg.c
Line
Count
Source
1
/*
2
 * Copyright (c) 1994-1997 Sam Leffler
3
 * Copyright (c) 1994-1997 Silicon Graphics, Inc.
4
 *
5
 * Permission to use, copy, modify, distribute, and sell this software and
6
 * its documentation for any purpose is hereby granted without fee, provided
7
 * that (i) the above copyright notices and this permission notice appear in
8
 * all copies of the software and related documentation, and (ii) the names of
9
 * Sam Leffler and Silicon Graphics may not be used in any advertising or
10
 * publicity relating to the software without the specific, prior written
11
 * permission of Sam Leffler and Silicon Graphics.
12
 *
13
 * THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
14
 * EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
15
 * WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
16
 *
17
 * IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
18
 * ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
19
 * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
20
 * WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
21
 * LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
22
 * OF THIS SOFTWARE.
23
 */
24
25
#define WIN32_LEAN_AND_MEAN
26
#define VC_EXTRALEAN
27
28
#include "tiffiop.h"
29
#include <stdlib.h>
30
31
#ifdef JPEG_SUPPORT
32
33
/*
34
 * TIFF Library
35
 *
36
 * JPEG Compression support per TIFF Technical Note #2
37
 * (*not* per the original TIFF 6.0 spec).
38
 *
39
 * This file is simply an interface to the libjpeg library written by
40
 * the Independent JPEG Group.  You need release 5 or later of the IJG
41
 * code, which you can find on the Internet at ftp.uu.net:/graphics/jpeg/.
42
 *
43
 * Contributed by Tom Lane <tgl@sss.pgh.pa.us>.
44
 */
45
#include <setjmp.h>
46
47
/* Settings that are independent of libjpeg ABI. Used when reinitializing the */
48
/* JPEGState from libjpegs 8 bit to libjpeg 12 bits, which have potentially */
49
/* different ABI */
50
typedef struct
51
{
52
    TIFFVGetMethod vgetparent;  /* super-class method */
53
    TIFFVSetMethod vsetparent;  /* super-class method */
54
    TIFFPrintMethod printdir;   /* super-class method */
55
    TIFFStripMethod defsparent; /* super-class method */
56
    TIFFTileMethod deftparent;  /* super-class method */
57
58
    /* pseudo-tag fields */
59
    void *jpegtables;           /* JPEGTables tag value, or NULL */
60
    uint32_t jpegtables_length; /* number of bytes in same */
61
    int jpegquality;            /* Compression quality level */
62
    int jpegcolormode;          /* Auto RGB<=>YCbCr convert? */
63
    int jpegtablesmode;         /* What to put in JPEGTables */
64
65
    int ycbcrsampling_fetched;
66
    int max_allowed_scan_number;
67
    int has_warned_about_progressive_mode;
68
} JPEGOtherSettings;
69
70
int TIFFFillStrip(TIFF *tif, uint32_t strip);
71
int TIFFFillTile(TIFF *tif, uint32_t tile);
72
int TIFFReInitJPEG_12(TIFF *tif, const JPEGOtherSettings *otherSettings,
73
                      int scheme, int is_encode);
74
int TIFFJPEGIsFullStripRequired_12(TIFF *tif);
75
76
#include "jerror.h"
77
#include "jpeglib.h"
78
79
/* Do optional compile-time version check */
80
#if defined(EXPECTED_JPEG_LIB_VERSION) && !defined(LIBJPEG_12_PATH)
81
#if EXPECTED_JPEG_LIB_VERSION != JPEG_LIB_VERSION
82
#error EXPECTED_JPEG_LIB_VERSION != JPEG_LIB_VERSION
83
#endif
84
#endif
85
86
/*
87
 * Do we want to do special processing suitable for when JSAMPLE is a
88
 * 16bit value?
89
 */
90
91
/* HAVE_JPEGTURBO_DUAL_MODE_8_12 is defined for libjpeg-turbo >= 3.0 which
92
 * adds a dual-mode 8/12 bit API in the same library.
93
 * (note: libjpeg-turbo 2.2 was actually released as 3.0)
94
 */
95
96
#if defined(HAVE_JPEGTURBO_DUAL_MODE_8_12)
97
#define JPEG_DUAL_MODE_8_12
98
/* Start by undefining BITS_IN_JSAMPLE which is always set to 8 in libjpeg-turbo
99
 * >= 3.0 Cf
100
 * https://github.com/libjpeg-turbo/libjpeg-turbo/commit/8b9bc4b9635a2a047fb23ebe70c9acd728d3f99b
101
 */
102
#undef BITS_IN_JSAMPLE
103
/* libjpeg-turbo >= 3.0 adds J12xxxx datatypes for the 12-bit mode. */
104
#if defined(FROM_TIF_JPEG_12)
105
#define BITS_IN_JSAMPLE 12
106
#define TIFF_JSAMPLE J12SAMPLE
107
#define TIFF_JSAMPARRAY J12SAMPARRAY
108
#define TIFF_JSAMPIMAGE J12SAMPIMAGE
109
#define TIFF_JSAMPROW J12SAMPROW
110
#else
111
#define BITS_IN_JSAMPLE 8
112
#define TIFF_JSAMPLE JSAMPLE
113
#define TIFF_JSAMPARRAY JSAMPARRAY
114
#define TIFF_JSAMPIMAGE JSAMPIMAGE
115
#define TIFF_JSAMPROW JSAMPROW
116
#endif
117
#else
118
0
#define TIFF_JSAMPLE JSAMPLE
119
0
#define TIFF_JSAMPARRAY JSAMPARRAY
120
#define TIFF_JSAMPIMAGE JSAMPIMAGE
121
0
#define TIFF_JSAMPROW JSAMPROW
122
#endif
123
124
#if defined(JPEG_LIB_MK1)
125
#define JPEG_LIB_MK1_OR_12BIT 1
126
#elif BITS_IN_JSAMPLE == 12
127
#define JPEG_LIB_MK1_OR_12BIT 1
128
#endif
129
130
/*
131
 * We are using width_in_blocks which is supposed to be private to
132
 * libjpeg. Unfortunately, the libjpeg delivered with Cygwin has
133
 * renamed this member to width_in_data_units.  Since the header has
134
 * also renamed a define, use that unique define name in order to
135
 * detect the problem header and adjust to suit.
136
 */
137
#if defined(D_MAX_DATA_UNITS_IN_MCU)
138
#define width_in_blocks width_in_data_units
139
#endif
140
141
/*
142
 * On some machines it may be worthwhile to use _setjmp or sigsetjmp
143
 * in place of plain setjmp.  These macros will make it easier.
144
 */
145
112k
#define SETJMP(jbuf) setjmp(jbuf)
146
1.56k
#define LONGJMP(jbuf, code) longjmp(jbuf, code)
147
#define JMP_BUF jmp_buf
148
149
#ifndef TIFF_jpeg_destination_mgr_defined
150
#define TIFF_jpeg_destination_mgr_defined
151
typedef struct jpeg_destination_mgr jpeg_destination_mgr;
152
#endif
153
154
#ifndef TIFF_jpeg_source_mgr_defined
155
#define TIFF_jpeg_source_mgr_defined
156
typedef struct jpeg_source_mgr jpeg_source_mgr;
157
#endif
158
159
#ifndef TIFF_jpeg_error_mgr_defined
160
#define TIFF_jpeg_error_mgr_defined
161
typedef struct jpeg_error_mgr jpeg_error_mgr;
162
#endif
163
164
/*
165
 * State block for each open TIFF file using
166
 * libjpeg to do JPEG compression/decompression.
167
 *
168
 * libjpeg's visible state is either a jpeg_compress_struct
169
 * or jpeg_decompress_struct depending on which way we
170
 * are going.  comm can be used to refer to the fields
171
 * which are common to both.
172
 *
173
 * NB: cinfo is required to be the first member of JPEGState,
174
 *     so we can safely cast JPEGState* -> jpeg_xxx_struct*
175
 *     and vice versa!
176
 */
177
typedef struct
178
{
179
    union
180
    {
181
        struct jpeg_compress_struct c;
182
        struct jpeg_decompress_struct d;
183
        struct jpeg_common_struct comm;
184
    } cinfo; /* NB: must be first */
185
    int cinfo_initialized;
186
187
    jpeg_error_mgr err;  /* libjpeg error manager */
188
    JMP_BUF exit_jmpbuf; /* for catching libjpeg failures */
189
190
    struct jpeg_progress_mgr progress;
191
    /*
192
     * The following two members could be a union, but
193
     * they're small enough that it's not worth the effort.
194
     */
195
    jpeg_destination_mgr dest; /* data dest for compression */
196
    jpeg_source_mgr src;       /* data source for decompression */
197
                               /* private state */
198
    TIFF *tif;                 /* back link needed by some code */
199
    uint16_t photometric;      /* copy of PhotometricInterpretation */
200
    uint16_t h_sampling;       /* luminance sampling factors */
201
    uint16_t v_sampling;
202
    tmsize_t bytesperline; /* decompressed bytes per scanline */
203
    /* pointers to intermediate buffers when processing downsampled data */
204
    TIFF_JSAMPARRAY ds_buffer[MAX_COMPONENTS];
205
    int scancount; /* number of "scanlines" accumulated */
206
    int samplesperclump;
207
208
    JPEGOtherSettings otherSettings;
209
210
    int encode_raw_error;
211
} JPEGState;
212
213
67.6k
#define JState(tif) ((JPEGState *)(tif)->tif_data)
214
215
static int JPEGDecode(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s);
216
static int JPEGDecodeRaw(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s);
217
static int JPEGEncode(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s);
218
static int JPEGEncodeRaw(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s);
219
static int JPEGInitializeLibJPEG(TIFF *tif, int decode);
220
static int DecodeRowError(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s);
221
222
#define FIELD_JPEGTABLES (FIELD_CODEC + 0)
223
224
static const TIFFField jpegFields[] = {
225
    {TIFFTAG_JPEGTABLES, -3, -3, TIFF_UNDEFINED, 0, TIFF_SETGET_C32_UINT8,
226
     FIELD_JPEGTABLES, FALSE, TRUE, "JPEGTables", NULL},
227
    {TIFFTAG_JPEGQUALITY, 0, 0, TIFF_ANY, 0, TIFF_SETGET_INT, FIELD_PSEUDO,
228
     TRUE, FALSE, "", NULL},
229
    {TIFFTAG_JPEGCOLORMODE, 0, 0, TIFF_ANY, 0, TIFF_SETGET_INT, FIELD_PSEUDO,
230
     FALSE, FALSE, "", NULL},
231
    {TIFFTAG_JPEGTABLESMODE, 0, 0, TIFF_ANY, 0, TIFF_SETGET_INT, FIELD_PSEUDO,
232
     FALSE, FALSE, "", NULL}};
233
234
/*
235
 * libjpeg interface layer.
236
 *
237
 * We use setjmp/longjmp to return control to libtiff
238
 * when a fatal error is encountered within the JPEG
239
 * library.  We also direct libjpeg error and warning
240
 * messages through the appropriate libtiff handlers.
241
 */
242
243
/*
244
 * Error handling routines (these replace corresponding
245
 * IJG routines from jerror.c).  These are used for both
246
 * compression and decompression.
247
 */
248
static void TIFFjpeg_error_exit(j_common_ptr cinfo)
249
1.56k
{
250
1.56k
    JPEGState *sp = (JPEGState *)cinfo; /* NB: cinfo assumed first */
251
1.56k
    char buffer[JMSG_LENGTH_MAX];
252
253
1.56k
    (*cinfo->err->format_message)(cinfo, buffer);
254
1.56k
    TIFFErrorExtR(sp->tif, "JPEGLib", "%s",
255
1.56k
                  buffer);       /* display the error message */
256
1.56k
    jpeg_abort(cinfo);           /* clean up libjpeg state */
257
1.56k
    LONGJMP(sp->exit_jmpbuf, 1); /* return to libtiff caller */
258
1.56k
}
259
260
/*
261
 * This routine is invoked only for warning messages,
262
 * since error_exit does its own thing and trace_level
263
 * is never set > 0.
264
 */
265
static void TIFFjpeg_output_message(j_common_ptr cinfo)
266
4.64k
{
267
4.64k
    char buffer[JMSG_LENGTH_MAX];
268
269
4.64k
    (*cinfo->err->format_message)(cinfo, buffer);
270
4.64k
    TIFFWarningExtR(((JPEGState *)cinfo)->tif, "JPEGLib", "%s", buffer);
271
4.64k
}
272
273
/* Avoid the risk of denial-of-service on crafted JPEGs with an insane */
274
/* number of scans. */
275
/* See
276
 * http://www.libjpeg-turbo.org/pmwiki/uploads/About/TwoIssueswiththeJPEGStandard.pdf
277
 */
278
static void TIFFjpeg_progress_monitor(j_common_ptr cinfo)
279
206k
{
280
206k
    JPEGState *sp = (JPEGState *)cinfo; /* NB: cinfo assumed first */
281
206k
    if (cinfo->is_decompressor)
282
206k
    {
283
206k
        const int scan_no = ((j_decompress_ptr)cinfo)->input_scan_number;
284
206k
        if (scan_no >= sp->otherSettings.max_allowed_scan_number)
285
2
        {
286
2
            TIFFErrorExtR(
287
2
                ((JPEGState *)cinfo)->tif, "TIFFjpeg_progress_monitor",
288
2
                "Scan number %d exceeds maximum scans (%d). This limit "
289
2
                "can be raised through the "
290
2
                "LIBTIFF_JPEG_MAX_ALLOWED_SCAN_NUMBER "
291
2
                "environment variable.",
292
2
                scan_no, sp->otherSettings.max_allowed_scan_number);
293
294
2
            jpeg_abort(cinfo);           /* clean up libjpeg state */
295
2
            LONGJMP(sp->exit_jmpbuf, 1); /* return to libtiff caller */
296
2
        }
297
206k
    }
298
206k
}
299
300
/*
301
 * Interface routines.  This layer of routines exists
302
 * primarily to limit side-effects from using setjmp.
303
 * Also, normal/error returns are converted into return
304
 * values per libtiff practice.
305
 */
306
112k
#define CALLJPEG(sp, fail, op) (SETJMP((sp)->exit_jmpbuf) ? (fail) : (op))
307
9.11k
#define CALLVJPEG(sp, op) CALLJPEG(sp, 0, ((op), 1))
308
309
static int TIFFjpeg_create_compress(JPEGState *sp)
310
0
{
311
    /* initialize JPEG error handling */
312
0
    sp->cinfo.c.err = jpeg_std_error(&sp->err);
313
0
    sp->err.error_exit = TIFFjpeg_error_exit;
314
0
    sp->err.output_message = TIFFjpeg_output_message;
315
316
    /* set client_data to avoid UMR warning from tools like Purify */
317
0
    sp->cinfo.c.client_data = NULL;
318
319
0
    return CALLVJPEG(sp, jpeg_create_compress(&sp->cinfo.c));
320
0
}
321
322
static int TIFFjpeg_create_decompress(JPEGState *sp)
323
167
{
324
    /* initialize JPEG error handling */
325
167
    sp->cinfo.d.err = jpeg_std_error(&sp->err);
326
167
    sp->err.error_exit = TIFFjpeg_error_exit;
327
167
    sp->err.output_message = TIFFjpeg_output_message;
328
329
    /* set client_data to avoid UMR warning from tools like Purify */
330
167
    sp->cinfo.d.client_data = NULL;
331
332
167
    return CALLVJPEG(sp, jpeg_create_decompress(&sp->cinfo.d));
333
167
}
334
335
static int TIFFjpeg_set_defaults(JPEGState *sp)
336
0
{
337
0
    return CALLVJPEG(sp, jpeg_set_defaults(&sp->cinfo.c));
338
0
}
339
340
static int TIFFjpeg_set_colorspace(JPEGState *sp, J_COLOR_SPACE colorspace)
341
0
{
342
0
    return CALLVJPEG(sp, jpeg_set_colorspace(&sp->cinfo.c, colorspace));
343
0
}
344
345
static int TIFFjpeg_set_quality(JPEGState *sp, int quality,
346
                                boolean force_baseline)
347
0
{
348
0
    return CALLVJPEG(sp,
349
0
                     jpeg_set_quality(&sp->cinfo.c, quality, force_baseline));
350
0
}
351
352
static int TIFFjpeg_suppress_tables(JPEGState *sp, boolean suppress)
353
0
{
354
0
    return CALLVJPEG(sp, jpeg_suppress_tables(&sp->cinfo.c, suppress));
355
0
}
356
357
static int TIFFjpeg_start_compress(JPEGState *sp, boolean write_all_tables)
358
0
{
359
0
    return CALLVJPEG(sp, jpeg_start_compress(&sp->cinfo.c, write_all_tables));
360
0
}
361
362
static int TIFFjpeg_write_scanlines(JPEGState *sp, TIFF_JSAMPARRAY scanlines,
363
                                    int num_lines)
364
0
{
365
#if defined(HAVE_JPEGTURBO_DUAL_MODE_8_12) && BITS_IN_JSAMPLE == 12
366
    return CALLJPEG(sp, -1,
367
                    (int)jpeg12_write_scanlines(&sp->cinfo.c, scanlines,
368
                                                (JDIMENSION)num_lines));
369
#else
370
0
    return CALLJPEG(sp, -1,
371
0
                    (int)jpeg_write_scanlines(&sp->cinfo.c, scanlines,
372
0
                                              (JDIMENSION)num_lines));
373
0
#endif
374
0
}
375
376
static int TIFFjpeg_write_raw_data(JPEGState *sp, TIFF_JSAMPIMAGE data,
377
                                   int num_lines)
378
0
{
379
#if defined(HAVE_JPEGTURBO_DUAL_MODE_8_12) && BITS_IN_JSAMPLE == 12
380
    return CALLJPEG(
381
        sp, -1,
382
        (int)jpeg12_write_raw_data(&sp->cinfo.c, data, (JDIMENSION)num_lines));
383
#else
384
0
    return CALLJPEG(
385
0
        sp, -1,
386
0
        (int)jpeg_write_raw_data(&sp->cinfo.c, data, (JDIMENSION)num_lines));
387
0
#endif
388
0
}
389
390
static int TIFFjpeg_finish_compress(JPEGState *sp)
391
0
{
392
0
    return CALLVJPEG(sp, jpeg_finish_compress(&sp->cinfo.c));
393
0
}
394
395
static int TIFFjpeg_write_tables(JPEGState *sp)
396
0
{
397
0
    return CALLVJPEG(sp, jpeg_write_tables(&sp->cinfo.c));
398
0
}
399
400
static int TIFFjpeg_read_header(JPEGState *sp, boolean require_image)
401
4.94k
{
402
4.94k
    return CALLJPEG(sp, -1, jpeg_read_header(&sp->cinfo.d, require_image));
403
4.94k
}
404
405
static int TIFFjpeg_has_multiple_scans(JPEGState *sp)
406
3.85k
{
407
3.85k
    return CALLJPEG(sp, 0, jpeg_has_multiple_scans(&sp->cinfo.d));
408
3.85k
}
409
410
static int TIFFjpeg_start_decompress(JPEGState *sp)
411
3.85k
{
412
3.85k
    const char *sz_max_allowed_scan_number;
413
    /* progress monitor */
414
3.85k
    sp->cinfo.d.progress = &sp->progress;
415
3.85k
    sp->progress.progress_monitor = TIFFjpeg_progress_monitor;
416
3.85k
    sp->otherSettings.max_allowed_scan_number = 100;
417
3.85k
    sz_max_allowed_scan_number = getenv("LIBTIFF_JPEG_MAX_ALLOWED_SCAN_NUMBER");
418
3.85k
    if (sz_max_allowed_scan_number)
419
0
        sp->otherSettings.max_allowed_scan_number =
420
0
            atoi(sz_max_allowed_scan_number);
421
422
3.85k
    return CALLVJPEG(sp, jpeg_start_decompress(&sp->cinfo.d));
423
3.85k
}
424
425
static int TIFFjpeg_read_scanlines(JPEGState *sp, TIFF_JSAMPARRAY scanlines,
426
                                   int max_lines)
427
91.1k
{
428
#if defined(HAVE_JPEGTURBO_DUAL_MODE_8_12) && BITS_IN_JSAMPLE == 12
429
    return CALLJPEG(sp, -1,
430
                    (int)jpeg12_read_scanlines(&sp->cinfo.d, scanlines,
431
                                               (JDIMENSION)max_lines));
432
#else
433
91.1k
    return CALLJPEG(sp, -1,
434
91.1k
                    (int)jpeg_read_scanlines(&sp->cinfo.d, scanlines,
435
91.1k
                                             (JDIMENSION)max_lines));
436
91.1k
#endif
437
91.1k
}
438
439
static int TIFFjpeg_read_raw_data(JPEGState *sp, TIFF_JSAMPIMAGE data,
440
                                  int max_lines)
441
0
{
442
#if defined(HAVE_JPEGTURBO_DUAL_MODE_8_12) && BITS_IN_JSAMPLE == 12
443
    return CALLJPEG(
444
        sp, -1,
445
        (int)jpeg12_read_raw_data(&sp->cinfo.d, data, (JDIMENSION)max_lines));
446
#else
447
0
    return CALLJPEG(
448
0
        sp, -1,
449
0
        (int)jpeg_read_raw_data(&sp->cinfo.d, data, (JDIMENSION)max_lines));
450
0
#endif
451
0
}
452
453
static int TIFFjpeg_finish_decompress(JPEGState *sp)
454
3.58k
{
455
3.58k
    return CALLJPEG(sp, -1, (int)jpeg_finish_decompress(&sp->cinfo.d));
456
3.58k
}
457
458
static int TIFFjpeg_abort(JPEGState *sp)
459
4.92k
{
460
4.92k
    return CALLVJPEG(sp, jpeg_abort(&sp->cinfo.comm));
461
4.92k
}
462
463
static int TIFFjpeg_destroy(JPEGState *sp)
464
167
{
465
167
    return CALLVJPEG(sp, jpeg_destroy(&sp->cinfo.comm));
466
167
}
467
468
static JSAMPARRAY TIFFjpeg_alloc_sarray(JPEGState *sp, int pool_id,
469
                                        JDIMENSION samplesperrow,
470
                                        JDIMENSION numrows)
471
0
{
472
0
    return CALLJPEG(sp, (JSAMPARRAY)NULL,
473
0
                    (*sp->cinfo.comm.mem->alloc_sarray)(
474
0
                        &sp->cinfo.comm, pool_id, samplesperrow, numrows));
475
0
}
476
477
/*
478
 * JPEG library destination data manager.
479
 * These routines direct compressed data from libjpeg into the
480
 * libtiff output buffer.
481
 */
482
483
static void std_init_destination(j_compress_ptr cinfo)
484
0
{
485
0
    JPEGState *sp = (JPEGState *)cinfo;
486
0
    TIFF *tif = sp->tif;
487
488
0
    sp->dest.next_output_byte = (JOCTET *)tif->tif_rawdata;
489
0
    sp->dest.free_in_buffer = (size_t)tif->tif_rawdatasize;
490
0
}
491
492
static boolean std_empty_output_buffer(j_compress_ptr cinfo)
493
0
{
494
0
    JPEGState *sp = (JPEGState *)cinfo;
495
0
    TIFF *tif = sp->tif;
496
497
    /* the entire buffer has been filled */
498
0
    tif->tif_rawcc = tif->tif_rawdatasize;
499
500
#ifdef IPPJ_HUFF
501
    /*
502
     * The Intel IPP performance library does not necessarily fill up
503
     * the whole output buffer on each pass, so only dump out the parts
504
     * that have been filled.
505
     *   http://trac.osgeo.org/gdal/wiki/JpegIPP
506
     */
507
    if (sp->dest.free_in_buffer >= 0)
508
    {
509
        tif->tif_rawcc = tif->tif_rawdatasize - sp->dest.free_in_buffer;
510
    }
511
#endif
512
513
0
    if (!TIFFFlushData1(tif))
514
0
        return FALSE;
515
0
    sp->dest.next_output_byte = (JOCTET *)tif->tif_rawdata;
516
0
    sp->dest.free_in_buffer = (size_t)tif->tif_rawdatasize;
517
518
0
    return (TRUE);
519
0
}
520
521
static void std_term_destination(j_compress_ptr cinfo)
522
0
{
523
0
    JPEGState *sp = (JPEGState *)cinfo;
524
0
    TIFF *tif = sp->tif;
525
526
0
    tif->tif_rawcp = (uint8_t *)sp->dest.next_output_byte;
527
0
    tif->tif_rawcc = tif->tif_rawdatasize - (tmsize_t)sp->dest.free_in_buffer;
528
    /* NB: libtiff does the final buffer flush */
529
0
}
530
531
static void TIFFjpeg_data_dest(JPEGState *sp, TIFF *tif)
532
0
{
533
0
    (void)tif;
534
0
    sp->cinfo.c.dest = &sp->dest;
535
0
    sp->dest.init_destination = std_init_destination;
536
0
    sp->dest.empty_output_buffer = std_empty_output_buffer;
537
0
    sp->dest.term_destination = std_term_destination;
538
0
}
539
540
/*
541
 * Alternate destination manager for outputting to JPEGTables field.
542
 */
543
544
static void tables_init_destination(j_compress_ptr cinfo)
545
0
{
546
0
    JPEGState *sp = (JPEGState *)cinfo;
547
548
    /* while building, otherSettings.jpegtables_length is allocated buffer size
549
     */
550
0
    sp->dest.next_output_byte = (JOCTET *)sp->otherSettings.jpegtables;
551
0
    sp->dest.free_in_buffer = (size_t)sp->otherSettings.jpegtables_length;
552
0
}
553
554
static boolean tables_empty_output_buffer(j_compress_ptr cinfo)
555
0
{
556
0
    JPEGState *sp = (JPEGState *)cinfo;
557
0
    void *newbuf;
558
559
    /* the entire buffer has been filled; enlarge it by 1000 bytes */
560
0
    newbuf =
561
0
        _TIFFreallocExt(sp->tif, (void *)sp->otherSettings.jpegtables,
562
0
                        (tmsize_t)(sp->otherSettings.jpegtables_length + 1000));
563
0
    if (newbuf == NULL)
564
0
        ERREXIT1(cinfo, JERR_OUT_OF_MEMORY, 100);
565
0
    sp->dest.next_output_byte =
566
0
        (JOCTET *)newbuf + sp->otherSettings.jpegtables_length;
567
0
    sp->dest.free_in_buffer = (size_t)1000;
568
0
    sp->otherSettings.jpegtables = newbuf;
569
0
    sp->otherSettings.jpegtables_length += 1000;
570
0
    return (TRUE);
571
0
}
572
573
static void tables_term_destination(j_compress_ptr cinfo)
574
0
{
575
0
    JPEGState *sp = (JPEGState *)cinfo;
576
577
    /* set tables length to number of bytes actually emitted */
578
0
    sp->otherSettings.jpegtables_length -= (uint32_t)sp->dest.free_in_buffer;
579
0
}
580
581
static int TIFFjpeg_tables_dest(JPEGState *sp, TIFF *tif)
582
0
{
583
0
    (void)tif;
584
    /*
585
     * Allocate a working buffer for building tables.
586
     * Initial size is 1000 bytes, which is usually adequate.
587
     */
588
0
    if (sp->otherSettings.jpegtables)
589
0
        _TIFFfreeExt(tif, sp->otherSettings.jpegtables);
590
0
    sp->otherSettings.jpegtables_length = 1000;
591
0
    sp->otherSettings.jpegtables = (void *)_TIFFmallocExt(
592
0
        tif, (tmsize_t)sp->otherSettings.jpegtables_length);
593
0
    if (sp->otherSettings.jpegtables == NULL)
594
0
    {
595
0
        sp->otherSettings.jpegtables_length = 0;
596
0
        TIFFErrorExtR(sp->tif, "TIFFjpeg_tables_dest",
597
0
                      "No space for JPEGTables");
598
0
        return (0);
599
0
    }
600
0
    sp->cinfo.c.dest = &sp->dest;
601
0
    sp->dest.init_destination = tables_init_destination;
602
0
    sp->dest.empty_output_buffer = tables_empty_output_buffer;
603
0
    sp->dest.term_destination = tables_term_destination;
604
0
    return (1);
605
0
}
606
607
/*
608
 * JPEG library source data manager.
609
 * These routines supply compressed data to libjpeg.
610
 */
611
612
static void std_init_source(j_decompress_ptr cinfo)
613
4.92k
{
614
4.92k
    JPEGState *sp = (JPEGState *)cinfo;
615
4.92k
    TIFF *tif = sp->tif;
616
617
4.92k
    sp->src.next_input_byte = (const JOCTET *)tif->tif_rawdata;
618
4.92k
    sp->src.bytes_in_buffer = (size_t)tif->tif_rawcc;
619
4.92k
}
620
621
static boolean std_fill_input_buffer(j_decompress_ptr cinfo)
622
6.73k
{
623
6.73k
    JPEGState *sp = (JPEGState *)cinfo;
624
6.73k
    static const JOCTET dummy_EOI[2] = {0xFF, JPEG_EOI};
625
626
#ifdef IPPJ_HUFF
627
    /*
628
     * The Intel IPP performance library does not necessarily read the whole
629
     * input buffer in one pass, so it is possible to get here with data
630
     * yet to read.
631
     *
632
     * We just return without doing anything, until the entire buffer has
633
     * been read.
634
     * http://trac.osgeo.org/gdal/wiki/JpegIPP
635
     */
636
    if (sp->src.bytes_in_buffer > 0)
637
    {
638
        return (TRUE);
639
    }
640
#endif
641
642
    /*
643
     * Normally the whole strip/tile is read and so we don't need to do
644
     * a fill.  In the case of CHUNKY_STRIP_READ_SUPPORT we might not have
645
     * all the data, but the rawdata is refreshed between scanlines and
646
     * we push this into the io machinery in JPEGDecode().
647
     * http://trac.osgeo.org/gdal/ticket/3894
648
     */
649
650
6.73k
    WARNMS(cinfo, JWRN_JPEG_EOF);
651
    /* insert a fake EOI marker */
652
6.73k
    sp->src.next_input_byte = dummy_EOI;
653
6.73k
    sp->src.bytes_in_buffer = 2;
654
6.73k
    return (TRUE);
655
6.73k
}
656
657
static void std_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
658
21.5k
{
659
21.5k
    JPEGState *sp = (JPEGState *)cinfo;
660
661
21.5k
    if (num_bytes > 0)
662
21.5k
    {
663
21.5k
        if ((size_t)num_bytes > sp->src.bytes_in_buffer)
664
2.54k
        {
665
            /* oops, buffer overrun */
666
2.54k
            (void)std_fill_input_buffer(cinfo);
667
2.54k
        }
668
18.9k
        else
669
18.9k
        {
670
18.9k
            sp->src.next_input_byte += (size_t)num_bytes;
671
18.9k
            sp->src.bytes_in_buffer -= (size_t)num_bytes;
672
18.9k
        }
673
21.5k
    }
674
21.5k
}
675
676
static void std_term_source(j_decompress_ptr cinfo)
677
3.35k
{
678
    /* No work necessary here */
679
3.35k
    (void)cinfo;
680
3.35k
}
681
682
static void TIFFjpeg_data_src(JPEGState *sp)
683
185
{
684
185
    sp->cinfo.d.src = &sp->src;
685
185
    sp->src.init_source = std_init_source;
686
185
    sp->src.fill_input_buffer = std_fill_input_buffer;
687
185
    sp->src.skip_input_data = std_skip_input_data;
688
185
    sp->src.resync_to_restart = jpeg_resync_to_restart;
689
185
    sp->src.term_source = std_term_source;
690
185
    sp->src.bytes_in_buffer = 0; /* for safety */
691
185
    sp->src.next_input_byte = NULL;
692
185
}
693
694
/*
695
 * Alternate source manager for reading from JPEGTables.
696
 * We can share all the code except for the init routine.
697
 */
698
699
static void tables_init_source(j_decompress_ptr cinfo)
700
21
{
701
21
    JPEGState *sp = (JPEGState *)cinfo;
702
703
21
    sp->src.next_input_byte = (const JOCTET *)sp->otherSettings.jpegtables;
704
21
    sp->src.bytes_in_buffer = (size_t)sp->otherSettings.jpegtables_length;
705
21
}
706
707
static void TIFFjpeg_tables_src(JPEGState *sp)
708
21
{
709
21
    TIFFjpeg_data_src(sp);
710
21
    sp->src.init_source = tables_init_source;
711
21
}
712
713
/*
714
 * Allocate downsampled-data buffers needed for downsampled I/O.
715
 * We use values computed in jpeg_start_compress or jpeg_start_decompress.
716
 * We use libjpeg's allocator so that buffers will be released automatically
717
 * when done with strip/tile.
718
 * This is also a handy place to compute samplesperclump, bytesperline.
719
 */
720
static int alloc_downsampled_buffers(TIFF *tif, jpeg_component_info *comp_info,
721
                                     int num_components)
722
0
{
723
0
    JPEGState *sp = JState(tif);
724
0
    int ci;
725
0
    jpeg_component_info *compptr;
726
0
    TIFF_JSAMPARRAY buf;
727
0
    int samples_per_clump = 0;
728
729
0
    for (ci = 0, compptr = comp_info; ci < num_components; ci++, compptr++)
730
0
    {
731
0
        samples_per_clump += compptr->h_samp_factor * compptr->v_samp_factor;
732
0
        buf = (TIFF_JSAMPARRAY)TIFFjpeg_alloc_sarray(
733
0
            sp, JPOOL_IMAGE, compptr->width_in_blocks * DCTSIZE,
734
0
            (JDIMENSION)(compptr->v_samp_factor * DCTSIZE));
735
0
        if (buf == NULL)
736
0
            return (0);
737
0
        sp->ds_buffer[ci] = buf;
738
0
    }
739
0
    sp->samplesperclump = samples_per_clump;
740
0
    return (1);
741
0
}
742
743
/*
744
 * JPEG Decoding.
745
 */
746
747
#ifdef CHECK_JPEG_YCBCR_SUBSAMPLING
748
749
307
#define JPEG_MARKER_SOF0 0xC0
750
307
#define JPEG_MARKER_SOF1 0xC1
751
316
#define JPEG_MARKER_SOF2 0xC2
752
323
#define JPEG_MARKER_SOF9 0xC9
753
323
#define JPEG_MARKER_SOF10 0xCA
754
569k
#define JPEG_MARKER_DHT 0xC4
755
2.45k
#define JPEG_MARKER_SOI 0xD8
756
568k
#define JPEG_MARKER_SOS 0xDA
757
314k
#define JPEG_MARKER_DQT 0xDB
758
570k
#define JPEG_MARKER_DRI 0xDD
759
3.23M
#define JPEG_MARKER_APP0 0xE0
760
3.27k
#define JPEG_MARKER_COM 0xFE
761
struct JPEGFixupTagsSubsamplingData
762
{
763
    TIFF *tif;
764
    void *buffer;
765
    uint32_t buffersize;
766
    uint8_t *buffercurrentbyte;
767
    uint32_t bufferbytesleft;
768
    uint64_t fileoffset;
769
    uint64_t filebytesleft;
770
    uint8_t filepositioned;
771
};
772
static void JPEGFixupTagsSubsampling(TIFF *tif);
773
static int
774
JPEGFixupTagsSubsamplingSec(struct JPEGFixupTagsSubsamplingData *data);
775
static int
776
JPEGFixupTagsSubsamplingReadByte(struct JPEGFixupTagsSubsamplingData *data,
777
                                 uint8_t *result);
778
static int
779
JPEGFixupTagsSubsamplingReadWord(struct JPEGFixupTagsSubsamplingData *data,
780
                                 uint16_t *result);
781
static void
782
JPEGFixupTagsSubsamplingSkip(struct JPEGFixupTagsSubsamplingData *data,
783
                             uint16_t skiplength);
784
785
#endif
786
787
static int JPEGFixupTags(TIFF *tif)
788
959
{
789
959
#ifdef CHECK_JPEG_YCBCR_SUBSAMPLING
790
959
    JPEGState *sp = JState(tif);
791
959
    if ((tif->tif_dir.td_photometric == PHOTOMETRIC_YCBCR) &&
792
796
        (tif->tif_dir.td_planarconfig == PLANARCONFIG_CONTIG) &&
793
793
        (tif->tif_dir.td_samplesperpixel == 3) &&
794
791
        !sp->otherSettings.ycbcrsampling_fetched)
795
779
        JPEGFixupTagsSubsampling(tif);
796
959
#endif
797
798
959
    return (1);
799
959
}
800
801
#ifdef CHECK_JPEG_YCBCR_SUBSAMPLING
802
803
static void JPEGFixupTagsSubsampling(TIFF *tif)
804
779
{
805
    /*
806
     * Some JPEG-in-TIFF produces do not emit the YCBCRSUBSAMPLING values in
807
     * the TIFF tags, but still use non-default (2,2) values within the jpeg
808
     * data stream itself.  In order for TIFF applications to work properly
809
     * - for instance to get the strip buffer size right - it is imperative
810
     * that the subsampling be available before we start reading the image
811
     * data normally.  This function will attempt to analyze the first strip in
812
     * order to get the sampling values from the jpeg data stream.
813
     *
814
     * Note that JPEGPreDeocode() will produce a fairly loud warning when the
815
     * discovered sampling does not match the default sampling (2,2) or whatever
816
     * was actually in the tiff tags.
817
     *
818
     * See the bug in bugzilla for details:
819
     *
820
     * http://bugzilla.remotesensing.org/show_bug.cgi?id=168
821
     *
822
     * Frank Warmerdam, July 2002
823
     * Joris Van Damme, May 2007
824
     */
825
779
    static const char module[] = "JPEGFixupTagsSubsampling";
826
779
    struct JPEGFixupTagsSubsamplingData m;
827
779
    uint64_t fileoffset = TIFFGetStrileOffset(tif, 0);
828
829
779
    if (fileoffset == 0)
830
6
    {
831
        /* Do not even try to check if the first strip/tile does not
832
           yet exist, as occurs when GDAL has created a new NULL file
833
           for instance. */
834
6
        return;
835
6
    }
836
837
773
    m.tif = tif;
838
773
    m.buffersize = 2048;
839
773
    m.buffer = _TIFFmallocExt(tif, m.buffersize);
840
773
    if (m.buffer == NULL)
841
0
    {
842
0
        TIFFWarningExtR(tif, module,
843
0
                        "Unable to allocate memory for auto-correcting of "
844
0
                        "subsampling values; auto-correcting skipped");
845
0
        return;
846
0
    }
847
773
    m.buffercurrentbyte = NULL;
848
773
    m.bufferbytesleft = 0;
849
773
    m.fileoffset = fileoffset;
850
773
    m.filepositioned = 0;
851
773
    m.filebytesleft = TIFFGetStrileByteCount(tif, 0);
852
773
    if (!JPEGFixupTagsSubsamplingSec(&m))
853
640
        TIFFWarningExtR(
854
640
            tif, module,
855
640
            "Unable to auto-correct subsampling values, likely corrupt JPEG "
856
640
            "compressed data in first strip/tile; auto-correcting skipped");
857
773
    _TIFFfreeExt(tif, m.buffer);
858
773
}
859
860
static int
861
JPEGFixupTagsSubsamplingSec(struct JPEGFixupTagsSubsamplingData *data)
862
773
{
863
773
    static const char module[] = "JPEGFixupTagsSubsamplingSec";
864
773
    uint8_t m;
865
573k
    while (1)
866
573k
    {
867
4.36M
        while (1)
868
4.36M
        {
869
4.36M
            if (!JPEGFixupTagsSubsamplingReadByte(data, &m))
870
229
                return (0);
871
4.36M
            if (m == 255)
872
573k
                break;
873
4.36M
        }
874
839k
        while (1)
875
839k
        {
876
839k
            if (!JPEGFixupTagsSubsamplingReadByte(data, &m))
877
89
                return (0);
878
839k
            if (m != 255)
879
573k
                break;
880
839k
        }
881
573k
        switch (m)
882
573k
        {
883
2.45k
            case JPEG_MARKER_SOI:
884
                /* this type of marker has no data and should be skipped */
885
2.45k
                break;
886
3.27k
            case JPEG_MARKER_COM:
887
105k
            case JPEG_MARKER_APP0:
888
114k
            case JPEG_MARKER_APP0 + 1:
889
117k
            case JPEG_MARKER_APP0 + 2:
890
125k
            case JPEG_MARKER_APP0 + 3:
891
162k
            case JPEG_MARKER_APP0 + 4:
892
175k
            case JPEG_MARKER_APP0 + 5:
893
181k
            case JPEG_MARKER_APP0 + 6:
894
201k
            case JPEG_MARKER_APP0 + 7:
895
208k
            case JPEG_MARKER_APP0 + 8:
896
215k
            case JPEG_MARKER_APP0 + 9:
897
222k
            case JPEG_MARKER_APP0 + 10:
898
224k
            case JPEG_MARKER_APP0 + 11:
899
270k
            case JPEG_MARKER_APP0 + 12:
900
297k
            case JPEG_MARKER_APP0 + 13:
901
300k
            case JPEG_MARKER_APP0 + 14:
902
310k
            case JPEG_MARKER_APP0 + 15:
903
314k
            case JPEG_MARKER_DQT:
904
568k
            case JPEG_MARKER_SOS:
905
569k
            case JPEG_MARKER_DHT:
906
570k
            case JPEG_MARKER_DRI:
907
                /* this type of marker has data, but it has no use to us and
908
                 * should be skipped */
909
570k
                {
910
570k
                    uint16_t n;
911
570k
                    if (!JPEGFixupTagsSubsamplingReadWord(data, &n))
912
55
                        return (0);
913
570k
                    if (n < 2)
914
6
                        return (0);
915
570k
                    n -= 2;
916
570k
                    if (n > 0)
917
531k
                        JPEGFixupTagsSubsamplingSkip(data, n);
918
570k
                }
919
0
                break;
920
307
            case JPEG_MARKER_SOF0:  /* Baseline sequential Huffman */
921
307
            case JPEG_MARKER_SOF1:  /* Extended sequential Huffman */
922
316
            case JPEG_MARKER_SOF2:  /* Progressive Huffman: normally not allowed
923
                                       by  TechNote, but that doesn't hurt
924
                                       supporting it */
925
323
            case JPEG_MARKER_SOF9:  /* Extended sequential arithmetic */
926
323
            case JPEG_MARKER_SOF10: /* Progressive arithmetic: normally not
927
                                       allowed by TechNote, but that doesn't
928
                                       hurt supporting it */
929
                /* this marker contains the subsampling factors we're scanning
930
                 * for */
931
323
                {
932
323
                    uint16_t n;
933
323
                    uint16_t o;
934
323
                    uint8_t p;
935
323
                    uint8_t ph, pv;
936
323
                    if (!JPEGFixupTagsSubsamplingReadWord(data, &n))
937
50
                        return (0);
938
273
                    if (n != 8 + data->tif->tif_dir.td_samplesperpixel * 3)
939
0
                        return (0);
940
273
                    JPEGFixupTagsSubsamplingSkip(data, 7);
941
273
                    if (!JPEGFixupTagsSubsamplingReadByte(data, &p))
942
74
                        return (0);
943
199
                    ph = (p >> 4);
944
199
                    pv = (p & 15);
945
199
                    JPEGFixupTagsSubsamplingSkip(data, 1);
946
361
                    for (o = 1; o < data->tif->tif_dir.td_samplesperpixel; o++)
947
347
                    {
948
347
                        JPEGFixupTagsSubsamplingSkip(data, 1);
949
347
                        if (!JPEGFixupTagsSubsamplingReadByte(data, &p))
950
66
                            return (0);
951
281
                        if (p != 0x11)
952
119
                        {
953
119
                            TIFFWarningExtR(data->tif, module,
954
119
                                            "Subsampling values inside JPEG "
955
119
                                            "compressed data "
956
119
                                            "have no TIFF equivalent, "
957
119
                                            "auto-correction of TIFF "
958
119
                                            "subsampling values failed");
959
119
                            return (1);
960
119
                        }
961
162
                        JPEGFixupTagsSubsamplingSkip(data, 1);
962
162
                    }
963
14
                    if (((ph != 1) && (ph != 2) && (ph != 4)) ||
964
12
                        ((pv != 1) && (pv != 2) && (pv != 4)))
965
2
                    {
966
2
                        TIFFWarningExtR(data->tif, module,
967
2
                                        "Subsampling values inside JPEG "
968
2
                                        "compressed data have no TIFF "
969
2
                                        "equivalent, auto-correction of TIFF "
970
2
                                        "subsampling values failed");
971
2
                        return (1);
972
2
                    }
973
12
                    if ((ph != data->tif->tif_dir.td_ycbcrsubsampling[0]) ||
974
5
                        (pv != data->tif->tif_dir.td_ycbcrsubsampling[1]))
975
7
                    {
976
7
                        TIFFWarningExtR(
977
7
                            data->tif, module,
978
7
                            "Auto-corrected former TIFF subsampling values "
979
7
                            "[%" PRIu16 ",%" PRIu16
980
7
                            "] to match subsampling values inside JPEG "
981
7
                            "compressed data [%" PRIu8 ",%" PRIu8 "]",
982
7
                            data->tif->tif_dir.td_ycbcrsubsampling[0],
983
7
                            data->tif->tif_dir.td_ycbcrsubsampling[1], ph, pv);
984
7
                        data->tif->tif_dir.td_ycbcrsubsampling[0] = ph;
985
7
                        data->tif->tif_dir.td_ycbcrsubsampling[1] = pv;
986
7
                    }
987
12
                }
988
0
                return (1);
989
71
            default:
990
71
                return (0);
991
573k
        }
992
573k
    }
993
773
}
994
995
static int
996
JPEGFixupTagsSubsamplingReadByte(struct JPEGFixupTagsSubsamplingData *data,
997
                                 uint8_t *result)
998
6.34M
{
999
6.34M
    if (data->bufferbytesleft == 0)
1000
7.33k
    {
1001
7.33k
        uint32_t m;
1002
7.33k
        if (data->filebytesleft == 0)
1003
80
            return (0);
1004
7.25k
        if (!data->filepositioned)
1005
4.05k
        {
1006
4.05k
            if (TIFFSeekFile(data->tif, data->fileoffset, SEEK_SET) ==
1007
4.05k
                (toff_t)-1)
1008
1
            {
1009
1
                return 0;
1010
1
            }
1011
4.05k
            data->filepositioned = 1;
1012
4.05k
        }
1013
7.24k
        m = data->buffersize;
1014
7.24k
        if ((uint64_t)m > data->filebytesleft)
1015
89
            m = (uint32_t)data->filebytesleft;
1016
7.24k
        assert(m < 0x80000000UL);
1017
7.24k
        if (TIFFReadFile(data->tif, data->buffer, (tmsize_t)m) != (tmsize_t)m)
1018
482
            return (0);
1019
6.76k
        data->buffercurrentbyte = data->buffer;
1020
6.76k
        data->bufferbytesleft = m;
1021
6.76k
        data->fileoffset += m;
1022
6.76k
        data->filebytesleft -= m;
1023
6.76k
    }
1024
6.34M
    *result = *data->buffercurrentbyte;
1025
6.34M
    data->buffercurrentbyte++;
1026
6.34M
    data->bufferbytesleft--;
1027
6.34M
    return (1);
1028
6.34M
}
1029
1030
static int
1031
JPEGFixupTagsSubsamplingReadWord(struct JPEGFixupTagsSubsamplingData *data,
1032
                                 uint16_t *result)
1033
571k
{
1034
571k
    uint8_t ma;
1035
571k
    uint8_t mb;
1036
571k
    if (!JPEGFixupTagsSubsamplingReadByte(data, &ma))
1037
53
        return (0);
1038
571k
    if (!JPEGFixupTagsSubsamplingReadByte(data, &mb))
1039
52
        return (0);
1040
570k
    *result = (ma << 8) | mb;
1041
570k
    return (1);
1042
571k
}
1043
1044
static void
1045
JPEGFixupTagsSubsamplingSkip(struct JPEGFixupTagsSubsamplingData *data,
1046
                             uint16_t skiplength)
1047
532k
{
1048
532k
    if ((uint32_t)skiplength <= data->bufferbytesleft)
1049
529k
    {
1050
529k
        data->buffercurrentbyte += skiplength;
1051
529k
        data->bufferbytesleft -= skiplength;
1052
529k
    }
1053
3.43k
    else
1054
3.43k
    {
1055
3.43k
        uint16_t m;
1056
3.43k
        m = (uint16_t)(skiplength - data->bufferbytesleft);
1057
3.43k
        if (m <= data->filebytesleft)
1058
3.38k
        {
1059
3.38k
            data->bufferbytesleft = 0;
1060
3.38k
            data->fileoffset += m;
1061
3.38k
            data->filebytesleft -= m;
1062
3.38k
            data->filepositioned = 0;
1063
3.38k
        }
1064
52
        else
1065
52
        {
1066
52
            data->bufferbytesleft = 0;
1067
52
            data->filebytesleft = 0;
1068
52
        }
1069
3.43k
    }
1070
532k
}
1071
1072
#endif
1073
1074
static int JPEGSetupDecode(TIFF *tif)
1075
167
{
1076
167
    JPEGState *sp = JState(tif);
1077
167
    TIFFDirectory *td = &tif->tif_dir;
1078
1079
#if defined(JPEG_DUAL_MODE_8_12) && !defined(FROM_TIF_JPEG_12)
1080
    if (tif->tif_dir.td_bitspersample == 12)
1081
    {
1082
        /* We pass a pointer to a copy of otherSettings, since */
1083
        /* TIFFReInitJPEG_12() will clear sp */
1084
        JPEGOtherSettings savedOtherSettings = sp->otherSettings;
1085
        return TIFFReInitJPEG_12(tif, &savedOtherSettings, COMPRESSION_JPEG, 0);
1086
    }
1087
#endif
1088
1089
167
    JPEGInitializeLibJPEG(tif, TRUE);
1090
1091
167
    assert(sp != NULL);
1092
167
    assert(sp->cinfo.comm.is_decompressor);
1093
1094
    /* Read JPEGTables if it is present */
1095
167
    if (TIFFFieldSet(tif, FIELD_JPEGTABLES))
1096
21
    {
1097
21
        TIFFjpeg_tables_src(sp);
1098
21
        if (TIFFjpeg_read_header(sp, FALSE) != JPEG_HEADER_TABLES_ONLY)
1099
3
        {
1100
3
            TIFFErrorExtR(tif, "JPEGSetupDecode", "Bogus JPEGTables field");
1101
3
            return (0);
1102
3
        }
1103
21
    }
1104
1105
    /* Grab parameters that are same for all strips/tiles */
1106
164
    sp->photometric = td->td_photometric;
1107
164
    switch (sp->photometric)
1108
164
    {
1109
73
        case PHOTOMETRIC_YCBCR:
1110
73
            sp->h_sampling = td->td_ycbcrsubsampling[0];
1111
73
            sp->v_sampling = td->td_ycbcrsubsampling[1];
1112
73
            break;
1113
91
        default:
1114
            /* TIFF 6.0 forbids subsampling of all other color spaces */
1115
91
            sp->h_sampling = 1;
1116
91
            sp->v_sampling = 1;
1117
91
            break;
1118
164
    }
1119
1120
    /* Set up for reading normal data */
1121
164
    TIFFjpeg_data_src(sp);
1122
164
    tif->tif_postdecode = _TIFFNoPostDecode; /* override byte swapping */
1123
164
    return (1);
1124
164
}
1125
1126
/* Returns 1 if the full strip should be read, even when doing scanline per */
1127
/* scanline decoding. This happens when the JPEG stream uses multiple scans. */
1128
/* Currently only called in CHUNKY_STRIP_READ_SUPPORT mode through */
1129
/* scanline interface. */
1130
/* Only reads tif->tif_dir.td_bitspersample, tif->tif_rawdata and */
1131
/* tif->tif_rawcc members. */
1132
/* Can be called independently of the usual setup/predecode/decode states */
1133
int TIFFJPEGIsFullStripRequired(TIFF *tif)
1134
0
{
1135
0
    int ret;
1136
0
    JPEGState state;
1137
1138
#if defined(JPEG_DUAL_MODE_8_12) && !defined(FROM_TIF_JPEG_12)
1139
    if (tif->tif_dir.td_bitspersample == 12)
1140
        return TIFFJPEGIsFullStripRequired_12(tif);
1141
#endif
1142
1143
0
    memset(&state, 0, sizeof(JPEGState));
1144
0
    state.tif = tif;
1145
1146
0
    TIFFjpeg_create_decompress(&state);
1147
1148
0
    TIFFjpeg_data_src(&state);
1149
1150
0
    if (TIFFjpeg_read_header(&state, TRUE) != JPEG_HEADER_OK)
1151
0
    {
1152
0
        TIFFjpeg_destroy(&state);
1153
0
        return (0);
1154
0
    }
1155
0
    ret = TIFFjpeg_has_multiple_scans(&state);
1156
1157
0
    TIFFjpeg_destroy(&state);
1158
1159
0
    return ret;
1160
0
}
1161
1162
/*
1163
 * Set up for decoding a strip or tile.
1164
 */
1165
/*ARGSUSED*/ static int JPEGPreDecode(TIFF *tif, uint16_t s)
1166
4.92k
{
1167
4.92k
    JPEGState *sp = JState(tif);
1168
4.92k
    TIFFDirectory *td = &tif->tif_dir;
1169
4.92k
    static const char module[] = "JPEGPreDecode";
1170
4.92k
    uint32_t segment_width, segment_height;
1171
4.92k
    int downsampled_output;
1172
4.92k
    int ci;
1173
1174
4.92k
    assert(sp != NULL);
1175
1176
4.92k
    if (sp->cinfo.comm.is_decompressor == 0)
1177
0
    {
1178
0
        tif->tif_setupdecode(tif);
1179
0
    }
1180
1181
4.92k
    assert(sp->cinfo.comm.is_decompressor);
1182
    /*
1183
     * Reset decoder state from any previous strip/tile,
1184
     * in case application didn't read the whole strip.
1185
     */
1186
4.92k
    if (!TIFFjpeg_abort(sp))
1187
0
        return (0);
1188
    /*
1189
     * Read the header for this strip/tile.
1190
     */
1191
1192
4.92k
    if (TIFFjpeg_read_header(sp, TRUE) != JPEG_HEADER_OK)
1193
1.07k
        return (0);
1194
1195
3.85k
    tif->tif_rawcp = (uint8_t *)sp->src.next_input_byte;
1196
3.85k
    tif->tif_rawcc = sp->src.bytes_in_buffer;
1197
1198
    /*
1199
     * Check image parameters and set decompression parameters.
1200
     */
1201
3.85k
    if (isTiled(tif))
1202
3.77k
    {
1203
3.77k
        segment_width = td->td_tilewidth;
1204
3.77k
        segment_height = td->td_tilelength;
1205
3.77k
        sp->bytesperline = TIFFTileRowSize(tif);
1206
3.77k
    }
1207
71
    else
1208
71
    {
1209
71
        segment_width = td->td_imagewidth;
1210
71
        segment_height = td->td_imagelength - tif->tif_row;
1211
71
        if (segment_height > td->td_rowsperstrip)
1212
13
            segment_height = td->td_rowsperstrip;
1213
71
        sp->bytesperline = TIFFScanlineSize(tif);
1214
71
    }
1215
3.85k
    if (td->td_planarconfig == PLANARCONFIG_SEPARATE && s > 0)
1216
1.47k
    {
1217
        /*
1218
         * For PC 2, scale down the expected strip/tile size
1219
         * to match a downsampled component
1220
         */
1221
1.47k
        if (sp->h_sampling == 0 || sp->v_sampling == 0)
1222
0
        {
1223
0
            TIFFErrorExtR(tif, module,
1224
0
                          "JPEG horizontal or vertical sampling is zero");
1225
0
            return (0);
1226
0
        }
1227
1.47k
        segment_width = TIFFhowmany_32(segment_width, sp->h_sampling);
1228
1.47k
        segment_height = TIFFhowmany_32(segment_height, sp->v_sampling);
1229
1.47k
    }
1230
3.85k
    if (sp->cinfo.d.image_width < segment_width ||
1231
27
        sp->cinfo.d.image_height < segment_height)
1232
3.83k
    {
1233
3.83k
        TIFFWarningExtR(tif, module,
1234
3.83k
                        "Improper JPEG strip/tile size, "
1235
3.83k
                        "expected %" PRIu32 "x%" PRIu32 ", got %ux%u",
1236
3.83k
                        segment_width, segment_height, sp->cinfo.d.image_width,
1237
3.83k
                        sp->cinfo.d.image_height);
1238
3.83k
    }
1239
3.85k
    if (sp->cinfo.d.image_width == segment_width &&
1240
27
        sp->cinfo.d.image_height > segment_height &&
1241
8
        tif->tif_row + segment_height == td->td_imagelength && !isTiled(tif))
1242
8
    {
1243
        /* Some files have a last strip, that should be truncated, */
1244
        /* but their JPEG codestream has still the maximum strip */
1245
        /* height. Warn about this as this is non compliant, but */
1246
        /* we can safely recover from that. */
1247
8
        TIFFWarningExtR(tif, module,
1248
8
                        "JPEG strip size exceeds expected dimensions,"
1249
8
                        " expected %" PRIu32 "x%" PRIu32 ", got %ux%u",
1250
8
                        segment_width, segment_height, sp->cinfo.d.image_width,
1251
8
                        sp->cinfo.d.image_height);
1252
8
    }
1253
3.84k
    else if (sp->cinfo.d.image_width > segment_width ||
1254
3.84k
             sp->cinfo.d.image_height > segment_height)
1255
0
    {
1256
        /*
1257
         * This case could be dangerous, if the strip or tile size has
1258
         * been reported as less than the amount of data jpeg will
1259
         * return, some potential security issues arise. Catch this
1260
         * case and error out.
1261
         */
1262
0
        TIFFErrorExtR(tif, module,
1263
0
                      "JPEG strip/tile size exceeds expected dimensions,"
1264
0
                      " expected %" PRIu32 "x%" PRIu32 ", got %ux%u",
1265
0
                      segment_width, segment_height, sp->cinfo.d.image_width,
1266
0
                      sp->cinfo.d.image_height);
1267
0
        return (0);
1268
0
    }
1269
3.85k
    if (sp->cinfo.d.num_components !=
1270
3.85k
        (td->td_planarconfig == PLANARCONFIG_CONTIG ? td->td_samplesperpixel
1271
3.85k
                                                    : 1))
1272
0
    {
1273
0
        TIFFErrorExtR(tif, module, "Improper JPEG component count");
1274
0
        return (0);
1275
0
    }
1276
#ifdef JPEG_LIB_MK1
1277
    if (12 != td->td_bitspersample && 8 != td->td_bitspersample)
1278
    {
1279
        TIFFErrorExtR(tif, module, "Improper JPEG data precision");
1280
        return (0);
1281
    }
1282
    sp->cinfo.d.data_precision = td->td_bitspersample;
1283
    sp->cinfo.d.bits_in_jsample = td->td_bitspersample;
1284
#else
1285
3.85k
    if (sp->cinfo.d.data_precision != td->td_bitspersample)
1286
0
    {
1287
0
        TIFFErrorExtR(tif, module, "Improper JPEG data precision");
1288
0
        return (0);
1289
0
    }
1290
3.85k
#endif
1291
1292
3.85k
    if (sp->cinfo.d.progressive_mode &&
1293
3.48k
        !sp->otherSettings.has_warned_about_progressive_mode)
1294
87
    {
1295
87
        TIFFWarningExtR(tif, module,
1296
87
                        "The JPEG strip/tile is encoded with progressive mode, "
1297
87
                        "which is normally not legal for JPEG-in-TIFF.\n"
1298
87
                        "libtiff should be able to decode it, but it might "
1299
87
                        "cause compatibility issues with other readers");
1300
87
        sp->otherSettings.has_warned_about_progressive_mode = TRUE;
1301
87
    }
1302
1303
    /* In some cases, libjpeg needs to allocate a lot of memory */
1304
    /* http://www.libjpeg-turbo.org/pmwiki/uploads/About/TwoIssueswiththeJPEGStandard.pdf
1305
     */
1306
3.85k
    if (TIFFjpeg_has_multiple_scans(sp))
1307
3.50k
    {
1308
        /* In this case libjpeg will need to allocate memory or backing */
1309
        /* store for all coefficients */
1310
        /* See call to jinit_d_coef_controller() from master_selection() */
1311
        /* in libjpeg */
1312
1313
        /* 1 MB for regular libjpeg usage */
1314
3.50k
        toff_t nRequiredMemory = 1024 * 1024;
1315
1316
7.06k
        for (ci = 0; ci < sp->cinfo.d.num_components; ci++)
1317
3.55k
        {
1318
3.55k
            const jpeg_component_info *compptr = &(sp->cinfo.d.comp_info[ci]);
1319
3.55k
            if (compptr->h_samp_factor > 0 && compptr->v_samp_factor > 0)
1320
3.55k
            {
1321
3.55k
                nRequiredMemory +=
1322
3.55k
                    (toff_t)(((compptr->width_in_blocks +
1323
3.55k
                               compptr->h_samp_factor - 1) /
1324
3.55k
                              compptr->h_samp_factor)) *
1325
3.55k
                    ((compptr->height_in_blocks + compptr->v_samp_factor - 1) /
1326
3.55k
                     compptr->v_samp_factor) *
1327
3.55k
                    sizeof(JBLOCK);
1328
3.55k
            }
1329
3.55k
        }
1330
1331
3.50k
        if (sp->cinfo.d.mem->max_memory_to_use > 0 &&
1332
0
            nRequiredMemory > (toff_t)(sp->cinfo.d.mem->max_memory_to_use) &&
1333
0
            getenv("LIBTIFF_ALLOW_LARGE_LIBJPEG_MEM_ALLOC") == NULL)
1334
0
        {
1335
0
            TIFFErrorExtR(
1336
0
                tif, module,
1337
0
                "Reading this image would require libjpeg to allocate "
1338
0
                "at least %" PRIu64 " bytes. "
1339
0
                "This is disabled since above the %ld threshold. "
1340
0
                "You may override this restriction by defining the "
1341
0
                "LIBTIFF_ALLOW_LARGE_LIBJPEG_MEM_ALLOC environment variable, "
1342
0
                "or setting the JPEGMEM environment variable to a value "
1343
0
                "greater "
1344
0
                "or equal to '%" PRIu64 "M'",
1345
0
                nRequiredMemory, sp->cinfo.d.mem->max_memory_to_use,
1346
0
                (nRequiredMemory + 1000000u - 1u) / 1000000u);
1347
0
            return 0;
1348
0
        }
1349
3.50k
    }
1350
1351
3.85k
    if (td->td_planarconfig == PLANARCONFIG_CONTIG)
1352
80
    {
1353
        /* Component 0 should have expected sampling factors */
1354
80
        if (sp->cinfo.d.comp_info[0].h_samp_factor != sp->h_sampling ||
1355
80
            sp->cinfo.d.comp_info[0].v_samp_factor != sp->v_sampling)
1356
0
        {
1357
0
            TIFFErrorExtR(tif, module,
1358
0
                          "Improper JPEG sampling factors %d,%d\n"
1359
0
                          "Apparently should be %" PRIu16 ",%" PRIu16 ".",
1360
0
                          sp->cinfo.d.comp_info[0].h_samp_factor,
1361
0
                          sp->cinfo.d.comp_info[0].v_samp_factor,
1362
0
                          sp->h_sampling, sp->v_sampling);
1363
0
            return (0);
1364
0
        }
1365
        /* Rest should have sampling factors 1,1 */
1366
149
        for (ci = 1; ci < sp->cinfo.d.num_components; ci++)
1367
69
        {
1368
69
            if (sp->cinfo.d.comp_info[ci].h_samp_factor != 1 ||
1369
69
                sp->cinfo.d.comp_info[ci].v_samp_factor != 1)
1370
0
            {
1371
0
                TIFFErrorExtR(tif, module, "Improper JPEG sampling factors");
1372
0
                return (0);
1373
0
            }
1374
69
        }
1375
80
    }
1376
3.77k
    else
1377
3.77k
    {
1378
        /* PC 2's single component should have sampling factors 1,1 */
1379
3.77k
        if (sp->cinfo.d.comp_info[0].h_samp_factor != 1 ||
1380
3.77k
            sp->cinfo.d.comp_info[0].v_samp_factor != 1)
1381
0
        {
1382
0
            TIFFErrorExtR(tif, module, "Improper JPEG sampling factors");
1383
0
            return (0);
1384
0
        }
1385
3.77k
    }
1386
3.85k
    downsampled_output = FALSE;
1387
3.85k
    if (td->td_planarconfig == PLANARCONFIG_CONTIG &&
1388
80
        sp->photometric == PHOTOMETRIC_YCBCR &&
1389
30
        sp->otherSettings.jpegcolormode == JPEGCOLORMODE_RGB)
1390
30
    {
1391
        /* Convert YCbCr to RGB */
1392
30
        sp->cinfo.d.jpeg_color_space = JCS_YCbCr;
1393
30
        sp->cinfo.d.out_color_space = JCS_RGB;
1394
30
    }
1395
3.82k
    else
1396
3.82k
    {
1397
        /* Suppress colorspace handling */
1398
3.82k
        sp->cinfo.d.jpeg_color_space = JCS_UNKNOWN;
1399
3.82k
        sp->cinfo.d.out_color_space = JCS_UNKNOWN;
1400
3.82k
        if (td->td_planarconfig == PLANARCONFIG_CONTIG &&
1401
50
            (sp->h_sampling != 1 || sp->v_sampling != 1))
1402
0
            downsampled_output = TRUE;
1403
        /* XXX what about up-sampling? */
1404
3.82k
    }
1405
3.85k
    if (downsampled_output)
1406
0
    {
1407
        /* Need to use raw-data interface to libjpeg */
1408
0
        sp->cinfo.d.raw_data_out = TRUE;
1409
0
#if JPEG_LIB_VERSION >= 70
1410
0
        sp->cinfo.d.do_fancy_upsampling = FALSE;
1411
0
#endif /* JPEG_LIB_VERSION >= 70 */
1412
0
        tif->tif_decoderow = DecodeRowError;
1413
0
        tif->tif_decodestrip = JPEGDecodeRaw;
1414
0
        tif->tif_decodetile = JPEGDecodeRaw;
1415
0
    }
1416
3.85k
    else
1417
3.85k
    {
1418
        /* Use normal interface to libjpeg */
1419
3.85k
        sp->cinfo.d.raw_data_out = FALSE;
1420
3.85k
        tif->tif_decoderow = JPEGDecode;
1421
3.85k
        tif->tif_decodestrip = JPEGDecode;
1422
3.85k
        tif->tif_decodetile = JPEGDecode;
1423
3.85k
    }
1424
    /* Start JPEG decompressor */
1425
3.85k
    if (!TIFFjpeg_start_decompress(sp))
1426
254
        return (0);
1427
    /* Allocate downsampled-data buffers if needed */
1428
3.59k
    if (downsampled_output)
1429
0
    {
1430
0
        if (!alloc_downsampled_buffers(tif, sp->cinfo.d.comp_info,
1431
0
                                       sp->cinfo.d.num_components))
1432
0
            return (0);
1433
0
        sp->scancount = DCTSIZE; /* mark buffer empty */
1434
0
    }
1435
3.59k
    return (1);
1436
3.59k
}
1437
1438
/*
1439
 * Decode a chunk of pixels.
1440
 * "Standard" case: returned data is not downsampled.
1441
 */
1442
#if !JPEG_LIB_MK1_OR_12BIT
1443
static int JPEGDecode(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s)
1444
3.59k
{
1445
3.59k
    JPEGState *sp = JState(tif);
1446
3.59k
    tmsize_t nrows;
1447
3.59k
    (void)s;
1448
1449
    /*
1450
    ** Update available information, buffer may have been refilled
1451
    ** between decode requests
1452
    */
1453
3.59k
    sp->src.next_input_byte = (const JOCTET *)tif->tif_rawcp;
1454
3.59k
    sp->src.bytes_in_buffer = (size_t)tif->tif_rawcc;
1455
1456
3.59k
    if (sp->bytesperline == 0)
1457
0
    {
1458
0
        memset(buf, 0, (size_t)cc);
1459
0
        return 0;
1460
0
    }
1461
1462
3.59k
    nrows = cc / sp->bytesperline;
1463
3.59k
    if (cc % sp->bytesperline)
1464
0
        TIFFWarningExtR(tif, tif->tif_name, "fractional scanline not read");
1465
1466
3.59k
    if (nrows > (tmsize_t)sp->cinfo.d.image_height)
1467
3.57k
        nrows = sp->cinfo.d.image_height;
1468
1469
    /* data is expected to be read in multiples of a scanline */
1470
3.59k
    if (nrows)
1471
3.59k
    {
1472
3.59k
        do
1473
91.1k
        {
1474
            /*
1475
             * In the libjpeg6b-9a 8bit case.  We read directly into
1476
             * the TIFF buffer.
1477
             */
1478
91.1k
            JSAMPROW bufptr = (JSAMPROW)buf;
1479
1480
91.1k
            if (TIFFjpeg_read_scanlines(sp, &bufptr, 1) != 1)
1481
0
            {
1482
0
                memset(buf, 0, (size_t)cc);
1483
0
                return (0);
1484
0
            }
1485
1486
91.1k
            ++tif->tif_row;
1487
91.1k
            buf += sp->bytesperline;
1488
91.1k
            cc -= sp->bytesperline;
1489
91.1k
        } while (--nrows > 0);
1490
3.59k
    }
1491
1492
    /* Update information on consumed data */
1493
3.59k
    tif->tif_rawcp = (uint8_t *)sp->src.next_input_byte;
1494
3.59k
    tif->tif_rawcc = sp->src.bytes_in_buffer;
1495
1496
    /* Close down the decompressor if we've finished the strip or tile. */
1497
3.59k
    return sp->cinfo.d.output_scanline < sp->cinfo.d.output_height ||
1498
3.58k
           TIFFjpeg_finish_decompress(sp);
1499
3.59k
}
1500
#endif /* !JPEG_LIB_MK1_OR_12BIT */
1501
1502
#if JPEG_LIB_MK1_OR_12BIT
1503
/*ARGSUSED*/ static int JPEGDecode(TIFF *tif, uint8_t *buf, tmsize_t cc,
1504
                                   uint16_t s)
1505
{
1506
    JPEGState *sp = JState(tif);
1507
    tmsize_t nrows;
1508
    (void)s;
1509
1510
    /*
1511
    ** Update available information, buffer may have been refilled
1512
    ** between decode requests
1513
    */
1514
    sp->src.next_input_byte = (const JOCTET *)tif->tif_rawcp;
1515
    sp->src.bytes_in_buffer = (size_t)tif->tif_rawcc;
1516
1517
    if (sp->bytesperline == 0)
1518
    {
1519
        memset(buf, 0, (size_t)cc);
1520
        return 0;
1521
    }
1522
1523
    nrows = cc / sp->bytesperline;
1524
    if (cc % sp->bytesperline)
1525
        TIFFWarningExtR(tif, tif->tif_name, "fractional scanline not read");
1526
1527
    if (nrows > (tmsize_t)sp->cinfo.d.image_height)
1528
        nrows = sp->cinfo.d.image_height;
1529
1530
    /* data is expected to be read in multiples of a scanline */
1531
    if (nrows)
1532
    {
1533
        TIFF_JSAMPROW line_work_buf = NULL;
1534
1535
        /*
1536
         * For 6B, only use temporary buffer for 12 bit imagery.
1537
         * For Mk1 always use it.
1538
         */
1539
        if (sp->cinfo.d.data_precision == 12)
1540
        {
1541
            line_work_buf = (TIFF_JSAMPROW)_TIFFmallocExt(
1542
                tif, sizeof(short) * sp->cinfo.d.output_width *
1543
                         sp->cinfo.d.num_components);
1544
        }
1545
1546
        do
1547
        {
1548
            if (line_work_buf != NULL)
1549
            {
1550
                /*
1551
                 * In the MK1 case, we always read into a 16bit
1552
                 * buffer, and then pack down to 12bit or 8bit.
1553
                 * In 6B case we only read into 16 bit buffer
1554
                 * for 12bit data, which we need to repack.
1555
                 */
1556
                if (TIFFjpeg_read_scanlines(sp, &line_work_buf, 1) != 1)
1557
                {
1558
                    memset(buf, 0, (size_t)cc);
1559
                    return (0);
1560
                }
1561
1562
                if (sp->cinfo.d.data_precision == 12)
1563
                {
1564
                    int value_pairs = (sp->cinfo.d.output_width *
1565
                                       sp->cinfo.d.num_components) /
1566
                                      2;
1567
                    int iPair;
1568
1569
                    for (iPair = 0; iPair < value_pairs; iPair++)
1570
                    {
1571
                        unsigned char *out_ptr =
1572
                            ((unsigned char *)buf) + iPair * 3;
1573
                        TIFF_JSAMPLE *in_ptr = line_work_buf + iPair * 2;
1574
1575
                        out_ptr[0] = (unsigned char)((in_ptr[0] & 0xff0) >> 4);
1576
                        out_ptr[1] =
1577
                            (unsigned char)(((in_ptr[0] & 0xf) << 4) |
1578
                                            ((in_ptr[1] & 0xf00) >> 8));
1579
                        out_ptr[2] = (unsigned char)(((in_ptr[1] & 0xff) >> 0));
1580
                    }
1581
                }
1582
                else if (sp->cinfo.d.data_precision == 8)
1583
                {
1584
                    int value_count =
1585
                        (sp->cinfo.d.output_width * sp->cinfo.d.num_components);
1586
                    int iValue;
1587
1588
                    for (iValue = 0; iValue < value_count; iValue++)
1589
                    {
1590
                        ((unsigned char *)buf)[iValue] =
1591
                            line_work_buf[iValue] & 0xff;
1592
                    }
1593
                }
1594
            }
1595
1596
            ++tif->tif_row;
1597
            buf += sp->bytesperline;
1598
            cc -= sp->bytesperline;
1599
        } while (--nrows > 0);
1600
1601
        if (line_work_buf != NULL)
1602
            _TIFFfreeExt(tif, line_work_buf);
1603
    }
1604
1605
    /* Update information on consumed data */
1606
    tif->tif_rawcp = (uint8_t *)sp->src.next_input_byte;
1607
    tif->tif_rawcc = sp->src.bytes_in_buffer;
1608
1609
    /* Close down the decompressor if we've finished the strip or tile. */
1610
    return sp->cinfo.d.output_scanline < sp->cinfo.d.output_height ||
1611
           TIFFjpeg_finish_decompress(sp);
1612
}
1613
#endif /* JPEG_LIB_MK1_OR_12BIT */
1614
1615
/*ARGSUSED*/ static int DecodeRowError(TIFF *tif, uint8_t *buf, tmsize_t cc,
1616
                                       uint16_t s)
1617
1618
0
{
1619
0
    (void)buf;
1620
0
    (void)cc;
1621
0
    (void)s;
1622
1623
0
    TIFFErrorExtR(
1624
0
        tif, "TIFFReadScanline",
1625
0
        "scanline oriented access is not supported for downsampled JPEG "
1626
0
        "compressed images, consider enabling TIFFTAG_JPEGCOLORMODE as "
1627
0
        "JPEGCOLORMODE_RGB.");
1628
0
    return 0;
1629
0
}
1630
1631
/*
1632
 * Decode a chunk of pixels.
1633
 * Returned data is downsampled per sampling factors.
1634
 */
1635
/*ARGSUSED*/ static int JPEGDecodeRaw(TIFF *tif, uint8_t *buf, tmsize_t cc,
1636
                                      uint16_t s)
1637
0
{
1638
0
    JPEGState *sp = JState(tif);
1639
0
    tmsize_t nrows;
1640
0
    TIFFDirectory *td = &tif->tif_dir;
1641
0
    (void)s;
1642
1643
0
    nrows = sp->cinfo.d.image_height;
1644
    /* For last strip, limit number of rows to its truncated height */
1645
    /* even if the codestream height is larger (which is not compliant, */
1646
    /* but that we tolerate) */
1647
0
    if ((uint32_t)nrows > td->td_imagelength - tif->tif_row && !isTiled(tif))
1648
0
        nrows = td->td_imagelength - tif->tif_row;
1649
1650
#if defined(JPEG_LIB_MK1_OR_12BIT)
1651
    unsigned short *tmpbuf = NULL;
1652
#endif
1653
1654
    /* data is expected to be read in multiples of a scanline */
1655
0
    if (nrows != 0)
1656
0
    {
1657
1658
        /* Cb,Cr both have sampling factors 1, so this is correct */
1659
0
        JDIMENSION clumps_per_line = sp->cinfo.d.comp_info[1].downsampled_width;
1660
0
        int samples_per_clump = sp->samplesperclump;
1661
1662
#if defined(JPEG_LIB_MK1_OR_12BIT)
1663
        tmpbuf = _TIFFmallocExt(tif, sizeof(unsigned short) *
1664
                                         sp->cinfo.d.output_width *
1665
                                         sp->cinfo.d.num_components);
1666
        if (tmpbuf == NULL)
1667
        {
1668
            TIFFErrorExtR(tif, "JPEGDecodeRaw", "Out of memory");
1669
            return 0;
1670
        }
1671
#endif
1672
1673
0
        do
1674
0
        {
1675
0
            jpeg_component_info *compptr;
1676
0
            int ci, clumpoffset;
1677
1678
0
            if (cc < sp->bytesperline)
1679
0
            {
1680
0
                TIFFErrorExtR(
1681
0
                    tif, "JPEGDecodeRaw",
1682
0
                    "application buffer not large enough for all data.");
1683
0
                goto error;
1684
0
            }
1685
1686
            /* Reload downsampled-data buffer if needed */
1687
0
            if (sp->scancount >= DCTSIZE)
1688
0
            {
1689
0
                int n = sp->cinfo.d.max_v_samp_factor * DCTSIZE;
1690
0
                if (TIFFjpeg_read_raw_data(sp, sp->ds_buffer, n) != n)
1691
0
                    goto error;
1692
0
                sp->scancount = 0;
1693
0
            }
1694
            /*
1695
             * Fastest way to unseparate data is to make one pass
1696
             * over the scanline for each row of each component.
1697
             */
1698
0
            clumpoffset = 0; /* first sample in clump */
1699
0
            for (ci = 0, compptr = sp->cinfo.d.comp_info;
1700
0
                 ci < sp->cinfo.d.num_components; ci++, compptr++)
1701
0
            {
1702
0
                int hsamp = compptr->h_samp_factor;
1703
0
                int vsamp = compptr->v_samp_factor;
1704
0
                int ypos;
1705
1706
0
                for (ypos = 0; ypos < vsamp; ypos++)
1707
0
                {
1708
0
                    TIFF_JSAMPLE *inptr =
1709
0
                        sp->ds_buffer[ci][sp->scancount * vsamp + ypos];
1710
0
                    JDIMENSION nclump;
1711
#if defined(JPEG_LIB_MK1_OR_12BIT)
1712
                    TIFF_JSAMPLE *outptr = (TIFF_JSAMPLE *)tmpbuf + clumpoffset;
1713
#else
1714
0
                    TIFF_JSAMPLE *outptr = (TIFF_JSAMPLE *)buf + clumpoffset;
1715
0
                    if (cc < (tmsize_t)(clumpoffset +
1716
0
                                        (tmsize_t)samples_per_clump *
1717
0
                                            (clumps_per_line - 1) +
1718
0
                                        hsamp))
1719
0
                    {
1720
0
                        TIFFErrorExtR(
1721
0
                            tif, "JPEGDecodeRaw",
1722
0
                            "application buffer not large enough for all data, "
1723
0
                            "possible subsampling issue");
1724
0
                        goto error;
1725
0
                    }
1726
0
#endif
1727
1728
0
                    if (hsamp == 1)
1729
0
                    {
1730
                        /* fast path for at least Cb and Cr */
1731
0
                        for (nclump = clumps_per_line; nclump-- > 0;)
1732
0
                        {
1733
0
                            outptr[0] = *inptr++;
1734
0
                            outptr += samples_per_clump;
1735
0
                        }
1736
0
                    }
1737
0
                    else
1738
0
                    {
1739
0
                        int xpos;
1740
1741
                        /* general case */
1742
0
                        for (nclump = clumps_per_line; nclump-- > 0;)
1743
0
                        {
1744
0
                            for (xpos = 0; xpos < hsamp; xpos++)
1745
0
                                outptr[xpos] = *inptr++;
1746
0
                            outptr += samples_per_clump;
1747
0
                        }
1748
0
                    }
1749
0
                    clumpoffset += hsamp;
1750
0
                }
1751
0
            }
1752
1753
#if defined(JPEG_LIB_MK1_OR_12BIT)
1754
            {
1755
                if (sp->cinfo.d.data_precision == 8)
1756
                {
1757
                    int i = 0;
1758
                    int len =
1759
                        sp->cinfo.d.output_width * sp->cinfo.d.num_components;
1760
                    for (i = 0; i < len; i++)
1761
                    {
1762
                        ((unsigned char *)buf)[i] = tmpbuf[i] & 0xff;
1763
                    }
1764
                }
1765
                else
1766
                { /* 12-bit */
1767
                    int value_pairs = (sp->cinfo.d.output_width *
1768
                                       sp->cinfo.d.num_components) /
1769
                                      2;
1770
                    int iPair;
1771
                    for (iPair = 0; iPair < value_pairs; iPair++)
1772
                    {
1773
                        unsigned char *out_ptr =
1774
                            ((unsigned char *)buf) + iPair * 3;
1775
                        TIFF_JSAMPLE *in_ptr =
1776
                            (TIFF_JSAMPLE *)(tmpbuf + iPair * 2);
1777
                        out_ptr[0] = (unsigned char)((in_ptr[0] & 0xff0) >> 4);
1778
                        out_ptr[1] =
1779
                            (unsigned char)(((in_ptr[0] & 0xf) << 4) |
1780
                                            ((in_ptr[1] & 0xf00) >> 8));
1781
                        out_ptr[2] = (unsigned char)(((in_ptr[1] & 0xff) >> 0));
1782
                    }
1783
                }
1784
            }
1785
#endif
1786
1787
0
            sp->scancount++;
1788
0
            tif->tif_row += sp->v_sampling;
1789
1790
0
            buf += sp->bytesperline;
1791
0
            cc -= sp->bytesperline;
1792
1793
0
            nrows -= sp->v_sampling;
1794
0
        } while (nrows > 0);
1795
1796
#if defined(JPEG_LIB_MK1_OR_12BIT)
1797
        _TIFFfreeExt(tif, tmpbuf);
1798
#endif
1799
0
    }
1800
1801
    /* Close down the decompressor if done. */
1802
0
    return sp->cinfo.d.output_scanline < sp->cinfo.d.output_height ||
1803
0
           TIFFjpeg_finish_decompress(sp);
1804
1805
0
error:
1806
#if defined(JPEG_LIB_MK1_OR_12BIT)
1807
    _TIFFfreeExt(tif, tmpbuf);
1808
#endif
1809
0
    return 0;
1810
0
}
1811
1812
/*
1813
 * JPEG Encoding.
1814
 */
1815
1816
static void unsuppress_quant_table(JPEGState *sp, int tblno)
1817
0
{
1818
0
    JQUANT_TBL *qtbl;
1819
1820
0
    if ((qtbl = sp->cinfo.c.quant_tbl_ptrs[tblno]) != NULL)
1821
0
        qtbl->sent_table = FALSE;
1822
0
}
1823
1824
static void suppress_quant_table(JPEGState *sp, int tblno)
1825
0
{
1826
0
    JQUANT_TBL *qtbl;
1827
1828
0
    if ((qtbl = sp->cinfo.c.quant_tbl_ptrs[tblno]) != NULL)
1829
0
        qtbl->sent_table = TRUE;
1830
0
}
1831
1832
static void unsuppress_huff_table(JPEGState *sp, int tblno)
1833
0
{
1834
0
    JHUFF_TBL *htbl;
1835
1836
0
    if ((htbl = sp->cinfo.c.dc_huff_tbl_ptrs[tblno]) != NULL)
1837
0
        htbl->sent_table = FALSE;
1838
0
    if ((htbl = sp->cinfo.c.ac_huff_tbl_ptrs[tblno]) != NULL)
1839
0
        htbl->sent_table = FALSE;
1840
0
}
1841
1842
static void suppress_huff_table(JPEGState *sp, int tblno)
1843
0
{
1844
0
    JHUFF_TBL *htbl;
1845
1846
0
    if ((htbl = sp->cinfo.c.dc_huff_tbl_ptrs[tblno]) != NULL)
1847
0
        htbl->sent_table = TRUE;
1848
0
    if ((htbl = sp->cinfo.c.ac_huff_tbl_ptrs[tblno]) != NULL)
1849
0
        htbl->sent_table = TRUE;
1850
0
}
1851
1852
static int prepare_JPEGTables(TIFF *tif)
1853
0
{
1854
0
    JPEGState *sp = JState(tif);
1855
1856
    /* Initialize quant tables for current quality setting */
1857
0
    if (!TIFFjpeg_set_quality(sp, sp->otherSettings.jpegquality, FALSE))
1858
0
        return (0);
1859
    /* Mark only the tables we want for output */
1860
    /* NB: chrominance tables are currently used only with YCbCr */
1861
0
    if (!TIFFjpeg_suppress_tables(sp, TRUE))
1862
0
        return (0);
1863
0
    if (sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_QUANT)
1864
0
    {
1865
0
        unsuppress_quant_table(sp, 0);
1866
0
        if (sp->photometric == PHOTOMETRIC_YCBCR)
1867
0
            unsuppress_quant_table(sp, 1);
1868
0
    }
1869
0
    if (sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_HUFF)
1870
0
    {
1871
0
        unsuppress_huff_table(sp, 0);
1872
0
        if (sp->photometric == PHOTOMETRIC_YCBCR)
1873
0
            unsuppress_huff_table(sp, 1);
1874
0
    }
1875
    /* Direct libjpeg output into otherSettings.jpegtables */
1876
0
    if (!TIFFjpeg_tables_dest(sp, tif))
1877
0
        return (0);
1878
    /* Emit tables-only datastream */
1879
0
    if (!TIFFjpeg_write_tables(sp))
1880
0
        return (0);
1881
1882
0
    return (1);
1883
0
}
1884
1885
#if defined(JPEG_LIB_VERSION_MAJOR) &&                                         \
1886
    (JPEG_LIB_VERSION_MAJOR > 9 ||                                             \
1887
     (JPEG_LIB_VERSION_MAJOR == 9 && JPEG_LIB_VERSION_MINOR >= 4))
1888
/* This is a modified version of std_huff_tables() from jcparam.c
1889
 * in libjpeg-9d because it no longer initializes default Huffman
1890
 * tables in jpeg_set_defaults(). */
1891
static void TIFF_std_huff_tables(j_compress_ptr cinfo)
1892
{
1893
1894
    if (cinfo->dc_huff_tbl_ptrs[0] == NULL)
1895
    {
1896
        (void)jpeg_std_huff_table((j_common_ptr)cinfo, TRUE, 0);
1897
    }
1898
    if (cinfo->ac_huff_tbl_ptrs[0] == NULL)
1899
    {
1900
        (void)jpeg_std_huff_table((j_common_ptr)cinfo, FALSE, 0);
1901
    }
1902
    if (cinfo->dc_huff_tbl_ptrs[1] == NULL)
1903
    {
1904
        (void)jpeg_std_huff_table((j_common_ptr)cinfo, TRUE, 1);
1905
    }
1906
    if (cinfo->ac_huff_tbl_ptrs[1] == NULL)
1907
    {
1908
        (void)jpeg_std_huff_table((j_common_ptr)cinfo, FALSE, 1);
1909
    }
1910
}
1911
#endif
1912
1913
static int JPEGSetupEncode(TIFF *tif)
1914
0
{
1915
0
    JPEGState *sp = JState(tif);
1916
0
    TIFFDirectory *td = &tif->tif_dir;
1917
0
    static const char module[] = "JPEGSetupEncode";
1918
1919
#if defined(JPEG_DUAL_MODE_8_12) && !defined(FROM_TIF_JPEG_12)
1920
    if (tif->tif_dir.td_bitspersample == 12)
1921
    {
1922
        /* We pass a pointer to a copy of otherSettings, since */
1923
        /* TIFFReInitJPEG_12() will clear sp */
1924
        JPEGOtherSettings savedOtherSettings = sp->otherSettings;
1925
        return TIFFReInitJPEG_12(tif, &savedOtherSettings, COMPRESSION_JPEG, 1);
1926
    }
1927
#endif
1928
1929
0
    JPEGInitializeLibJPEG(tif, FALSE);
1930
1931
0
    assert(sp != NULL);
1932
0
    assert(!sp->cinfo.comm.is_decompressor);
1933
1934
0
    sp->photometric = td->td_photometric;
1935
1936
    /*
1937
     * Initialize all JPEG parameters to default values.
1938
     * Note that jpeg_set_defaults needs legal values for
1939
     * in_color_space and input_components.
1940
     */
1941
0
    if (td->td_planarconfig == PLANARCONFIG_CONTIG)
1942
0
    {
1943
0
        sp->cinfo.c.input_components = td->td_samplesperpixel;
1944
0
        if (sp->photometric == PHOTOMETRIC_YCBCR)
1945
0
        {
1946
0
            if (sp->otherSettings.jpegcolormode == JPEGCOLORMODE_RGB)
1947
0
            {
1948
0
                sp->cinfo.c.in_color_space = JCS_RGB;
1949
0
            }
1950
0
            else
1951
0
            {
1952
0
                sp->cinfo.c.in_color_space = JCS_YCbCr;
1953
0
            }
1954
0
        }
1955
0
        else
1956
0
        {
1957
0
            if ((td->td_photometric == PHOTOMETRIC_MINISWHITE ||
1958
0
                 td->td_photometric == PHOTOMETRIC_MINISBLACK) &&
1959
0
                td->td_samplesperpixel == 1)
1960
0
                sp->cinfo.c.in_color_space = JCS_GRAYSCALE;
1961
0
            else if (td->td_photometric == PHOTOMETRIC_RGB &&
1962
0
                     td->td_samplesperpixel == 3)
1963
0
                sp->cinfo.c.in_color_space = JCS_RGB;
1964
0
            else if (td->td_photometric == PHOTOMETRIC_SEPARATED &&
1965
0
                     td->td_samplesperpixel == 4)
1966
0
                sp->cinfo.c.in_color_space = JCS_CMYK;
1967
0
            else
1968
0
                sp->cinfo.c.in_color_space = JCS_UNKNOWN;
1969
0
        }
1970
0
    }
1971
0
    else
1972
0
    {
1973
0
        sp->cinfo.c.input_components = 1;
1974
0
        sp->cinfo.c.in_color_space = JCS_UNKNOWN;
1975
0
    }
1976
0
    if (!TIFFjpeg_set_defaults(sp))
1977
0
        return (0);
1978
1979
    /* mozjpeg by default enables progressive JPEG, which is illegal in
1980
     * JPEG-in-TIFF */
1981
    /* So explicitly disable it. */
1982
0
    if (sp->cinfo.c.num_scans != 0 &&
1983
0
        (sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_HUFF) != 0)
1984
0
    {
1985
        /* it has been found that mozjpeg could create corrupt strips/tiles */
1986
        /* in non optimize_coding mode. */
1987
0
        TIFFWarningExtR(
1988
0
            tif, module,
1989
0
            "mozjpeg library likely detected. Disable emission of "
1990
0
            "Huffman tables in JpegTables tag, and use optimize_coding "
1991
0
            "to avoid potential issues");
1992
0
        sp->otherSettings.jpegtablesmode &= ~JPEGTABLESMODE_HUFF;
1993
0
    }
1994
0
    sp->cinfo.c.num_scans = 0;
1995
0
    sp->cinfo.c.scan_info = NULL;
1996
1997
    /* Set per-file parameters */
1998
0
    switch (sp->photometric)
1999
0
    {
2000
0
        case PHOTOMETRIC_YCBCR:
2001
0
            sp->h_sampling = td->td_ycbcrsubsampling[0];
2002
0
            sp->v_sampling = td->td_ycbcrsubsampling[1];
2003
0
            if (sp->h_sampling == 0 || sp->v_sampling == 0)
2004
0
            {
2005
0
                TIFFErrorExtR(tif, module,
2006
0
                              "Invalig horizontal/vertical sampling value");
2007
0
                return (0);
2008
0
            }
2009
0
            if (td->td_bitspersample > 16)
2010
0
            {
2011
0
                TIFFErrorExtR(tif, module,
2012
0
                              "BitsPerSample %" PRIu16 " not allowed for JPEG",
2013
0
                              td->td_bitspersample);
2014
0
                return (0);
2015
0
            }
2016
2017
            /*
2018
             * A ReferenceBlackWhite field *must* be present since the
2019
             * default value is inappropriate for YCbCr.  Fill in the
2020
             * proper value if application didn't set it.
2021
             */
2022
0
            {
2023
0
                float *ref;
2024
0
                if (!TIFFGetField(tif, TIFFTAG_REFERENCEBLACKWHITE, &ref))
2025
0
                {
2026
0
                    float refbw[6];
2027
0
                    long top = 1L << td->td_bitspersample;
2028
0
                    refbw[0] = 0;
2029
0
                    refbw[1] = (float)(top - 1L);
2030
0
                    refbw[2] = (float)(top >> 1);
2031
0
                    refbw[3] = refbw[1];
2032
0
                    refbw[4] = refbw[2];
2033
0
                    refbw[5] = refbw[1];
2034
0
                    TIFFSetField(tif, TIFFTAG_REFERENCEBLACKWHITE, refbw);
2035
0
                }
2036
0
            }
2037
0
            break;
2038
0
        case PHOTOMETRIC_PALETTE: /* disallowed by Tech Note */
2039
0
        case PHOTOMETRIC_MASK:
2040
0
            TIFFErrorExtR(tif, module,
2041
0
                          "PhotometricInterpretation %" PRIu16
2042
0
                          " not allowed for JPEG",
2043
0
                          sp->photometric);
2044
0
            return (0);
2045
0
        default:
2046
            /* TIFF 6.0 forbids subsampling of all other color spaces */
2047
0
            sp->h_sampling = 1;
2048
0
            sp->v_sampling = 1;
2049
0
            break;
2050
0
    }
2051
2052
        /* Verify miscellaneous parameters */
2053
2054
        /*
2055
         * This would need work if libtiff ever supports different
2056
         * depths for different components, or if libjpeg ever supports
2057
         * run-time selection of depth.  Neither is imminent.
2058
         */
2059
#ifdef JPEG_LIB_MK1
2060
    /* BITS_IN_JSAMPLE now permits 8 and 12 --- dgilbert */
2061
    if (td->td_bitspersample != 8 && td->td_bitspersample != 12)
2062
#else
2063
0
    if (td->td_bitspersample != BITS_IN_JSAMPLE)
2064
0
#endif
2065
0
    {
2066
0
        TIFFErrorExtR(tif, module,
2067
0
                      "BitsPerSample %" PRIu16 " not allowed for JPEG",
2068
0
                      td->td_bitspersample);
2069
0
        return (0);
2070
0
    }
2071
0
    sp->cinfo.c.data_precision = td->td_bitspersample;
2072
#ifdef JPEG_LIB_MK1
2073
    sp->cinfo.c.bits_in_jsample = td->td_bitspersample;
2074
#endif
2075
0
    if (isTiled(tif))
2076
0
    {
2077
0
        if ((td->td_tilelength % (sp->v_sampling * DCTSIZE)) != 0)
2078
0
        {
2079
0
            TIFFErrorExtR(tif, module,
2080
0
                          "JPEG tile height must be multiple of %" PRIu32,
2081
0
                          (uint32_t)(sp->v_sampling * DCTSIZE));
2082
0
            return (0);
2083
0
        }
2084
0
        if ((td->td_tilewidth % (sp->h_sampling * DCTSIZE)) != 0)
2085
0
        {
2086
0
            TIFFErrorExtR(tif, module,
2087
0
                          "JPEG tile width must be multiple of %" PRIu32,
2088
0
                          (uint32_t)(sp->h_sampling * DCTSIZE));
2089
0
            return (0);
2090
0
        }
2091
0
    }
2092
0
    else
2093
0
    {
2094
0
        if (td->td_rowsperstrip < td->td_imagelength &&
2095
0
            (td->td_rowsperstrip % (sp->v_sampling * DCTSIZE)) != 0)
2096
0
        {
2097
0
            TIFFErrorExtR(tif, module,
2098
0
                          "RowsPerStrip must be multiple of %" PRIu32
2099
0
                          " for JPEG",
2100
0
                          (uint32_t)(sp->v_sampling * DCTSIZE));
2101
0
            return (0);
2102
0
        }
2103
0
    }
2104
2105
    /* Create a JPEGTables field if appropriate */
2106
0
    if (sp->otherSettings.jpegtablesmode &
2107
0
        (JPEGTABLESMODE_QUANT | JPEGTABLESMODE_HUFF))
2108
0
    {
2109
0
        if (sp->otherSettings.jpegtables == NULL ||
2110
0
            memcmp(sp->otherSettings.jpegtables, "\0\0\0\0\0\0\0\0\0", 8) == 0)
2111
0
        {
2112
#if defined(JPEG_LIB_VERSION_MAJOR) &&                                         \
2113
    (JPEG_LIB_VERSION_MAJOR > 9 ||                                             \
2114
     (JPEG_LIB_VERSION_MAJOR == 9 && JPEG_LIB_VERSION_MINOR >= 4))
2115
            if ((sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_HUFF) != 0 &&
2116
                (sp->cinfo.c.dc_huff_tbl_ptrs[0] == NULL ||
2117
                 sp->cinfo.c.dc_huff_tbl_ptrs[1] == NULL ||
2118
                 sp->cinfo.c.ac_huff_tbl_ptrs[0] == NULL ||
2119
                 sp->cinfo.c.ac_huff_tbl_ptrs[1] == NULL))
2120
            {
2121
                /* libjpeg-9d no longer initializes default Huffman tables in */
2122
                /* jpeg_set_defaults() */
2123
                TIFF_std_huff_tables(&sp->cinfo.c);
2124
            }
2125
#endif
2126
2127
0
            if (!prepare_JPEGTables(tif))
2128
0
                return (0);
2129
            /* Mark the field present */
2130
            /* Can't use TIFFSetField since BEENWRITING is already set! */
2131
0
            tif->tif_flags |= TIFF_DIRTYDIRECT;
2132
0
            TIFFSetFieldBit(tif, FIELD_JPEGTABLES);
2133
0
        }
2134
0
    }
2135
0
    else
2136
0
    {
2137
        /* We do not support application-supplied JPEGTables, */
2138
        /* so mark the field not present */
2139
0
        TIFFClrFieldBit(tif, FIELD_JPEGTABLES);
2140
0
    }
2141
2142
    /* Direct libjpeg output to libtiff's output buffer */
2143
0
    TIFFjpeg_data_dest(sp, tif);
2144
2145
0
    return (1);
2146
0
}
2147
2148
/*
2149
 * Set encoding state at the start of a strip or tile.
2150
 */
2151
static int JPEGPreEncode(TIFF *tif, uint16_t s)
2152
0
{
2153
0
    JPEGState *sp = JState(tif);
2154
0
    TIFFDirectory *td = &tif->tif_dir;
2155
0
    static const char module[] = "JPEGPreEncode";
2156
0
    uint32_t segment_width, segment_height;
2157
0
    int downsampled_input;
2158
2159
0
    assert(sp != NULL);
2160
2161
0
    if (sp->cinfo.comm.is_decompressor == 1)
2162
0
    {
2163
0
        tif->tif_setupencode(tif);
2164
0
    }
2165
2166
0
    assert(!sp->cinfo.comm.is_decompressor);
2167
    /*
2168
     * Set encoding parameters for this strip/tile.
2169
     */
2170
0
    if (isTiled(tif))
2171
0
    {
2172
0
        segment_width = td->td_tilewidth;
2173
0
        segment_height = td->td_tilelength;
2174
0
        sp->bytesperline = TIFFTileRowSize(tif);
2175
0
    }
2176
0
    else
2177
0
    {
2178
0
        segment_width = td->td_imagewidth;
2179
0
        segment_height = td->td_imagelength - tif->tif_row;
2180
0
        if (segment_height > td->td_rowsperstrip)
2181
0
            segment_height = td->td_rowsperstrip;
2182
0
        sp->bytesperline = TIFFScanlineSize(tif);
2183
0
    }
2184
0
    if (td->td_planarconfig == PLANARCONFIG_SEPARATE && s > 0)
2185
0
    {
2186
        /* for PC 2, scale down the strip/tile size
2187
         * to match a downsampled component
2188
         */
2189
0
        if (sp->h_sampling == 0 || sp->v_sampling == 0)
2190
0
        {
2191
0
            TIFFErrorExtR(tif, module,
2192
0
                          "JPEG horizontal or vertical sampling is zero");
2193
0
            return (0);
2194
0
        }
2195
0
        segment_width = TIFFhowmany_32(segment_width, sp->h_sampling);
2196
0
        segment_height = TIFFhowmany_32(segment_height, sp->v_sampling);
2197
0
    }
2198
0
    if (segment_width > (uint32_t)JPEG_MAX_DIMENSION ||
2199
0
        segment_height > (uint32_t)JPEG_MAX_DIMENSION)
2200
0
    {
2201
0
        TIFFErrorExtR(tif, module,
2202
0
                      "Strip/tile too large for JPEG. Maximum dimension is %d",
2203
0
                      (int)JPEG_MAX_DIMENSION);
2204
0
        return (0);
2205
0
    }
2206
0
    sp->cinfo.c.image_width = segment_width;
2207
0
    sp->cinfo.c.image_height = segment_height;
2208
0
    downsampled_input = FALSE;
2209
0
    if (td->td_planarconfig == PLANARCONFIG_CONTIG)
2210
0
    {
2211
0
        sp->cinfo.c.input_components = td->td_samplesperpixel;
2212
0
        if (sp->photometric == PHOTOMETRIC_YCBCR)
2213
0
        {
2214
0
            if (sp->otherSettings.jpegcolormode != JPEGCOLORMODE_RGB)
2215
0
            {
2216
0
                if (sp->h_sampling != 1 || sp->v_sampling != 1)
2217
0
                    downsampled_input = TRUE;
2218
0
            }
2219
0
            if (!TIFFjpeg_set_colorspace(sp, JCS_YCbCr))
2220
0
                return (0);
2221
            /*
2222
             * Set Y sampling factors;
2223
             * we assume jpeg_set_colorspace() set the rest to 1
2224
             */
2225
0
            sp->cinfo.c.comp_info[0].h_samp_factor = sp->h_sampling;
2226
0
            sp->cinfo.c.comp_info[0].v_samp_factor = sp->v_sampling;
2227
0
        }
2228
0
        else
2229
0
        {
2230
0
            if (!TIFFjpeg_set_colorspace(sp, sp->cinfo.c.in_color_space))
2231
0
                return (0);
2232
            /* jpeg_set_colorspace set all sampling factors to 1 */
2233
0
        }
2234
0
    }
2235
0
    else
2236
0
    {
2237
0
        if (!TIFFjpeg_set_colorspace(sp, JCS_UNKNOWN))
2238
0
            return (0);
2239
0
        sp->cinfo.c.comp_info[0].component_id = s;
2240
        /* jpeg_set_colorspace() set sampling factors to 1 */
2241
0
        if (sp->photometric == PHOTOMETRIC_YCBCR && s > 0)
2242
0
        {
2243
0
            sp->cinfo.c.comp_info[0].quant_tbl_no = 1;
2244
0
            sp->cinfo.c.comp_info[0].dc_tbl_no = 1;
2245
0
            sp->cinfo.c.comp_info[0].ac_tbl_no = 1;
2246
0
        }
2247
0
    }
2248
    /* ensure libjpeg won't write any extraneous markers */
2249
0
    sp->cinfo.c.write_JFIF_header = FALSE;
2250
0
    sp->cinfo.c.write_Adobe_marker = FALSE;
2251
    /* set up table handling correctly */
2252
    /* calling TIFFjpeg_set_quality() causes quantization tables to be flagged
2253
     */
2254
    /* as being to be emitted, which we don't want in the JPEGTABLESMODE_QUANT
2255
     */
2256
    /* mode, so we must manually suppress them. However TIFFjpeg_set_quality()
2257
     */
2258
    /* should really be called when dealing with files with directories with */
2259
    /* mixed qualities. see http://trac.osgeo.org/gdal/ticket/3539 */
2260
0
    if (!TIFFjpeg_set_quality(sp, sp->otherSettings.jpegquality, FALSE))
2261
0
        return (0);
2262
0
    if (sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_QUANT)
2263
0
    {
2264
0
        suppress_quant_table(sp, 0);
2265
0
        suppress_quant_table(sp, 1);
2266
0
    }
2267
0
    else
2268
0
    {
2269
0
        unsuppress_quant_table(sp, 0);
2270
0
        unsuppress_quant_table(sp, 1);
2271
0
    }
2272
0
    if (sp->otherSettings.jpegtablesmode & JPEGTABLESMODE_HUFF)
2273
0
    {
2274
        /* Explicit suppression is only needed if we did not go through the */
2275
        /* prepare_JPEGTables() code path, which may be the case if updating */
2276
        /* an existing file */
2277
0
        suppress_huff_table(sp, 0);
2278
0
        suppress_huff_table(sp, 1);
2279
0
        sp->cinfo.c.optimize_coding = FALSE;
2280
0
    }
2281
0
    else
2282
0
        sp->cinfo.c.optimize_coding = TRUE;
2283
0
    if (downsampled_input)
2284
0
    {
2285
        /* Need to use raw-data interface to libjpeg */
2286
0
        sp->cinfo.c.raw_data_in = TRUE;
2287
0
        tif->tif_encoderow = JPEGEncodeRaw;
2288
0
        tif->tif_encodestrip = JPEGEncodeRaw;
2289
0
        tif->tif_encodetile = JPEGEncodeRaw;
2290
0
    }
2291
0
    else
2292
0
    {
2293
        /* Use normal interface to libjpeg */
2294
0
        sp->cinfo.c.raw_data_in = FALSE;
2295
0
        tif->tif_encoderow = JPEGEncode;
2296
0
        tif->tif_encodestrip = JPEGEncode;
2297
0
        tif->tif_encodetile = JPEGEncode;
2298
0
    }
2299
    /* Start JPEG compressor */
2300
0
    if (!TIFFjpeg_start_compress(sp, FALSE))
2301
0
        return (0);
2302
    /* Allocate downsampled-data buffers if needed */
2303
0
    if (downsampled_input)
2304
0
    {
2305
0
        if (!alloc_downsampled_buffers(tif, sp->cinfo.c.comp_info,
2306
0
                                       sp->cinfo.c.num_components))
2307
0
            return (0);
2308
0
    }
2309
0
    sp->scancount = 0;
2310
0
    sp->encode_raw_error = FALSE;
2311
2312
0
    return (1);
2313
0
}
2314
2315
/*
2316
 * Encode a chunk of pixels.
2317
 * "Standard" case: incoming data is not downsampled.
2318
 */
2319
static int JPEGEncode(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s)
2320
0
{
2321
0
    JPEGState *sp = JState(tif);
2322
0
    tmsize_t nrows;
2323
0
    TIFF_JSAMPROW bufptr[1];
2324
0
    short *line16 = NULL;
2325
0
    int line16_count = 0;
2326
2327
0
    (void)s;
2328
0
    assert(sp != NULL);
2329
    /* data is expected to be supplied in multiples of a scanline */
2330
0
    nrows = cc / sp->bytesperline;
2331
0
    if (cc % sp->bytesperline)
2332
0
        TIFFWarningExtR(tif, tif->tif_name, "fractional scanline discarded");
2333
2334
    /* The last strip will be limited to image size */
2335
0
    if (!isTiled(tif) && tif->tif_row + nrows > tif->tif_dir.td_imagelength)
2336
0
        nrows = tif->tif_dir.td_imagelength - tif->tif_row;
2337
2338
0
    if (sp->cinfo.c.data_precision == 12)
2339
0
    {
2340
0
        line16_count = (int)((sp->bytesperline * 2) / 3);
2341
0
        line16 = (short *)_TIFFmallocExt(tif, sizeof(short) * line16_count);
2342
0
        if (!line16)
2343
0
        {
2344
0
            TIFFErrorExtR(tif, "JPEGEncode", "Failed to allocate memory");
2345
2346
0
            return 0;
2347
0
        }
2348
0
    }
2349
2350
0
    while (nrows-- > 0)
2351
0
    {
2352
2353
0
        if (sp->cinfo.c.data_precision == 12)
2354
0
        {
2355
2356
0
            int value_pairs = line16_count / 2;
2357
0
            int iPair;
2358
2359
0
            bufptr[0] = (TIFF_JSAMPROW)line16;
2360
2361
0
            for (iPair = 0; iPair < value_pairs; iPair++)
2362
0
            {
2363
0
                unsigned char *in_ptr = ((unsigned char *)buf) + iPair * 3;
2364
0
                TIFF_JSAMPLE *out_ptr = (TIFF_JSAMPLE *)(line16 + iPair * 2);
2365
2366
0
                out_ptr[0] = (in_ptr[0] << 4) | ((in_ptr[1] & 0xf0) >> 4);
2367
0
                out_ptr[1] = ((in_ptr[1] & 0x0f) << 8) | in_ptr[2];
2368
0
            }
2369
0
        }
2370
0
        else
2371
0
        {
2372
0
            bufptr[0] = (TIFF_JSAMPROW)buf;
2373
0
        }
2374
0
        if (TIFFjpeg_write_scanlines(sp, bufptr, 1) != 1)
2375
0
            return (0);
2376
0
        if (nrows > 0)
2377
0
            tif->tif_row++;
2378
0
        buf += sp->bytesperline;
2379
0
    }
2380
2381
0
    if (sp->cinfo.c.data_precision == 12)
2382
0
    {
2383
0
        _TIFFfreeExt(tif, line16);
2384
0
    }
2385
2386
0
    return (1);
2387
0
}
2388
2389
/*
2390
 * Encode a chunk of pixels.
2391
 * Incoming data is expected to be downsampled per sampling factors.
2392
 */
2393
static int JPEGEncodeRaw(TIFF *tif, uint8_t *buf, tmsize_t cc, uint16_t s)
2394
0
{
2395
0
    JPEGState *sp = JState(tif);
2396
0
    TIFF_JSAMPLE *inptr;
2397
0
    TIFF_JSAMPLE *outptr;
2398
0
    tmsize_t nrows;
2399
0
    JDIMENSION clumps_per_line, nclump;
2400
0
    int clumpoffset, ci, xpos, ypos;
2401
0
    jpeg_component_info *compptr;
2402
0
    int samples_per_clump = sp->samplesperclump;
2403
0
    tmsize_t bytesperclumpline;
2404
2405
0
    (void)s;
2406
0
    assert(sp != NULL);
2407
2408
0
    if (sp->encode_raw_error)
2409
0
    {
2410
0
        TIFFErrorExtR(tif, tif->tif_name, "JPEGEncodeRaw() already failed");
2411
0
        return 0;
2412
0
    }
2413
2414
    /* data is expected to be supplied in multiples of a clumpline */
2415
    /* a clumpline is equivalent to v_sampling desubsampled scanlines */
2416
    /* TODO: the following calculation of bytesperclumpline, should substitute
2417
     * calculation of sp->bytesperline, except that it is per v_sampling lines
2418
     */
2419
0
    bytesperclumpline =
2420
0
        ((((tmsize_t)sp->cinfo.c.image_width + sp->h_sampling - 1) /
2421
0
          sp->h_sampling) *
2422
0
             ((tmsize_t)sp->h_sampling * sp->v_sampling + 2) *
2423
0
             sp->cinfo.c.data_precision +
2424
0
         7) /
2425
0
        8;
2426
2427
0
    nrows = (cc / bytesperclumpline) * sp->v_sampling;
2428
0
    if (cc % bytesperclumpline)
2429
0
        TIFFWarningExtR(tif, tif->tif_name, "fractional scanline discarded");
2430
2431
    /* Cb,Cr both have sampling factors 1, so this is correct */
2432
0
    clumps_per_line = sp->cinfo.c.comp_info[1].downsampled_width;
2433
2434
0
    while (nrows > 0)
2435
0
    {
2436
        /*
2437
         * Fastest way to separate the data is to make one pass
2438
         * over the scanline for each row of each component.
2439
         */
2440
0
        clumpoffset = 0; /* first sample in clump */
2441
0
        for (ci = 0, compptr = sp->cinfo.c.comp_info;
2442
0
             ci < sp->cinfo.c.num_components; ci++, compptr++)
2443
0
        {
2444
0
            int hsamp = compptr->h_samp_factor;
2445
0
            int vsamp = compptr->v_samp_factor;
2446
0
            int padding = (int)(compptr->width_in_blocks * DCTSIZE -
2447
0
                                clumps_per_line * hsamp);
2448
0
            for (ypos = 0; ypos < vsamp; ypos++)
2449
0
            {
2450
0
                inptr = ((TIFF_JSAMPLE *)buf) + clumpoffset;
2451
0
                outptr = sp->ds_buffer[ci][sp->scancount * vsamp + ypos];
2452
0
                if (hsamp == 1)
2453
0
                {
2454
                    /* fast path for at least Cb and Cr */
2455
0
                    for (nclump = clumps_per_line; nclump-- > 0;)
2456
0
                    {
2457
0
                        *outptr++ = inptr[0];
2458
0
                        inptr += samples_per_clump;
2459
0
                    }
2460
0
                }
2461
0
                else
2462
0
                {
2463
                    /* general case */
2464
0
                    for (nclump = clumps_per_line; nclump-- > 0;)
2465
0
                    {
2466
0
                        for (xpos = 0; xpos < hsamp; xpos++)
2467
0
                            *outptr++ = inptr[xpos];
2468
0
                        inptr += samples_per_clump;
2469
0
                    }
2470
0
                }
2471
                /* pad each scanline as needed */
2472
0
                for (xpos = 0; xpos < padding; xpos++)
2473
0
                {
2474
0
                    *outptr = outptr[-1];
2475
0
                    outptr++;
2476
0
                }
2477
0
                clumpoffset += hsamp;
2478
0
            }
2479
0
        }
2480
0
        sp->scancount++;
2481
0
        if (sp->scancount >= DCTSIZE)
2482
0
        {
2483
0
            int n = sp->cinfo.c.max_v_samp_factor * DCTSIZE;
2484
0
            if (TIFFjpeg_write_raw_data(sp, sp->ds_buffer, n) != n)
2485
0
            {
2486
0
                sp->encode_raw_error = TRUE;
2487
0
                return (0);
2488
0
            }
2489
0
            sp->scancount = 0;
2490
0
        }
2491
0
        tif->tif_row += sp->v_sampling;
2492
0
        buf += bytesperclumpline;
2493
0
        nrows -= sp->v_sampling;
2494
0
    }
2495
0
    return (1);
2496
0
}
2497
2498
/*
2499
 * Finish up at the end of a strip or tile.
2500
 */
2501
static int JPEGPostEncode(TIFF *tif)
2502
0
{
2503
0
    JPEGState *sp = JState(tif);
2504
2505
0
    if (sp->scancount > 0)
2506
0
    {
2507
        /*
2508
         * Need to emit a partial bufferload of downsampled data.
2509
         * Pad the data vertically.
2510
         */
2511
0
        int ci, ypos, n;
2512
0
        jpeg_component_info *compptr;
2513
2514
0
        for (ci = 0, compptr = sp->cinfo.c.comp_info;
2515
0
             ci < sp->cinfo.c.num_components; ci++, compptr++)
2516
0
        {
2517
0
            int vsamp = compptr->v_samp_factor;
2518
0
            tmsize_t row_width =
2519
0
                compptr->width_in_blocks * DCTSIZE * sizeof(JSAMPLE);
2520
0
            for (ypos = sp->scancount * vsamp; ypos < DCTSIZE * vsamp; ypos++)
2521
0
            {
2522
0
                _TIFFmemcpy((void *)sp->ds_buffer[ci][ypos],
2523
0
                            (void *)sp->ds_buffer[ci][ypos - 1], row_width);
2524
0
            }
2525
0
        }
2526
0
        n = sp->cinfo.c.max_v_samp_factor * DCTSIZE;
2527
0
        if (TIFFjpeg_write_raw_data(sp, sp->ds_buffer, n) != n)
2528
0
            return (0);
2529
0
    }
2530
2531
0
    return (TIFFjpeg_finish_compress(JState(tif)));
2532
0
}
2533
2534
static void JPEGCleanup(TIFF *tif)
2535
977
{
2536
977
    JPEGState *sp = JState(tif);
2537
2538
977
    assert(sp != 0);
2539
2540
977
    tif->tif_tagmethods.vgetfield = sp->otherSettings.vgetparent;
2541
977
    tif->tif_tagmethods.vsetfield = sp->otherSettings.vsetparent;
2542
977
    tif->tif_tagmethods.printdir = sp->otherSettings.printdir;
2543
977
    if (sp->cinfo_initialized)
2544
167
        TIFFjpeg_destroy(sp);         /* release libjpeg resources */
2545
977
    if (sp->otherSettings.jpegtables) /* tag value */
2546
22
        _TIFFfreeExt(tif, sp->otherSettings.jpegtables);
2547
977
    _TIFFfreeExt(tif, tif->tif_data); /* release local state */
2548
977
    tif->tif_data = NULL;
2549
2550
977
    _TIFFSetDefaultCompressionState(tif);
2551
977
}
2552
2553
static void JPEGResetUpsampled(TIFF *tif)
2554
1.63k
{
2555
1.63k
    JPEGState *sp = JState(tif);
2556
1.63k
    TIFFDirectory *td = &tif->tif_dir;
2557
2558
    /*
2559
     * Mark whether returned data is up-sampled or not so TIFFStripSize
2560
     * and TIFFTileSize return values that reflect the true amount of
2561
     * data.
2562
     */
2563
1.63k
    tif->tif_flags &= ~TIFF_UPSAMPLED;
2564
1.63k
    if (td->td_planarconfig == PLANARCONFIG_CONTIG)
2565
1.58k
    {
2566
1.58k
        if (td->td_photometric == PHOTOMETRIC_YCBCR &&
2567
1.46k
            sp->otherSettings.jpegcolormode == JPEGCOLORMODE_RGB)
2568
663
        {
2569
663
            tif->tif_flags |= TIFF_UPSAMPLED;
2570
663
        }
2571
923
        else
2572
923
        {
2573
#ifdef notdef
2574
            if (td->td_ycbcrsubsampling[0] != 1 ||
2575
                td->td_ycbcrsubsampling[1] != 1)
2576
                ; /* XXX what about up-sampling? */
2577
#endif
2578
923
        }
2579
1.58k
    }
2580
2581
    /*
2582
     * Must recalculate cached tile size in case sampling state changed.
2583
     * Should we really be doing this now if image size isn't set?
2584
     */
2585
1.63k
    if (tif->tif_tilesize > 0)
2586
495
        tif->tif_tilesize = isTiled(tif) ? TIFFTileSize(tif) : (tmsize_t)(-1);
2587
1.63k
    if (tif->tif_scanlinesize > 0)
2588
663
        tif->tif_scanlinesize = TIFFScanlineSize(tif);
2589
1.63k
}
2590
2591
static int JPEGVSetField(TIFF *tif, uint32_t tag, va_list ap)
2592
7.22k
{
2593
7.22k
    JPEGState *sp = JState(tif);
2594
7.22k
    const TIFFField *fip;
2595
7.22k
    uint32_t v32;
2596
2597
7.22k
    assert(sp != NULL);
2598
2599
7.22k
    switch (tag)
2600
7.22k
    {
2601
22
        case TIFFTAG_JPEGTABLES:
2602
22
            v32 = (uint32_t)va_arg(ap, uint32_t);
2603
22
            if (v32 == 0)
2604
0
            {
2605
                /* XXX */
2606
0
                return (0);
2607
0
            }
2608
22
            _TIFFsetByteArrayExt(tif, &sp->otherSettings.jpegtables,
2609
22
                                 va_arg(ap, void *), v32);
2610
22
            sp->otherSettings.jpegtables_length = v32;
2611
22
            TIFFSetFieldBit(tif, FIELD_JPEGTABLES);
2612
22
            break;
2613
0
        case TIFFTAG_JPEGQUALITY:
2614
0
            sp->otherSettings.jpegquality = (int)va_arg(ap, int);
2615
0
            return (1); /* pseudo tag */
2616
663
        case TIFFTAG_JPEGCOLORMODE:
2617
663
            sp->otherSettings.jpegcolormode = (int)va_arg(ap, int);
2618
663
            JPEGResetUpsampled(tif);
2619
663
            return (1); /* pseudo tag */
2620
973
        case TIFFTAG_PHOTOMETRIC:
2621
973
        {
2622
973
            int ret_value = (*sp->otherSettings.vsetparent)(tif, tag, ap);
2623
973
            JPEGResetUpsampled(tif);
2624
973
            return ret_value;
2625
22
        }
2626
0
        case TIFFTAG_JPEGTABLESMODE:
2627
0
            sp->otherSettings.jpegtablesmode = (int)va_arg(ap, int);
2628
0
            return (1); /* pseudo tag */
2629
12
        case TIFFTAG_YCBCRSUBSAMPLING:
2630
            /* mark the fact that we have a real ycbcrsubsampling! */
2631
12
            sp->otherSettings.ycbcrsampling_fetched = 1;
2632
            /* should we be recomputing upsampling info here? */
2633
12
            return (*sp->otherSettings.vsetparent)(tif, tag, ap);
2634
5.55k
        default:
2635
5.55k
            return (*sp->otherSettings.vsetparent)(tif, tag, ap);
2636
7.22k
    }
2637
2638
22
    if ((fip = TIFFFieldWithTag(tif, tag)) != NULL)
2639
22
    {
2640
22
        TIFFSetFieldBit(tif, fip->field_bit);
2641
22
    }
2642
0
    else
2643
0
    {
2644
0
        return (0);
2645
0
    }
2646
2647
22
    tif->tif_flags |= TIFF_DIRTYDIRECT;
2648
22
    return (1);
2649
22
}
2650
2651
static int JPEGVGetField(TIFF *tif, uint32_t tag, va_list ap)
2652
46.0k
{
2653
46.0k
    JPEGState *sp = JState(tif);
2654
2655
46.0k
    assert(sp != NULL);
2656
2657
46.0k
    switch (tag)
2658
46.0k
    {
2659
0
        case TIFFTAG_JPEGTABLES:
2660
0
            *va_arg(ap, uint32_t *) = sp->otherSettings.jpegtables_length;
2661
0
            *va_arg(ap, const void **) = sp->otherSettings.jpegtables;
2662
0
            break;
2663
0
        case TIFFTAG_JPEGQUALITY:
2664
0
            *va_arg(ap, int *) = sp->otherSettings.jpegquality;
2665
0
            break;
2666
0
        case TIFFTAG_JPEGCOLORMODE:
2667
0
            *va_arg(ap, int *) = sp->otherSettings.jpegcolormode;
2668
0
            break;
2669
0
        case TIFFTAG_JPEGTABLESMODE:
2670
0
            *va_arg(ap, int *) = sp->otherSettings.jpegtablesmode;
2671
0
            break;
2672
46.0k
        default:
2673
46.0k
            return (*sp->otherSettings.vgetparent)(tif, tag, ap);
2674
46.0k
    }
2675
0
    return (1);
2676
46.0k
}
2677
2678
static void JPEGPrintDir(TIFF *tif, FILE *fd, long flags)
2679
0
{
2680
0
    JPEGState *sp = JState(tif);
2681
2682
0
    assert(sp != NULL);
2683
0
    (void)flags;
2684
2685
0
    if (sp != NULL)
2686
0
    {
2687
0
        if (TIFFFieldSet(tif, FIELD_JPEGTABLES))
2688
0
            fprintf(fd, "  JPEG Tables: (%" PRIu32 " bytes)\n",
2689
0
                    sp->otherSettings.jpegtables_length);
2690
0
        if (sp->otherSettings.printdir)
2691
0
            (*sp->otherSettings.printdir)(tif, fd, flags);
2692
0
    }
2693
0
}
2694
2695
static uint32_t JPEGDefaultStripSize(TIFF *tif, uint32_t s)
2696
0
{
2697
0
    JPEGState *sp = JState(tif);
2698
0
    TIFFDirectory *td = &tif->tif_dir;
2699
2700
0
    s = (*sp->otherSettings.defsparent)(tif, s);
2701
0
    if (s < td->td_imagelength)
2702
0
        s = TIFFroundup_32(s, td->td_ycbcrsubsampling[1] * DCTSIZE);
2703
0
    return (s);
2704
0
}
2705
2706
static void JPEGDefaultTileSize(TIFF *tif, uint32_t *tw, uint32_t *th)
2707
0
{
2708
0
    JPEGState *sp = JState(tif);
2709
0
    TIFFDirectory *td = &tif->tif_dir;
2710
2711
0
    (*sp->otherSettings.deftparent)(tif, tw, th);
2712
0
    *tw = TIFFroundup_32(*tw, td->td_ycbcrsubsampling[0] * DCTSIZE);
2713
0
    *th = TIFFroundup_32(*th, td->td_ycbcrsubsampling[1] * DCTSIZE);
2714
0
}
2715
2716
/*
2717
 * The JPEG library initialized used to be done in TIFFInitJPEG(), but
2718
 * now that we allow a TIFF file to be opened in update mode it is necessary
2719
 * to have some way of deciding whether compression or decompression is
2720
 * desired other than looking at tif->tif_mode.  We accomplish this by
2721
 * examining {TILE/STRIP}BYTECOUNTS to see if there is a non-zero entry.
2722
 * If so, we assume decompression is desired.
2723
 *
2724
 * This is tricky, because TIFFInitJPEG() is called while the directory is
2725
 * being read, and generally speaking the BYTECOUNTS tag won't have been read
2726
 * at that point.  So we try to defer jpeg library initialization till we
2727
 * do have that tag ... basically any access that might require the compressor
2728
 * or decompressor that occurs after the reading of the directory.
2729
 *
2730
 * In an ideal world compressors or decompressors would be setup
2731
 * at the point where a single tile or strip was accessed (for read or write)
2732
 * so that stuff like update of missing tiles, or replacement of tiles could
2733
 * be done. However, we aren't trying to crack that nut just yet ...
2734
 *
2735
 * NFW, Feb 3rd, 2003.
2736
 */
2737
2738
static int JPEGInitializeLibJPEG(TIFF *tif, int decompress)
2739
167
{
2740
167
    JPEGState *sp = JState(tif);
2741
2742
167
    if (sp->cinfo_initialized)
2743
0
    {
2744
0
        if (!decompress && sp->cinfo.comm.is_decompressor)
2745
0
            TIFFjpeg_destroy(sp);
2746
0
        else if (decompress && !sp->cinfo.comm.is_decompressor)
2747
0
            TIFFjpeg_destroy(sp);
2748
0
        else
2749
0
            return 1;
2750
2751
0
        sp->cinfo_initialized = 0;
2752
0
    }
2753
2754
    /*
2755
     * Initialize libjpeg.
2756
     */
2757
167
    if (decompress)
2758
167
    {
2759
167
        if (!TIFFjpeg_create_decompress(sp))
2760
0
            return (0);
2761
167
    }
2762
0
    else
2763
0
    {
2764
0
        if (!TIFFjpeg_create_compress(sp))
2765
0
            return (0);
2766
0
#ifndef TIFF_JPEG_MAX_MEMORY_TO_USE
2767
0
#define TIFF_JPEG_MAX_MEMORY_TO_USE (10 * 1024 * 1024)
2768
0
#endif
2769
        /* libjpeg turbo 1.5.2 honours max_memory_to_use, but has no backing */
2770
        /* store implementation, so better not set max_memory_to_use ourselves.
2771
         */
2772
        /* See https://github.com/libjpeg-turbo/libjpeg-turbo/issues/162 */
2773
0
        if (sp->cinfo.c.mem->max_memory_to_use > 0)
2774
0
        {
2775
            /* This is to address bug related in ticket GDAL #1795. */
2776
0
            if (getenv("JPEGMEM") == NULL)
2777
0
            {
2778
                /* Increase the max memory usable. This helps when creating
2779
                 * files */
2780
                /* with "big" tile, without using libjpeg temporary files. */
2781
                /* For example a 512x512 tile with 3 bands */
2782
                /* requires 1.5 MB which is above libjpeg 1MB default */
2783
0
                if (sp->cinfo.c.mem->max_memory_to_use <
2784
0
                    TIFF_JPEG_MAX_MEMORY_TO_USE)
2785
0
                    sp->cinfo.c.mem->max_memory_to_use =
2786
0
                        TIFF_JPEG_MAX_MEMORY_TO_USE;
2787
0
            }
2788
0
        }
2789
0
    }
2790
2791
167
    sp->cinfo_initialized = TRUE;
2792
2793
167
    return 1;
2794
167
}
2795
2796
/* Common to tif_jpeg.c and tif_jpeg_12.c */
2797
static void TIFFInitJPEGCommon(TIFF *tif)
2798
977
{
2799
977
    JPEGState *sp;
2800
2801
977
    sp = JState(tif);
2802
977
    sp->tif = tif; /* back link */
2803
2804
    /* Default values for codec-specific fields */
2805
977
    sp->otherSettings.jpegtables = NULL;
2806
977
    sp->otherSettings.jpegtables_length = 0;
2807
977
    sp->otherSettings.jpegquality = 75; /* Default IJG quality */
2808
977
    sp->otherSettings.jpegcolormode = JPEGCOLORMODE_RAW;
2809
977
    sp->otherSettings.jpegtablesmode =
2810
977
        JPEGTABLESMODE_QUANT | JPEGTABLESMODE_HUFF;
2811
977
    sp->otherSettings.ycbcrsampling_fetched = 0;
2812
2813
977
    tif->tif_tagmethods.vgetfield = JPEGVGetField; /* hook for codec tags */
2814
977
    tif->tif_tagmethods.vsetfield = JPEGVSetField; /* hook for codec tags */
2815
977
    tif->tif_tagmethods.printdir = JPEGPrintDir;   /* hook for codec tags */
2816
2817
    /*
2818
     * Install codec methods.
2819
     */
2820
977
    tif->tif_fixuptags = JPEGFixupTags;
2821
977
    tif->tif_setupdecode = JPEGSetupDecode;
2822
977
    tif->tif_predecode = JPEGPreDecode;
2823
977
    tif->tif_decoderow = JPEGDecode;
2824
977
    tif->tif_decodestrip = JPEGDecode;
2825
977
    tif->tif_decodetile = JPEGDecode;
2826
977
    tif->tif_setupencode = JPEGSetupEncode;
2827
977
    tif->tif_preencode = JPEGPreEncode;
2828
977
    tif->tif_postencode = JPEGPostEncode;
2829
977
    tif->tif_encoderow = JPEGEncode;
2830
977
    tif->tif_encodestrip = JPEGEncode;
2831
977
    tif->tif_encodetile = JPEGEncode;
2832
977
    tif->tif_cleanup = JPEGCleanup;
2833
2834
977
    tif->tif_defstripsize = JPEGDefaultStripSize;
2835
977
    tif->tif_deftilesize = JPEGDefaultTileSize;
2836
977
    tif->tif_flags |= TIFF_NOBITREV; /* no bit reversal, please */
2837
977
    sp->cinfo_initialized = FALSE;
2838
977
}
2839
2840
int TIFFInitJPEG(TIFF *tif, int scheme)
2841
977
{
2842
977
    JPEGState *sp;
2843
2844
977
    (void)scheme;
2845
977
    assert(scheme == COMPRESSION_JPEG);
2846
2847
    /*
2848
     * Merge codec-specific tag information.
2849
     */
2850
977
    if (!_TIFFMergeFields(tif, jpegFields, TIFFArrayCount(jpegFields)))
2851
0
    {
2852
0
        TIFFErrorExtR(tif, "TIFFInitJPEG",
2853
0
                      "Merging JPEG codec-specific tags failed");
2854
0
        return 0;
2855
0
    }
2856
2857
    /*
2858
     * Allocate state block so tag methods have storage to record values.
2859
     */
2860
977
    tif->tif_data = (uint8_t *)_TIFFmallocExt(tif, sizeof(JPEGState));
2861
2862
977
    if (tif->tif_data == NULL)
2863
0
    {
2864
0
        TIFFErrorExtR(tif, "TIFFInitJPEG", "No space for JPEG state block");
2865
0
        return 0;
2866
0
    }
2867
977
    _TIFFmemset(tif->tif_data, 0, sizeof(JPEGState));
2868
2869
977
    sp = JState(tif);
2870
    /*
2871
     * Override parent get/set field methods.
2872
     */
2873
977
    sp->otherSettings.vgetparent = tif->tif_tagmethods.vgetfield;
2874
977
    sp->otherSettings.vsetparent = tif->tif_tagmethods.vsetfield;
2875
977
    sp->otherSettings.printdir = tif->tif_tagmethods.printdir;
2876
2877
977
    sp->otherSettings.defsparent = tif->tif_defstripsize;
2878
977
    sp->otherSettings.deftparent = tif->tif_deftilesize;
2879
2880
977
    TIFFInitJPEGCommon(tif);
2881
2882
    /*
2883
    ** Create a JPEGTables field if no directory has yet been created.
2884
    ** We do this just to ensure that sufficient space is reserved for
2885
    ** the JPEGTables field.  It will be properly created the right
2886
    ** size later.
2887
    */
2888
977
    if (tif->tif_diroff == 0)
2889
0
    {
2890
0
#define SIZE_OF_JPEGTABLES 2000
2891
        /*
2892
        The following line assumes incorrectly that all JPEG-in-TIFF files will
2893
        have a JPEGTABLES tag generated and causes null-filled JPEGTABLES tags
2894
        to be written when the JPEG data is placed with TIFFWriteRawStrip.  The
2895
        field bit should be set, anyway, later when actual JPEGTABLES header is
2896
        generated, so removing it here hopefully is harmless.
2897
        TIFFSetFieldBit(tif, FIELD_JPEGTABLES);
2898
        */
2899
0
        sp->otherSettings.jpegtables_length = SIZE_OF_JPEGTABLES;
2900
0
        sp->otherSettings.jpegtables =
2901
0
            (void *)_TIFFmallocExt(tif, sp->otherSettings.jpegtables_length);
2902
0
        if (sp->otherSettings.jpegtables)
2903
0
        {
2904
0
            _TIFFmemset(sp->otherSettings.jpegtables, 0, SIZE_OF_JPEGTABLES);
2905
0
        }
2906
0
        else
2907
0
        {
2908
0
            TIFFErrorExtR(tif, "TIFFInitJPEG",
2909
0
                          "Failed to allocate memory for JPEG tables");
2910
0
            return 0;
2911
0
        }
2912
0
#undef SIZE_OF_JPEGTABLES
2913
0
    }
2914
977
    return 1;
2915
977
}
2916
#endif /* JPEG_SUPPORT */