mirror of
				https://git.code.sf.net/p/libpng/code.git
				synced 2025-07-10 18:04:09 +02:00 
			
		
		
		
	
		
			
				
	
	
		
			961 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			961 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| 
 | |
| /* pngunknown.c - test the read side unknown chunk handling
 | |
|  *
 | |
|  * Last changed in libpng 1.6.0 [(PENDING RELEASE)]
 | |
|  * Copyright (c) 2013 Glenn Randers-Pehrson
 | |
|  * Written by John Cunningham Bowler
 | |
|  *
 | |
|  * This code is released under the libpng license.
 | |
|  * For conditions of distribution and use, see the disclaimer
 | |
|  * and license in png.h
 | |
|  *
 | |
|  * NOTES:
 | |
|  *   This is a C program that is intended to be linked against libpng.  It
 | |
|  *   allows the libpng unknown handling code to be tested by interpreting
 | |
|  *   arguments to save or discard combinations of chunks.  The program is
 | |
|  *   currently just a minimal validation for the built-in libpng facilities.
 | |
|  */
 | |
| 
 | |
| #include <stdlib.h>
 | |
| #include <string.h>
 | |
| #include <stdio.h>
 | |
| #include <setjmp.h>
 | |
| 
 | |
| /* Define the following to use this test against your installed libpng, rather
 | |
|  * than the one being built here:
 | |
|  */
 | |
| #ifdef PNG_FREESTANDING_TESTS
 | |
| #  include <png.h>
 | |
| #else
 | |
| #  include "../../png.h"
 | |
| #endif
 | |
| 
 | |
| #ifdef PNG_READ_SUPPORTED
 | |
| 
 | |
| #if PNG_LIBPNG_VER < 10500
 | |
| /* This deliberately lacks the PNG_CONST. */
 | |
| typedef png_byte *png_const_bytep;
 | |
| 
 | |
| /* This is copied from 1.5.1 png.h: */
 | |
| #define PNG_INTERLACE_ADAM7_PASSES 7
 | |
| #define PNG_PASS_START_ROW(pass) (((1U&~(pass))<<(3-((pass)>>1)))&7)
 | |
| #define PNG_PASS_START_COL(pass) (((1U& (pass))<<(3-(((pass)+1)>>1)))&7)
 | |
| #define PNG_PASS_ROW_SHIFT(pass) ((pass)>2?(8-(pass))>>1:3)
 | |
| #define PNG_PASS_COL_SHIFT(pass) ((pass)>1?(7-(pass))>>1:3)
 | |
| #define PNG_PASS_ROWS(height, pass) (((height)+(((1<<PNG_PASS_ROW_SHIFT(pass))\
 | |
|    -1)-PNG_PASS_START_ROW(pass)))>>PNG_PASS_ROW_SHIFT(pass))
 | |
| #define PNG_PASS_COLS(width, pass) (((width)+(((1<<PNG_PASS_COL_SHIFT(pass))\
 | |
|    -1)-PNG_PASS_START_COL(pass)))>>PNG_PASS_COL_SHIFT(pass))
 | |
| #define PNG_ROW_FROM_PASS_ROW(yIn, pass) \
 | |
|    (((yIn)<<PNG_PASS_ROW_SHIFT(pass))+PNG_PASS_START_ROW(pass))
 | |
| #define PNG_COL_FROM_PASS_COL(xIn, pass) \
 | |
|    (((xIn)<<PNG_PASS_COL_SHIFT(pass))+PNG_PASS_START_COL(pass))
 | |
| #define PNG_PASS_MASK(pass,off) ( \
 | |
|    ((0x110145AFU>>(((7-(off))-(pass))<<2)) & 0xFU) | \
 | |
|    ((0x01145AF0U>>(((7-(off))-(pass))<<2)) & 0xF0U))
 | |
| #define PNG_ROW_IN_INTERLACE_PASS(y, pass) \
 | |
|    ((PNG_PASS_MASK(pass,0) >> ((y)&7)) & 1)
 | |
| #define PNG_COL_IN_INTERLACE_PASS(x, pass) \
 | |
|    ((PNG_PASS_MASK(pass,1) >> ((x)&7)) & 1)
 | |
| 
 | |
| /* These are needed too for the default build: */
 | |
| #define PNG_WRITE_16BIT_SUPPORTED
 | |
| #define PNG_READ_16BIT_SUPPORTED
 | |
| 
 | |
| /* This comes from pnglibconf.h afer 1.5: */
 | |
| #define PNG_FP_1 100000
 | |
| #define PNG_GAMMA_THRESHOLD_FIXED\
 | |
|    ((png_fixed_point)(PNG_GAMMA_THRESHOLD * PNG_FP_1))
 | |
| #endif
 | |
| 
 | |
| #if PNG_LIBPNG_VER < 10600
 | |
|    /* 1.6.0 constifies many APIs. The following exists to allow pngvalid to be
 | |
|     * compiled against earlier versions.
 | |
|     */
 | |
| #  define png_const_structp png_structp
 | |
| #endif
 | |
| 
 | |
| 
 | |
| /* Copied from pngpriv.h */
 | |
| #define PNG_32b(b,s) ((png_uint_32)(b) << (s))
 | |
| #define PNG_CHUNK(b1,b2,b3,b4) \
 | |
|    (PNG_32b(b1,24) | PNG_32b(b2,16) | PNG_32b(b3,8) | PNG_32b(b4,0))
 | |
| 
 | |
