pngtest.c 48.6 KB
Newer Older
A
Andreas Dilger 已提交
1

G
Guy Schalnat 已提交
2
/* pngtest.c - a simple test program to test libpng
3
 *
4
 * Last changed in libpng 1.5.0 [(PENDING RELEASE)]
5
 * Copyright (c) 1998-2010 Glenn Randers-Pehrson
6 7
 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
8
 *
9
 * This code is released under the libpng license.
10
 * For conditions of distribution and use, see the disclaimer
11
 * and license in png.h
12
 *
13 14 15 16 17 18
 * 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.
 *
19
 * The program will report "FAIL" in certain legitimate cases:
20
 * 1) when the compression level or filter selection method is changed.
21
 * 2) when the maximum IDAT size (PNG_ZBUF_SIZE in pngconf.h) is not 8192.
22 23
 * 3) unknown unsafe-to-copy ancillary chunks or unknown critical chunks
 *    exist in the input file.
24 25
 * 4) others not listed here...
 * In these cases, it is best to check with another tool such as "pngcheck"
26
 * to see what the differences between the two files are.
27 28 29
 *
 * 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
30 31
 * 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 ..."
32
 */
G
Guy Schalnat 已提交
33

G
[devel]  
Glenn Randers-Pehrson 已提交
34
#include "zlib.h"
35
#include "png.h"
36 37 38 39
/* Copied from pngpriv.h but only used in error messages below. */
#ifndef PNG_ZBUF_SIZE
#  define PNG_ZBUF_SIZE 8192
#endif
40 41
#  include <stdio.h>
#  include <stdlib.h>
42
#  include <string.h>
43
#  define FCLOSE(file) fclose(file)
44

45
#ifndef PNG_STDIO_SUPPORTED
46
typedef FILE                * png_FILE_p;
47 48
#endif

49
/* Makes pngtest verbose so we can find problems. */
A
Andreas Dilger 已提交
50
#ifndef PNG_DEBUG
51 52 53
#  define PNG_DEBUG 0
#endif

54 55 56 57 58 59 60 61 62
#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
63

64
#if !PNG_DEBUG
65
#  define SINGLE_ROWBUF_ALLOC  /* Makes buffer overruns easier to nail */
66
#endif
A
Andreas Dilger 已提交
67

G
[devel]  
Glenn Randers-Pehrson 已提交
68 69 70 71 72 73 74 75 76 77 78
/* 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

79 80 81 82
/* Turn on CPU timing
#define PNGTEST_TIMING
*/

83
#ifndef PNG_FLOATING_POINT_SUPPORTED
84 85 86
#undef PNGTEST_TIMING
#endif

87 88 89 90 91
#ifdef PNGTEST_TIMING
static float t_start, t_stop, t_decode, t_encode, t_misc;
#include <time.h>
#endif

92
#ifdef PNG_TIME_RFC1123_SUPPORTED
93
#define PNG_tIME_STRING_LENGTH 29
94
static int tIME_chunk_present = 0;
95
static char tIME_string[PNG_tIME_STRING_LENGTH] = "tIME chunk is not present";
96 97
#endif

98 99
static int verbose = 0;

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

G
Guy Schalnat 已提交
102 103 104 105
#ifdef __TURBOC__
#include <mem.h>
#endif

106
/* Defined so I can write to a file on gui/windowing platforms */
G
Guy Schalnat 已提交
107
/*  #define STDERR stderr  */
108
#define STDERR stdout   /* For DOS */
G
Guy Schalnat 已提交
109

110 111 112 113 114
/* 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

115
/* Example of using row callbacks to make a simple progress meter */
116 117 118
static int status_pass = 1;
static int status_dots_requested = 0;
static int status_dots = 1;
119

120
void PNGCBAPI
121
read_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass);
122
void PNGCBAPI
123
read_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)
124
{
125 126
   if (png_ptr == NULL || row_number > PNG_UINT_31_MAX)
      return;
127

128 129 130 131 132 133
   if (status_pass != pass)
   {
      fprintf(stdout, "\n Pass %d: ", pass);
      status_pass = pass;
      status_dots = 31;
   }
134

135
   status_dots--;
136

137 138 139 140 141
   if (status_dots == 0)
   {
      fprintf(stdout, "\n         ");
      status_dots=30;
   }
142

143
   fprintf(stdout, "r");
144
}
145

146
void PNGCBAPI
147
write_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass);
148
void PNGCBAPI
149
write_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)
150
{
151 152
   if (png_ptr == NULL || row_number > PNG_UINT_31_MAX || pass > 7)
      return;
153

154
   fprintf(stdout, "w");
155 156 157
}


158
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
159
/* Example of using user transform callback (we don't transform anything,
160 161 162
 * but merely examine the row filters.  We set this to 256 rather than
 * 5 in case illegal filter values are present.)
 */
163
static png_uint_32 filters_used[256];
164
void PNGCBAPI
165
count_filters(png_structp png_ptr, png_row_infop row_info, png_bytep data);
166
void PNGCBAPI
167 168
count_filters(png_structp png_ptr, png_row_infop row_info, png_bytep data)
{
169
   if (png_ptr != NULL && row_info != NULL)
170
      ++filters_used[*(data - 1)];
171 172 173
}
#endif

174
#ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
175
/* Example of using user transform callback (we don't transform anything,
176 177
 * but merely count the zero samples)
 */
178

179
static png_uint_32 zero_samples;
180

181
void PNGCBAPI
182
count_zero_samples(png_structp png_ptr, png_row_infop row_info, png_bytep data);
183
void PNGCBAPI
184
count_zero_samples(png_structp png_ptr, png_row_infop row_info, png_bytep data)
185 186
{
   png_bytep dp = data;
187 188
   if (png_ptr == NULL)
      return;
189

190
   /* Contents of row_info:
191 192 193 194 195 196 197 198
    *  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)
    */

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

201
    if (row_info->color_type == 0 || row_info->color_type == 3)
202
    {
203
       int pos = 0;
204
       png_uint_32 n, nstop;
205

206
       for (n = 0, nstop=row_info->width; n<nstop; n++)
207
       {
208
          if (row_info->bit_depth == 1)
209
          {
210 211
             if (((*dp << pos++ ) & 0x80) == 0)
                zero_samples++;
212

213
             if (pos == 8)
214
             {
215
                pos = 0;
216 217
                dp++;
             }
218
          }
219

220
          if (row_info->bit_depth == 2)
221
          {
222 223
             if (((*dp << (pos+=2)) & 0xc0) == 0)
                zero_samples++;
224

225
             if (pos == 8)
226
             {
227
                pos = 0;
228 229
                dp++;
             }
230
          }
231

232
          if (row_info->bit_depth == 4)
233
          {
234 235
             if (((*dp << (pos+=4)) & 0xf0) == 0)
                zero_samples++;
236

237
             if (pos == 8)
238
             {
239
                pos = 0;
240 241
                dp++;
             }
242
          }
243

244
          if (row_info->bit_depth == 8)
245 246
             if (*dp++ == 0)
                zero_samples++;
247

248
          if (row_info->bit_depth == 16)
249
          {
250 251
             if ((*dp | *(dp+1)) == 0)
                zero_samples++;
252 253 254 255
             dp+=2;
          }
       }
    }
256
    else /* Other color types */
257
    {
258
       png_uint_32 n, nstop;
259 260
       int channel;
       int color_channels = row_info->channels;
261
       if (row_info->color_type > 3)color_channels--;
262

263
       for (n = 0, nstop=row_info->width; n<nstop; n++)
264 265 266
       {
          for (channel = 0; channel < color_channels; channel++)
          {
267
             if (row_info->bit_depth == 8)
268 269
                if (*dp++ == 0)
                   zero_samples++;
270

271
             if (row_info->bit_depth == 16)
272
             {
273 274
                if ((*dp | *(dp+1)) == 0)
                   zero_samples++;
275

276 277 278
                dp+=2;
             }
          }
279
          if (row_info->color_type > 3)
280 281
          {
             dp++;
282 283
             if (row_info->bit_depth == 16)
                dp++;
284 285 286 287
          }
       }
    }
}
288
#endif /* PNG_WRITE_USER_TRANSFORM_SUPPORTED */
289

290
static int wrote_question = 0;
291

292
#ifndef PNG_STDIO_SUPPORTED
293
/* START of code to validate stdio-free compilation */
294 295 296 297 298 299 300
/* 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.
 */
301

