/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/* pngtest.c - a simple test program to test libpng

 *

 * This file is available under and governed by the GNU General Public

 * License version 2 only, as published by the Free Software Foundation.

 * However, the following notice accompanied the original version of this

 * file and, per its terms, should not be removed:

 *

 * Last changed in libpng 1.5.25 [December 3, 2015]

 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)

 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)

 *

 * This code is released under the libpng license.

 * For conditions of distribution and use, see the disclaimer

 * and license in png.h

 *

 * This program reads in a PNG image, writes it out again, and then

 * compares the two files.  If the files are identical, this shows that

 * the basic chunk handling, filtering, and (de)compression code is working

 * properly.  It does not currently test all of the transforms, although

 * it probably should.

 *

 * The program will report "FAIL" in certain legitimate cases:

 * 1) when the compression level or filter selection method is changed.

 * 2) when the maximum IDAT size (PNG_ZBUF_SIZE in pngconf.h) is not 8192.

 * 3) unknown unsafe-to-copy ancillary chunks or unknown critical chunks

 *    exist in the input file.

 * 4) others not listed here...

 * In these cases, it is best to check with another tool such as "pngcheck"

 * to see what the differences between the two files are.

 *

 * If a filename is given on the command-line, then this file is used

 * for the input, rather than the default "pngtest.png".  This allows

 * testing a wide variety of files easily.  You can also test a number

 * of files at once by typing "pngtest -m file1.png file2.png ..."

 */

#define _POSIX_SOURCE 1

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

/* Defined so I can write to a file on gui/windowing platforms */

/*  #define STDERR stderr  */

#define STDERR stdout   /* For DOS */

#include "png.h"

/* Known chunks that exist in pngtest.png must be supported or pngtest will fail

 * simply as a result of re-ordering them.  This may be fixed in 1.7

 *

 * pngtest allocates a single row buffer for each row and overwrites it,

 * therefore if the write side doesn't support the writing of interlaced images

 * nothing can be done for an interlaced image (and the code below will fail

 * horribly trying to write extra data after writing garbage).

 */

#if defined PNG_READ_SUPPORTED && /* else nothing can be done */\

   defined PNG_READ_bKGD_SUPPORTED &&\

   defined PNG_READ_cHRM_SUPPORTED &&\

   defined PNG_READ_gAMA_SUPPORTED &&\

   defined PNG_READ_oFFs_SUPPORTED &&\

   defined PNG_READ_pCAL_SUPPORTED &&\

   defined PNG_READ_pHYs_SUPPORTED &&\

   defined PNG_READ_sBIT_SUPPORTED &&\

   defined PNG_READ_sCAL_SUPPORTED &&\

   defined PNG_READ_sRGB_SUPPORTED &&\

   defined PNG_READ_sPLT_SUPPORTED &&\

   defined PNG_READ_tEXt_SUPPORTED &&\

   defined PNG_READ_tIME_SUPPORTED &&\

   defined PNG_READ_zTXt_SUPPORTED &&\

   (defined PNG_WRITE_INTERLACING_SUPPORTED || PNG_LIBPNG_VER >= 10700)

#ifdef PNG_ZLIB_HEADER

#  include PNG_ZLIB_HEADER /* defined by pnglibconf.h from 1.7 */

#else

#  include "zlib.h"

#endif

/* Copied from pngpriv.h but only used in error messages below. */

#ifndef PNG_ZBUF_SIZE

#  define PNG_ZBUF_SIZE 8192

#endif

#define FCLOSE(file) fclose(file)

#ifndef PNG_STDIO_SUPPORTED

typedef FILE                * png_FILE_p;

#endif

/* Makes pngtest verbose so we can find problems. */

#ifndef PNG_DEBUG

#  define PNG_DEBUG 0

#endif

#if PNG_DEBUG > 1

#  define pngtest_debug(m)        ((void)fprintf(stderr, m "\n"))

#  define pngtest_debug1(m,p1)    ((void)fprintf(stderr, m "\n", p1))

#  define pngtest_debug2(m,p1,p2) ((void)fprintf(stderr, m "\n", p1, p2))

#else

#  define pngtest_debug(m)        ((void)0)

#  define pngtest_debug1(m,p1)    ((void)0)

#  define pngtest_debug2(m,p1,p2) ((void)0)

#endif

#if !PNG_DEBUG

#  define SINGLE_ROWBUF_ALLOC  /* Makes buffer overruns easier to nail */

#endif

#ifndef PNG_UNUSED

#  define PNG_UNUSED(param) (void)param;

#endif

/* Turn on CPU timing

#define PNGTEST_TIMING

*/

#ifndef PNG_FLOATING_POINT_SUPPORTED

#undef PNGTEST_TIMING

#endif

#ifdef PNGTEST_TIMING

static float t_start, t_stop, t_decode, t_encode, t_misc;

#include <time.h>

#endif

