Coverage Report

Created: 2023-06-07 06:03

/src/libjpeg-turbo.2.1.x/rdppm.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
 * rdppm.c
3
 *
4
 * This file was part of the Independent JPEG Group's software:
5
 * Copyright (C) 1991-1997, Thomas G. Lane.
6
 * Modified 2009 by Bill Allombert, Guido Vollbeding.
7
 * libjpeg-turbo Modifications:
8
 * Copyright (C) 2015-2017, 2020-2023, D. R. Commander.
9
 * For conditions of distribution and use, see the accompanying README.ijg
10
 * file.
11
 *
12
 * This file contains routines to read input images in PPM/PGM format.
13
 * The extended 2-byte-per-sample raw PPM/PGM formats are supported.
14
 * The PBMPLUS library is NOT required to compile this software
15
 * (but it is highly useful as a set of PPM image manipulation programs).
16
 *
17
 * These routines may need modification for non-Unix environments or
18
 * specialized applications.  As they stand, they assume input from
19
 * an ordinary stdio stream.  They further assume that reading begins
20
 * at the start of the file; start_input may need work if the
21
 * user interface has already read some data (e.g., to determine that
22
 * the file is indeed PPM format).
23
 */
24
25
#include "cmyk.h"
26
#include "cdjpeg.h"             /* Common decls for cjpeg/djpeg applications */
27
28
#ifdef PPM_SUPPORTED
29
30
31
/* Portions of this code are based on the PBMPLUS library, which is:
32
**
33
** Copyright (C) 1988 by Jef Poskanzer.
34
**
35
** Permission to use, copy, modify, and distribute this software and its
36
** documentation for any purpose and without fee is hereby granted, provided
37
** that the above copyright notice appear in all copies and that both that
38
** copyright notice and this permission notice appear in supporting
39
** documentation.  This software is provided "as is" without express or
40
** implied warranty.
41
*/
42
43
44
/* Macros to deal with unsigned chars as efficiently as compiler allows */
45
46
typedef unsigned char U_CHAR;
47
38.2k
#define UCH(x)  ((int)(x))
48
49
50
#define ReadOK(file, buffer, len) \
51
3.43k
  (fread(buffer, 1, len, file) == ((size_t)(len)))
52
53
static int alpha_index[JPEG_NUMCS] = {
54
  -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0, -1
55
};
56
57
58
/* Private version of data source object */
59
60
typedef struct {
61
  struct cjpeg_source_struct pub; /* public fields */
62
63
  /* Usually these two pointers point to the same place: */
64
  U_CHAR *iobuffer;             /* fread's I/O buffer */
65
  JSAMPROW pixrow;              /* compressor input buffer */
66
  size_t buffer_width;          /* width of I/O buffer */
67
  JSAMPLE *rescale;             /* => maxval-remapping array, or NULL */
68
  unsigned int maxval;
69
} ppm_source_struct;
70
71
typedef ppm_source_struct *ppm_source_ptr;
72
73
74
LOCAL(int)
75
pbm_getc(FILE *infile)
76
/* Read next char, skipping over any comments */
77
/* A comment/newline sequence is returned as a newline */
78
29.3k
{
79
29.3k
  register int ch;
80
81
29.3k
  ch = getc(infile);
82
29.3k
  if (ch == '#') {
83
1.62k
    do {
84
1.62k
      ch = getc(infile);
85
1.62k
    } while (ch != '\n' && ch != EOF);
86
664
  }
87
29.3k
  return ch;
88
29.3k
}
89
90
91
LOCAL(unsigned int)
92
read_pbm_integer(j_compress_ptr cinfo, FILE *infile, unsigned int maxval)
93
/* Read an unsigned decimal integer from the PPM file */
94
/* Swallows one trailing character after the integer */
95
/* Note that on a 16-bit-int machine, only values up to 64k can be read. */
96
/* This should not be a problem in practice. */
97
11.0k
{
98
11.0k
  register int ch;
99
11.0k
  register unsigned int val;
100
101
  /* Skip any leading whitespace */
102
13.9k
  do {
103
13.9k
    ch = pbm_getc(infile);
104
13.9k
    if (ch == EOF)
105
570
      ERREXIT(cinfo, JERR_INPUT_EOF);
106
13.9k
  } while (ch == ' ' || ch == '\t' || ch == '\n' || ch == '\r');
107
108
11.0k
  if (ch < '0' || ch > '9')
109
59
    ERREXIT(cinfo, JERR_PPM_NONNUMERIC);
110
111
11.0k
  val = ch - '0';
112
16.0k
  while ((ch = pbm_getc(infile)) >= '0' && ch <= '9') {
113
5.02k
    val *= 10;
114
5.02k
    val += ch - '0';
115
5.02k
    if (val > maxval)
116
41
      ERREXIT(cinfo, JERR_PPM_OUTOFRANGE);
117
5.02k
  }
118
119
11.0k
  return val;
120
11.0k
}
121
122
123
/*
124
 * Read one row of pixels.
125
 *
126
 * We provide several different versions depending on input file format.
127
 * In all cases, input is scaled to the size of JSAMPLE.
128
 *
129
 * A really fast path is provided for reading byte/sample raw files with
130
 * maxval = MAXJSAMPLE, which is the normal case for 8-bit data.
131
 */