302
#ifndef USE_FAR_KEYWORD
303
static void PNGCBAPI
304
pngtest_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
305
{
306 307
   png_size_t check = 0;
   png_voidp io_ptr;
308 309 310 311

   /* 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.
    */
312 313 314 315 316
   io_ptr = png_get_io_ptr(png_ptr);
   if (io_ptr != NULL)
   {
      check = fread(data, 1, length, (png_FILE_p)io_ptr);
   }
317

318 319
   if (check != length)
   {
320
      png_error(png_ptr, "Read Error!");
321 322
   }
}
G
Guy Schalnat 已提交
323
#else
324
/* This is the model-independent version. Since the standard I/O library
325 326 327
   can't handle far buffers in the medium and small models, we have to copy
   the data.
*/
328

329 330
#define NEAR_BUF_SIZE 1024
#define MIN(a,b) (a <= b ? a : b)
331

332
static void PNGCBAPI
333
pngtest_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
334
{
335
   png_size_t check;
336
   png_byte *n_data;
337
   png_FILE_p io_ptr;
338 339 340

   /* Check if data really is near. If so, use usual code. */
   n_data = (png_byte *)CVT_PTR_NOCHECK(data);
341
   io_ptr = (png_FILE_p)CVT_PTR(png_get_io_ptr(png_ptr));
342 343
   if ((png_bytep)n_data == data)
   {
344
      check = fread(n_data, 1, length, io_ptr);
345 346 347 348 349 350 351
   }
   else
   {
      png_byte buf[NEAR_BUF_SIZE];
      png_size_t read, remaining, err;
      check = 0;
      remaining = length;
352

353 354 355
      do
      {
         read = MIN(NEAR_BUF_SIZE, remaining);
356
         err = fread(buf, 1, 1, io_ptr);
357
         png_memcpy(data, buf, read); /* Copy far buffer to near buffer */
358
         if (err != read)
359 360 361 362 363 364 365 366
            break;
         else
            check += err;
         data += read;
         remaining -= read;
      }
      while (remaining != 0);
   }
367

368 369 370
   if (check != length)
      png_error(png_ptr, "read Error");
}
371
#endif /* USE_FAR_KEYWORD */
G
Guy Schalnat 已提交
372

373
#ifdef PNG_WRITE_FLUSH_SUPPORTED
374
static void PNGCBAPI
375
pngtest_flush(png_structp png_ptr)
376
{
377
   /* Do nothing; fflush() is said to be just a waste of energy. */
378
   png_ptr = png_ptr;  /* Stifle compiler warning */
379 380
}
#endif
G
Guy Schalnat 已提交
381

382
/* This is the function that does the actual writing of data.  If you are
383 384 385 386
 * 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.
 */
387
#ifndef USE_FAR_KEYWORD
388
static void PNGCBAPI
389
pngtest_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
390
{
391
   png_size_t check;
392

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

395 396 397 398 399 400
   if (check != length)
   {
      png_error(png_ptr, "Write Error");
   }
}
#else
401
/* This is the model-independent version. Since the standard I/O library
402 403 404 405 406 407 408
   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)

409
static void PNGCBAPI
410
pngtest_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
411
{
412
   png_size_t check;
413
   png_byte *near_data;  /* Needs to be "png_byte *" instead of "png_bytep" */
414
   png_FILE_p io_ptr;
415 416 417

   /* Check if data really is near. If so, use usual code. */
   near_data = (png_byte *)CVT_PTR_NOCHECK(data);
418
   io_ptr = (png_FILE_p)CVT_PTR(png_get_io_ptr(png_ptr));
419

420 421
   if ((png_bytep)near_data == data)
   {
422
      check = fwrite(near_data, 1, length, io_ptr);
423
   }
424

425 426 427 428 429 430
   else
   {
      png_byte buf[NEAR_BUF_SIZE];
      png_size_t written, remaining, err;
      check = 0;
      remaining = length;
431

432 433 434
      do
      {
         written = MIN(NEAR_BUF_SIZE, remaining);
435
         png_memcpy(buf, data, written); /* Copy far buffer to near buffer */
436
         err = fwrite(buf, 1, written, io_ptr);
437 438 439 440 441 442 443 444 445
         if (err != written)
            break;
         else
            check += err;
         data += written;
         remaining -= written;
      }
      while (remaining != 0);
   }
446

447 448 449 450 451
   if (check != length)
   {
      png_error(png_ptr, "Write Error");
   }
}
452
#endif /* USE_FAR_KEYWORD */
453 454 455 456 457 458

/* 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.
 */
459
static void PNGCBAPI
460
pngtest_warning(png_structp png_ptr, png_const_charp message)
461 462
{
   PNG_CONST char *name = "UNKNOWN (ERROR!)";
463 464
   char *test;
   test = png_get_error_ptr(png_ptr);
465

466 467
   if (test == NULL)
     fprintf(STDERR, "%s: libpng warning: %s\n", name, message);
468

469 470
   else
     fprintf(STDERR, "%s: libpng warning: %s\n", test, message);
471 472 473 474 475 476 477
}

/* 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().
 */
478
static void PNGCBAPI
479
pngtest_error(png_structp png_ptr, png_const_charp message)
480
{
481
   pngtest_warning(png_ptr, message);
482
   /* We can return because png_error calls the default handler, which is
483 484
    * actually OK in this case.
    */
485
}
486
#endif /* !PNG_STDIO_SUPPORTED */
487 488
/* END of code to validate stdio-free compilation */

489
/* START of code to validate memory allocation and deallocation */
490
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
491 492

/* Allocate memory.  For reasonable files, size should never exceed
493 494 495 496 497 498 499 500
 * 64K.  However, zlib may allocate more then 64K if you don't tell
 * it not to.  See zconf.h and png.h for more information.  zlib does
 * need to allocate exactly 64K, so whatever you call here must
 * have the ability to do that.
 *
 * This piece of code can be compiled to validate max 64K allocations
 * by setting MAXSEG_64K in zlib zconf.h *or* PNG_MAX_MALLOC_64K.
 */
501 502
typedef struct memory_information
{
503
   png_alloc_size_t          size;
504
   png_voidp                 pointer;
505 506 507 508 509 510 511
   struct memory_information FAR *next;
} memory_information;
typedef memory_information FAR *memory_infop;

static memory_infop pinformation = NULL;
static int current_allocation = 0;
static int maximum_allocation = 0;
512 513
static int total_allocation = 0;
static int num_allocations = 0;
514

515 516 517
png_voidp PNGCBAPI png_debug_malloc PNGARG((png_structp png_ptr,
    png_alloc_size_t size));
void PNGCBAPI png_debug_free PNGARG((png_structp png_ptr, png_voidp ptr));
518 519

png_voidp
520
PNGCBAPI png_debug_malloc(png_structp png_ptr, png_alloc_size_t size)
521
{
522

523
   /* png_malloc has already tested for NULL; png_create_struct calls
524 525
    * png_debug_malloc directly, with png_ptr == NULL which is OK
    */
526

527
   if (size == 0)
528
      return (NULL);
529 530 531 532

   /* This calls the library allocator twice, once to get the requested
      buffer and once to get a new free list entry. */
   {
533
      /* Disable malloc_fn and free_fn */
534
      memory_infop pinfo;
535
      png_set_mem_fn(png_ptr, NULL, NULL, NULL);
536 537
      pinfo = (memory_infop)png_malloc(png_ptr,
         png_sizeof(*pinfo));
538 539
      pinfo->size = size;
      current_allocation += size;
540 541
      total_allocation += size;
      num_allocations ++;
542

543 544
      if (current_allocation > maximum_allocation)
         maximum_allocation = current_allocation;
545

546
      pinfo->pointer = png_malloc(png_ptr, size);
547
      /* Restore malloc_fn and free_fn */
548

549 550
      png_set_mem_fn(png_ptr,
          NULL, png_debug_malloc, png_debug_free);
551

552 553 554 555
      if (size != 0 && pinfo->pointer == NULL)
      {
         current_allocation -= size;
         total_allocation -= size;
556
         png_error(png_ptr,
557
           "out of memory in pngtest->png_debug_malloc");
558
      }
559

560 561 562 563
      pinfo->next = pinformation;
      pinformation = pinfo;
      /* Make sure the caller isn't assuming zeroed memory. */
      png_memset(pinfo->pointer, 0xdd, pinfo->size);
564

565
      if (verbose)
566
         printf("png_malloc %lu bytes at %p\n", (unsigned long)size,
567
            pinfo->pointer);
568

569
      return (png_voidp)(pinfo->pointer);
570 571 572 573
   }
}