#ifdef PNG_TIME_RFC1123_SUPPORTED

#define PNG_tIME_STRING_LENGTH 29

static int tIME_chunk_present = 0;

static char tIME_string[PNG_tIME_STRING_LENGTH] = "tIME chunk is not present";

#if PNG_LIBPNG_VER < 10619

#define png_convert_to_rfc1123_buffer(ts, t) tIME_to_str(read_ptr, ts, t)

static int

tIME_to_str(png_structp png_ptr, png_charp ts, png_const_timep t)

{

    png_const_charp str = png_convert_to_rfc1123(png_ptr, t);

    if (str == NULL)

        return 0;

    strcpy(ts, str);

    return 1;

}

#endif /* older libpng */

#endif

static int verbose = 0;

static int strict = 0;

static int relaxed = 0;

static int unsupported_chunks = 0; /* chunk unsupported by libpng in input */

static int error_count = 0; /* count calls to png_error */

static int warning_count = 0; /* count calls to png_warning */

/* Define png_jmpbuf() in case we are using a pre-1.0.6 version of libpng */

#ifndef png_jmpbuf

#  define png_jmpbuf(png_ptr) png_ptr->jmpbuf

#endif

/* Defines for unknown chunk handling if required. */

#ifndef PNG_HANDLE_CHUNK_ALWAYS

#  define PNG_HANDLE_CHUNK_ALWAYS       3

#endif

#ifndef PNG_HANDLE_CHUNK_IF_SAFE

#  define PNG_HANDLE_CHUNK_IF_SAFE      2

#endif

/* Utility to save typing/errors, the argument must be a name */

#define MEMZERO(var) ((void)memset(&var, 0, sizeof var))

/* Example of using row callbacks to make a simple progress meter */

static int status_pass = 1;

static int status_dots_requested = 0;

static int status_dots = 1;

static void PNGCBAPI

read_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)

{

   if (png_ptr == NULL || row_number > PNG_UINT_31_MAX)

      return;

   if (status_pass != pass)

   {

      fprintf(stdout, "\n Pass %d: ", pass);

      status_pass = pass;

      status_dots = 31;

   }

   status_dots--;

   if (status_dots == 0)

   {

      fprintf(stdout, "\n         ");

      status_dots=30;

   }

   fprintf(stdout, "r");

}

#ifdef PNG_WRITE_SUPPORTED

static void PNGCBAPI

write_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)

{

   if (png_ptr == NULL || row_number > PNG_UINT_31_MAX || pass > 7)

      return;

   fprintf(stdout, "w");

}

#endif

#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED

/* Example of using a user transform callback (doesn't do anything at present).

 */

static void PNGCBAPI

read_user_callback(png_structp png_ptr, png_row_infop row_info, png_bytep data)

{

   PNG_UNUSED(png_ptr)

   PNG_UNUSED(row_info)

   PNG_UNUSED(data)

}

#endif

#ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED

/* Example of using user transform callback (we don't transform anything,

 * but merely count the zero samples)

 */

static png_uint_32 zero_samples;

static void PNGCBAPI

count_zero_samples(png_structp png_ptr, png_row_infop row_info, png_bytep data)

{

   png_bytep dp = data;

   if (png_ptr == NULL)

      return;

   /* Contents of row_info:

    *  png_uint_32 width      width of row

    *  png_uint_32 rowbytes   number of bytes in row

    *  png_byte color_type    color type of pixels

    *  png_byte bit_depth     bit depth of samples

    *  png_byte channels      number of channels (1-4)

    *  png_byte pixel_depth   bits per pixel (depth*channels)

    */

    /* Counts the number of zero samples (or zero pixels if color_type is 3 */

    if (row_info->color_type == 0 || row_info->color_type == 3)

    {

       int pos = 0;

       png_uint_32 n, nstop;

       for (n = 0, nstop=row_info->width; n<nstop; n++)

       {

          if (row_info->bit_depth == 1)

          {

             if (((*dp << pos++ ) & 0x80) == 0)

                zero_samples++;

             if (pos == 8)

             {

                pos = 0;

                dp++;

             }

          }

          if (row_info->bit_depth == 2)

          {

             if (((*dp << (pos+=2)) & 0xc0) == 0)

                zero_samples++;

             if (pos == 8)

             {

                pos = 0;

                dp++;

             }

          }

          if (row_info->bit_depth == 4)

          {

             if (((*dp << (pos+=4)) & 0xf0) == 0)

                zero_samples++;

             if (pos == 8)

             {

                pos = 0;

                dp++;

             }

          }

          if (row_info->bit_depth == 8)

             if (*dp++ == 0)

                zero_samples++;

          if (row_info->bit_depth == 16)

          {

             if ((*dp | *(dp+1)) == 0)

                zero_samples++;

             dp+=2;

          }

       }

    }

    else /* Other color types */

    {

       png_uint_32 n, nstop;

       int channel;

       int color_channels = row_info->channels;

       if (row_info->color_type > 3)

          color_channels--;

       for (n = 0, nstop=row_info->width; n<nstop; n++)

       {

          for (channel = 0; channel < color_channels; channel++)

          {

             if (row_info->bit_depth == 8)

                if (*dp++ == 0)

                   zero_samples++;

             if (row_info->bit_depth == 16)

             {

                if ((*dp | *(dp+1)) == 0)

                   zero_samples++;

                dp+=2;

             }

          }

          if (row_info->color_type > 3)

          {

             dp++;

             if (row_info->bit_depth == 16)

                dp++;

          }

       }

    }

}

