/*

 * 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.4 [July 7, 2011]

 * Copyright (c) 1998-2011 Glenn Randers-Pehrson

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

#include "png.h"

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

#ifndef PNG_ZBUF_SIZE

#  define PNG_ZBUF_SIZE 8192

#endif

#  include <stdio.h>

#  include <stdlib.h>

#  include <string.h>

#  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

/* The code uses memcmp and memcpy on large objects (typically row pointers) so

 * it is necessary to do soemthing special on certain architectures, note that

 * the actual support for this was effectively removed in 1.4, so only the

 * memory remains in this program:

 */

#define CVT_PTR(ptr)         (ptr)

#define CVT_PTR_NOCHECK(ptr) (ptr)

#define png_memcmp  memcmp

#define png_memcpy  memcpy

#define png_memset  memset

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

int test_one_file PNGARG((PNG_CONST char *inname, PNG_CONST char *outname));

#ifdef __TURBOC__

#include <mem.h>

#endif

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

/*  #define STDERR stderr  */

#define STDERR stdout   /* For DOS */

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

void PNGCBAPI

read_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass);

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

}

void PNGCBAPI

write_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass);

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

}

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

void PNGCBAPI

count_filters(png_structp png_ptr, png_row_infop row_info, png_bytep data);

void PNGCBAPI

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;

void PNGCBAPI

count_zero_samples(png_structp png_ptr, png_row_infop row_info, png_bytep data);

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

static int wrote_question = 0;

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

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

}

#endif

#ifndef USE_FAR_KEYWORD

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

}

#else

/* This is the model-independent version. Since the standard I/O library

   can't handle far buffers in the medium and small models, we have to copy

   the data.

*/

#define NEAR_BUF_SIZE 1024

#define MIN(a,b) (a <= b ? a : b)

static void PNGCBAPI

pngtest_read_data(png_structp png_ptr, png_bytep data, png_size_t length)

{

   png_size_t check;

   png_byte *n_data;

   png_FILE_p io_ptr;

   /* Check if data really is near. If so, use usual code. */

   n_data = (png_byte *)CVT_PTR_NOCHECK(data);

   io_ptr = (png_FILE_p)CVT_PTR(png_get_io_ptr(png_ptr));

   if ((png_bytep)n_data == data)

   {

      check = fread(n_data, 1, length, io_ptr);

   }

   else

   {

      png_byte buf[NEAR_BUF_SIZE];

      png_size_t read, remaining, err;

      check = 0;

      remaining = length;

      do

      {

         read = MIN(NEAR_BUF_SIZE, remaining);

         err = fread(buf, 1, 1, io_ptr);

         png_memcpy(data, buf, read); /* Copy far buffer to near buffer */

         if (err != read)

            break;

         else

            check += err;

         data += read;

         remaining -= read;

      }

      while (remaining != 0);

   }

   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

}

#endif /* USE_FAR_KEYWORD */

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

 */

#ifndef USE_FAR_KEYWORD

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

}

#else

/* This is the model-independent version. Since the standard I/O library

   can't handle far buffers in the medium and small models, we have to copy

   the data.

*/

#define NEAR_BUF_SIZE 1024

#define MIN(a,b) (a <= b ? a : b)

static void PNGCBAPI

pngtest_write_data(png_structp png_ptr, png_bytep data, png_size_t length)

{

   png_size_t check;

   png_byte *near_data;  /* Needs to be "png_byte *" instead of "png_bytep" */

   png_FILE_p io_ptr;

   /* Check if data really is near. If so, use usual code. */

   near_data = (png_byte *)CVT_PTR_NOCHECK(data);

   io_ptr = (png_FILE_p)CVT_PTR(png_get_io_ptr(png_ptr));