/* Free a pointer.  It is removed from the list at the same time. */
574
void PNGCBAPI
575
png_debug_free(png_structp png_ptr, png_voidp ptr)
576 577
{
   if (png_ptr == NULL)
578
      fprintf(STDERR, "NULL pointer to png_debug_free.\n");
579

580 581
   if (ptr == 0)
   {
582 583 584 585 586 587 588 589 590
#if 0 /* This happens all the time. */
      fprintf(STDERR, "WARNING: freeing NULL pointer\n");
#endif
      return;
   }

   /* Unlink the element from the list. */
   {
      memory_infop FAR *ppinfo = &pinformation;
591

592 593
      for (;;)
      {
594
         memory_infop pinfo = *ppinfo;
595

596 597
         if (pinfo->pointer == ptr)
         {
598 599 600 601 602
            *ppinfo = pinfo->next;
            current_allocation -= pinfo->size;
            if (current_allocation < 0)
               fprintf(STDERR, "Duplicate free of memory\n");
            /* We must free the list element too, but first kill
603
               the memory that is to be freed. */
604
            png_memset(ptr, 0x55, pinfo->size);
605
            png_free_default(png_ptr, pinfo);
606
            pinfo = NULL;
607 608
            break;
         }
609

610 611
         if (pinfo->next == NULL)
         {
612
            fprintf(STDERR, "Pointer %x not found\n", (unsigned int)ptr);
613 614
            break;
         }
615

616 617 618 619 620
         ppinfo = &pinfo->next;
      }
   }

   /* Finally free the data. */
621
   if (verbose)
622
      printf("Freeing %p\n", ptr);
623

624
   png_free_default(png_ptr, ptr);
625
   ptr = NULL;
626
}
627
#endif /* PNG_USER_MEM_SUPPORTED && PNG_DEBUG */
628 629
/* END of code to test memory allocation/deallocation */

630 631

/* Demonstration of user chunk support of the sTER and vpAg chunks */
632
#ifdef PNG_UNKNOWN_CHUNKS_SUPPORTED
633

634
/* (sTER is a public chunk not yet known by libpng.  vpAg is a private
635 636 637 638
chunk used in ImageMagick to store "virtual page" size).  */

static png_uint_32 user_chunk_data[4];

639 640 641 642
    /* 0: sTER mode + 1
     * 1: vpAg width
     * 2: vpAg height
     * 3: vpAg units
643 644
     */

645
static int PNGCBAPI read_user_chunk_callback(png_struct *png_ptr,
646 647
   png_unknown_chunkp chunk)
{
648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669
   png_uint_32
     *my_user_chunk_data;

   /* Return one of the following:
    *    return (-n);  chunk had an error
    *    return (0);  did not recognize
    *    return (n);  success
    *
    * The unknown chunk structure contains the chunk data:
    * png_byte name[5];
    * png_byte *data;
    * png_size_t size;
    *
    * Note that libpng has already taken care of the CRC handling.
    */

   if (chunk->name[0] == 115 && chunk->name[1] ==  84 &&     /* s  T */
       chunk->name[2] ==  69 && chunk->name[3] ==  82)       /* E  R */
      {
         /* Found sTER chunk */
         if (chunk->size != 1)
            return (-1); /* Error return */
670

671 672
         if (chunk->data[0] != 0 && chunk->data[0] != 1)
            return (-1);  /* Invalid mode */
673

674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
         my_user_chunk_data=(png_uint_32 *) png_get_user_chunk_ptr(png_ptr);
         my_user_chunk_data[0]=chunk->data[0]+1;
         return (1);
      }

   if (chunk->name[0] != 118 || chunk->name[1] != 112 ||    /* v  p */
       chunk->name[2] !=  65 || chunk->name[3] != 103)      /* A  g */
      return (0); /* Did not recognize */

   /* Found ImageMagick vpAg chunk */

   if (chunk->size != 9)
      return (-1); /* Error return */

   my_user_chunk_data=(png_uint_32 *) png_get_user_chunk_ptr(png_ptr);

   my_user_chunk_data[1]=png_get_uint_31(png_ptr, chunk->data);
   my_user_chunk_data[2]=png_get_uint_31(png_ptr, chunk->data + 4);
   my_user_chunk_data[3]=(png_uint_32)chunk->data[8];

   return (1);
695 696 697 698 699

}
#endif
/* END of code to demonstrate user chunk support */

700
/* Test one file */
701 702
int
test_one_file(PNG_CONST char *inname, PNG_CONST char *outname)
G
Guy Schalnat 已提交
703
{
704 705
   static png_FILE_p fpin;
   static png_FILE_p fpout;  /* "static" prevents setjmp corruption */
706 707 708 709 710 711 712 713 714 715 716
   png_structp read_ptr;
   png_infop read_info_ptr, end_info_ptr;
#ifdef PNG_WRITE_SUPPORTED
   png_structp write_ptr;
   png_infop write_info_ptr;
   png_infop write_end_info_ptr;
#else
   png_structp write_ptr = NULL;
   png_infop write_info_ptr = NULL;
   png_infop write_end_info_ptr = NULL;
#endif
G
Guy Schalnat 已提交
717
   png_bytep row_buf;
G
Guy Schalnat 已提交
718
   png_uint_32 y;
A
Andreas Dilger 已提交
719 720 721
   png_uint_32 width, height;
   int num_pass, pass;
   int bit_depth, color_type;
722
#ifdef PNG_SETJMP_SUPPORTED
A
Andreas Dilger 已提交
723
#ifdef USE_FAR_KEYWORD
724
   jmp_buf png_jmpbuf;
725
#endif
726
#endif
727

728
   char inbuf[256], outbuf[256];
G
Guy Schalnat 已提交
729

730
   row_buf = NULL;
G
Guy Schalnat 已提交
731

A
Andreas Dilger 已提交
732
   if ((fpin = fopen(inname, "rb")) == NULL)
G
Guy Schalnat 已提交
733 734
   {
      fprintf(STDERR, "Could not find input file %s\n", inname);
735
      return (1);
G
Guy Schalnat 已提交
736 737
   }

A
Andreas Dilger 已提交
738
   if ((fpout = fopen(outname, "wb")) == NULL)
G
Guy Schalnat 已提交
739
   {
G
Guy Schalnat 已提交
740
      fprintf(STDERR, "Could not open output file %s\n", outname);
741
      FCLOSE(fpin);
742
      return (1);
G
Guy Schalnat 已提交
743 744
   }

745
   pngtest_debug("Allocating read and write structures");
746
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
747 748
   read_ptr =
      png_create_read_struct_2(PNG_LIBPNG_VER_STRING, NULL,
749
      NULL, NULL, NULL, png_debug_malloc, png_debug_free);
750
#else
751 752
   read_ptr =
      png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
753
#endif
754
#ifndef PNG_STDIO_SUPPORTED
755 756
   png_set_error_fn(read_ptr, (png_voidp)inname, pngtest_error,
       pngtest_warning);
757
#endif
758

759
#ifdef PNG_UNKNOWN_CHUNKS_SUPPORTED
760 761 762 763 764 765
   user_chunk_data[0] = 0;
   user_chunk_data[1] = 0;
   user_chunk_data[2] = 0;
   user_chunk_data[3] = 0;
   png_set_read_user_chunk_fn(read_ptr, user_chunk_data,
     read_user_chunk_callback);
766

767
#endif
768
#ifdef PNG_WRITE_SUPPORTED
769
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
770 771 772
   write_ptr =
      png_create_write_struct_2(PNG_LIBPNG_VER_STRING, NULL,
      NULL, NULL, NULL, png_debug_malloc, png_debug_free);
773
#else
774 775
   write_ptr =
      png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
776
#endif
777
#ifndef PNG_STDIO_SUPPORTED
778 779
   png_set_error_fn(write_ptr, (png_voidp)inname, pngtest_error,
       pngtest_warning);
780
#endif
781
#endif
782
   pngtest_debug("Allocating read_info, write_info and end_info structures");
A
Andreas Dilger 已提交
783
   read_info_ptr = png_create_info_struct(read_ptr);
784
   end_info_ptr = png_create_info_struct(read_ptr);
785 786
#ifdef PNG_WRITE_SUPPORTED
   write_info_ptr = png_create_info_struct(write_ptr);
787
   write_end_info_ptr = png_create_info_struct(write_ptr);
788
#endif
G
Guy Schalnat 已提交
789

790
#ifdef PNG_SETJMP_SUPPORTED
791
   pngtest_debug("Setting jmpbuf for read struct");
A
Andreas Dilger 已提交
792
#ifdef USE_FAR_KEYWORD
793
   if (setjmp(png_jmpbuf))
A
Andreas Dilger 已提交
794
#else
795
   if (setjmp(png_jmpbuf(read_ptr)))
A
Andreas Dilger 已提交
796
#endif
G
Guy Schalnat 已提交
797
   {
798
      fprintf(STDERR, "%s -> %s: libpng read error\n", inname, outname);
799 800
      png_free(read_ptr, row_buf);
      row_buf = NULL;
A
Andreas Dilger 已提交
801
      png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
802
#ifdef PNG_WRITE_SUPPORTED
803
      png_destroy_info_struct(write_ptr, &write_end_info_ptr);
A
Andreas Dilger 已提交
804
      png_destroy_write_struct(&write_ptr, &write_info_ptr);
805
#endif
806 807
      FCLOSE(fpin);
      FCLOSE(fpout);
808
      return (1);
G
Guy Schalnat 已提交
809
   }
810
#ifdef USE_FAR_KEYWORD
811
   png_memcpy(png_jmpbuf(read_ptr), png_jmpbuf, png_sizeof(jmp_buf));
812
#endif
A
Andreas Dilger 已提交
813

814
#ifdef PNG_WRITE_SUPPORTED
815
   pngtest_debug("Setting jmpbuf for write struct");
A
Andreas Dilger 已提交
816
#ifdef USE_FAR_KEYWORD
817

818
   if (setjmp(png_jmpbuf))
A
Andreas Dilger 已提交
819
#else
820
   if (setjmp(png_jmpbuf(write_ptr)))
A
Andreas Dilger 已提交
821
#endif
G
Guy Schalnat 已提交
822
   {
823
      fprintf(STDERR, "%s -> %s: libpng write error\n", inname, outname);
A
Andreas Dilger 已提交
824
      png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
825
      png_destroy_info_struct(write_ptr, &write_end_info_ptr);
826
#ifdef PNG_WRITE_SUPPORTED
A
Andreas Dilger 已提交
827
      png_destroy_write_struct(&write_ptr, &write_info_ptr);
828
#endif
829 830
      FCLOSE(fpin);
      FCLOSE(fpout);
831
      return (1);
G
Guy Schalnat 已提交
832
   }
833

A
Andreas Dilger 已提交
834
#ifdef USE_FAR_KEYWORD
835
   png_memcpy(png_jmpbuf(write_ptr), png_jmpbuf, png_sizeof(jmp_buf));
836
#endif
837
#endif
A
Andreas Dilger 已提交
838
#endif
839

840
   pngtest_debug("Initializing input and output streams");
841
#ifdef PNG_STDIO_SUPPORTED
G
Guy Schalnat 已提交
842
   png_init_io(read_ptr, fpin);
843
#  ifdef PNG_WRITE_SUPPORTED
G
Guy Schalnat 已提交
844
   png_init_io(write_ptr, fpout);
845
#  endif
846
#else
847
   png_set_read_fn(read_ptr, (png_voidp)fpin, pngtest_read_data);
848
#  ifdef PNG_WRITE_SUPPORTED
849
   png_set_write_fn(write_ptr, (png_voidp)fpout,  pngtest_write_data,
850
#    ifdef PNG_WRITE_FLUSH_SUPPORTED
851
      pngtest_flush);
852
#    else
853
      NULL);
