pngwrite.c

   1
   2 /* pngwrite.c - general routines to write a PNG file
   3  *
   4  * Copyright (c) 2018-2019 Cosmin Truta
   5  * Copyright (c) 1998-2002,2004,2006-2018 Glenn Randers-Pehrson
   6  * Copyright (c) 1996-1997 Andreas Dilger
   7  * Copyright (c) 1995-1996 Guy Eric Schalnat, Group 42, Inc.
   8  *
   9  * This code is released under the libpng license.
  10  * For conditions of distribution and use, see the disclaimer
  11  * and license in png.h
  12  */
  13
  14 #include "pngpriv.h"
  15 #ifdef PNG_SIMPLIFIED_WRITE_STDIO_SUPPORTED
  16 #  include <errno.h>
  17 #endif /* SIMPLIFIED_WRITE_STDIO */
  18
  19 #ifdef PNG_WRITE_SUPPORTED
  20
  21 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
  22 /* Write out all the unknown chunks for the current given location */
  23 static void
  24 write_unknown_chunks(png_structrp png_ptr, png_const_inforp info_ptr,
  25     unsigned int where)
  26 {
  27    if (info_ptr->unknown_chunks_num != 0)
  28    {
  29       png_const_unknown_chunkp up;
  30
  31       png_debug(5, "writing extra chunks");
  32
  33       for (up = info_ptr->unknown_chunks;
  34            up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num;
  35            ++up)
  36          if ((up->location & where) != 0)
  37       {
  38          /* If per-chunk unknown chunk handling is enabled use it, otherwise
  39           * just write the chunks the application has set.
  40           */
  41 #ifdef PNG_SET_UNKNOWN_CHUNKS_SUPPORTED
  42          int keep = png_handle_as_unknown(png_ptr, up->name);
  43
  44          /* NOTE: this code is radically different from the read side in the
  45           * matter of handling an ancillary unknown chunk.  In the read side
  46           * the default behavior is to discard it, in the code below the default
  47           * behavior is to write it.  Critical chunks are, however, only
  48           * written if explicitly listed or if the default is set to write all
  49           * unknown chunks.
  50           *
  51           * The default handling is also slightly weird - it is not possible to
  52           * stop the writing of all unsafe-to-copy chunks!
  53           *
  54           * TODO: REVIEW: this would seem to be a bug.
  55           */
  56          if (keep != PNG_HANDLE_CHUNK_NEVER &&
  57              ((up->name[3] & 0x20) /* safe-to-copy overrides everything */ ||
  58               keep == PNG_HANDLE_CHUNK_ALWAYS ||
  59               (keep == PNG_HANDLE_CHUNK_AS_DEFAULT &&
  60                png_ptr->unknown_default == PNG_HANDLE_CHUNK_ALWAYS)))
  61 #endif
  62          {
  63             /* TODO: review, what is wrong with a zero length unknown chunk? */
  64             if (up->size == 0)
  65                png_warning(png_ptr, "Writing zero-length unknown chunk");
  66
  67             png_write_chunk(png_ptr, up->name, up->data, up->size);
  68          }
  69       }
  70    }
  71 }
  72 #endif /* WRITE_UNKNOWN_CHUNKS */
  73
  74 /* Writes all the PNG information.  This is the suggested way to use the
  75  * library.  If you have a new chunk to add, make a function to write it,
  76  * and put it in the correct location here.  If you want the chunk written
  77  * after the image data, put it in png_write_end().  I strongly encourage
  78  * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing
  79  * the chunk, as that will keep the code from breaking if you want to just
  80  * write a plain PNG file.  If you have long comments, I suggest writing
  81  * them in png_write_end(), and compressing them.
  82  */
  83 void PNGAPI
  84 png_write_info_before_PLTE(png_structrp png_ptr, png_const_inforp info_ptr)
  85 {
  86    png_debug(1, "in png_write_info_before_PLTE");
  87
  88    if (png_ptr == NULL || info_ptr == NULL)
  89       return;
  90
  91    if ((png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE) == 0)
  92    {
  93       /* Write PNG signature */
  94       png_write_sig(png_ptr);
  95
  96 #ifdef PNG_MNG_FEATURES_SUPPORTED
  97       if ((png_ptr->mode & PNG_HAVE_PNG_SIGNATURE) != 0 && \
  98           png_ptr->mng_features_permitted != 0)
  99       {
 100          png_warning(png_ptr,
 101              "MNG features are not allowed in a PNG datastream");
 102          png_ptr->mng_features_permitted = 0;
 103       }
 104 #endif
 105
 106       /* Write IHDR information. */
 107       png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height,
 108           info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type,
 109           info_ptr->filter_type,
 110 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
 111           info_ptr->interlace_type
 112 #else
 113           0
 114 #endif
 115          );
 116
 117       /* The rest of these check to see if the valid field has the appropriate
 118        * flag set, and if it does, writes the chunk.
 119        *
 120        * 1.6.0: COLORSPACE support controls the writing of these chunks too, and
 121        * the chunks will be written if the WRITE routine is there and
 122        * information * is available in the COLORSPACE. (See
 123        * png_colorspace_sync_info in png.c for where the valid flags get set.)
 124        *
 125        * Under certain circumstances the colorspace can be invalidated without
 126        * syncing the info_struct 'valid' flags; this happens if libpng detects
 127        * an error and calls png_error while the color space is being set, yet
 128        * the application continues writing the PNG.  So check the 'invalid'
 129        * flag here too.
 130        */
 131 #ifdef PNG_WRITE_APNG_SUPPORTED
 132       if (info_ptr->valid & PNG_INFO_acTL)
 133          png_write_acTL(png_ptr, info_ptr->num_frames, info_ptr->num_plays);
 134 #endif
 135 #ifdef PNG_GAMMA_SUPPORTED
 136 #  ifdef PNG_WRITE_gAMA_SUPPORTED
 137       if ((info_ptr->colorspace.flags & PNG_COLORSPACE_INVALID) == 0 &&
 138           (info_ptr->colorspace.flags & PNG_COLORSPACE_FROM_gAMA) != 0 &&
 139           (info_ptr->valid & PNG_INFO_gAMA) != 0)
 140          png_write_gAMA_fixed(png_ptr, info_ptr->colorspace.gamma);
 141 #  endif
 142 #endif
 143
 144 #ifdef PNG_COLORSPACE_SUPPORTED
 145       /* Write only one of sRGB or an ICC profile.  If a profile was supplied
 146        * and it matches one of the known sRGB ones issue a warning.
 147        */
 148 #  ifdef PNG_WRITE_iCCP_SUPPORTED
 149          if ((info_ptr->colorspace.flags & PNG_COLORSPACE_INVALID) == 0 &&
 150              (info_ptr->valid & PNG_INFO_iCCP) != 0)
 151          {
 152 #    ifdef PNG_WRITE_sRGB_SUPPORTED
 153                if ((info_ptr->valid & PNG_INFO_sRGB) != 0)
 154                   png_app_warning(png_ptr,
 155                       "profile matches sRGB but writing iCCP instead");
 156 #     endif
 157
 158             png_write_iCCP(png_ptr, info_ptr->iccp_name,
 159                 info_ptr->iccp_profile);
 160          }
 161 #     ifdef PNG_WRITE_sRGB_SUPPORTED
 162          else
 163 #     endif
 164 #  endif
 165
 166 #  ifdef PNG_WRITE_sRGB_SUPPORTED
 167          if ((info_ptr->colorspace.flags & PNG_COLORSPACE_INVALID) == 0 &&
 168              (info_ptr->valid & PNG_INFO_sRGB) != 0)
 169             png_write_sRGB(png_ptr, info_ptr->colorspace.rendering_intent);
 170 #  endif /* WRITE_sRGB */
 171 #endif /* COLORSPACE */
 172
 173 #ifdef PNG_WRITE_sBIT_SUPPORTED
 174          if ((info_ptr->valid & PNG_INFO_sBIT) != 0)
 175             png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type);
 176 #endif
 177
 178 #ifdef PNG_COLORSPACE_SUPPORTED
 179 #  ifdef PNG_WRITE_cHRM_SUPPORTED
 180          if ((info_ptr->colorspace.flags & PNG_COLORSPACE_INVALID) == 0 &&
 181              (info_ptr->colorspace.flags & PNG_COLORSPACE_FROM_cHRM) != 0 &&
 182              (info_ptr->valid & PNG_INFO_cHRM) != 0)
 183             png_write_cHRM_fixed(png_ptr, &info_ptr->colorspace.end_points_xy);
 184 #  endif
 185 #endif
 186
 187 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
 188          write_unknown_chunks(png_ptr, info_ptr, PNG_HAVE_IHDR);
 189 #endif
 190
 191       png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE;
 192    }
 193 }
 194
 195 void PNGAPI
 196 png_write_info(png_structrp png_ptr, png_const_inforp info_ptr)
 197 {
 198 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
 199    int i;
 200 #endif
 201
 202    png_debug(1, "in png_write_info");
 203
 204    if (png_ptr == NULL || info_ptr == NULL)
 205       return;
 206
 207    png_write_info_before_PLTE(png_ptr, info_ptr);
 208
 209    if ((info_ptr->valid & PNG_INFO_PLTE) != 0)
 210       png_write_PLTE(png_ptr, info_ptr->palette,
 211           (png_uint_32)info_ptr->num_palette);
 212
 213    else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
 214       png_error(png_ptr, "Valid palette required for paletted images");
 215
 216 #ifdef PNG_WRITE_tRNS_SUPPORTED
 217    if ((info_ptr->valid & PNG_INFO_tRNS) !=0)
 218    {
 219 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED
 220       /* Invert the alpha channel (in tRNS) */
 221       if ((png_ptr->transformations & PNG_INVERT_ALPHA) != 0 &&
 222           info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
 223       {
 224          int j, jend;
 225
 226          jend = info_ptr->num_trans;
 227          if (jend > PNG_MAX_PALETTE_LENGTH)
 228             jend = PNG_MAX_PALETTE_LENGTH;
 229
 230          for (j = 0; j<jend; ++j)
 231             info_ptr->trans_alpha[j] =
 232                (png_byte)(255 - info_ptr->trans_alpha[j]);
 233       }
 234 #endif
 235       png_write_tRNS(png_ptr, info_ptr->trans_alpha, &(info_ptr->trans_color),
 236           info_ptr->num_trans, info_ptr->color_type);
 237    }
 238 #endif
 239 #ifdef PNG_WRITE_bKGD_SUPPORTED
 240    if ((info_ptr->valid & PNG_INFO_bKGD) != 0)
 241       png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type);
 242 #endif
 243
 244 #ifdef PNG_WRITE_eXIf_SUPPORTED
 245    if ((info_ptr->valid & PNG_INFO_eXIf) != 0)
 246       png_write_eXIf(png_ptr, info_ptr->exif, info_ptr->num_exif);
 247 #endif
 248
 249 #ifdef PNG_WRITE_hIST_SUPPORTED
 250    if ((info_ptr->valid & PNG_INFO_hIST) != 0)
 251       png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette);
 252 #endif
 253
 254 #ifdef PNG_WRITE_oFFs_SUPPORTED
 255    if ((info_ptr->valid & PNG_INFO_oFFs) != 0)
 256       png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset,
 257           info_ptr->offset_unit_type);
 258 #endif
 259
 260 #ifdef PNG_WRITE_pCAL_SUPPORTED
 261    if ((info_ptr->valid & PNG_INFO_pCAL) != 0)
 262       png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0,
 263           info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams,
 264           info_ptr->pcal_units, info_ptr->pcal_params);
 265 #endif
 266
 267 #ifdef PNG_WRITE_sCAL_SUPPORTED
 268    if ((info_ptr->valid & PNG_INFO_sCAL) != 0)
 269       png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit,
 270           info_ptr->scal_s_width, info_ptr->scal_s_height);
 271 #endif /* sCAL */
 272
 273 #ifdef PNG_WRITE_pHYs_SUPPORTED
 274    if ((info_ptr->valid & PNG_INFO_pHYs) != 0)
 275       png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit,
 276           info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type);
 277 #endif /* pHYs */
 278
 279 #ifdef PNG_WRITE_tIME_SUPPORTED
 280    if ((info_ptr->valid & PNG_INFO_tIME) != 0)
 281    {
 282       png_write_tIME(png_ptr, &(info_ptr->mod_time));
 283       png_ptr->mode |= PNG_WROTE_tIME;
 284    }
 285 #endif /* tIME */
 286
 287 #ifdef PNG_WRITE_sPLT_SUPPORTED
 288    if ((info_ptr->valid & PNG_INFO_sPLT) != 0)
 289       for (i = 0; i < (int)info_ptr->splt_palettes_num; i++)
 290          png_write_sPLT(png_ptr, info_ptr->splt_palettes + i);
 291 #endif /* sPLT */
 292
 293 #ifdef PNG_WRITE_TEXT_SUPPORTED
 294    /* Check to see if we need to write text chunks */
 295    for (i = 0; i < info_ptr->num_text; i++)
 296    {
 297       png_debug2(2, "Writing header text chunk %d, type %d", i,
 298           info_ptr->text[i].compression);
 299       /* An internationalized chunk? */
 300       if (info_ptr->text[i].compression > 0)
 301       {
 302 #ifdef PNG_WRITE_iTXt_SUPPORTED
 303          /* Write international chunk */
 304          png_write_iTXt(png_ptr,
 305              info_ptr->text[i].compression,
 306              info_ptr->text[i].key,
 307              info_ptr->text[i].lang,
 308              info_ptr->text[i].lang_key,
 309              info_ptr->text[i].text);
 310          /* Mark this chunk as written */
 311          if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
 312             info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
 313          else
 314             info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
 315 #else
 316          png_warning(png_ptr, "Unable to write international text");
 317 #endif
 318       }
 319
 320       /* If we want a compressed text chunk */
 321       else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt)
 322       {
 323 #ifdef PNG_WRITE_zTXt_SUPPORTED
 324          /* Write compressed chunk */
 325          png_write_zTXt(png_ptr, info_ptr->text[i].key,
 326              info_ptr->text[i].text, info_ptr->text[i].compression);
 327          /* Mark this chunk as written */
 328          info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
 329 #else
 330          png_warning(png_ptr, "Unable to write compressed text");
 331 #endif
 332       }
 333
 334       else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
 335       {
 336 #ifdef PNG_WRITE_tEXt_SUPPORTED
 337          /* Write uncompressed chunk */
 338          png_write_tEXt(png_ptr, info_ptr->text[i].key,
 339              info_ptr->text[i].text,
 340              0);
 341          /* Mark this chunk as written */
 342          info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
 343 #else
 344          /* Can't get here */
 345          png_warning(png_ptr, "Unable to write uncompressed text");
 346 #endif
 347       }
 348    }
 349 #endif /* tEXt */
 350
 351 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
 352    write_unknown_chunks(png_ptr, info_ptr, PNG_HAVE_PLTE);
 353 #endif
 354 }
 355
 356 /* Writes the end of the PNG file.  If you don't want to write comments or
 357  * time information, you can pass NULL for info.  If you already wrote these
 358  * in png_write_info(), do not write them again here.  If you have long
 359  * comments, I suggest writing them here, and compressing them.
 360  */
 361 void PNGAPI
 362 png_write_end(png_structrp png_ptr, png_inforp info_ptr)
 363 {
 364    png_debug(1, "in png_write_end");
 365
 366    if (png_ptr == NULL)
 367       return;
 368
 369    if ((png_ptr->mode & PNG_HAVE_IDAT) == 0)
 370       png_error(png_ptr, "No IDATs written into file");
 371
 372 #ifdef PNG_WRITE_CHECK_FOR_INVALID_INDEX_SUPPORTED
 373    if (png_ptr->num_palette_max > png_ptr->num_palette)
 374       png_benign_error(png_ptr, "Wrote palette index exceeding num_palette");
 375 #endif
 376
 377 #ifdef PNG_WRITE_APNG_SUPPORTED
 378    if (png_ptr->num_frames_written != png_ptr->num_frames_to_write)
 379       png_error(png_ptr, "Not enough frames written");
 380 #endif
 381
 382    /* See if user wants us to write information chunks */
 383    if (info_ptr != NULL)
 384    {
 385 #ifdef PNG_WRITE_TEXT_SUPPORTED
 386       int i; /* local index variable */
 387 #endif
 388 #ifdef PNG_WRITE_tIME_SUPPORTED
 389       /* Check to see if user has supplied a time chunk */
 390       if ((info_ptr->valid & PNG_INFO_tIME) != 0 &&
 391           (png_ptr->mode & PNG_WROTE_tIME) == 0)
 392          png_write_tIME(png_ptr, &(info_ptr->mod_time));
 393
 394 #endif
 395 #ifdef PNG_WRITE_TEXT_SUPPORTED
 396       /* Loop through comment chunks */
 397       for (i = 0; i < info_ptr->num_text; i++)
 398       {
 399          png_debug2(2, "Writing trailer text chunk %d, type %d", i,
 400              info_ptr->text[i].compression);
 401          /* An internationalized chunk? */
 402          if (info_ptr->text[i].compression > 0)
 403          {
 404 #ifdef PNG_WRITE_iTXt_SUPPORTED
 405             /* Write international chunk */
 406             png_write_iTXt(png_ptr,
 407                 info_ptr->text[i].compression,
 408                 info_ptr->text[i].key,
 409                 info_ptr->text[i].lang,
 410                 info_ptr->text[i].lang_key,
 411                 info_ptr->text[i].text);
 412             /* Mark this chunk as written */
 413             if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
 414                info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
 415             else
 416                info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
 417 #else
 418             png_warning(png_ptr, "Unable to write international text");
 419 #endif
 420          }
 421
 422          else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt)
 423          {
 424 #ifdef PNG_WRITE_zTXt_SUPPORTED
 425             /* Write compressed chunk */
 426             png_write_zTXt(png_ptr, info_ptr->text[i].key,
 427                 info_ptr->text[i].text, info_ptr->text[i].compression);
 428             /* Mark this chunk as written */
 429             info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR;
 430 #else
 431             png_warning(png_ptr, "Unable to write compressed text");
 432 #endif
 433          }
 434
 435          else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE)
 436          {
 437 #ifdef PNG_WRITE_tEXt_SUPPORTED
 438             /* Write uncompressed chunk */
 439             png_write_tEXt(png_ptr, info_ptr->text[i].key,
 440                 info_ptr->text[i].text, 0);
 441             /* Mark this chunk as written */
 442             info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR;
 443 #else
 444             png_warning(png_ptr, "Unable to write uncompressed text");
 445 #endif
 446          }
 447       }
 448 #endif
 449
 450 #ifdef PNG_WRITE_eXIf_SUPPORTED
 451    if ((info_ptr->valid & PNG_INFO_eXIf) != 0)
 452       png_write_eXIf(png_ptr, info_ptr->exif, info_ptr->num_exif);
 453 #endif
 454
 455 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
 456       write_unknown_chunks(png_ptr, info_ptr, PNG_AFTER_IDAT);
 457 #endif
 458    }
 459
 460    png_ptr->mode |= PNG_AFTER_IDAT;
 461
 462    /* Write end of PNG file */
 463    png_write_IEND(png_ptr);
 464
 465    /* This flush, added in libpng-1.0.8, removed from libpng-1.0.9beta03,
 466     * and restored again in libpng-1.2.30, may cause some applications that
 467     * do not set png_ptr->output_flush_fn to crash.  If your application
 468     * experiences a problem, please try building libpng with
 469     * PNG_WRITE_FLUSH_AFTER_IEND_SUPPORTED defined, and report the event to
 470     * png-mng-implement at lists.sf.net .
 471     */
 472 #ifdef PNG_WRITE_FLUSH_SUPPORTED
 473 #  ifdef PNG_WRITE_FLUSH_AFTER_IEND_SUPPORTED
 474    png_flush(png_ptr);
 475 #  endif
 476 #endif
 477 }
 478
 479 #ifdef PNG_CONVERT_tIME_SUPPORTED
 480 void PNGAPI
 481 png_convert_from_struct_tm(png_timep ptime, const struct tm * ttime)
 482 {
 483    png_debug(1, "in png_convert_from_struct_tm");
 484
 485    ptime->year = (png_uint_16)(1900 + ttime->tm_year);
 486    ptime->month = (png_byte)(ttime->tm_mon + 1);
 487    ptime->day = (png_byte)ttime->tm_mday;
 488    ptime->hour = (png_byte)ttime->tm_hour;
 489    ptime->minute = (png_byte)ttime->tm_min;
 490    ptime->second = (png_byte)ttime->tm_sec;
 491 }
 492
 493 void PNGAPI
 494 png_convert_from_time_t(png_timep ptime, time_t ttime)
 495 {
 496    struct tm *tbuf;
 497
 498    png_debug(1, "in png_convert_from_time_t");
 499
 500    tbuf = gmtime(&ttime);
 501    png_convert_from_struct_tm(ptime, tbuf);
 502 }
 503 #endif
 504
 505 /* Initialize png_ptr structure, and allocate any memory needed */
 506 PNG_FUNCTION(png_structp,PNGAPI
 507 png_create_write_struct,(png_const_charp user_png_ver, png_voidp error_ptr,
 508     png_error_ptr error_fn, png_error_ptr warn_fn),PNG_ALLOCATED)
 509 {
 510 #ifndef PNG_USER_MEM_SUPPORTED
 511    png_structrp png_ptr = png_create_png_struct(user_png_ver, error_ptr,
 512        error_fn, warn_fn, NULL, NULL, NULL);
 513 #else
 514    return png_create_write_struct_2(user_png_ver, error_ptr, error_fn,
 515        warn_fn, NULL, NULL, NULL);
 516 }
 517
 518 /* Alternate initialize png_ptr structure, and allocate any memory needed */
 519 PNG_FUNCTION(png_structp,PNGAPI
 520 png_create_write_struct_2,(png_const_charp user_png_ver, png_voidp error_ptr,
 521     png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr,
 522     png_malloc_ptr malloc_fn, png_free_ptr free_fn),PNG_ALLOCATED)
 523 {
 524    png_structrp png_ptr = png_create_png_struct(user_png_ver, error_ptr,
 525        error_fn, warn_fn, mem_ptr, malloc_fn, free_fn);
 526 #endif /* USER_MEM */
 527    if (png_ptr != NULL)
 528    {
 529       /* Set the zlib control values to defaults; they can be overridden by the
 530        * application after the struct has been created.
 531        */
 532       png_ptr->zbuffer_size = PNG_ZBUF_SIZE;
 533
 534       /* The 'zlib_strategy' setting is irrelevant because png_default_claim in
 535        * pngwutil.c defaults it according to whether or not filters will be
 536        * used, and ignores this setting.
 537        */
 538       png_ptr->zlib_strategy = PNG_Z_DEFAULT_STRATEGY;
 539       png_ptr->zlib_level = PNG_Z_DEFAULT_COMPRESSION;
 540       png_ptr->zlib_mem_level = 8;
 541       png_ptr->zlib_window_bits = 15;
 542       png_ptr->zlib_method = 8;
 543
 544 #ifdef PNG_WRITE_COMPRESSED_TEXT_SUPPORTED
 545       png_ptr->zlib_text_strategy = PNG_TEXT_Z_DEFAULT_STRATEGY;
 546       png_ptr->zlib_text_level = PNG_TEXT_Z_DEFAULT_COMPRESSION;
 547       png_ptr->zlib_text_mem_level = 8;
 548       png_ptr->zlib_text_window_bits = 15;
 549       png_ptr->zlib_text_method = 8;
 550 #endif /* WRITE_COMPRESSED_TEXT */
 551
 552       /* This is a highly dubious configuration option; by default it is off,
 553        * but it may be appropriate for private builds that are testing
 554        * extensions not conformant to the current specification, or of
 555        * applications that must not fail to write at all costs!
 556        */
 557 #ifdef PNG_BENIGN_WRITE_ERRORS_SUPPORTED
 558       /* In stable builds only warn if an application error can be completely
 559        * handled.
 560        */
 561       png_ptr->flags |= PNG_FLAG_BENIGN_ERRORS_WARN;
 562 #endif
 563
 564       /* App warnings are warnings in release (or release candidate) builds but
 565        * are errors during development.
 566        */
 567 #if PNG_RELEASE_BUILD
 568       png_ptr->flags |= PNG_FLAG_APP_WARNINGS_WARN;
 569 #endif
 570
 571       /* TODO: delay this, it can be done in png_init_io() (if the app doesn't
 572        * do it itself) avoiding setting the default function if it is not
 573        * required.
 574        */
 575       png_set_write_fn(png_ptr, NULL, NULL, NULL);
 576    }
 577
 578    return png_ptr;
 579 }
 580
 581
 582 /* Write a few rows of image data.  If the image is interlaced,
 583  * either you will have to write the 7 sub images, or, if you
 584  * have called png_set_interlace_handling(), you will have to
 585  * "write" the image seven times.
 586  */
 587 void PNGAPI
 588 png_write_rows(png_structrp png_ptr, png_bytepp row,
 589     png_uint_32 num_rows)
 590 {
 591    png_uint_32 i; /* row counter */
 592    png_bytepp rp; /* row pointer */
 593
 594    png_debug(1, "in png_write_rows");
 595
 596    if (png_ptr == NULL)
 597       return;
 598
 599    /* Loop through the rows */
 600    for (i = 0, rp = row; i < num_rows; i++, rp++)
 601    {
 602       png_write_row(png_ptr, *rp);
 603    }
 604 }
 605
 606 /* Write the image.  You only need to call this function once, even
 607  * if you are writing an interlaced image.
 608  */
 609 void PNGAPI
 610 png_write_image(png_structrp png_ptr, png_bytepp image)
 611 {
 612    png_uint_32 i; /* row index */
 613    int pass, num_pass; /* pass variables */
 614    png_bytepp rp; /* points to current row */
 615
 616    if (png_ptr == NULL)
 617       return;
 618
 619    png_debug(1, "in png_write_image");
 620
 621 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
 622    /* Initialize interlace handling.  If image is not interlaced,
 623     * this will set pass to 1
 624     */
 625    num_pass = png_set_interlace_handling(png_ptr);
 626 #else
 627    num_pass = 1;
 628 #endif
 629    /* Loop through passes */
 630    for (pass = 0; pass < num_pass; pass++)
 631    {
 632       /* Loop through image */
 633       for (i = 0, rp = image; i < png_ptr->height; i++, rp++)
 634       {
 635          png_write_row(png_ptr, *rp);
 636       }
 637    }
 638 }
 639
 640 #ifdef PNG_MNG_FEATURES_SUPPORTED
 641 /* Performs intrapixel differencing  */
 642 static void
 643 png_do_write_intrapixel(png_row_infop row_info, png_bytep row)
 644 {
 645    png_debug(1, "in png_do_write_intrapixel");
 646
 647    if ((row_info->color_type & PNG_COLOR_MASK_COLOR) != 0)
 648    {
 649       int bytes_per_pixel;
 650       png_uint_32 row_width = row_info->width;
 651       if (row_info->bit_depth == 8)
 652       {
 653          png_bytep rp;
 654          png_uint_32 i;
 655
 656          if (row_info->color_type == PNG_COLOR_TYPE_RGB)
 657             bytes_per_pixel = 3;
 658
 659          else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 660             bytes_per_pixel = 4;
 661
 662          else
 663             return;
 664
 665          for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
 666          {
 667             *(rp)     = (png_byte)(*rp       - *(rp + 1));
 668             *(rp + 2) = (png_byte)(*(rp + 2) - *(rp + 1));
 669          }
 670       }
 671
 672 #ifdef PNG_WRITE_16BIT_SUPPORTED
 673       else if (row_info->bit_depth == 16)
 674       {
 675          png_bytep rp;
 676          png_uint_32 i;
 677
 678          if (row_info->color_type == PNG_COLOR_TYPE_RGB)
 679             bytes_per_pixel = 6;
 680
 681          else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 682             bytes_per_pixel = 8;
 683
 684          else
 685             return;
 686
 687          for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
 688          {
 689             png_uint_32 s0   = (png_uint_32)(*(rp    ) << 8) | *(rp + 1);
 690             png_uint_32 s1   = (png_uint_32)(*(rp + 2) << 8) | *(rp + 3);
 691             png_uint_32 s2   = (png_uint_32)(*(rp + 4) << 8) | *(rp + 5);
 692             png_uint_32 red  = (png_uint_32)((s0 - s1) & 0xffffL);
 693             png_uint_32 blue = (png_uint_32)((s2 - s1) & 0xffffL);
 694             *(rp    ) = (png_byte)(red >> 8);
 695             *(rp + 1) = (png_byte)red;
 696             *(rp + 4) = (png_byte)(blue >> 8);
 697             *(rp + 5) = (png_byte)blue;
 698          }
 699       }
 700 #endif /* WRITE_16BIT */
 701    }
 702 }
 703 #endif /* MNG_FEATURES */
 704
 705 /* Called by user to write a row of image data */
 706 void PNGAPI
 707 png_write_row(png_structrp png_ptr, png_const_bytep row)
 708 {
 709    /* 1.5.6: moved from png_struct to be a local structure: */
 710    png_row_info row_info;
 711
 712    if (png_ptr == NULL)
 713       return;
 714
 715    png_debug2(1, "in png_write_row (row %u, pass %d)",
 716        png_ptr->row_number, png_ptr->pass);
 717
 718    /* Initialize transformations and other stuff if first time */
 719    if (png_ptr->row_number == 0 && png_ptr->pass == 0)
 720    {
 721       /* Make sure we wrote the header info */
 722       if ((png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE) == 0)
 723          png_error(png_ptr,
 724              "png_write_info was never called before png_write_row");
 725
 726       /* Check for transforms that have been set but were defined out */
 727 #if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED)
 728       if ((png_ptr->transformations & PNG_INVERT_MONO) != 0)
 729          png_warning(png_ptr, "PNG_WRITE_INVERT_SUPPORTED is not defined");
 730 #endif
 731
 732 #if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED)
 733       if ((png_ptr->transformations & PNG_FILLER) != 0)
 734          png_warning(png_ptr, "PNG_WRITE_FILLER_SUPPORTED is not defined");
 735 #endif
 736 #if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && \
 737     defined(PNG_READ_PACKSWAP_SUPPORTED)
 738       if ((png_ptr->transformations & PNG_PACKSWAP) != 0)
 739          png_warning(png_ptr,
 740              "PNG_WRITE_PACKSWAP_SUPPORTED is not defined");
 741 #endif
 742
 743 #if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED)
 744       if ((png_ptr->transformations & PNG_PACK) != 0)
 745          png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined");
 746 #endif
 747
 748 #if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED)
 749       if ((png_ptr->transformations & PNG_SHIFT) != 0)
 750          png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined");
 751 #endif
 752
 753 #if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED)
 754       if ((png_ptr->transformations & PNG_BGR) != 0)
 755          png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined");
 756 #endif
 757
 758 #if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED)
 759       if ((png_ptr->transformations & PNG_SWAP_BYTES) != 0)
 760          png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined");
 761 #endif
 762
 763       png_write_start_row(png_ptr);
 764    }
 765
 766 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
 767    /* If interlaced and not interested in row, return */
 768    if (png_ptr->interlaced != 0 &&
 769        (png_ptr->transformations & PNG_INTERLACE) != 0)
 770    {
 771       switch (png_ptr->pass)
 772       {
 773          case 0:
 774             if ((png_ptr->row_number & 0x07) != 0)
 775             {
 776                png_write_finish_row(png_ptr);
 777                return;
 778             }
 779             break;
 780
 781          case 1:
 782             if ((png_ptr->row_number & 0x07) != 0 || png_ptr->width < 5)
 783             {
 784                png_write_finish_row(png_ptr);
 785                return;
 786             }
 787             break;
 788
 789          case 2:
 790             if ((png_ptr->row_number & 0x07) != 4)
 791             {
 792                png_write_finish_row(png_ptr);
 793                return;
 794             }
 795             break;
 796
 797          case 3:
 798             if ((png_ptr->row_number & 0x03) != 0 || png_ptr->width < 3)
 799             {
 800                png_write_finish_row(png_ptr);
 801                return;
 802             }
 803             break;
 804
 805          case 4:
 806             if ((png_ptr->row_number & 0x03) != 2)
 807             {
 808                png_write_finish_row(png_ptr);
 809                return;
 810             }
 811             break;
 812
 813          case 5:
 814             if ((png_ptr->row_number & 0x01) != 0 || png_ptr->width < 2)
 815             {
 816                png_write_finish_row(png_ptr);
 817                return;
 818             }
 819             break;
 820
 821          case 6:
 822             if ((png_ptr->row_number & 0x01) == 0)
 823             {
 824                png_write_finish_row(png_ptr);
 825                return;
 826             }
 827             break;
 828
 829          default: /* error: ignore it */
 830             break;
 831       }
 832    }
 833 #endif
 834
 835    /* Set up row info for transformations */
 836    row_info.color_type = png_ptr->color_type;
 837    row_info.width = png_ptr->usr_width;
 838    row_info.channels = png_ptr->usr_channels;
 839    row_info.bit_depth = png_ptr->usr_bit_depth;
 840    row_info.pixel_depth = (png_byte)(row_info.bit_depth * row_info.channels);
 841    row_info.rowbytes = PNG_ROWBYTES(row_info.pixel_depth, row_info.width);
 842
 843    png_debug1(3, "row_info->color_type = %d", row_info.color_type);
 844    png_debug1(3, "row_info->width = %u", row_info.width);
 845    png_debug1(3, "row_info->channels = %d", row_info.channels);
 846    png_debug1(3, "row_info->bit_depth = %d", row_info.bit_depth);
 847    png_debug1(3, "row_info->pixel_depth = %d", row_info.pixel_depth);
 848    png_debug1(3, "row_info->rowbytes = %lu", (unsigned long)row_info.rowbytes);
 849
 850    /* Copy user's row into buffer, leaving room for filter byte. */
 851    memcpy(png_ptr->row_buf + 1, row, row_info.rowbytes);
 852
 853 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
 854    /* Handle interlacing */
 855    if (png_ptr->interlaced && png_ptr->pass < 6 &&
 856        (png_ptr->transformations & PNG_INTERLACE) != 0)
 857    {
 858       png_do_write_interlace(&row_info, png_ptr->row_buf + 1, png_ptr->pass);
 859       /* This should always get caught above, but still ... */
 860       if (row_info.width == 0)
 861       {
 862          png_write_finish_row(png_ptr);
 863          return;
 864       }
 865    }
 866 #endif
 867
 868 #ifdef PNG_WRITE_TRANSFORMS_SUPPORTED
 869    /* Handle other transformations */
 870    if (png_ptr->transformations != 0)
 871       png_do_write_transformations(png_ptr, &row_info);
 872 #endif
 873
 874    /* At this point the row_info pixel depth must match the 'transformed' depth,
 875     * which is also the output depth.
 876     */
 877    if (row_info.pixel_depth != png_ptr->pixel_depth ||
 878        row_info.pixel_depth != png_ptr->transformed_pixel_depth)
 879       png_error(png_ptr, "internal write transform logic error");
 880
 881 #ifdef PNG_MNG_FEATURES_SUPPORTED
 882    /* Write filter_method 64 (intrapixel differencing) only if
 883     * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
 884     * 2. Libpng did not write a PNG signature (this filter_method is only
 885     *    used in PNG datastreams that are embedded in MNG datastreams) and
 886     * 3. The application called png_permit_mng_features with a mask that
 887     *    included PNG_FLAG_MNG_FILTER_64 and
 888     * 4. The filter_method is 64 and
 889     * 5. The color_type is RGB or RGBA
 890     */
 891    if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) != 0 &&
 892        (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING))
 893    {
 894       /* Intrapixel differencing */
 895       png_do_write_intrapixel(&row_info, png_ptr->row_buf + 1);
 896    }
 897 #endif
 898
 899 /* Added at libpng-1.5.10 */
 900 #ifdef PNG_WRITE_CHECK_FOR_INVALID_INDEX_SUPPORTED
 901    /* Check for out-of-range palette index */
 902    if (row_info.color_type == PNG_COLOR_TYPE_PALETTE &&
 903        png_ptr->num_palette_max >= 0)
 904       png_do_check_palette_indexes(png_ptr, &row_info);
 905 #endif
 906
 907    /* Find a filter if necessary, filter the row and write it out. */
 908    png_write_find_filter(png_ptr, &row_info);
 909
 910    if (png_ptr->write_row_fn != NULL)
 911       (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass);
 912 }
 913
 914 #ifdef PNG_WRITE_FLUSH_SUPPORTED
 915 /* Set the automatic flush interval or 0 to turn flushing off */
 916 void PNGAPI
 917 png_set_flush(png_structrp png_ptr, int nrows)
 918 {
 919    png_debug(1, "in png_set_flush");
 920
 921    if (png_ptr == NULL)
 922       return;
 923
 924    png_ptr->flush_dist = (nrows < 0 ? 0 : (png_uint_32)nrows);
 925 }
 926
 927 /* Flush the current output buffers now */
 928 void PNGAPI
 929 png_write_flush(png_structrp png_ptr)
 930 {
 931    png_debug(1, "in png_write_flush");
 932
 933    if (png_ptr == NULL)
 934       return;
 935
 936    /* We have already written out all of the data */
 937    if (png_ptr->row_number >= png_ptr->num_rows)
 938       return;
 939
 940    png_compress_IDAT(png_ptr, NULL, 0, Z_SYNC_FLUSH);
 941    png_ptr->flush_rows = 0;
 942    png_flush(png_ptr);
 943 }
 944 #endif /* WRITE_FLUSH */
 945
 946 /* Free any memory used in png_ptr struct without freeing the struct itself. */
 947 static void
 948 png_write_destroy(png_structrp png_ptr)
 949 {
 950    png_debug(1, "in png_write_destroy");
 951
 952    /* Free any memory zlib uses */
 953    if ((png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED) != 0)
 954       deflateEnd(&png_ptr->zstream);
 955
 956    /* Free our memory.  png_free checks NULL for us. */
 957    png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list);
 958    png_free(png_ptr, png_ptr->row_buf);
 959    png_ptr->row_buf = NULL;
 960 #ifdef PNG_WRITE_FILTER_SUPPORTED
 961    png_free(png_ptr, png_ptr->prev_row);
 962    png_free(png_ptr, png_ptr->try_row);
 963    png_free(png_ptr, png_ptr->tst_row);
 964    png_ptr->prev_row = NULL;
 965    png_ptr->try_row = NULL;
 966    png_ptr->tst_row = NULL;
 967 #endif
 968
 969 #ifdef PNG_SET_UNKNOWN_CHUNKS_SUPPORTED
 970    png_free(png_ptr, png_ptr->chunk_list);
 971    png_ptr->chunk_list = NULL;
 972 #endif
 973
 974    /* The error handling and memory handling information is left intact at this
 975     * point: the jmp_buf may still have to be freed.  See png_destroy_png_struct
 976     * for how this happens.
 977     */
 978 }
 979
 980 /* Free all memory used by the write.
 981  * In libpng 1.6.0 this API changed quietly to no longer accept a NULL value for
 982  * *png_ptr_ptr.  Prior to 1.6.0 it would accept such a value and it would free
 983  * the passed in info_structs but it would quietly fail to free any of the data
 984  * inside them.  In 1.6.0 it quietly does nothing (it has to be quiet because it
 985  * has no png_ptr.)
 986  */
 987 void PNGAPI
 988 png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr)
 989 {
 990    png_debug(1, "in png_destroy_write_struct");
 991
 992    if (png_ptr_ptr != NULL)
 993    {
 994       png_structrp png_ptr = *png_ptr_ptr;
 995
 996       if (png_ptr != NULL) /* added in libpng 1.6.0 */
 997       {
 998          png_destroy_info_struct(png_ptr, info_ptr_ptr);
 999
1000          *png_ptr_ptr = NULL;
1001          png_write_destroy(png_ptr);
1002          png_destroy_png_struct(png_ptr);
1003       }
1004    }
1005 }
1006
1007 /* Allow the application to select one or more row filters to use. */
1008 void PNGAPI
1009 png_set_filter(png_structrp png_ptr, int method, int filters)
1010 {
1011    png_debug(1, "in png_set_filter");
1012
1013    if (png_ptr == NULL)
1014       return;
1015
1016 #ifdef PNG_MNG_FEATURES_SUPPORTED
1017    if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) != 0 &&
1018        (method == PNG_INTRAPIXEL_DIFFERENCING))
1019       method = PNG_FILTER_TYPE_BASE;
1020
1021 #endif
1022    if (method == PNG_FILTER_TYPE_BASE)
1023    {
1024       switch (filters & (PNG_ALL_FILTERS | 0x07))
1025       {
1026 #ifdef PNG_WRITE_FILTER_SUPPORTED
1027          case 5:
1028          case 6:
1029          case 7: png_app_error(png_ptr, "Unknown row filter for method 0");
1030 #endif /* WRITE_FILTER */
1031             /* FALLTHROUGH */
1032 #ifdef __TIZEN__
1033             /* 'switch ~ case' is dead code.
1034              * If filters is #PNG_FILTER_VALUE_NONE, #PNG_FILTER_VALUE_SUB ...,
1035              * filter flag will be assigned to 'png_ptr->do_filter'. But
1036              * 'png_ptr->do_filter' is overwritten as 'filters' at below code.
1037              * Issue code(#1155) is 'png_ptr->do_filter = (png_byte)filters;'.
1038              * So the result of 'switch ~ case' has been lost at the line.
1039              * To keep the value of 'switch ~ case', we fixed it to apply
1040              * the result into 'filters'.
1041              * It will be updated into 'png_ptr->do_filter' at #1155 line.
1042              */
1043          case PNG_FILTER_VALUE_NONE:
1044             filters = PNG_FILTER_NONE; break;
1045 #else
1046          case PNG_FILTER_VALUE_NONE:
1047             png_ptr->do_filter = PNG_FILTER_NONE; break;
1048 #endif /* __TIZEN__ */
1049
1050 #ifdef PNG_WRITE_FILTER_SUPPORTED
1051 #ifdef __TIZEN__
1052          case PNG_FILTER_VALUE_SUB:
1053             filters = PNG_FILTER_SUB; break;
1054
1055          case PNG_FILTER_VALUE_UP:
1056             filters = PNG_FILTER_UP; break;
1057
1058          case PNG_FILTER_VALUE_AVG:
1059             filters = PNG_FILTER_AVG; break;
1060
1061          case PNG_FILTER_VALUE_PAETH:
1062             filters = PNG_FILTER_PAETH; break;
1063 #else
1064          case PNG_FILTER_VALUE_SUB:
1065             png_ptr->do_filter = PNG_FILTER_SUB; break;
1066
1067          case PNG_FILTER_VALUE_UP:
1068             png_ptr->do_filter = PNG_FILTER_UP; break;
1069
1070          case PNG_FILTER_VALUE_AVG:
1071             png_ptr->do_filter = PNG_FILTER_AVG; break;
1072
1073          case PNG_FILTER_VALUE_PAETH:
1074             png_ptr->do_filter = PNG_FILTER_PAETH; break;
1075 #endif /* __TIZEN__ */
1076          default:
1077             png_ptr->do_filter = (png_byte)filters; break;
1078 #else
1079          default:
1080             png_app_error(png_ptr, "Unknown row filter for method 0");
1081 #endif /* WRITE_FILTER */
1082       }
1083
1084 #ifdef PNG_WRITE_FILTER_SUPPORTED
1085       /* If we have allocated the row_buf, this means we have already started
1086        * with the image and we should have allocated all of the filter buffers
1087        * that have been selected.  If prev_row isn't already allocated, then
1088        * it is too late to start using the filters that need it, since we
1089        * will be missing the data in the previous row.  If an application
1090        * wants to start and stop using particular filters during compression,
1091        * it should start out with all of the filters, and then remove them
1092        * or add them back after the start of compression.
1093        *
1094        * NOTE: this is a nasty constraint on the code, because it means that the
1095        * prev_row buffer must be maintained even if there are currently no
1096        * 'prev_row' requiring filters active.
1097        */
1098       if (png_ptr->row_buf != NULL)
1099       {
1100          int num_filters;
1101          png_alloc_size_t buf_size;
1102
1103          /* Repeat the checks in png_write_start_row; 1 pixel high or wide
1104           * images cannot benefit from certain filters.  If this isn't done here
1105           * the check below will fire on 1 pixel high images.
1106           */
1107          if (png_ptr->height == 1)
1108             filters &= ~(PNG_FILTER_UP|PNG_FILTER_AVG|PNG_FILTER_PAETH);
1109
1110          if (png_ptr->width == 1)
1111             filters &= ~(PNG_FILTER_SUB|PNG_FILTER_AVG|PNG_FILTER_PAETH);
1112
1113          if ((filters & (PNG_FILTER_UP|PNG_FILTER_AVG|PNG_FILTER_PAETH)) != 0
1114             && png_ptr->prev_row == NULL)
1115          {
1116             /* This is the error case, however it is benign - the previous row
1117              * is not available so the filter can't be used.  Just warn here.
1118              */
1119             png_app_warning(png_ptr,
1120                 "png_set_filter: UP/AVG/PAETH cannot be added after start");
1121             filters &= ~(PNG_FILTER_UP|PNG_FILTER_AVG|PNG_FILTER_PAETH);
1122          }
1123
1124          num_filters = 0;
1125
1126          if (filters & PNG_FILTER_SUB)
1127             num_filters++;
1128
1129          if (filters & PNG_FILTER_UP)
1130             num_filters++;
1131
1132          if (filters & PNG_FILTER_AVG)
1133             num_filters++;
1134
1135          if (filters & PNG_FILTER_PAETH)
1136             num_filters++;
1137
1138          /* Allocate needed row buffers if they have not already been
1139           * allocated.
1140           */
1141          buf_size = PNG_ROWBYTES(png_ptr->usr_channels * png_ptr->usr_bit_depth,
1142              png_ptr->width) + 1;
1143
1144          if (png_ptr->try_row == NULL)
1145             png_ptr->try_row = png_voidcast(png_bytep,
1146                 png_malloc(png_ptr, buf_size));
1147
1148          if (num_filters > 1)
1149          {
1150             if (png_ptr->tst_row == NULL)
1151                png_ptr->tst_row = png_voidcast(png_bytep,
1152                    png_malloc(png_ptr, buf_size));
1153          }
1154       }
1155       png_ptr->do_filter = (png_byte)filters;
1156 #endif
1157    }
1158    else
1159       png_error(png_ptr, "Unknown custom filter method");
1160 }
1161
1162 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED /* DEPRECATED */
1163 /* Provide floating and fixed point APIs */
1164 #ifdef PNG_FLOATING_POINT_SUPPORTED
1165 void PNGAPI
1166 png_set_filter_heuristics(png_structrp png_ptr, int heuristic_method,
1167     int num_weights, png_const_doublep filter_weights,
1168     png_const_doublep filter_costs)
1169 {
1170    PNG_UNUSED(png_ptr)
1171    PNG_UNUSED(heuristic_method)
1172    PNG_UNUSED(num_weights)
1173    PNG_UNUSED(filter_weights)
1174    PNG_UNUSED(filter_costs)
1175 }
1176 #endif /* FLOATING_POINT */
1177
1178 #ifdef PNG_FIXED_POINT_SUPPORTED
1179 void PNGAPI
1180 png_set_filter_heuristics_fixed(png_structrp png_ptr, int heuristic_method,
1181     int num_weights, png_const_fixed_point_p filter_weights,
1182     png_const_fixed_point_p filter_costs)
1183 {
1184    PNG_UNUSED(png_ptr)
1185    PNG_UNUSED(heuristic_method)
1186    PNG_UNUSED(num_weights)
1187    PNG_UNUSED(filter_weights)
1188    PNG_UNUSED(filter_costs)
1189 }
1190 #endif /* FIXED_POINT */
1191 #endif /* WRITE_WEIGHTED_FILTER */
1192
1193 #ifdef PNG_WRITE_CUSTOMIZE_COMPRESSION_SUPPORTED
1194 void PNGAPI
1195 png_set_compression_level(png_structrp png_ptr, int level)
1196 {
1197    png_debug(1, "in png_set_compression_level");
1198
1199    if (png_ptr == NULL)
1200       return;
1201
1202    png_ptr->zlib_level = level;
1203 }
1204
1205 void PNGAPI
1206 png_set_compression_mem_level(png_structrp png_ptr, int mem_level)
1207 {
1208    png_debug(1, "in png_set_compression_mem_level");
1209
1210    if (png_ptr == NULL)
1211       return;
1212
1213    png_ptr->zlib_mem_level = mem_level;
1214 }
1215
1216 void PNGAPI
1217 png_set_compression_strategy(png_structrp png_ptr, int strategy)
1218 {
1219    png_debug(1, "in png_set_compression_strategy");
1220
1221    if (png_ptr == NULL)
1222       return;
1223
1224    /* The flag setting here prevents the libpng dynamic selection of strategy.
1225     */
1226    png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY;
1227    png_ptr->zlib_strategy = strategy;
1228 }
1229
1230 /* If PNG_WRITE_OPTIMIZE_CMF_SUPPORTED is defined, libpng will use a
1231  * smaller value of window_bits if it can do so safely.
1232  */
1233 void PNGAPI
1234 png_set_compression_window_bits(png_structrp png_ptr, int window_bits)
1235 {
1236    if (png_ptr == NULL)
1237       return;
1238
1239    /* Prior to 1.6.0 this would warn but then set the window_bits value. This
1240     * meant that negative window bits values could be selected that would cause
1241     * libpng to write a non-standard PNG file with raw deflate or gzip
1242     * compressed IDAT or ancillary chunks.  Such files can be read and there is
1243     * no warning on read, so this seems like a very bad idea.
1244     */
1245    if (window_bits > 15)
1246    {
1247       png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
1248       window_bits = 15;
1249    }
1250
1251    else if (window_bits < 8)
1252    {
1253       png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
1254       window_bits = 8;
1255    }
1256
1257    png_ptr->zlib_window_bits = window_bits;
1258 }
1259
1260 void PNGAPI
1261 png_set_compression_method(png_structrp png_ptr, int method)
1262 {
1263    png_debug(1, "in png_set_compression_method");
1264
1265    if (png_ptr == NULL)
1266       return;
1267
1268    /* This would produce an invalid PNG file if it worked, but it doesn't and
1269     * deflate will fault it, so it is harmless to just warn here.
1270     */
1271    if (method != 8)
1272       png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1273
1274    png_ptr->zlib_method = method;
1275 }
1276 #endif /* WRITE_CUSTOMIZE_COMPRESSION */
1277
1278 /* The following were added to libpng-1.5.4 */
1279 #ifdef PNG_WRITE_CUSTOMIZE_ZTXT_COMPRESSION_SUPPORTED
1280 void PNGAPI
1281 png_set_text_compression_level(png_structrp png_ptr, int level)
1282 {
1283    png_debug(1, "in png_set_text_compression_level");
1284
1285    if (png_ptr == NULL)
1286       return;
1287
1288    png_ptr->zlib_text_level = level;
1289 }
1290
1291 void PNGAPI
1292 png_set_text_compression_mem_level(png_structrp png_ptr, int mem_level)
1293 {
1294    png_debug(1, "in png_set_text_compression_mem_level");
1295
1296    if (png_ptr == NULL)
1297       return;
1298
1299    png_ptr->zlib_text_mem_level = mem_level;
1300 }
1301
1302 void PNGAPI
1303 png_set_text_compression_strategy(png_structrp png_ptr, int strategy)
1304 {
1305    png_debug(1, "in png_set_text_compression_strategy");
1306
1307    if (png_ptr == NULL)
1308       return;
1309
1310    png_ptr->zlib_text_strategy = strategy;
1311 }
1312
1313 /* If PNG_WRITE_OPTIMIZE_CMF_SUPPORTED is defined, libpng will use a
1314  * smaller value of window_bits if it can do so safely.
1315  */
1316 void PNGAPI
1317 png_set_text_compression_window_bits(png_structrp png_ptr, int window_bits)
1318 {
1319    if (png_ptr == NULL)
1320       return;
1321
1322    if (window_bits > 15)
1323    {
1324       png_warning(png_ptr, "Only compression windows <= 32k supported by PNG");
1325       window_bits = 15;
1326    }
1327
1328    else if (window_bits < 8)
1329    {
1330       png_warning(png_ptr, "Only compression windows >= 256 supported by PNG");
1331       window_bits = 8;
1332    }
1333
1334    png_ptr->zlib_text_window_bits = window_bits;
1335 }
1336
1337 void PNGAPI
1338 png_set_text_compression_method(png_structrp png_ptr, int method)
1339 {
1340    png_debug(1, "in png_set_text_compression_method");
1341
1342    if (png_ptr == NULL)
1343       return;
1344
1345    if (method != 8)
1346       png_warning(png_ptr, "Only compression method 8 is supported by PNG");
1347
1348    png_ptr->zlib_text_method = method;
1349 }
1350 #endif /* WRITE_CUSTOMIZE_ZTXT_COMPRESSION */
1351 /* end of API added to libpng-1.5.4 */
1352
1353 void PNGAPI
1354 png_set_write_status_fn(png_structrp png_ptr, png_write_status_ptr write_row_fn)
1355 {
1356    if (png_ptr == NULL)
1357       return;
1358
1359    png_ptr->write_row_fn = write_row_fn;
1360 }
1361
1362 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
1363 void PNGAPI
1364 png_set_write_user_transform_fn(png_structrp png_ptr, png_user_transform_ptr
1365     write_user_transform_fn)
1366 {
1367    png_debug(1, "in png_set_write_user_transform_fn");
1368
1369    if (png_ptr == NULL)
1370       return;
1371
1372    png_ptr->transformations |= PNG_USER_TRANSFORM;
1373    png_ptr->write_user_transform_fn = write_user_transform_fn;
1374 }
1375 #endif
1376
1377
1378 #ifdef PNG_INFO_IMAGE_SUPPORTED
1379 void PNGAPI
1380 png_write_png(png_structrp png_ptr, png_inforp info_ptr,
1381     int transforms, voidp params)
1382 {
1383    if (png_ptr == NULL || info_ptr == NULL)
1384       return;
1385
1386    if ((info_ptr->valid & PNG_INFO_IDAT) == 0)
1387    {
1388       png_app_error(png_ptr, "no rows for png_write_image to write");
1389       return;
1390    }
1391
1392    /* Write the file header information. */
1393    png_write_info(png_ptr, info_ptr);
1394
1395    /* ------ these transformations don't touch the info structure ------- */
1396
1397    /* Invert monochrome pixels */
1398    if ((transforms & PNG_TRANSFORM_INVERT_MONO) != 0)
1399 #ifdef PNG_WRITE_INVERT_SUPPORTED
1400       png_set_invert_mono(png_ptr);
1401 #else
1402       png_app_error(png_ptr, "PNG_TRANSFORM_INVERT_MONO not supported");
1403 #endif
1404
1405    /* Shift the pixels up to a legal bit depth and fill in
1406     * as appropriate to correctly scale the image.
1407     */
1408    if ((transforms & PNG_TRANSFORM_SHIFT) != 0)
1409 #ifdef PNG_WRITE_SHIFT_SUPPORTED
1410       if ((info_ptr->valid & PNG_INFO_sBIT) != 0)
1411          png_set_shift(png_ptr, &info_ptr->sig_bit);
1412 #else
1413       png_app_error(png_ptr, "PNG_TRANSFORM_SHIFT not supported");
1414 #endif
1415
1416    /* Pack pixels into bytes */
1417    if ((transforms & PNG_TRANSFORM_PACKING) != 0)
1418 #ifdef PNG_WRITE_PACK_SUPPORTED
1419       png_set_packing(png_ptr);
1420 #else
1421       png_app_error(png_ptr, "PNG_TRANSFORM_PACKING not supported");
1422 #endif
1423
1424    /* Swap location of alpha bytes from ARGB to RGBA */
1425    if ((transforms & PNG_TRANSFORM_SWAP_ALPHA) != 0)
1426 #ifdef PNG_WRITE_SWAP_ALPHA_SUPPORTED
1427       png_set_swap_alpha(png_ptr);
1428 #else
1429       png_app_error(png_ptr, "PNG_TRANSFORM_SWAP_ALPHA not supported");
1430 #endif
1431
1432    /* Remove a filler (X) from XRGB/RGBX/AG/GA into to convert it into
1433     * RGB, note that the code expects the input color type to be G or RGB; no
1434     * alpha channel.
1435     */
1436    if ((transforms & (PNG_TRANSFORM_STRIP_FILLER_AFTER|
1437        PNG_TRANSFORM_STRIP_FILLER_BEFORE)) != 0)
1438    {
1439 #ifdef PNG_WRITE_FILLER_SUPPORTED
1440       if ((transforms & PNG_TRANSFORM_STRIP_FILLER_AFTER) != 0)
1441       {
1442          if ((transforms & PNG_TRANSFORM_STRIP_FILLER_BEFORE) != 0)
1443             png_app_error(png_ptr,
1444                 "PNG_TRANSFORM_STRIP_FILLER: BEFORE+AFTER not supported");
1445
1446          /* Continue if ignored - this is the pre-1.6.10 behavior */
1447          png_set_filler(png_ptr, 0, PNG_FILLER_AFTER);
1448       }
1449
1450       else if ((transforms & PNG_TRANSFORM_STRIP_FILLER_BEFORE) != 0)
1451          png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE);
1452 #else
1453       png_app_error(png_ptr, "PNG_TRANSFORM_STRIP_FILLER not supported");
1454 #endif
1455    }
1456
1457    /* Flip BGR pixels to RGB */
1458    if ((transforms & PNG_TRANSFORM_BGR) != 0)
1459 #ifdef PNG_WRITE_BGR_SUPPORTED
1460       png_set_bgr(png_ptr);
1461 #else
1462       png_app_error(png_ptr, "PNG_TRANSFORM_BGR not supported");
1463 #endif
1464
1465    /* Swap bytes of 16-bit files to most significant byte first */
1466    if ((transforms & PNG_TRANSFORM_SWAP_ENDIAN) != 0)
1467 #ifdef PNG_WRITE_SWAP_SUPPORTED
1468       png_set_swap(png_ptr);
1469 #else
1470       png_app_error(png_ptr, "PNG_TRANSFORM_SWAP_ENDIAN not supported");
1471 #endif
1472
1473    /* Swap bits of 1-bit, 2-bit, 4-bit packed pixel formats */
1474    if ((transforms & PNG_TRANSFORM_PACKSWAP) != 0)
1475 #ifdef PNG_WRITE_PACKSWAP_SUPPORTED
1476       png_set_packswap(png_ptr);
1477 #else
1478       png_app_error(png_ptr, "PNG_TRANSFORM_PACKSWAP not supported");
1479 #endif
1480
1481    /* Invert the alpha channel from opacity to transparency */
1482    if ((transforms & PNG_TRANSFORM_INVERT_ALPHA) != 0)
1483 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED
1484       png_set_invert_alpha(png_ptr);
1485 #else
1486       png_app_error(png_ptr, "PNG_TRANSFORM_INVERT_ALPHA not supported");
1487 #endif
1488
1489    /* ----------------------- end of transformations ------------------- */
1490
1491    /* Write the bits */
1492    png_write_image(png_ptr, info_ptr->row_pointers);
1493
1494    /* It is REQUIRED to call this to finish writing the rest of the file */
1495    png_write_end(png_ptr, info_ptr);
1496
1497    PNG_UNUSED(params)
1498 }
1499 #endif
1500
1501 #ifdef PNG_WRITE_APNG_SUPPORTED
1502 void PNGAPI
1503 png_write_frame_head(png_structp png_ptr, png_infop info_ptr,
1504     png_bytepp row_pointers, png_uint_32 width, png_uint_32 height,
1505     png_uint_32 x_offset, png_uint_32 y_offset,
1506     png_uint_16 delay_num, png_uint_16 delay_den, png_byte dispose_op,
1507     png_byte blend_op)
1508 {
1509     png_debug(1, "in png_write_frame_head");
1510
1511     /* there is a chance this has been set after png_write_info was called,
1512     * so it would be set but not written. is there a way to be sure? */
1513     if (!(info_ptr->valid & PNG_INFO_acTL))
1514         png_error(png_ptr, "png_write_frame_head(): acTL not set");
1515
1516     png_write_reset(png_ptr);
1517
1518     png_write_reinit(png_ptr, info_ptr, width, height);
1519
1520     if ( !(png_ptr->num_frames_written == 0 &&
1521            (png_ptr->apng_flags & PNG_FIRST_FRAME_HIDDEN) ) )
1522         png_write_fcTL(png_ptr, width, height, x_offset, y_offset,
1523                        delay_num, delay_den, dispose_op, blend_op);
1524
1525     PNG_UNUSED(row_pointers)
1526 }
1527
1528 void PNGAPI
1529 png_write_frame_tail(png_structp png_ptr, png_infop info_ptr)
1530 {
1531     png_debug(1, "in png_write_frame_tail");
1532
1533     png_ptr->num_frames_written++;
1534
1535     PNG_UNUSED(info_ptr)
1536 }
1537 #endif /* PNG_WRITE_APNG_SUPPORTED */
1538
1539 #ifdef PNG_SIMPLIFIED_WRITE_SUPPORTED
1540 /* Initialize the write structure - general purpose utility. */
1541 static int
1542 png_image_write_init(png_imagep image)
1543 {
1544    png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, image,
1545        png_safe_error, png_safe_warning);
1546
1547    if (png_ptr != NULL)
1548    {
1549       png_infop info_ptr = png_create_info_struct(png_ptr);
1550
1551       if (info_ptr != NULL)
1552       {
1553          png_controlp control = png_voidcast(png_controlp,
1554              png_malloc_warn(png_ptr, (sizeof *control)));
1555
1556          if (control != NULL)
1557          {
1558             memset(control, 0, (sizeof *control));
1559
1560             control->png_ptr = png_ptr;
1561             control->info_ptr = info_ptr;
1562             control->for_write = 1;
1563
1564             image->opaque = control;
1565             return 1;
1566          }
1567
1568          /* Error clean up */
1569          png_destroy_info_struct(png_ptr, &info_ptr);
1570       }
1571
1572       png_destroy_write_struct(&png_ptr, NULL);
1573    }
1574
1575    return png_image_error(image, "png_image_write_: out of memory");
1576 }
1577
1578 /* Arguments to png_image_write_main: */
1579 typedef struct
1580 {
1581    /* Arguments: */
1582    png_imagep      image;
1583    png_const_voidp buffer;
1584    png_int_32      row_stride;
1585    png_const_voidp colormap;
1586    int             convert_to_8bit;
1587    /* Local variables: */
1588    png_const_voidp first_row;
1589    ptrdiff_t       row_bytes;
1590    png_voidp       local_row;
1591    /* Byte count for memory writing */
1592    png_bytep        memory;
1593    png_alloc_size_t memory_bytes; /* not used for STDIO */
1594    png_alloc_size_t output_bytes; /* running total */
1595 } png_image_write_control;
1596
1597 /* Write png_uint_16 input to a 16-bit PNG; the png_ptr has already been set to
1598  * do any necessary byte swapping.  The component order is defined by the
1599  * png_image format value.
1600  */
1601 static int
1602 png_write_image_16bit(png_voidp argument)
1603 {
1604    png_image_write_control *display = png_voidcast(png_image_write_control*,
1605        argument);
1606    png_imagep image = display->image;
1607    png_structrp png_ptr = image->opaque->png_ptr;
1608
1609    png_const_uint_16p input_row = png_voidcast(png_const_uint_16p,
1610        display->first_row);
1611    png_uint_16p output_row = png_voidcast(png_uint_16p, display->local_row);
1612    png_uint_16p row_end;
1613    unsigned int channels = (image->format & PNG_FORMAT_FLAG_COLOR) != 0 ?
1614        3 : 1;
1615    int aindex = 0;
1616    png_uint_32 y = image->height;
1617
1618    if ((image->format & PNG_FORMAT_FLAG_ALPHA) != 0)
1619    {
1620 #   ifdef PNG_SIMPLIFIED_WRITE_AFIRST_SUPPORTED
1621       if ((image->format & PNG_FORMAT_FLAG_AFIRST) != 0)
1622       {
1623          aindex = -1;
1624          ++input_row; /* To point to the first component */
1625          ++output_row;
1626       }
1627          else
1628             aindex = (int)channels;
1629 #     else
1630          aindex = (int)channels;
1631 #     endif
1632    }
1633
1634    else
1635       png_error(png_ptr, "png_write_image: internal call error");
1636
1637    /* Work out the output row end and count over this, note that the increment
1638     * above to 'row' means that row_end can actually be beyond the end of the
1639     * row; this is correct.
1640     */
1641    row_end = output_row + image->width * (channels+1);
1642
1643    for (; y > 0; --y)
1644    {
1645       png_const_uint_16p in_ptr = input_row;
1646       png_uint_16p out_ptr = output_row;
1647
1648       while (out_ptr < row_end)
1649       {
1650          png_uint_16 alpha = in_ptr[aindex];
1651          png_uint_32 reciprocal = 0;
1652          int c;
1653
1654          out_ptr[aindex] = alpha;
1655
1656          /* Calculate a reciprocal.  The correct calculation is simply
1657           * component/alpha*65535 << 15. (I.e. 15 bits of precision); this
1658           * allows correct rounding by adding .5 before the shift.  'reciprocal'
1659           * is only initialized when required.
1660           */
1661          if (alpha > 0 && alpha < 65535)
1662             reciprocal = ((0xffff<<15)+(alpha>>1))/alpha;
1663
1664          c = (int)channels;
1665          do /* always at least one channel */
1666          {
1667             png_uint_16 component = *in_ptr++;
1668
1669             /* The following gives 65535 for an alpha of 0, which is fine,
1670              * otherwise if 0/0 is represented as some other value there is more
1671              * likely to be a discontinuity which will probably damage
1672              * compression when moving from a fully transparent area to a
1673              * nearly transparent one.  (The assumption here is that opaque
1674              * areas tend not to be 0 intensity.)
1675              */
1676             if (component >= alpha)
1677                component = 65535;
1678
1679             /* component<alpha, so component/alpha is less than one and
1680              * component*reciprocal is less than 2^31.
1681              */
1682             else if (component > 0 && alpha < 65535)
1683             {
1684                png_uint_32 calc = component * reciprocal;
1685                calc += 16384; /* round to nearest */
1686                component = (png_uint_16)(calc >> 15);
1687             }
1688
1689             *out_ptr++ = component;
1690          }
1691          while (--c > 0);
1692
1693          /* Skip to next component (skip the intervening alpha channel) */
1694          ++in_ptr;
1695          ++out_ptr;
1696       }
1697
1698       png_write_row(png_ptr, png_voidcast(png_const_bytep, display->local_row));
1699       input_row += (png_uint_16)display->row_bytes/(sizeof (png_uint_16));
1700    }
1701
1702    return 1;
1703 }
1704
1705 /* Given 16-bit input (1 to 4 channels) write 8-bit output.  If an alpha channel
1706  * is present it must be removed from the components, the components are then
1707  * written in sRGB encoding.  No components are added or removed.
1708  *
1709  * Calculate an alpha reciprocal to reverse pre-multiplication.  As above the
1710  * calculation can be done to 15 bits of accuracy; however, the output needs to
1711  * be scaled in the range 0..255*65535, so include that scaling here.
1712  */
1713 #   define UNP_RECIPROCAL(alpha) ((((0xffff*0xff)<<7)+((alpha)>>1))/(alpha))
1714
1715 static png_byte
1716 png_unpremultiply(png_uint_32 component, png_uint_32 alpha,
1717     png_uint_32 reciprocal/*from the above macro*/)
1718 {
1719    /* The following gives 1.0 for an alpha of 0, which is fine, otherwise if 0/0
1720     * is represented as some other value there is more likely to be a
1721     * discontinuity which will probably damage compression when moving from a
1722     * fully transparent area to a nearly transparent one.  (The assumption here
1723     * is that opaque areas tend not to be 0 intensity.)
1724     *
1725     * There is a rounding problem here; if alpha is less than 128 it will end up
1726     * as 0 when scaled to 8 bits.  To avoid introducing spurious colors into the
1727     * output change for this too.
1728     */
1729    if (component >= alpha || alpha < 128)
1730       return 255;
1731
1732    /* component<alpha, so component/alpha is less than one and
1733     * component*reciprocal is less than 2^31.
1734     */
1735    else if (component > 0)
1736    {
1737       /* The test is that alpha/257 (rounded) is less than 255, the first value
1738        * that becomes 255 is 65407.
1739        * NOTE: this must agree with the PNG_DIV257 macro (which must, therefore,
1740        * be exact!)  [Could also test reciprocal != 0]
1741        */
1742       if (alpha < 65407)
1743       {
1744          component *= reciprocal;
1745          component += 64; /* round to nearest */
1746          component >>= 7;
1747       }
1748
1749       else
1750          component *= 255;
1751
1752       /* Convert the component to sRGB. */
1753       return (png_byte)PNG_sRGB_FROM_LINEAR(component);
1754    }
1755
1756    else
1757       return 0;
1758 }
1759
1760 static int
1761 png_write_image_8bit(png_voidp argument)
1762 {
1763    png_image_write_control *display = png_voidcast(png_image_write_control*,
1764        argument);
1765    png_imagep image = display->image;
1766    png_structrp png_ptr = image->opaque->png_ptr;
1767
1768    png_const_uint_16p input_row = png_voidcast(png_const_uint_16p,
1769        display->first_row);
1770    png_bytep output_row = png_voidcast(png_bytep, display->local_row);
1771    png_uint_32 y = image->height;
1772    unsigned int channels = (image->format & PNG_FORMAT_FLAG_COLOR) != 0 ?
1773        3 : 1;
1774
1775    if ((image->format & PNG_FORMAT_FLAG_ALPHA) != 0)
1776    {
1777       png_bytep row_end;
1778       int aindex;
1779
1780 #   ifdef PNG_SIMPLIFIED_WRITE_AFIRST_SUPPORTED
1781       if ((image->format & PNG_FORMAT_FLAG_AFIRST) != 0)
1782       {
1783          aindex = -1;
1784          ++input_row; /* To point to the first component */
1785          ++output_row;
1786       }
1787
1788       else
1789 #   endif
1790       aindex = (int)channels;
1791
1792       /* Use row_end in place of a loop counter: */
1793       row_end = output_row + image->width * (channels+1);
1794
1795       for (; y > 0; --y)
1796       {
1797          png_const_uint_16p in_ptr = input_row;
1798          png_bytep out_ptr = output_row;
1799
1800          while (out_ptr < row_end)
1801          {
1802             png_uint_16 alpha = in_ptr[aindex];
1803             png_byte alphabyte = (png_byte)PNG_DIV257(alpha);
1804             png_uint_32 reciprocal = 0;
1805             int c;
1806
1807             /* Scale and write the alpha channel. */
1808             out_ptr[aindex] = alphabyte;
1809
1810             if (alphabyte > 0 && alphabyte < 255)
1811                reciprocal = UNP_RECIPROCAL(alpha);
1812
1813             c = (int)channels;
1814             do /* always at least one channel */
1815                *out_ptr++ = png_unpremultiply(*in_ptr++, alpha, reciprocal);
1816             while (--c > 0);
1817
1818             /* Skip to next component (skip the intervening alpha channel) */
1819             ++in_ptr;
1820             ++out_ptr;
1821          } /* while out_ptr < row_end */
1822
1823          png_write_row(png_ptr, png_voidcast(png_const_bytep,
1824              display->local_row));
1825          input_row += (png_uint_16)display->row_bytes/(sizeof (png_uint_16));
1826       } /* while y */
1827    }
1828
1829    else
1830    {
1831       /* No alpha channel, so the row_end really is the end of the row and it
1832        * is sufficient to loop over the components one by one.
1833        */
1834       png_bytep row_end = output_row + image->width * channels;
1835
1836       for (; y > 0; --y)
1837       {
1838          png_const_uint_16p in_ptr = input_row;
1839          png_bytep out_ptr = output_row;
1840
1841          while (out_ptr < row_end)
1842          {
1843             png_uint_32 component = *in_ptr++;
1844
1845             component *= 255;
1846             *out_ptr++ = (png_byte)PNG_sRGB_FROM_LINEAR(component);
1847          }
1848
1849          png_write_row(png_ptr, output_row);
1850          input_row += (png_uint_16)display->row_bytes/(sizeof (png_uint_16));
1851       }
1852    }
1853
1854    return 1;
1855 }
1856
1857 static void
1858 png_image_set_PLTE(png_image_write_control *display)
1859 {
1860    png_imagep image = display->image;
1861    const void *cmap = display->colormap;
1862    int entries = image->colormap_entries > 256 ? 256 :
1863        (int)image->colormap_entries;
1864
1865    /* NOTE: the caller must check for cmap != NULL and entries != 0 */
1866    png_uint_32 format = image->format;
1867    unsigned int channels = PNG_IMAGE_SAMPLE_CHANNELS(format);
1868
1869 #   if defined(PNG_FORMAT_BGR_SUPPORTED) &&\
1870       defined(PNG_SIMPLIFIED_WRITE_AFIRST_SUPPORTED)
1871       int afirst = (format & PNG_FORMAT_FLAG_AFIRST) != 0 &&
1872           (format & PNG_FORMAT_FLAG_ALPHA) != 0;
1873 #   else
1874 #     define afirst 0
1875 #   endif
1876
1877 #   ifdef PNG_FORMAT_BGR_SUPPORTED
1878       int bgr = (format & PNG_FORMAT_FLAG_BGR) != 0 ? 2 : 0;
1879 #   else
1880 #     define bgr 0
1881 #   endif
1882
1883    int i, num_trans;
1884    png_color palette[256];
1885    png_byte tRNS[256];
1886
1887    memset(tRNS, 255, (sizeof tRNS));
1888    memset(palette, 0, (sizeof palette));
1889
1890    for (i=num_trans=0; i<entries; ++i)
1891    {
1892       /* This gets automatically converted to sRGB with reversal of the
1893        * pre-multiplication if the color-map has an alpha channel.
1894        */
1895       if ((format & PNG_FORMAT_FLAG_LINEAR) != 0)
1896       {
1897          png_const_uint_16p entry = png_voidcast(png_const_uint_16p, cmap);
1898
1899          entry += (unsigned int)i * channels;
1900
1901          if ((channels & 1) != 0) /* no alpha */
1902          {
1903             if (channels >= 3) /* RGB */
1904             {
1905                palette[i].blue = (png_byte)PNG_sRGB_FROM_LINEAR(255 *
1906                    entry[(2 ^ bgr)]);
1907                palette[i].green = (png_byte)PNG_sRGB_FROM_LINEAR(255 *
1908                    entry[1]);
1909                palette[i].red = (png_byte)PNG_sRGB_FROM_LINEAR(255 *
1910                    entry[bgr]);
1911             }
1912
1913             else /* Gray */
1914                palette[i].blue = palette[i].red = palette[i].green =
1915                   (png_byte)PNG_sRGB_FROM_LINEAR(255 * *entry);
1916          }
1917
1918          else /* alpha */
1919          {
1920             png_uint_16 alpha = entry[afirst ? 0 : channels-1];
1921             png_byte alphabyte = (png_byte)PNG_DIV257(alpha);
1922             png_uint_32 reciprocal = 0;
1923
1924             /* Calculate a reciprocal, as in the png_write_image_8bit code above
1925              * this is designed to produce a value scaled to 255*65535 when
1926              * divided by 128 (i.e. asr 7).
1927              */
1928             if (alphabyte > 0 && alphabyte < 255)
1929                reciprocal = (((0xffff*0xff)<<7)+(alpha>>1))/alpha;
1930
1931             tRNS[i] = alphabyte;
1932             if (alphabyte < 255)
1933                num_trans = i+1;
1934
1935             if (channels >= 3) /* RGB */
1936             {
1937                palette[i].blue = png_unpremultiply(entry[afirst + (2 ^ bgr)],
1938                    alpha, reciprocal);
1939                palette[i].green = png_unpremultiply(entry[afirst + 1], alpha,
1940                    reciprocal);
1941                palette[i].red = png_unpremultiply(entry[afirst + bgr], alpha,
1942                    reciprocal);
1943             }
1944
1945             else /* gray */
1946                palette[i].blue = palette[i].red = palette[i].green =
1947                    png_unpremultiply(entry[afirst], alpha, reciprocal);
1948          }
1949       }
1950
1951       else /* Color-map has sRGB values */
1952       {
1953          png_const_bytep entry = png_voidcast(png_const_bytep, cmap);
1954
1955          entry += (unsigned int)i * channels;
1956
1957          switch (channels)
1958          {
1959             case 4:
1960                tRNS[i] = entry[afirst ? 0 : 3];
1961                if (tRNS[i] < 255)
1962                   num_trans = i+1;
1963                /* FALLTHROUGH */
1964             case 3:
1965                palette[i].blue = entry[afirst + (2 ^ bgr)];
1966                palette[i].green = entry[afirst + 1];
1967                palette[i].red = entry[afirst + bgr];
1968                break;
1969
1970             case 2:
1971                tRNS[i] = entry[1 ^ afirst];
1972                if (tRNS[i] < 255)
1973                   num_trans = i+1;
1974                /* FALLTHROUGH */
1975             case 1:
1976                palette[i].blue = palette[i].red = palette[i].green =
1977                   entry[afirst];
1978                break;
1979
1980             default:
1981                break;
1982          }
1983       }
1984    }
1985
1986 #   ifdef afirst
1987 #     undef afirst
1988 #   endif
1989 #   ifdef bgr
1990 #     undef bgr
1991 #   endif
1992
1993    png_set_PLTE(image->opaque->png_ptr, image->opaque->info_ptr, palette,
1994        entries);
1995
1996    if (num_trans > 0)
1997       png_set_tRNS(image->opaque->png_ptr, image->opaque->info_ptr, tRNS,
1998           num_trans, NULL);
1999
2000    image->colormap_entries = (png_uint_32)entries;
2001 }
2002
2003 static int
2004 png_image_write_main(png_voidp argument)
2005 {
2006    png_image_write_control *display = png_voidcast(png_image_write_control*,
2007        argument);
2008    png_imagep image = display->image;
2009    png_structrp png_ptr = image->opaque->png_ptr;
2010    png_inforp info_ptr = image->opaque->info_ptr;
2011    png_uint_32 format = image->format;
2012
2013    /* The following four ints are actually booleans */
2014    int colormap = (format & PNG_FORMAT_FLAG_COLORMAP);
2015    int linear = !colormap && (format & PNG_FORMAT_FLAG_LINEAR); /* input */
2016    int alpha = !colormap && (format & PNG_FORMAT_FLAG_ALPHA);
2017    int write_16bit = linear && (display->convert_to_8bit == 0);
2018
2019 #   ifdef PNG_BENIGN_ERRORS_SUPPORTED
2020       /* Make sure we error out on any bad situation */
2021       png_set_benign_errors(png_ptr, 0/*error*/);
2022 #   endif
2023
2024    /* Default the 'row_stride' parameter if required, also check the row stride
2025     * and total image size to ensure that they are within the system limits.
2026     */
2027    {
2028       unsigned int channels = PNG_IMAGE_PIXEL_CHANNELS(image->format);
2029
2030       if (image->width <= 0x7fffffffU/channels) /* no overflow */
2031       {
2032          png_uint_32 check;
2033          png_uint_32 png_row_stride = image->width * channels;
2034
2035          if (display->row_stride == 0)
2036             display->row_stride = (png_int_32)/*SAFE*/png_row_stride;
2037
2038          if (display->row_stride < 0)
2039             check = (png_uint_32)(-display->row_stride);
2040
2041          else
2042             check = (png_uint_32)display->row_stride;
2043
2044          if (check >= png_row_stride)
2045          {
2046             /* Now check for overflow of the image buffer calculation; this
2047              * limits the whole image size to 32 bits for API compatibility with
2048              * the current, 32-bit, PNG_IMAGE_BUFFER_SIZE macro.
2049              */
2050             if (image->height > 0xffffffffU/png_row_stride)
2051                png_error(image->opaque->png_ptr, "memory image too large");
2052          }
2053
2054          else
2055             png_error(image->opaque->png_ptr, "supplied row stride too small");
2056       }
2057
2058       else
2059          png_error(image->opaque->png_ptr, "image row stride too large");
2060    }
2061
2062    /* Set the required transforms then write the rows in the correct order. */
2063    if ((format & PNG_FORMAT_FLAG_COLORMAP) != 0)
2064    {
2065       if (display->colormap != NULL && image->colormap_entries > 0)
2066       {
2067          png_uint_32 entries = image->colormap_entries;
2068
2069          png_set_IHDR(png_ptr, info_ptr, image->width, image->height,
2070              entries > 16 ? 8 : (entries > 4 ? 4 : (entries > 2 ? 2 : 1)),
2071              PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
2072              PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
2073
2074          png_image_set_PLTE(display);
2075       }
2076
2077       else
2078          png_error(image->opaque->png_ptr,
2079              "no color-map for color-mapped image");
2080    }
2081
2082    else
2083       png_set_IHDR(png_ptr, info_ptr, image->width, image->height,
2084           write_16bit ? 16 : 8,
2085           ((format & PNG_FORMAT_FLAG_COLOR) ? PNG_COLOR_MASK_COLOR : 0) +
2086           ((format & PNG_FORMAT_FLAG_ALPHA) ? PNG_COLOR_MASK_ALPHA : 0),
2087           PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
2088
2089    /* Counter-intuitively the data transformations must be called *after*
2090     * png_write_info, not before as in the read code, but the 'set' functions
2091     * must still be called before.  Just set the color space information, never
2092     * write an interlaced image.
2093     */
2094
2095    if (write_16bit != 0)
2096    {
2097       /* The gamma here is 1.0 (linear) and the cHRM chunk matches sRGB. */
2098       png_set_gAMA_fixed(png_ptr, info_ptr, PNG_GAMMA_LINEAR);
2099
2100       if ((image->flags & PNG_IMAGE_FLAG_COLORSPACE_NOT_sRGB) == 0)
2101          png_set_cHRM_fixed(png_ptr, info_ptr,
2102              /* color      x       y */
2103              /* white */ 31270, 32900,
2104              /* red   */ 64000, 33000,
2105              /* green */ 30000, 60000,
2106              /* blue  */ 15000,  6000
2107          );
2108    }
2109
2110    else if ((image->flags & PNG_IMAGE_FLAG_COLORSPACE_NOT_sRGB) == 0)
2111       png_set_sRGB(png_ptr, info_ptr, PNG_sRGB_INTENT_PERCEPTUAL);
2112
2113    /* Else writing an 8-bit file and the *colors* aren't sRGB, but the 8-bit
2114     * space must still be gamma encoded.
2115     */
2116    else
2117       png_set_gAMA_fixed(png_ptr, info_ptr, PNG_GAMMA_sRGB_INVERSE);
2118
2119    /* Write the file header. */
2120    png_write_info(png_ptr, info_ptr);
2121
2122    /* Now set up the data transformations (*after* the header is written),
2123     * remove the handled transformations from the 'format' flags for checking.
2124     *
2125     * First check for a little endian system if writing 16-bit files.
2126     */
2127    if (write_16bit != 0)
2128    {
2129       png_uint_16 le = 0x0001;
2130
2131       if ((*(png_const_bytep) & le) != 0)
2132          png_set_swap(png_ptr);
2133    }
2134
2135 #   ifdef PNG_SIMPLIFIED_WRITE_BGR_SUPPORTED
2136       if ((format & PNG_FORMAT_FLAG_BGR) != 0)
2137       {
2138          if (colormap == 0 && (format & PNG_FORMAT_FLAG_COLOR) != 0)
2139             png_set_bgr(png_ptr);
2140          format &= ~PNG_FORMAT_FLAG_BGR;
2141       }
2142 #   endif
2143
2144 #   ifdef PNG_SIMPLIFIED_WRITE_AFIRST_SUPPORTED
2145       if ((format & PNG_FORMAT_FLAG_AFIRST) != 0)
2146       {
2147          if (colormap == 0 && (format & PNG_FORMAT_FLAG_ALPHA) != 0)
2148             png_set_swap_alpha(png_ptr);
2149          format &= ~PNG_FORMAT_FLAG_AFIRST;
2150       }
2151 #   endif
2152
2153    /* If there are 16 or fewer color-map entries we wrote a lower bit depth
2154     * above, but the application data is still byte packed.
2155     */
2156    if (colormap != 0 && image->colormap_entries <= 16)
2157       png_set_packing(png_ptr);
2158
2159    /* That should have handled all (both) the transforms. */
2160    if ((format & ~(png_uint_32)(PNG_FORMAT_FLAG_COLOR | PNG_FORMAT_FLAG_LINEAR |
2161          PNG_FORMAT_FLAG_ALPHA | PNG_FORMAT_FLAG_COLORMAP)) != 0)
2162       png_error(png_ptr, "png_write_image: unsupported transformation");
2163
2164    {
2165       png_const_bytep row = png_voidcast(png_const_bytep, display->buffer);
2166       ptrdiff_t row_bytes = display->row_stride;
2167
2168       if (linear != 0)
2169          row_bytes *= (sizeof (png_uint_16));
2170
2171       if (row_bytes < 0)
2172          row += (image->height-1) * (-row_bytes);
2173
2174       display->first_row = row;
2175       display->row_bytes = row_bytes;
2176    }
2177
2178    /* Apply 'fast' options if the flag is set. */
2179    if ((image->flags & PNG_IMAGE_FLAG_FAST) != 0)
2180    {
2181       png_set_filter(png_ptr, PNG_FILTER_TYPE_BASE, PNG_NO_FILTERS);
2182       /* NOTE: determined by experiment using pngstest, this reflects some
2183        * balance between the time to write the image once and the time to read
2184        * it about 50 times.  The speed-up in pngstest was about 10-20% of the
2185        * total (user) time on a heavily loaded system.
2186        */
2187 #   ifdef PNG_WRITE_CUSTOMIZE_COMPRESSION_SUPPORTED
2188       png_set_compression_level(png_ptr, 3);
2189 #   endif
2190    }
2191
2192    /* Check for the cases that currently require a pre-transform on the row
2193     * before it is written.  This only applies when the input is 16-bit and
2194     * either there is an alpha channel or it is converted to 8-bit.
2195     */
2196    if ((linear != 0 && alpha != 0 ) ||
2197        (colormap == 0 && display->convert_to_8bit != 0))
2198    {
2199       png_bytep row = png_voidcast(png_bytep, png_malloc(png_ptr,
2200           png_get_rowbytes(png_ptr, info_ptr)));
2201       int result;
2202
2203       display->local_row = row;
2204       if (write_16bit != 0)
2205          result = png_safe_execute(image, png_write_image_16bit, display);
2206       else
2207          result = png_safe_execute(image, png_write_image_8bit, display);
2208       display->local_row = NULL;
2209
2210       png_free(png_ptr, row);
2211
2212       /* Skip the 'write_end' on error: */
2213       if (result == 0)
2214          return 0;
2215    }
2216
2217    /* Otherwise this is the case where the input is in a format currently
2218     * supported by the rest of the libpng write code; call it directly.
2219     */
2220    else
2221    {
2222       png_const_bytep row = png_voidcast(png_const_bytep, display->first_row);
2223       ptrdiff_t row_bytes = display->row_bytes;
2224       png_uint_32 y = image->height;
2225
2226       for (; y > 0; --y)
2227       {
2228          png_write_row(png_ptr, row);
2229          row += row_bytes;
2230       }
2231    }
2232
2233    png_write_end(png_ptr, info_ptr);
2234    return 1;
2235 }
2236
2237
2238 static void (PNGCBAPI
2239 image_memory_write)(png_structp png_ptr, png_bytep/*const*/ data, size_t size)
2240 {
2241    png_image_write_control *display = png_voidcast(png_image_write_control*,
2242        png_ptr->io_ptr/*backdoor: png_get_io_ptr(png_ptr)*/);
2243    png_alloc_size_t ob = display->output_bytes;
2244
2245    /* Check for overflow; this should never happen: */
2246    if (size <= ((png_alloc_size_t)-1) - ob)
2247    {
2248       /* I don't think libpng ever does this, but just in case: */
2249       if (size > 0)
2250       {
2251          if (display->memory_bytes >= ob+size) /* writing */
2252             memcpy(display->memory+ob, data, size);
2253
2254          /* Always update the size: */
2255          display->output_bytes = ob+size;
2256       }
2257    }
2258
2259    else
2260       png_error(png_ptr, "png_image_write_to_memory: PNG too big");
2261 }
2262
2263 static void (PNGCBAPI
2264 image_memory_flush)(png_structp png_ptr)
2265 {
2266    PNG_UNUSED(png_ptr)
2267 }
2268
2269 static int
2270 png_image_write_memory(png_voidp argument)
2271 {
2272    png_image_write_control *display = png_voidcast(png_image_write_control*,
2273        argument);
2274
2275    /* The rest of the memory-specific init and write_main in an error protected
2276     * environment.  This case needs to use callbacks for the write operations
2277     * since libpng has no built in support for writing to memory.
2278     */
2279    png_set_write_fn(display->image->opaque->png_ptr, display/*io_ptr*/,
2280        image_memory_write, image_memory_flush);
2281
2282    return png_image_write_main(display);
2283 }
2284
2285 int PNGAPI
2286 png_image_write_to_memory(png_imagep image, void *memory,
2287     png_alloc_size_t * PNG_RESTRICT memory_bytes, int convert_to_8bit,
2288     const void *buffer, png_int_32 row_stride, const void *colormap)
2289 {
2290    /* Write the image to the given buffer, or count the bytes if it is NULL */
2291    if (image != NULL && image->version == PNG_IMAGE_VERSION)
2292    {
2293       if (memory_bytes != NULL && buffer != NULL)
2294       {
2295          /* This is to give the caller an easier error detection in the NULL
2296           * case and guard against uninitialized variable problems:
2297           */
2298          if (memory == NULL)
2299             *memory_bytes = 0;
2300
2301          if (png_image_write_init(image) != 0)
2302          {
2303             png_image_write_control display;
2304             int result;
2305
2306             memset(&display, 0, (sizeof display));
2307             display.image = image;
2308             display.buffer = buffer;
2309             display.row_stride = row_stride;
2310             display.colormap = colormap;
2311             display.convert_to_8bit = convert_to_8bit;
2312             display.memory = png_voidcast(png_bytep, memory);
2313             display.memory_bytes = *memory_bytes;
2314             display.output_bytes = 0;
2315
2316             result = png_safe_execute(image, png_image_write_memory, &display);
2317             png_image_free(image);
2318
2319             /* write_memory returns true even if we ran out of buffer. */
2320             if (result)
2321             {
2322                /* On out-of-buffer this function returns '0' but still updates
2323                 * memory_bytes:
2324                 */
2325                if (memory != NULL && display.output_bytes > *memory_bytes)
2326                   result = 0;
2327
2328                *memory_bytes = display.output_bytes;
2329             }
2330
2331             return result;
2332          }
2333
2334          else
2335             return 0;
2336       }
2337
2338       else
2339          return png_image_error(image,
2340              "png_image_write_to_memory: invalid argument");
2341    }
2342
2343    else if (image != NULL)
2344       return png_image_error(image,
2345           "png_image_write_to_memory: incorrect PNG_IMAGE_VERSION");
2346
2347    else
2348       return 0;
2349 }
2350
2351 #ifdef PNG_SIMPLIFIED_WRITE_STDIO_SUPPORTED
2352 int PNGAPI
2353 png_image_write_to_stdio(png_imagep image, FILE *file, int convert_to_8bit,
2354     const void *buffer, png_int_32 row_stride, const void *colormap)
2355 {
2356    /* Write the image to the given (FILE*). */
2357    if (image != NULL && image->version == PNG_IMAGE_VERSION)
2358    {
2359       if (file != NULL && buffer != NULL)
2360       {
2361          if (png_image_write_init(image) != 0)
2362          {
2363             png_image_write_control display;
2364             int result;
2365
2366             /* This is slightly evil, but png_init_io doesn't do anything other
2367              * than this and we haven't changed the standard IO functions so
2368              * this saves a 'safe' function.
2369              */
2370             image->opaque->png_ptr->io_ptr = file;
2371
2372             memset(&display, 0, (sizeof display));
2373             display.image = image;
2374             display.buffer = buffer;
2375             display.row_stride = row_stride;
2376             display.colormap = colormap;
2377             display.convert_to_8bit = convert_to_8bit;
2378
2379             result = png_safe_execute(image, png_image_write_main, &display);
2380             png_image_free(image);
2381             return result;
2382          }
2383
2384          else
2385             return 0;
2386       }
2387
2388       else
2389          return png_image_error(image,
2390              "png_image_write_to_stdio: invalid argument");
2391    }
2392
2393    else if (image != NULL)
2394       return png_image_error(image,
2395           "png_image_write_to_stdio: incorrect PNG_IMAGE_VERSION");
2396
2397    else
2398       return 0;
2399 }
2400
2401 int PNGAPI
2402 png_image_write_to_file(png_imagep image, const char *file_name,
2403     int convert_to_8bit, const void *buffer, png_int_32 row_stride,
2404     const void *colormap)
2405 {
2406    /* Write the image to the named file. */
2407    if (image != NULL && image->version == PNG_IMAGE_VERSION)
2408    {
2409       if (file_name != NULL && buffer != NULL)
2410       {
2411          FILE *fp = fopen(file_name, "wb");
2412
2413          if (fp != NULL)
2414          {
2415             if (png_image_write_to_stdio(image, fp, convert_to_8bit, buffer,
2416                 row_stride, colormap) != 0)
2417             {
2418                int error; /* from fflush/fclose */
2419
2420                /* Make sure the file is flushed correctly. */
2421                if (fflush(fp) == 0 && ferror(fp) == 0)
2422                {
2423                   if (fclose(fp) == 0)
2424                      return 1;
2425
2426                   error = errno; /* from fclose */
2427                }
2428
2429                else
2430                {
2431                   error = errno; /* from fflush or ferror */
2432                   (void)fclose(fp);
2433                }
2434
2435                (void)remove(file_name);
2436                /* The image has already been cleaned up; this is just used to
2437                 * set the error (because the original write succeeded).
2438                 */
2439                return png_image_error(image, strerror(error));
2440             }
2441
2442             else
2443             {
2444                /* Clean up: just the opened file. */
2445                (void)fclose(fp);
2446                (void)remove(file_name);
2447                return 0;
2448             }
2449          }
2450
2451          else
2452             return png_image_error(image, strerror(errno));
2453       }
2454
2455       else
2456          return png_image_error(image,
2457              "png_image_write_to_file: invalid argument");
2458    }
2459
2460    else if (image != NULL)
2461       return png_image_error(image,
2462           "png_image_write_to_file: incorrect PNG_IMAGE_VERSION");
2463
2464    else
2465       return 0;
2466 }
2467 #endif /* SIMPLIFIED_WRITE_STDIO */
2468 #endif /* SIMPLIFIED_WRITE */
2469 #endif /* WRITE */