132
133
134
METHODDEF(JDIMENSION)
135
get_text_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
136
/* This version is for reading text-format PGM files with any maxval */
137
370
{
138
370
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
139
370
  FILE *infile = source->pub.input_file;
140
370
  register JSAMPROW ptr;
141
370
  register JSAMPLE *rescale = source->rescale;
142
370
  JDIMENSION col;
143
370
  unsigned int maxval = source->maxval;
144
145
370
  ptr = source->pub.buffer[0];
146
1.95k
  for (col = cinfo->image_width; col > 0; col--) {
147
1.58k
    *ptr++ = rescale[read_pbm_integer(cinfo, infile, maxval)];
148
1.58k
  }
149
370
  return 1;
150
370
}
151
152
153
0
#define GRAY_RGB_READ_LOOP(read_op, alpha_set_op) { \
154
0
  for (col = cinfo->image_width; col > 0; col--) { \
155
0
    ptr[rindex] = ptr[gindex] = ptr[bindex] = read_op; \
156
0
    alpha_set_op \
157
0
    ptr += ps; \
158
0
  } \
159
0
}
160
161
METHODDEF(JDIMENSION)
162
get_text_gray_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
163
/* This version is for reading text-format PGM files with any maxval and
164
   converting to extended RGB */
165
0
{
166
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
167
0
  FILE *infile = source->pub.input_file;
168
0
  register JSAMPROW ptr;
169
0
  register JSAMPLE *rescale = source->rescale;
170
0
  JDIMENSION col;
171
0
  unsigned int maxval = source->maxval;
172
0
  register int rindex = rgb_red[cinfo->in_color_space];
173
0
  register int gindex = rgb_green[cinfo->in_color_space];
174
0
  register int bindex = rgb_blue[cinfo->in_color_space];
175
0
  register int aindex = alpha_index[cinfo->in_color_space];
176
0
  register int ps = rgb_pixelsize[cinfo->in_color_space];
177
178
0
  ptr = source->pub.buffer[0];
179
0
  if (maxval == MAXJSAMPLE) {
180
0
    if (aindex >= 0)
181
0
      GRAY_RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval),
182
0
                         ptr[aindex] = MAXJSAMPLE;)
183
0
    else
184
0
      GRAY_RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), {})
185
0
  } else {
186
0
    if (aindex >= 0)
187
0
      GRAY_RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)],
188
0
                         ptr[aindex] = MAXJSAMPLE;)
189
0
    else
190
0
      GRAY_RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], {})
191
0
  }
192
0
  return 1;
193
0
}
194
195
196
METHODDEF(JDIMENSION)
197
get_text_gray_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
198
/* This version is for reading text-format PGM files with any maxval and
199
   converting to CMYK */
200
0
{
201
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
202
0
  FILE *infile = source->pub.input_file;
203
0
  register JSAMPROW ptr;
204
0
  register JSAMPLE *rescale = source->rescale;
205
0
  JDIMENSION col;
206
0
  unsigned int maxval = source->maxval;
207
208
0
  ptr = source->pub.buffer[0];
209
0
  if (maxval == MAXJSAMPLE) {
210
0
    for (col = cinfo->image_width; col > 0; col--) {
211
0
      JSAMPLE gray = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval);
212
0
      rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3);
213
0
      ptr += 4;
214
0
    }