854 855
#    endif
#  endif
856
#endif
857

858
   if (status_dots_requested == 1)
859
   {
860
#ifdef PNG_WRITE_SUPPORTED
861
      png_set_write_status_fn(write_ptr, write_row_callback);
862
#endif
863 864
      png_set_read_status_fn(read_ptr, read_row_callback);
   }
865

866 867
   else
   {
868
#ifdef PNG_WRITE_SUPPORTED
869
      png_set_write_status_fn(write_ptr, NULL);
870
#endif
871
      png_set_read_status_fn(read_ptr, NULL);
872 873
   }

874
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
875
   {
876
      int i;
877

878 879
      for (i = 0; i<256; i++)
         filters_used[i] = 0;
880

881
      png_set_read_user_transform_fn(read_ptr, count_filters);
882 883
   }
#endif
884
#ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
885
   zero_samples = 0;
886 887
   png_set_write_user_transform_fn(write_ptr, count_zero_samples);
#endif
G
Guy Schalnat 已提交
888

889
#ifdef PNG_READ_UNKNOWN_CHUNKS_SUPPORTED
890 891 892
#  ifndef PNG_HANDLE_CHUNK_ALWAYS
#    define PNG_HANDLE_CHUNK_ALWAYS       3
#  endif
893 894
   png_set_keep_unknown_chunks(read_ptr, PNG_HANDLE_CHUNK_ALWAYS,
      NULL, 0);
895
#endif
896
#ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
897 898 899
#  ifndef PNG_HANDLE_CHUNK_IF_SAFE
#    define PNG_HANDLE_CHUNK_IF_SAFE      2
#  endif
900 901
   png_set_keep_unknown_chunks(write_ptr, PNG_HANDLE_CHUNK_IF_SAFE,
      NULL, 0);
902 903
#endif

904
   pngtest_debug("Reading info struct");
A
Andreas Dilger 已提交
905
   png_read_info(read_ptr, read_info_ptr);
G
Guy Schalnat 已提交
906

907
   pngtest_debug("Transferring info struct");
A
Andreas Dilger 已提交
908 909
   {
      int interlace_type, compression_type, filter_type;
G
Guy Schalnat 已提交
910

A
Andreas Dilger 已提交
911 912 913 914
      if (png_get_IHDR(read_ptr, read_info_ptr, &width, &height, &bit_depth,
          &color_type, &interlace_type, &compression_type, &filter_type))
      {
         png_set_IHDR(write_ptr, write_info_ptr, width, height, bit_depth,
915
#ifdef PNG_WRITE_INTERLACING_SUPPORTED
A
Andreas Dilger 已提交
916
            color_type, interlace_type, compression_type, filter_type);
917 918 919
#else
            color_type, PNG_INTERLACE_NONE, compression_type, filter_type);
#endif
A
Andreas Dilger 已提交
920 921
      }
   }
922 923
#ifdef PNG_FIXED_POINT_SUPPORTED
#ifdef PNG_cHRM_SUPPORTED
G
Guy Schalnat 已提交
924
   {
925
      png_fixed_point white_x, white_y, red_x, red_y, green_x, green_y, blue_x,
926
         blue_y;
927

928 929
      if (png_get_cHRM_fixed(read_ptr, read_info_ptr, &white_x, &white_y,
         &red_x, &red_y, &green_x, &green_y, &blue_x, &blue_y))
A
Andreas Dilger 已提交
930
      {
931 932
         png_set_cHRM_fixed(write_ptr, write_info_ptr, white_x, white_y, red_x,
            red_y, green_x, green_y, blue_x, blue_y);
A
Andreas Dilger 已提交
933
      }
G
Guy Schalnat 已提交
934
   }
A
Andreas Dilger 已提交
935
#endif
936
#ifdef PNG_gAMA_SUPPORTED
A
Andreas Dilger 已提交
937
   {
938
      png_fixed_point gamma;
G
Guy Schalnat 已提交
939

940 941
      if (png_get_gAMA_fixed(read_ptr, read_info_ptr, &gamma))
         png_set_gAMA_fixed(write_ptr, write_info_ptr, gamma);
A
Andreas Dilger 已提交
942 943
   }
#endif
944
#else /* Use floating point versions */
945 946
#ifdef PNG_FLOATING_POINT_SUPPORTED
#ifdef PNG_cHRM_SUPPORTED
947 948 949
   {
      double white_x, white_y, red_x, red_y, green_x, green_y, blue_x,
         blue_y;
950

951 952 953 954 955 956 957 958
      if (png_get_cHRM(read_ptr, read_info_ptr, &white_x, &white_y, &red_x,
         &red_y, &green_x, &green_y, &blue_x, &blue_y))
      {
         png_set_cHRM(write_ptr, write_info_ptr, white_x, white_y, red_x,
            red_y, green_x, green_y, blue_x, blue_y);
      }
   }
