]> rtime.felk.cvut.cz Git - l4.git/blob - l4/pkg/libpng/lib/dist/pngtest.c
update
[l4.git] / l4 / pkg / libpng / lib / dist / pngtest.c
1
2 /* pngtest.c - a simple test program to test libpng
3  *
4  * Last changed in libpng 1.6.1 [March 28, 2013]
5  * Copyright (c) 1998-2013 Glenn Randers-Pehrson
6  * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
7  * (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  * This program reads in a PNG image, writes it out again, and then
14  * compares the two files.  If the files are identical, this shows that
15  * the basic chunk handling, filtering, and (de)compression code is working
16  * properly.  It does not currently test all of the transforms, although
17  * it probably should.
18  *
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  * 3) unknown unsafe-to-copy ancillary chunks or unknown critical chunks
23  *    exist in the input file.
24  * 4) others not listed here...
25  * 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  * If a filename is given on the command-line, then this file is used
29  * for the input, rather than the default "pngtest.png".  This allows
30  * testing a wide variety of files easily.  You can also test a number
31  * of files at once by typing "pngtest -m file1.png file2.png ..."
32  */
33
34 #define _POSIX_SOURCE 1
35
36 #include <stdio.h>
37 #include <stdlib.h>
38 #include <string.h>
39
40 /* Defined so I can write to a file on gui/windowing platforms */
41 /*  #define STDERR stderr  */
42 #define STDERR stdout   /* For DOS */
43
44 #include "png.h"
45
46 #ifdef PNG_READ_SUPPORTED /* else nothing can be done */
47 #include "zlib.h"
48 /* Copied from pngpriv.h but only used in error messages below. */
49 #ifndef PNG_ZBUF_SIZE
50 #  define PNG_ZBUF_SIZE 8192
51 #endif
52 #define FCLOSE(file) fclose(file)
53
54 #ifndef PNG_STDIO_SUPPORTED
55 typedef FILE                * png_FILE_p;
56 #endif
57
58 /* Makes pngtest verbose so we can find problems. */
59 #ifndef PNG_DEBUG
60 #  define PNG_DEBUG 0
61 #endif
62
63 #if PNG_DEBUG > 1
64 #  define pngtest_debug(m)        ((void)fprintf(stderr, m "\n"))
65 #  define pngtest_debug1(m,p1)    ((void)fprintf(stderr, m "\n", p1))
66 #  define pngtest_debug2(m,p1,p2) ((void)fprintf(stderr, m "\n", p1, p2))
67 #else
68 #  define pngtest_debug(m)        ((void)0)
69 #  define pngtest_debug1(m,p1)    ((void)0)
70 #  define pngtest_debug2(m,p1,p2) ((void)0)
71 #endif
72
73 #if !PNG_DEBUG
74 #  define SINGLE_ROWBUF_ALLOC  /* Makes buffer overruns easier to nail */
75 #endif
76
77 /* Turn on CPU timing
78 #define PNGTEST_TIMING
79 */
80
81 #ifndef PNG_FLOATING_POINT_SUPPORTED
82 #undef PNGTEST_TIMING
83 #endif
84
85 #ifdef PNGTEST_TIMING
86 static float t_start, t_stop, t_decode, t_encode, t_misc;
87 #include <time.h>
88 #endif
89
90 #ifdef PNG_TIME_RFC1123_SUPPORTED
91 #define PNG_tIME_STRING_LENGTH 29
92 static int tIME_chunk_present = 0;
93 static char tIME_string[PNG_tIME_STRING_LENGTH] = "tIME chunk is not present";
94 #endif
95
96 static int verbose = 0;
97 static int strict = 0;
98 static int relaxed = 0;
99 static int unsupported_chunks = 0; /* chunk unsupported by libpng in input */
100 static int error_count = 0; /* count calls to png_error */
101 static int warning_count = 0; /* count calls to png_warning */
102
103 #ifdef __TURBOC__
104 #include <mem.h>
105 #endif
106
107 /* Define png_jmpbuf() in case we are using a pre-1.0.6 version of libpng */
108 #ifndef png_jmpbuf
109 #  define png_jmpbuf(png_ptr) png_ptr->jmpbuf
110 #endif
111
112 /* Defines for unknown chunk handling if required. */
113 #ifndef PNG_HANDLE_CHUNK_ALWAYS
114 #  define PNG_HANDLE_CHUNK_ALWAYS       3
115 #endif
116 #ifndef PNG_HANDLE_CHUNK_IF_SAFE
117 #  define PNG_HANDLE_CHUNK_IF_SAFE      2
118 #endif
119
120 /* Utility to save typing/errors, the argument must be a name */
121 #define MEMZERO(var) ((void)memset(&var, 0, sizeof var))
122
123 /* Example of using row callbacks to make a simple progress meter */
124 static int status_pass = 1;
125 static int status_dots_requested = 0;
126 static int status_dots = 1;
127
128 static void PNGCBAPI
129 read_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)
130 {
131    if (png_ptr == NULL || row_number > PNG_UINT_31_MAX)
132       return;
133
134    if (status_pass != pass)
135    {
136       fprintf(stdout, "\n Pass %d: ", pass);
137       status_pass = pass;
138       status_dots = 31;
139    }
140
141    status_dots--;
142
143    if (status_dots == 0)
144    {
145       fprintf(stdout, "\n         ");
146       status_dots=30;
147    }
148
149    fprintf(stdout, "r");
150 }
151
152 #ifdef PNG_WRITE_SUPPORTED
153 static void PNGCBAPI
154 write_row_callback(png_structp png_ptr, png_uint_32 row_number, int pass)
155 {
156    if (png_ptr == NULL || row_number > PNG_UINT_31_MAX || pass > 7)
157       return;
158
159    fprintf(stdout, "w");
160 }
161 #endif
162
163
164 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
165 /* Example of using user transform callback (we don't transform anything,
166  * but merely examine the row filters.  We set this to 256 rather than
167  * 5 in case illegal filter values are present.)
168  */
169 static png_uint_32 filters_used[256];
170 static void PNGCBAPI
171 count_filters(png_structp png_ptr, png_row_infop row_info, png_bytep data)
172 {
173    if (png_ptr != NULL && row_info != NULL)
174       ++filters_used[*(data - 1)];
175 }
176 #endif
177
178 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
179 /* Example of using user transform callback (we don't transform anything,
180  * but merely count the zero samples)
181  */
182
183 static png_uint_32 zero_samples;
184
185 static void PNGCBAPI
186 count_zero_samples(png_structp png_ptr, png_row_infop row_info, png_bytep data)
187 {
188    png_bytep dp = data;
189    if (png_ptr == NULL)
190       return;
191
192    /* Contents of row_info:
193     *  png_uint_32 width      width of row
194     *  png_uint_32 rowbytes   number of bytes in row
195     *  png_byte color_type    color type of pixels
196     *  png_byte bit_depth     bit depth of samples
197     *  png_byte channels      number of channels (1-4)
198     *  png_byte pixel_depth   bits per pixel (depth*channels)
199     */
200
201     /* Counts the number of zero samples (or zero pixels if color_type is 3 */
202
203     if (row_info->color_type == 0 || row_info->color_type == 3)
204     {
205        int pos = 0;
206        png_uint_32 n, nstop;
207
208        for (n = 0, nstop=row_info->width; n<nstop; n++)
209        {
210           if (row_info->bit_depth == 1)
211           {
212              if (((*dp << pos++ ) & 0x80) == 0)
213                 zero_samples++;
214
215              if (pos == 8)
216              {
217                 pos = 0;
218                 dp++;
219              }
220           }
221
222           if (row_info->bit_depth == 2)
223           {
224              if (((*dp << (pos+=2)) & 0xc0) == 0)
225                 zero_samples++;
226
227              if (pos == 8)
228              {
229                 pos = 0;
230                 dp++;
231              }
232           }
233
234           if (row_info->bit_depth == 4)
235           {
236              if (((*dp << (pos+=4)) & 0xf0) == 0)
237                 zero_samples++;
238
239              if (pos == 8)
240              {
241                 pos = 0;
242                 dp++;
243              }
244           }
245
246           if (row_info->bit_depth == 8)
247              if (*dp++ == 0)
248                 zero_samples++;
249
250           if (row_info->bit_depth == 16)
251           {
252              if ((*dp | *(dp+1)) == 0)
253                 zero_samples++;
254              dp+=2;
255           }
256        }
257     }
258     else /* Other color types */
259     {
260        png_uint_32 n, nstop;
261        int channel;
262        int color_channels = row_info->channels;
263        if (row_info->color_type > 3)color_channels--;
264
265        for (n = 0, nstop=row_info->width; n<nstop; n++)
266        {
267           for (channel = 0; channel < color_channels; channel++)
268           {
269              if (row_info->bit_depth == 8)
270                 if (*dp++ == 0)
271                    zero_samples++;
272
273              if (row_info->bit_depth == 16)
274              {
275                 if ((*dp | *(dp+1)) == 0)
276                    zero_samples++;
277
278                 dp+=2;
279              }
280           }
281           if (row_info->color_type > 3)
282           {
283              dp++;
284              if (row_info->bit_depth == 16)
285                 dp++;
286           }
287        }
288     }
289 }
290 #endif /* PNG_WRITE_USER_TRANSFORM_SUPPORTED */
291
292 #ifndef PNG_STDIO_SUPPORTED
293 /* START of code to validate stdio-free compilation */
294 /* These copies of the default read/write functions come from pngrio.c and
295  * pngwio.c.  They allow "don't include stdio" testing of the library.
296  * This is the function that does the actual reading of data.  If you are
297  * not reading from a standard C stream, you should create a replacement
298  * read_data function and use it at run time with png_set_read_fn(), rather
299  * than changing the library.
300  */
301
302 #ifdef PNG_IO_STATE_SUPPORTED
303 void
304 pngtest_check_io_state(png_structp png_ptr, png_size_t data_length,
305    png_uint_32 io_op);
306 void
307 pngtest_check_io_state(png_structp png_ptr, png_size_t data_length,
308    png_uint_32 io_op)
309 {
310    png_uint_32 io_state = png_get_io_state(png_ptr);
311    int err = 0;
312
313    /* Check if the current operation (reading / writing) is as expected. */
314    if ((io_state & PNG_IO_MASK_OP) != io_op)
315       png_error(png_ptr, "Incorrect operation in I/O state");
316
317    /* Check if the buffer size specific to the current location
318     * (file signature / header / data / crc) is as expected.
319     */
320    switch (io_state & PNG_IO_MASK_LOC)
321    {
322    case PNG_IO_SIGNATURE:
323       if (data_length > 8)
324          err = 1;
325       break;
326    case PNG_IO_CHUNK_HDR:
327       if (data_length != 8)
328          err = 1;
329       break;
330    case PNG_IO_CHUNK_DATA:
331       break;  /* no restrictions here */
332    case PNG_IO_CHUNK_CRC:
333       if (data_length != 4)
334          err = 1;
335       break;
336    default:
337       err = 1;  /* uninitialized */
338    }
339    if (err)
340       png_error(png_ptr, "Bad I/O state or buffer size");
341 }
342 #endif
343
344 static void PNGCBAPI
345 pngtest_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
346 {
347    png_size_t check = 0;
348    png_voidp io_ptr;
349
350    /* fread() returns 0 on error, so it is OK to store this in a png_size_t
351     * instead of an int, which is what fread() actually returns.
352     */
353    io_ptr = png_get_io_ptr(png_ptr);
354    if (io_ptr != NULL)
355    {
356       check = fread(data, 1, length, (png_FILE_p)io_ptr);
357    }
358
359    if (check != length)
360    {
361       png_error(png_ptr, "Read Error");
362    }
363
364 #ifdef PNG_IO_STATE_SUPPORTED
365    pngtest_check_io_state(png_ptr, length, PNG_IO_READING);
366 #endif
367 }
368
369 #ifdef PNG_WRITE_FLUSH_SUPPORTED
370 static void PNGCBAPI
371 pngtest_flush(png_structp png_ptr)
372 {
373    /* Do nothing; fflush() is said to be just a waste of energy. */
374    PNG_UNUSED(png_ptr)   /* Stifle compiler warning */
375 }
376 #endif
377
378 /* This is the function that does the actual writing of data.  If you are
379  * not writing to a standard C stream, you should create a replacement
380  * write_data function and use it at run time with png_set_write_fn(), rather
381  * than changing the library.
382  */
383 static void PNGCBAPI
384 pngtest_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
385 {
386    png_size_t check;
387
388    check = fwrite(data, 1, length, (png_FILE_p)png_get_io_ptr(png_ptr));
389
390    if (check != length)
391    {
392       png_error(png_ptr, "Write Error");
393    }
394
395 #ifdef PNG_IO_STATE_SUPPORTED
396    pngtest_check_io_state(png_ptr, length, PNG_IO_WRITING);
397 #endif
398 }
399 #endif /* !PNG_STDIO_SUPPORTED */
400
401 /* This function is called when there is a warning, but the library thinks
402  * it can continue anyway.  Replacement functions don't have to do anything
403  * here if you don't want to.  In the default configuration, png_ptr is
404  * not used, but it is passed in case it may be useful.
405  */
406 typedef struct
407 {
408    PNG_CONST char *file_name;
409 }  pngtest_error_parameters;
410
411 static void PNGCBAPI
412 pngtest_warning(png_structp png_ptr, png_const_charp message)
413 {
414    PNG_CONST char *name = "UNKNOWN (ERROR!)";
415    pngtest_error_parameters *test =
416       (pngtest_error_parameters*)png_get_error_ptr(png_ptr);
417
418    ++warning_count;
419
420    if (test != NULL && test->file_name != NULL)
421       name = test->file_name;
422
423    fprintf(STDERR, "%s: libpng warning: %s\n", name, message);
424 }
425
426 /* This is the default error handling function.  Note that replacements for
427  * this function MUST NOT RETURN, or the program will likely crash.  This
428  * function is used by default, or if the program supplies NULL for the
429  * error function pointer in png_set_error_fn().
430  */
431 static void PNGCBAPI
432 pngtest_error(png_structp png_ptr, png_const_charp message)
433 {
434    ++error_count;
435
436    pngtest_warning(png_ptr, message);
437    /* We can return because png_error calls the default handler, which is
438     * actually OK in this case.
439     */
440 }
441
442 /* END of code to validate stdio-free compilation */
443
444 /* START of code to validate memory allocation and deallocation */
445 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
446
447 /* Allocate memory.  For reasonable files, size should never exceed
448  * 64K.  However, zlib may allocate more then 64K if you don't tell
449  * it not to.  See zconf.h and png.h for more information.  zlib does
450  * need to allocate exactly 64K, so whatever you call here must
451  * have the ability to do that.
452  *
453  * This piece of code can be compiled to validate max 64K allocations
454  * by setting MAXSEG_64K in zlib zconf.h *or* PNG_MAX_MALLOC_64K.
455  */
456 typedef struct memory_information
457 {
458    png_alloc_size_t          size;
459    png_voidp                 pointer;
460    struct memory_information *next;
461 } memory_information;
462 typedef memory_information *memory_infop;
463
464 static memory_infop pinformation = NULL;
465 static int current_allocation = 0;
466 static int maximum_allocation = 0;
467 static int total_allocation = 0;
468 static int num_allocations = 0;
469
470 png_voidp PNGCBAPI png_debug_malloc PNGARG((png_structp png_ptr,
471     png_alloc_size_t size));
472 void PNGCBAPI png_debug_free PNGARG((png_structp png_ptr, png_voidp ptr));
473
474 png_voidp
475 PNGCBAPI png_debug_malloc(png_structp png_ptr, png_alloc_size_t size)
476 {
477
478    /* png_malloc has already tested for NULL; png_create_struct calls
479     * png_debug_malloc directly, with png_ptr == NULL which is OK
480     */
481
482    if (size == 0)
483       return (NULL);
484
485    /* This calls the library allocator twice, once to get the requested
486       buffer and once to get a new free list entry. */
487    {
488       /* Disable malloc_fn and free_fn */
489       memory_infop pinfo;
490       png_set_mem_fn(png_ptr, NULL, NULL, NULL);
491       pinfo = (memory_infop)png_malloc(png_ptr,
492          (sizeof *pinfo));
493       pinfo->size = size;
494       current_allocation += size;
495       total_allocation += size;
496       num_allocations ++;
497
498       if (current_allocation > maximum_allocation)
499          maximum_allocation = current_allocation;
500
501       pinfo->pointer = png_malloc(png_ptr, size);
502       /* Restore malloc_fn and free_fn */
503
504       png_set_mem_fn(png_ptr,
505           NULL, png_debug_malloc, png_debug_free);
506
507       if (size != 0 && pinfo->pointer == NULL)
508       {
509          current_allocation -= size;
510          total_allocation -= size;
511          png_error(png_ptr,
512            "out of memory in pngtest->png_debug_malloc");
513       }
514
515       pinfo->next = pinformation;
516       pinformation = pinfo;
517       /* Make sure the caller isn't assuming zeroed memory. */
518       memset(pinfo->pointer, 0xdd, pinfo->size);
519
520       if (verbose)
521          printf("png_malloc %lu bytes at %p\n", (unsigned long)size,
522             pinfo->pointer);
523
524       return (png_voidp)(pinfo->pointer);
525    }
526 }
527
528 /* Free a pointer.  It is removed from the list at the same time. */
529 void PNGCBAPI
530 png_debug_free(png_structp png_ptr, png_voidp ptr)
531 {
532    if (png_ptr == NULL)
533       fprintf(STDERR, "NULL pointer to png_debug_free.\n");
534
535    if (ptr == 0)
536    {
537 #if 0 /* This happens all the time. */
538       fprintf(STDERR, "WARNING: freeing NULL pointer\n");
539 #endif
540       return;
541    }
542
543    /* Unlink the element from the list. */
544    {
545       memory_infop *ppinfo = &pinformation;
546
547       for (;;)
548       {
549          memory_infop pinfo = *ppinfo;
550
551          if (pinfo->pointer == ptr)
552          {
553             *ppinfo = pinfo->next;
554             current_allocation -= pinfo->size;
555             if (current_allocation < 0)
556                fprintf(STDERR, "Duplicate free of memory\n");
557             /* We must free the list element too, but first kill
558                the memory that is to be freed. */
559             memset(ptr, 0x55, pinfo->size);
560             png_free_default(png_ptr, pinfo);
561             pinfo = NULL;
562             break;
563          }
564
565          if (pinfo->next == NULL)
566          {
567             fprintf(STDERR, "Pointer %x not found\n", (unsigned int)ptr);
568             break;
569          }
570
571          ppinfo = &pinfo->next;
572       }
573    }
574
575    /* Finally free the data. */
576    if (verbose)
577       printf("Freeing %p\n", ptr);
578
579    png_free_default(png_ptr, ptr);
580    ptr = NULL;
581 }
582 #endif /* PNG_USER_MEM_SUPPORTED && PNG_DEBUG */
583 /* END of code to test memory allocation/deallocation */
584
585
586 #ifdef PNG_READ_USER_CHUNKS_SUPPORTED
587 /* Demonstration of user chunk support of the sTER and vpAg chunks */
588
589 /* (sTER is a public chunk not yet known by libpng.  vpAg is a private
590 chunk used in ImageMagick to store "virtual page" size).  */
591
592 static struct user_chunk_data
593 {
594    png_const_infop info_ptr;
595    png_uint_32     vpAg_width, vpAg_height;
596    png_byte        vpAg_units;
597    png_byte        sTER_mode;
598    int             location[2];
599 }
600 user_chunk_data;
601
602 /* Used for location and order; zero means nothing. */
603 #define have_sTER   0x01
604 #define have_vpAg   0x02
605 #define before_PLTE 0x10
606 #define before_IDAT 0x20
607 #define after_IDAT  0x40
608
609 static void
610 init_callback_info(png_const_infop info_ptr)
611 {
612    MEMZERO(user_chunk_data);
613    user_chunk_data.info_ptr = info_ptr;
614 }
615
616 static int
617 set_location(png_structp png_ptr, struct user_chunk_data *data, int what)
618 {
619    int location;
620
621    if ((data->location[0] & what) || (data->location[1] & what))
622       return 0; /* already have one of these */
623
624    /* Find where we are (the code below zeros info_ptr to indicate that the
625     * chunks before the first IDAT have been read.)
626     */
627    if (data->info_ptr == NULL) /* after IDAT */
628       location = what | after_IDAT;
629
630    else if (png_get_valid(png_ptr, data->info_ptr, PNG_INFO_PLTE))
631       location = what | before_IDAT;
632
633    else
634       location = what | before_PLTE;
635
636    if (data->location[0] == 0)
637       data->location[0] = location;
638
639    else
640       data->location[1] = location;
641
642    return 1; /* handled */
643 }
644
645 static int PNGCBAPI read_user_chunk_callback(png_struct *png_ptr,
646    png_unknown_chunkp chunk)
647 {
648    struct user_chunk_data *my_user_chunk_data =
649       (struct user_chunk_data*)png_get_user_chunk_ptr(png_ptr);
650
651    if (my_user_chunk_data == NULL)
652       png_error(png_ptr, "lost user chunk pointer");
653
654    /* Return one of the following:
655     *    return (-n);  chunk had an error
656     *    return (0);  did not recognize
657     *    return (n);  success
658     *
659     * The unknown chunk structure contains the chunk data:
660     * png_byte name[5];
661     * png_byte *data;
662     * png_size_t size;
663     *
664     * Note that libpng has already taken care of the CRC handling.
665     */
666
667    if (chunk->name[0] == 115 && chunk->name[1] ==  84 &&     /* s  T */
668        chunk->name[2] ==  69 && chunk->name[3] ==  82)       /* E  R */
669       {
670          /* Found sTER chunk */
671          if (chunk->size != 1)
672             return (-1); /* Error return */
673
674          if (chunk->data[0] != 0 && chunk->data[0] != 1)
675             return (-1);  /* Invalid mode */
676
677          if (set_location(png_ptr, my_user_chunk_data, have_sTER))
678          {
679             my_user_chunk_data->sTER_mode=chunk->data[0];
680             return (1);
681          }
682
683          else
684             return (0); /* duplicate sTER - give it to libpng */
685       }
686
687    if (chunk->name[0] != 118 || chunk->name[1] != 112 ||    /* v  p */
688        chunk->name[2] !=  65 || chunk->name[3] != 103)      /* A  g */
689       return (0); /* Did not recognize */
690
691    /* Found ImageMagick vpAg chunk */
692
693    if (chunk->size != 9)
694       return (-1); /* Error return */
695
696    if (!set_location(png_ptr, my_user_chunk_data, have_vpAg))
697       return (0);  /* duplicate vpAg */
698
699    my_user_chunk_data->vpAg_width = png_get_uint_31(png_ptr, chunk->data);
700    my_user_chunk_data->vpAg_height = png_get_uint_31(png_ptr, chunk->data + 4);
701    my_user_chunk_data->vpAg_units = chunk->data[8];
702
703    return (1);
704 }
705
706 #ifdef PNG_WRITE_SUPPORTED
707 static void
708 write_sTER_chunk(png_structp write_ptr)
709 {
710    png_byte png_sTER[5] = {115,  84,  69,  82, '\0'};
711
712    if (verbose)
713       fprintf(STDERR, "\n stereo mode = %d\n", user_chunk_data.sTER_mode);
714
715    png_write_chunk(write_ptr, png_sTER, &user_chunk_data.sTER_mode, 1);
716 }
717
718 static void
719 write_vpAg_chunk(png_structp write_ptr)
720 {
721    png_byte png_vpAg[5] = {118, 112,  65, 103, '\0'};
722
723    png_byte vpag_chunk_data[9];
724
725    if (verbose)
726       fprintf(STDERR, " vpAg = %lu x %lu, units = %d\n",
727         (unsigned long)user_chunk_data.vpAg_width,
728         (unsigned long)user_chunk_data.vpAg_height,
729         user_chunk_data.vpAg_units);
730
731    png_save_uint_32(vpag_chunk_data, user_chunk_data.vpAg_width);
732    png_save_uint_32(vpag_chunk_data + 4, user_chunk_data.vpAg_height);
733    vpag_chunk_data[8] = user_chunk_data.vpAg_units;
734    png_write_chunk(write_ptr, png_vpAg, vpag_chunk_data, 9);
735 }
736
737 static void
738 write_chunks(png_structp write_ptr, int location)
739 {
740    int i;
741
742    /* Notice that this preserves the original chunk order, however chunks
743     * intercepted by the callback will be written *after* chunks passed to
744     * libpng.  This will actually reverse a pair of sTER chunks or a pair of
745     * vpAg chunks, resulting in an error later.  This is not worth worrying
746     * about - the chunks should not be duplicated!
747     */
748    for (i=0; i<2; ++i)
749    {
750       if (user_chunk_data.location[i] == (location | have_sTER))
751          write_sTER_chunk(write_ptr);
752
753       else if (user_chunk_data.location[i] == (location | have_vpAg))
754          write_vpAg_chunk(write_ptr);
755    }
756 }
757 #endif /* PNG_WRITE_SUPPORTED */
758 #else /* !PNG_READ_USER_CHUNKS_SUPPORTED */
759 #  define write_chunks(pp,loc) ((void)0)
760 #endif
761 /* END of code to demonstrate user chunk support */
762
763 /* START of code to check that libpng has the required text support; this only
764  * checks for the write support because if read support is missing the chunk
765  * will simply not be reported back to pngtest.
766  */
767 #ifdef PNG_TEXT_SUPPORTED
768 static void
769 pngtest_check_text_support(png_const_structp png_ptr, png_textp text_ptr,
770    int num_text)
771 {
772    while (num_text > 0)
773    {
774       switch (text_ptr[--num_text].compression)
775       {
776          case PNG_TEXT_COMPRESSION_NONE:
777             break;
778
779          case PNG_TEXT_COMPRESSION_zTXt:
780 #           ifndef PNG_WRITE_zTXt_SUPPORTED
781                ++unsupported_chunks;
782 #           endif
783             break;
784
785          case PNG_ITXT_COMPRESSION_NONE:
786          case PNG_ITXT_COMPRESSION_zTXt:
787 #           ifndef PNG_WRITE_iTXt_SUPPORTED
788                ++unsupported_chunks;
789 #           endif
790             break;
791
792          default:
793             /* This is an error */
794             png_error(png_ptr, "invalid text chunk compression field");
795             break;
796       }
797    }
798 }
799 #endif
800 /* END of code to check that libpng has the required text support */
801
802 /* Test one file */
803 static int
804 test_one_file(PNG_CONST char *inname, PNG_CONST char *outname)
805 {
806    static png_FILE_p fpin;
807    static png_FILE_p fpout;  /* "static" prevents setjmp corruption */
808    pngtest_error_parameters error_parameters;
809    png_structp read_ptr;
810    png_infop read_info_ptr, end_info_ptr;
811 #ifdef PNG_WRITE_SUPPORTED
812    png_structp write_ptr;
813    png_infop write_info_ptr;
814    png_infop write_end_info_ptr;
815 #else
816    png_structp write_ptr = NULL;
817    png_infop write_info_ptr = NULL;
818    png_infop write_end_info_ptr = NULL;
819 #endif
820    png_bytep row_buf;
821    png_uint_32 y;
822    png_uint_32 width, height;
823    int num_pass, pass;
824    int bit_depth, color_type;
825
826    row_buf = NULL;
827    error_parameters.file_name = inname;
828
829    if ((fpin = fopen(inname, "rb")) == NULL)
830    {
831       fprintf(STDERR, "Could not find input file %s\n", inname);
832       return (1);
833    }
834
835    if ((fpout = fopen(outname, "wb")) == NULL)
836    {
837       fprintf(STDERR, "Could not open output file %s\n", outname);
838       FCLOSE(fpin);
839       return (1);
840    }
841
842    pngtest_debug("Allocating read and write structures");
843 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
844    read_ptr =
845       png_create_read_struct_2(PNG_LIBPNG_VER_STRING, NULL,
846       NULL, NULL, NULL, png_debug_malloc, png_debug_free);
847 #else
848    read_ptr =
849       png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
850 #endif
851    png_set_error_fn(read_ptr, &error_parameters, pngtest_error,
852       pngtest_warning);
853
854 #ifdef PNG_WRITE_SUPPORTED
855 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
856    write_ptr =
857       png_create_write_struct_2(PNG_LIBPNG_VER_STRING, NULL,
858       NULL, NULL, NULL, png_debug_malloc, png_debug_free);
859 #else
860    write_ptr =
861       png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
862 #endif
863    png_set_error_fn(write_ptr, &error_parameters, pngtest_error,
864       pngtest_warning);
865 #endif
866    pngtest_debug("Allocating read_info, write_info and end_info structures");
867    read_info_ptr = png_create_info_struct(read_ptr);
868    end_info_ptr = png_create_info_struct(read_ptr);
869 #ifdef PNG_WRITE_SUPPORTED
870    write_info_ptr = png_create_info_struct(write_ptr);
871    write_end_info_ptr = png_create_info_struct(write_ptr);
872 #endif
873
874 #ifdef PNG_READ_USER_CHUNKS_SUPPORTED
875    init_callback_info(read_info_ptr);
876    png_set_read_user_chunk_fn(read_ptr, &user_chunk_data,
877      read_user_chunk_callback);
878 #endif
879
880 #ifdef PNG_SETJMP_SUPPORTED
881    pngtest_debug("Setting jmpbuf for read struct");
882    if (setjmp(png_jmpbuf(read_ptr)))
883    {
884       fprintf(STDERR, "%s -> %s: libpng read error\n", inname, outname);
885       png_free(read_ptr, row_buf);
886       row_buf = NULL;
887       png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
888 #ifdef PNG_WRITE_SUPPORTED
889       png_destroy_info_struct(write_ptr, &write_end_info_ptr);
890       png_destroy_write_struct(&write_ptr, &write_info_ptr);
891 #endif
892       FCLOSE(fpin);
893       FCLOSE(fpout);
894       return (1);
895    }
896
897 #ifdef PNG_WRITE_SUPPORTED
898    pngtest_debug("Setting jmpbuf for write struct");
899
900    if (setjmp(png_jmpbuf(write_ptr)))
901    {
902       fprintf(STDERR, "%s -> %s: libpng write error\n", inname, outname);
903       png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
904       png_destroy_info_struct(write_ptr, &write_end_info_ptr);
905 #ifdef PNG_WRITE_SUPPORTED
906       png_destroy_write_struct(&write_ptr, &write_info_ptr);
907 #endif
908       FCLOSE(fpin);
909       FCLOSE(fpout);
910       return (1);
911    }
912 #endif
913 #endif
914
915    if (strict)
916    {
917       /* Treat png_benign_error() as errors on read */
918       png_set_benign_errors(read_ptr, 0);
919
920 #ifdef PNG_WRITE_SUPPORTED
921       /* Treat them as errors on write */
922       png_set_benign_errors(write_ptr, 0);
923 #endif
924
925       /* if strict is not set, then app warnings and errors are treated as
926        * warnings in release builds, but not in unstable builds; this can be
927        * changed with '--relaxed'.
928        */
929    }
930
931    else if (relaxed)
932    {
933       /* Allow application (pngtest) errors and warnings to pass */
934       png_set_benign_errors(read_ptr, 1);
935
936 #ifdef PNG_WRITE_SUPPORTED
937       png_set_benign_errors(write_ptr, 1);
938 #endif
939    }
940
941    pngtest_debug("Initializing input and output streams");
942 #ifdef PNG_STDIO_SUPPORTED
943    png_init_io(read_ptr, fpin);
944 #  ifdef PNG_WRITE_SUPPORTED
945    png_init_io(write_ptr, fpout);
946 #  endif
947 #else
948    png_set_read_fn(read_ptr, (png_voidp)fpin, pngtest_read_data);
949 #  ifdef PNG_WRITE_SUPPORTED
950    png_set_write_fn(write_ptr, (png_voidp)fpout,  pngtest_write_data,
951 #    ifdef PNG_WRITE_FLUSH_SUPPORTED
952       pngtest_flush);
953 #    else
954       NULL);
955 #    endif
956 #  endif
957 #endif
958
959    if (status_dots_requested == 1)
960    {
961 #ifdef PNG_WRITE_SUPPORTED
962       png_set_write_status_fn(write_ptr, write_row_callback);
963 #endif
964       png_set_read_status_fn(read_ptr, read_row_callback);
965    }
966
967    else
968    {
969 #ifdef PNG_WRITE_SUPPORTED
970       png_set_write_status_fn(write_ptr, NULL);
971 #endif
972       png_set_read_status_fn(read_ptr, NULL);
973    }
974
975 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
976    {
977       int i;
978
979       for (i = 0; i<256; i++)
980          filters_used[i] = 0;
981
982       png_set_read_user_transform_fn(read_ptr, count_filters);
983    }
984 #endif
985 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
986    zero_samples = 0;
987    png_set_write_user_transform_fn(write_ptr, count_zero_samples);
988 #endif
989
990 #ifdef PNG_SET_UNKNOWN_CHUNKS_SUPPORTED
991    /* Preserve all the unknown chunks, if possible.  If this is disabled then,
992     * even if the png_{get,set}_unknown_chunks stuff is enabled, we can't use
993     * libpng to *save* the unknown chunks on read (because we can't switch the
994     * save option on!)
995     *
996     * Notice that if SET_UNKNOWN_CHUNKS is *not* supported read will discard all
997     * unknown chunks and write will write them all.
998     */
999 #ifdef PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
1000    png_set_keep_unknown_chunks(read_ptr, PNG_HANDLE_CHUNK_ALWAYS,
1001       NULL, 0);
1002 #endif
1003 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1004    png_set_keep_unknown_chunks(write_ptr, PNG_HANDLE_CHUNK_ALWAYS,
1005       NULL, 0);
1006 #endif
1007 #endif
1008
1009    pngtest_debug("Reading info struct");
1010    png_read_info(read_ptr, read_info_ptr);
1011
1012 #ifdef PNG_READ_USER_CHUNKS_SUPPORTED
1013    /* This is a bit of a hack; there is no obvious way in the callback function
1014     * to determine that the chunks before the first IDAT have been read, so
1015     * remove the info_ptr (which is only used to determine position relative to
1016     * PLTE) here to indicate that we are after the IDAT.
1017     */
1018    user_chunk_data.info_ptr = NULL;
1019 #endif
1020
1021    pngtest_debug("Transferring info struct");
1022    {
1023       int interlace_type, compression_type, filter_type;
1024
1025       if (png_get_IHDR(read_ptr, read_info_ptr, &width, &height, &bit_depth,
1026           &color_type, &interlace_type, &compression_type, &filter_type))
1027       {
1028          png_set_IHDR(write_ptr, write_info_ptr, width, height, bit_depth,
1029 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1030             color_type, interlace_type, compression_type, filter_type);
1031 #else
1032             color_type, PNG_INTERLACE_NONE, compression_type, filter_type);
1033 #endif
1034       }
1035    }
1036 #ifdef PNG_FIXED_POINT_SUPPORTED
1037 #ifdef PNG_cHRM_SUPPORTED
1038    {
1039       png_fixed_point white_x, white_y, red_x, red_y, green_x, green_y, blue_x,
1040          blue_y;
1041
1042       if (png_get_cHRM_fixed(read_ptr, read_info_ptr, &white_x, &white_y,
1043          &red_x, &red_y, &green_x, &green_y, &blue_x, &blue_y))
1044       {
1045          png_set_cHRM_fixed(write_ptr, write_info_ptr, white_x, white_y, red_x,
1046             red_y, green_x, green_y, blue_x, blue_y);
1047       }
1048    }
1049 #endif
1050 #ifdef PNG_gAMA_SUPPORTED
1051    {
1052       png_fixed_point gamma;
1053
1054       if (png_get_gAMA_fixed(read_ptr, read_info_ptr, &gamma))
1055          png_set_gAMA_fixed(write_ptr, write_info_ptr, gamma);
1056    }
1057 #endif
1058 #else /* Use floating point versions */
1059 #ifdef PNG_FLOATING_POINT_SUPPORTED
1060 #ifdef PNG_cHRM_SUPPORTED
1061    {
1062       double white_x, white_y, red_x, red_y, green_x, green_y, blue_x,
1063          blue_y;
1064
1065       if (png_get_cHRM(read_ptr, read_info_ptr, &white_x, &white_y, &red_x,
1066          &red_y, &green_x, &green_y, &blue_x, &blue_y))
1067       {
1068          png_set_cHRM(write_ptr, write_info_ptr, white_x, white_y, red_x,
1069             red_y, green_x, green_y, blue_x, blue_y);
1070       }
1071    }
1072 #endif
1073 #ifdef PNG_gAMA_SUPPORTED
1074    {
1075       double gamma;
1076
1077       if (png_get_gAMA(read_ptr, read_info_ptr, &gamma))
1078          png_set_gAMA(write_ptr, write_info_ptr, gamma);
1079    }
1080 #endif
1081 #endif /* Floating point */
1082 #endif /* Fixed point */
1083 #ifdef PNG_iCCP_SUPPORTED
1084    {
1085       png_charp name;
1086       png_bytep profile;
1087       png_uint_32 proflen;
1088       int compression_type;
1089
1090       if (png_get_iCCP(read_ptr, read_info_ptr, &name, &compression_type,
1091                       &profile, &proflen))
1092       {
1093          png_set_iCCP(write_ptr, write_info_ptr, name, compression_type,
1094                       profile, proflen);
1095       }
1096    }
1097 #endif
1098 #ifdef PNG_sRGB_SUPPORTED
1099    {
1100       int intent;
1101
1102       if (png_get_sRGB(read_ptr, read_info_ptr, &intent))
1103          png_set_sRGB(write_ptr, write_info_ptr, intent);
1104    }
1105 #endif
1106    {
1107       png_colorp palette;
1108       int num_palette;
1109
1110       if (png_get_PLTE(read_ptr, read_info_ptr, &palette, &num_palette))
1111          png_set_PLTE(write_ptr, write_info_ptr, palette, num_palette);
1112    }
1113 #ifdef PNG_bKGD_SUPPORTED
1114    {
1115       png_color_16p background;
1116
1117       if (png_get_bKGD(read_ptr, read_info_ptr, &background))
1118       {
1119          png_set_bKGD(write_ptr, write_info_ptr, background);
1120       }
1121    }
1122 #endif
1123 #ifdef PNG_hIST_SUPPORTED
1124    {
1125       png_uint_16p hist;
1126
1127       if (png_get_hIST(read_ptr, read_info_ptr, &hist))
1128          png_set_hIST(write_ptr, write_info_ptr, hist);
1129    }
1130 #endif
1131 #ifdef PNG_oFFs_SUPPORTED
1132    {
1133       png_int_32 offset_x, offset_y;
1134       int unit_type;
1135
1136       if (png_get_oFFs(read_ptr, read_info_ptr, &offset_x, &offset_y,
1137           &unit_type))
1138       {
1139          png_set_oFFs(write_ptr, write_info_ptr, offset_x, offset_y, unit_type);
1140       }
1141    }
1142 #endif
1143 #ifdef PNG_pCAL_SUPPORTED
1144    {
1145       png_charp purpose, units;
1146       png_charpp params;
1147       png_int_32 X0, X1;
1148       int type, nparams;
1149
1150       if (png_get_pCAL(read_ptr, read_info_ptr, &purpose, &X0, &X1, &type,
1151          &nparams, &units, &params))
1152       {
1153          png_set_pCAL(write_ptr, write_info_ptr, purpose, X0, X1, type,
1154             nparams, units, params);
1155       }
1156    }
1157 #endif
1158 #ifdef PNG_pHYs_SUPPORTED
1159    {
1160       png_uint_32 res_x, res_y;
1161       int unit_type;
1162
1163       if (png_get_pHYs(read_ptr, read_info_ptr, &res_x, &res_y, &unit_type))
1164          png_set_pHYs(write_ptr, write_info_ptr, res_x, res_y, unit_type);
1165    }
1166 #endif
1167 #ifdef PNG_sBIT_SUPPORTED
1168    {
1169       png_color_8p sig_bit;
1170
1171       if (png_get_sBIT(read_ptr, read_info_ptr, &sig_bit))
1172          png_set_sBIT(write_ptr, write_info_ptr, sig_bit);
1173    }
1174 #endif
1175 #ifdef PNG_sCAL_SUPPORTED
1176 #ifdef PNG_FLOATING_POINT_SUPPORTED
1177    {
1178       int unit;
1179       double scal_width, scal_height;
1180
1181       if (png_get_sCAL(read_ptr, read_info_ptr, &unit, &scal_width,
1182          &scal_height))
1183       {
1184          png_set_sCAL(write_ptr, write_info_ptr, unit, scal_width, scal_height);
1185       }
1186    }
1187 #else
1188 #ifdef PNG_FIXED_POINT_SUPPORTED
1189    {
1190       int unit;
1191       png_charp scal_width, scal_height;
1192
1193       if (png_get_sCAL_s(read_ptr, read_info_ptr, &unit, &scal_width,
1194           &scal_height))
1195       {
1196          png_set_sCAL_s(write_ptr, write_info_ptr, unit, scal_width,
1197              scal_height);
1198       }
1199    }
1200 #endif
1201 #endif
1202 #endif
1203 #ifdef PNG_TEXT_SUPPORTED
1204    {
1205       png_textp text_ptr;
1206       int num_text;
1207
1208       if (png_get_text(read_ptr, read_info_ptr, &text_ptr, &num_text) > 0)
1209       {
1210          pngtest_debug1("Handling %d iTXt/tEXt/zTXt chunks", num_text);
1211
1212          pngtest_check_text_support(read_ptr, text_ptr, num_text);
1213
1214          if (verbose)
1215          {
1216             int i;
1217
1218             printf("\n");
1219             for (i=0; i<num_text; i++)
1220             {
1221                printf("   Text compression[%d]=%d\n",
1222                      i, text_ptr[i].compression);
1223             }
1224          }
1225
1226          png_set_text(write_ptr, write_info_ptr, text_ptr, num_text);
1227       }
1228    }
1229 #endif
1230 #ifdef PNG_tIME_SUPPORTED
1231    {
1232       png_timep mod_time;
1233
1234       if (png_get_tIME(read_ptr, read_info_ptr, &mod_time))
1235       {
1236          png_set_tIME(write_ptr, write_info_ptr, mod_time);
1237 #ifdef PNG_TIME_RFC1123_SUPPORTED
1238          if (png_convert_to_rfc1123_buffer(tIME_string, mod_time))
1239             tIME_string[(sizeof tIME_string) - 1] = '\0';
1240
1241          else
1242          {
1243             strncpy(tIME_string, "*** invalid time ***", (sizeof tIME_string));
1244             tIME_string[(sizeof tIME_string) - 1] = '\0';
1245          }
1246
1247          tIME_chunk_present++;
1248 #endif /* PNG_TIME_RFC1123_SUPPORTED */
1249       }
1250    }
1251 #endif
1252 #ifdef PNG_tRNS_SUPPORTED
1253    {
1254       png_bytep trans_alpha;
1255       int num_trans;
1256       png_color_16p trans_color;
1257
1258       if (png_get_tRNS(read_ptr, read_info_ptr, &trans_alpha, &num_trans,
1259          &trans_color))
1260       {
1261          int sample_max = (1 << bit_depth);
1262          /* libpng doesn't reject a tRNS chunk with out-of-range samples */
1263          if (!((color_type == PNG_COLOR_TYPE_GRAY &&
1264              (int)trans_color->gray > sample_max) ||
1265              (color_type == PNG_COLOR_TYPE_RGB &&
1266              ((int)trans_color->red > sample_max ||
1267              (int)trans_color->green > sample_max ||
1268              (int)trans_color->blue > sample_max))))
1269             png_set_tRNS(write_ptr, write_info_ptr, trans_alpha, num_trans,
1270                trans_color);
1271       }
1272    }
1273 #endif
1274 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1275    {
1276       png_unknown_chunkp unknowns;
1277       int num_unknowns = png_get_unknown_chunks(read_ptr, read_info_ptr,
1278          &unknowns);
1279
1280       if (num_unknowns)
1281       {
1282          png_set_unknown_chunks(write_ptr, write_info_ptr, unknowns,
1283            num_unknowns);
1284 #if PNG_LIBPNG_VER < 10600
1285          /* Copy the locations from the read_info_ptr.  The automatically
1286           * generated locations in write_end_info_ptr are wrong prior to 1.6.0
1287           * because they are reset from the write pointer (removed in 1.6.0).
1288           */
1289          {
1290             int i;
1291             for (i = 0; i < num_unknowns; i++)
1292               png_set_unknown_chunk_location(write_ptr, write_info_ptr, i,
1293                 unknowns[i].location);
1294          }
1295 #endif
1296       }
1297    }
1298 #endif
1299
1300 #ifdef PNG_WRITE_SUPPORTED
1301    pngtest_debug("Writing info struct");
1302
1303    /* Write the info in two steps so that if we write the 'unknown' chunks here
1304     * they go to the correct place.
1305     */
1306    png_write_info_before_PLTE(write_ptr, write_info_ptr);
1307
1308    write_chunks(write_ptr, before_PLTE); /* before PLTE */
1309
1310    png_write_info(write_ptr, write_info_ptr);
1311
1312    write_chunks(write_ptr, before_IDAT); /* after PLTE */
1313 #endif
1314
1315 #ifdef SINGLE_ROWBUF_ALLOC
1316    pngtest_debug("Allocating row buffer...");
1317    row_buf = (png_bytep)png_malloc(read_ptr,
1318       png_get_rowbytes(read_ptr, read_info_ptr));
1319
1320    pngtest_debug1("\t0x%08lx", (unsigned long)row_buf);
1321 #endif /* SINGLE_ROWBUF_ALLOC */
1322    pngtest_debug("Writing row data");
1323
1324 #if defined(PNG_READ_INTERLACING_SUPPORTED) || \
1325   defined(PNG_WRITE_INTERLACING_SUPPORTED)
1326    num_pass = png_set_interlace_handling(read_ptr);
1327 #  ifdef PNG_WRITE_SUPPORTED
1328    png_set_interlace_handling(write_ptr);
1329 #  endif
1330 #else
1331    num_pass = 1;
1332 #endif
1333
1334 #ifdef PNGTEST_TIMING
1335    t_stop = (float)clock();
1336    t_misc += (t_stop - t_start);
1337    t_start = t_stop;
1338 #endif
1339    for (pass = 0; pass < num_pass; pass++)
1340    {
1341       pngtest_debug1("Writing row data for pass %d", pass);
1342       for (y = 0; y < height; y++)
1343       {
1344 #ifndef SINGLE_ROWBUF_ALLOC
1345          pngtest_debug2("Allocating row buffer (pass %d, y = %u)...", pass, y);
1346          row_buf = (png_bytep)png_malloc(read_ptr,
1347             png_get_rowbytes(read_ptr, read_info_ptr));
1348
1349          pngtest_debug2("\t0x%08lx (%u bytes)", (unsigned long)row_buf,
1350             png_get_rowbytes(read_ptr, read_info_ptr));
1351
1352 #endif /* !SINGLE_ROWBUF_ALLOC */
1353          png_read_rows(read_ptr, (png_bytepp)&row_buf, NULL, 1);
1354
1355 #ifdef PNG_WRITE_SUPPORTED
1356 #ifdef PNGTEST_TIMING
1357          t_stop = (float)clock();
1358          t_decode += (t_stop - t_start);
1359          t_start = t_stop;
1360 #endif
1361          png_write_rows(write_ptr, (png_bytepp)&row_buf, 1);
1362 #ifdef PNGTEST_TIMING
1363          t_stop = (float)clock();
1364          t_encode += (t_stop - t_start);
1365          t_start = t_stop;
1366 #endif
1367 #endif /* PNG_WRITE_SUPPORTED */
1368
1369 #ifndef SINGLE_ROWBUF_ALLOC
1370          pngtest_debug2("Freeing row buffer (pass %d, y = %u)", pass, y);
1371          png_free(read_ptr, row_buf);
1372          row_buf = NULL;
1373 #endif /* !SINGLE_ROWBUF_ALLOC */
1374       }
1375    }
1376
1377 #ifdef PNG_READ_UNKNOWN_CHUNKS_SUPPORTED
1378    png_free_data(read_ptr, read_info_ptr, PNG_FREE_UNKN, -1);
1379 #endif
1380 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1381    png_free_data(write_ptr, write_info_ptr, PNG_FREE_UNKN, -1);
1382 #endif
1383
1384    pngtest_debug("Reading and writing end_info data");
1385
1386    png_read_end(read_ptr, end_info_ptr);
1387 #ifdef PNG_TEXT_SUPPORTED
1388    {
1389       png_textp text_ptr;
1390       int num_text;
1391
1392       if (png_get_text(read_ptr, end_info_ptr, &text_ptr, &num_text) > 0)
1393       {
1394          pngtest_debug1("Handling %d iTXt/tEXt/zTXt chunks", num_text);
1395
1396          pngtest_check_text_support(read_ptr, text_ptr, num_text);
1397
1398          if (verbose)
1399          {
1400             int i;
1401
1402             printf("\n");
1403             for (i=0; i<num_text; i++)
1404             {
1405                printf("   Text compression[%d]=%d\n",
1406                      i, text_ptr[i].compression);
1407             }
1408          }
1409
1410          png_set_text(write_ptr, write_end_info_ptr, text_ptr, num_text);
1411       }
1412    }
1413 #endif
1414 #ifdef PNG_tIME_SUPPORTED
1415    {
1416       png_timep mod_time;
1417
1418       if (png_get_tIME(read_ptr, end_info_ptr, &mod_time))
1419       {
1420          png_set_tIME(write_ptr, write_end_info_ptr, mod_time);
1421 #ifdef PNG_TIME_RFC1123_SUPPORTED
1422          if (png_convert_to_rfc1123_buffer(tIME_string, mod_time))
1423             tIME_string[(sizeof tIME_string) - 1] = '\0';
1424
1425          else
1426          {
1427             strncpy(tIME_string, "*** invalid time ***", sizeof tIME_string);
1428             tIME_string[(sizeof tIME_string)-1] = '\0';
1429          }
1430
1431          tIME_chunk_present++;
1432 #endif /* PNG_TIME_RFC1123_SUPPORTED */
1433       }
1434    }
1435 #endif
1436 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
1437    {
1438       png_unknown_chunkp unknowns;
1439       int num_unknowns = png_get_unknown_chunks(read_ptr, end_info_ptr,
1440          &unknowns);
1441
1442       if (num_unknowns)
1443       {
1444          png_set_unknown_chunks(write_ptr, write_end_info_ptr, unknowns,
1445            num_unknowns);
1446 #if PNG_LIBPNG_VER < 10600
1447          /* Copy the locations from the read_info_ptr.  The automatically
1448           * generated locations in write_end_info_ptr are wrong prior to 1.6.0
1449           * because they are reset from the write pointer (removed in 1.6.0).
1450           */
1451          {
1452             int i;
1453             for (i = 0; i < num_unknowns; i++)
1454               png_set_unknown_chunk_location(write_ptr, write_end_info_ptr, i,
1455                 unknowns[i].location);
1456          }
1457 #endif
1458       }
1459    }
1460 #endif
1461
1462 #ifdef PNG_WRITE_SUPPORTED
1463 #ifdef PNG_WRITE_CUSTOMIZE_ZTXT_COMPRESSION_SUPPORTED
1464    /* Normally one would use Z_DEFAULT_STRATEGY for text compression.
1465     * This is here just to make pngtest replicate the results from libpng
1466     * versions prior to 1.5.4, and to test this new API.
1467     */
1468    png_set_text_compression_strategy(write_ptr, Z_FILTERED);
1469 #endif
1470
1471    /* When the unknown vpAg/sTER chunks are written by pngtest the only way to
1472     * do it is to write them *before* calling png_write_end.  When unknown
1473     * chunks are written by libpng, however, they are written just before IEND.
1474     * There seems to be no way round this, however vpAg/sTER are not expected
1475     * after IDAT.
1476     */
1477    write_chunks(write_ptr, after_IDAT);
1478
1479    png_write_end(write_ptr, write_end_info_ptr);
1480 #endif
1481
1482 #ifdef PNG_EASY_ACCESS_SUPPORTED
1483    if (verbose)
1484    {
1485       png_uint_32 iwidth, iheight;
1486       iwidth = png_get_image_width(write_ptr, write_info_ptr);
1487       iheight = png_get_image_height(write_ptr, write_info_ptr);
1488       fprintf(STDERR, "\n Image width = %lu, height = %lu\n",
1489          (unsigned long)iwidth, (unsigned long)iheight);
1490    }
1491 #endif
1492
1493    pngtest_debug("Destroying data structs");
1494 #ifdef SINGLE_ROWBUF_ALLOC
1495    pngtest_debug("destroying row_buf for read_ptr");
1496    png_free(read_ptr, row_buf);
1497    row_buf = NULL;
1498 #endif /* SINGLE_ROWBUF_ALLOC */
1499    pngtest_debug("destroying read_ptr, read_info_ptr, end_info_ptr");
1500    png_destroy_read_struct(&read_ptr, &read_info_ptr, &end_info_ptr);
1501 #ifdef PNG_WRITE_SUPPORTED
1502    pngtest_debug("destroying write_end_info_ptr");
1503    png_destroy_info_struct(write_ptr, &write_end_info_ptr);
1504    pngtest_debug("destroying write_ptr, write_info_ptr");
1505    png_destroy_write_struct(&write_ptr, &write_info_ptr);
1506 #endif
1507    pngtest_debug("Destruction complete.");
1508
1509    FCLOSE(fpin);
1510    FCLOSE(fpout);
1511
1512    /* Summarize any warnings or errors and in 'strict' mode fail the test.
1513     * Unsupported chunks can result in warnings, in that case ignore the strict
1514     * setting, otherwise fail the test on warnings as well as errors.
1515     */
1516    if (error_count > 0)
1517    {
1518       /* We don't really expect to get here because of the setjmp handling
1519        * above, but this is safe.
1520        */
1521       fprintf(STDERR, "\n  %s: %d libpng errors found (%d warnings)",
1522          inname, error_count, warning_count);
1523
1524       if (strict != 0)
1525          return (1);
1526    }
1527
1528 #  ifdef PNG_WRITE_SUPPORTED
1529       /* If there we no write support nothing was written! */
1530       else if (unsupported_chunks > 0)
1531       {
1532          fprintf(STDERR, "\n  %s: unsupported chunks (%d)%s",
1533             inname, unsupported_chunks, strict ? ": IGNORED --strict!" : "");
1534       }
1535 #  endif
1536
1537    else if (warning_count > 0)
1538    {
1539       fprintf(STDERR, "\n  %s: %d libpng warnings found",
1540          inname, warning_count);
1541
1542       if (strict != 0)
1543          return (1);
1544    }
1545
1546    pngtest_debug("Opening files for comparison");
1547    if ((fpin = fopen(inname, "rb")) == NULL)
1548    {
1549       fprintf(STDERR, "Could not find file %s\n", inname);
1550       return (1);
1551    }
1552
1553    if ((fpout = fopen(outname, "rb")) == NULL)
1554    {
1555       fprintf(STDERR, "Could not find file %s\n", outname);
1556       FCLOSE(fpin);
1557       return (1);
1558    }
1559
1560 #ifdef PNG_WRITE_SUPPORTED /* else nothing was written */
1561    {
1562       int wrote_question = 0;
1563
1564       for (;;)
1565       {
1566          png_size_t num_in, num_out;
1567          char inbuf[256], outbuf[256];
1568
1569
1570          num_in = fread(inbuf, 1, sizeof inbuf, fpin);
1571          num_out = fread(outbuf, 1, sizeof outbuf, fpout);
1572
1573          if (num_in != num_out)
1574          {
1575             fprintf(STDERR, "\nFiles %s and %s are of a different size\n",
1576                     inname, outname);
1577
1578             if (wrote_question == 0 && unsupported_chunks == 0)
1579             {
1580                fprintf(STDERR,
1581          "   Was %s written with the same maximum IDAT chunk size (%d bytes),",
1582                  inname, PNG_ZBUF_SIZE);
1583                fprintf(STDERR,
1584                  "\n   filtering heuristic (libpng default), compression");
1585                fprintf(STDERR,
1586                  " level (zlib default),\n   and zlib version (%s)?\n\n",
1587                  ZLIB_VERSION);
1588                wrote_question = 1;
1589             }
1590
1591             FCLOSE(fpin);
1592             FCLOSE(fpout);
1593
1594             if (strict != 0 && unsupported_chunks == 0)
1595               return (1);
1596
1597             else
1598               return (0);
1599          }
1600
1601          if (!num_in)
1602             break;
1603
1604          if (memcmp(inbuf, outbuf, num_in))
1605          {
1606             fprintf(STDERR, "\nFiles %s and %s are different\n", inname,
1607                outname);
1608
1609             if (wrote_question == 0 && unsupported_chunks == 0)
1610             {
1611                fprintf(STDERR,
1612          "   Was %s written with the same maximum IDAT chunk size (%d bytes),",
1613                     inname, PNG_ZBUF_SIZE);
1614                fprintf(STDERR,
1615                  "\n   filtering heuristic (libpng default), compression");
1616                fprintf(STDERR,
1617                  " level (zlib default),\n   and zlib version (%s)?\n\n",
1618                  ZLIB_VERSION);
1619                wrote_question = 1;
1620             }
1621
1622             FCLOSE(fpin);
1623             FCLOSE(fpout);
1624
1625             /* NOTE: the unsupported_chunks escape is permitted here because
1626              * unsupported text chunk compression will result in the compression
1627              * mode being changed (to NONE) yet, in the test case, the result
1628              * can be exactly the same size!
1629              */
1630             if (strict != 0 && unsupported_chunks == 0)
1631               return (1);
1632
1633             else
1634               return (0);
1635          }
1636       }
1637    }
1638 #endif /* PNG_WRITE_SUPPORTED */
1639
1640    FCLOSE(fpin);
1641    FCLOSE(fpout);
1642
1643    return (0);
1644 }
1645
1646 /* Input and output filenames */
1647 #ifdef RISCOS
1648 static PNG_CONST char *inname = "pngtest/png";
1649 static PNG_CONST char *outname = "pngout/png";
1650 #else
1651 static PNG_CONST char *inname = "pngtest.png";
1652 static PNG_CONST char *outname = "pngout.png";
1653 #endif
1654
1655 int
1656 main(int argc, char *argv[])
1657 {
1658    int multiple = 0;
1659    int ierror = 0;
1660
1661    fprintf(STDERR, "\n Testing libpng version %s\n", PNG_LIBPNG_VER_STRING);
1662    fprintf(STDERR, "   with zlib   version %s\n", ZLIB_VERSION);
1663    fprintf(STDERR, "%s", png_get_copyright(NULL));
1664    /* Show the version of libpng used in building the library */
1665    fprintf(STDERR, " library (%lu):%s",
1666       (unsigned long)png_access_version_number(),
1667       png_get_header_version(NULL));
1668
1669    /* Show the version of libpng used in building the application */
1670    fprintf(STDERR, " pngtest (%lu):%s", (unsigned long)PNG_LIBPNG_VER,
1671       PNG_HEADER_VERSION_STRING);
1672
1673    /* Do some consistency checking on the memory allocation settings, I'm
1674     * not sure this matters, but it is nice to know, the first of these
1675     * tests should be impossible because of the way the macros are set
1676     * in pngconf.h
1677     */
1678 #if defined(MAXSEG_64K) && !defined(PNG_MAX_MALLOC_64K)
1679       fprintf(STDERR, " NOTE: Zlib compiled for max 64k, libpng not\n");
1680 #endif
1681    /* I think the following can happen. */
1682 #if !defined(MAXSEG_64K) && defined(PNG_MAX_MALLOC_64K)
1683       fprintf(STDERR, " NOTE: libpng compiled for max 64k, zlib not\n");
1684 #endif
1685
1686    if (strcmp(png_libpng_ver, PNG_LIBPNG_VER_STRING))
1687    {
1688       fprintf(STDERR,
1689          "Warning: versions are different between png.h and png.c\n");
1690       fprintf(STDERR, "  png.h version: %s\n", PNG_LIBPNG_VER_STRING);
1691       fprintf(STDERR, "  png.c version: %s\n\n", png_libpng_ver);
1692       ++ierror;
1693    }
1694
1695    if (argc > 1)
1696    {
1697       if (strcmp(argv[1], "-m") == 0)
1698       {
1699          multiple = 1;
1700          status_dots_requested = 0;
1701       }
1702
1703       else if (strcmp(argv[1], "-mv") == 0 ||
1704                strcmp(argv[1], "-vm") == 0 )
1705       {
1706          multiple = 1;
1707          verbose = 1;
1708          status_dots_requested = 1;
1709       }
1710
1711       else if (strcmp(argv[1], "-v") == 0)
1712       {
1713          verbose = 1;
1714          status_dots_requested = 1;
1715          inname = argv[2];
1716       }
1717
1718       else if (strcmp(argv[1], "--strict") == 0)
1719       {
1720          status_dots_requested = 0;
1721          verbose = 1;
1722          inname = argv[2];
1723          strict++;
1724          relaxed = 0;
1725       }
1726
1727       else if (strcmp(argv[1], "--relaxed") == 0)
1728       {
1729          status_dots_requested = 0;
1730          verbose = 1;
1731          inname = argv[2];
1732          strict = 0;
1733          relaxed++;
1734       }
1735
1736       else
1737       {
1738          inname = argv[1];
1739          status_dots_requested = 0;
1740       }
1741    }
1742
1743    if (!multiple && argc == 3 + verbose)
1744      outname = argv[2 + verbose];
1745
1746    if ((!multiple && argc > 3 + verbose) || (multiple && argc < 2))
1747    {
1748      fprintf(STDERR,
1749        "usage: %s [infile.png] [outfile.png]\n\t%s -m {infile.png}\n",
1750         argv[0], argv[0]);
1751      fprintf(STDERR,
1752        "  reads/writes one PNG file (without -m) or multiple files (-m)\n");
1753      fprintf(STDERR,
1754        "  with -m %s is used as a temporary file\n", outname);
1755      exit(1);
1756    }
1757
1758    if (multiple)
1759    {
1760       int i;
1761 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1762       int allocation_now = current_allocation;
1763 #endif
1764       for (i=2; i<argc; ++i)
1765       {
1766          int kerror;
1767          fprintf(STDERR, "\n Testing %s:", argv[i]);
1768          kerror = test_one_file(argv[i], outname);
1769          if (kerror == 0)
1770          {
1771 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1772             int k;
1773 #endif
1774 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
1775             fprintf(STDERR, "\n PASS (%lu zero samples)\n",
1776                (unsigned long)zero_samples);
1777 #else
1778             fprintf(STDERR, " PASS\n");
1779 #endif
1780 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1781             for (k = 0; k<256; k++)
1782                if (filters_used[k])
1783                   fprintf(STDERR, " Filter %d was used %lu times\n",
1784                      k, (unsigned long)filters_used[k]);
1785 #endif
1786 #ifdef PNG_TIME_RFC1123_SUPPORTED
1787          if (tIME_chunk_present != 0)
1788             fprintf(STDERR, " tIME = %s\n", tIME_string);
1789
1790          tIME_chunk_present = 0;
1791 #endif /* PNG_TIME_RFC1123_SUPPORTED */
1792          }
1793
1794          else
1795          {
1796             fprintf(STDERR, " FAIL\n");
1797             ierror += kerror;
1798          }
1799 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1800          if (allocation_now != current_allocation)
1801             fprintf(STDERR, "MEMORY ERROR: %d bytes lost\n",
1802                current_allocation - allocation_now);
1803
1804          if (current_allocation != 0)
1805          {
1806             memory_infop pinfo = pinformation;
1807
1808             fprintf(STDERR, "MEMORY ERROR: %d bytes still allocated\n",
1809                current_allocation);
1810
1811             while (pinfo != NULL)
1812             {
1813                fprintf(STDERR, " %lu bytes at %x\n",
1814                  (unsigned long)pinfo->size,
1815                  (unsigned int)pinfo->pointer);
1816                pinfo = pinfo->next;
1817             }
1818          }
1819 #endif
1820       }
1821 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1822          fprintf(STDERR, " Current memory allocation: %10d bytes\n",
1823             current_allocation);
1824          fprintf(STDERR, " Maximum memory allocation: %10d bytes\n",
1825             maximum_allocation);
1826          fprintf(STDERR, " Total   memory allocation: %10d bytes\n",
1827             total_allocation);
1828          fprintf(STDERR, "     Number of allocations: %10d\n",
1829             num_allocations);
1830 #endif
1831    }
1832
1833    else
1834    {
1835       int i;
1836       for (i = 0; i<3; ++i)
1837       {
1838          int kerror;
1839 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1840          int allocation_now = current_allocation;
1841 #endif
1842          if (i == 1)
1843             status_dots_requested = 1;
1844
1845          else if (verbose == 0)
1846             status_dots_requested = 0;
1847
1848          if (i == 0 || verbose == 1 || ierror != 0)
1849             fprintf(STDERR, "\n Testing %s:", inname);
1850
1851          kerror = test_one_file(inname, outname);
1852
1853          if (kerror == 0)
1854          {
1855             if (verbose == 1 || i == 2)
1856             {
1857 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1858                 int k;
1859 #endif
1860 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
1861                 fprintf(STDERR, "\n PASS (%lu zero samples)\n",
1862                    (unsigned long)zero_samples);
1863 #else
1864                 fprintf(STDERR, " PASS\n");
1865 #endif
1866 #ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1867                 for (k = 0; k<256; k++)
1868                    if (filters_used[k])
1869                       fprintf(STDERR, " Filter %d was used %lu times\n",
1870                          k, (unsigned long)filters_used[k]);
1871 #endif
1872 #ifdef PNG_TIME_RFC1123_SUPPORTED
1873              if (tIME_chunk_present != 0)
1874                 fprintf(STDERR, " tIME = %s\n", tIME_string);
1875 #endif /* PNG_TIME_RFC1123_SUPPORTED */
1876             }
1877          }
1878
1879          else
1880          {
1881             if (verbose == 0 && i != 2)
1882                fprintf(STDERR, "\n Testing %s:", inname);
1883
1884             fprintf(STDERR, " FAIL\n");
1885             ierror += kerror;
1886          }
1887 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1888          if (allocation_now != current_allocation)
1889              fprintf(STDERR, "MEMORY ERROR: %d bytes lost\n",
1890                current_allocation - allocation_now);
1891
1892          if (current_allocation != 0)
1893          {
1894              memory_infop pinfo = pinformation;
1895
1896              fprintf(STDERR, "MEMORY ERROR: %d bytes still allocated\n",
1897                 current_allocation);
1898
1899              while (pinfo != NULL)
1900              {
1901                 fprintf(STDERR, " %lu bytes at %x\n",
1902                    (unsigned long)pinfo->size, (unsigned int)pinfo->pointer);
1903                 pinfo = pinfo->next;
1904              }
1905           }
1906 #endif
1907        }
1908 #if defined(PNG_USER_MEM_SUPPORTED) && PNG_DEBUG
1909        fprintf(STDERR, " Current memory allocation: %10d bytes\n",
1910           current_allocation);
1911        fprintf(STDERR, " Maximum memory allocation: %10d bytes\n",
1912           maximum_allocation);
1913        fprintf(STDERR, " Total   memory allocation: %10d bytes\n",
1914           total_allocation);
1915        fprintf(STDERR, "     Number of allocations: %10d\n",
1916             num_allocations);
1917 #endif
1918    }
1919
1920 #ifdef PNGTEST_TIMING
1921    t_stop = (float)clock();
1922    t_misc += (t_stop - t_start);
1923    t_start = t_stop;
1924    fprintf(STDERR, " CPU time used = %.3f seconds",
1925       (t_misc+t_decode+t_encode)/(float)CLOCKS_PER_SEC);
1926    fprintf(STDERR, " (decoding %.3f,\n",
1927       t_decode/(float)CLOCKS_PER_SEC);
1928    fprintf(STDERR, "        encoding %.3f ,",
1929       t_encode/(float)CLOCKS_PER_SEC);
1930    fprintf(STDERR, " other %.3f seconds)\n\n",
1931       t_misc/(float)CLOCKS_PER_SEC);
1932 #endif
1933
1934    if (ierror == 0)
1935       fprintf(STDERR, " libpng passes test\n");
1936
1937    else
1938       fprintf(STDERR, " libpng FAILS test\n");
1939
1940    return (int)(ierror != 0);
1941 }
1942 #else
1943 int
1944 main(void)
1945 {
1946    fprintf(STDERR,
1947       " test ignored because libpng was not built with read support\n");
1948    return 0;
1949 }
1950 #endif
1951
1952 /* Generate a compiler error if there is an old png.h in the search path. */
1953 typedef png_libpng_version_1_6_1 Your_png_h_is_not_version_1_6_1;