215
0
  } else {
216
0
    for (col = cinfo->image_width; col > 0; col--) {
217
0
      JSAMPLE gray = rescale[read_pbm_integer(cinfo, infile, maxval)];
218
0
      rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3);
219
0
      ptr += 4;
220
0
    }
221
0
  }
222
0
  return 1;
223
0
}
224
225
226
2.11k
#define RGB_READ_LOOP(read_op, alpha_set_op) { \
227
5.21k
  for (col = cinfo->image_width; col > 0; col--) { \
228
3.09k
    ptr[rindex] = read_op; \
229
3.09k
    ptr[gindex] = read_op; \
230
3.09k
    ptr[bindex] = read_op; \
231
3.09k
    alpha_set_op \
232
3.09k
    ptr += ps; \
233
3.09k
  } \
234
2.11k
}
235
236
METHODDEF(JDIMENSION)
237
get_text_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
238
/* This version is for reading text-format PPM files with any maxval */
239
800
{
240
800
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
241
800
  FILE *infile = source->pub.input_file;
242
800
  register JSAMPROW ptr;
243
800
  register JSAMPLE *rescale = source->rescale;
244
800
  JDIMENSION col;
245
800
  unsigned int maxval = source->maxval;
246
800
  register int rindex = rgb_red[cinfo->in_color_space];
247
800
  register int gindex = rgb_green[cinfo->in_color_space];
248
800
  register int bindex = rgb_blue[cinfo->in_color_space];
249
800
  register int aindex = alpha_index[cinfo->in_color_space];
250
800
  register int ps = rgb_pixelsize[cinfo->in_color_space];
251
252
800
  ptr = source->pub.buffer[0];
253
800
  if (maxval == MAXJSAMPLE) {
254
232
    if (aindex >= 0)
255
0
      RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval),
256
232
                    ptr[aindex] = MAXJSAMPLE;)
257
232
    else
258
232
      RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), {})
259
568
  } else {
260
568
    if (aindex >= 0)
261
0
      RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)],
262
568
                    ptr[aindex] = MAXJSAMPLE;)
263
568
    else
264
568
      RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], {})
265
568
  }
266
800
  return 1;
267
800
}
268
269
270
METHODDEF(JDIMENSION)
271
get_text_rgb_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
272
/* This version is for reading text-format PPM files with any maxval and
273
   converting to CMYK */
274
0
{
275
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
276
0
  FILE *infile = source->pub.input_file;
277
0
  register JSAMPROW ptr;
278
0
  register JSAMPLE *rescale = source->rescale;
279
0
  JDIMENSION col;
280
0
  unsigned int maxval = source->maxval;
281
282
0
  ptr = source->pub.buffer[0];
283
0
  if (maxval == MAXJSAMPLE) {
284
0
    for (col = cinfo->image_width; col > 0; col--) {
285
0
      JSAMPLE r = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval);
286
0
      JSAMPLE g = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval);
287
0
      JSAMPLE b = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval);
288
0
      rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3);
289
0
      ptr += 4;
290
0
    }
291
0
  } else {
292
0
    for (col = cinfo->image_width; col > 0; col--) {
293
0
      JSAMPLE r = rescale[read_pbm_integer(cinfo, infile, maxval)];
294
0
      JSAMPLE g = rescale[read_pbm_integer(cinfo, infile, maxval)];
295
0
      JSAMPLE b = rescale[read_pbm_integer(cinfo, infile, maxval)];
296
0
      rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3);
297
0
      ptr += 4;
298
0
    }
299
0
  }
300
0
  return 1;
301
0
}
302
303
304
METHODDEF(JDIMENSION)
305
get_scaled_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
306
/* This version is for reading raw-byte-format PGM files with any maxval */
307
671
{
308
671
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
309
671
  register JSAMPROW ptr;
310
671
  register U_CHAR *bufferptr;
311
671
  register JSAMPLE *rescale = source->rescale;
312
671
  JDIMENSION col;
313
314
671
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
315
77
    ERREXIT(cinfo, JERR_INPUT_EOF);
316
671
  ptr = source->pub.buffer[0];
317
671
  bufferptr = source->iobuffer;
318
32.6k
  for (col = cinfo->image_width; col > 0; col--) {
319
32.0k
    *ptr++ = rescale[UCH(*bufferptr++)];
320
32.0k
  }
321
671
  return 1;
322
671
}
323
324
325
METHODDEF(JDIMENSION)
326
get_gray_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
327
/* This version is for reading raw-byte-format PGM files with any maxval
328
   and converting to extended RGB */