#endif
959
#ifdef PNG_gAMA_SUPPORTED
960 961 962 963 964 965 966
   {
      double gamma;

      if (png_get_gAMA(read_ptr, read_info_ptr, &gamma))
         png_set_gAMA(write_ptr, write_info_ptr, gamma);
   }
#endif
967 968
#endif /* Floating point */
#endif /* Fixed point */
969
#ifdef PNG_iCCP_SUPPORTED
G
Guy Schalnat 已提交
970
   {
971
      png_charp name;
G
[devel]  
Glenn Randers-Pehrson 已提交
972
      png_bytep profile;
973
      png_uint_32 proflen;
974
      int compression_type;
A
Andreas Dilger 已提交
975

976 977
      if (png_get_iCCP(read_ptr, read_info_ptr, &name, &compression_type,
                      &profile, &proflen))
A
Andreas Dilger 已提交
978
      {
979 980
         png_set_iCCP(write_ptr, write_info_ptr, name, compression_type,
                      profile, proflen);
A
Andreas Dilger 已提交
981
      }
G
Guy Schalnat 已提交
982
   }
983
#endif
984
#ifdef PNG_sRGB_SUPPORTED
985
   {
986
      int intent;
987 988 989 990

      if (png_get_sRGB(read_ptr, read_info_ptr, &intent))
         png_set_sRGB(write_ptr, write_info_ptr, intent);
   }
A
Andreas Dilger 已提交
991
#endif
992 993 994 995 996 997 998
   {
      png_colorp palette;
      int num_palette;

      if (png_get_PLTE(read_ptr, read_info_ptr, &palette, &num_palette))
         png_set_PLTE(write_ptr, write_info_ptr, palette, num_palette);
   }
999
#ifdef PNG_bKGD_SUPPORTED
1000 1001 1002 1003 1004 1005 1006 1007 1008
   {
      png_color_16p background;

      if (png_get_bKGD(read_ptr, read_info_ptr, &background))
      {
         png_set_bKGD(write_ptr, write_info_ptr, background);
      }
   }
#endif
1009
#ifdef PNG_hIST_SUPPORTED
G
Guy Schalnat 已提交
1010
   {
A
Andreas Dilger 已提交
1011 1012 1013 1014
      png_uint_16p hist;

      if (png_get_hIST(read_ptr, read_info_ptr, &hist))
         png_set_hIST(write_ptr, write_info_ptr, hist);
G
Guy Schalnat 已提交
1015
   }
A
Andreas Dilger 已提交
1016
#endif
1017
#ifdef PNG_oFFs_SUPPORTED
A
Andreas Dilger 已提交
1018
   {
1019
      png_int_32 offset_x, offset_y;
A
Andreas Dilger 已提交
1020
      int unit_type;
G
Guy Schalnat 已提交
1021

1022
      if (png_get_oFFs(read_ptr, read_info_ptr, &offset_x, &offset_y,
1023
          &unit_type))
A
Andreas Dilger 已提交
1024 1025 1026 1027 1028
      {
         png_set_oFFs(write_ptr, write_info_ptr, offset_x, offset_y, unit_type);
      }
   }
#endif
1029
#ifdef PNG_pCAL_SUPPORTED
A
Andreas Dilger 已提交
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043
   {
      png_charp purpose, units;
      png_charpp params;
      png_int_32 X0, X1;
      int type, nparams;

      if (png_get_pCAL(read_ptr, read_info_ptr, &purpose, &X0, &X1, &type,
         &nparams, &units, &params))
      {
         png_set_pCAL(write_ptr, write_info_ptr, purpose, X0, X1, type,
            nparams, units, params);
      }
   }
#endif
1044
#ifdef PNG_pHYs_SUPPORTED
A
Andreas Dilger 已提交
1045 1046 1047 1048 1049 1050 1051 1052
   {
      png_uint_32 res_x, res_y;
      int unit_type;

      if (png_get_pHYs(read_ptr, read_info_ptr, &res_x, &res_y, &unit_type))
         png_set_pHYs(write_ptr, write_info_ptr, res_x, res_y, unit_type);
   }
#endif
1053
#ifdef PNG_sBIT_SUPPORTED
A
Andreas Dilger 已提交
1054
   {
1055
      png_color_8p sig_bit;
A
Andreas Dilger 已提交
1056

1057 1058
      if (png_get_sBIT(read_ptr, read_info_ptr, &sig_bit))
         png_set_sBIT(write_ptr, write_info_ptr, sig_bit);
A
Andreas Dilger 已提交
1059
   }
1060
#endif
1061
#ifdef PNG_sCAL_SUPPORTED
1062
#ifdef PNG_FLOATING_POINT_SUPPORTED
G
Guy Schalnat 已提交
1063
   {
1064
      int unit;
1065
      double scal_width, scal_height;
A
Andreas Dilger 已提交
1066

1067 1068
      if (png_get_sCAL(read_ptr, read_info_ptr, &unit, &scal_width,
         &scal_height))
G
Guy Schalnat 已提交
1069
      {
1070
         png_set_sCAL(write_ptr, write_info_ptr, unit, scal_width, scal_height);
1071 1072 1073
      }
   }
#else
1074
#ifdef PNG_FIXED_POINT_SUPPORTED
1075
   {
1076
      int unit;
1077
      png_charp scal_width, scal_height;
1078

1079 1080
      if (png_get_sCAL_s(read_ptr, read_info_ptr, &unit, &scal_width,
          &scal_height))
1081
      {
1082 1083
         png_set_sCAL_s(write_ptr, write_info_ptr, unit, scal_width,
             scal_height);
A
Andreas Dilger 已提交
1084 1085
      }
   }
G
Guy Schalnat 已提交
1086
#endif
1087 1088
#endif
#endif
1089
#ifdef PNG_TEXT_SUPPORTED
A
Andreas Dilger 已提交
1090 1091 1092 1093 1094 1095
   {
      png_textp text_ptr;
      int num_text;

      if (png_get_text(read_ptr, read_info_ptr, &text_ptr, &num_text) > 0)
      {
1096
         pngtest_debug1("Handling %d iTXt/tEXt/zTXt chunks", num_text);
A
Andreas Dilger 已提交
1097 1098 1099 1100
         png_set_text(write_ptr, write_info_ptr, text_ptr, num_text);
      }
   }
#endif
1101
#ifdef PNG_tIME_SUPPORTED
A
Andreas Dilger 已提交
1102 1103 1104 1105 1106 1107
   {
      png_timep mod_time;

      if (png_get_tIME(read_ptr, read_info_ptr, &mod_time))
      {
         png_set_tIME(write_ptr, write_info_ptr, mod_time);
1108
#ifdef PNG_TIME_RFC1123_SUPPORTED
1109
         /* We have to use png_memcpy instead of "=" because the string
1110 1111 1112
          * pointed to by png_convert_to_rfc1123() gets free'ed before
          * we use it.
          */
1113 1114 1115
         png_memcpy(tIME_string,
                    png_convert_to_rfc1123(read_ptr, mod_time),
                    png_sizeof(tIME_string));
1116

1117
         tIME_string[png_sizeof(tIME_string) - 1] = '\0';
1118 1119
         tIME_chunk_present++;
#endif /* PNG_TIME_RFC1123_SUPPORTED */
1120
      }
A
Andreas Dilger 已提交
1121 1122
   }
#endif
1123
#ifdef PNG_tRNS_SUPPORTED
A
Andreas Dilger 已提交
1124
   {
1125
      png_bytep trans_alpha;
A
Andreas Dilger 已提交
1126
      int num_trans;
1127
      png_color_16p trans_color;
A
Andreas Dilger 已提交
1128

1129
      if (png_get_tRNS(read_ptr, read_info_ptr, &trans_alpha, &num_trans,
1130
         &trans_color))
A
Andreas Dilger 已提交
1131
      {
1132
         int sample_max = (1 << bit_depth);
1133
         /* libpng doesn't reject a tRNS chunk with out-of-range samples */
1134
         if (!((color_type == PNG_COLOR_TYPE_GRAY &&
1135
             (int)trans_color->gray > sample_max) ||
1136
             (color_type == PNG_COLOR_TYPE_RGB &&
1137 1138 1139
             ((int)trans_color->red > sample_max ||
             (int)trans_color->green > sample_max ||
             (int)trans_color->blue > sample_max))))
1140
            png_set_tRNS(write_ptr, write_info_ptr, trans_alpha, num_trans,
1141
               trans_color);
A
Andreas Dilger 已提交
1142 1143 1144
      }
   }