| #define png_IHDR PNG_CHUNK( 73,  72,  68,  82)
 | |
| #define png_IDAT PNG_CHUNK( 73,  68,  65,  84)
 | |
| #define png_IEND PNG_CHUNK( 73,  69,  78,  68)
 | |
| #define png_PLTE PNG_CHUNK( 80,  76,  84,  69)
 | |
| #define png_bKGD PNG_CHUNK( 98,  75,  71,  68)
 | |
| #define png_cHRM PNG_CHUNK( 99,  72,  82,  77)
 | |
| #define png_gAMA PNG_CHUNK(103,  65,  77,  65)
 | |
| #define png_hIST PNG_CHUNK(104,  73,  83,  84)
 | |
| #define png_iCCP PNG_CHUNK(105,  67,  67,  80)
 | |
| #define png_iTXt PNG_CHUNK(105,  84,  88, 116)
 | |
| #define png_oFFs PNG_CHUNK(111,  70,  70, 115)
 | |
| #define png_pCAL PNG_CHUNK(112,  67,  65,  76)
 | |
| #define png_sCAL PNG_CHUNK(115,  67,  65,  76)
 | |
| #define png_pHYs PNG_CHUNK(112,  72,  89, 115)
 | |
| #define png_sBIT PNG_CHUNK(115,  66,  73,  84)
 | |
| #define png_sPLT PNG_CHUNK(115,  80,  76,  84)
 | |
| #define png_sRGB PNG_CHUNK(115,  82,  71,  66)
 | |
| #define png_sTER PNG_CHUNK(115,  84,  69,  82)
 | |
| #define png_tEXt PNG_CHUNK(116,  69,  88, 116)
 | |
| #define png_tIME PNG_CHUNK(116,  73,  77,  69)
 | |
| #define png_tRNS PNG_CHUNK(116,  82,  78,  83)
 | |
| #define png_zTXt PNG_CHUNK(122,  84,  88, 116)
 | |
| #define png_vpAg PNG_CHUNK('v', 'p', 'A', 'g')
 | |
| 
 | |
| /* Test on flag values as defined in the spec (section 5.4): */
 | |
| #define PNG_CHUNK_ANCILLARY(c )   (1 & ((c) >> 29))
 | |
| #define PNG_CHUNK_CRITICAL(c)     (!PNG_CHUNK_ANCILLARY(c))
 | |
| #define PNG_CHUNK_PRIVATE(c)      (1 & ((c) >> 21))
 | |
| #define PNG_CHUNK_RESERVED(c)     (1 & ((c) >> 13))
 | |
| #define PNG_CHUNK_SAFE_TO_COPY(c) (1 & ((c) >>  5))
 | |
| 
 | |
| /* Chunk information */
 | |
| #define PNG_INFO_tEXt 0x10000000U
 | |
| #define PNG_INFO_iTXt 0x20000000U
 | |
| #define PNG_INFO_zTXt 0x40000000U
 | |
| 
 | |
| #define PNG_INFO_sTER 0x01000000U
 | |
| #define PNG_INFO_vpAg 0x02000000U
 | |
| 
 | |
| #define ABSENT  0
 | |
| #define START   1
 | |
| #define END     2
 | |
| 
 | |
| static struct
 | |
| {
 | |
|    char        name[5];
 | |
|    png_uint_32 flag;
 | |
|    png_uint_32 tag;
 | |
|    int         unknown;    /* Chunk not known to libpng */
 | |
|    int         all;        /* Chunk set by the '-1' option */
 | |
|    int         position;   /* position in pngtest.png */
 | |
|    int         keep;       /* unknown handling setting */
 | |
| } chunk_info[] = {
 | |
|    /* Critical chunks */
 | |
|    { "IDAT", PNG_INFO_IDAT, png_IDAT, 0, 0,  START, 0 }, /* must be [0] */
 | |
|    { "PLTE", PNG_INFO_PLTE, png_PLTE, 0, 0, ABSENT, 0 },
 | |
| 
 | |
|    /* Non-critical chunks that libpng handles */
 | |
|    { "bKGD", PNG_INFO_bKGD, png_bKGD, 0, 1,  START, 0 },
 | |
|    { "cHRM", PNG_INFO_cHRM, png_cHRM, 0, 1,  START, 0 },
 | |
|    { "gAMA", PNG_INFO_gAMA, png_gAMA, 0, 1,  START, 0 },
 | |
|    { "hIST", PNG_INFO_hIST, png_hIST, 0, 1, ABSENT, 0 },
 | |
|    { "iCCP", PNG_INFO_iCCP, png_iCCP, 0, 1, ABSENT, 0 },
 | |
|    { "iTXt", PNG_INFO_iTXt, png_iTXt, 0, 1, ABSENT, 0 },
 | |
|    { "oFFs", PNG_INFO_oFFs, png_oFFs, 0, 1,  START, 0 },
 | |
|    { "pCAL", PNG_INFO_pCAL, png_pCAL, 0, 1,  START, 0 },
 | |
|    { "pHYs", PNG_INFO_pHYs, png_pHYs, 0, 1,  START, 0 },
 | |
|    { "sBIT", PNG_INFO_sBIT, png_sBIT, 0, 1,  START, 0 },
 | |
|    { "sCAL", PNG_INFO_sCAL, png_sCAL, 0, 1,  START, 0 },
 | |
|    { "sPLT", PNG_INFO_sPLT, png_sPLT, 0, 1, ABSENT, 0 },
 | |
|    { "sRGB", PNG_INFO_sRGB, png_sRGB, 0, 1,  START, 0 },
 | |
|    { "tEXt", PNG_INFO_tEXt, png_tEXt, 0, 1,  START, 0 },
 | |
|    { "tIME", PNG_INFO_tIME, png_tIME, 0, 1,  START, 0 },
 | |
|    { "tRNS", PNG_INFO_tRNS, png_tRNS, 0, 0, ABSENT, 0 },
 | |
|    { "zTXt", PNG_INFO_zTXt, png_zTXt, 0, 1,    END, 0 },
 | |
| 
 | |
|    /* No libpng handling */
 | |
|    { "sTER", PNG_INFO_sTER, png_sTER, 1, 1,  START, 0 },
 | |
|    { "vpAg", PNG_INFO_vpAg, png_vpAg, 1, 0,  START, 0 },
 | |
| };
 | |