329
0
{
330
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
331
0
  register JSAMPROW ptr;
332
0
  register U_CHAR *bufferptr;
333
0
  register JSAMPLE *rescale = source->rescale;
334
0
  JDIMENSION col;
335
0
  unsigned int maxval = source->maxval;
336
0
  register int rindex = rgb_red[cinfo->in_color_space];
337
0
  register int gindex = rgb_green[cinfo->in_color_space];
338
0
  register int bindex = rgb_blue[cinfo->in_color_space];
339
0
  register int aindex = alpha_index[cinfo->in_color_space];
340
0
  register int ps = rgb_pixelsize[cinfo->in_color_space];
341
342
0
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
343
0
    ERREXIT(cinfo, JERR_INPUT_EOF);
344
0
  ptr = source->pub.buffer[0];
345
0
  bufferptr = source->iobuffer;
346
0
  if (maxval == MAXJSAMPLE) {
347
0
    if (aindex >= 0)
348
0
      GRAY_RGB_READ_LOOP(*bufferptr++, ptr[aindex] = MAXJSAMPLE;)
349
0
    else
350
0
      GRAY_RGB_READ_LOOP(*bufferptr++, {})
351
0
  } else {
352
0
    if (aindex >= 0)
353
0
      GRAY_RGB_READ_LOOP(rescale[UCH(*bufferptr++)], ptr[aindex] = MAXJSAMPLE;)
354
0
    else
355
0
      GRAY_RGB_READ_LOOP(rescale[UCH(*bufferptr++)], {})
356
0
  }
357
0
  return 1;
358
0
}
359
360
361
METHODDEF(JDIMENSION)
362
get_gray_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
363
/* This version is for reading raw-byte-format PGM files with any maxval
364
   and converting to CMYK */
365
0
{
366
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
367
0
  register JSAMPROW ptr;
368
0
  register U_CHAR *bufferptr;
369
0
  register JSAMPLE *rescale = source->rescale;
370
0
  JDIMENSION col;
371
0
  unsigned int maxval = source->maxval;
372
373
0
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
374
0
    ERREXIT(cinfo, JERR_INPUT_EOF);
375
0
  ptr = source->pub.buffer[0];
376
0
  bufferptr = source->iobuffer;
377
0
  if (maxval == MAXJSAMPLE) {
378
0
    for (col = cinfo->image_width; col > 0; col--) {
379
0
      JSAMPLE gray = *bufferptr++;
380
0
      rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3);
381
0
      ptr += 4;
382
0
    }
383
0
  } else {
384
0
    for (col = cinfo->image_width; col > 0; col--) {
385
0
      JSAMPLE gray = rescale[UCH(*bufferptr++)];
386
0
      rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3);
387
0
      ptr += 4;
388
0
    }
389
0
  }
390
0
  return 1;
391
0
}
392
393
394
METHODDEF(JDIMENSION)
395
get_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
396
/* This version is for reading raw-byte-format PPM files with any maxval */
397
1.31k
{
398
1.31k
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
399
1.31k
  register JSAMPROW ptr;
400
1.31k
  register U_CHAR *bufferptr;
401
1.31k
  register JSAMPLE *rescale = source->rescale;
402
1.31k
  JDIMENSION col;
403
1.31k
  unsigned int maxval = source->maxval;
404
1.31k
  register int rindex = rgb_red[cinfo->in_color_space];
405
1.31k
  register int gindex = rgb_green[cinfo->in_color_space];
406
1.31k
  register int bindex = rgb_blue[cinfo->in_color_space];
407
1.31k
  register int aindex = alpha_index[cinfo->in_color_space];
408
1.31k
  register int ps = rgb_pixelsize[cinfo->in_color_space];
409
410
1.31k
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
411
170
    ERREXIT(cinfo, JERR_INPUT_EOF);
412
1.31k
  ptr = source->pub.buffer[0];
413
1.31k
  bufferptr = source->iobuffer;
414
1.31k
  if (maxval == MAXJSAMPLE) {
415
0
    if (aindex >= 0)
416
0
      RGB_READ_LOOP(*bufferptr++, ptr[aindex] = MAXJSAMPLE;)
417
0
    else
418
0
      RGB_READ_LOOP(*bufferptr++, {})
419
1.31k
  } else {
420
1.31k
    if (aindex >= 0)
421
0
      RGB_READ_LOOP(rescale[UCH(*bufferptr++)], ptr[aindex] = MAXJSAMPLE;)
422
1.31k
    else
423
1.31k
      RGB_READ_LOOP(rescale[UCH(*bufferptr++)], {})
424
1.31k
  }
425
1.31k
  return 1;
426
1.31k
}
427
428
429
METHODDEF(JDIMENSION)
430
get_rgb_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
431
/* This version is for reading raw-byte-format PPM files with any maxval and
432
   converting to CMYK */