#endif
1145
#ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1146 1147
   {
      png_unknown_chunkp unknowns;
1148
      int num_unknowns = png_get_unknown_chunks(read_ptr, read_info_ptr,
1149
         &unknowns);
1150

1151 1152
      if (num_unknowns)
      {
1153
         int i;
1154 1155
         png_set_unknown_chunks(write_ptr, write_info_ptr, unknowns,
           num_unknowns);
1156
         /* Copy the locations from the read_info_ptr.  The automatically
1157 1158 1159
          * generated locations in write_info_ptr are wrong because we
          * haven't written anything yet.
          */
1160
         for (i = 0; i < num_unknowns; i++)
1161 1162
           png_set_unknown_chunk_location(write_ptr, write_info_ptr, i,
             unknowns[i].location);
1163 1164 1165
      }
   }
#endif
A
Andreas Dilger 已提交
1166

1167
#ifdef PNG_WRITE_SUPPORTED
1168
   pngtest_debug("Writing info struct");
1169 1170

/* If we wanted, we could write info in two steps:
1171
 * png_write_info_before_PLTE(write_ptr, write_info_ptr);
1172
 */
A
Andreas Dilger 已提交
1173
   png_write_info(write_ptr, write_info_ptr);
1174

1175
#ifdef PNG_UNKNOWN_CHUNKS_SUPPORTED
1176 1177
   if (user_chunk_data[0] != 0)
   {
1178
      png_byte png_sTER[5] = {115,  84,  69,  82, '\0'};
1179

1180 1181
      unsigned char
        ster_chunk_data[1];
1182

1183 1184 1185
      if (verbose)
         fprintf(STDERR, "\n stereo mode = %lu\n",
           (unsigned long)(user_chunk_data[0] - 1));
1186

1187 1188
      ster_chunk_data[0]=(unsigned char)(user_chunk_data[0] - 1);
      png_write_chunk(write_ptr, png_sTER, ster_chunk_data, 1);
1189
   }
1190

1191 1192
   if (user_chunk_data[1] != 0 || user_chunk_data[2] != 0)
   {
1193 1194 1195 1196 1197 1198 1199 1200 1201 1202
      png_byte png_vpAg[5] = {118, 112,  65, 103, '\0'};

      unsigned char
        vpag_chunk_data[9];

      if (verbose)
         fprintf(STDERR, " vpAg = %lu x %lu, units = %lu\n",
           (unsigned long)user_chunk_data[1],
           (unsigned long)user_chunk_data[2],
           (unsigned long)user_chunk_data[3]);
1203

1204 1205 1206 1207
      png_save_uint_32(vpag_chunk_data, user_chunk_data[1]);
      png_save_uint_32(vpag_chunk_data + 4, user_chunk_data[2]);
      vpag_chunk_data[8] = (unsigned char)(user_chunk_data[3] & 0xff);
      png_write_chunk(write_ptr, png_vpAg, vpag_chunk_data, 9);
1208 1209 1210
   }

#endif
1211
#endif
A
Andreas Dilger 已提交
1212

1213
#ifdef SINGLE_ROWBUF_ALLOC
1214
   pngtest_debug("Allocating row buffer...");
1215
   row_buf = (png_bytep)png_malloc(read_ptr,
A
Andreas Dilger 已提交
1216
      png_get_rowbytes(read_ptr, read_info_ptr));
1217

1218
   pngtest_debug1("\t0x%08lx", (unsigned long)row_buf);
1219
#endif /* SINGLE_ROWBUF_ALLOC */
1220
   pngtest_debug("Writing row data");
A
Andreas Dilger 已提交
1221

1222 1223
#if defined(PNG_READ_INTERLACING_SUPPORTED) || \
  defined(PNG_WRITE_INTERLACING_SUPPORTED)
A
Andreas Dilger 已提交
1224
   num_pass = png_set_interlace_handling(read_ptr);
1225
#  ifdef PNG_WRITE_SUPPORTED
A
Andreas Dilger 已提交
1226
   png_set_interlace_handling(write_ptr);
1227
#  endif
1228
#else
1229
   num_pass = 1;
1230
#endif
A
Andreas Dilger 已提交
1231

1232 1233 1234 1235 1236
#ifdef PNGTEST_TIMING
   t_stop = (float)clock();
   t_misc += (t_stop - t_start);
   t_start = t_stop;
#endif
A
Andreas Dilger 已提交
1237 1238
   for (pass = 0; pass < num_pass; pass++)
   {
1239
      pngtest_debug1("Writing row data for pass %d", pass);
A
Andreas Dilger 已提交
1240 1241
      for (y = 0; y < height; y++)
      {
1242
#ifndef SINGLE_ROWBUF_ALLOC
1243
         pngtest_debug2("Allocating row buffer (pass %d, y = %u)...", pass, y);
1244 1245
         row_buf = (png_bytep)png_malloc(read_ptr,
            png_get_rowbytes(read_ptr, read_info_ptr));
1246

1247
         pngtest_debug2("\t0x%08lx (%u bytes)", (unsigned long)row_buf,
1248
            png_get_rowbytes(read_ptr, read_info_ptr));
1249

1250
#endif /* !SINGLE_ROWBUF_ALLOC */
1251
         png_read_rows(read_ptr, (png_bytepp)&row_buf, NULL, 1);
1252 1253

#ifdef PNG_WRITE_SUPPORTED
1254 1255 1256 1257 1258
#ifdef PNGTEST_TIMING
         t_stop = (float)clock();
         t_decode += (t_stop - t_start);
         t_start = t_stop;
#endif
G
Guy Schalnat 已提交
1259
         png_write_rows(write_ptr, (png_bytepp)&row_buf, 1);
1260 1261 1262 1263 1264
#ifdef PNGTEST_TIMING
         t_stop = (float)clock();
         t_encode += (t_stop - t_start);
         t_start = t_stop;
#endif
1265 1266 1267
#endif /* PNG_WRITE_SUPPORTED */

#ifndef SINGLE_ROWBUF_ALLOC
1268
         pngtest_debug2("Freeing row buffer (pass %d, y = %u)", pass, y);
1269
         png_free(read_ptr, row_buf);
1270
         row_buf = NULL;
1271
#endif /* !SINGLE_ROWBUF_ALLOC */
G
Guy Schalnat 已提交
1272 1273 1274
      }
   }

1275
#ifdef PNG_READ_UNKNOWN_CHUNKS_SUPPORTED
1276
   png_free_data(read_ptr, read_info_ptr, PNG_FREE_UNKN, -1);
1277
#endif
1278
#ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1279
   png_free_data(write_ptr, write_info_ptr, PNG_FREE_UNKN, -1);
1280 1281
#endif

1282
   pngtest_debug("Reading and writing end_info data");
1283

A
Andreas Dilger 已提交
1284
   png_read_end(read_ptr, end_info_ptr);
1285
#ifdef PNG_TEXT_SUPPORTED
1286 1287 1288 1289 1290 1291
   {
      png_textp text_ptr;
      int num_text;

      if (png_get_text(read_ptr, end_info_ptr, &text_ptr, &num_text) > 0)
      {
1292
         pngtest_debug1("Handling %d iTXt/tEXt/zTXt chunks", num_text);
1293 1294 1295 1296
         png_set_text(write_ptr, write_end_info_ptr, text_ptr, num_text);
      }
   }
#endif
1297
#ifdef PNG_tIME_SUPPORTED
1298 1299 1300 1301 1302 1303
   {
      png_timep mod_time;

      if (png_get_tIME(read_ptr, end_info_ptr, &mod_time))
      {
         png_set_tIME(write_ptr, write_end_info_ptr, mod_time);
1304
#ifdef PNG_TIME_RFC1123_SUPPORTED
1305
         /* We have to use png_memcpy instead of "=" because the string
1306 1307
            pointed to by png_convert_to_rfc1123() gets free'ed before
            we use it */
1308 1309 1310
         png_memcpy(tIME_string,
                    png_convert_to_rfc1123(read_ptr, mod_time),
                    png_sizeof(tIME_string));
1311

1312
         tIME_string[png_sizeof(tIME_string) - 1] = '\0';
1313 1314 1315 1316
         tIME_chunk_present++;
#endif /* PNG_TIME_RFC1123_SUPPORTED */
      }
   }