| 
 | |
| #define NINFO ((int)((sizeof chunk_info)/(sizeof chunk_info[0])))
 | |
| 
 | |
| static void
 | |
| clear_keep(void)
 | |
| {
 | |
|    int i = NINFO;
 | |
|    while (--i >= 0)
 | |
|       chunk_info[i].keep = 0;
 | |
| }
 | |
| 
 | |
| static int
 | |
| find(const char *name)
 | |
| {
 | |
|    int i = NINFO;
 | |
|    while (--i >= 0)
 | |
|    {
 | |
|       if (memcmp(chunk_info[i].name, name, 4) == 0)
 | |
|          break;
 | |
|    }
 | |
| 
 | |
|    return i;
 | |
| }
 | |
| 
 | |
| static int
 | |
| findb(const png_byte *name)
 | |
| {
 | |
|    int i = NINFO;
 | |
|    while (--i >= 0)
 | |
|    {
 | |
|       if (memcmp(chunk_info[i].name, name, 4) == 0)
 | |
|          break;
 | |
|    }
 | |
| 
 | |
|    return i;
 | |
| }
 | |
| 
 | |
| static int
 | |
| find_by_flag(png_uint_32 flag)
 | |
| {
 | |
|    int i = NINFO;
 | |
| 
 | |
|    while (--i >= 0) if (chunk_info[i].flag == flag) return i;
 | |
| 
 | |
|    fprintf(stderr, "pngunknown: internal error\n");
 | |
|    exit(4);
 | |
| }
 | |
| 
 | |
| static int
 | |
| ancillary(const char *name)
 | |
| {
 | |
|    return PNG_CHUNK_ANCILLARY(PNG_CHUNK(name[0], name[1], name[2], name[3]));
 | |
| }
 | |
| 
 | |
| static int
 | |
| ancillaryb(const png_byte *name)
 | |
| {
 | |
|    return PNG_CHUNK_ANCILLARY(PNG_CHUNK(name[0], name[1], name[2], name[3]));
 | |
| }
 | |
| 
 | |
| /* Type of an error_ptr */
 | |
| typedef struct
 | |
| {
 | |
|    jmp_buf     error_return;
 | |
|    png_structp png_ptr;
 | |
|    png_infop   info_ptr, end_ptr;
 | |
|    int         error_count;
 | |
|    int         warning_count;
 | |
|    const char *program;
 | |
|    const char *file;
 | |
|    const char *test;
 | |
| } display;
 | |
| 
 | |
| static const char init[] = "initialization";
 | |
| static const char cmd[] = "command line";
 | |
| 
 | |
| static void
 | |
| init_display(display *d, const char *program)
 | |
| {
 | |
|    memset(d, 0, sizeof *d);
 | |
|    d->png_ptr = NULL;
 | |
|    d->info_ptr = d->end_ptr = NULL;
 | |
|    d->error_count = d->warning_count = 0;
 | |
|    d->program = program;
 | |
|    d->file = program;
 | |
|    d->test = init;
 | |
| }
 | |
| 
 | |
| static void
 | |
| clean_display(display *d)
 | |
| {
 | |
|    png_destroy_read_struct(&d->png_ptr, &d->info_ptr, &d->end_ptr);
 | |
| 
 | |
|    /* This must not happen - it might cause an app crash */
 | |
|    if (d->png_ptr != NULL || d->info_ptr != NULL || d->end_ptr != NULL)
 | |
|    {
 | |
|       fprintf(stderr, "%s(%s): png_destroy_read_struct error\n", d->file,
 | |
|          d->test);
 | |
|       exit(1);
 | |
|    }
 | |
| 
 | |
|    /* Invalidate the test */
 | |
|    d->test = init;
 | |
| }
 | |
| 
 | |
| PNG_FUNCTION(void, display_exit, (display *d), static PNG_NORETURN)
 | |
| {
 | |
|    ++(d->error_count);
 | |
| 
 | |
|    if (d->png_ptr != NULL)
 | |
|       clean_display(d);
 | |
| 
 | |
|    /* During initialization and if this is a single command line argument set
 | |
|     * exit now - there is only one test, otherwise longjmp to do the next test.
 | |
|     */
 | |
|    if (d->test == init || d->test == cmd)
 | |
|       exit(1);
 | |
| 
 | |
|    longjmp(d->error_return, 1);
 | |
| }
 | |
| 
 | |
| static int
 | |
| display_rc(const display *d, int strict)
 | |
| {
 | |
|    return d->error_count + (strict ? d->warning_count : 0);
 | |
| }
 | |
| 
 | |
