Home | History | Annotate | Download | only in lpng_v163
      1 
      2 /* pngwutil.c - utilities to write a PNG file
      3  *
      4  * Last changed in libpng 1.6.2 [April 25, 2013]
      5  * Copyright (c) 1998-2013 Glenn Randers-Pehrson
      6  * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
      7  * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
      8  *
      9  * This code is released under the libpng license.
     10  * For conditions of distribution and use, see the disclaimer
     11  * and license in png.h
     12  */
     13 #include "pngpriv.h"
     14 
     15 #ifdef PNG_WRITE_SUPPORTED
     16 
     17 #ifdef PNG_WRITE_INT_FUNCTIONS_SUPPORTED
     18 /* Place a 32-bit number into a buffer in PNG byte order.  We work
     19  * with unsigned numbers for convenience, although one supported
     20  * ancillary chunk uses signed (two's complement) numbers.
     21  */
     22 void PNGAPI
     23 png_save_uint_32(png_bytep buf, png_uint_32 i)
     24 {
     25    buf[0] = (png_byte)((i >> 24) & 0xff);
     26    buf[1] = (png_byte)((i >> 16) & 0xff);
     27    buf[2] = (png_byte)((i >> 8) & 0xff);
     28    buf[3] = (png_byte)(i & 0xff);
     29 }
     30 
     31 /* Place a 16-bit number into a buffer in PNG byte order.
     32  * The parameter is declared unsigned int, not png_uint_16,
     33  * just to avoid potential problems on pre-ANSI C compilers.
     34  */
     35 void PNGAPI
     36 png_save_uint_16(png_bytep buf, unsigned int i)
     37 {
     38    buf[0] = (png_byte)((i >> 8) & 0xff);
     39    buf[1] = (png_byte)(i & 0xff);
     40 }
     41 #endif
     42 
     43 /* Simple function to write the signature.  If we have already written
     44  * the magic bytes of the signature, or more likely, the PNG stream is
     45  * being embedded into another stream and doesn't need its own signature,
     46  * we should call png_set_sig_bytes() to tell libpng how many of the
     47  * bytes have already been written.
     48  */
     49 void PNGAPI
     50 png_write_sig(png_structrp png_ptr)
     51 {
     52    png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10};
     53 
     54 #ifdef PNG_IO_STATE_SUPPORTED
     55    /* Inform the I/O callback that the signature is being written */
     56    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_SIGNATURE;
     57 #endif
     58 
     59    /* Write the rest of the 8 byte signature */
     60    png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes],
     61       (png_size_t)(8 - png_ptr->sig_bytes));
     62 
     63    if (png_ptr->sig_bytes < 3)
     64       png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE;
     65 }
     66 
     67 /* Write the start of a PNG chunk.  The type is the chunk type.
     68  * The total_length is the sum of the lengths of all the data you will be
     69  * passing in png_write_chunk_data().
     70  */
     71 static void
     72 png_write_chunk_header(png_structrp png_ptr, png_uint_32 chunk_name,
     73     png_uint_32 length)
     74 {
     75    png_byte buf[8];
     76 
     77 #if defined(PNG_DEBUG) && (PNG_DEBUG > 0)
     78    PNG_CSTRING_FROM_CHUNK(buf, chunk_name);
     79    png_debug2(0, "Writing %s chunk, length = %lu", buf, (unsigned long)length);
     80 #endif
     81 
     82    if (png_ptr == NULL)
     83       return;
     84 
     85 #ifdef PNG_IO_STATE_SUPPORTED
     86    /* Inform the I/O callback that the chunk header is being written.
     87     * PNG_IO_CHUNK_HDR requires a single I/O call.
     88     */
     89    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_HDR;
     90 #endif
     91 
     92    /* Write the length and the chunk name */
     93    png_save_uint_32(buf, length);
     94    png_save_uint_32(buf + 4, chunk_name);
     95    png_write_data(png_ptr, buf, 8);
     96 
     97    /* Put the chunk name into png_ptr->chunk_name */
     98    png_ptr->chunk_name = chunk_name;
     99 
    100    /* Reset the crc and run it over the chunk name */
    101    png_reset_crc(png_ptr);
    102 
    103    png_calculate_crc(png_ptr, buf + 4, 4);
    104 
    105 #ifdef PNG_IO_STATE_SUPPORTED
    106    /* Inform the I/O callback that chunk data will (possibly) be written.
    107     * PNG_IO_CHUNK_DATA does NOT require a specific number of I/O calls.
    108     */
    109    png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_DATA;
    110 #endif
    111 }
    112 
    113 void PNGAPI
    114 png_write_chunk_start(png_structrp png_ptr, png_const_bytep chunk_string,
    115     png_uint_32 length)
    116 {
    117    png_write_chunk_header(png_ptr, PNG_CHUNK_FROM_STRING(chunk_string), length);
    118 }
    119 
    120 /* Write the data of a PNG chunk started with png_write_chunk_header().
    121  * Note that multiple calls to this function are allowed, and that the
    122  * sum of the lengths from these calls *must* add up to the total_length
    123  * given to png_write_chunk_header().
    124  */
    125 void PNGAPI
    126 png_write_chunk_data(png_structrp png_ptr, png_const_bytep data,
    127     png_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 that 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, (png_size_t)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, png_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 maxima");
    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, png_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)
    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 #else
    289 #  define optimize_cmf(dp,dl) ((void)0)
    290 #endif /* PNG_WRITE_OPTIMIZE_CMF_SUPPORTED */
    291 
    292 /* Initialize the compressor for the appropriate type of compression. */
    293 static int
    294 png_deflate_claim(png_structrp png_ptr, png_uint_32 owner,
    295    png_alloc_size_t data_size)
    296 {
    297    if (png_ptr->zowner != 0)
    298    {
    299       char msg[64];
    300 
    301       PNG_STRING_FROM_CHUNK(msg, owner);
    302       msg[4] = ':';
    303       msg[5] = ' ';
    304       PNG_STRING_FROM_CHUNK(msg+6, png_ptr->zowner);
    305       /* So the message that results is "<chunk> using zstream"; this is an
    306        * internal error, but is very useful for debugging.  i18n requirements
    307        * are minimal.
    308        */
    309       (void)png_safecat(msg, (sizeof msg), 10, " using zstream");
    310 #     if PNG_LIBPNG_BUILD_BASE_TYPE >= PNG_LIBPNG_BUILD_RC
    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)
    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) &&
    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        * to a simple reset to the previous parameters.
    411        */
    412       if (png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED)
    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          /* Fix up the deflate header, if required */
    620          optimize_cmf(comp->output, comp->input_len);
    621 
    622          /* But Z_OK is returned, not Z_STREAM_END; this allows the claim
    623           * function above to return Z_STREAM_END on an error (though it never
    624           * does in the current versions of zlib.)
    625           */
    626          return Z_OK;
    627       }
    628 
    629       else
    630          return ret;
    631    }
    632 }
    633 
    634 /* Ship the compressed text out via chunk writes */
    635 static void
    636 png_write_compressed_data_out(png_structrp png_ptr, compression_state *comp)
    637 {
    638    png_uint_32 output_len = comp->output_len;
    639    png_const_bytep output = comp->output;
    640    png_uint_32 avail = (sizeof comp->output);
    641    png_compression_buffer *next = png_ptr->zbuffer_list;
    642 
    643    for (;;)
    644    {
    645       if (avail > output_len)
    646          avail = output_len;
    647 
    648       png_write_chunk_data(png_ptr, output, avail);
    649 
    650       output_len -= avail;
    651 
    652       if (output_len == 0 || next == NULL)
    653          break;
    654 
    655       avail = png_ptr->zbuffer_size;
    656       output = next->output;
    657       next = next->next;
    658    }
    659 
    660    /* This is an internal error; 'next' must have been NULL! */
    661    if (output_len > 0)
    662       png_error(png_ptr, "error writing ancillary chunked compressed data");
    663 }
    664 #endif /* PNG_WRITE_COMPRESSED_TEXT_SUPPORTED */
    665 
    666 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_pCAL_SUPPORTED) || \
    667     defined(PNG_WRITE_iCCP_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
    668 /* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
    669  * and if invalid, correct the keyword rather than discarding the entire
    670  * chunk.  The PNG 1.0 specification requires keywords 1-79 characters in
    671  * length, forbids leading or trailing whitespace, multiple internal spaces,
    672  * and the non-break space (0x80) from ISO 8859-1.  Returns keyword length.
    673  *
    674  * The 'new_key' buffer must be 80 characters in size (for the keyword plus a
    675  * trailing '\0').  If this routine returns 0 then there was no keyword, or a
    676  * valid one could not be generated, and the caller must png_error.
    677  */
    678 static png_uint_32
    679 png_check_keyword(png_structrp png_ptr, png_const_charp key, png_bytep new_key)
    680 {
    681    png_const_charp orig_key = key;
    682    png_uint_32 key_len = 0;
    683    int bad_character = 0;
    684    int space = 1;
    685 
    686    png_debug(1, "in png_check_keyword");
    687 
    688    if (key == NULL)
    689    {
    690       *new_key = 0;
    691       return 0;
    692    }
    693 
    694    while (*key && key_len < 79)
    695    {
    696       png_byte ch = (png_byte)(0xff & *key++);
    697 
    698       if ((ch > 32 && ch <= 126) || (ch >= 161 /*&& ch <= 255*/))
    699          *new_key++ = ch, ++key_len, space = 0;
    700 
    701       else if (!space)
    702       {
    703          /* A space or an invalid character when one wasn't seen immediately
    704           * before; output just a space.
    705           */
    706          *new_key++ = 32, ++key_len, space = 1;
    707 
    708          /* If the character was not a space then it is invalid. */
    709          if (ch != 32)
    710             bad_character = ch;
    711       }
    712 
    713       else if (!bad_character)
    714          bad_character = ch; /* just skip it, record the first error */
    715    }
    716 
    717    if (key_len > 0 && space) /* trailing space */
    718    {
    719       --key_len, --new_key;
    720       if (!bad_character)
    721          bad_character = 32;
    722    }
    723 
    724    /* Terminate the keyword */
    725    *new_key = 0;
    726 
    727    if (key_len == 0)
    728       return 0;
    729 
    730    /* Try to only output one warning per keyword: */
    731    if (*key) /* keyword too long */
    732       png_warning(png_ptr, "keyword truncated");
    733 
    734    else if (bad_character)
    735    {
    736       PNG_WARNING_PARAMETERS(p)
    737 
    738       png_warning_parameter(p, 1, orig_key);
    739       png_warning_parameter_signed(p, 2, PNG_NUMBER_FORMAT_02x, bad_character);
    740 
    741       png_formatted_warning(png_ptr, p, "keyword \"@1\": bad character '0x@2'");
    742    }
    743 
    744    return key_len;
    745 }
    746 #endif
    747 
    748 /* Write the IHDR chunk, and update the png_struct with the necessary
    749  * information.  Note that the rest of this code depends upon this
    750  * information being correct.
    751  */
    752 void /* PRIVATE */
    753 png_write_IHDR(png_structrp png_ptr, png_uint_32 width, png_uint_32 height,
    754     int bit_depth, int color_type, int compression_type, int filter_type,
    755     int interlace_type)
    756 {
    757    png_byte buf[13]; /* Buffer to store the IHDR info */
    758 
    759    png_debug(1, "in png_write_IHDR");
    760 
    761    /* Check that we have valid input data from the application info */
    762    switch (color_type)
    763    {
    764       case PNG_COLOR_TYPE_GRAY:
    765          switch (bit_depth)
    766          {
    767             case 1:
    768             case 2:
    769             case 4:
    770             case 8:
    771 #ifdef PNG_WRITE_16BIT_SUPPORTED
    772             case 16:
    773 #endif
    774                png_ptr->channels = 1; break;
    775 
    776             default:
    777                png_error(png_ptr,
    778                    "Invalid bit depth for grayscale image");
    779          }
    780          break;
    781 
    782       case PNG_COLOR_TYPE_RGB:
    783 #ifdef PNG_WRITE_16BIT_SUPPORTED
    784          if (bit_depth != 8 && bit_depth != 16)
    785 #else
    786          if (bit_depth != 8)
    787 #endif
    788             png_error(png_ptr, "Invalid bit depth for RGB image");
    789 
    790          png_ptr->channels = 3;
    791          break;
    792 
    793       case PNG_COLOR_TYPE_PALETTE:
    794          switch (bit_depth)
    795          {
    796             case 1:
    797             case 2:
    798             case 4:
    799             case 8:
    800                png_ptr->channels = 1;
    801                break;
    802 
    803             default:
    804                png_error(png_ptr, "Invalid bit depth for paletted image");
    805          }
    806          break;
    807 
    808       case PNG_COLOR_TYPE_GRAY_ALPHA:
    809          if (bit_depth != 8 && bit_depth != 16)
    810             png_error(png_ptr, "Invalid bit depth for grayscale+alpha image");
    811 
    812          png_ptr->channels = 2;
    813          break;
    814 
    815       case PNG_COLOR_TYPE_RGB_ALPHA:
    816 #ifdef PNG_WRITE_16BIT_SUPPORTED
    817          if (bit_depth != 8 && bit_depth != 16)
    818 #else
    819          if (bit_depth != 8)
    820 #endif
    821             png_error(png_ptr, "Invalid bit depth for RGBA image");
    822 
    823          png_ptr->channels = 4;
    824          break;
    825 
    826       default:
    827          png_error(png_ptr, "Invalid image color type specified");
    828    }
    829 
    830    if (compression_type != PNG_COMPRESSION_TYPE_BASE)
    831    {
    832       png_warning(png_ptr, "Invalid compression type specified");
    833       compression_type = PNG_COMPRESSION_TYPE_BASE;
    834    }
    835 
    836    /* Write filter_method 64 (intrapixel differencing) only if
    837     * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
    838     * 2. Libpng did not write a PNG signature (this filter_method is only
    839     *    used in PNG datastreams that are embedded in MNG datastreams) and
    840     * 3. The application called png_permit_mng_features with a mask that
    841     *    included PNG_FLAG_MNG_FILTER_64 and
    842     * 4. The filter_method is 64 and
    843     * 5. The color_type is RGB or RGBA
    844     */
    845    if (
    846 #ifdef PNG_MNG_FEATURES_SUPPORTED
    847        !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) &&
    848        ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE) == 0) &&
    849        (color_type == PNG_COLOR_TYPE_RGB ||
    850         color_type == PNG_COLOR_TYPE_RGB_ALPHA) &&
    851        (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) &&
    852 #endif
    853        filter_type != PNG_FILTER_TYPE_BASE)
    854    {
    855       png_warning(png_ptr, "Invalid filter type specified");
    856       filter_type = PNG_FILTER_TYPE_BASE;
    857    }
    858 
    859 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
    860    if (interlace_type != PNG_INTERLACE_NONE &&
    861        interlace_type != PNG_INTERLACE_ADAM7)
    862    {
    863       png_warning(png_ptr, "Invalid interlace type specified");
    864       interlace_type = PNG_INTERLACE_ADAM7;
    865    }
    866 #else
    867    interlace_type=PNG_INTERLACE_NONE;
    868 #endif
    869 
    870    /* Save the relevent information */
    871    png_ptr->bit_depth = (png_byte)bit_depth;
    872    png_ptr->color_type = (png_byte)color_type;
    873    png_ptr->interlaced = (png_byte)interlace_type;
    874 #ifdef PNG_MNG_FEATURES_SUPPORTED
    875    png_ptr->filter_type = (png_byte)filter_type;
    876 #endif
    877    png_ptr->compression_type = (png_byte)compression_type;
    878    png_ptr->width = width;
    879    png_ptr->height = height;
    880 
    881    png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels);
    882    png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width);
    883    /* Set the usr info, so any transformations can modify it */
    884    png_ptr->usr_width = png_ptr->width;
    885    png_ptr->usr_bit_depth = png_ptr->bit_depth;
    886    png_ptr->usr_channels = png_ptr->channels;
    887 
    888    /* Pack the header information into the buffer */
    889    png_save_uint_32(buf, width);
    890    png_save_uint_32(buf + 4, height);
    891    buf[8] = (png_byte)bit_depth;
    892    buf[9] = (png_byte)color_type;
    893    buf[10] = (png_byte)compression_type;
    894    buf[11] = (png_byte)filter_type;
    895    buf[12] = (png_byte)interlace_type;
    896 
    897    /* Write the chunk */
    898    png_write_complete_chunk(png_ptr, png_IHDR, buf, (png_size_t)13);
    899 
    900    if (!(png_ptr->do_filter))
    901    {
    902       if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE ||
    903           png_ptr->bit_depth < 8)
    904          png_ptr->do_filter = PNG_FILTER_NONE;
    905 
    906       else
    907          png_ptr->do_filter = PNG_ALL_FILTERS;
    908    }
    909 
    910    png_ptr->mode = PNG_HAVE_IHDR; /* not READY_FOR_ZTXT */
    911 }
    912 
    913 /* Write the palette.  We are careful not to trust png_color to be in the
    914  * correct order for PNG, so people can redefine it to any convenient
    915  * structure.
    916  */
    917 void /* PRIVATE */
    918 png_write_PLTE(png_structrp png_ptr, png_const_colorp palette,
    919     png_uint_32 num_pal)
    920 {
    921    png_uint_32 i;
    922    png_const_colorp pal_ptr;
    923    png_byte buf[3];
    924 
    925    png_debug(1, "in png_write_PLTE");
    926 
    927    if ((
    928 #ifdef PNG_MNG_FEATURES_SUPPORTED
    929        !(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) &&
    930 #endif
    931        num_pal == 0) || num_pal > 256)
    932    {
    933       if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
    934       {
    935          png_error(png_ptr, "Invalid number of colors in palette");
    936       }
    937 
    938       else
    939       {
    940          png_warning(png_ptr, "Invalid number of colors in palette");
    941          return;
    942       }
    943    }
    944 
    945    if (!(png_ptr->color_type&PNG_COLOR_MASK_COLOR))
    946    {
    947       png_warning(png_ptr,
    948           "Ignoring request to write a PLTE chunk in grayscale PNG");
    949 
    950       return;
    951    }
    952 
    953    png_ptr->num_palette = (png_uint_16)num_pal;
    954    png_debug1(3, "num_palette = %d", png_ptr->num_palette);
    955 
    956    png_write_chunk_header(png_ptr, png_PLTE, (png_uint_32)(num_pal * 3));
    957 #ifdef PNG_POINTER_INDEXING_SUPPORTED
    958 
    959    for (i = 0, pal_ptr = palette; i < num_pal; i++, pal_ptr++)
    960    {
    961       buf[0] = pal_ptr->red;
    962       buf[1] = pal_ptr->green;
    963       buf[2] = pal_ptr->blue;
    964       png_write_chunk_data(png_ptr, buf, (png_size_t)3);
    965    }
    966 
    967 #else
    968    /* This is a little slower but some buggy compilers need to do this
    969     * instead
    970     */
    971    pal_ptr=palette;
    972 
    973    for (i = 0; i < num_pal; i++)
    974    {
    975       buf[0] = pal_ptr[i].red;
    976       buf[1] = pal_ptr[i].green;
    977       buf[2] = pal_ptr[i].blue;
    978       png_write_chunk_data(png_ptr, buf, (png_size_t)3);
    979    }
    980 
    981 #endif
    982    png_write_chunk_end(png_ptr);
    983    png_ptr->mode |= PNG_HAVE_PLTE;
    984 }
    985 
    986 /* This is similar to png_text_compress, above, except that it does not require
    987  * all of the data at once and, instead of buffering the compressed result,
    988  * writes it as IDAT chunks.  Unlike png_text_compress it *can* png_error out
    989  * because it calls the write interface.  As a result it does its own error
    990  * reporting and does not return an error code.  In the event of error it will
    991  * just call png_error.  The input data length may exceed 32-bits.  The 'flush'
    992  * parameter is exactly the same as that to deflate, with the following
    993  * meanings:
    994  *
    995  * Z_NO_FLUSH: normal incremental output of compressed data
    996  * Z_SYNC_FLUSH: do a SYNC_FLUSH, used by png_write_flush
    997  * Z_FINISH: this is the end of the input, do a Z_FINISH and clean up
    998  *
    999  * The routine manages the acquire and release of the png_ptr->zstream by
   1000  * checking and (at the end) clearing png_ptr->zowner, it does some sanity
   1001  * checks on the 'mode' flags while doing this.
   1002  */
   1003 void /* PRIVATE */
   1004 png_compress_IDAT(png_structrp png_ptr, png_const_bytep input,
   1005    png_alloc_size_t input_len, int flush)
   1006 {
   1007    if (png_ptr->zowner != png_IDAT)
   1008    {
   1009       /* First time.   Ensure we have a temporary buffer for compression and
   1010        * trim the buffer list if it has more than one entry to free memory.
   1011        * If 'WRITE_COMPRESSED_TEXT' is not set the list will never have been
   1012        * created at this point, but the check here is quick and safe.
   1013        */
   1014       if (png_ptr->zbuffer_list == NULL)
   1015       {
   1016          png_ptr->zbuffer_list = png_voidcast(png_compression_bufferp,
   1017             png_malloc(png_ptr, PNG_COMPRESSION_BUFFER_SIZE(png_ptr)));
   1018          png_ptr->zbuffer_list->next = NULL;
   1019       }
   1020 
   1021       else
   1022          png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list->next);
   1023 
   1024       /* It is a terminal error if we can't claim the zstream. */
   1025       if (png_deflate_claim(png_ptr, png_IDAT, png_image_size(png_ptr)) != Z_OK)
   1026          png_error(png_ptr, png_ptr->zstream.msg);
   1027 
   1028       /* The output state is maintained in png_ptr->zstream, so it must be
   1029        * initialized here after the claim.
   1030        */
   1031       png_ptr->zstream.next_out = png_ptr->zbuffer_list->output;
   1032       png_ptr->zstream.avail_out = png_ptr->zbuffer_size;
   1033    }
   1034 
   1035    /* Now loop reading and writing until all the input is consumed or an error
   1036     * terminates the operation.  The _out values are maintained across calls to
   1037     * this function, but the input must be reset each time.
   1038     */
   1039    png_ptr->zstream.next_in = PNGZ_INPUT_CAST(input);
   1040    png_ptr->zstream.avail_in = 0; /* set below */
   1041    for (;;)
   1042    {
   1043       int ret;
   1044 
   1045       /* INPUT: from the row data */
   1046       uInt avail = ZLIB_IO_MAX;
   1047 
   1048       if (avail > input_len)
   1049          avail = (uInt)input_len; /* safe because of the check */
   1050 
   1051       png_ptr->zstream.avail_in = avail;
   1052       input_len -= avail;
   1053 
   1054       ret = deflate(&png_ptr->zstream, input_len > 0 ? Z_NO_FLUSH : flush);
   1055 
   1056       /* Include as-yet unconsumed input */
   1057       input_len += png_ptr->zstream.avail_in;
   1058       png_ptr->zstream.avail_in = 0;
   1059 
   1060       /* OUTPUT: write complete IDAT chunks when avail_out drops to zero, note
   1061        * that these two zstream fields are preserved across the calls, therefore
   1062        * there is no need to set these up on entry to the loop.
   1063        */
   1064       if (png_ptr->zstream.avail_out == 0)
   1065       {
   1066          png_bytep data = png_ptr->zbuffer_list->output;
   1067          uInt size = png_ptr->zbuffer_size;
   1068 
   1069          /* Write an IDAT containing the data then reset the buffer.  The
   1070           * first IDAT may need deflate header optimization.
   1071           */
   1072 #        ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
   1073             if (!(png_ptr->mode & PNG_HAVE_IDAT) &&
   1074                png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
   1075                optimize_cmf(data, png_image_size(png_ptr));
   1076 #        endif
   1077 
   1078          png_write_complete_chunk(png_ptr, png_IDAT, data, size);
   1079          png_ptr->mode |= PNG_HAVE_IDAT;
   1080 
   1081          png_ptr->zstream.next_out = data;
   1082          png_ptr->zstream.avail_out = size;
   1083 
   1084          /* For SYNC_FLUSH or FINISH it is essential to keep calling zlib with
   1085           * the same flush parameter until it has finished output, for NO_FLUSH
   1086           * it doesn't matter.
   1087           */
   1088          if (ret == Z_OK && flush != Z_NO_FLUSH)
   1089             continue;
   1090       }
   1091 
   1092       /* The order of these checks doesn't matter much; it just effect which
   1093        * possible error might be detected if multiple things go wrong at once.
   1094        */
   1095       if (ret == Z_OK) /* most likely return code! */
   1096       {
   1097          /* If all the input has been consumed then just return.  If Z_FINISH
   1098           * was used as the flush parameter something has gone wrong if we get
   1099           * here.
   1100           */
   1101          if (input_len == 0)
   1102          {
   1103             if (flush == Z_FINISH)
   1104                png_error(png_ptr, "Z_OK on Z_FINISH with output space");
   1105 
   1106             return;
   1107          }
   1108       }
   1109 
   1110       else if (ret == Z_STREAM_END && flush == Z_FINISH)
   1111       {
   1112          /* This is the end of the IDAT data; any pending output must be
   1113           * flushed.  For small PNG files we may still be at the beginning.
   1114           */
   1115          png_bytep data = png_ptr->zbuffer_list->output;
   1116          uInt size = png_ptr->zbuffer_size - png_ptr->zstream.avail_out;
   1117 
   1118 #        ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
   1119             if (!(png_ptr->mode & PNG_HAVE_IDAT) &&
   1120                png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
   1121                optimize_cmf(data, png_image_size(png_ptr));
   1122 #        endif
   1123 
   1124          png_write_complete_chunk(png_ptr, png_IDAT, data, size);
   1125          png_ptr->zstream.avail_out = 0;
   1126          png_ptr->zstream.next_out = NULL;
   1127          png_ptr->mode |= PNG_HAVE_IDAT | PNG_AFTER_IDAT;
   1128 
   1129          png_ptr->zowner = 0; /* Release the stream */
   1130          return;
   1131       }
   1132 
   1133       else
   1134       {
   1135          /* This is an error condition. */
   1136          png_zstream_error(png_ptr, ret);
   1137          png_error(png_ptr, png_ptr->zstream.msg);
   1138       }
   1139    }
   1140 }
   1141 
   1142 /* Write an IEND chunk */
   1143 void /* PRIVATE */
   1144 png_write_IEND(png_structrp png_ptr)
   1145 {
   1146    png_debug(1, "in png_write_IEND");
   1147 
   1148    png_write_complete_chunk(png_ptr, png_IEND, NULL, (png_size_t)0);
   1149    png_ptr->mode |= PNG_HAVE_IEND;
   1150 }
   1151 
   1152 #ifdef PNG_WRITE_gAMA_SUPPORTED
   1153 /* Write a gAMA chunk */
   1154 void /* PRIVATE */
   1155 png_write_gAMA_fixed(png_structrp png_ptr, png_fixed_point file_gamma)
   1156 {
   1157    png_byte buf[4];
   1158 
   1159    png_debug(1, "in png_write_gAMA");
   1160 
   1161    /* file_gamma is saved in 1/100,000ths */
   1162    png_save_uint_32(buf, (png_uint_32)file_gamma);
   1163    png_write_complete_chunk(png_ptr, png_gAMA, buf, (png_size_t)4);
   1164 }
   1165 #endif
   1166 
   1167 #ifdef PNG_WRITE_sRGB_SUPPORTED
   1168 /* Write a sRGB chunk */
   1169 void /* PRIVATE */
   1170 png_write_sRGB(png_structrp png_ptr, int srgb_intent)
   1171 {
   1172    png_byte buf[1];
   1173 
   1174    png_debug(1, "in png_write_sRGB");
   1175 
   1176    if (srgb_intent >= PNG_sRGB_INTENT_LAST)
   1177       png_warning(png_ptr,
   1178           "Invalid sRGB rendering intent specified");
   1179 
   1180    buf[0]=(png_byte)srgb_intent;
   1181    png_write_complete_chunk(png_ptr, png_sRGB, buf, (png_size_t)1);
   1182 }
   1183 #endif
   1184 
   1185 #ifdef PNG_WRITE_iCCP_SUPPORTED
   1186 /* Write an iCCP chunk */
   1187 void /* PRIVATE */
   1188 png_write_iCCP(png_structrp png_ptr, png_const_charp name,
   1189     png_const_bytep profile)
   1190 {
   1191    png_uint_32 name_len;
   1192    png_uint_32 profile_len;
   1193    png_byte new_name[81]; /* 1 byte for the compression byte */
   1194    compression_state comp;
   1195 
   1196    png_debug(1, "in png_write_iCCP");
   1197 
   1198    /* These are all internal problems: the profile should have been checked
   1199     * before when it was stored.
   1200     */
   1201    if (profile == NULL)
   1202       png_error(png_ptr, "No profile for iCCP chunk"); /* internal error */
   1203 
   1204    profile_len = png_get_uint_32(profile);
   1205 
   1206    if (profile_len < 132)
   1207       png_error(png_ptr, "ICC profile too short");
   1208 
   1209    if (profile_len & 0x03)
   1210       png_error(png_ptr, "ICC profile length invalid (not a multiple of 4)");
   1211 
   1212    {
   1213       png_uint_32 embedded_profile_len = png_get_uint_32(profile);
   1214 
   1215       if (profile_len != embedded_profile_len)
   1216          png_error(png_ptr, "Profile length does not match profile");
   1217    }
   1218 
   1219    name_len = png_check_keyword(png_ptr, name, new_name);
   1220 
   1221    if (name_len == 0)
   1222       png_error(png_ptr, "iCCP: invalid keyword");
   1223 
   1224    new_name[++name_len] = PNG_COMPRESSION_TYPE_BASE;
   1225 
   1226    /* Make sure we include the NULL after the name and the compression type */
   1227    ++name_len;
   1228 
   1229    png_text_compress_init(&comp, profile, profile_len);
   1230 
   1231    /* Allow for keyword terminator and compression byte */
   1232    if (png_text_compress(png_ptr, png_iCCP, &comp, name_len) != Z_OK)
   1233       png_error(png_ptr, png_ptr->zstream.msg);
   1234 
   1235    png_write_chunk_header(png_ptr, png_iCCP, name_len + comp.output_len);
   1236 
   1237    png_write_chunk_data(png_ptr, new_name, name_len);
   1238 
   1239    png_write_compressed_data_out(png_ptr, &comp);
   1240 
   1241    png_write_chunk_end(png_ptr);
   1242 }
   1243 #endif
   1244 
   1245 #ifdef PNG_WRITE_sPLT_SUPPORTED
   1246 /* Write a sPLT chunk */
   1247 void /* PRIVATE */
   1248 png_write_sPLT(png_structrp png_ptr, png_const_sPLT_tp spalette)
   1249 {
   1250    png_uint_32 name_len;
   1251    png_byte new_name[80];
   1252    png_byte entrybuf[10];
   1253    png_size_t entry_size = (spalette->depth == 8 ? 6 : 10);
   1254    png_size_t palette_size = entry_size * spalette->nentries;
   1255    png_sPLT_entryp ep;
   1256 #ifndef PNG_POINTER_INDEXING_SUPPORTED
   1257    int i;
   1258 #endif
   1259 
   1260    png_debug(1, "in png_write_sPLT");
   1261 
   1262    name_len = png_check_keyword(png_ptr, spalette->name, new_name);
   1263 
   1264    if (name_len == 0)
   1265       png_error(png_ptr, "sPLT: invalid keyword");
   1266 
   1267    /* Make sure we include the NULL after the name */
   1268    png_write_chunk_header(png_ptr, png_sPLT,
   1269        (png_uint_32)(name_len + 2 + palette_size));
   1270 
   1271    png_write_chunk_data(png_ptr, (png_bytep)new_name,
   1272        (png_size_t)(name_len + 1));
   1273 
   1274    png_write_chunk_data(png_ptr, &spalette->depth, (png_size_t)1);
   1275 
   1276    /* Loop through each palette entry, writing appropriately */
   1277 #ifdef PNG_POINTER_INDEXING_SUPPORTED
   1278    for (ep = spalette->entries; ep<spalette->entries + spalette->nentries; ep++)
   1279    {
   1280       if (spalette->depth == 8)
   1281       {
   1282          entrybuf[0] = (png_byte)ep->red;
   1283          entrybuf[1] = (png_byte)ep->green;
   1284          entrybuf[2] = (png_byte)ep->blue;
   1285          entrybuf[3] = (png_byte)ep->alpha;
   1286          png_save_uint_16(entrybuf + 4, ep->frequency);
   1287       }
   1288 
   1289       else
   1290       {
   1291          png_save_uint_16(entrybuf + 0, ep->red);
   1292          png_save_uint_16(entrybuf + 2, ep->green);
   1293          png_save_uint_16(entrybuf + 4, ep->blue);
   1294          png_save_uint_16(entrybuf + 6, ep->alpha);
   1295          png_save_uint_16(entrybuf + 8, ep->frequency);
   1296       }
   1297 
   1298       png_write_chunk_data(png_ptr, entrybuf, entry_size);
   1299    }
   1300 #else
   1301    ep=spalette->entries;
   1302    for (i = 0; i>spalette->nentries; i++)
   1303    {
   1304       if (spalette->depth == 8)
   1305       {
   1306          entrybuf[0] = (png_byte)ep[i].red;
   1307          entrybuf[1] = (png_byte)ep[i].green;
   1308          entrybuf[2] = (png_byte)ep[i].blue;
   1309          entrybuf[3] = (png_byte)ep[i].alpha;
   1310          png_save_uint_16(entrybuf + 4, ep[i].frequency);
   1311       }
   1312 
   1313       else
   1314       {
   1315          png_save_uint_16(entrybuf + 0, ep[i].red);
   1316          png_save_uint_16(entrybuf + 2, ep[i].green);
   1317          png_save_uint_16(entrybuf + 4, ep[i].blue);
   1318          png_save_uint_16(entrybuf + 6, ep[i].alpha);
   1319          png_save_uint_16(entrybuf + 8, ep[i].frequency);
   1320       }
   1321 
   1322       png_write_chunk_data(png_ptr, entrybuf, entry_size);
   1323    }
   1324 #endif
   1325 
   1326    png_write_chunk_end(png_ptr);
   1327 }
   1328 #endif
   1329 
   1330 #ifdef PNG_WRITE_sBIT_SUPPORTED
   1331 /* Write the sBIT chunk */
   1332 void /* PRIVATE */
   1333 png_write_sBIT(png_structrp png_ptr, png_const_color_8p sbit, int color_type)
   1334 {
   1335    png_byte buf[4];
   1336    png_size_t size;
   1337 
   1338    png_debug(1, "in png_write_sBIT");
   1339 
   1340    /* Make sure we don't depend upon the order of PNG_COLOR_8 */
   1341    if (color_type & PNG_COLOR_MASK_COLOR)
   1342    {
   1343       png_byte maxbits;
   1344 
   1345       maxbits = (png_byte)(color_type==PNG_COLOR_TYPE_PALETTE ? 8 :
   1346           png_ptr->usr_bit_depth);
   1347 
   1348       if (sbit->red == 0 || sbit->red > maxbits ||
   1349           sbit->green == 0 || sbit->green > maxbits ||
   1350           sbit->blue == 0 || sbit->blue > maxbits)
   1351       {
   1352          png_warning(png_ptr, "Invalid sBIT depth specified");
   1353          return;
   1354       }
   1355 
   1356       buf[0] = sbit->red;
   1357       buf[1] = sbit->green;
   1358       buf[2] = sbit->blue;
   1359       size = 3;
   1360    }
   1361 
   1362    else
   1363    {
   1364       if (sbit->gray == 0 || sbit->gray > png_ptr->usr_bit_depth)
   1365       {
   1366          png_warning(png_ptr, "Invalid sBIT depth specified");
   1367          return;
   1368       }
   1369 
   1370       buf[0] = sbit->gray;
   1371       size = 1;
   1372    }
   1373 
   1374    if (color_type & PNG_COLOR_MASK_ALPHA)
   1375    {
   1376       if (sbit->alpha == 0 || sbit->alpha > png_ptr->usr_bit_depth)
   1377       {
   1378          png_warning(png_ptr, "Invalid sBIT depth specified");
   1379          return;
   1380       }
   1381 
   1382       buf[size++] = sbit->alpha;
   1383    }
   1384 
   1385    png_write_complete_chunk(png_ptr, png_sBIT, buf, size);
   1386 }
   1387 #endif
   1388 
   1389 #ifdef PNG_WRITE_cHRM_SUPPORTED
   1390 /* Write the cHRM chunk */
   1391 void /* PRIVATE */
   1392 png_write_cHRM_fixed(png_structrp png_ptr, const png_xy *xy)
   1393 {
   1394    png_byte buf[32];
   1395 
   1396    png_debug(1, "in png_write_cHRM");
   1397 
   1398    /* Each value is saved in 1/100,000ths */
   1399    png_save_int_32(buf,      xy->whitex);
   1400    png_save_int_32(buf +  4, xy->whitey);
   1401 
   1402    png_save_int_32(buf +  8, xy->redx);
   1403    png_save_int_32(buf + 12, xy->redy);
   1404 
   1405    png_save_int_32(buf + 16, xy->greenx);
   1406    png_save_int_32(buf + 20, xy->greeny);
   1407 
   1408    png_save_int_32(buf + 24, xy->bluex);
   1409    png_save_int_32(buf + 28, xy->bluey);
   1410 
   1411    png_write_complete_chunk(png_ptr, png_cHRM, buf, 32);
   1412 }
   1413 #endif
   1414 
   1415 #ifdef PNG_WRITE_tRNS_SUPPORTED
   1416 /* Write the tRNS chunk */
   1417 void /* PRIVATE */
   1418 png_write_tRNS(png_structrp png_ptr, png_const_bytep trans_alpha,
   1419     png_const_color_16p tran, int num_trans, int color_type)
   1420 {
   1421    png_byte buf[6];
   1422 
   1423    png_debug(1, "in png_write_tRNS");
   1424 
   1425    if (color_type == PNG_COLOR_TYPE_PALETTE)
   1426    {
   1427       if (num_trans <= 0 || num_trans > (int)png_ptr->num_palette)
   1428       {
   1429          png_app_warning(png_ptr,
   1430              "Invalid number of transparent colors specified");
   1431          return;
   1432       }
   1433 
   1434       /* Write the chunk out as it is */
   1435       png_write_complete_chunk(png_ptr, png_tRNS, trans_alpha,
   1436          (png_size_t)num_trans);
   1437    }
   1438 
   1439    else if (color_type == PNG_COLOR_TYPE_GRAY)
   1440    {
   1441       /* One 16 bit value */
   1442       if (tran->gray >= (1 << png_ptr->bit_depth))
   1443       {
   1444          png_app_warning(png_ptr,
   1445              "Ignoring attempt to write tRNS chunk out-of-range for bit_depth");
   1446 
   1447          return;
   1448       }
   1449 
   1450       png_save_uint_16(buf, tran->gray);
   1451       png_write_complete_chunk(png_ptr, png_tRNS, buf, (png_size_t)2);
   1452    }
   1453 
   1454    else if (color_type == PNG_COLOR_TYPE_RGB)
   1455    {
   1456       /* Three 16 bit values */
   1457       png_save_uint_16(buf, tran->red);
   1458       png_save_uint_16(buf + 2, tran->green);
   1459       png_save_uint_16(buf + 4, tran->blue);
   1460 #ifdef PNG_WRITE_16BIT_SUPPORTED
   1461       if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]))
   1462 #else
   1463       if (buf[0] | buf[2] | buf[4])
   1464 #endif
   1465       {
   1466          png_app_warning(png_ptr,
   1467            "Ignoring attempt to write 16-bit tRNS chunk when bit_depth is 8");
   1468          return;
   1469       }
   1470 
   1471       png_write_complete_chunk(png_ptr, png_tRNS, buf, (png_size_t)6);
   1472    }
   1473 
   1474    else
   1475    {
   1476       png_app_warning(png_ptr, "Can't write tRNS with an alpha channel");
   1477    }
   1478 }
   1479 #endif
   1480 
   1481 #ifdef PNG_WRITE_bKGD_SUPPORTED
   1482 /* Write the background chunk */
   1483 void /* PRIVATE */
   1484 png_write_bKGD(png_structrp png_ptr, png_const_color_16p back, int color_type)
   1485 {
   1486    png_byte buf[6];
   1487 
   1488    png_debug(1, "in png_write_bKGD");
   1489 
   1490    if (color_type == PNG_COLOR_TYPE_PALETTE)
   1491    {
   1492       if (
   1493 #ifdef PNG_MNG_FEATURES_SUPPORTED
   1494           (png_ptr->num_palette ||
   1495           (!(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE))) &&
   1496 #endif
   1497          back->index >= png_ptr->num_palette)
   1498       {
   1499          png_warning(png_ptr, "Invalid background palette index");
   1500          return;
   1501       }
   1502 
   1503       buf[0] = back->index;
   1504       png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)1);
   1505    }
   1506 
   1507    else if (color_type & PNG_COLOR_MASK_COLOR)
   1508    {
   1509       png_save_uint_16(buf, back->red);
   1510       png_save_uint_16(buf + 2, back->green);
   1511       png_save_uint_16(buf + 4, back->blue);
   1512 #ifdef PNG_WRITE_16BIT_SUPPORTED
   1513       if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]))
   1514 #else
   1515       if (buf[0] | buf[2] | buf[4])
   1516 #endif
   1517       {
   1518          png_warning(png_ptr,
   1519              "Ignoring attempt to write 16-bit bKGD chunk when bit_depth is 8");
   1520 
   1521          return;
   1522       }
   1523 
   1524       png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)6);
   1525    }
   1526 
   1527    else
   1528    {
   1529       if (back->gray >= (1 << png_ptr->bit_depth))
   1530       {
   1531          png_warning(png_ptr,
   1532              "Ignoring attempt to write bKGD chunk out-of-range for bit_depth");
   1533 
   1534          return;
   1535       }
   1536 
   1537       png_save_uint_16(buf, back->gray);
   1538       png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)2);
   1539    }
   1540 }
   1541 #endif
   1542 
   1543 #ifdef PNG_WRITE_hIST_SUPPORTED
   1544 /* Write the histogram */
   1545 void /* PRIVATE */
   1546 png_write_hIST(png_structrp png_ptr, png_const_uint_16p hist, int num_hist)
   1547 {
   1548    int i;
   1549    png_byte buf[3];
   1550 
   1551    png_debug(1, "in png_write_hIST");
   1552 
   1553    if (num_hist > (int)png_ptr->num_palette)
   1554    {
   1555       png_debug2(3, "num_hist = %d, num_palette = %d", num_hist,
   1556           png_ptr->num_palette);
   1557 
   1558       png_warning(png_ptr, "Invalid number of histogram entries specified");
   1559       return;
   1560    }
   1561 
   1562    png_write_chunk_header(png_ptr, png_hIST, (png_uint_32)(num_hist * 2));
   1563 
   1564    for (i = 0; i < num_hist; i++)
   1565    {
   1566       png_save_uint_16(buf, hist[i]);
   1567       png_write_chunk_data(png_ptr, buf, (png_size_t)2);
   1568    }
   1569 
   1570    png_write_chunk_end(png_ptr);
   1571 }
   1572 #endif
   1573 
   1574 #ifdef PNG_WRITE_tEXt_SUPPORTED
   1575 /* Write a tEXt chunk */
   1576 void /* PRIVATE */
   1577 png_write_tEXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
   1578     png_size_t text_len)
   1579 {
   1580    png_uint_32 key_len;
   1581    png_byte new_key[80];
   1582 
   1583    png_debug(1, "in png_write_tEXt");
   1584 
   1585    key_len = png_check_keyword(png_ptr, key, new_key);
   1586 
   1587    if (key_len == 0)
   1588       png_error(png_ptr, "tEXt: invalid keyword");
   1589 
   1590    if (text == NULL || *text == '\0')
   1591       text_len = 0;
   1592 
   1593    else
   1594       text_len = strlen(text);
   1595 
   1596    if (text_len > PNG_UINT_31_MAX - (key_len+1))
   1597       png_error(png_ptr, "tEXt: text too long");
   1598 
   1599    /* Make sure we include the 0 after the key */
   1600    png_write_chunk_header(png_ptr, png_tEXt,
   1601        (png_uint_32)/*checked above*/(key_len + text_len + 1));
   1602    /*
   1603     * We leave it to the application to meet PNG-1.0 requirements on the
   1604     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
   1605     * any non-Latin-1 characters except for NEWLINE.  ISO PNG will forbid them.
   1606     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
   1607     */
   1608    png_write_chunk_data(png_ptr, new_key, key_len + 1);
   1609 
   1610    if (text_len)
   1611       png_write_chunk_data(png_ptr, (png_const_bytep)text, text_len);
   1612 
   1613    png_write_chunk_end(png_ptr);
   1614 }
   1615 #endif
   1616 
   1617 #ifdef PNG_WRITE_zTXt_SUPPORTED
   1618 /* Write a compressed text chunk */
   1619 void /* PRIVATE */
   1620 png_write_zTXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
   1621     png_size_t text_len, int compression)
   1622 {
   1623    png_uint_32 key_len;
   1624    png_byte new_key[81];
   1625    compression_state comp;
   1626 
   1627    png_debug(1, "in png_write_zTXt");
   1628    PNG_UNUSED(text_len) /* Always use strlen */
   1629 
   1630    if (compression == PNG_TEXT_COMPRESSION_NONE)
   1631    {
   1632       png_write_tEXt(png_ptr, key, text, 0);
   1633       return;
   1634    }
   1635 
   1636    if (compression != PNG_TEXT_COMPRESSION_zTXt)
   1637       png_error(png_ptr, "zTXt: invalid compression type");
   1638 
   1639    key_len = png_check_keyword(png_ptr, key, new_key);
   1640 
   1641    if (key_len == 0)
   1642       png_error(png_ptr, "zTXt: invalid keyword");
   1643 
   1644    /* Add the compression method and 1 for the keyword separator. */
   1645    new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
   1646    ++key_len;
   1647 
   1648    /* Compute the compressed data; do it now for the length */
   1649    png_text_compress_init(&comp, (png_const_bytep)text,
   1650       text == NULL ? 0 : strlen(text));
   1651 
   1652    if (png_text_compress(png_ptr, png_zTXt, &comp, key_len) != Z_OK)
   1653       png_error(png_ptr, png_ptr->zstream.msg);
   1654 
   1655    /* Write start of chunk */
   1656    png_write_chunk_header(png_ptr, png_zTXt, key_len + comp.output_len);
   1657 
   1658    /* Write key */
   1659    png_write_chunk_data(png_ptr, new_key, key_len);
   1660 
   1661    /* Write the compressed data */
   1662    png_write_compressed_data_out(png_ptr, &comp);
   1663 
   1664    /* Close the chunk */
   1665    png_write_chunk_end(png_ptr);
   1666 }
   1667 #endif
   1668 
   1669 #ifdef PNG_WRITE_iTXt_SUPPORTED
   1670 /* Write an iTXt chunk */
   1671 void /* PRIVATE */
   1672 png_write_iTXt(png_structrp png_ptr, int compression, png_const_charp key,
   1673     png_const_charp lang, png_const_charp lang_key, png_const_charp text)
   1674 {
   1675    png_uint_32 key_len, prefix_len;
   1676    png_size_t lang_len, lang_key_len;
   1677    png_byte new_key[82];
   1678    compression_state comp;
   1679 
   1680    png_debug(1, "in png_write_iTXt");
   1681 
   1682    key_len = png_check_keyword(png_ptr, key, new_key);
   1683 
   1684    if (key_len == 0)
   1685       png_error(png_ptr, "iTXt: invalid keyword");
   1686 
   1687    /* Set the compression flag */
   1688    switch (compression)
   1689    {
   1690       case PNG_ITXT_COMPRESSION_NONE:
   1691       case PNG_TEXT_COMPRESSION_NONE:
   1692          compression = new_key[++key_len] = 0; /* no compression */
   1693          break;
   1694 
   1695       case PNG_TEXT_COMPRESSION_zTXt:
   1696       case PNG_ITXT_COMPRESSION_zTXt:
   1697          compression = new_key[++key_len] = 1; /* compressed */
   1698          break;
   1699 
   1700       default:
   1701          png_error(png_ptr, "iTXt: invalid compression");
   1702    }
   1703 
   1704    new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
   1705    ++key_len; /* for the keywod separator */
   1706 
   1707    /* We leave it to the application to meet PNG-1.0 requirements on the
   1708     * contents of the text.  PNG-1.0 through PNG-1.2 discourage the use of
   1709     * any non-Latin-1 characters except for NEWLINE.  ISO PNG, however,
   1710     * specifies that the text is UTF-8 and this really doesn't require any
   1711     * checking.
   1712     *
   1713     * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
   1714     *
   1715     * TODO: validate the language tag correctly (see the spec.)
   1716     */
   1717    if (lang == NULL) lang = ""; /* empty language is valid */
   1718    lang_len = strlen(lang)+1;
   1719    if (lang_key == NULL) lang_key = ""; /* may be empty */
   1720    lang_key_len = strlen(lang_key)+1;
   1721    if (text == NULL) text = ""; /* may be empty */
   1722 
   1723    prefix_len = key_len;
   1724    if (lang_len > PNG_UINT_31_MAX-prefix_len)
   1725       prefix_len = PNG_UINT_31_MAX;
   1726    else
   1727       prefix_len = (png_uint_32)(prefix_len + lang_len);
   1728 
   1729    if (lang_key_len > PNG_UINT_31_MAX-prefix_len)
   1730       prefix_len = PNG_UINT_31_MAX;
   1731    else
   1732       prefix_len = (png_uint_32)(prefix_len + lang_key_len);
   1733 
   1734    png_text_compress_init(&comp, (png_const_bytep)text, strlen(text));
   1735 
   1736    if (compression)
   1737    {
   1738       if (png_text_compress(png_ptr, png_iTXt, &comp, prefix_len) != Z_OK)
   1739          png_error(png_ptr, png_ptr->zstream.msg);
   1740    }
   1741 
   1742    else
   1743    {
   1744       if (comp.input_len > PNG_UINT_31_MAX-prefix_len)
   1745          png_error(png_ptr, "iTXt: uncompressed text too long");
   1746 
   1747       /* So the string will fit in a chunk: */
   1748       comp.output_len = (png_uint_32)/*SAFE*/comp.input_len;
   1749    }
   1750 
   1751    png_write_chunk_header(png_ptr, png_iTXt, comp.output_len + prefix_len);
   1752 
   1753    png_write_chunk_data(png_ptr, new_key, key_len);
   1754 
   1755    png_write_chunk_data(png_ptr, (png_const_bytep)lang, lang_len);
   1756 
   1757    png_write_chunk_data(png_ptr, (png_const_bytep)lang_key, lang_key_len);
   1758 
   1759    if (compression)
   1760       png_write_compressed_data_out(png_ptr, &comp);
   1761 
   1762    else
   1763       png_write_chunk_data(png_ptr, (png_const_bytep)text, comp.input_len);
   1764 
   1765    png_write_chunk_end(png_ptr);
   1766 }
   1767 #endif
   1768 
   1769 #ifdef PNG_WRITE_oFFs_SUPPORTED
   1770 /* Write the oFFs chunk */
   1771 void /* PRIVATE */
   1772 png_write_oFFs(png_structrp png_ptr, png_int_32 x_offset, png_int_32 y_offset,
   1773     int unit_type)
   1774 {
   1775    png_byte buf[9];
   1776 
   1777    png_debug(1, "in png_write_oFFs");
   1778 
   1779    if (unit_type >= PNG_OFFSET_LAST)
   1780       png_warning(png_ptr, "Unrecognized unit type for oFFs chunk");
   1781 
   1782    png_save_int_32(buf, x_offset);
   1783    png_save_int_32(buf + 4, y_offset);
   1784    buf[8] = (png_byte)unit_type;
   1785 
   1786    png_write_complete_chunk(png_ptr, png_oFFs, buf, (png_size_t)9);
   1787 }
   1788 #endif
   1789 #ifdef PNG_WRITE_pCAL_SUPPORTED
   1790 /* Write the pCAL chunk (described in the PNG extensions document) */
   1791 void /* PRIVATE */
   1792 png_write_pCAL(png_structrp png_ptr, png_charp purpose, png_int_32 X0,
   1793     png_int_32 X1, int type, int nparams, png_const_charp units,
   1794     png_charpp params)
   1795 {
   1796    png_uint_32 purpose_len;
   1797    png_size_t units_len, total_len;
   1798    png_size_tp params_len;
   1799    png_byte buf[10];
   1800    png_byte new_purpose[80];
   1801    int i;
   1802 
   1803    png_debug1(1, "in png_write_pCAL (%d parameters)", nparams);
   1804 
   1805    if (type >= PNG_EQUATION_LAST)
   1806       png_error(png_ptr, "Unrecognized equation type for pCAL chunk");
   1807 
   1808    purpose_len = png_check_keyword(png_ptr, purpose, new_purpose);
   1809 
   1810    if (purpose_len == 0)
   1811       png_error(png_ptr, "pCAL: invalid keyword");
   1812 
   1813    ++purpose_len; /* terminator */
   1814 
   1815    png_debug1(3, "pCAL purpose length = %d", (int)purpose_len);
   1816    units_len = strlen(units) + (nparams == 0 ? 0 : 1);
   1817    png_debug1(3, "pCAL units length = %d", (int)units_len);
   1818    total_len = purpose_len + units_len + 10;
   1819 
   1820    params_len = (png_size_tp)png_malloc(png_ptr,
   1821        (png_alloc_size_t)(nparams * (sizeof (png_size_t))));
   1822 
   1823    /* Find the length of each parameter, making sure we don't count the
   1824     * null terminator for the last parameter.
   1825     */
   1826    for (i = 0; i < nparams; i++)
   1827    {
   1828       params_len[i] = strlen(params[i]) + (i == nparams - 1 ? 0 : 1);
   1829       png_debug2(3, "pCAL parameter %d length = %lu", i,
   1830           (unsigned long)params_len[i]);
   1831       total_len += params_len[i];
   1832    }
   1833 
   1834    png_debug1(3, "pCAL total length = %d", (int)total_len);
   1835    png_write_chunk_header(png_ptr, png_pCAL, (png_uint_32)total_len);
   1836    png_write_chunk_data(png_ptr, new_purpose, purpose_len);
   1837    png_save_int_32(buf, X0);
   1838    png_save_int_32(buf + 4, X1);
   1839    buf[8] = (png_byte)type;
   1840    buf[9] = (png_byte)nparams;
   1841    png_write_chunk_data(png_ptr, buf, (png_size_t)10);
   1842    png_write_chunk_data(png_ptr, (png_const_bytep)units, (png_size_t)units_len);
   1843 
   1844    for (i = 0; i < nparams; i++)
   1845    {
   1846       png_write_chunk_data(png_ptr, (png_const_bytep)params[i], params_len[i]);
   1847    }
   1848 
   1849    png_free(png_ptr, params_len);
   1850    png_write_chunk_end(png_ptr);
   1851 }
   1852 #endif
   1853 
   1854 #ifdef PNG_WRITE_sCAL_SUPPORTED
   1855 /* Write the sCAL chunk */
   1856 void /* PRIVATE */
   1857 png_write_sCAL_s(png_structrp png_ptr, int unit, png_const_charp width,
   1858     png_const_charp height)
   1859 {
   1860    png_byte buf[64];
   1861    png_size_t wlen, hlen, total_len;
   1862 
   1863    png_debug(1, "in png_write_sCAL_s");
   1864 
   1865    wlen = strlen(width);
   1866    hlen = strlen(height);
   1867    total_len = wlen + hlen + 2;
   1868 
   1869    if (total_len > 64)
   1870    {
   1871       png_warning(png_ptr, "Can't write sCAL (buffer too small)");
   1872       return;
   1873    }
   1874 
   1875    buf[0] = (png_byte)unit;
   1876    memcpy(buf + 1, width, wlen + 1);      /* Append the '\0' here */
   1877    memcpy(buf + wlen + 2, height, hlen);  /* Do NOT append the '\0' here */
   1878 
   1879    png_debug1(3, "sCAL total length = %u", (unsigned int)total_len);
   1880    png_write_complete_chunk(png_ptr, png_sCAL, buf, total_len);
   1881 }
   1882 #endif
   1883 
   1884 #ifdef PNG_WRITE_pHYs_SUPPORTED
   1885 /* Write the pHYs chunk */
   1886 void /* PRIVATE */
   1887 png_write_pHYs(png_structrp png_ptr, png_uint_32 x_pixels_per_unit,
   1888     png_uint_32 y_pixels_per_unit,
   1889     int unit_type)
   1890 {
   1891    png_byte buf[9];
   1892 
   1893    png_debug(1, "in png_write_pHYs");
   1894 
   1895    if (unit_type >= PNG_RESOLUTION_LAST)
   1896       png_warning(png_ptr, "Unrecognized unit type for pHYs chunk");
   1897 
   1898    png_save_uint_32(buf, x_pixels_per_unit);
   1899    png_save_uint_32(buf + 4, y_pixels_per_unit);
   1900    buf[8] = (png_byte)unit_type;
   1901 
   1902    png_write_complete_chunk(png_ptr, png_pHYs, buf, (png_size_t)9);
   1903 }
   1904 #endif
   1905 
   1906 #ifdef PNG_WRITE_tIME_SUPPORTED
   1907 /* Write the tIME chunk.  Use either png_convert_from_struct_tm()
   1908  * or png_convert_from_time_t(), or fill in the structure yourself.
   1909  */
   1910 void /* PRIVATE */
   1911 png_write_tIME(png_structrp png_ptr, png_const_timep mod_time)
   1912 {
   1913    png_byte buf[7];
   1914 
   1915    png_debug(1, "in png_write_tIME");
   1916 
   1917    if (mod_time->month  > 12 || mod_time->month  < 1 ||
   1918        mod_time->day    > 31 || mod_time->day    < 1 ||
   1919        mod_time->hour   > 23 || mod_time->second > 60)
   1920    {
   1921       png_warning(png_ptr, "Invalid time specified for tIME chunk");
   1922       return;
   1923    }
   1924 
   1925    png_save_uint_16(buf, mod_time->year);
   1926    buf[2] = mod_time->month;
   1927    buf[3] = mod_time->day;
   1928    buf[4] = mod_time->hour;
   1929    buf[5] = mod_time->minute;
   1930    buf[6] = mod_time->second;
   1931 
   1932    png_write_complete_chunk(png_ptr, png_tIME, buf, (png_size_t)7);
   1933 }
   1934 #endif
   1935 
   1936 /* Initializes the row writing capability of libpng */
   1937 void /* PRIVATE */
   1938 png_write_start_row(png_structrp png_ptr)
   1939 {
   1940 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
   1941    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
   1942 
   1943    /* Start of interlace block */
   1944    static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
   1945 
   1946    /* Offset to next interlace block */
   1947    static PNG_CONST png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
   1948 
   1949    /* Start of interlace block in the y direction */
   1950    static PNG_CONST png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
   1951 
   1952    /* Offset to next interlace block in the y direction */
   1953    static PNG_CONST png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
   1954 #endif
   1955 
   1956    png_alloc_size_t buf_size;
   1957    int usr_pixel_depth;
   1958 
   1959    png_debug(1, "in png_write_start_row");
   1960 
   1961    usr_pixel_depth = png_ptr->usr_channels * png_ptr->usr_bit_depth;
   1962    buf_size = PNG_ROWBYTES(usr_pixel_depth, png_ptr->width) + 1;
   1963 
   1964    /* 1.5.6: added to allow checking in the row write code. */
   1965    png_ptr->transformed_pixel_depth = png_ptr->pixel_depth;
   1966    png_ptr->maximum_pixel_depth = (png_byte)usr_pixel_depth;
   1967 
   1968    /* Set up row buffer */
   1969    png_ptr->row_buf = (png_bytep)png_malloc(png_ptr, buf_size);
   1970 
   1971    png_ptr->row_buf[0] = PNG_FILTER_VALUE_NONE;
   1972 
   1973 #ifdef PNG_WRITE_FILTER_SUPPORTED
   1974    /* Set up filtering buffer, if using this filter */
   1975    if (png_ptr->do_filter & PNG_FILTER_SUB)
   1976    {
   1977       png_ptr->sub_row = (png_bytep)png_malloc(png_ptr, png_ptr->rowbytes + 1);
   1978 
   1979       png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB;
   1980    }
   1981 
   1982    /* We only need to keep the previous row if we are using one of these. */
   1983    if (png_ptr->do_filter & (PNG_FILTER_AVG | PNG_FILTER_UP | PNG_FILTER_PAETH))
   1984    {
   1985       /* Set up previous row buffer */
   1986       png_ptr->prev_row = (png_bytep)png_calloc(png_ptr, buf_size);
   1987 
   1988       if (png_ptr->do_filter & PNG_FILTER_UP)
   1989       {
   1990          png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
   1991             png_ptr->rowbytes + 1);
   1992 
   1993          png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
   1994       }
   1995 
   1996       if (png_ptr->do_filter & PNG_FILTER_AVG)
   1997       {
   1998          png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
   1999              png_ptr->rowbytes + 1);
   2000 
   2001          png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
   2002       }
   2003 
   2004       if (png_ptr->do_filter & PNG_FILTER_PAETH)
   2005       {
   2006          png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
   2007              png_ptr->rowbytes + 1);
   2008 
   2009          png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
   2010       }
   2011    }
   2012 #endif /* PNG_WRITE_FILTER_SUPPORTED */
   2013 
   2014 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
   2015    /* If interlaced, we need to set up width and height of pass */
   2016    if (png_ptr->interlaced)
   2017    {
   2018       if (!(png_ptr->transformations & PNG_INTERLACE))
   2019       {
   2020          png_ptr->num_rows = (png_ptr->height + png_pass_yinc[0] - 1 -
   2021              png_pass_ystart[0]) / png_pass_yinc[0];
   2022 
   2023          png_ptr->usr_width = (png_ptr->width + png_pass_inc[0] - 1 -
   2024              png_pass_start[0]) / png_pass_inc[0];
   2025       }
   2026 
   2027       else
   2028       {
   2029          png_ptr->num_rows = png_ptr->height;
   2030          png_ptr->usr_width = png_ptr->width;
   2031       }
   2032    }
   2033 
   2034    else
   2035 #endif
   2036    {
   2037       png_ptr->num_rows = png_ptr->height;
   2038       png_ptr->usr_width = png_ptr->width;
   2039    }
   2040 }
   2041 
   2042 /* Internal use only.  Called when finished processing a row of data. */
   2043 void /* PRIVATE */
   2044 png_write_finish_row(png_structrp png_ptr)
   2045 {
   2046 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
   2047    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
   2048 
   2049    /* Start of interlace block */
   2050    static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
   2051 
   2052    /* Offset to next interlace block */
   2053    static PNG_CONST png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
   2054 
   2055    /* Start of interlace block in the y direction */
   2056    static PNG_CONST png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
   2057 
   2058    /* Offset to next interlace block in the y direction */
   2059    static PNG_CONST png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
   2060 #endif
   2061 
   2062    png_debug(1, "in png_write_finish_row");
   2063 
   2064    /* Next row */
   2065    png_ptr->row_number++;
   2066 
   2067    /* See if we are done */
   2068    if (png_ptr->row_number < png_ptr->num_rows)
   2069       return;
   2070 
   2071 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
   2072    /* If interlaced, go to next pass */
   2073    if (png_ptr->interlaced)
   2074    {
   2075       png_ptr->row_number = 0;
   2076       if (png_ptr->transformations & PNG_INTERLACE)
   2077       {
   2078          png_ptr->pass++;
   2079       }
   2080 
   2081       else
   2082       {
   2083          /* Loop until we find a non-zero width or height pass */
   2084          do
   2085          {
   2086             png_ptr->pass++;
   2087 
   2088             if (png_ptr->pass >= 7)
   2089                break;
   2090 
   2091             png_ptr->usr_width = (png_ptr->width +
   2092                 png_pass_inc[png_ptr->pass] - 1 -
   2093                 png_pass_start[png_ptr->pass]) /
   2094                 png_pass_inc[png_ptr->pass];
   2095 
   2096             png_ptr->num_rows = (png_ptr->height +
   2097                 png_pass_yinc[png_ptr->pass] - 1 -
   2098                 png_pass_ystart[png_ptr->pass]) /
   2099                 png_pass_yinc[png_ptr->pass];
   2100 
   2101             if (png_ptr->transformations & PNG_INTERLACE)
   2102                break;
   2103 
   2104          } while (png_ptr->usr_width == 0 || png_ptr->num_rows == 0);
   2105 
   2106       }
   2107 
   2108       /* Reset the row above the image for the next pass */
   2109       if (png_ptr->pass < 7)
   2110       {
   2111          if (png_ptr->prev_row != NULL)
   2112             memset(png_ptr->prev_row, 0,
   2113                 (png_size_t)(PNG_ROWBYTES(png_ptr->usr_channels*
   2114                 png_ptr->usr_bit_depth, png_ptr->width)) + 1);
   2115 
   2116          return;
   2117       }
   2118    }
   2119 #endif
   2120 
   2121    /* If we get here, we've just written the last row, so we need
   2122       to flush the compressor */
   2123    png_compress_IDAT(png_ptr, NULL, 0, Z_FINISH);
   2124 }
   2125 
   2126 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
   2127 /* Pick out the correct pixels for the interlace pass.
   2128  * The basic idea here is to go through the row with a source
   2129  * pointer and a destination pointer (sp and dp), and copy the
   2130  * correct pixels for the pass.  As the row gets compacted,
   2131  * sp will always be >= dp, so we should never overwrite anything.
   2132  * See the default: case for the easiest code to understand.
   2133  */
   2134 void /* PRIVATE */
   2135 png_do_write_interlace(png_row_infop row_info, png_bytep row, int pass)
   2136 {
   2137    /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
   2138 
   2139    /* Start of interlace block */
   2140    static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
   2141 
   2142    /* Offset to next interlace block */
   2143    static PNG_CONST png_byte  png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
   2144 
   2145    png_debug(1, "in png_do_write_interlace");
   2146 
   2147    /* We don't have to do anything on the last pass (6) */
   2148    if (pass < 6)
   2149    {
   2150       /* Each pixel depth is handled separately */
   2151       switch (row_info->pixel_depth)
   2152       {
   2153          case 1:
   2154          {
   2155             png_bytep sp;
   2156             png_bytep dp;
   2157             int shift;
   2158             int d;
   2159             int value;
   2160             png_uint_32 i;
   2161             png_uint_32 row_width = row_info->width;
   2162 
   2163             dp = row;
   2164             d = 0;
   2165             shift = 7;
   2166 
   2167             for (i = png_pass_start[pass]; i < row_width;
   2168                i += png_pass_inc[pass])
   2169             {
   2170                sp = row + (png_size_t)(i >> 3);
   2171                value = (int)(*sp >> (7 - (int)(i & 0x07))) & 0x01;
   2172                d |= (value << shift);
   2173 
   2174                if (shift == 0)
   2175                {
   2176                   shift = 7;
   2177                   *dp++ = (png_byte)d;
   2178                   d = 0;
   2179                }
   2180 
   2181                else
   2182                   shift--;
   2183 
   2184             }
   2185             if (shift != 7)
   2186                *dp = (png_byte)d;
   2187 
   2188             break;
   2189          }
   2190 
   2191          case 2:
   2192          {
   2193             png_bytep sp;
   2194             png_bytep dp;
   2195             int shift;
   2196             int d;
   2197             int value;
   2198             png_uint_32 i;
   2199             png_uint_32 row_width = row_info->width;
   2200 
   2201             dp = row;
   2202             shift = 6;
   2203             d = 0;
   2204 
   2205             for (i = png_pass_start[pass]; i < row_width;
   2206                i += png_pass_inc[pass])
   2207             {
   2208                sp = row + (png_size_t)(i >> 2);
   2209                value = (*sp >> ((3 - (int)(i & 0x03)) << 1)) & 0x03;
   2210                d |= (value << shift);
   2211 
   2212                if (shift == 0)
   2213                {
   2214                   shift = 6;
   2215                   *dp++ = (png_byte)d;
   2216                   d = 0;
   2217                }
   2218 
   2219                else
   2220                   shift -= 2;
   2221             }
   2222             if (shift != 6)
   2223                *dp = (png_byte)d;
   2224 
   2225             break;
   2226          }
   2227 
   2228          case 4:
   2229          {
   2230             png_bytep sp;
   2231             png_bytep dp;
   2232             int shift;
   2233             int d;
   2234             int value;
   2235             png_uint_32 i;
   2236             png_uint_32 row_width = row_info->width;
   2237 
   2238             dp = row;
   2239             shift = 4;
   2240             d = 0;
   2241             for (i = png_pass_start[pass]; i < row_width;
   2242                 i += png_pass_inc[pass])
   2243             {
   2244                sp = row + (png_size_t)(i >> 1);
   2245                value = (*sp >> ((1 - (int)(i & 0x01)) << 2)) & 0x0f;
   2246                d |= (value << shift);
   2247 
   2248                if (shift == 0)
   2249                {
   2250                   shift = 4;
   2251                   *dp++ = (png_byte)d;
   2252                   d = 0;
   2253                }
   2254 
   2255                else
   2256                   shift -= 4;
   2257             }
   2258             if (shift != 4)
   2259                *dp = (png_byte)d;
   2260 
   2261             break;
   2262          }
   2263 
   2264          default:
   2265          {
   2266             png_bytep sp;
   2267             png_bytep dp;
   2268             png_uint_32 i;
   2269             png_uint_32 row_width = row_info->width;
   2270             png_size_t pixel_bytes;
   2271 
   2272             /* Start at the beginning */
   2273             dp = row;
   2274 
   2275             /* Find out how many bytes each pixel takes up */
   2276             pixel_bytes = (row_info->pixel_depth >> 3);
   2277 
   2278             /* Loop through the row, only looking at the pixels that matter */
   2279             for (i = png_pass_start[pass]; i < row_width;
   2280                i += png_pass_inc[pass])
   2281             {
   2282                /* Find out where the original pixel is */
   2283                sp = row + (png_size_t)i * pixel_bytes;
   2284 
   2285                /* Move the pixel */
   2286                if (dp != sp)
   2287                   memcpy(dp, sp, pixel_bytes);
   2288 
   2289                /* Next pixel */
   2290                dp += pixel_bytes;
   2291             }
   2292             break;
   2293          }
   2294       }
   2295       /* Set new row width */
   2296       row_info->width = (row_info->width +
   2297           png_pass_inc[pass] - 1 -
   2298           png_pass_start[pass]) /
   2299           png_pass_inc[pass];
   2300 
   2301       row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth,
   2302           row_info->width);
   2303    }
   2304 }
   2305 #endif
   2306 
   2307 /* This filters the row, chooses which filter to use, if it has not already
   2308  * been specified by the application, and then writes the row out with the
   2309  * chosen filter.
   2310  */
   2311 static void png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
   2312    png_size_t row_bytes);
   2313 
   2314 #define PNG_MAXSUM (((png_uint_32)(-1)) >> 1)
   2315 #define PNG_HISHIFT 10
   2316 #define PNG_LOMASK ((png_uint_32)0xffffL)
   2317 #define PNG_HIMASK ((png_uint_32)(~PNG_LOMASK >> PNG_HISHIFT))
   2318 void /* PRIVATE */
   2319 png_write_find_filter(png_structrp png_ptr, png_row_infop row_info)
   2320 {
   2321    png_bytep best_row;
   2322 #ifdef PNG_WRITE_FILTER_SUPPORTED
   2323    png_bytep prev_row, row_buf;
   2324    png_uint_32 mins, bpp;
   2325    png_byte filter_to_do = png_ptr->do_filter;
   2326    png_size_t row_bytes = row_info->rowbytes;
   2327 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2328    int num_p_filters = png_ptr->num_prev_filters;
   2329 #endif
   2330 
   2331    png_debug(1, "in png_write_find_filter");
   2332 
   2333 #ifndef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2334   if (png_ptr->row_number == 0 && filter_to_do == PNG_ALL_FILTERS)
   2335   {
   2336      /* These will never be selected so we need not test them. */
   2337      filter_to_do &= ~(PNG_FILTER_UP | PNG_FILTER_PAETH);
   2338   }
   2339 #endif
   2340 
   2341    /* Find out how many bytes offset each pixel is */
   2342    bpp = (row_info->pixel_depth + 7) >> 3;
   2343 
   2344    prev_row = png_ptr->prev_row;
   2345 #endif
   2346    best_row = png_ptr->row_buf;
   2347 #ifdef PNG_WRITE_FILTER_SUPPORTED
   2348    row_buf = best_row;
   2349    mins = PNG_MAXSUM;
   2350 
   2351    /* The prediction method we use is to find which method provides the
   2352     * smallest value when summing the absolute values of the distances
   2353     * from zero, using anything >= 128 as negative numbers.  This is known
   2354     * as the "minimum sum of absolute differences" heuristic.  Other
   2355     * heuristics are the "weighted minimum sum of absolute differences"
   2356     * (experimental and can in theory improve compression), and the "zlib
   2357     * predictive" method (not implemented yet), which does test compressions
   2358     * of lines using different filter methods, and then chooses the
   2359     * (series of) filter(s) that give minimum compressed data size (VERY
   2360     * computationally expensive).
   2361     *
   2362     * GRR 980525:  consider also
   2363     *
   2364     *   (1) minimum sum of absolute differences from running average (i.e.,
   2365     *       keep running sum of non-absolute differences & count of bytes)
   2366     *       [track dispersion, too?  restart average if dispersion too large?]
   2367     *
   2368     *  (1b) minimum sum of absolute differences from sliding average, probably
   2369     *       with window size <= deflate window (usually 32K)
   2370     *
   2371     *   (2) minimum sum of squared differences from zero or running average
   2372     *       (i.e., ~ root-mean-square approach)
   2373     */
   2374 
   2375 
   2376    /* We don't need to test the 'no filter' case if this is the only filter
   2377     * that has been chosen, as it doesn't actually do anything to the data.
   2378     */
   2379    if ((filter_to_do & PNG_FILTER_NONE) && filter_to_do != PNG_FILTER_NONE)
   2380    {
   2381       png_bytep rp;
   2382       png_uint_32 sum = 0;
   2383       png_size_t i;
   2384       int v;
   2385 
   2386       for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++)
   2387       {
   2388          v = *rp;
   2389          sum += (v < 128) ? v : 256 - v;
   2390       }
   2391 
   2392 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2393       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2394       {
   2395          png_uint_32 sumhi, sumlo;
   2396          int j;
   2397          sumlo = sum & PNG_LOMASK;
   2398          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK; /* Gives us some footroom */
   2399 
   2400          /* Reduce the sum if we match any of the previous rows */
   2401          for (j = 0; j < num_p_filters; j++)
   2402          {
   2403             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_NONE)
   2404             {
   2405                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
   2406                    PNG_WEIGHT_SHIFT;
   2407 
   2408                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
   2409                    PNG_WEIGHT_SHIFT;
   2410             }
   2411          }
   2412 
   2413          /* Factor in the cost of this filter (this is here for completeness,
   2414           * but it makes no sense to have a "cost" for the NONE filter, as
   2415           * it has the minimum possible computational cost - none).
   2416           */
   2417          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_NONE]) >>
   2418              PNG_COST_SHIFT;
   2419 
   2420          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_NONE]) >>
   2421              PNG_COST_SHIFT;
   2422 
   2423          if (sumhi > PNG_HIMASK)
   2424             sum = PNG_MAXSUM;
   2425 
   2426          else
   2427             sum = (sumhi << PNG_HISHIFT) + sumlo;
   2428       }
   2429 #endif
   2430       mins = sum;
   2431    }
   2432 
   2433    /* Sub filter */
   2434    if (filter_to_do == PNG_FILTER_SUB)
   2435    /* It's the only filter so no testing is needed */
   2436    {
   2437       png_bytep rp, lp, dp;
   2438       png_size_t i;
   2439 
   2440       for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp;
   2441            i++, rp++, dp++)
   2442       {
   2443          *dp = *rp;
   2444       }
   2445 
   2446       for (lp = row_buf + 1; i < row_bytes;
   2447          i++, rp++, lp++, dp++)
   2448       {
   2449          *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
   2450       }
   2451 
   2452       best_row = png_ptr->sub_row;
   2453    }
   2454 
   2455    else if (filter_to_do & PNG_FILTER_SUB)
   2456    {
   2457       png_bytep rp, dp, lp;
   2458       png_uint_32 sum = 0, lmins = mins;
   2459       png_size_t i;
   2460       int v;
   2461 
   2462 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2463       /* We temporarily increase the "minimum sum" by the factor we
   2464        * would reduce the sum of this filter, so that we can do the
   2465        * early exit comparison without scaling the sum each time.
   2466        */
   2467       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2468       {
   2469          int j;
   2470          png_uint_32 lmhi, lmlo;
   2471          lmlo = lmins & PNG_LOMASK;
   2472          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
   2473 
   2474          for (j = 0; j < num_p_filters; j++)
   2475          {
   2476             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_SUB)
   2477             {
   2478                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
   2479                    PNG_WEIGHT_SHIFT;
   2480 
   2481                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
   2482                    PNG_WEIGHT_SHIFT;
   2483             }
   2484          }
   2485 
   2486          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
   2487              PNG_COST_SHIFT;
   2488 
   2489          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
   2490              PNG_COST_SHIFT;
   2491 
   2492          if (lmhi > PNG_HIMASK)
   2493             lmins = PNG_MAXSUM;
   2494 
   2495          else
   2496             lmins = (lmhi << PNG_HISHIFT) + lmlo;
   2497       }
   2498 #endif
   2499 
   2500       for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp;
   2501            i++, rp++, dp++)
   2502       {
   2503          v = *dp = *rp;
   2504 
   2505          sum += (v < 128) ? v : 256 - v;
   2506       }
   2507 
   2508       for (lp = row_buf + 1; i < row_bytes;
   2509          i++, rp++, lp++, dp++)
   2510       {
   2511          v = *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
   2512 
   2513          sum += (v < 128) ? v : 256 - v;
   2514 
   2515          if (sum > lmins)  /* We are already worse, don't continue. */
   2516             break;
   2517       }
   2518 
   2519 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2520       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2521       {
   2522          int j;
   2523          png_uint_32 sumhi, sumlo;
   2524          sumlo = sum & PNG_LOMASK;
   2525          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
   2526 
   2527          for (j = 0; j < num_p_filters; j++)
   2528          {
   2529             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_SUB)
   2530             {
   2531                sumlo = (sumlo * png_ptr->inv_filter_weights[j]) >>
   2532                    PNG_WEIGHT_SHIFT;
   2533 
   2534                sumhi = (sumhi * png_ptr->inv_filter_weights[j]) >>
   2535                    PNG_WEIGHT_SHIFT;
   2536             }
   2537          }
   2538 
   2539          sumlo = (sumlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
   2540              PNG_COST_SHIFT;
   2541 
   2542          sumhi = (sumhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_SUB]) >>
   2543              PNG_COST_SHIFT;
   2544 
   2545          if (sumhi > PNG_HIMASK)
   2546             sum = PNG_MAXSUM;
   2547 
   2548          else
   2549             sum = (sumhi << PNG_HISHIFT) + sumlo;
   2550       }
   2551 #endif
   2552 
   2553       if (sum < mins)
   2554       {
   2555          mins = sum;
   2556          best_row = png_ptr->sub_row;
   2557       }
   2558    }
   2559 
   2560    /* Up filter */
   2561    if (filter_to_do == PNG_FILTER_UP)
   2562    {
   2563       png_bytep rp, dp, pp;
   2564       png_size_t i;
   2565 
   2566       for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1,
   2567           pp = prev_row + 1; i < row_bytes;
   2568           i++, rp++, pp++, dp++)
   2569       {
   2570          *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
   2571       }
   2572 
   2573       best_row = png_ptr->up_row;
   2574    }
   2575 
   2576    else if (filter_to_do & PNG_FILTER_UP)
   2577    {
   2578       png_bytep rp, dp, pp;
   2579       png_uint_32 sum = 0, lmins = mins;
   2580       png_size_t i;
   2581       int v;
   2582 
   2583 
   2584 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2585       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2586       {
   2587          int j;
   2588          png_uint_32 lmhi, lmlo;
   2589          lmlo = lmins & PNG_LOMASK;
   2590          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
   2591 
   2592          for (j = 0; j < num_p_filters; j++)
   2593          {
   2594             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_UP)
   2595             {
   2596                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
   2597                    PNG_WEIGHT_SHIFT;
   2598 
   2599                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
   2600                    PNG_WEIGHT_SHIFT;
   2601             }
   2602          }
   2603 
   2604          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_UP]) >>
   2605              PNG_COST_SHIFT;
   2606 
   2607          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_UP]) >>
   2608              PNG_COST_SHIFT;
   2609 
   2610          if (lmhi > PNG_HIMASK)
   2611             lmins = PNG_MAXSUM;
   2612 
   2613          else
   2614             lmins = (lmhi << PNG_HISHIFT) + lmlo;
   2615       }
   2616 #endif
   2617 
   2618       for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1,
   2619           pp = prev_row + 1; i < row_bytes; i++)
   2620       {
   2621          v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
   2622 
   2623          sum += (v < 128) ? v : 256 - v;
   2624 
   2625          if (sum > lmins)  /* We are already worse, don't continue. */
   2626             break;
   2627       }
   2628 
   2629 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2630       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2631       {
   2632          int j;
   2633          png_uint_32 sumhi, sumlo;
   2634          sumlo = sum & PNG_LOMASK;
   2635          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
   2636 
   2637          for (j = 0; j < num_p_filters; j++)
   2638          {
   2639             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_UP)
   2640             {
   2641                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
   2642                    PNG_WEIGHT_SHIFT;
   2643 
   2644                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
   2645                    PNG_WEIGHT_SHIFT;
   2646             }
   2647          }
   2648 
   2649          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_UP]) >>
   2650              PNG_COST_SHIFT;
   2651 
   2652          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_UP]) >>
   2653              PNG_COST_SHIFT;
   2654 
   2655          if (sumhi > PNG_HIMASK)
   2656             sum = PNG_MAXSUM;
   2657 
   2658          else
   2659             sum = (sumhi << PNG_HISHIFT) + sumlo;
   2660       }
   2661 #endif
   2662 
   2663       if (sum < mins)
   2664       {
   2665          mins = sum;
   2666          best_row = png_ptr->up_row;
   2667       }
   2668    }
   2669 
   2670    /* Avg filter */
   2671    if (filter_to_do == PNG_FILTER_AVG)
   2672    {
   2673       png_bytep rp, dp, pp, lp;
   2674       png_uint_32 i;
   2675 
   2676       for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1,
   2677            pp = prev_row + 1; i < bpp; i++)
   2678       {
   2679          *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
   2680       }
   2681 
   2682       for (lp = row_buf + 1; i < row_bytes; i++)
   2683       {
   2684          *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
   2685                  & 0xff);
   2686       }
   2687       best_row = png_ptr->avg_row;
   2688    }
   2689 
   2690    else if (filter_to_do & PNG_FILTER_AVG)
   2691    {
   2692       png_bytep rp, dp, pp, lp;
   2693       png_uint_32 sum = 0, lmins = mins;
   2694       png_size_t i;
   2695       int v;
   2696 
   2697 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2698       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2699       {
   2700          int j;
   2701          png_uint_32 lmhi, lmlo;
   2702          lmlo = lmins & PNG_LOMASK;
   2703          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
   2704 
   2705          for (j = 0; j < num_p_filters; j++)
   2706          {
   2707             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_AVG)
   2708             {
   2709                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
   2710                    PNG_WEIGHT_SHIFT;
   2711 
   2712                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
   2713                    PNG_WEIGHT_SHIFT;
   2714             }
   2715          }
   2716 
   2717          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_AVG]) >>
   2718              PNG_COST_SHIFT;
   2719 
   2720          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_AVG]) >>
   2721              PNG_COST_SHIFT;
   2722 
   2723          if (lmhi > PNG_HIMASK)
   2724             lmins = PNG_MAXSUM;
   2725 
   2726          else
   2727             lmins = (lmhi << PNG_HISHIFT) + lmlo;
   2728       }
   2729 #endif
   2730 
   2731       for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1,
   2732            pp = prev_row + 1; i < bpp; i++)
   2733       {
   2734          v = *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
   2735 
   2736          sum += (v < 128) ? v : 256 - v;
   2737       }
   2738 
   2739       for (lp = row_buf + 1; i < row_bytes; i++)
   2740       {
   2741          v = *dp++ =
   2742              (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2)) & 0xff);
   2743 
   2744          sum += (v < 128) ? v : 256 - v;
   2745 
   2746          if (sum > lmins)  /* We are already worse, don't continue. */
   2747             break;
   2748       }
   2749 
   2750 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2751       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2752       {
   2753          int j;
   2754          png_uint_32 sumhi, sumlo;
   2755          sumlo = sum & PNG_LOMASK;
   2756          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
   2757 
   2758          for (j = 0; j < num_p_filters; j++)
   2759          {
   2760             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_NONE)
   2761             {
   2762                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
   2763                    PNG_WEIGHT_SHIFT;
   2764 
   2765                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
   2766                    PNG_WEIGHT_SHIFT;
   2767             }
   2768          }
   2769 
   2770          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_AVG]) >>
   2771              PNG_COST_SHIFT;
   2772 
   2773          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_AVG]) >>
   2774              PNG_COST_SHIFT;
   2775 
   2776          if (sumhi > PNG_HIMASK)
   2777             sum = PNG_MAXSUM;
   2778 
   2779          else
   2780             sum = (sumhi << PNG_HISHIFT) + sumlo;
   2781       }
   2782 #endif
   2783 
   2784       if (sum < mins)
   2785       {
   2786          mins = sum;
   2787          best_row = png_ptr->avg_row;
   2788       }
   2789    }
   2790 
   2791    /* Paeth filter */
   2792    if (filter_to_do == PNG_FILTER_PAETH)
   2793    {
   2794       png_bytep rp, dp, pp, cp, lp;
   2795       png_size_t i;
   2796 
   2797       for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1,
   2798           pp = prev_row + 1; i < bpp; i++)
   2799       {
   2800          *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
   2801       }
   2802 
   2803       for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
   2804       {
   2805          int a, b, c, pa, pb, pc, p;
   2806 
   2807          b = *pp++;
   2808          c = *cp++;
   2809          a = *lp++;
   2810 
   2811          p = b - c;
   2812          pc = a - c;
   2813 
   2814 #ifdef PNG_USE_ABS
   2815          pa = abs(p);
   2816          pb = abs(pc);
   2817          pc = abs(p + pc);
   2818 #else
   2819          pa = p < 0 ? -p : p;
   2820          pb = pc < 0 ? -pc : pc;
   2821          pc = (p + pc) < 0 ? -(p + pc) : p + pc;
   2822 #endif
   2823 
   2824          p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
   2825 
   2826          *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
   2827       }
   2828       best_row = png_ptr->paeth_row;
   2829    }
   2830 
   2831    else if (filter_to_do & PNG_FILTER_PAETH)
   2832    {
   2833       png_bytep rp, dp, pp, cp, lp;
   2834       png_uint_32 sum = 0, lmins = mins;
   2835       png_size_t i;
   2836       int v;
   2837 
   2838 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2839       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2840       {
   2841          int j;
   2842          png_uint_32 lmhi, lmlo;
   2843          lmlo = lmins & PNG_LOMASK;
   2844          lmhi = (lmins >> PNG_HISHIFT) & PNG_HIMASK;
   2845 
   2846          for (j = 0; j < num_p_filters; j++)
   2847          {
   2848             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_PAETH)
   2849             {
   2850                lmlo = (lmlo * png_ptr->inv_filter_weights[j]) >>
   2851                    PNG_WEIGHT_SHIFT;
   2852 
   2853                lmhi = (lmhi * png_ptr->inv_filter_weights[j]) >>
   2854                    PNG_WEIGHT_SHIFT;
   2855             }
   2856          }
   2857 
   2858          lmlo = (lmlo * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_PAETH]) >>
   2859              PNG_COST_SHIFT;
   2860 
   2861          lmhi = (lmhi * png_ptr->inv_filter_costs[PNG_FILTER_VALUE_PAETH]) >>
   2862              PNG_COST_SHIFT;
   2863 
   2864          if (lmhi > PNG_HIMASK)
   2865             lmins = PNG_MAXSUM;
   2866 
   2867          else
   2868             lmins = (lmhi << PNG_HISHIFT) + lmlo;
   2869       }
   2870 #endif
   2871 
   2872       for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1,
   2873           pp = prev_row + 1; i < bpp; i++)
   2874       {
   2875          v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
   2876 
   2877          sum += (v < 128) ? v : 256 - v;
   2878       }
   2879 
   2880       for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
   2881       {
   2882          int a, b, c, pa, pb, pc, p;
   2883 
   2884          b = *pp++;
   2885          c = *cp++;
   2886          a = *lp++;
   2887 
   2888 #ifndef PNG_SLOW_PAETH
   2889          p = b - c;
   2890          pc = a - c;
   2891 #ifdef PNG_USE_ABS
   2892          pa = abs(p);
   2893          pb = abs(pc);
   2894          pc = abs(p + pc);
   2895 #else
   2896          pa = p < 0 ? -p : p;
   2897          pb = pc < 0 ? -pc : pc;
   2898          pc = (p + pc) < 0 ? -(p + pc) : p + pc;
   2899 #endif
   2900          p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
   2901 #else /* PNG_SLOW_PAETH */
   2902          p = a + b - c;
   2903          pa = abs(p - a);
   2904          pb = abs(p - b);
   2905          pc = abs(p - c);
   2906 
   2907          if (pa <= pb && pa <= pc)
   2908             p = a;
   2909 
   2910          else if (pb <= pc)
   2911             p = b;
   2912 
   2913          else
   2914             p = c;
   2915 #endif /* PNG_SLOW_PAETH */
   2916 
   2917          v = *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
   2918 
   2919          sum += (v < 128) ? v : 256 - v;
   2920 
   2921          if (sum > lmins)  /* We are already worse, don't continue. */
   2922             break;
   2923       }
   2924 
   2925 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2926       if (png_ptr->heuristic_method == PNG_FILTER_HEURISTIC_WEIGHTED)
   2927       {
   2928          int j;
   2929          png_uint_32 sumhi, sumlo;
   2930          sumlo = sum & PNG_LOMASK;
   2931          sumhi = (sum >> PNG_HISHIFT) & PNG_HIMASK;
   2932 
   2933          for (j = 0; j < num_p_filters; j++)
   2934          {
   2935             if (png_ptr->prev_filters[j] == PNG_FILTER_VALUE_PAETH)
   2936             {
   2937                sumlo = (sumlo * png_ptr->filter_weights[j]) >>
   2938                    PNG_WEIGHT_SHIFT;
   2939 
   2940                sumhi = (sumhi * png_ptr->filter_weights[j]) >>
   2941                    PNG_WEIGHT_SHIFT;
   2942             }
   2943          }
   2944 
   2945          sumlo = (sumlo * png_ptr->filter_costs[PNG_FILTER_VALUE_PAETH]) >>
   2946              PNG_COST_SHIFT;
   2947 
   2948          sumhi = (sumhi * png_ptr->filter_costs[PNG_FILTER_VALUE_PAETH]) >>
   2949              PNG_COST_SHIFT;
   2950 
   2951          if (sumhi > PNG_HIMASK)
   2952             sum = PNG_MAXSUM;
   2953 
   2954          else
   2955             sum = (sumhi << PNG_HISHIFT) + sumlo;
   2956       }
   2957 #endif
   2958 
   2959       if (sum < mins)
   2960       {
   2961          best_row = png_ptr->paeth_row;
   2962       }
   2963    }
   2964 #endif /* PNG_WRITE_FILTER_SUPPORTED */
   2965 
   2966    /* Do the actual writing of the filtered row data from the chosen filter. */
   2967    png_write_filtered_row(png_ptr, best_row, row_info->rowbytes+1);
   2968 
   2969 #ifdef PNG_WRITE_FILTER_SUPPORTED
   2970 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
   2971    /* Save the type of filter we picked this time for future calculations */
   2972    if (png_ptr->num_prev_filters > 0)
   2973    {
   2974       int j;
   2975 
   2976       for (j = 1; j < num_p_filters; j++)
   2977       {
   2978          png_ptr->prev_filters[j] = png_ptr->prev_filters[j - 1];
   2979       }
   2980 
   2981       png_ptr->prev_filters[j] = best_row[0];
   2982    }
   2983 #endif
   2984 #endif /* PNG_WRITE_FILTER_SUPPORTED */
   2985 }
   2986 
   2987 
   2988 /* Do the actual writing of a previously filtered row. */
   2989 static void
   2990 png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
   2991    png_size_t full_row_length/*includes filter byte*/)
   2992 {
   2993    png_debug(1, "in png_write_filtered_row");
   2994 
   2995    png_debug1(2, "filter = %d", filtered_row[0]);
   2996 
   2997    png_compress_IDAT(png_ptr, filtered_row, full_row_length, Z_NO_FLUSH);
   2998 
   2999    /* Swap the current and previous rows */
   3000    if (png_ptr->prev_row != NULL)
   3001    {
   3002       png_bytep tptr;
   3003 
   3004       tptr = png_ptr->prev_row;
   3005       png_ptr->prev_row = png_ptr->row_buf;
   3006       png_ptr->row_buf = tptr;
   3007    }
   3008 
   3009    /* Finish row - updates counters and flushes zlib if last row */
   3010    png_write_finish_row(png_ptr);
   3011 
   3012 #ifdef PNG_WRITE_FLUSH_SUPPORTED
   3013    png_ptr->flush_rows++;
   3014 
   3015    if (png_ptr->flush_dist > 0 &&
   3016        png_ptr->flush_rows >= png_ptr->flush_dist)
   3017    {
   3018       png_write_flush(png_ptr);
   3019    }
   3020 #endif
   3021 }
   3022 #endif /* PNG_WRITE_SUPPORTED */
   3023