#endif /* WRITE_USER_TRANSFORM */

#ifndef PNG_STDIO_SUPPORTED

/* START of code to validate stdio-free compilation */

/* These copies of the default read/write functions come from pngrio.c and

 * pngwio.c.  They allow "don't include stdio" testing of the library.

 * This is the function that does the actual reading of data.  If you are

 * not reading from a standard C stream, you should create a replacement

 * read_data function and use it at run time with png_set_read_fn(), rather

 * than changing the library.

 */

#ifdef PNG_IO_STATE_SUPPORTED

void

pngtest_check_io_state(png_structp png_ptr, png_size_t data_length,

   png_uint_32 io_op);

void

pngtest_check_io_state(png_structp png_ptr, png_size_t data_length,

   png_uint_32 io_op)

{

   png_uint_32 io_state = png_get_io_state(png_ptr);

   int err = 0;

   /* Check if the current operation (reading / writing) is as expected. */

   if ((io_state & PNG_IO_MASK_OP) != io_op)

      png_error(png_ptr, "Incorrect operation in I/O state");

   /* Check if the buffer size specific to the current location

    * (file signature / header / data / crc) is as expected.

    */

   switch (io_state & PNG_IO_MASK_LOC)

   {

   case PNG_IO_SIGNATURE:

      if (data_length > 8)

         err = 1;

      break;

   case PNG_IO_CHUNK_HDR:

      if (data_length != 8)

         err = 1;

      break;

   case PNG_IO_CHUNK_DATA:

      break;  /* no restrictions here */

   case PNG_IO_CHUNK_CRC:

      if (data_length != 4)

         err = 1;

      break;

   default:

      err = 1;  /* uninitialized */

   }

   if (err != 0)

      png_error(png_ptr, "Bad I/O state or buffer size");

}

#endif

static void PNGCBAPI

pngtest_read_data(png_structp png_ptr, png_bytep data, png_size_t length)

{

   png_size_t check = 0;

   png_voidp io_ptr;

   /* fread() returns 0 on error, so it is OK to store this in a png_size_t

    * instead of an int, which is what fread() actually returns.

    */

   io_ptr = png_get_io_ptr(png_ptr);

   if (io_ptr != NULL)

   {

      check = fread(data, 1, length, (png_FILE_p)io_ptr);

   }

   if (check != length)

   {

      png_error(png_ptr, "Read Error");

   }

#ifdef PNG_IO_STATE_SUPPORTED

   pngtest_check_io_state(png_ptr, length, PNG_IO_READING);

#endif

}

#ifdef PNG_WRITE_FLUSH_SUPPORTED

static void PNGCBAPI

pngtest_flush(png_structp png_ptr)

{

   /* Do nothing; fflush() is said to be just a waste of energy. */

   PNG_UNUSED(png_ptr)   /* Stifle compiler warning */

}

#endif

/* This is the function that does the actual writing of data.  If you are

 * not writing to a standard C stream, you should create a replacement

 * write_data function and use it at run time with png_set_write_fn(), rather

 * than changing the library.

 */

static void PNGCBAPI

pngtest_write_data(png_structp png_ptr, png_bytep data, png_size_t length)

{

   png_size_t check;

   check = fwrite(data, 1, length, (png_FILE_p)png_get_io_ptr(png_ptr));

   if (check != length)

   {

      png_error(png_ptr, "Write Error");

   }

#ifdef PNG_IO_STATE_SUPPORTED

   pngtest_check_io_state(png_ptr, length, PNG_IO_WRITING);

#endif

}

#endif /* !STDIO */

/* This function is called when there is a warning, but the library thinks

 * it can continue anyway.  Replacement functions don't have to do anything

 * here if you don't want to.  In the default configuration, png_ptr is

 * not used, but it is passed in case it may be useful.

 */

typedef struct

{

   PNG_CONST char *file_name;

}  pngtest_error_parameters;

static void PNGCBAPI

pngtest_warning(png_structp png_ptr, png_const_charp message)

{

   PNG_CONST char *name = "UNKNOWN (ERROR!)";

   pngtest_error_parameters *test =

      (pngtest_error_parameters*)png_get_error_ptr(png_ptr);

   ++warning_count;