Merge branch 'upstream' into tizen
[platform/upstream/libpng.git] / pngwutil.c
1
2 /* pngwutil.c - utilities to write a PNG file
3  *
4  * Copyright (c) 2018-2022 Cosmin Truta
5  * Copyright (c) 1998-2002,2004,2006-2018 Glenn Randers-Pehrson
6  * Copyright (c) 1996-1997 Andreas Dilger
7  * 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 #include "pngpriv.h"
15
16 #ifdef PNG_WRITE_SUPPORTED
17
18 #ifdef PNG_WRITE_INT_FUNCTIONS_SUPPORTED
19 /* Place a 32-bit number into a buffer in PNG byte order.  We work
20  * with unsigned numbers for convenience, although one supported
21  * ancillary chunk uses signed (two's complement) numbers.
22  */
23 void PNGAPI
24 png_save_uint_32(png_bytep buf, png_uint_32 i)
25 {
26    buf[0] = (png_byte)((i >> 24) & 0xffU);
27    buf[1] = (png_byte)((i >> 16) & 0xffU);
28    buf[2] = (png_byte)((i >>  8) & 0xffU);
29    buf[3] = (png_byte)( i        & 0xffU);
30 }
31
32 /* Place a 16-bit number into a buffer in PNG byte order.
33  * The parameter is declared unsigned int, not png_uint_16,
34  * just to avoid potential problems on pre-ANSI C compilers.
35  */
36 void PNGAPI
37 png_save_uint_16(png_bytep buf, unsigned int i)
38 {
39    buf[0] = (png_byte)((i >> 8) & 0xffU);
40    buf[1] = (png_byte)( i       & 0xffU);
41 }
42 #endif
43
44 /* Simple function to write the signature.  If we have already written
45  * the magic bytes of the signature, or more likely, the PNG stream is
46  * being embedded into another stream and doesn't need its own signature,
47  * we should call png_set_sig_bytes() to tell libpng how many of the
48  * bytes have already been written.
49  */
50 void PNGAPI
51 png_write_sig(png_structrp png_ptr)
52 {
53    png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10};
54
55 #ifdef PNG_IO_STATE_SUPPORTED
56    /* Inform the I/O callback that the signature is being written */
57    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_SIGNATURE;
58 #endif
59
60    /* Write the rest of the 8 byte signature */
61    png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes],
62        (size_t)(8 - png_ptr->sig_bytes));
63
64    if (png_ptr->sig_bytes < 3)
65       png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE;
66 }
67
68 /* Write the start of a PNG chunk.  The type is the chunk type.
69  * The total_length is the sum of the lengths of all the data you will be
70  * passing in png_write_chunk_data().
71  */
72 static void
73 png_write_chunk_header(png_structrp png_ptr, png_uint_32 chunk_name,
74     png_uint_32 length)
75 {
76    png_byte buf[8];
77
78 #if defined(PNG_DEBUG) && (PNG_DEBUG > 0)
79    PNG_CSTRING_FROM_CHUNK(buf, chunk_name);
80    png_debug2(0, "Writing %s chunk, length = %lu", buf, (unsigned long)length);
81 #endif
82
83    if (png_ptr == NULL)
84       return;
85
86 #ifdef PNG_IO_STATE_SUPPORTED
87    /* Inform the I/O callback that the chunk header is being written.
88     * PNG_IO_CHUNK_HDR requires a single I/O call.
89     */
90    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_HDR;
91 #endif
92
93    /* Write the length and the chunk name */
94    png_save_uint_32(buf, length);
95    png_save_uint_32(buf + 4, chunk_name);
96    png_write_data(png_ptr, buf, 8);
97
98    /* Put the chunk name into png_ptr->chunk_name */
99    png_ptr->chunk_name = chunk_name;
100
101    /* Reset the crc and run it over the chunk name */
102    png_reset_crc(png_ptr);
103
104    png_calculate_crc(png_ptr, buf + 4, 4);
105
106 #ifdef PNG_IO_STATE_SUPPORTED
107    /* Inform the I/O callback that chunk data will (possibly) be written.
108     * PNG_IO_CHUNK_DATA does NOT require a specific number of I/O calls.
109     */
110    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_DATA;
111 #endif
112 }
113
114 void PNGAPI
115 png_write_chunk_start(png_structrp png_ptr, png_const_bytep chunk_string,
116     png_uint_32 length)
117 {
118    png_write_chunk_header(png_ptr, PNG_CHUNK_FROM_STRING(chunk_string), length);
119 }
120
121 /* Write the data of a PNG chunk started with png_write_chunk_header().
122  * Note that multiple calls to this function are allowed, and that the
123  * sum of the lengths from these calls *must* add up to the total_length
124  * given to png_write_chunk_header().
125  */
126 void PNGAPI
127 png_write_chunk_data(png_structrp png_ptr, png_const_bytep data, size_t length)
128 {
129    /* Write the data, and run the CRC over it */
130    if (png_ptr == NULL)
131       return;
132
133    if (data != NULL && length > 0)
134    {
135       png_write_data(png_ptr, data, length);
136
137       /* Update the CRC after writing the data,
138        * in case the user I/O routine alters it.
139        */
140       png_calculate_crc(png_ptr, data, length);
141    }
142 }
143
144 /* Finish a chunk started with png_write_chunk_header(). */
145 void PNGAPI
146 png_write_chunk_end(png_structrp png_ptr)
147 {
148    png_byte buf[4];
149
150    if (png_ptr == NULL) return;
151
152 #ifdef PNG_IO_STATE_SUPPORTED
153    /* Inform the I/O callback that the chunk CRC is being written.
154     * PNG_IO_CHUNK_CRC requires a single I/O function call.
155     */
156    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_CRC;
157 #endif
158
159    /* Write the crc in a single operation */
160    png_save_uint_32(buf, png_ptr->crc);
161
162    png_write_data(png_ptr, buf, 4);
163 }
164
165 /* Write a PNG chunk all at once.  The type is an array of ASCII characters
166  * representing the chunk name.  The array must be at least 4 bytes in
167  * length, and does not need to be null terminated.  To be safe, pass the
168  * pre-defined chunk names here, and if you need a new one, define it
169  * where the others are defined.  The length is the length of the data.
170  * All the data must be present.  If that is not possible, use the
171  * png_write_chunk_start(), png_write_chunk_data(), and png_write_chunk_end()
172  * functions instead.
173  */
174 static void
175 png_write_complete_chunk(png_structrp png_ptr, png_uint_32 chunk_name,
176     png_const_bytep data, size_t length)
177 {
178    if (png_ptr == NULL)
179       return;
180
181    /* On 64-bit architectures 'length' may not fit in a png_uint_32. */
182    if (length > PNG_UINT_31_MAX)
183       png_error(png_ptr, "length exceeds PNG maximum");
184
185    png_write_chunk_header(png_ptr, chunk_name, (png_uint_32)length);
186    png_write_chunk_data(png_ptr, data, length);
187    png_write_chunk_end(png_ptr);
188 }
189
190 /* This is the API that calls the internal function above. */
191 void PNGAPI
192 png_write_chunk(png_structrp png_ptr, png_const_bytep chunk_string,
193     png_const_bytep data, size_t length)
194 {
195    png_write_complete_chunk(png_ptr, PNG_CHUNK_FROM_STRING(chunk_string), data,
196        length);
197 }
198
199 /* This is used below to find the size of an image to pass to png_deflate_claim,
200  * so it only needs to be accurate if the size is less than 16384 bytes (the
201  * point at which a lower LZ window size can be used.)
202  */
203 static png_alloc_size_t
204 png_image_size(png_structrp png_ptr)
205 {
206    /* Only return sizes up to the maximum of a png_uint_32; do this by limiting
207     * the width and height used to 15 bits.
208     */
209    png_uint_32 h = png_ptr->height;
210
211    if (png_ptr->rowbytes < 32768 && h < 32768)
212    {
213       if (png_ptr->interlaced != 0)
214       {
215          /* Interlacing makes the image larger because of the replication of
216           * both the filter byte and the padding to a byte boundary.
217           */
218          png_uint_32 w = png_ptr->width;
219          unsigned int pd = png_ptr->pixel_depth;
220          png_alloc_size_t cb_base;
221          int pass;
222
223          for (cb_base=0, pass=0; pass<=6; ++pass)
224          {
225             png_uint_32 pw = PNG_PASS_COLS(w, pass);
226
227             if (pw > 0)
228                cb_base += (PNG_ROWBYTES(pd, pw)+1) * PNG_PASS_ROWS(h, pass);
229          }
230
231          return cb_base;
232       }
233
234       else
235          return (png_ptr->rowbytes+1) * h;
236    }
237
238    else
239       return 0xffffffffU;
240 }
241
242 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
243    /* This is the code to hack the first two bytes of the deflate stream (the
244     * deflate header) to correct the windowBits value to match the actual data
245     * size.  Note that the second argument is the *uncompressed* size but the
246     * first argument is the *compressed* data (and it must be deflate
247     * compressed.)
248     */
249 static void
250 optimize_cmf(png_bytep data, png_alloc_size_t data_size)
251 {
252    /* Optimize the CMF field in the zlib stream.  The resultant zlib stream is
253     * still compliant to the stream specification.
254     */
255    if (data_size <= 16384) /* else windowBits must be 15 */
256    {
257       unsigned int z_cmf = data[0];  /* zlib compression method and flags */
258
259       if ((z_cmf & 0x0f) == 8 && (z_cmf & 0xf0) <= 0x70)
260       {
261          unsigned int z_cinfo;
262          unsigned int half_z_window_size;
263
264          z_cinfo = z_cmf >> 4;
265          half_z_window_size = 1U << (z_cinfo + 7);
266
267          if (data_size <= half_z_window_size) /* else no change */
268          {
269             unsigned int tmp;
270
271             do
272             {
273                half_z_window_size >>= 1;
274                --z_cinfo;
275             }
276             while (z_cinfo > 0 && data_size <= half_z_window_size);
277
278             z_cmf = (z_cmf & 0x0f) | (z_cinfo << 4);
279
280             data[0] = (png_byte)z_cmf;
281             tmp = data[1] & 0xe0;
282             tmp += 0x1f - ((z_cmf << 8) + tmp) % 0x1f;
283             data[1] = (png_byte)tmp;
284          }
285       }
286    }
287 }
288 #endif /* WRITE_OPTIMIZE_CMF */
289
290 /* Initialize the compressor for the appropriate type of compression. */
291 static int
292 png_deflate_claim(png_structrp png_ptr, png_uint_32 owner,
293     png_alloc_size_t data_size)
294 {
295    if (png_ptr->zowner != 0)
296    {
297 #if defined(PNG_WARNINGS_SUPPORTED) || defined(PNG_ERROR_TEXT_SUPPORTED)
298       char msg[64];
299
300       PNG_STRING_FROM_CHUNK(msg, owner);
301       msg[4] = ':';
302       msg[5] = ' ';
303       PNG_STRING_FROM_CHUNK(msg+6, png_ptr->zowner);
304       /* So the message that results is "<chunk> using zstream"; this is an
305        * internal error, but is very useful for debugging.  i18n requirements
306        * are minimal.
307        */
308       (void)png_safecat(msg, (sizeof msg), 10, " using zstream");
309 #endif
310 #if PNG_RELEASE_BUILD
311          png_warning(png_ptr, msg);
312
313          /* Attempt sane error recovery */
314          if (png_ptr->zowner == png_IDAT) /* don't steal from IDAT */
315          {
316             png_ptr->zstream.msg = PNGZ_MSG_CAST("in use by IDAT");
317             return Z_STREAM_ERROR;
318          }
319
320          png_ptr->zowner = 0;
321 #else
322          png_error(png_ptr, msg);
323 #endif
324    }
325
326    {
327       int level = png_ptr->zlib_level;
328       int method = png_ptr->zlib_method;
329       int windowBits = png_ptr->zlib_window_bits;
330       int memLevel = png_ptr->zlib_mem_level;
331       int strategy; /* set below */
332       int ret; /* zlib return code */
333
334       if (owner == png_IDAT)
335       {
336          if ((png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_STRATEGY) != 0)
337             strategy = png_ptr->zlib_strategy;
338
339          else if (png_ptr->do_filter != PNG_FILTER_NONE)
340             strategy = PNG_Z_DEFAULT_STRATEGY;
341
342          else
343             strategy = PNG_Z_DEFAULT_NOFILTER_STRATEGY;
344       }
345
346       else
347       {
348 #ifdef PNG_WRITE_CUSTOMIZE_ZTXT_COMPRESSION_SUPPORTED
349             level = png_ptr->zlib_text_level;
350             method = png_ptr->zlib_text_method;
351             windowBits = png_ptr->zlib_text_window_bits;
352             memLevel = png_ptr->zlib_text_mem_level;
353             strategy = png_ptr->zlib_text_strategy;
354 #else
355             /* If customization is not supported the values all come from the
356              * IDAT values except for the strategy, which is fixed to the
357              * default.  (This is the pre-1.6.0 behavior too, although it was
358              * implemented in a very different way.)
359              */
360             strategy = Z_DEFAULT_STRATEGY;
361 #endif
362       }
363
364       /* Adjust 'windowBits' down if larger than 'data_size'; to stop this
365        * happening just pass 32768 as the data_size parameter.  Notice that zlib
366        * requires an extra 262 bytes in the window in addition to the data to be
367        * able to see the whole of the data, so if data_size+262 takes us to the
368        * next windowBits size we need to fix up the value later.  (Because even
369        * though deflate needs the extra window, inflate does not!)
370        */
371       if (data_size <= 16384)
372       {
373          /* IMPLEMENTATION NOTE: this 'half_window_size' stuff is only here to
374           * work round a Microsoft Visual C misbehavior which, contrary to C-90,
375           * widens the result of the following shift to 64-bits if (and,
376           * apparently, only if) it is used in a test.
377           */
378          unsigned int half_window_size = 1U << (windowBits-1);
379
380          while (data_size + 262 <= half_window_size)
381          {
382             half_window_size >>= 1;
383             --windowBits;
384          }
385       }
386
387       /* Check against the previous initialized values, if any. */
388       if ((png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED) != 0 &&
389          (png_ptr->zlib_set_level != level ||
390          png_ptr->zlib_set_method != method ||
391          png_ptr->zlib_set_window_bits != windowBits ||
392          png_ptr->zlib_set_mem_level != memLevel ||
393          png_ptr->zlib_set_strategy != strategy))
394       {
395          if (deflateEnd(&png_ptr->zstream) != Z_OK)
396             png_warning(png_ptr, "deflateEnd failed (ignored)");
397
398          png_ptr->flags &= ~PNG_FLAG_ZSTREAM_INITIALIZED;
399       }
400
401       /* For safety clear out the input and output pointers (currently zlib
402        * doesn't use them on Init, but it might in the future).
403        */
404       png_ptr->zstream.next_in = NULL;
405       png_ptr->zstream.avail_in = 0;
406       png_ptr->zstream.next_out = NULL;
407       png_ptr->zstream.avail_out = 0;
408
409       /* Now initialize if required, setting the new parameters, otherwise just
410        * do a simple reset to the previous parameters.
411        */
412       if ((png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED) != 0)
413          ret = deflateReset(&png_ptr->zstream);
414
415       else
416       {
417          ret = deflateInit2(&png_ptr->zstream, level, method, windowBits,
418              memLevel, strategy);
419
420          if (ret == Z_OK)
421             png_ptr->flags |= PNG_FLAG_ZSTREAM_INITIALIZED;
422       }
423
424       /* The return code is from either deflateReset or deflateInit2; they have
425        * pretty much the same set of error codes.
426        */
427       if (ret == Z_OK)
428          png_ptr->zowner = owner;
429
430       else
431          png_zstream_error(png_ptr, ret);
432
433       return ret;
434    }
435 }
436
437 /* Clean up (or trim) a linked list of compression buffers. */
438 void /* PRIVATE */
439 png_free_buffer_list(png_structrp png_ptr, png_compression_bufferp *listp)
440 {
441    png_compression_bufferp list = *listp;
442
443    if (list != NULL)
444    {
445       *listp = NULL;
446
447       do
448       {
449          png_compression_bufferp next = list->next;
450
451          png_free(png_ptr, list);
452          list = next;
453       }
454       while (list != NULL);
455    }
456 }
457
458 #ifdef PNG_WRITE_COMPRESSED_TEXT_SUPPORTED
459 /* This pair of functions encapsulates the operation of (a) compressing a
460  * text string, and (b) issuing it later as a series of chunk data writes.
461  * The compression_state structure is shared context for these functions
462  * set up by the caller to allow access to the relevant local variables.
463  *
464  * compression_buffer (new in 1.6.0) is just a linked list of zbuffer_size
465  * temporary buffers.  From 1.6.0 it is retained in png_struct so that it will
466  * be correctly freed in the event of a write error (previous implementations
467  * just leaked memory.)
468  */
469 typedef struct
470 {
471    png_const_bytep      input;        /* The uncompressed input data */
472    png_alloc_size_t     input_len;    /* Its length */
473    png_uint_32          output_len;   /* Final compressed length */
474    png_byte             output[1024]; /* First block of output */
475 } compression_state;
476
477 static void
478 png_text_compress_init(compression_state *comp, png_const_bytep input,
479     png_alloc_size_t input_len)
480 {
481    comp->input = input;
482    comp->input_len = input_len;
483    comp->output_len = 0;
484 }
485
486 /* Compress the data in the compression state input */
487 static int
488 png_text_compress(png_structrp png_ptr, png_uint_32 chunk_name,
489     compression_state *comp, png_uint_32 prefix_len)
490 {
491    int ret;
492
493    /* To find the length of the output it is necessary to first compress the
494     * input. The result is buffered rather than using the two-pass algorithm
495     * that is used on the inflate side; deflate is assumed to be slower and a
496     * PNG writer is assumed to have more memory available than a PNG reader.
497     *
498     * IMPLEMENTATION NOTE: the zlib API deflateBound() can be used to find an
499     * upper limit on the output size, but it is always bigger than the input
500     * size so it is likely to be more efficient to use this linked-list
501     * approach.
502     */
503    ret = png_deflate_claim(png_ptr, chunk_name, comp->input_len);
504
505    if (ret != Z_OK)
506       return ret;
507
508    /* Set up the compression buffers, we need a loop here to avoid overflowing a
509     * uInt.  Use ZLIB_IO_MAX to limit the input.  The output is always limited
510     * by the output buffer size, so there is no need to check that.  Since this
511     * is ANSI-C we know that an 'int', hence a uInt, is always at least 16 bits
512     * in size.
513     */
514    {
515       png_compression_bufferp *end = &png_ptr->zbuffer_list;
516       png_alloc_size_t input_len = comp->input_len; /* may be zero! */
517       png_uint_32 output_len;
518
519       /* zlib updates these for us: */
520       png_ptr->zstream.next_in = PNGZ_INPUT_CAST(comp->input);
521       png_ptr->zstream.avail_in = 0; /* Set below */
522       png_ptr->zstream.next_out = comp->output;
523       png_ptr->zstream.avail_out = (sizeof comp->output);
524
525       output_len = png_ptr->zstream.avail_out;
526
527       do
528       {
529          uInt avail_in = ZLIB_IO_MAX;
530
531          if (avail_in > input_len)
532             avail_in = (uInt)input_len;
533
534          input_len -= avail_in;
535
536          png_ptr->zstream.avail_in = avail_in;
537
538          if (png_ptr->zstream.avail_out == 0)
539          {
540             png_compression_buffer *next;
541
542             /* Chunk data is limited to 2^31 bytes in length, so the prefix
543              * length must be counted here.
544              */
545             if (output_len + prefix_len > PNG_UINT_31_MAX)
546             {
547                ret = Z_MEM_ERROR;
548                break;
549             }
550
551             /* Need a new (malloc'ed) buffer, but there may be one present
552              * already.
553              */
554             next = *end;
555             if (next == NULL)
556             {
557                next = png_voidcast(png_compression_bufferp, png_malloc_base
558                   (png_ptr, PNG_COMPRESSION_BUFFER_SIZE(png_ptr)));
559
560                if (next == NULL)
561                {
562                   ret = Z_MEM_ERROR;
563                   break;
564                }
565
566                /* Link in this buffer (so that it will be freed later) */
567                next->next = NULL;
568                *end = next;
569             }
570
571             png_ptr->zstream.next_out = next->output;
572             png_ptr->zstream.avail_out = png_ptr->zbuffer_size;
573             output_len += png_ptr->zstream.avail_out;
574
575             /* Move 'end' to the next buffer pointer. */
576             end = &next->next;
577          }
578
579          /* Compress the data */
580          ret = deflate(&png_ptr->zstream,
581              input_len > 0 ? Z_NO_FLUSH : Z_FINISH);
582
583          /* Claw back input data that was not consumed (because avail_in is
584           * reset above every time round the loop).
585           */
586          input_len += png_ptr->zstream.avail_in;
587          png_ptr->zstream.avail_in = 0; /* safety */
588       }
589       while (ret == Z_OK);
590
591       /* There may be some space left in the last output buffer. This needs to
592        * be subtracted from output_len.
593        */
594       output_len -= png_ptr->zstream.avail_out;
595       png_ptr->zstream.avail_out = 0; /* safety */
596       comp->output_len = output_len;
597
598       /* Now double check the output length, put in a custom message if it is
599        * too long.  Otherwise ensure the z_stream::msg pointer is set to
600        * something.
601        */
602       if (output_len + prefix_len >= PNG_UINT_31_MAX)
603       {
604          png_ptr->zstream.msg = PNGZ_MSG_CAST("compressed data too long");
605          ret = Z_MEM_ERROR;
606       }
607
608       else
609          png_zstream_error(png_ptr, ret);
610
611       /* Reset zlib for another zTXt/iTXt or image data */
612       png_ptr->zowner = 0;
613
614       /* The only success case is Z_STREAM_END, input_len must be 0; if not this
615        * is an internal error.
616        */
617       if (ret == Z_STREAM_END && input_len == 0)
618       {
619 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
620          /* Fix up the deflate header, if required */
621          optimize_cmf(comp->output, comp->input_len);
622 #endif
623          /* But Z_OK is returned, not Z_STREAM_END; this allows the claim
624           * function above to return Z_STREAM_END on an error (though it never
625           * does in the current versions of zlib.)
626           */
627          return Z_OK;
628       }
629
630       else
631          return ret;
632    }
633 }
634
635 /* Ship the compressed text out via chunk writes */
636 static void
637 png_write_compressed_data_out(png_structrp png_ptr, compression_state *comp)
638 {
639    png_uint_32 output_len = comp->output_len;
640    png_const_bytep output = comp->output;
641    png_uint_32 avail = (sizeof comp->output);
642    png_compression_buffer *next = png_ptr->zbuffer_list;
643
644    for (;;)
645    {
646       if (avail > output_len)
647          avail = output_len;
648
649       png_write_chunk_data(png_ptr, output, avail);
650
651       output_len -= avail;
652
653       if (output_len == 0 || next == NULL)
654          break;
655
656       avail = png_ptr->zbuffer_size;
657       output = next->output;
658       next = next->next;
659    }
660
661    /* This is an internal error; 'next' must have been NULL! */
662    if (output_len > 0)
663       png_error(png_ptr, "error writing ancillary chunked compressed data");
664 }
665 #endif /* WRITE_COMPRESSED_TEXT */
666
667 /* Write the IHDR chunk, and update the png_struct with the necessary
668  * information.  Note that the rest of this code depends upon this
669  * information being correct.
670  */
671 void /* PRIVATE */
672 png_write_IHDR(png_structrp png_ptr, png_uint_32 width, png_uint_32 height,
673     int bit_depth, int color_type, int compression_type, int filter_type,
674     int interlace_type)
675 {
676    png_byte buf[13]; /* Buffer to store the IHDR info */
677    int is_invalid_depth;
678
679    png_debug(1, "in png_write_IHDR");
680
681    /* Check that we have valid input data from the application info */
682    switch (color_type)
683    {
684       case PNG_COLOR_TYPE_GRAY:
685          switch (bit_depth)
686          {
687             case 1:
688             case 2:
689             case 4:
690             case 8:
691 #ifdef PNG_WRITE_16BIT_SUPPORTED
692             case 16:
693 #endif
694                png_ptr->channels = 1; break;
695
696             default:
697                png_error(png_ptr,
698                    "Invalid bit depth for grayscale image");
699          }
700          break;
701
702       case PNG_COLOR_TYPE_RGB:
703          is_invalid_depth = (bit_depth != 8);
704 #ifdef PNG_WRITE_16BIT_SUPPORTED
705          is_invalid_depth = (is_invalid_depth && bit_depth != 16);
706 #endif
707          if (is_invalid_depth)
708             png_error(png_ptr, "Invalid bit depth for RGB image");
709
710          png_ptr->channels = 3;
711          break;
712
713       case PNG_COLOR_TYPE_PALETTE:
714          switch (bit_depth)
715          {
716             case 1:
717             case 2:
718             case 4:
719             case 8:
720                png_ptr->channels = 1;
721                break;
722
723             default:
724                png_error(png_ptr, "Invalid bit depth for paletted image");
725          }
726          break;
727
728       case PNG_COLOR_TYPE_GRAY_ALPHA:
729          is_invalid_depth = (bit_depth != 8);
730 #ifdef PNG_WRITE_16BIT_SUPPORTED
731          is_invalid_depth = (is_invalid_depth && bit_depth != 16);
732 #endif
733          if (is_invalid_depth)
734             png_error(png_ptr, "Invalid bit depth for grayscale+alpha image");
735
736          png_ptr->channels = 2;
737          break;
738
739       case PNG_COLOR_TYPE_RGB_ALPHA:
740          is_invalid_depth = (bit_depth != 8);
741 #ifdef PNG_WRITE_16BIT_SUPPORTED
742          is_invalid_depth = (is_invalid_depth && bit_depth != 16);
743 #endif
744          if (is_invalid_depth)
745             png_error(png_ptr, "Invalid bit depth for RGBA image");
746
747          png_ptr->channels = 4;
748          break;
749
750       default:
751          png_error(png_ptr, "Invalid image color type specified");
752    }
753
754    if (compression_type != PNG_COMPRESSION_TYPE_BASE)
755    {
756       png_warning(png_ptr, "Invalid compression type specified");
757       compression_type = PNG_COMPRESSION_TYPE_BASE;
758    }
759
760    /* Write filter_method 64 (intrapixel differencing) only if
761     * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
762     * 2. Libpng did not write a PNG signature (this filter_method is only
763     *    used in PNG datastreams that are embedded in MNG datastreams) and
764     * 3. The application called png_permit_mng_features with a mask that
765     *    included PNG_FLAG_MNG_FILTER_64 and
766     * 4. The filter_method is 64 and
767     * 5. The color_type is RGB or RGBA
768     */
769    if (
770 #ifdef PNG_MNG_FEATURES_SUPPORTED
771        !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) != 0 &&
772        ((png_ptr->mode & PNG_HAVE_PNG_SIGNATURE) == 0) &&
773        (color_type == PNG_COLOR_TYPE_RGB ||
774         color_type == PNG_COLOR_TYPE_RGB_ALPHA) &&
775        (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) &&
776 #endif
777        filter_type != PNG_FILTER_TYPE_BASE)
778    {
779       png_warning(png_ptr, "Invalid filter type specified");
780       filter_type = PNG_FILTER_TYPE_BASE;
781    }
782
783 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
784    if (interlace_type != PNG_INTERLACE_NONE &&
785        interlace_type != PNG_INTERLACE_ADAM7)
786    {
787       png_warning(png_ptr, "Invalid interlace type specified");
788       interlace_type = PNG_INTERLACE_ADAM7;
789    }
790 #else
791    interlace_type=PNG_INTERLACE_NONE;
792 #endif
793
794    /* Save the relevant information */
795    png_ptr->bit_depth = (png_byte)bit_depth;
796    png_ptr->color_type = (png_byte)color_type;
797    png_ptr->interlaced = (png_byte)interlace_type;
798 #ifdef PNG_MNG_FEATURES_SUPPORTED
799    png_ptr->filter_type = (png_byte)filter_type;
800 #endif
801    png_ptr->compression_type = (png_byte)compression_type;
802    png_ptr->width = width;
803    png_ptr->height = height;
804
805    png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels);
806    png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width);
807    /* Set the usr info, so any transformations can modify it */
808    png_ptr->usr_width = png_ptr->width;
809    png_ptr->usr_bit_depth = png_ptr->bit_depth;
810    png_ptr->usr_channels = png_ptr->channels;
811
812    /* Pack the header information into the buffer */
813    png_save_uint_32(buf, width);
814    png_save_uint_32(buf + 4, height);
815    buf[8] = (png_byte)bit_depth;
816    buf[9] = (png_byte)color_type;
817    buf[10] = (png_byte)compression_type;
818    buf[11] = (png_byte)filter_type;
819    buf[12] = (png_byte)interlace_type;
820
821    /* Write the chunk */
822    png_write_complete_chunk(png_ptr, png_IHDR, buf, 13);
823
824 #ifdef PNG_WRITE_APNG_SUPPORTED
825    png_ptr->first_frame_width = width;
826    png_ptr->first_frame_height = height;
827 #endif
828
829    if ((png_ptr->do_filter) == PNG_NO_FILTERS)
830    {
831       if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE ||
832           png_ptr->bit_depth < 8)
833          png_ptr->do_filter = PNG_FILTER_NONE;
834
835       else
836          png_ptr->do_filter = PNG_ALL_FILTERS;
837    }
838
839    png_ptr->mode = PNG_HAVE_IHDR; /* not READY_FOR_ZTXT */
840 }
841
842 /* Write the palette.  We are careful not to trust png_color to be in the
843  * correct order for PNG, so people can redefine it to any convenient
844  * structure.
845  */
846 void /* PRIVATE */
847 png_write_PLTE(png_structrp png_ptr, png_const_colorp palette,
848     png_uint_32 num_pal)
849 {
850    png_uint_32 max_palette_length, i;
851    png_const_colorp pal_ptr;
852    png_byte buf[3];
853
854    png_debug(1, "in png_write_PLTE");
855
856    max_palette_length = (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE) ?
857       (1 << png_ptr->bit_depth) : PNG_MAX_PALETTE_LENGTH;
858
859    if ((
860 #ifdef PNG_MNG_FEATURES_SUPPORTED
861        (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0 &&
862 #endif
863        num_pal == 0) || num_pal > max_palette_length)
864    {
865       if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
866       {
867          png_error(png_ptr, "Invalid number of colors in palette");
868       }
869
870       else
871       {
872          png_warning(png_ptr, "Invalid number of colors in palette");
873          return;
874       }
875    }
876
877    if ((png_ptr->color_type & PNG_COLOR_MASK_COLOR) == 0)
878    {
879       png_warning(png_ptr,
880           "Ignoring request to write a PLTE chunk in grayscale PNG");
881
882       return;
883    }
884
885    png_ptr->num_palette = (png_uint_16)num_pal;
886    png_debug1(3, "num_palette = %d", png_ptr->num_palette);
887
888    png_write_chunk_header(png_ptr, png_PLTE, (png_uint_32)(num_pal * 3));
889 #ifdef PNG_POINTER_INDEXING_SUPPORTED
890
891    for (i = 0, pal_ptr = palette; i < num_pal; i++, pal_ptr++)
892    {
893       buf[0] = pal_ptr->red;
894       buf[1] = pal_ptr->green;
895       buf[2] = pal_ptr->blue;
896       png_write_chunk_data(png_ptr, buf, 3);
897    }
898
899 #else
900    /* This is a little slower but some buggy compilers need to do this
901     * instead
902     */
903    pal_ptr=palette;
904
905    for (i = 0; i < num_pal; i++)
906    {
907       buf[0] = pal_ptr[i].red;
908       buf[1] = pal_ptr[i].green;
909       buf[2] = pal_ptr[i].blue;
910       png_write_chunk_data(png_ptr, buf, 3);
911    }
912
913 #endif
914    png_write_chunk_end(png_ptr);
915    png_ptr->mode |= PNG_HAVE_PLTE;
916 }
917
918 /* This is similar to png_text_compress, above, except that it does not require
919  * all of the data at once and, instead of buffering the compressed result,
920  * writes it as IDAT chunks.  Unlike png_text_compress it *can* png_error out
921  * because it calls the write interface.  As a result it does its own error
922  * reporting and does not return an error code.  In the event of error it will
923  * just call png_error.  The input data length may exceed 32-bits.  The 'flush'
924  * parameter is exactly the same as that to deflate, with the following
925  * meanings:
926  *
927  * Z_NO_FLUSH: normal incremental output of compressed data
928  * Z_SYNC_FLUSH: do a SYNC_FLUSH, used by png_write_flush
929  * Z_FINISH: this is the end of the input, do a Z_FINISH and clean up
930  *
931  * The routine manages the acquire and release of the png_ptr->zstream by
932  * checking and (at the end) clearing png_ptr->zowner; it does some sanity
933  * checks on the 'mode' flags while doing this.
934  */
935 void /* PRIVATE */
936 png_compress_IDAT(png_structrp png_ptr, png_const_bytep input,
937     png_alloc_size_t input_len, int flush)
938 {
939    if (png_ptr->zowner != png_IDAT)
940    {
941       /* First time.   Ensure we have a temporary buffer for compression and
942        * trim the buffer list if it has more than one entry to free memory.
943        * If 'WRITE_COMPRESSED_TEXT' is not set the list will never have been
944        * created at this point, but the check here is quick and safe.
945        */
946       if (png_ptr->zbuffer_list == NULL)
947       {
948          png_ptr->zbuffer_list = png_voidcast(png_compression_bufferp,
949              png_malloc(png_ptr, PNG_COMPRESSION_BUFFER_SIZE(png_ptr)));
950          png_ptr->zbuffer_list->next = NULL;
951       }
952
953       else
954          png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list->next);
955
956       /* It is a terminal error if we can't claim the zstream. */
957       if (png_deflate_claim(png_ptr, png_IDAT, png_image_size(png_ptr)) != Z_OK)
958          png_error(png_ptr, png_ptr->zstream.msg);
959
960       /* The output state is maintained in png_ptr->zstream, so it must be
961        * initialized here after the claim.
962        */
963       png_ptr->zstream.next_out = png_ptr->zbuffer_list->output;
964       png_ptr->zstream.avail_out = png_ptr->zbuffer_size;
965    }
966
967    /* Now loop reading and writing until all the input is consumed or an error
968     * terminates the operation.  The _out values are maintained across calls to
969     * this function, but the input must be reset each time.
970     */
971    png_ptr->zstream.next_in = PNGZ_INPUT_CAST(input);
972    png_ptr->zstream.avail_in = 0; /* set below */
973    for (;;)
974    {
975       int ret;
976
977       /* INPUT: from the row data */
978       uInt avail = ZLIB_IO_MAX;
979
980       if (avail > input_len)
981          avail = (uInt)input_len; /* safe because of the check */
982
983       png_ptr->zstream.avail_in = avail;
984       input_len -= avail;
985
986       ret = deflate(&png_ptr->zstream, input_len > 0 ? Z_NO_FLUSH : flush);
987
988       /* Include as-yet unconsumed input */
989       input_len += png_ptr->zstream.avail_in;
990       png_ptr->zstream.avail_in = 0;
991
992       /* OUTPUT: write complete IDAT chunks when avail_out drops to zero. Note
993        * that these two zstream fields are preserved across the calls, therefore
994        * there is no need to set these up on entry to the loop.
995        */
996       if (png_ptr->zstream.avail_out == 0)
997       {
998          png_bytep data = png_ptr->zbuffer_list->output;
999          uInt size = png_ptr->zbuffer_size;
1000
1001          /* Write an IDAT containing the data then reset the buffer.  The
1002           * first IDAT may need deflate header optimization.
1003           */
1004 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
1005             if ((png_ptr->mode & PNG_HAVE_IDAT) == 0 &&
1006                 png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
1007                optimize_cmf(data, png_image_size(png_ptr));
1008 #endif
1009
1010             if (size > 0)
1011 #ifdef PNG_WRITE_APNG_SUPPORTED
1012             {
1013                if (png_ptr->num_frames_written == 0)
1014 #endif
1015                png_write_complete_chunk(png_ptr, png_IDAT, data, size);
1016 #ifdef PNG_WRITE_APNG_SUPPORTED
1017                else
1018                   png_write_fdAT(png_ptr, data, size);
1019             }
1020 #endif /* PNG_WRITE_APNG_SUPPORTED */
1021          png_ptr->mode |= PNG_HAVE_IDAT;
1022
1023          png_ptr->zstream.next_out = data;
1024          png_ptr->zstream.avail_out = size;
1025
1026          /* For SYNC_FLUSH or FINISH it is essential to keep calling zlib with
1027           * the same flush parameter until it has finished output, for NO_FLUSH
1028           * it doesn't matter.
1029           */
1030          if (ret == Z_OK && flush != Z_NO_FLUSH)
1031             continue;
1032       }
1033
1034       /* The order of these checks doesn't matter much; it just affects which
1035        * possible error might be detected if multiple things go wrong at once.
1036        */
1037       if (ret == Z_OK) /* most likely return code! */
1038       {
1039          /* If all the input has been consumed then just return.  If Z_FINISH
1040           * was used as the flush parameter something has gone wrong if we get
1041           * here.
1042           */
1043          if (input_len == 0)
1044          {
1045             if (flush == Z_FINISH)
1046                png_error(png_ptr, "Z_OK on Z_FINISH with output space");
1047
1048             return;
1049          }
1050       }
1051
1052       else if (ret == Z_STREAM_END && flush == Z_FINISH)
1053       {
1054          /* This is the end of the IDAT data; any pending output must be
1055           * flushed.  For small PNG files we may still be at the beginning.
1056           */
1057          png_bytep data = png_ptr->zbuffer_list->output;
1058          uInt size = png_ptr->zbuffer_size - png_ptr->zstream.avail_out;
1059
1060 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
1061          if ((png_ptr->mode & PNG_HAVE_IDAT) == 0 &&
1062              png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
1063             optimize_cmf(data, png_image_size(png_ptr));
1064 #endif
1065
1066          if (size > 0)
1067 #ifdef PNG_WRITE_APNG_SUPPORTED
1068          {
1069             if (png_ptr->num_frames_written == 0)
1070 #endif
1071             png_write_complete_chunk(png_ptr, png_IDAT, data, size);
1072 #ifdef PNG_WRITE_APNG_SUPPORTED
1073             else
1074                png_write_fdAT(png_ptr, data, size);
1075          }
1076 #endif /* PNG_WRITE_APNG_SUPPORTED */
1077
1078          png_ptr->zstream.avail_out = 0;
1079          png_ptr->zstream.next_out = NULL;
1080          png_ptr->mode |= PNG_HAVE_IDAT | PNG_AFTER_IDAT;
1081
1082          png_ptr->zowner = 0; /* Release the stream */
1083          return;
1084       }
1085
1086       else
1087       {
1088          /* This is an error condition. */
1089          png_zstream_error(png_ptr, ret);
1090          png_error(png_ptr, png_ptr->zstream.msg);
1091       }
1092    }
1093 }
1094
1095 /* Write an IEND chunk */
1096 void /* PRIVATE */
1097 png_write_IEND(png_structrp png_ptr)
1098 {
1099    png_debug(1, "in png_write_IEND");
1100
1101    png_write_complete_chunk(png_ptr, png_IEND, NULL, 0);
1102    png_ptr->mode |= PNG_HAVE_IEND;
1103 }
1104
1105 #ifdef PNG_WRITE_gAMA_SUPPORTED
1106 /* Write a gAMA chunk */
1107 void /* PRIVATE */
1108 png_write_gAMA_fixed(png_structrp png_ptr, png_fixed_point file_gamma)
1109 {
1110    png_byte buf[4];
1111
1112    png_debug(1, "in png_write_gAMA");
1113
1114    /* file_gamma is saved in 1/100,000ths */
1115    png_save_uint_32(buf, (png_uint_32)file_gamma);
1116    png_write_complete_chunk(png_ptr, png_gAMA, buf, 4);
1117 }
1118 #endif
1119
1120 #ifdef PNG_WRITE_sRGB_SUPPORTED
1121 /* Write a sRGB chunk */
1122 void /* PRIVATE */
1123 png_write_sRGB(png_structrp png_ptr, int srgb_intent)
1124 {
1125    png_byte buf[1];
1126
1127    png_debug(1, "in png_write_sRGB");
1128
1129    if (srgb_intent >= PNG_sRGB_INTENT_LAST)
1130       png_warning(png_ptr,
1131           "Invalid sRGB rendering intent specified");
1132
1133    buf[0]=(png_byte)srgb_intent;
1134    png_write_complete_chunk(png_ptr, png_sRGB, buf, 1);
1135 }
1136 #endif
1137
1138 #ifdef PNG_WRITE_iCCP_SUPPORTED
1139 /* Write an iCCP chunk */
1140 void /* PRIVATE */
1141 png_write_iCCP(png_structrp png_ptr, png_const_charp name,
1142     png_const_bytep profile)
1143 {
1144    png_uint_32 name_len;
1145    png_uint_32 profile_len;
1146    png_byte new_name[81]; /* 1 byte for the compression byte */
1147    compression_state comp;
1148    png_uint_32 temp;
1149
1150    png_debug(1, "in png_write_iCCP");
1151
1152    /* These are all internal problems: the profile should have been checked
1153     * before when it was stored.
1154     */
1155    if (profile == NULL)
1156       png_error(png_ptr, "No profile for iCCP chunk"); /* internal error */
1157
1158    profile_len = png_get_uint_32(profile);
1159
1160    if (profile_len < 132)
1161       png_error(png_ptr, "ICC profile too short");
1162
1163    temp = (png_uint_32) (*(profile+8));
1164    if (temp > 3 && (profile_len & 0x03))
1165       png_error(png_ptr, "ICC profile length invalid (not a multiple of 4)");
1166
1167    {
1168       png_uint_32 embedded_profile_len = png_get_uint_32(profile);
1169
1170       if (profile_len != embedded_profile_len)
1171          png_error(png_ptr, "Profile length does not match profile");
1172    }
1173
1174    name_len = png_check_keyword(png_ptr, name, new_name);
1175
1176    if (name_len == 0)
1177       png_error(png_ptr, "iCCP: invalid keyword");
1178
1179    new_name[++name_len] = PNG_COMPRESSION_TYPE_BASE;
1180
1181    /* Make sure we include the NULL after the name and the compression type */
1182    ++name_len;
1183
1184    png_text_compress_init(&comp, profile, profile_len);
1185
1186    /* Allow for keyword terminator and compression byte */
1187    if (png_text_compress(png_ptr, png_iCCP, &comp, name_len) != Z_OK)
1188       png_error(png_ptr, png_ptr->zstream.msg);
1189
1190    png_write_chunk_header(png_ptr, png_iCCP, name_len + comp.output_len);
1191
1192    png_write_chunk_data(png_ptr, new_name, name_len);
1193
1194    png_write_compressed_data_out(png_ptr, &comp);
1195
1196    png_write_chunk_end(png_ptr);
1197 }
1198 #endif
1199
1200 #ifdef PNG_WRITE_sPLT_SUPPORTED
1201 /* Write a sPLT chunk */
1202 void /* PRIVATE */
1203 png_write_sPLT(png_structrp png_ptr, png_const_sPLT_tp spalette)
1204 {
1205    png_uint_32 name_len;
1206    png_byte new_name[80];
1207    png_byte entrybuf[10];
1208    size_t entry_size = (spalette->depth == 8 ? 6 : 10);
1209    size_t palette_size = entry_size * (size_t)spalette->nentries;
1210    png_sPLT_entryp ep;
1211 #ifndef PNG_POINTER_INDEXING_SUPPORTED
1212    int i;
1213 #endif
1214
1215    png_debug(1, "in png_write_sPLT");
1216
1217    name_len = png_check_keyword(png_ptr, spalette->name, new_name);
1218
1219    if (name_len == 0)
1220       png_error(png_ptr, "sPLT: invalid keyword");
1221
1222    /* Make sure we include the NULL after the name */
1223    png_write_chunk_header(png_ptr, png_sPLT,
1224        (png_uint_32)(name_len + 2 + palette_size));
1225
1226    png_write_chunk_data(png_ptr, (png_bytep)new_name, (size_t)(name_len + 1));
1227
1228    png_write_chunk_data(png_ptr, &spalette->depth, 1);
1229
1230    /* Loop through each palette entry, writing appropriately */
1231 #ifdef PNG_POINTER_INDEXING_SUPPORTED
1232    for (ep = spalette->entries; ep<spalette->entries + spalette->nentries; ep++)
1233    {
1234       if (spalette->depth == 8)
1235       {
1236          entrybuf[0] = (png_byte)ep->red;
1237          entrybuf[1] = (png_byte)ep->green;
1238          entrybuf[2] = (png_byte)ep->blue;
1239          entrybuf[3] = (png_byte)ep->alpha;
1240          png_save_uint_16(entrybuf + 4, ep->frequency);
1241       }
1242
1243       else
1244       {
1245          png_save_uint_16(entrybuf + 0, ep->red);
1246          png_save_uint_16(entrybuf + 2, ep->green);
1247          png_save_uint_16(entrybuf + 4, ep->blue);
1248          png_save_uint_16(entrybuf + 6, ep->alpha);
1249          png_save_uint_16(entrybuf + 8, ep->frequency);
1250       }
1251
1252       png_write_chunk_data(png_ptr, entrybuf, entry_size);
1253    }
1254 #else
1255    ep=spalette->entries;
1256    for (i = 0; i>spalette->nentries; i++)
1257    {
1258       if (spalette->depth == 8)
1259       {
1260          entrybuf[0] = (png_byte)ep[i].red;
1261          entrybuf[1] = (png_byte)ep[i].green;
1262          entrybuf[2] = (png_byte)ep[i].blue;
1263          entrybuf[3] = (png_byte)ep[i].alpha;
1264          png_save_uint_16(entrybuf + 4, ep[i].frequency);
1265       }
1266
1267       else
1268       {
1269          png_save_uint_16(entrybuf + 0, ep[i].red);
1270          png_save_uint_16(entrybuf + 2, ep[i].green);
1271          png_save_uint_16(entrybuf + 4, ep[i].blue);
1272          png_save_uint_16(entrybuf + 6, ep[i].alpha);
1273          png_save_uint_16(entrybuf + 8, ep[i].frequency);
1274       }
1275
1276       png_write_chunk_data(png_ptr, entrybuf, entry_size);
1277    }
1278 #endif
1279
1280    png_write_chunk_end(png_ptr);
1281 }
1282 #endif
1283
1284 #ifdef PNG_WRITE_sBIT_SUPPORTED
1285 /* Write the sBIT chunk */
1286 void /* PRIVATE */
1287 png_write_sBIT(png_structrp png_ptr, png_const_color_8p sbit, int color_type)
1288 {
1289    png_byte buf[4];
1290    size_t size;
1291
1292    png_debug(1, "in png_write_sBIT");
1293
1294    /* Make sure we don't depend upon the order of PNG_COLOR_8 */
1295    if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
1296    {
1297       png_byte maxbits;
1298
1299       maxbits = (png_byte)(color_type==PNG_COLOR_TYPE_PALETTE ? 8 :
1300           png_ptr->usr_bit_depth);
1301
1302       if (sbit->red == 0 || sbit->red > maxbits ||
1303           sbit->green == 0 || sbit->green > maxbits ||
1304           sbit->blue == 0 || sbit->blue > maxbits)
1305       {
1306          png_warning(png_ptr, "Invalid sBIT depth specified");
1307          return;
1308       }
1309
1310       buf[0] = sbit->red;
1311       buf[1] = sbit->green;
1312       buf[2] = sbit->blue;
1313       size = 3;
1314    }
1315
1316    else
1317    {
1318       if (sbit->gray == 0 || sbit->gray > png_ptr->usr_bit_depth)
1319       {
1320          png_warning(png_ptr, "Invalid sBIT depth specified");
1321          return;
1322       }
1323
1324       buf[0] = sbit->gray;
1325       size = 1;
1326    }
1327
1328    if ((color_type & PNG_COLOR_MASK_ALPHA) != 0)
1329    {
1330       if (sbit->alpha == 0 || sbit->alpha > png_ptr->usr_bit_depth)
1331       {
1332          png_warning(png_ptr, "Invalid sBIT depth specified");
1333          return;
1334       }
1335
1336       buf[size++] = sbit->alpha;
1337    }
1338
1339    png_write_complete_chunk(png_ptr, png_sBIT, buf, size);
1340 }
1341 #endif
1342
1343 #ifdef PNG_WRITE_cHRM_SUPPORTED
1344 /* Write the cHRM chunk */
1345 void /* PRIVATE */
1346 png_write_cHRM_fixed(png_structrp png_ptr, const png_xy *xy)
1347 {
1348    png_byte buf[32];
1349
1350    png_debug(1, "in png_write_cHRM");
1351
1352    /* Each value is saved in 1/100,000ths */
1353    png_save_int_32(buf,      xy->whitex);
1354    png_save_int_32(buf +  4, xy->whitey);
1355
1356    png_save_int_32(buf +  8, xy->redx);
1357    png_save_int_32(buf + 12, xy->redy);
1358
1359    png_save_int_32(buf + 16, xy->greenx);
1360    png_save_int_32(buf + 20, xy->greeny);
1361
1362    png_save_int_32(buf + 24, xy->bluex);
1363    png_save_int_32(buf + 28, xy->bluey);
1364
1365    png_write_complete_chunk(png_ptr, png_cHRM, buf, 32);
1366 }
1367 #endif
1368
1369 #ifdef PNG_WRITE_tRNS_SUPPORTED
1370 /* Write the tRNS chunk */
1371 void /* PRIVATE */
1372 png_write_tRNS(png_structrp png_ptr, png_const_bytep trans_alpha,
1373     png_const_color_16p tran, int num_trans, int color_type)
1374 {
1375    png_byte buf[6];
1376
1377    png_debug(1, "in png_write_tRNS");
1378
1379    if (color_type == PNG_COLOR_TYPE_PALETTE)
1380    {
1381       if (num_trans <= 0 || num_trans > (int)png_ptr->num_palette)
1382       {
1383          png_app_warning(png_ptr,
1384              "Invalid number of transparent colors specified");
1385          return;
1386       }
1387
1388       /* Write the chunk out as it is */
1389       png_write_complete_chunk(png_ptr, png_tRNS, trans_alpha,
1390           (size_t)num_trans);
1391    }
1392
1393    else if (color_type == PNG_COLOR_TYPE_GRAY)
1394    {
1395       /* One 16-bit value */
1396       if (tran->gray >= (1 << png_ptr->bit_depth))
1397       {
1398          png_app_warning(png_ptr,
1399              "Ignoring attempt to write tRNS chunk out-of-range for bit_depth");
1400
1401          return;
1402       }
1403
1404       png_save_uint_16(buf, tran->gray);
1405       png_write_complete_chunk(png_ptr, png_tRNS, buf, 2);
1406    }
1407
1408    else if (color_type == PNG_COLOR_TYPE_RGB)
1409    {
1410       /* Three 16-bit values */
1411       png_save_uint_16(buf, tran->red);
1412       png_save_uint_16(buf + 2, tran->green);
1413       png_save_uint_16(buf + 4, tran->blue);
1414 #ifdef PNG_WRITE_16BIT_SUPPORTED
1415       if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]) != 0)
1416 #else
1417       if ((buf[0] | buf[2] | buf[4]) != 0)
1418 #endif
1419       {
1420          png_app_warning(png_ptr,
1421              "Ignoring attempt to write 16-bit tRNS chunk when bit_depth is 8");
1422          return;
1423       }
1424
1425       png_write_complete_chunk(png_ptr, png_tRNS, buf, 6);
1426    }
1427
1428    else
1429    {
1430       png_app_warning(png_ptr, "Can't write tRNS with an alpha channel");
1431    }
1432 }
1433 #endif
1434
1435 #ifdef PNG_WRITE_bKGD_SUPPORTED
1436 /* Write the background chunk */
1437 void /* PRIVATE */
1438 png_write_bKGD(png_structrp png_ptr, png_const_color_16p back, int color_type)
1439 {
1440    png_byte buf[6];
1441
1442    png_debug(1, "in png_write_bKGD");
1443
1444    if (color_type == PNG_COLOR_TYPE_PALETTE)
1445    {
1446       if (
1447 #ifdef PNG_MNG_FEATURES_SUPPORTED
1448           (png_ptr->num_palette != 0 ||
1449           (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0) &&
1450 #endif
1451          back->index >= png_ptr->num_palette)
1452       {
1453          png_warning(png_ptr, "Invalid background palette index");
1454          return;
1455       }
1456
1457       buf[0] = back->index;
1458       png_write_complete_chunk(png_ptr, png_bKGD, buf, 1);
1459    }
1460
1461    else if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
1462    {
1463       png_save_uint_16(buf, back->red);
1464       png_save_uint_16(buf + 2, back->green);
1465       png_save_uint_16(buf + 4, back->blue);
1466 #ifdef PNG_WRITE_16BIT_SUPPORTED
1467       if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]) != 0)
1468 #else
1469       if ((buf[0] | buf[2] | buf[4]) != 0)
1470 #endif
1471       {
1472          png_warning(png_ptr,
1473              "Ignoring attempt to write 16-bit bKGD chunk "
1474              "when bit_depth is 8");
1475
1476          return;
1477       }
1478
1479       png_write_complete_chunk(png_ptr, png_bKGD, buf, 6);
1480    }
1481
1482    else
1483    {
1484       if (back->gray >= (1 << png_ptr->bit_depth))
1485       {
1486          png_warning(png_ptr,
1487              "Ignoring attempt to write bKGD chunk out-of-range for bit_depth");
1488
1489          return;
1490       }
1491
1492       png_save_uint_16(buf, back->gray);
1493       png_write_complete_chunk(png_ptr, png_bKGD, buf, 2);
1494    }
1495 }
1496 #endif
1497
1498 #ifdef PNG_WRITE_eXIf_SUPPORTED
1499 /* Write the Exif data */
1500 void /* PRIVATE */
1501 png_write_eXIf(png_structrp png_ptr, png_bytep exif, int num_exif)
1502 {
1503    int i;
1504    png_byte buf[1];
1505
1506    png_debug(1, "in png_write_eXIf");
1507
1508    png_write_chunk_header(png_ptr, png_eXIf, (png_uint_32)(num_exif));
1509
1510    for (i = 0; i < num_exif; i++)
1511    {
1512       buf[0] = exif[i];
1513       png_write_chunk_data(png_ptr, buf, 1);
1514    }
1515
1516    png_write_chunk_end(png_ptr);
1517 }
1518 #endif
1519
1520 #ifdef PNG_WRITE_hIST_SUPPORTED
1521 /* Write the histogram */
1522 void /* PRIVATE */
1523 png_write_hIST(png_structrp png_ptr, png_const_uint_16p hist, int num_hist)
1524 {
1525    int i;
1526    png_byte buf[3];
1527
1528    png_debug(1, "in png_write_hIST");
1529
1530    if (num_hist > (int)png_ptr->num_palette)
1531    {
1532       png_debug2(3, "num_hist = %d, num_palette = %d", num_hist,
1533           png_ptr->num_palette);
1534
1535       png_warning(png_ptr, "Invalid number of histogram entries specified");
1536       return;
1537    }
1538
1539    png_write_chunk_header(png_ptr, png_hIST, (png_uint_32)(num_hist * 2));
1540
1541    for (i = 0; i < num_hist; i++)
1542    {
1543       png_save_uint_16(buf, hist[i]);
1544       png_write_chunk_data(png_ptr, buf, 2);
1545    }
1546
1547    png_write_chunk_end(png_ptr);
1548 }
1549 #endif
1550
1551 #ifdef PNG_WRITE_tEXt_SUPPORTED
1552 /* Write a tEXt chunk */
1553 void /* PRIVATE */
1554 png_write_tEXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
1555     size_t text_len)
1556 {
1557    png_uint_32 key_len;
1558    png_byte new_key[80];
1559
1560    png_debug(1, "in png_write_tEXt");
1561
1562    key_len = png_check_keyword(png_ptr, key, new_key);
1563
1564    if (key_len == 0)
1565       png_error(png_ptr, "tEXt: invalid keyword");
1566
1567    if (text == NULL || *text == '\0')
1568       text_len = 0;
1569
1570    else
1571       text_len = strlen(text);
1572
1573    if (text_len > PNG_UINT_31_MAX - (key_len+1))
1574       png_error(png_ptr, "tEXt: text too long");
1575
1576    /* Make sure we include the 0 after the key */
1577    png_write_chunk_header(png_ptr, png_tEXt,
1578        (png_uint_32)/*checked above*/(key_len + text_len + 1));
1579    /*
1580     * We leave it to the application to meet PNG-1.0 requirements on the
1581     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
1582     * any non-Latin-1 characters except for NEWLINE.  ISO PNG will forbid them.
1583     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1584     */
1585    png_write_chunk_data(png_ptr, new_key, key_len + 1);
1586
1587    if (text_len != 0)
1588       png_write_chunk_data(png_ptr, (png_const_bytep)text, text_len);
1589
1590    png_write_chunk_end(png_ptr);
1591 }
1592 #endif
1593
1594 #ifdef PNG_WRITE_zTXt_SUPPORTED
1595 /* Write a compressed text chunk */
1596 void /* PRIVATE */
1597 png_write_zTXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
1598     int compression)
1599 {
1600    png_uint_32 key_len;
1601    png_byte new_key[81];
1602    compression_state comp;
1603
1604    png_debug(1, "in png_write_zTXt");
1605
1606    if (compression == PNG_TEXT_COMPRESSION_NONE)
1607    {
1608       png_write_tEXt(png_ptr, key, text, 0);
1609       return;
1610    }
1611
1612    if (compression != PNG_TEXT_COMPRESSION_zTXt)
1613       png_error(png_ptr, "zTXt: invalid compression type");
1614
1615    key_len = png_check_keyword(png_ptr, key, new_key);
1616
1617    if (key_len == 0)
1618       png_error(png_ptr, "zTXt: invalid keyword");
1619
1620    /* Add the compression method and 1 for the keyword separator. */
1621    new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
1622    ++key_len;
1623
1624    /* Compute the compressed data; do it now for the length */
1625    png_text_compress_init(&comp, (png_const_bytep)text,
1626        text == NULL ? 0 : strlen(text));
1627
1628    if (png_text_compress(png_ptr, png_zTXt, &comp, key_len) != Z_OK)
1629       png_error(png_ptr, png_ptr->zstream.msg);
1630
1631    /* Write start of chunk */
1632    png_write_chunk_header(png_ptr, png_zTXt, key_len + comp.output_len);
1633
1634    /* Write key */
1635    png_write_chunk_data(png_ptr, new_key, key_len);
1636
1637    /* Write the compressed data */
1638    png_write_compressed_data_out(png_ptr, &comp);
1639
1640    /* Close the chunk */
1641    png_write_chunk_end(png_ptr);
1642 }
1643 #endif
1644
1645 #ifdef PNG_WRITE_iTXt_SUPPORTED
1646 /* Write an iTXt chunk */
1647 void /* PRIVATE */
1648 png_write_iTXt(png_structrp png_ptr, int compression, png_const_charp key,
1649     png_const_charp lang, png_const_charp lang_key, png_const_charp text)
1650 {
1651    png_uint_32 key_len, prefix_len;
1652    size_t lang_len, lang_key_len;
1653    png_byte new_key[82];
1654    compression_state comp;
1655
1656    png_debug(1, "in png_write_iTXt");
1657
1658    key_len = png_check_keyword(png_ptr, key, new_key);
1659
1660    if (key_len == 0)
1661       png_error(png_ptr, "iTXt: invalid keyword");
1662
1663    /* Set the compression flag */
1664    switch (compression)
1665    {
1666       case PNG_ITXT_COMPRESSION_NONE:
1667       case PNG_TEXT_COMPRESSION_NONE:
1668          compression = new_key[++key_len] = 0; /* no compression */
1669          break;
1670
1671       case PNG_TEXT_COMPRESSION_zTXt:
1672       case PNG_ITXT_COMPRESSION_zTXt:
1673          compression = new_key[++key_len] = 1; /* compressed */
1674          break;
1675
1676       default:
1677          png_error(png_ptr, "iTXt: invalid compression");
1678    }
1679
1680    new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
1681    ++key_len; /* for the keywod separator */
1682
1683    /* We leave it to the application to meet PNG-1.0 requirements on the
1684     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
1685     * any non-Latin-1 characters except for NEWLINE.  ISO PNG, however,
1686     * specifies that the text is UTF-8 and this really doesn't require any
1687     * checking.
1688     *
1689     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1690     *
1691     * TODO: validate the language tag correctly (see the spec.)
1692     */
1693    if (lang == NULL) lang = ""; /* empty language is valid */
1694    lang_len = strlen(lang)+1;
1695    if (lang_key == NULL) lang_key = ""; /* may be empty */
1696    lang_key_len = strlen(lang_key)+1;
1697    if (text == NULL) text = ""; /* may be empty */
1698
1699    prefix_len = key_len;
1700    if (lang_len > PNG_UINT_31_MAX-prefix_len)
1701       prefix_len = PNG_UINT_31_MAX;
1702    else
1703       prefix_len = (png_uint_32)(prefix_len + lang_len);
1704
1705    if (lang_key_len > PNG_UINT_31_MAX-prefix_len)
1706       prefix_len = PNG_UINT_31_MAX;
1707    else
1708       prefix_len = (png_uint_32)(prefix_len + lang_key_len);
1709
1710    png_text_compress_init(&comp, (png_const_bytep)text, strlen(text));
1711
1712    if (compression != 0)
1713    {
1714       if (png_text_compress(png_ptr, png_iTXt, &comp, prefix_len) != Z_OK)
1715          png_error(png_ptr, png_ptr->zstream.msg);
1716    }
1717
1718    else
1719    {
1720       if (comp.input_len > PNG_UINT_31_MAX-prefix_len)
1721          png_error(png_ptr, "iTXt: uncompressed text too long");
1722
1723       /* So the string will fit in a chunk: */
1724       comp.output_len = (png_uint_32)/*SAFE*/comp.input_len;
1725    }
1726
1727    png_write_chunk_header(png_ptr, png_iTXt, comp.output_len + prefix_len);
1728
1729    png_write_chunk_data(png_ptr, new_key, key_len);
1730
1731    png_write_chunk_data(png_ptr, (png_const_bytep)lang, lang_len);
1732
1733    png_write_chunk_data(png_ptr, (png_const_bytep)lang_key, lang_key_len);
1734
1735    if (compression != 0)
1736       png_write_compressed_data_out(png_ptr, &comp);
1737
1738    else
1739       png_write_chunk_data(png_ptr, (png_const_bytep)text, comp.output_len);
1740
1741    png_write_chunk_end(png_ptr);
1742 }
1743 #endif
1744
1745 #ifdef PNG_WRITE_oFFs_SUPPORTED
1746 /* Write the oFFs chunk */
1747 void /* PRIVATE */
1748 png_write_oFFs(png_structrp png_ptr, png_int_32 x_offset, png_int_32 y_offset,
1749     int unit_type)
1750 {
1751    png_byte buf[9];
1752
1753    png_debug(1, "in png_write_oFFs");
1754
1755    if (unit_type >= PNG_OFFSET_LAST)
1756       png_warning(png_ptr, "Unrecognized unit type for oFFs chunk");
1757
1758    png_save_int_32(buf, x_offset);
1759    png_save_int_32(buf + 4, y_offset);
1760    buf[8] = (png_byte)unit_type;
1761
1762    png_write_complete_chunk(png_ptr, png_oFFs, buf, 9);
1763 }
1764 #endif
1765 #ifdef PNG_WRITE_pCAL_SUPPORTED
1766 /* Write the pCAL chunk (described in the PNG extensions document) */
1767 void /* PRIVATE */
1768 png_write_pCAL(png_structrp png_ptr, png_charp purpose, png_int_32 X0,
1769     png_int_32 X1, int type, int nparams, png_const_charp units,
1770     png_charpp params)
1771 {
1772    png_uint_32 purpose_len;
1773    size_t units_len, total_len;
1774    size_t *params_len;
1775    png_byte buf[10];
1776    png_byte new_purpose[80];
1777    int i;
1778
1779    png_debug1(1, "in png_write_pCAL (%d parameters)", nparams);
1780
1781    if (type >= PNG_EQUATION_LAST)
1782       png_error(png_ptr, "Unrecognized equation type for pCAL chunk");
1783
1784    purpose_len = png_check_keyword(png_ptr, purpose, new_purpose);
1785
1786    if (purpose_len == 0)
1787       png_error(png_ptr, "pCAL: invalid keyword");
1788
1789    ++purpose_len; /* terminator */
1790
1791    png_debug1(3, "pCAL purpose length = %d", (int)purpose_len);
1792    units_len = strlen(units) + (nparams == 0 ? 0 : 1);
1793    png_debug1(3, "pCAL units length = %d", (int)units_len);
1794    total_len = purpose_len + units_len + 10;
1795
1796    params_len = (size_t *)png_malloc(png_ptr,
1797        (png_alloc_size_t)((png_alloc_size_t)nparams * (sizeof (size_t))));
1798
1799    /* Find the length of each parameter, making sure we don't count the
1800     * null terminator for the last parameter.
1801     */
1802    for (i = 0; i < nparams; i++)
1803    {
1804       params_len[i] = strlen(params[i]) + (i == nparams - 1 ? 0 : 1);
1805       png_debug2(3, "pCAL parameter %d length = %lu", i,
1806           (unsigned long)params_len[i]);
1807       total_len += params_len[i];
1808    }
1809
1810    png_debug1(3, "pCAL total length = %d", (int)total_len);
1811    png_write_chunk_header(png_ptr, png_pCAL, (png_uint_32)total_len);
1812    png_write_chunk_data(png_ptr, new_purpose, purpose_len);
1813    png_save_int_32(buf, X0);
1814    png_save_int_32(buf + 4, X1);
1815    buf[8] = (png_byte)type;
1816    buf[9] = (png_byte)nparams;
1817    png_write_chunk_data(png_ptr, buf, 10);
1818    png_write_chunk_data(png_ptr, (png_const_bytep)units, (size_t)units_len);
1819
1820    for (i = 0; i < nparams; i++)
1821    {
1822       png_write_chunk_data(png_ptr, (png_const_bytep)params[i], params_len[i]);
1823    }
1824
1825    png_free(png_ptr, params_len);
1826    png_write_chunk_end(png_ptr);
1827 }
1828 #endif
1829
1830 #ifdef PNG_WRITE_sCAL_SUPPORTED
1831 /* Write the sCAL chunk */
1832 void /* PRIVATE */
1833 png_write_sCAL_s(png_structrp png_ptr, int unit, png_const_charp width,
1834     png_const_charp height)
1835 {
1836    png_byte buf[64];
1837    size_t wlen, hlen, total_len;
1838
1839    png_debug(1, "in png_write_sCAL_s");
1840
1841    wlen = strlen(width);
1842    hlen = strlen(height);
1843    total_len = wlen + hlen + 2;
1844
1845    if (total_len > 64)
1846    {
1847       png_warning(png_ptr, "Can't write sCAL (buffer too small)");
1848       return;
1849    }
1850
1851    buf[0] = (png_byte)unit;
1852    memcpy(buf + 1, width, wlen + 1);      /* Append the '\0' here */
1853    memcpy(buf + wlen + 2, height, hlen);  /* Do NOT append the '\0' here */
1854
1855    png_debug1(3, "sCAL total length = %u", (unsigned int)total_len);
1856    png_write_complete_chunk(png_ptr, png_sCAL, buf, total_len);
1857 }
1858 #endif
1859
1860 #ifdef PNG_WRITE_pHYs_SUPPORTED
1861 /* Write the pHYs chunk */
1862 void /* PRIVATE */
1863 png_write_pHYs(png_structrp png_ptr, png_uint_32 x_pixels_per_unit,
1864     png_uint_32 y_pixels_per_unit,
1865     int unit_type)
1866 {
1867    png_byte buf[9];
1868
1869    png_debug(1, "in png_write_pHYs");
1870
1871    if (unit_type >= PNG_RESOLUTION_LAST)
1872       png_warning(png_ptr, "Unrecognized unit type for pHYs chunk");
1873
1874    png_save_uint_32(buf, x_pixels_per_unit);
1875    png_save_uint_32(buf + 4, y_pixels_per_unit);
1876    buf[8] = (png_byte)unit_type;
1877
1878    png_write_complete_chunk(png_ptr, png_pHYs, buf, 9);
1879 }
1880 #endif
1881
1882 #ifdef PNG_WRITE_tIME_SUPPORTED
1883 /* Write the tIME chunk.  Use either png_convert_from_struct_tm()
1884  * or png_convert_from_time_t(), or fill in the structure yourself.
1885  */
1886 void /* PRIVATE */
1887 png_write_tIME(png_structrp png_ptr, png_const_timep mod_time)
1888 {
1889    png_byte buf[7];
1890
1891    png_debug(1, "in png_write_tIME");
1892
1893    if (mod_time->month  > 12 || mod_time->month  < 1 ||
1894        mod_time->day    > 31 || mod_time->day    < 1 ||
1895        mod_time->hour   > 23 || mod_time->second > 60)
1896    {
1897       png_warning(png_ptr, "Invalid time specified for tIME chunk");
1898       return;
1899    }
1900
1901    png_save_uint_16(buf, mod_time->year);
1902    buf[2] = mod_time->month;
1903    buf[3] = mod_time->day;
1904    buf[4] = mod_time->hour;
1905    buf[5] = mod_time->minute;
1906    buf[6] = mod_time->second;
1907
1908    png_write_complete_chunk(png_ptr, png_tIME, buf, 7);
1909 }
1910 #endif
1911
1912 #ifdef PNG_WRITE_APNG_SUPPORTED
1913 void /* PRIVATE */
1914 png_write_acTL(png_structp png_ptr,
1915     png_uint_32 num_frames, png_uint_32 num_plays)
1916 {
1917     png_byte buf[8];
1918
1919     png_debug(1, "in png_write_acTL");
1920
1921     png_ptr->num_frames_to_write = num_frames;
1922
1923     if (png_ptr->apng_flags & PNG_FIRST_FRAME_HIDDEN)
1924         num_frames--;
1925
1926     png_save_uint_32(buf, num_frames);
1927     png_save_uint_32(buf + 4, num_plays);
1928
1929     png_write_complete_chunk(png_ptr, png_acTL, buf, (png_size_t)8);
1930 }
1931
1932 void /* PRIVATE */
1933 png_write_fcTL(png_structp png_ptr, png_uint_32 width, png_uint_32 height,
1934     png_uint_32 x_offset, png_uint_32 y_offset,
1935     png_uint_16 delay_num, png_uint_16 delay_den, png_byte dispose_op,
1936     png_byte blend_op)
1937 {
1938     png_byte buf[26];
1939
1940     png_debug(1, "in png_write_fcTL");
1941
1942     if (png_ptr->num_frames_written == 0 && (x_offset != 0 || y_offset != 0))
1943         png_error(png_ptr, "x and/or y offset for the first frame aren't 0");
1944     if (png_ptr->num_frames_written == 0 &&
1945         (width != png_ptr->first_frame_width ||
1946          height != png_ptr->first_frame_height))
1947         png_error(png_ptr, "width and/or height in the first frame's fcTL "
1948                            "don't match the ones in IHDR");
1949
1950     /* more error checking */
1951     png_ensure_fcTL_is_valid(png_ptr, width, height, x_offset, y_offset,
1952                              delay_num, delay_den, dispose_op, blend_op);
1953
1954     png_save_uint_32(buf, png_ptr->next_seq_num);
1955     png_save_uint_32(buf + 4, width);
1956     png_save_uint_32(buf + 8, height);
1957     png_save_uint_32(buf + 12, x_offset);
1958     png_save_uint_32(buf + 16, y_offset);
1959     png_save_uint_16(buf + 20, delay_num);
1960     png_save_uint_16(buf + 22, delay_den);
1961     buf[24] = dispose_op;
1962     buf[25] = blend_op;
1963
1964     png_write_complete_chunk(png_ptr, png_fcTL, buf, (png_size_t)26);
1965
1966     png_ptr->next_seq_num++;
1967 }
1968
1969 void /* PRIVATE */
1970 png_write_fdAT(png_structp png_ptr,
1971     png_const_bytep data, png_size_t length)
1972 {
1973     png_byte buf[4];
1974
1975     png_write_chunk_header(png_ptr, png_fdAT, (png_uint_32)(4 + length));
1976
1977     png_save_uint_32(buf, png_ptr->next_seq_num);
1978     png_write_chunk_data(png_ptr, buf, 4);
1979
1980     png_write_chunk_data(png_ptr, data, length);
1981
1982     png_write_chunk_end(png_ptr);
1983
1984     png_ptr->next_seq_num++;
1985 }
1986 #endif /* PNG_WRITE_APNG_SUPPORTED */
1987
1988 /* Initializes the row writing capability of libpng */
1989 void /* PRIVATE */
1990 png_write_start_row(png_structrp png_ptr)
1991 {
1992 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1993    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1994
1995    /* Start of interlace block */
1996    static const png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1997
1998    /* Offset to next interlace block */
1999    static const png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
2000
2001    /* Start of interlace block in the y direction */
2002    static const png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
2003
2004    /* Offset to next interlace block in the y direction */
2005    static const png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
2006 #endif
2007
2008    png_alloc_size_t buf_size;
2009    int usr_pixel_depth;
2010
2011 #ifdef PNG_WRITE_FILTER_SUPPORTED
2012    png_byte filters;
2013 #endif
2014
2015    png_debug(1, "in png_write_start_row");
2016
2017    usr_pixel_depth = png_ptr->usr_channels * png_ptr->usr_bit_depth;
2018    buf_size = PNG_ROWBYTES(usr_pixel_depth, png_ptr->width) + 1;
2019
2020    /* 1.5.6: added to allow checking in the row write code. */
2021    png_ptr->transformed_pixel_depth = png_ptr->pixel_depth;
2022    png_ptr->maximum_pixel_depth = (png_byte)usr_pixel_depth;
2023
2024    /* Set up row buffer */
2025    png_ptr->row_buf = png_voidcast(png_bytep, png_malloc(png_ptr, buf_size));
2026
2027    png_ptr->row_buf[0] = PNG_FILTER_VALUE_NONE;
2028
2029 #ifdef PNG_WRITE_FILTER_SUPPORTED
2030    filters = png_ptr->do_filter;
2031
2032    if (png_ptr->height == 1)
2033       filters &= 0xff & ~(PNG_FILTER_UP|PNG_FILTER_AVG|PNG_FILTER_PAETH);
2034
2035    if (png_ptr->width == 1)
2036       filters &= 0xff & ~(PNG_FILTER_SUB|PNG_FILTER_AVG|PNG_FILTER_PAETH);
2037
2038    if (filters == 0)
2039       filters = PNG_FILTER_NONE;
2040
2041    png_ptr->do_filter = filters;
2042
2043    if (((filters & (PNG_FILTER_SUB | PNG_FILTER_UP | PNG_FILTER_AVG |
2044        PNG_FILTER_PAETH)) != 0) && png_ptr->try_row == NULL)
2045    {
2046       int num_filters = 0;
2047
2048       png_ptr->try_row = png_voidcast(png_bytep, png_malloc(png_ptr, buf_size));
2049
2050       if (filters & PNG_FILTER_SUB)
2051          num_filters++;
2052
2053       if (filters & PNG_FILTER_UP)
2054          num_filters++;
2055
2056       if (filters & PNG_FILTER_AVG)
2057          num_filters++;
2058
2059       if (filters & PNG_FILTER_PAETH)
2060          num_filters++;
2061
2062       if (num_filters > 1)
2063          png_ptr->tst_row = png_voidcast(png_bytep, png_malloc(png_ptr,
2064              buf_size));
2065    }
2066
2067    /* We only need to keep the previous row if we are using one of the following
2068     * filters.
2069     */
2070    if ((filters & (PNG_FILTER_AVG | PNG_FILTER_UP | PNG_FILTER_PAETH)) != 0)
2071       png_ptr->prev_row = png_voidcast(png_bytep,
2072           png_calloc(png_ptr, buf_size));
2073 #endif /* WRITE_FILTER */
2074
2075 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
2076    /* If interlaced, we need to set up width and height of pass */
2077    if (png_ptr->interlaced != 0)
2078    {
2079       if ((png_ptr->transformations & PNG_INTERLACE) == 0)
2080       {
2081          png_ptr->num_rows = (png_ptr->height + png_pass_yinc[0] - 1 -
2082              png_pass_ystart[0]) / png_pass_yinc[0];
2083
2084          png_ptr->usr_width = (png_ptr->width + png_pass_inc[0] - 1 -
2085              png_pass_start[0]) / png_pass_inc[0];
2086       }
2087
2088       else
2089       {
2090          png_ptr->num_rows = png_ptr->height;
2091          png_ptr->usr_width = png_ptr->width;
2092       }
2093    }
2094
2095    else
2096 #endif
2097    {
2098       png_ptr->num_rows = png_ptr->height;
2099       png_ptr->usr_width = png_ptr->width;
2100    }
2101 }
2102
2103 /* Internal use only.  Called when finished processing a row of data. */
2104 void /* PRIVATE */
2105 png_write_finish_row(png_structrp png_ptr)
2106 {
2107 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
2108    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
2109
2110    /* Start of interlace block */
2111    static const png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
2112
2113    /* Offset to next interlace block */
2114    static const png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
2115
2116    /* Start of interlace block in the y direction */
2117    static const png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
2118
2119    /* Offset to next interlace block in the y direction */
2120    static const png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
2121 #endif
2122
2123    png_debug(1, "in png_write_finish_row");
2124
2125    /* Next row */
2126    png_ptr->row_number++;
2127
2128    /* See if we are done */
2129    if (png_ptr->row_number < png_ptr->num_rows)
2130       return;
2131
2132 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
2133    /* If interlaced, go to next pass */
2134    if (png_ptr->interlaced != 0)
2135    {
2136       png_ptr->row_number = 0;
2137       if ((png_ptr->transformations & PNG_INTERLACE) != 0)
2138       {
2139          png_ptr->pass++;
2140       }
2141
2142       else
2143       {
2144          /* Loop until we find a non-zero width or height pass */
2145          do
2146          {
2147             png_ptr->pass++;
2148
2149             if (png_ptr->pass >= 7)
2150                break;
2151
2152             png_ptr->usr_width = (png_ptr->width +
2153                 png_pass_inc[png_ptr->pass] - 1 -
2154                 png_pass_start[png_ptr->pass]) /
2155                 png_pass_inc[png_ptr->pass];
2156
2157             png_ptr->num_rows = (png_ptr->height +
2158                 png_pass_yinc[png_ptr->pass] - 1 -
2159                 png_pass_ystart[png_ptr->pass]) /
2160                 png_pass_yinc[png_ptr->pass];
2161
2162             if ((png_ptr->transformations & PNG_INTERLACE) != 0)
2163                break;
2164
2165          } while (png_ptr->usr_width == 0 || png_ptr->num_rows == 0);
2166
2167       }
2168
2169       /* Reset the row above the image for the next pass */
2170       if (png_ptr->pass < 7)
2171       {
2172          if (png_ptr->prev_row != NULL)
2173             memset(png_ptr->prev_row, 0,
2174                 PNG_ROWBYTES(png_ptr->usr_channels *
2175                 png_ptr->usr_bit_depth, png_ptr->width) + 1);
2176
2177          return;
2178       }
2179    }
2180 #endif
2181
2182    /* If we get here, we've just written the last row, so we need
2183       to flush the compressor */
2184    png_compress_IDAT(png_ptr, NULL, 0, Z_FINISH);
2185 }
2186
2187 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
2188 /* Pick out the correct pixels for the interlace pass.
2189  * The basic idea here is to go through the row with a source
2190  * pointer and a destination pointer (sp and dp), and copy the
2191  * correct pixels for the pass.  As the row gets compacted,
2192  * sp will always be >= dp, so we should never overwrite anything.
2193  * See the default: case for the easiest code to understand.
2194  */
2195 void /* PRIVATE */
2196 png_do_write_interlace(png_row_infop row_info, png_bytep row, int pass)
2197 {
2198    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
2199
2200    /* Start of interlace block */
2201    static const png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
2202
2203    /* Offset to next interlace block */
2204    static const png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
2205
2206    png_debug(1, "in png_do_write_interlace");
2207
2208    /* We don't have to do anything on the last pass (6) */
2209    if (pass < 6)
2210    {
2211       /* Each pixel depth is handled separately */
2212       switch (row_info->pixel_depth)
2213       {
2214          case 1:
2215          {
2216             png_bytep sp;
2217             png_bytep dp;
2218             unsigned int shift;
2219             int d;
2220             int value;
2221             png_uint_32 i;
2222             png_uint_32 row_width = row_info->width;
2223
2224             dp = row;
2225             d = 0;
2226             shift = 7;
2227
2228             for (i = png_pass_start[pass]; i < row_width;
2229                i += png_pass_inc[pass])
2230             {
2231                sp = row + (size_t)(i >> 3);
2232                value = (int)(*sp >> (7 - (int)(i & 0x07))) & 0x01;
2233                d |= (value << shift);
2234
2235                if (shift == 0)
2236                {
2237                   shift = 7;
2238                   *dp++ = (png_byte)d;
2239                   d = 0;
2240                }
2241
2242                else
2243                   shift--;
2244
2245             }
2246             if (shift != 7)
2247                *dp = (png_byte)d;
2248
2249             break;
2250          }
2251
2252          case 2:
2253          {
2254             png_bytep sp;
2255             png_bytep dp;
2256             unsigned int shift;
2257             int d;
2258             int value;
2259             png_uint_32 i;
2260             png_uint_32 row_width = row_info->width;
2261
2262             dp = row;
2263             shift = 6;
2264             d = 0;
2265
2266             for (i = png_pass_start[pass]; i < row_width;
2267                i += png_pass_inc[pass])
2268             {
2269                sp = row + (size_t)(i >> 2);
2270                value = (*sp >> ((3 - (int)(i & 0x03)) << 1)) & 0x03;
2271                d |= (value << shift);
2272
2273                if (shift == 0)
2274                {
2275                   shift = 6;
2276                   *dp++ = (png_byte)d;
2277                   d = 0;
2278                }
2279
2280                else
2281                   shift -= 2;
2282             }
2283             if (shift != 6)
2284                *dp = (png_byte)d;
2285
2286             break;
2287          }
2288
2289          case 4:
2290          {
2291             png_bytep sp;
2292             png_bytep dp;
2293             unsigned int shift;
2294             int d;
2295             int value;
2296             png_uint_32 i;
2297             png_uint_32 row_width = row_info->width;
2298
2299             dp = row;
2300             shift = 4;
2301             d = 0;
2302             for (i = png_pass_start[pass]; i < row_width;
2303                 i += png_pass_inc[pass])
2304             {
2305                sp = row + (size_t)(i >> 1);
2306                value = (*sp >> ((1 - (int)(i & 0x01)) << 2)) & 0x0f;
2307                d |= (value << shift);
2308
2309                if (shift == 0)
2310                {
2311                   shift = 4;
2312                   *dp++ = (png_byte)d;
2313                   d = 0;
2314                }
2315
2316                else
2317                   shift -= 4;
2318             }
2319             if (shift != 4)
2320                *dp = (png_byte)d;
2321
2322             break;
2323          }
2324
2325          default:
2326          {
2327             png_bytep sp;
2328             png_bytep dp;
2329             png_uint_32 i;
2330             png_uint_32 row_width = row_info->width;
2331             size_t pixel_bytes;
2332
2333             /* Start at the beginning */
2334             dp = row;
2335
2336             /* Find out how many bytes each pixel takes up */
2337             pixel_bytes = (row_info->pixel_depth >> 3);
2338
2339             /* Loop through the row, only looking at the pixels that matter */
2340             for (i = png_pass_start[pass]; i < row_width;
2341                i += png_pass_inc[pass])
2342             {
2343                /* Find out where the original pixel is */
2344                sp = row + (size_t)i * pixel_bytes;
2345
2346                /* Move the pixel */
2347                if (dp != sp)
2348                   memcpy(dp, sp, pixel_bytes);
2349
2350                /* Next pixel */
2351                dp += pixel_bytes;
2352             }
2353             break;
2354          }
2355       }
2356       /* Set new row width */
2357       row_info->width = (row_info->width +
2358           png_pass_inc[pass] - 1 -
2359           png_pass_start[pass]) /
2360           png_pass_inc[pass];
2361
2362       row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth,
2363           row_info->width);
2364    }
2365 }
2366 #endif
2367
2368
2369 /* This filters the row, chooses which filter to use, if it has not already
2370  * been specified by the application, and then writes the row out with the
2371  * chosen filter.
2372  */
2373 static void /* PRIVATE */
2374 png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
2375     size_t row_bytes);
2376
2377 #ifdef PNG_WRITE_FILTER_SUPPORTED
2378 static size_t /* PRIVATE */
2379 png_setup_sub_row(png_structrp png_ptr, png_uint_32 bpp,
2380     size_t row_bytes, size_t lmins)
2381 {
2382    png_bytep rp, dp, lp;
2383    size_t i;
2384    size_t sum = 0;
2385    unsigned int v;
2386
2387    png_ptr->try_row[0] = PNG_FILTER_VALUE_SUB;
2388
2389    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1; i < bpp;
2390         i++, rp++, dp++)
2391    {
2392       v = *dp = *rp;
2393 #ifdef PNG_USE_ABS
2394       sum += 128 - abs((int)v - 128);
2395 #else
2396       sum += (v < 128) ? v : 256 - v;
2397 #endif
2398    }
2399
2400    for (lp = png_ptr->row_buf + 1; i < row_bytes;
2401       i++, rp++, lp++, dp++)
2402    {
2403       v = *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2404 #ifdef PNG_USE_ABS
2405       sum += 128 - abs((int)v - 128);
2406 #else
2407       sum += (v < 128) ? v : 256 - v;
2408 #endif
2409
2410       if (sum > lmins)  /* We are already worse, don't continue. */
2411         break;
2412    }
2413
2414    return (sum);
2415 }
2416
2417 static void /* PRIVATE */
2418 png_setup_sub_row_only(png_structrp png_ptr, png_uint_32 bpp,
2419     size_t row_bytes)
2420 {
2421    png_bytep rp, dp, lp;
2422    size_t i;
2423
2424    png_ptr->try_row[0] = PNG_FILTER_VALUE_SUB;
2425
2426    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1; i < bpp;
2427         i++, rp++, dp++)
2428    {
2429       *dp = *rp;
2430    }
2431
2432    for (lp = png_ptr->row_buf + 1; i < row_bytes;
2433       i++, rp++, lp++, dp++)
2434    {
2435       *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2436    }
2437 }
2438
2439 static size_t /* PRIVATE */
2440 png_setup_up_row(png_structrp png_ptr, size_t row_bytes, size_t lmins)
2441 {
2442    png_bytep rp, dp, pp;
2443    size_t i;
2444    size_t sum = 0;
2445    unsigned int v;
2446
2447    png_ptr->try_row[0] = PNG_FILTER_VALUE_UP;
2448
2449    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2450        pp = png_ptr->prev_row + 1; i < row_bytes;
2451        i++, rp++, pp++, dp++)
2452    {
2453       v = *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
2454 #ifdef PNG_USE_ABS
2455       sum += 128 - abs((int)v - 128);
2456 #else
2457       sum += (v < 128) ? v : 256 - v;
2458 #endif
2459
2460       if (sum > lmins)  /* We are already worse, don't continue. */
2461         break;
2462    }
2463
2464    return (sum);
2465 }
2466 static void /* PRIVATE */
2467 png_setup_up_row_only(png_structrp png_ptr, size_t row_bytes)
2468 {
2469    png_bytep rp, dp, pp;
2470    size_t i;
2471
2472    png_ptr->try_row[0] = PNG_FILTER_VALUE_UP;
2473
2474    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2475        pp = png_ptr->prev_row + 1; i < row_bytes;
2476        i++, rp++, pp++, dp++)
2477    {
2478       *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
2479    }
2480 }
2481
2482 static size_t /* PRIVATE */
2483 png_setup_avg_row(png_structrp png_ptr, png_uint_32 bpp,
2484     size_t row_bytes, size_t lmins)
2485 {
2486    png_bytep rp, dp, pp, lp;
2487    png_uint_32 i;
2488    size_t sum = 0;
2489    unsigned int v;
2490
2491    png_ptr->try_row[0] = PNG_FILTER_VALUE_AVG;
2492
2493    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2494        pp = png_ptr->prev_row + 1; i < bpp; i++)
2495    {
2496       v = *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2497
2498 #ifdef PNG_USE_ABS
2499       sum += 128 - abs((int)v - 128);
2500 #else
2501       sum += (v < 128) ? v : 256 - v;
2502 #endif
2503    }
2504
2505    for (lp = png_ptr->row_buf + 1; i < row_bytes; i++)
2506    {
2507       v = *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
2508           & 0xff);
2509
2510 #ifdef PNG_USE_ABS
2511       sum += 128 - abs((int)v - 128);
2512 #else
2513       sum += (v < 128) ? v : 256 - v;
2514 #endif
2515
2516       if (sum > lmins)  /* We are already worse, don't continue. */
2517         break;
2518    }
2519
2520    return (sum);
2521 }
2522 static void /* PRIVATE */
2523 png_setup_avg_row_only(png_structrp png_ptr, png_uint_32 bpp,
2524     size_t row_bytes)
2525 {
2526    png_bytep rp, dp, pp, lp;
2527    png_uint_32 i;
2528
2529    png_ptr->try_row[0] = PNG_FILTER_VALUE_AVG;
2530
2531    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2532        pp = png_ptr->prev_row + 1; i < bpp; i++)
2533    {
2534       *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2535    }
2536
2537    for (lp = png_ptr->row_buf + 1; i < row_bytes; i++)
2538    {
2539       *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
2540           & 0xff);
2541    }
2542 }
2543
2544 static size_t /* PRIVATE */
2545 png_setup_paeth_row(png_structrp png_ptr, png_uint_32 bpp,
2546     size_t row_bytes, size_t lmins)
2547 {
2548    png_bytep rp, dp, pp, cp, lp;
2549    size_t i;
2550    size_t sum = 0;
2551    unsigned int v;
2552
2553    png_ptr->try_row[0] = PNG_FILTER_VALUE_PAETH;
2554
2555    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2556        pp = png_ptr->prev_row + 1; i < bpp; i++)
2557    {
2558       v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2559
2560 #ifdef PNG_USE_ABS
2561       sum += 128 - abs((int)v - 128);
2562 #else
2563       sum += (v < 128) ? v : 256 - v;
2564 #endif
2565    }
2566
2567    for (lp = png_ptr->row_buf + 1, cp = png_ptr->prev_row + 1; i < row_bytes;
2568         i++)
2569    {
2570       int a, b, c, pa, pb, pc, p;
2571
2572       b = *pp++;
2573       c = *cp++;
2574       a = *lp++;
2575
2576       p = b - c;
2577       pc = a - c;
2578
2579 #ifdef PNG_USE_ABS
2580       pa = abs(p);
2581       pb = abs(pc);
2582       pc = abs(p + pc);
2583 #else
2584       pa = p < 0 ? -p : p;
2585       pb = pc < 0 ? -pc : pc;
2586       pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2587 #endif
2588
2589       p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2590
2591       v = *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2592
2593 #ifdef PNG_USE_ABS
2594       sum += 128 - abs((int)v - 128);
2595 #else
2596       sum += (v < 128) ? v : 256 - v;
2597 #endif
2598
2599       if (sum > lmins)  /* We are already worse, don't continue. */
2600         break;
2601    }
2602
2603    return (sum);
2604 }
2605 static void /* PRIVATE */
2606 png_setup_paeth_row_only(png_structrp png_ptr, png_uint_32 bpp,
2607     size_t row_bytes)
2608 {
2609    png_bytep rp, dp, pp, cp, lp;
2610    size_t i;
2611
2612    png_ptr->try_row[0] = PNG_FILTER_VALUE_PAETH;
2613
2614    for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2615        pp = png_ptr->prev_row + 1; i < bpp; i++)
2616    {
2617       *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2618    }
2619
2620    for (lp = png_ptr->row_buf + 1, cp = png_ptr->prev_row + 1; i < row_bytes;
2621         i++)
2622    {
2623       int a, b, c, pa, pb, pc, p;
2624
2625       b = *pp++;
2626       c = *cp++;
2627       a = *lp++;
2628
2629       p = b - c;
2630       pc = a - c;
2631
2632 #ifdef PNG_USE_ABS
2633       pa = abs(p);
2634       pb = abs(pc);
2635       pc = abs(p + pc);
2636 #else
2637       pa = p < 0 ? -p : p;
2638       pb = pc < 0 ? -pc : pc;
2639       pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2640 #endif
2641
2642       p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2643
2644       *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2645    }
2646 }
2647 #endif /* WRITE_FILTER */
2648
2649 void /* PRIVATE */
2650 png_write_find_filter(png_structrp png_ptr, png_row_infop row_info)
2651 {
2652 #ifndef PNG_WRITE_FILTER_SUPPORTED
2653    png_write_filtered_row(png_ptr, png_ptr->row_buf, row_info->rowbytes+1);
2654 #else
2655    unsigned int filter_to_do = png_ptr->do_filter;
2656    png_bytep row_buf;
2657    png_bytep best_row;
2658    png_uint_32 bpp;
2659    size_t mins;
2660    size_t row_bytes = row_info->rowbytes;
2661
2662    png_debug(1, "in png_write_find_filter");
2663
2664    /* Find out how many bytes offset each pixel is */
2665    bpp = (row_info->pixel_depth + 7) >> 3;
2666
2667    row_buf = png_ptr->row_buf;
2668    mins = PNG_SIZE_MAX - 256/* so we can detect potential overflow of the
2669                                running sum */;
2670
2671    /* The prediction method we use is to find which method provides the
2672     * smallest value when summing the absolute values of the distances
2673     * from zero, using anything >= 128 as negative numbers.  This is known
2674     * as the "minimum sum of absolute differences" heuristic.  Other
2675     * heuristics are the "weighted minimum sum of absolute differences"
2676     * (experimental and can in theory improve compression), and the "zlib
2677     * predictive" method (not implemented yet), which does test compressions
2678     * of lines using different filter methods, and then chooses the
2679     * (series of) filter(s) that give minimum compressed data size (VERY
2680     * computationally expensive).
2681     *
2682     * GRR 980525:  consider also
2683     *
2684     *   (1) minimum sum of absolute differences from running average (i.e.,
2685     *       keep running sum of non-absolute differences & count of bytes)
2686     *       [track dispersion, too?  restart average if dispersion too large?]
2687     *
2688     *  (1b) minimum sum of absolute differences from sliding average, probably
2689     *       with window size <= deflate window (usually 32K)
2690     *
2691     *   (2) minimum sum of squared differences from zero or running average
2692     *       (i.e., ~ root-mean-square approach)
2693     */
2694
2695
2696    /* We don't need to test the 'no filter' case if this is the only filter
2697     * that has been chosen, as it doesn't actually do anything to the data.
2698     */
2699    best_row = png_ptr->row_buf;
2700
2701    if (PNG_SIZE_MAX/128 <= row_bytes)
2702    {
2703       /* Overflow can occur in the calculation, just select the lowest set
2704        * filter.
2705        */
2706       filter_to_do &= 0U-filter_to_do;
2707    }
2708    else if ((filter_to_do & PNG_FILTER_NONE) != 0 &&
2709          filter_to_do != PNG_FILTER_NONE)
2710    {
2711       /* Overflow not possible and multiple filters in the list, including the
2712        * 'none' filter.
2713        */
2714       png_bytep rp;
2715       size_t sum = 0;
2716       size_t i;
2717       unsigned int v;
2718
2719       {
2720          for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++)
2721          {
2722             v = *rp;
2723 #ifdef PNG_USE_ABS
2724             sum += 128 - abs((int)v - 128);
2725 #else
2726             sum += (v < 128) ? v : 256 - v;
2727 #endif
2728          }
2729       }
2730
2731       mins = sum;
2732    }
2733
2734    /* Sub filter */
2735    if (filter_to_do == PNG_FILTER_SUB)
2736    /* It's the only filter so no testing is needed */
2737    {
2738       png_setup_sub_row_only(png_ptr, bpp, row_bytes);
2739       best_row = png_ptr->try_row;
2740    }
2741
2742    else if ((filter_to_do & PNG_FILTER_SUB) != 0)
2743    {
2744       size_t sum;
2745       size_t lmins = mins;
2746
2747       sum = png_setup_sub_row(png_ptr, bpp, row_bytes, lmins);
2748
2749       if (sum < mins)
2750       {
2751          mins = sum;
2752          best_row = png_ptr->try_row;
2753          if (png_ptr->tst_row != NULL)
2754          {
2755             png_ptr->try_row = png_ptr->tst_row;
2756             png_ptr->tst_row = best_row;
2757          }
2758       }
2759    }
2760
2761    /* Up filter */
2762    if (filter_to_do == PNG_FILTER_UP)
2763    {
2764       png_setup_up_row_only(png_ptr, row_bytes);
2765       best_row = png_ptr->try_row;
2766    }
2767
2768    else if ((filter_to_do & PNG_FILTER_UP) != 0)
2769    {
2770       size_t sum;
2771       size_t lmins = mins;
2772
2773       sum = png_setup_up_row(png_ptr, row_bytes, lmins);
2774
2775       if (sum < mins)
2776       {
2777          mins = sum;
2778          best_row = png_ptr->try_row;
2779          if (png_ptr->tst_row != NULL)
2780          {
2781             png_ptr->try_row = png_ptr->tst_row;
2782             png_ptr->tst_row = best_row;
2783          }
2784       }
2785    }
2786
2787    /* Avg filter */
2788    if (filter_to_do == PNG_FILTER_AVG)
2789    {
2790       png_setup_avg_row_only(png_ptr, bpp, row_bytes);
2791       best_row = png_ptr->try_row;
2792    }
2793
2794    else if ((filter_to_do & PNG_FILTER_AVG) != 0)
2795    {
2796       size_t sum;
2797       size_t lmins = mins;
2798
2799       sum= png_setup_avg_row(png_ptr, bpp, row_bytes, lmins);
2800
2801       if (sum < mins)
2802       {
2803          mins = sum;
2804          best_row = png_ptr->try_row;
2805          if (png_ptr->tst_row != NULL)
2806          {
2807             png_ptr->try_row = png_ptr->tst_row;
2808             png_ptr->tst_row = best_row;
2809          }
2810       }
2811    }
2812
2813    /* Paeth filter */
2814    if (filter_to_do == PNG_FILTER_PAETH)
2815    {
2816       png_setup_paeth_row_only(png_ptr, bpp, row_bytes);
2817       best_row = png_ptr->try_row;
2818    }
2819
2820    else if ((filter_to_do & PNG_FILTER_PAETH) != 0)
2821    {
2822       size_t sum;
2823       size_t lmins = mins;
2824
2825       sum = png_setup_paeth_row(png_ptr, bpp, row_bytes, lmins);
2826
2827       if (sum < mins)
2828       {
2829          best_row = png_ptr->try_row;
2830          if (png_ptr->tst_row != NULL)
2831          {
2832             png_ptr->try_row = png_ptr->tst_row;
2833             png_ptr->tst_row = best_row;
2834          }
2835       }
2836    }
2837
2838    /* Do the actual writing of the filtered row data from the chosen filter. */
2839    png_write_filtered_row(png_ptr, best_row, row_info->rowbytes+1);
2840
2841 #endif /* WRITE_FILTER */
2842 }
2843
2844
2845 /* Do the actual writing of a previously filtered row. */
2846 static void
2847 png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
2848     size_t full_row_length/*includes filter byte*/)
2849 {
2850    png_debug(1, "in png_write_filtered_row");
2851
2852    png_debug1(2, "filter = %d", filtered_row[0]);
2853
2854    png_compress_IDAT(png_ptr, filtered_row, full_row_length, Z_NO_FLUSH);
2855
2856 #ifdef PNG_WRITE_FILTER_SUPPORTED
2857    /* Swap the current and previous rows */
2858    if (png_ptr->prev_row != NULL)
2859    {
2860       png_bytep tptr;
2861
2862       tptr = png_ptr->prev_row;
2863       png_ptr->prev_row = png_ptr->row_buf;
2864       png_ptr->row_buf = tptr;
2865    }
2866 #endif /* WRITE_FILTER */
2867
2868    /* Finish row - updates counters and flushes zlib if last row */
2869    png_write_finish_row(png_ptr);
2870
2871 #ifdef PNG_WRITE_FLUSH_SUPPORTED
2872    png_ptr->flush_rows++;
2873
2874    if (png_ptr->flush_dist > 0 &&
2875        png_ptr->flush_rows >= png_ptr->flush_dist)
2876    {
2877       png_write_flush(png_ptr);
2878    }
2879 #endif /* WRITE_FLUSH */
2880 }
2881
2882 #ifdef PNG_WRITE_APNG_SUPPORTED
2883 void /* PRIVATE */
2884 png_write_reset(png_structp png_ptr)
2885 {
2886     png_ptr->row_number = 0;
2887     png_ptr->pass = 0;
2888     png_ptr->mode &= ~PNG_HAVE_IDAT;
2889 }
2890
2891 void /* PRIVATE */
2892 png_write_reinit(png_structp png_ptr, png_infop info_ptr,
2893                  png_uint_32 width, png_uint_32 height)
2894 {
2895     if (png_ptr->num_frames_written == 0 &&
2896         (width != png_ptr->first_frame_width ||
2897          height != png_ptr->first_frame_height))
2898         png_error(png_ptr, "width and/or height in the first frame's fcTL "
2899                            "don't match the ones in IHDR");
2900     if (width > png_ptr->first_frame_width ||
2901         height > png_ptr->first_frame_height)
2902         png_error(png_ptr, "width and/or height for a frame greater than"
2903                            "the ones in IHDR");
2904
2905     png_set_IHDR(png_ptr, info_ptr, width, height,
2906                  info_ptr->bit_depth, info_ptr->color_type,
2907                  info_ptr->interlace_type, info_ptr->compression_type,
2908                  info_ptr->filter_type);
2909
2910     png_ptr->width = width;
2911     png_ptr->height = height;
2912     png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width);
2913     png_ptr->usr_width = png_ptr->width;
2914 }
2915 #endif /* PNG_WRITE_APNG_SUPPORTED */
2916 #endif /* WRITE */