| /* libpng error and warning callbacks */
 | |
| PNG_FUNCTION(void, error, (png_structp png_ptr, const char *message),
 | |
|    static PNG_NORETURN)
 | |
| {
 | |
|    display *d = (display*)png_get_error_ptr(png_ptr);
 | |
| 
 | |
|    fprintf(stderr, "%s(%s): libpng error: %s\n", d->file, d->test, message);
 | |
|    display_exit(d);
 | |
| }
 | |
| 
 | |
| static void
 | |
| warning(png_structp png_ptr, const char *message)
 | |
| {
 | |
|    display *d = (display*)png_get_error_ptr(png_ptr);
 | |
| 
 | |
|    fprintf(stderr, "%s(%s): libpng warning: %s\n", d->file, d->test, message);
 | |
|    ++(d->warning_count);
 | |
| }
 | |
| 
 | |
| static png_uint_32
 | |
| get_valid(display *d, png_infop info_ptr)
 | |
| {
 | |
|    png_uint_32 flags = png_get_valid(d->png_ptr, info_ptr, (png_uint_32)~0);
 | |
| 
 | |
|    /* Map the text chunks back into the flags */
 | |
|    {
 | |
|       png_textp text;
 | |
|       png_uint_32 ntext = png_get_text(d->png_ptr, info_ptr, &text, NULL);
 | |
| 
 | |
|       while (ntext-- > 0) switch (text[ntext].compression)
 | |
|       {
 | |
|          case -1:
 | |
|             flags |= PNG_INFO_tEXt;
 | |
|             break;
 | |
|          case 0:
 | |
|             flags |= PNG_INFO_zTXt;
 | |
|             break;
 | |
|          case 1:
 | |
|          case 2:
 | |
|             flags |= PNG_INFO_iTXt;
 | |
|             break;
 | |
|          default:
 | |
|             fprintf(stderr, "%s(%s): unknown text compression %d\n", d->file,
 | |
|                d->test, text[ntext].compression);
 | |
|             display_exit(d);
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    return flags;
 | |
| }
 | |
| 
 | |
| static png_uint_32
 | |
| get_unknown(display *d, int def, png_infop info_ptr)
 | |
| {
 | |
|    /* Create corresponding 'unknown' flags */
 | |
|    png_uint_32 flags = 0;
 | |
|    {
 | |
|       png_unknown_chunkp unknown;
 | |
|       int num_unknown = png_get_unknown_chunks(d->png_ptr, info_ptr, &unknown);
 | |
| 
 | |
|       while (--num_unknown >= 0)
 | |
|       {
 | |
|          int chunk = findb(unknown[num_unknown].name);
 | |
| 
 | |
|          /* Chunks not known to pngunknown must be validated here; since they
 | |
|           * must also be unknown to libpng the 'def' behavior should have been
 | |
|           * used.
 | |
|           */
 | |
|          if (chunk < 0) switch (def)
 | |
|          {
 | |
|             default: /* impossible */
 | |
|             case PNG_HANDLE_CHUNK_AS_DEFAULT:
 | |
|             case PNG_HANDLE_CHUNK_NEVER:
 | |
|                fprintf(stderr, "%s(%s): %s: %s: unknown chunk saved\n",
 | |
|                   d->file, d->test, def ? "discard" : "default",
 | |
|                   unknown[num_unknown].name);
 | |
|                ++(d->error_count);
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_IF_SAFE:
 | |
|                if (!ancillaryb(unknown[num_unknown].name))
 | |
|                {
 | |
|                   fprintf(stderr,
 | |
|                      "%s(%s): if-safe: %s: unknown critical chunk saved\n",
 | |
|                      d->file, d->test, unknown[num_unknown].name);
 | |
|                   ++(d->error_count);
 | |
|                   break;
 | |
|                }
 | |
|                /* FALL THROUGH (safe) */
 | |
|             case PNG_HANDLE_CHUNK_ALWAYS:
 | |
|                break;
 | |
|          }
 | |
| 
 | |
|          else
 | |
|             flags |= chunk_info[chunk].flag;
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    return flags;
 | |
| }
 | |
| 
 | |
| static int
 | |
| check(FILE *fp, int argc, const char **argv, png_uint_32p flags/*out*/,
 | |
|    display *d)
 | |
| {
 | |
|    int i, def = PNG_HANDLE_CHUNK_AS_DEFAULT, npasses, ipass;
 | |
|    png_uint_32 height;
 | |
| 
 | |
|    /* Some of these errors are permanently fatal and cause an exit here, others
 | |
|     * are per-test and cause an error return.
 | |
|     */
 | |
|    d->png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, d, error,
 | |
|       warning);
 | |
|    if (d->png_ptr == NULL)
 | |
|    {
 | |
|       fprintf(stderr, "%s(%s): could not allocate png struct\n", d->file,
 | |
|          d->test);
 | |
|       /* Terminate here, this error is not test specific. */
 | |
|       exit(1);
 | |
|    }
 | |
| 
 | |
|    d->info_ptr = png_create_info_struct(d->png_ptr);
 | |
|    d->end_ptr = png_create_info_struct(d->png_ptr);
 | |
|    if (d->info_ptr == NULL || d->end_ptr == NULL)
 | |
|    {
 | |
|       fprintf(stderr, "%s(%s): could not allocate png info\n", d->file,
 | |
|          d->test);
 | |
|       clean_display(d);
 | |
|       exit(1);
 | |
|    }
 | |
| 
 | |
|    png_init_io(d->png_ptr, fp);
 | |
| 
 | |
|    /* Handle each argument in turn; multiple settings are possible for the same
 | |
|     * chunk and multiple calls will occur (the last one should override all
 | |
|     * preceding ones).
 | |
|     */
 | |
|    for (i=0; i<argc; ++i)
 | |
|    {
 | |
|       const char *equals = strchr(argv[i], '=');
 | |
| 
 | |
|       if (equals != NULL)
 | |
|       {
 | |
|          int chunk, option;
 | |
| 
 | |
|          if (strcmp(equals+1, "default") == 0)
 | |
|             option = PNG_HANDLE_CHUNK_AS_DEFAULT;
 | |
|          else if (strcmp(equals+1, "discard") == 0)
 | |
|             option = PNG_HANDLE_CHUNK_NEVER;
 | |
|          else if (strcmp(equals+1, "if-safe") == 0)
 | |
|             option = PNG_HANDLE_CHUNK_IF_SAFE;
 | |
|          else if (strcmp(equals+1, "save") == 0)
 | |
|             option = PNG_HANDLE_CHUNK_ALWAYS;
 | |
|          else
 | |
|          {
 | |
|             fprintf(stderr, "%s(%s): %s: unrecognized chunk option\n", d->file,
 | |
|                d->test, argv[i]);
 | |
|             display_exit(d);
 | |
|          }
 | |
| 
 | |
|          switch (equals - argv[i])
 | |
|          {
 | |
|             case 4: /* chunk name */
 | |
|                chunk = find(argv[i]);
 | |
| 
 | |
|                if (chunk >= 0)
 | |
|                {
 | |
|                   /* These #if tests have the effect of skipping the arguments
 | |
|                    * if SAVE support is unavailable - we can't do a useful test
 | |
|                    * in this case, so we just check the arguments!  This could
 | |
|                    * be improved in the future by using the read callback.
 | |
|                    */
 | |
| #                 ifdef PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
 | |
|                      png_byte name[5];
 | |
| 
 | |
|                      memcpy(name, chunk_info[chunk].name, 5);
 | |
|                      png_set_keep_unknown_chunks(d->png_ptr, option, name, 1);
 | |
|                      chunk_info[chunk].keep = option;
 | |
| #                 endif
 | |
|                   continue;
 | |
|                }
 | |
| 
 | |
|                break;
 | |
| 
 | |
|             case 7: /* default */
 | |
|                if (memcmp(argv[i], "default", 7) == 0)
 | |
|                {
 | |
| #                 ifdef PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
 | |
|                      png_set_keep_unknown_chunks(d->png_ptr, option, NULL, 0);
 | |
| #                 endif
 | |
|                   def = option;
 | |
|                   continue;
 | |
|                }
 | |
| 
 | |
|                break;
 | |
| 
 | |
|             case 3: /* all */
 | |
|                if (memcmp(argv[i], "all", 3) == 0)
 | |
|                {
 | |
| #                 ifdef PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
 | |
|                      png_set_keep_unknown_chunks(d->png_ptr, option, NULL, -1);
 | |
|                      def = option;
 | |
| 
 | |
|                      for (chunk = 0; chunk < NINFO; ++chunk)
 | |
|                         if (chunk_info[chunk].all)
 | |
|                            chunk_info[chunk].keep = option;
 | |
| #                 endif
 | |
|                   continue;
 | |
|                }
 | |
| 
 | |
|                break;
 | |
| 
 | |
|             default: /* some misplaced = */
 | |
| 
 | |
|                break;
 | |
|          }
 | |
|       }
 | |
| 
 | |
|       fprintf(stderr, "%s(%s): %s: unrecognized chunk argument\n", d->file,
 | |
|          d->test, argv[i]);
 | |
|       display_exit(d);
 | |
|    }
 | |
| 
 | |
|    png_read_info(d->png_ptr, d->info_ptr);
 | |
| 
 | |
|    switch (png_get_interlace_type(d->png_ptr, d->info_ptr))
 | |
|    {
 | |
|       case PNG_INTERLACE_NONE:
 | |
|          npasses = 1;
 | |
|          break;
 | |
| 
 | |
|       case PNG_INTERLACE_ADAM7:
 | |
|          npasses = PNG_INTERLACE_ADAM7_PASSES;
 | |
|          break;
 | |
| 
 | |
|       default:
 | |
|          /* Hard error because it is not test specific */
 | |
|          fprintf(stderr, "%s(%s): invalid interlace type\n", d->file, d->test);
 | |
|          clean_display(d);
 | |
|          exit(1);
 | |
|    }
 | |
| 
 | |
|    /* Skip the image data, if IDAT is not being handled then don't do this
 | |
|     * because it will cause a CRC error.
 | |
|     */
 | |
|    if (chunk_info[0/*IDAT*/].keep == PNG_HANDLE_CHUNK_AS_DEFAULT)
 | |
|    {
 | |
|       png_start_read_image(d->png_ptr);
 | |
|       height = png_get_image_height(d->png_ptr, d->info_ptr);
 | |
| 
 | |
|       if (npasses > 1)
 | |
|       {
 | |
|          png_uint_32 width = png_get_image_width(d->png_ptr, d->info_ptr);
 | |
| 
 | |
|          for (ipass=0; ipass<npasses; ++ipass)
 | |
|          {
 | |
|             png_uint_32 wPass = PNG_PASS_COLS(width, ipass);
 | |
| 
 | |
|             if (wPass > 0)
 | |
|             {
 | |
|                png_uint_32 y;
 | |
| 
 | |
|                for (y=0; y<height; ++y) if (PNG_ROW_IN_INTERLACE_PASS(y, ipass))
 | |
|                   png_read_row(d->png_ptr, NULL, NULL);
 | |
|             }
 | |
|          }
 | |
|       } /* interlaced */
 | |
| 
 | |
|       else /* not interlaced */
 | |
|       {
 | |
|          png_uint_32 y;
 | |
| 
 | |
|          for (y=0; y<height; ++y)
 | |
|             png_read_row(d->png_ptr, NULL, NULL);
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    png_read_end(d->png_ptr, d->end_ptr);
 | |
| 
 | |
|    flags[0] = get_valid(d, d->info_ptr);
 | |
|    flags[1] = get_unknown(d, def, d->info_ptr);
 | |
| 
 | |
|    /* Only png_read_png sets PNG_INFO_IDAT! */
 | |
|    flags[chunk_info[0/*IDAT*/].keep != PNG_HANDLE_CHUNK_AS_DEFAULT] |=
 | |
|       PNG_INFO_IDAT;
 | |
| 
 | |
|    flags[2] = get_valid(d, d->end_ptr);
 | |
|    flags[3] = get_unknown(d, def, d->end_ptr);
 | |
| 
 | |
|    clean_display(d);
 | |
| 
 | |
|    return def;
 | |
| }
 | |
| 
 | |
| static void
 | |
| check_error(display *d, png_uint_32 flags, const char *message)
 | |
| {
 | |
|    while (flags)
 | |
|    {
 | |
|       png_uint_32 flag = flags & -(png_int_32)flags;
 | |
|       int i = find_by_flag(flag);
 | |
| 
 | |
|       fprintf(stderr, "%s(%s): chunk %s: %s\n", d->file, d->test,
 | |
|          chunk_info[i].name, message);
 | |
|       ++(d->error_count);
 | |
| 
 | |
|       flags &= ~flag;
 | |
|    }
 | |
| }
 | |
| 
 | |
| static void
 | |
| check_handling(display *d, int def, png_uint_32 chunks, png_uint_32 known,
 | |
|    png_uint_32 unknown, const char *position)
 | |
| {
 | |
|    while (chunks)
 | |
|    {
 | |
|       png_uint_32 flag = chunks & -(png_int_32)chunks;
 | |
|       int i = find_by_flag(flag);
 | |
|       int keep = chunk_info[i].keep;
 | |
|       const char *type;
 | |
|       const char *errorx = NULL;
 | |
| 
 | |
|       if (chunk_info[i].unknown)
 | |
|       {
 | |
|          if (keep == PNG_HANDLE_CHUNK_AS_DEFAULT)
 | |
|          {
 | |
|             type = "UNKNOWN (default)";
 | |
|             keep = def;
 | |
|          }
 | |
| 
 | |
|          else
 | |
|             type = "UNKNOWN (specified)";
 | |
| 
 | |
|          if (flag & known)
 | |
|             errorx = "chunk processed";
 | |
| 
 | |
|          else switch (keep)
 | |
|          {
 | |
|             case PNG_HANDLE_CHUNK_AS_DEFAULT:
 | |
|                if (flag & unknown)
 | |
|                   errorx = "DEFAULT: unknown chunk saved";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_NEVER:
 | |
|                if (flag & unknown)
 | |
|                   errorx = "DISCARD: unknown chunk saved";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_IF_SAFE:
 | |
|                if (ancillary(chunk_info[i].name))
 | |
|                {
 | |
|                   if (!(flag & unknown))
 | |
|                      errorx = "IF-SAFE: unknown ancillary chunk lost";
 | |
|                }
 | |
| 
 | |
|                else if (flag & unknown)
 | |
|                   errorx = "IF-SAFE: unknown critical chunk saved";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_ALWAYS:
 | |
|                if (!(flag & unknown))
 | |
|                   errorx = "SAVE: unknown chunk lost";
 | |
|                break;
 | |
| 
 | |
|             default:
 | |
|                errorx = "internal error: bad keep";
 | |
|                break;
 | |
|          }
 | |
|       } /* unknown chunk */
 | |
| 
 | |
|       else /* known chunk */
 | |
|       {
 | |
|          type = "KNOWN";
 | |
| 
 | |
|          if (flag & known)
 | |
|          {
 | |
|             /* chunk was processed, it won't have been saved because that is
 | |
|              * caught below when checking for inconsistent processing.
 | |
|              */
 | |
|             if (keep != PNG_HANDLE_CHUNK_AS_DEFAULT)
 | |
|                errorx = "!DEFAULT: known chunk processed";
 | |
|          }
 | |
| 
 | |
|          else /* not processed */ switch (keep)
 | |
|          {
 | |
|             case PNG_HANDLE_CHUNK_AS_DEFAULT:
 | |
|                errorx = "DEFAULT: known chunk not processed";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_NEVER:
 | |
|                if (flag & unknown)
 | |
|                   errorx = "DISCARD: known chunk saved";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_IF_SAFE:
 | |
|                if (ancillary(chunk_info[i].name))
 | |
|                {
 | |
|                   if (!(flag & unknown))
 | |
|                      errorx = "IF-SAFE: known ancillary chunk lost";
 | |
|                }
 | |
| 
 | |
|                else if (flag & unknown)
 | |
|                   errorx = "IF-SAFE: known critical chunk saved";
 | |
|                break;
 | |
| 
 | |
|             case PNG_HANDLE_CHUNK_ALWAYS:
 | |
|                if (!(flag & unknown))
 | |
|                   errorx = "SAVE: known chunk lost";
 | |
|                break;
 | |
| 
 | |
|             default:
 | |
|                errorx = "internal error: bad keep (2)";
 | |
|                break;
 | |
|          }
 | |
|       }
 | |
| 
 | |
|       if (errorx != NULL)
 | |
|       {
 | |
|          ++(d->error_count);
 | |
|          fprintf(stderr, "%s(%s): %s %s %s: %s\n",
 | |
|             d->file, d->test, type, chunk_info[i].name, position, errorx);
 | |
|       }
 | |
| 
 | |
|       chunks &= ~flag;
 | |
|    }
 | |
| }
 | |
| 
 | |
| static void
 | |
| perform_one_test(FILE *fp, int argc, const char **argv,
 | |
|    png_uint_32 *default_flags, display *d)
 | |
| {
 | |
|    int def;
 | |
|    png_uint_32 flags[2][4];
 | |
| 
 | |
|    rewind(fp);
 | |
|    clear_keep();
 | |
|    memcpy(flags[0], default_flags, sizeof flags[0]);
 | |
| 
 | |
|    def = check(fp, argc, argv, flags[1], d);
 | |
| 
 | |
|    /* Chunks should either be known or unknown, never both and this should apply
 | |
|     * whether the chunk is before or after the IDAT (actually, the app can
 | |
|     * probably change this by swapping the handling after the image, but this
 | |
|     * test does not do that.)
 | |
|     */
 | |
|    check_error(d, (flags[0][0]|flags[0][2]) & (flags[0][1]|flags[0][3]),
 | |
|       "chunk handled inconsistently in count tests");
 | |
|    check_error(d, (flags[1][0]|flags[1][2]) & (flags[1][1]|flags[1][3]),
 | |
|       "chunk handled inconsistently in option tests");
 | |
| 
 | |
|    /* Now find out what happened to each chunk before and after the IDAT and
 | |
|     * determine if the behavior was correct.  First some basic sanity checks,
 | |
|     * any known chunk should be known in the original count, any unknown chunk
 | |
|     * should be either known or unknown in the original.
 | |
|     */
 | |
|    {
 | |
|       png_uint_32 test;
 | |
| 
 | |
|       test = flags[1][0] & ~flags[0][0];
 | |
|       check_error(d, test, "new known chunk before IDAT");
 | |
|       test = flags[1][1] & ~(flags[0][0] | flags[0][1]);
 | |
|       check_error(d, test, "new unknown chunk before IDAT");
 | |
|       test = flags[1][2] & ~flags[0][2];
 | |
|       check_error(d, test, "new known chunk after IDAT");
 | |
|       test = flags[1][3] & ~(flags[0][2] | flags[0][3]);
 | |
|       check_error(d, test, "new unknown chunk after IDAT");
 | |
|    }
 | |
| 
 | |
|    /* Now each chunk in the original list should have been handled according to
 | |
|     * the options set for that chunk, regardless of whether libpng knows about
 | |
|     * it or not.
 | |
|     */
 | |
|    check_handling(d, def, flags[0][0] | flags[0][1], flags[1][0], flags[1][1],
 | |
|       "before IDAT");
 | |
|    check_handling(d, def, flags[0][2] | flags[0][3], flags[1][2], flags[1][3],
 | |
|       "after IDAT");
 | |
| }
 | |
| 
 | |
| static void
 | |
| perform_one_test_safe(FILE *fp, int argc, const char **argv,
 | |
|    png_uint_32 *default_flags, display *d, const char *test)
 | |
| {
 | |
|    if (setjmp(d->error_return) == 0)
 | |
|    {
 | |
|       d->test = test; /* allow use of d->error_return */
 | |
|       perform_one_test(fp, argc, argv, default_flags, d);
 | |
|       d->test = init; /* prevent use of d->error_return */
 | |
|    }
 | |
| }
 | |
| 
 | |
| static const char *standard_tests[] =
 | |
| {
 | |
|  "discard", "default=discard", 0,
 | |
|  "save", "default=save", 0,
 | |
|  "if-safe", "default=if-safe", 0,
 | |
|  "vpAg", "vpAg=if-safe", 0,
 | |
|  "sTER", "sTER=if-safe", 0,
 | |
|  "IDAT", "default=discard", "IDAT=save", 0,
 | |
|  "sAPI", "bKGD=save", "cHRM=save", "gAMA=save", "all=discard", "iCCP=save",
 | |
|    "sBIT=save", "sRGB=save", 0,
 | |
|  0/*end*/
 | |
| };
 | |
| 
 | |
| static PNG_NORETURN void
 | |
| usage(const char *program, const char *reason)
 | |
| {
 | |
|    fprintf(stderr, "pngunknown: %s: usage:\n %s [--strict] "
 | |
|       "--default|{(CHNK|default|all)=(default|discard|if-safe|save)} "
 | |
|       "testfile.png\n", reason, program);
 | |
|    exit(2);
 | |
| }
 | |
| 
 | |
| int
 | |
| main(int argc, const char **argv)
 | |
| {
 | |
|    FILE *fp;
 | |
|    png_uint_32 default_flags[4/*valid,unknown{before,after}*/];
 | |
|    int strict = 0, default_tests = 0;
 | |
|    const char *count_argv = "default=save";
 | |
|    const char *touch_file = NULL;
 | |
|    display d;
 | |
| 
 | |
|    init_display(&d, argv[0]);
 | |
| 
 | |
|    while (++argv, --argc > 0)
 | |
|    {
 | |
|       if (strcmp(*argv, "--strict") == 0)
 | |
|          strict = 1;
 | |
| 
 | |
|       else if (strcmp(*argv, "--default") == 0)
 | |
|          default_tests = 1;
 | |
| 
 | |
|       else if (strcmp(*argv, "--touch") == 0)
 | |
|       {
 | |
|          if (argc > 1)
 | |
|             touch_file = *++argv, --argc;
 | |
| 
 | |
|          else
 | |
|             usage(d.program, "--touch: missing file name");
 | |
|       }
 | |
| 
 | |
|       else
 | |
|          break;
 | |
|    }
 | |
| 
 | |
|    /* A file name is required, but there should be no other arguments if
 | |
|     * --default was specified.
 | |
|     */
 | |
|    if (argc <= 0)
 | |
|       usage(d.program, "missing test file");
 | |
| 
 | |
|    /* GCC BUG: if (default_tests && argc != 1) triggers some weird GCC argc
 | |
|     * optimization which causes warnings with -Wstrict-overflow!
 | |
|     */
 | |
|    else if (default_tests) if (argc != 1)
 | |
|       usage(d.program, "extra arguments");
 | |
| 
 | |
| #  ifndef PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
 | |
|       fprintf(stderr, "%s: warning: no 'save' support so arguments ignored\n",
 | |
|          d.program);
 | |
| #  endif
 | |
| 
 | |
|    /* The name of the test file is the last argument; remove it. */
 | |
|    d.file = argv[--argc];
 | |
| 
 | |
|    fp = fopen(d.file, "rb");
 | |
|    if (fp == NULL)
 | |
|    {
 | |
|       perror(d.file);
 | |
|       exit(2);
 | |
|    }
 | |
| 
 | |
|    /* First find all the chunks, known and unknown, in the test file, a failure
 | |
|     * here aborts the whole test.
 | |
|     */
 | |
|    if (check(fp, 1, &count_argv, default_flags, &d) !=
 | |
|       PNG_HANDLE_CHUNK_ALWAYS)
 | |
|    {
 | |
|       fprintf(stderr, "%s: %s: internal error\n", d.program, d.file);
 | |
|       exit(3);
 | |
|    }
 | |
| 
 | |
|    /* Now find what the various supplied options cause to change: */
 | |
|    if (!default_tests)
 | |
|    {
 | |
|       d.test = cmd; /* acts as a flag to say exit, do not longjmp */
 | |
|       perform_one_test(fp, argc, argv, default_flags, &d);
 | |
|       d.test = init;
 | |
|    }
 | |
| 
 | |
|    else
 | |
|    {
 | |
|       const char **test = standard_tests;
 | |
| 
 | |
|       /* Set the exit_test pointer here so we can continue after a libpng error.
 | |
|        * NOTE: this leaks memory because the png_struct data from the failing
 | |
|        * test is never freed.
 | |
|        */
 | |
|       while (*test)
 | |
|       {
 | |
|          const char *this_test = *test++;
 | |
|          const char **next = test;
 | |
|          int count = display_rc(&d, strict), new_count;
 | |
|          const char *result;
 | |
|          int arg_count = 0;
 | |
| 
 | |
|          while (*next) ++next, ++arg_count;
 | |
| 
 | |
|          perform_one_test_safe(fp, arg_count, test, default_flags, &d,
 | |
|             this_test);
 | |
| 
 | |
|          new_count = display_rc(&d, strict);
 | |
| 
 | |
|          if (new_count == count)
 | |
|             result = "PASS";
 | |
| 
 | |
|          else
 | |
|             result = "FAIL";
 | |
| 
 | |
|          printf("%s: %s %s\n", result, d.program, this_test);
 | |
| 
 | |
|          test = next+1;
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    fclose(fp);
 | |
| 
 | |
|    if (display_rc(&d, strict) == 0)
 | |
|    {
 | |
|       /* Success, touch the success file if appropriate */
 | |
|       if (touch_file != NULL)
 | |
|       {
 | |
|          FILE *fsuccess = fopen(touch_file, "wt");
 | |
| 
 | |
|          if (fsuccess != NULL)
 | |
|          {
 | |
|             int err = 0;
 | |
|             fprintf(fsuccess, "PNG unknown tests succeeded\n");
 | |
|             fflush(fsuccess);
 | |
|             err = ferror(fsuccess);
 | |
| 
 | |
|             if (fclose(fsuccess) || err)
 | |
|             {
 | |
|                fprintf(stderr, "%s: write failed\n", touch_file);
 | |
|                exit(1);
 | |
|             }
 | |
|          }
 | |
| 
 | |
|          else
 | |
|          {
 | |
|             fprintf(stderr, "%s: open failed\n", touch_file);
 | |
|             exit(1);
 | |
|          }
 | |
|       }
 | |
| 
 | |
|       return 0;
 | |
|    }
 | |
| 
 | |
|    return 1;
 | |
| }
 | |
| 
 | |
| #else
 | |
| int
 | |
| main(void)
 | |
| {
 | |
|    fprintf(stderr,
 | |
|    " test ignored because libpng was not built with unknown chunk support\n");
 | |
|    return 0;
 | |
| }
 | |
| #endif
 | 