1317
#endif
1318
#ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1319 1320
   {
      png_unknown_chunkp unknowns;
1321
      int num_unknowns = png_get_unknown_chunks(read_ptr, end_info_ptr,
1322
         &unknowns);
1323

1324 1325
      if (num_unknowns)
      {
1326
         int i;
1327 1328
         png_set_unknown_chunks(write_ptr, write_end_info_ptr, unknowns,
           num_unknowns);
1329
         /* Copy the locations from the read_info_ptr.  The automatically
1330 1331 1332
          * generated locations in write_end_info_ptr are wrong because we
          * haven't written the end_info yet.
          */
1333
         for (i = 0; i < num_unknowns; i++)
1334 1335
           png_set_unknown_chunk_location(write_ptr, write_end_info_ptr, i,
             unknowns[i].location);
1336 1337
      }
   }
1338
#endif
1339
#ifdef PNG_WRITE_SUPPORTED
1340
   png_write_end(write_ptr, write_end_info_ptr);
1341
#endif
1342

1343
#ifdef PNG_EASY_ACCESS_SUPPORTED
1344
   if (verbose)
1345 1346 1347 1348
   {
      png_uint_32 iwidth, iheight;
      iwidth = png_get_image_width(write_ptr, write_info_ptr);
      iheight = png_get_image_height(write_ptr, write_info_ptr);
1349
      fprintf(STDERR, "\n Image width = %lu, height = %lu\n",
1350
         (unsigned long)iwidth, (unsigned long)iheight);
1351 1352
   }
#endif
G
Guy Schalnat 已提交
1353

1354
   pngtest_debug("Destroying data structs");
1355
#ifdef SINGLE_ROWBUF_ALLOC
1356
   pngtest_debug("destroying row_buf for read_ptr");
1357
   png_free(read_ptr, row_buf);
1358
   row_buf = NULL;
1359
#endif /* SINGLE_ROWBUF_ALLOC */
1360
   pngtest_debug("destroying read_ptr, read_info_ptr, end_info_ptr");
A
Andreas Dilger 已提交
1361
   png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
1362
#ifdef PNG_WRITE_SUPPORTED
1363
   pngtest_debug("destroying write_end_info_ptr");
1364
   png_destroy_info_struct(write_ptr, &write_end_info_ptr);
1365
   pngtest_debug("destroying write_ptr, write_info_ptr");
A
Andreas Dilger 已提交
1366
   png_destroy_write_struct(&write_ptr, &write_info_ptr);
1367
#endif
1368
   pngtest_debug("Destruction complete.");
G
Guy Schalnat 已提交
1369

1370 1371
   FCLOSE(fpin);
   FCLOSE(fpout);
G
Guy Schalnat 已提交
1372

1373
   pngtest_debug("Opening files for comparison");
A
Andreas Dilger 已提交
1374
   if ((fpin = fopen(inname, "rb")) == NULL)
G
Guy Schalnat 已提交
1375
   {
G
Guy Schalnat 已提交
1376
      fprintf(STDERR, "Could not find file %s\n", inname);
1377
      return (1);
G
Guy Schalnat 已提交
1378 1379
   }

A
Andreas Dilger 已提交
1380
   if ((fpout = fopen(outname, "rb")) == NULL)
G
Guy Schalnat 已提交
1381
   {
G
Guy Schalnat 已提交
1382
      fprintf(STDERR, "Could not find file %s\n", outname);
1383
      FCLOSE(fpin);
1384
      return (1);
G
Guy Schalnat 已提交
1385
   }
A
Andreas Dilger 已提交
1386

1387
   for (;;)
G
Guy Schalnat 已提交
1388
   {
A
Andreas Dilger 已提交
1389
      png_size_t num_in, num_out;
G
Guy Schalnat 已提交
1390

1391 1392
         num_in = fread(inbuf, 1, 1, fpin);
         num_out = fread(outbuf, 1, 1, fpout);
G
Guy Schalnat 已提交
1393 1394 1395

      if (num_in != num_out)
      {
1396
         fprintf(STDERR, "\nFiles %s and %s are of a different size\n",
G
Guy Schalnat 已提交
1397
                 inname, outname);
1398

1399
         if (wrote_question == 0)
1400 1401
         {
            fprintf(STDERR,
1402
         "   Was %s written with the same maximum IDAT chunk size (%d bytes),",
1403
              inname, PNG_ZBUF_SIZE);
1404
            fprintf(STDERR,
1405
              "\n   filtering heuristic (libpng default), compression");
1406
            fprintf(STDERR,
1407
              " level (zlib default),\n   and zlib version (%s)?\n\n",
1408
              ZLIB_VERSION);
1409
            wrote_question = 1;
1410
         }
1411

1412 1413
         FCLOSE(fpin);
         FCLOSE(fpout);
1414
         return (0);
G
Guy Schalnat 已提交
1415 1416 1417 1418 1419
      }

      if (!num_in)
         break;

A
Andreas Dilger 已提交
1420
      if (png_memcmp(inbuf, outbuf, num_in))
G
Guy Schalnat 已提交
1421
      {
1422
         fprintf(STDERR, "\nFiles %s and %s are different\n", inname, outname);
1423

1424
         if (wrote_question == 0)
1425 1426
         {
            fprintf(STDERR,
1427
         "   Was %s written with the same maximum IDAT chunk size (%d bytes),",
1428
                 inname, PNG_ZBUF_SIZE);
1429
            fprintf(STDERR,
1430
              "\n   filtering heuristic (libpng default), compression");
1431
            fprintf(STDERR,
1432
              " level (zlib default),\n   and zlib version (%s)?\n\n",
1433
              ZLIB_VERSION);
1434
            wrote_question = 1;
1435
         }
1436

1437 1438
         FCLOSE(fpin);
         FCLOSE(fpout);
1439
         return (0);
G
Guy Schalnat 已提交
1440 1441 1442
      }
   }

1443 1444
   FCLOSE(fpin);
   FCLOSE(fpout);
G
Guy Schalnat 已提交
1445

1446
   return (0);
G
Guy Schalnat 已提交
1447
}
G
Guy Schalnat 已提交
1448

1449
/* Input and output filenames */
1450
#ifdef RISCOS
1451 1452
static PNG_CONST char *inname = "pngtest/png";
static PNG_CONST char *outname = "pngout/png";
1453
#else
1454 1455
static PNG_CONST char *inname = "pngtest.png";
static PNG_CONST char *outname = "pngout.png";
1456 1457 1458 1459 1460 1461 1462 1463
#endif