433
0
{
434
0
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
435
0
  register JSAMPROW ptr;
436
0
  register U_CHAR *bufferptr;
437
0
  register JSAMPLE *rescale = source->rescale;
438
0
  JDIMENSION col;
439
0
  unsigned int maxval = source->maxval;
440
441
0
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
442
0
    ERREXIT(cinfo, JERR_INPUT_EOF);
443
0
  ptr = source->pub.buffer[0];
444
0
  bufferptr = source->iobuffer;
445
0
  if (maxval == MAXJSAMPLE) {
446
0
    for (col = cinfo->image_width; col > 0; col--) {
447
0
      JSAMPLE r = *bufferptr++;
448
0
      JSAMPLE g = *bufferptr++;
449
0
      JSAMPLE b = *bufferptr++;
450
0
      rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3);
451
0
      ptr += 4;
452
0
    }
453
0
  } else {
454
0
    for (col = cinfo->image_width; col > 0; col--) {
455
0
      JSAMPLE r = rescale[UCH(*bufferptr++)];
456
0
      JSAMPLE g = rescale[UCH(*bufferptr++)];
457
0
      JSAMPLE b = rescale[UCH(*bufferptr++)];
458
0
      rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3);
459
0
      ptr += 4;
460
0
    }
461
0
  }
462
0
  return 1;
463
0
}
464
465
466
METHODDEF(JDIMENSION)
467
get_raw_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
468
/* This version is for reading raw-byte-format files with maxval = MAXJSAMPLE.
469
 * In this case we just read right into the JSAMPLE buffer!
470
 * Note that same code works for PPM and PGM files.
471
 */
472
396
{
473
396
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
474
475
396
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
476
22
    ERREXIT(cinfo, JERR_INPUT_EOF);
477
396
  return 1;
478
396
}
479
480
481
METHODDEF(JDIMENSION)
482
get_word_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
483
/* This version is for reading raw-word-format PGM files with any maxval */
484
425
{
485
425
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
486
425
  register JSAMPROW ptr;
487
425
  register U_CHAR *bufferptr;
488
425
  register JSAMPLE *rescale = source->rescale;
489
425
  JDIMENSION col;
490
425
  unsigned int maxval = source->maxval;
491
492
425
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
493
49
    ERREXIT(cinfo, JERR_INPUT_EOF);
494
425
  ptr = source->pub.buffer[0];
495
425
  bufferptr = source->iobuffer;
496
1.00k
  for (col = cinfo->image_width; col > 0; col--) {
497
578
    register unsigned int temp;
498
578
    temp  = UCH(*bufferptr++) << 8;
499
578
    temp |= UCH(*bufferptr++);
500
578
    if (temp > maxval)
501
13
      ERREXIT(cinfo, JERR_PPM_OUTOFRANGE);
502
578
    *ptr++ = rescale[temp];
503
578
  }
504
425
  return 1;
505
425
}
506
507
508
METHODDEF(JDIMENSION)
509
get_word_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
510
/* This version is for reading raw-word-format PPM files with any maxval */
511
626
{
512
626
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
513
626
  register JSAMPROW ptr;
514
626
  register U_CHAR *bufferptr;
515
626
  register JSAMPLE *rescale = source->rescale;
516
626
  JDIMENSION col;
517
626
  unsigned int maxval = source->maxval;
518
626
  register int rindex = rgb_red[cinfo->in_color_space];
519
626
  register int gindex = rgb_green[cinfo->in_color_space];
520
626
  register int bindex = rgb_blue[cinfo->in_color_space];
521
626
  register int aindex = alpha_index[cinfo->in_color_space];
522
626
  register int ps = rgb_pixelsize[cinfo->in_color_space];
523
524
626
  if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width))
525
96
    ERREXIT(cinfo, JERR_INPUT_EOF);
526
626
  ptr = source->pub.buffer[0];
