| /* |
| * rdtarga.c |
| * |
| * This file was part of the Independent JPEG Group's software: |
| * Copyright (C) 1991-1996, Thomas G. Lane. |
| * Modified 2017 by Guido Vollbeding. |
| * libjpeg-turbo Modifications: |
| * Copyright (C) 2018, 2021-2022, D. R. Commander. |
| * For conditions of distribution and use, see the accompanying README.ijg |
| * file. |
| * |
| * This file contains routines to read input images in Targa format. |
| * |
| * These routines may need modification for non-Unix environments or |
| * specialized applications. As they stand, they assume input from |
| * an ordinary stdio stream. They further assume that reading begins |
| * at the start of the file; start_input may need work if the |
| * user interface has already read some data (e.g., to determine that |
| * the file is indeed Targa format). |
| * |
| * Based on code contributed by Lee Daniel Crocker. |
| */ |
| |
| #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */ |
| |
| #ifdef TARGA_SUPPORTED |
| |
| |
| /* Macros to deal with unsigned chars as efficiently as compiler allows */ |
| |
| typedef unsigned char U_CHAR; |
| #define UCH(x) ((int)(x)) |
| |
| |
| #define ReadOK(file, buffer, len) \ |
| (fread(buffer, 1, len, file) == ((size_t)(len))) |
| |
| |
| /* Private version of data source object */ |
| |
| typedef struct _tga_source_struct *tga_source_ptr; |
| |
| typedef struct _tga_source_struct { |
| struct cjpeg_source_struct pub; /* public fields */ |
| |
| j_compress_ptr cinfo; /* back link saves passing separate parm */ |
| |
| JSAMPARRAY colormap; /* Targa colormap (converted to my format) */ |
| |
| jvirt_sarray_ptr whole_image; /* Needed if funny input row order */ |
| JDIMENSION current_row; /* Current logical row number to read */ |
| |
| /* Pointer to routine to extract next Targa pixel from input file */ |
| void (*read_pixel) (tga_source_ptr sinfo); |
| |
| /* Result of read_pixel is delivered here: */ |
| U_CHAR tga_pixel[4]; |
| |
| int pixel_size; /* Bytes per Targa pixel (1 to 4) */ |
| int cmap_length; /* colormap length */ |
| |
| /* State info for reading RLE-coded pixels; both counts must be init to 0 */ |
| int block_count; /* # of pixels remaining in RLE block */ |
| int dup_pixel_count; /* # of times to duplicate previous pixel */ |
| |
| /* This saves the correct pixel-row-expansion method for preload_image */ |
| JDIMENSION (*get_pixel_rows) (j_compress_ptr cinfo, cjpeg_source_ptr sinfo); |
| } tga_source_struct; |
| |
| |
| /* For expanding 5-bit pixel values to 8-bit with best rounding */ |
| |
| static const UINT8 c5to8bits[32] = { |
| 0, 8, 16, 25, 33, 41, 49, 58, |
| 66, 74, 82, 90, 99, 107, 115, 123, |
| 132, 140, 148, 156, 165, 173, 181, 189, |
| 197, 206, 214, 222, 230, 239, 247, 255 |
| }; |
| |
| |
| |
| LOCAL(int) |
| read_byte(tga_source_ptr sinfo) |
| /* Read next byte from Targa file */ |
| { |
| register FILE *infile = sinfo->pub.input_file; |
| register int c; |
| |
| if ((c = getc(infile)) == EOF) |
| ERREXIT(sinfo->cinfo, JERR_INPUT_EOF); |
| return c; |
| } |
| |
| |
| LOCAL(void) |
| read_colormap(tga_source_ptr sinfo, int cmaplen, int mapentrysize) |
| /* Read the colormap from a Targa file */ |
| { |
| int i; |
| |
| /* Presently only handles 24-bit BGR format */ |
| if (mapentrysize != 24) |
| ERREXIT(sinfo->cinfo, JERR_TGA_BADCMAP); |
| |
| for (i = 0; i < cmaplen; i++) { |
| sinfo->colormap[2][i] = (JSAMPLE)read_byte(sinfo); |
| sinfo->colormap[1][i] = (JSAMPLE)read_byte(sinfo); |
| sinfo->colormap[0][i] = (JSAMPLE)read_byte(sinfo); |
| } |
| } |
| |
| |
| /* |
| * read_pixel methods: get a single pixel from Targa file into tga_pixel[] |
| */ |
| |
| METHODDEF(void) |
| read_non_rle_pixel(tga_source_ptr sinfo) |
| /* Read one Targa pixel from the input file; no RLE expansion */ |
| { |
| register int i; |
| |
| for (i = 0; i < sinfo->pixel_size; i++) { |
| sinfo->tga_pixel[i] = (U_CHAR)read_byte(sinfo); |
| } |
| } |
| |
| |
| METHODDEF(void) |
| read_rle_pixel(tga_source_ptr sinfo) |
| /* Read one Targa pixel from the input file, expanding RLE data as needed */ |
| { |
| register int i; |
| |
| /* Duplicate previously read pixel? */ |
| if (sinfo->dup_pixel_count > 0) { |
| sinfo->dup_pixel_count--; |
| return; |
| } |
| |
| /* Time to read RLE block header? */ |
| if (--sinfo->block_count < 0) { /* decrement pixels remaining in block */ |
| i = read_byte(sinfo); |
| if (i & 0x80) { /* Start of duplicate-pixel block? */ |
| sinfo->dup_pixel_count = i & 0x7F; /* number of dups after this one */ |
| sinfo->block_count = 0; /* then read new block header */ |
| } else { |
| sinfo->block_count = i & 0x7F; /* number of pixels after this one */ |
| } |
| } |
| |
| /* Read next pixel */ |
| for (i = 0; i < sinfo->pixel_size; i++) { |
| sinfo->tga_pixel[i] = (U_CHAR)read_byte(sinfo); |
| } |
| } |
| |
| |
| /* |
| * Read one row of pixels. |
| * |
| * We provide several different versions depending on input file format. |
| */ |
| |
| |
| METHODDEF(JDIMENSION) |
| get_8bit_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| /* This version is for reading 8-bit grayscale pixels */ |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| register JSAMPROW ptr; |
| register JDIMENSION col; |
| |
| ptr = source->pub.buffer[0]; |
| for (col = cinfo->image_width; col > 0; col--) { |
| (*source->read_pixel) (source); /* Load next pixel into tga_pixel */ |
| *ptr++ = (JSAMPLE)UCH(source->tga_pixel[0]); |
| } |
| return 1; |
| } |
| |
| METHODDEF(JDIMENSION) |
| get_8bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| /* This version is for reading 8-bit colormap indexes */ |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| register int t; |
| register JSAMPROW ptr; |
| register JDIMENSION col; |
| register JSAMPARRAY colormap = source->colormap; |
| int cmaplen = source->cmap_length; |
| |
| ptr = source->pub.buffer[0]; |
| for (col = cinfo->image_width; col > 0; col--) { |
| (*source->read_pixel) (source); /* Load next pixel into tga_pixel */ |
| t = UCH(source->tga_pixel[0]); |
| if (t >= cmaplen) |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| *ptr++ = colormap[0][t]; |
| *ptr++ = colormap[1][t]; |
| *ptr++ = colormap[2][t]; |
| } |
| return 1; |
| } |
| |
| METHODDEF(JDIMENSION) |
| get_16bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| /* This version is for reading 16-bit pixels */ |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| register int t; |
| register JSAMPROW ptr; |
| register JDIMENSION col; |
| |
| ptr = source->pub.buffer[0]; |
| for (col = cinfo->image_width; col > 0; col--) { |
| (*source->read_pixel) (source); /* Load next pixel into tga_pixel */ |
| t = UCH(source->tga_pixel[0]); |
| t += UCH(source->tga_pixel[1]) << 8; |
| /* We expand 5 bit data to 8 bit sample width. |
| * The format of the 16-bit (LSB first) input word is |
| * xRRRRRGGGGGBBBBB |
| */ |
| ptr[2] = (JSAMPLE)c5to8bits[t & 0x1F]; |
| t >>= 5; |
| ptr[1] = (JSAMPLE)c5to8bits[t & 0x1F]; |
| t >>= 5; |
| ptr[0] = (JSAMPLE)c5to8bits[t & 0x1F]; |
| ptr += 3; |
| } |
| return 1; |
| } |
| |
| METHODDEF(JDIMENSION) |
| get_24bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| /* This version is for reading 24-bit pixels */ |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| register JSAMPROW ptr; |
| register JDIMENSION col; |
| |
| ptr = source->pub.buffer[0]; |
| for (col = cinfo->image_width; col > 0; col--) { |
| (*source->read_pixel) (source); /* Load next pixel into tga_pixel */ |
| *ptr++ = (JSAMPLE)UCH(source->tga_pixel[2]); /* change BGR to RGB order */ |
| *ptr++ = (JSAMPLE)UCH(source->tga_pixel[1]); |
| *ptr++ = (JSAMPLE)UCH(source->tga_pixel[0]); |
| } |
| return 1; |
| } |
| |
| /* |
| * Targa also defines a 32-bit pixel format with order B,G,R,A. |
| * We presently ignore the attribute byte, so the code for reading |
| * these pixels is identical to the 24-bit routine above. |
| * This works because the actual pixel length is only known to read_pixel. |
| */ |
| |
| #define get_32bit_row get_24bit_row |
| |
| |
| /* |
| * This method is for re-reading the input data in standard top-down |
| * row order. The entire image has already been read into whole_image |
| * with proper conversion of pixel format, but it's in a funny row order. |
| */ |
| |
| METHODDEF(JDIMENSION) |
| get_memory_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| JDIMENSION source_row; |
| |
| /* Compute row of source that maps to current_row of normal order */ |
| /* For now, assume image is bottom-up and not interlaced. */ |
| /* NEEDS WORK to support interlaced images! */ |
| source_row = cinfo->image_height - source->current_row - 1; |
| |
| /* Fetch that row from virtual array */ |
| source->pub.buffer = (*cinfo->mem->access_virt_sarray) |
| ((j_common_ptr)cinfo, source->whole_image, |
| source_row, (JDIMENSION)1, FALSE); |
| |
| source->current_row++; |
| return 1; |
| } |
| |
| |
| /* |
| * This method loads the image into whole_image during the first call on |
| * get_pixel_rows. The get_pixel_rows pointer is then adjusted to call |
| * get_memory_row on subsequent calls. |
| */ |
| |
| METHODDEF(JDIMENSION) |
| preload_image(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| JDIMENSION row; |
| cd_progress_ptr progress = (cd_progress_ptr)cinfo->progress; |
| |
| /* Read the data into a virtual array in input-file row order. */ |
| for (row = 0; row < cinfo->image_height; row++) { |
| if (progress != NULL) { |
| progress->pub.pass_counter = (long)row; |
| progress->pub.pass_limit = (long)cinfo->image_height; |
| (*progress->pub.progress_monitor) ((j_common_ptr)cinfo); |
| } |
| source->pub.buffer = (*cinfo->mem->access_virt_sarray) |
| ((j_common_ptr)cinfo, source->whole_image, row, (JDIMENSION)1, TRUE); |
| (*source->get_pixel_rows) (cinfo, sinfo); |
| } |
| if (progress != NULL) |
| progress->completed_extra_passes++; |
| |
| /* Set up to read from the virtual array in unscrambled order */ |
| source->pub.get_pixel_rows = get_memory_row; |
| source->current_row = 0; |
| /* And read the first row */ |
| return get_memory_row(cinfo, sinfo); |
| } |
| |
| |
| /* |
| * Read the file header; return image size and component count. |
| */ |
| |
| METHODDEF(void) |
| start_input_tga(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| { |
| tga_source_ptr source = (tga_source_ptr)sinfo; |
| U_CHAR targaheader[18]; |
| int idlen, cmaptype, subtype, flags, interlace_type, components; |
| unsigned int width, height, maplen; |
| boolean is_bottom_up; |
| |
| #define GET_2B(offset) \ |
| ((unsigned int)UCH(targaheader[offset]) + \ |
| (((unsigned int)UCH(targaheader[offset + 1])) << 8)) |
| |
| if (!ReadOK(source->pub.input_file, targaheader, 18)) |
| ERREXIT(cinfo, JERR_INPUT_EOF); |
| |
| /* Pretend "15-bit" pixels are 16-bit --- we ignore attribute bit anyway */ |
| if (targaheader[16] == 15) |
| targaheader[16] = 16; |
| |
| idlen = UCH(targaheader[0]); |
| cmaptype = UCH(targaheader[1]); |
| subtype = UCH(targaheader[2]); |
| maplen = GET_2B(5); |
| width = GET_2B(12); |
| height = GET_2B(14); |
| source->pixel_size = UCH(targaheader[16]) >> 3; |
| flags = UCH(targaheader[17]); /* Image Descriptor byte */ |
| |
| is_bottom_up = ((flags & 0x20) == 0); /* bit 5 set => top-down */ |
| interlace_type = flags >> 6; /* bits 6/7 are interlace code */ |
| |
| if (cmaptype > 1 || /* cmaptype must be 0 or 1 */ |
| source->pixel_size < 1 || source->pixel_size > 4 || |
| (UCH(targaheader[16]) & 7) != 0 || /* bits/pixel must be multiple of 8 */ |
| interlace_type != 0 || /* currently don't allow interlaced image */ |
| width == 0 || height == 0) /* image width/height must be non-zero */ |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| #ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION |
| if (sinfo->max_pixels && |
| (unsigned long long)width * height > sinfo->max_pixels) |
| ERREXIT(cinfo, JERR_WIDTH_OVERFLOW); |
| #endif |
| |
| if (subtype > 8) { |
| /* It's an RLE-coded file */ |
| source->read_pixel = read_rle_pixel; |
| source->block_count = source->dup_pixel_count = 0; |
| subtype -= 8; |
| } else { |
| /* Non-RLE file */ |
| source->read_pixel = read_non_rle_pixel; |
| } |
| |
| /* Now should have subtype 1, 2, or 3 */ |
| components = 3; /* until proven different */ |
| cinfo->in_color_space = JCS_RGB; |
| |
| switch (subtype) { |
| case 1: /* Colormapped image */ |
| if (source->pixel_size == 1 && cmaptype == 1) |
| source->get_pixel_rows = get_8bit_row; |
| else |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| TRACEMS2(cinfo, 1, JTRC_TGA_MAPPED, width, height); |
| break; |
| case 2: /* RGB image */ |
| switch (source->pixel_size) { |
| case 2: |
| source->get_pixel_rows = get_16bit_row; |
| break; |
| case 3: |
| source->get_pixel_rows = get_24bit_row; |
| break; |
| case 4: |
| source->get_pixel_rows = get_32bit_row; |
| break; |
| default: |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| break; |
| } |
| TRACEMS2(cinfo, 1, JTRC_TGA, width, height); |
| break; |
| case 3: /* Grayscale image */ |
| components = 1; |
| cinfo->in_color_space = JCS_GRAYSCALE; |
| if (source->pixel_size == 1) |
| source->get_pixel_rows = get_8bit_gray_row; |
| else |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| TRACEMS2(cinfo, 1, JTRC_TGA_GRAY, width, height); |
| break; |
| default: |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| break; |
| } |
| |
| if (is_bottom_up) { |
| /* Create a virtual array to buffer the upside-down image. */ |
| source->whole_image = (*cinfo->mem->request_virt_sarray) |
| ((j_common_ptr)cinfo, JPOOL_IMAGE, FALSE, |
| (JDIMENSION)width * components, (JDIMENSION)height, (JDIMENSION)1); |
| if (cinfo->progress != NULL) { |
| cd_progress_ptr progress = (cd_progress_ptr)cinfo->progress; |
| progress->total_extra_passes++; /* count file input as separate pass */ |
| } |
| /* source->pub.buffer will point to the virtual array. */ |
| source->pub.buffer_height = 1; /* in case anyone looks at it */ |
| source->pub.get_pixel_rows = preload_image; |
| } else { |
| /* Don't need a virtual array, but do need a one-row input buffer. */ |
| source->whole_image = NULL; |
| source->pub.buffer = (*cinfo->mem->alloc_sarray) |
| ((j_common_ptr)cinfo, JPOOL_IMAGE, |
| (JDIMENSION)width * components, (JDIMENSION)1); |
| source->pub.buffer_height = 1; |
| source->pub.get_pixel_rows = source->get_pixel_rows; |
| } |
| |
| while (idlen--) /* Throw away ID field */ |
| (void)read_byte(source); |
| |
| if (maplen > 0) { |
| if (maplen > 256 || GET_2B(3) != 0) |
| ERREXIT(cinfo, JERR_TGA_BADCMAP); |
| /* Allocate space to store the colormap */ |
| source->colormap = (*cinfo->mem->alloc_sarray) |
| ((j_common_ptr)cinfo, JPOOL_IMAGE, (JDIMENSION)maplen, (JDIMENSION)3); |
| source->cmap_length = (int)maplen; |
| /* and read it from the file */ |
| read_colormap(source, (int)maplen, UCH(targaheader[7])); |
| } else { |
| if (cmaptype) /* but you promised a cmap! */ |
| ERREXIT(cinfo, JERR_TGA_BADPARMS); |
| source->colormap = NULL; |
| source->cmap_length = 0; |
| } |
| |
| cinfo->input_components = components; |
| cinfo->data_precision = 8; |
| cinfo->image_width = width; |
| cinfo->image_height = height; |
| } |
| |
| |
| /* |
| * Finish up at the end of the file. |
| */ |
| |
| METHODDEF(void) |
| finish_input_tga(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
| { |
| /* no work */ |
| } |
| |
| |
| /* |
| * The module selection routine for Targa format input. |
| */ |
| |
| GLOBAL(cjpeg_source_ptr) |
| jinit_read_targa(j_compress_ptr cinfo) |
| { |
| tga_source_ptr source; |
| |
| if (cinfo->data_precision != 8) |
| ERREXIT1(cinfo, JERR_BAD_PRECISION, cinfo->data_precision); |
| |
| /* Create module interface object */ |
| source = (tga_source_ptr) |
| (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE, |
| sizeof(tga_source_struct)); |
| source->cinfo = cinfo; /* make back link for subroutines */ |
| /* Fill in method ptrs, except get_pixel_rows which start_input sets */ |
| source->pub.start_input = start_input_tga; |
| source->pub.finish_input = finish_input_tga; |
| #ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION |
| source->pub.max_pixels = 0; |
| #endif |
| |
| return (cjpeg_source_ptr)source; |
| } |
| |
| #endif /* TARGA_SUPPORTED */ |