int
main(int argc, char *argv[])
{
   int multiple = 0;
   int ierror = 0;

1464
   fprintf(STDERR, "\n Testing libpng version %s\n", PNG_LIBPNG_VER_STRING);
1465
   fprintf(STDERR, "   with zlib   version %s\n", ZLIB_VERSION);
1466
   fprintf(STDERR, "%s", png_get_copyright(NULL));
1467
   /* Show the version of libpng used in building the library */
1468 1469
   fprintf(STDERR, " library (%lu):%s",
      (unsigned long)png_access_version_number(),
1470
      png_get_header_version(NULL));
1471

1472
   /* Show the version of libpng used in building the application */
1473
   fprintf(STDERR, " pngtest (%lu):%s", (unsigned long)PNG_LIBPNG_VER,
1474
      PNG_HEADER_VERSION_STRING);
1475 1476

   /* Do some consistency checking on the memory allocation settings, I'm
1477 1478 1479 1480
    * not sure this matters, but it is nice to know, the first of these
    * tests should be impossible because of the way the macros are set
    * in pngconf.h
    */
1481
#if defined(MAXSEG_64K) && !defined(PNG_MAX_MALLOC_64K)
1482
      fprintf(STDERR, " NOTE: Zlib compiled for max 64k, libpng not\n");
1483
#endif
1484
   /* I think the following can happen. */
1485
#if !defined(MAXSEG_64K) && defined(PNG_MAX_MALLOC_64K)
1486
      fprintf(STDERR, " NOTE: libpng compiled for max 64k, zlib not\n");
1487
#endif
1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499

   if (strcmp(png_libpng_ver, PNG_LIBPNG_VER_STRING))
   {
      fprintf(STDERR,
         "Warning: versions are different between png.h and png.c\n");
      fprintf(STDERR, "  png.h version: %s\n", PNG_LIBPNG_VER_STRING);
      fprintf(STDERR, "  png.c version: %s\n\n", png_libpng_ver);
      ++ierror;
   }

   if (argc > 1)
   {
1500
      if (strcmp(argv[1], "-m") == 0)
1501
      {
1502
         multiple = 1;
1503 1504
         status_dots_requested = 0;
      }
1505

1506 1507 1508 1509 1510
      else if (strcmp(argv[1], "-mv") == 0 ||
               strcmp(argv[1], "-vm") == 0 )
      {
         multiple = 1;
         verbose = 1;
1511
         status_dots_requested = 1;
1512
      }
1513

1514 1515 1516
      else if (strcmp(argv[1], "-v") == 0)
      {
         verbose = 1;
1517
         status_dots_requested = 1;
1518 1519
         inname = argv[2];
      }
1520

1521
      else
1522
      {
1523
         inname = argv[1];
1524 1525
         status_dots_requested = 0;
      }
1526 1527
   }

1528 1529
   if (!multiple && argc == 3 + verbose)
     outname = argv[2 + verbose];
1530

1531
   if ((!multiple && argc > 3 + verbose) || (multiple && argc < 2))
1532
   {
1533 1534
     fprintf(STDERR,
       "usage: %s [infile.png] [outfile.png]\n\t%s -m {infile.png}\n",
1535
        argv[0], argv[0]);
1536 1537 1538 1539
     fprintf(STDERR,
       "  reads/writes one PNG file (without -m) or multiple files (-m)\n");
     fprintf(STDERR,
       "  with -m %s is used as a temporary file\n", outname);
1540 1541 1542 1543 1544 1545
     exit(1);
   }

   if (multiple)
   {
      int i;
1546
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1547 1548
      int allocation_now = current_allocation;
#endif
1549
      for (i=2; i<argc; ++i)
1550
      {
1551
         int kerror;
1552
         fprintf(STDERR, "\n Testing %s:", argv[i]);
1553
         kerror = test_one_file(argv[i], outname);
1554
         if (kerror == 0)
1555
         {
1556 1557 1558
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
            int k;
#endif
1559
#ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
1560
            fprintf(STDERR, "\n PASS (%lu zero samples)\n",
1561
               (unsigned long)zero_samples);
1562
#else
1563
            fprintf(STDERR, " PASS\n");
1564
#endif
1565
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1566 1567
            for (k = 0; k<256; k++)
               if (filters_used[k])
1568
                  fprintf(STDERR, " Filter %d was used %lu times\n",
1569
                     k, (unsigned long)filters_used[k]);
1570
#endif
1571
#ifdef PNG_TIME_RFC1123_SUPPORTED
1572 1573
         if (tIME_chunk_present != 0)
            fprintf(STDERR, " tIME = %s\n", tIME_string);
1574

1575 1576 1577
         tIME_chunk_present = 0;
#endif /* PNG_TIME_RFC1123_SUPPORTED */
         }
1578

1579 1580
         else
         {
1581 1582
            fprintf(STDERR, " FAIL\n");
            ierror += kerror;
1583
         }
1584
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1585 1586
         if (allocation_now != current_allocation)
            fprintf(STDERR, "MEMORY ERROR: %d bytes lost\n",
1587
               current_allocation - allocation_now);
1588

1589 1590
         if (current_allocation != 0)
         {
1591 1592 1593 1594
            memory_infop pinfo = pinformation;

            fprintf(STDERR, "MEMORY ERROR: %d bytes still allocated\n",
               current_allocation);
1595

1596 1597
            while (pinfo != NULL)
            {
1598
               fprintf(STDERR, " %lu bytes at %x\n",
1599
                 (unsigned long)pinfo->size,
1600
                 (unsigned int)pinfo->pointer);
1601
               pinfo = pinfo->next;
1602
            }
1603
         }
1604 1605
#endif
      }
1606
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1607
         fprintf(STDERR, " Current memory allocation: %10d bytes\n",
1608
            current_allocation);
1609
         fprintf(STDERR, " Maximum memory allocation: %10d bytes\n",
1610
            maximum_allocation);
1611 1612 1613 1614
         fprintf(STDERR, " Total   memory allocation: %10d bytes\n",
            total_allocation);
         fprintf(STDERR, "     Number of allocations: %10d\n",
            num_allocations);
1615
#endif
1616
   }
1617

1618 1619
   else
   {
1620
      int i;
1621
      for (i = 0; i<3; ++i)
1622
      {
1623
         int kerror;
1624
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1625 1626
         int allocation_now = current_allocation;
#endif
1627 1628 1629 1630 1631 1632
         if (i == 1)
            status_dots_requested = 1;

         else if (verbose == 0)
            status_dots_requested = 0;

1633
         if (i == 0 || verbose == 1 || ierror != 0)
1634
            fprintf(STDERR, "\n Testing %s:", inname);
1635

1636
         kerror = test_one_file(inname, outname);
1637

1638
         if (kerror == 0)
1639
         {
1640
            if (verbose == 1 || i == 2)
1641
            {
1642
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1643
                int k;
1644
#endif
1645
#ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
1646
                fprintf(STDERR, "\n PASS (%lu zero samples)\n",
1647
                   (unsigned long)zero_samples);
1648 1649 1650
#else
                fprintf(STDERR, " PASS\n");
#endif
1651
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1652 1653
                for (k = 0; k<256; k++)
                   if (filters_used[k])
1654
                      fprintf(STDERR, " Filter %d was used %lu times\n",
1655
                         k, (unsigned long)filters_used[k]);
1656
#endif
1657
#ifdef PNG_TIME_RFC1123_SUPPORTED
1658 1659
             if (tIME_chunk_present != 0)
                fprintf(STDERR, " tIME = %s\n", tIME_string);
1660 1661
#endif /* PNG_TIME_RFC1123_SUPPORTED */
            }
1662
         }
1663

1664 1665
         else
         {
1666
            if (verbose == 0 && i != 2)
1667
               fprintf(STDERR, "\n Testing %s:", inname);
1668

1669 1670
            fprintf(STDERR, " FAIL\n");
            ierror += kerror;
1671
         }
1672
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1673 1674
         if (allocation_now != current_allocation)
             fprintf(STDERR, "MEMORY ERROR: %d bytes lost\n",
1675
               current_allocation - allocation_now);
1676

1677 1678
         if (current_allocation != 0)
         {
1679
             memory_infop pinfo = pinformation;
1680

1681 1682
             fprintf(STDERR, "MEMORY ERROR: %d bytes still allocated\n",
                current_allocation);
1683

1684 1685
             while (pinfo != NULL)
             {
1686 1687
                fprintf(STDERR, " %lu bytes at %x\n",
                   (unsigned long)pinfo->size, (unsigned int)pinfo->pointer);
1688 1689 1690 1691
                pinfo = pinfo->next;
             }
          }
#endif
1692
       }
1693
#if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1694
       fprintf(STDERR, " Current memory allocation: %10d bytes\n",
1695
          current_allocation);
1696
       fprintf(STDERR, " Maximum memory allocation: %10d bytes\n",
1697
          maximum_allocation);
1698 1699 1700 1701
       fprintf(STDERR, " Total   memory allocation: %10d bytes\n",
          total_allocation);
       fprintf(STDERR, "     Number of allocations: %10d\n",
            num_allocations);
1702
#endif
1703 1704
   }

1705 1706 1707 1708
#ifdef PNGTEST_TIMING
   t_stop = (float)clock();
   t_misc += (t_stop - t_start);
   t_start = t_stop;
1709
   fprintf(STDERR, " CPU time used = %.3f seconds",
1710
      (t_misc+t_decode+t_encode)/(float)CLOCKS_PER_SEC);
1711
   fprintf(STDERR, " (decoding %.3f,\n",
1712
      t_decode/(float)CLOCKS_PER_SEC);
1713
   fprintf(STDERR, "        encoding %.3f ,",
1714
      t_encode/(float)CLOCKS_PER_SEC);
1715
   fprintf(STDERR, " other %.3f seconds)\n\n",
1716 1717 1718
      t_misc/(float)CLOCKS_PER_SEC);
#endif

1719
   if (ierror == 0)
1720
      fprintf(STDERR, " libpng passes test\n");
1721

1722
   else
1723
      fprintf(STDERR, " libpng FAILS test\n");
1724

1725
   return (int)(ierror != 0);
1726
}
1727

1728
/* Generate a compiler error if there is an old png.h in the search path. */
1729
typedef png_libpng_version_1_5_0beta54 Your_png_h_is_not_version_1_5_0beta54;