527
626
  bufferptr = source->iobuffer;
528
1.46k
  for (col = cinfo->image_width; col > 0; col--) {
529
842
    register unsigned int temp;
530
842
    temp  = UCH(*bufferptr++) << 8;
531
842
    temp |= UCH(*bufferptr++);
532
842
    if (temp > maxval)
533
44
      ERREXIT(cinfo, JERR_PPM_OUTOFRANGE);
534
842
    ptr[rindex] = rescale[temp];
535
842
    temp  = UCH(*bufferptr++) << 8;
536
842
    temp |= UCH(*bufferptr++);
537
842
    if (temp > maxval)
538
40
      ERREXIT(cinfo, JERR_PPM_OUTOFRANGE);
539
842
    ptr[gindex] = rescale[temp];
540
842
    temp  = UCH(*bufferptr++) << 8;
541
842
    temp |= UCH(*bufferptr++);
542
842
    if (temp > maxval)
543
32
      ERREXIT(cinfo, JERR_PPM_OUTOFRANGE);
544
842
    ptr[bindex] = rescale[temp];
545
842
    if (aindex >= 0)
546
0
      ptr[aindex] = MAXJSAMPLE;
547
842
    ptr += ps;
548
842
  }
549
626
  return 1;
550
626
}
551
552
553
/*
554
 * Read the file header; return image size and component count.
555
 */
