/*

 * 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:

 *

 * (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 "png.h"

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

#ifndef PNG_ZBUF_SIZE

#  define PNG_ZBUF_SIZE 8192

#endif

#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

/* 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";

#endif

static int verbose = 0;

/* 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

/* 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;

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");

}

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");

}

#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED

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

 * but merely examine the row filters.  We set this to 256 rather than

 * 5 in case illegal filter values are present.)

 */

static png_uint_32 filters_used[256];

count_filters(png_structp png_ptr, png_row_infop row_info, png_bytep data)

{

   if (png_ptr != NULL && row_info != NULL)

      ++filters_used[*(data - 1)];

}

#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;

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;

       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++;

          }

       }

    }

}

#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 */

   }

      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

}

/* 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.

 */

static void PNGCBAPI

pngtest_warning(png_structp png_ptr, png_const_charp message)

{

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

}

/* This is the default error handling function.  Note that replacements for

 * this function MUST NOT RETURN, or the program will likely crash.  This

 * function is used by default, or if the program supplies NULL for the

 * error function pointer in png_set_error_fn().

 */

static void PNGCBAPI

pngtest_error(png_structp png_ptr, png_const_charp message)

{

   pngtest_warning(png_ptr, message);

   /* We can return because png_error calls the default handler, which is

    * actually OK in this case.

    */

}

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

/* START of code to validate memory allocation and deallocation */

#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG

/* Allocate memory.  For reasonable files, size should never exceed

 * 64K.  However, zlib may allocate more then 64K if you don't tell