556
557
METHODDEF(void)
558
start_input_ppm(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
559
2.12k
{
560
2.12k
  ppm_source_ptr source = (ppm_source_ptr)sinfo;
561
2.12k
  int c;
562
2.12k
  unsigned int w, h, maxval;
563
2.12k
  boolean need_iobuffer, use_raw_buffer, need_rescale;
564
565
2.12k
  if (getc(source->pub.input_file) != 'P')
566
0
    ERREXIT(cinfo, JERR_PPM_NOT);
567
568
2.12k
  c = getc(source->pub.input_file); /* subformat discriminator character */
569
570
  /* detect unsupported variants (ie, PBM) before trying to read header */
571
2.12k
  switch (c) {
572
368
  case '2':                     /* it's a text-format PGM file */
573
680
  case '3':                     /* it's a text-format PPM file */
574
1.21k
  case '5':                     /* it's a raw-format PGM file */
575
2.11k
  case '6':                     /* it's a raw-format PPM file */
576
2.11k
    break;
577
4
  default:
578
4
    ERREXIT(cinfo, JERR_PPM_NOT);
579
4
    break;
580
2.12k
  }
581
582
  /* fetch the remaining header info */
583
2.11k
  w = read_pbm_integer(cinfo, source->pub.input_file, 65535);
584
2.11k
  h = read_pbm_integer(cinfo, source->pub.input_file, 65535);
585
2.11k
  maxval = read_pbm_integer(cinfo, source->pub.input_file, 65535);
586
587
2.11k
  if (w <= 0 || h <= 0 || maxval <= 0) /* error check */
588
6
    ERREXIT(cinfo, JERR_PPM_NOT);
589
2.11k
#ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
590
2.11k
  if (sinfo->max_pixels && (unsigned long long)w * h > sinfo->max_pixels)
591
52
    ERREXIT(cinfo, JERR_WIDTH_OVERFLOW);
592
2.11k
#endif
593
594
2.11k
  cinfo->data_precision = BITS_IN_JSAMPLE; /* we always rescale data to this */
595
2.11k
  cinfo->image_width = (JDIMENSION)w;
596
2.11k
  cinfo->image_height = (JDIMENSION)h;
597
2.11k
  source->maxval = maxval;
598
599
  /* initialize flags to most common settings */
600
2.11k
  need_iobuffer = TRUE;         /* do we need an I/O buffer? */
601
2.11k
  use_raw_buffer = FALSE;       /* do we map input buffer onto I/O buffer? */
602
2.11k
  need_rescale = TRUE;          /* do we need a rescale array? */
603
604
2.11k
  switch (c) {
605
244
  case '2':                     /* it's a text-format PGM file */
606
244
    if (cinfo->in_color_space == JCS_UNKNOWN ||
607
244
        cinfo->in_color_space == JCS_RGB)
608
244
      cinfo->in_color_space = JCS_GRAYSCALE;
609
244
    TRACEMS2(cinfo, 1, JTRC_PGM_TEXT, w, h);
610
244
    if (cinfo->in_color_space == JCS_GRAYSCALE)
611
244
      source->pub.get_pixel_rows = get_text_gray_row;
612
0
    else if (IsExtRGB(cinfo->in_color_space))
613
0
      source->pub.get_pixel_rows = get_text_gray_rgb_row;
614
0
    else if (cinfo->in_color_space == JCS_CMYK)
615
0
      source->pub.get_pixel_rows = get_text_gray_cmyk_row;
616
0
    else
617
0
      ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
618
244
    need_iobuffer = FALSE;
619
244
    break;
620
621
218
  case '3':                     /* it's a text-format PPM file */
622
218
    if (cinfo->in_color_space == JCS_UNKNOWN)
623
0
      cinfo->in_color_space = JCS_EXT_RGB;
624
218
    TRACEMS2(cinfo, 1, JTRC_PPM_TEXT, w, h);
625
218
    if (IsExtRGB(cinfo->in_color_space))
626
218
      source->pub.get_pixel_rows = get_text_rgb_row;
627
0
    else if (cinfo->in_color_space == JCS_CMYK)
628
0
      source->pub.get_pixel_rows = get_text_rgb_cmyk_row;
629
0
    else
630
0
      ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
631
218
    need_iobuffer = FALSE;
632
218
    break;
633
634
438
  case '5':                     /* it's a raw-format PGM file */
635
438
    if (cinfo->in_color_space == JCS_UNKNOWN ||
636
438
        cinfo->in_color_space == JCS_RGB)
637
438
      cinfo->in_color_space = JCS_GRAYSCALE;
638
438
    TRACEMS2(cinfo, 1, JTRC_PGM, w, h);
639
438
    if (maxval > 255) {
640
130
      if (cinfo->in_color_space == JCS_GRAYSCALE)
641
130
        source->pub.get_pixel_rows = get_word_gray_row;
642
0
      else
643
0
        ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
644
308
    } else if (maxval == MAXJSAMPLE && sizeof(JSAMPLE) == sizeof(U_CHAR) &&
645
308
               cinfo->in_color_space == JCS_GRAYSCALE) {
646
40
      source->pub.get_pixel_rows = get_raw_row;
647
40
      use_raw_buffer = TRUE;
648
40
      need_rescale = FALSE;
649
268
    } else {
650
268
      if (cinfo->in_color_space == JCS_GRAYSCALE)
651
268
        source->pub.get_pixel_rows = get_scaled_gray_row;
652
0
      else if (IsExtRGB(cinfo->in_color_space))
653
0
        source->pub.get_pixel_rows = get_gray_rgb_row;
654
0
      else if (cinfo->in_color_space == JCS_CMYK)
655
0
        source->pub.get_pixel_rows = get_gray_cmyk_row;
656
0
      else
657
0
        ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
658
268
    }
659
438
    break;
660
661
810
  case '6':                     /* it's a raw-format PPM file */
662
810
    if (cinfo->in_color_space == JCS_UNKNOWN)
663
0
      cinfo->in_color_space = JCS_EXT_RGB;
664
810
    TRACEMS2(cinfo, 1, JTRC_PPM, w, h);
665
810
    if (maxval > 255) {
666
214
      if (IsExtRGB(cinfo->in_color_space))
667
214
        source->pub.get_pixel_rows = get_word_rgb_row;
668
0
      else
669
0
        ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
670
596
    } else if (maxval == MAXJSAMPLE && sizeof(JSAMPLE) == sizeof(U_CHAR) &&
671
596
#if RGB_RED == 0 && RGB_GREEN == 1 && RGB_BLUE == 2 && RGB_PIXELSIZE == 3
672
596
               (cinfo->in_color_space == JCS_EXT_RGB ||
673
4
                cinfo->in_color_space == JCS_RGB)) {
674
#else
675
               cinfo->in_color_space == JCS_EXT_RGB) {
676
#endif
677
4
      source->pub.get_pixel_rows = get_raw_row;
678
4
      use_raw_buffer = TRUE;
679
4
      need_rescale = FALSE;
680
592
    } else {
681
592
      if (IsExtRGB(cinfo->in_color_space))
682
592
        source->pub.get_pixel_rows = get_rgb_row;
683
0
      else if (cinfo->in_color_space == JCS_CMYK)
684
0
        source->pub.get_pixel_rows = get_rgb_cmyk_row;
685
0
      else
686
0
        ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE);
687
592
    }
688
810
    break;
689
2.11k
  }
690
691
1.71k
  if (IsExtRGB(cinfo->in_color_space))
692
1.02k
    cinfo->input_components = rgb_pixelsize[cinfo->in_color_space];
693
682
  else if (cinfo->in_color_space == JCS_GRAYSCALE)
694
682
    cinfo->input_components = 1;
695
0
  else if (cinfo->in_color_space == JCS_CMYK)
696
0
    cinfo->input_components = 4;
697
698
  /* Allocate space for I/O buffer: 1 or 3 bytes or words/pixel. */
699
1.71k
  if (need_iobuffer) {
700
1.24k
    if (c == '6')
701
810
      source->buffer_width = (size_t)w * 3 *
702
810
        ((maxval <= 255) ? sizeof(U_CHAR) : (2 * sizeof(U_CHAR)));
703
438
    else
704
438
      source->buffer_width = (size_t)w *
705
438
        ((maxval <= 255) ? sizeof(U_CHAR) : (2 * sizeof(U_CHAR)));
706
1.24k
    source->iobuffer = (U_CHAR *)
707
1.24k
      (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
708
1.24k
                                  source->buffer_width);
709
1.24k
  }
710
711
  /* Create compressor input buffer. */
712
1.71k
  if (use_raw_buffer) {
713
    /* For unscaled raw-input case, we can just map it onto the I/O buffer. */
714
    /* Synthesize a JSAMPARRAY pointer structure */
715
44
    source->pixrow = (JSAMPROW)source->iobuffer;
716
44
    source->pub.buffer = &source->pixrow;
717
44
    source->pub.buffer_height = 1;
718
1.66k
  } else {
719
    /* Need to translate anyway, so make a separate sample buffer. */
720
1.66k
    source->pub.buffer = (*cinfo->mem->alloc_sarray)
721
1.66k
      ((j_common_ptr)cinfo, JPOOL_IMAGE,
722
1.66k
       (JDIMENSION)w * cinfo->input_components, (JDIMENSION)1);
723
1.66k
    source->pub.buffer_height = 1;
724
1.66k
  }
725
726
  /* Compute the rescaling array if required. */
727
1.71k
  if (need_rescale) {
728
1.66k
    long val, half_maxval;
729
730
    /* On 16-bit-int machines we have to be careful of maxval = 65535 */
731
1.66k
    source->rescale = (JSAMPLE *)
732
1.66k
      (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
733
1.66k
                                  (size_t)(((long)MAX(maxval, 255) + 1L) *
734
1.66k
                                           sizeof(JSAMPLE)));
735
1.66k
    memset(source->rescale, 0, (size_t)(((long)MAX(maxval, 255) + 1L) *
736
1.66k
                                        sizeof(JSAMPLE)));
737
1.66k
    half_maxval = maxval / 2;
738
6.60M
    for (val = 0; val <= (long)maxval; val++) {
739
      /* The multiplication here must be done in 32 bits to avoid overflow */
740
6.60M
      source->rescale[val] = (JSAMPLE)((val * MAXJSAMPLE + half_maxval) /
741
6.60M
                                        maxval);
742
6.60M
    }
743
1.66k
  }
744
1.71k
}
745
746
747
/*
748
 * Finish up at the end of the file.
749
 */
750
751
METHODDEF(void)
752
finish_input_ppm(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
753
500
{
754
  /* no work */
755
500
}
756
757
758
/*
759
 * The module selection routine for PPM format input.
760
 */
761
762
GLOBAL(cjpeg_source_ptr)
763
jinit_read_ppm(j_compress_ptr cinfo)
764
2.12k
{
765
2.12k
  ppm_source_ptr source;
766
767
  /* Create module interface object */
768
2.12k
  source = (ppm_source_ptr)
769
2.12k
    (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
770
2.12k
                                sizeof(ppm_source_struct));
771
  /* Fill in method ptrs, except get_pixel_rows which start_input sets */
772
2.12k
  source->pub.start_input = start_input_ppm;
773
2.12k
  source->pub.finish_input = finish_input_ppm;
774
2.12k
#ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION
775
2.12k
  source->pub.max_pixels = 0;
776
2.12k
#endif
777
778
2.12k
  return (cjpeg_source_ptr)source;
779
2.12k
}
780
781
#endif /* PPM_SUPPORTED */