pngset.c 50.3 KB
Newer Older
1 2 3

/* pngset.c - storage of image information into info struct
 *
4 5
 * Last changed in libpng 1.6.32 [August 24, 2017]
 * Copyright (c) 1998-2017 Glenn Randers-Pehrson
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
 *
 * This code is released under the libpng license.
 * For conditions of distribution and use, see the disclaimer
 * and license in png.h
 *
 * The functions here are used during reads to store data from the file
 * into the info struct, and during writes to store application data
 * into the info struct for writing into the file.  This abstracts the
 * info struct and allows us to change the structure in the future.
 */

#include "pngpriv.h"

#if defined(PNG_READ_SUPPORTED) || defined(PNG_WRITE_SUPPORTED)

#ifdef PNG_bKGD_SUPPORTED
void PNGAPI
25
png_set_bKGD(png_const_structrp png_ptr, png_inforp info_ptr,
26 27 28 29
    png_const_color_16p background)
{
   png_debug1(1, "in %s storage function", "bKGD");

30
   if (png_ptr == NULL || info_ptr == NULL || background == NULL)
31 32
      return;

33
   info_ptr->background = *background;
34 35 36 37 38 39
   info_ptr->valid |= PNG_INFO_bKGD;
}
#endif

#ifdef PNG_cHRM_SUPPORTED
void PNGFAPI
40
png_set_cHRM_fixed(png_const_structrp png_ptr, png_inforp info_ptr,
41 42 43 44
    png_fixed_point white_x, png_fixed_point white_y, png_fixed_point red_x,
    png_fixed_point red_y, png_fixed_point green_x, png_fixed_point green_y,
    png_fixed_point blue_x, png_fixed_point blue_y)
{
45 46
   png_xy xy;

47 48 49 50 51
   png_debug1(1, "in %s storage function", "cHRM fixed");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

52 53 54 55 56 57 58 59 60 61 62 63 64 65
   xy.redx = red_x;
   xy.redy = red_y;
   xy.greenx = green_x;
   xy.greeny = green_y;
   xy.bluex = blue_x;
   xy.bluey = blue_y;
   xy.whitex = white_x;
   xy.whitey = white_y;

   if (png_colorspace_set_chromaticities(png_ptr, &info_ptr->colorspace, &xy,
       2/* override with app values*/) != 0)
      info_ptr->colorspace.flags |= PNG_COLORSPACE_FROM_cHRM;

   png_colorspace_sync_info(png_ptr, info_ptr);
66 67 68
}

void PNGFAPI
69
png_set_cHRM_XYZ_fixed(png_const_structrp png_ptr, png_inforp info_ptr,
70 71 72 73 74 75 76 77 78 79 80 81 82
    png_fixed_point int_red_X, png_fixed_point int_red_Y,
    png_fixed_point int_red_Z, png_fixed_point int_green_X,
    png_fixed_point int_green_Y, png_fixed_point int_green_Z,
    png_fixed_point int_blue_X, png_fixed_point int_blue_Y,
    png_fixed_point int_blue_Z)
{
   png_XYZ XYZ;

   png_debug1(1, "in %s storage function", "cHRM XYZ fixed");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
   XYZ.red_X = int_red_X;
   XYZ.red_Y = int_red_Y;
   XYZ.red_Z = int_red_Z;
   XYZ.green_X = int_green_X;
   XYZ.green_Y = int_green_Y;
   XYZ.green_Z = int_green_Z;
   XYZ.blue_X = int_blue_X;
   XYZ.blue_Y = int_blue_Y;
   XYZ.blue_Z = int_blue_Z;

   if (png_colorspace_set_endpoints(png_ptr, &info_ptr->colorspace,
       &XYZ, 2) != 0)
      info_ptr->colorspace.flags |= PNG_COLORSPACE_FROM_cHRM;

   png_colorspace_sync_info(png_ptr, info_ptr);
98 99 100 101
}

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
102
png_set_cHRM(png_const_structrp png_ptr, png_inforp info_ptr,
103 104 105 106
    double white_x, double white_y, double red_x, double red_y,
    double green_x, double green_y, double blue_x, double blue_y)
{
   png_set_cHRM_fixed(png_ptr, info_ptr,
kurenai's avatar
kurenai committed
107 108 109 110 111 112 113 114
       png_fixed(png_ptr, white_x, "cHRM White X"),
       png_fixed(png_ptr, white_y, "cHRM White Y"),
       png_fixed(png_ptr, red_x, "cHRM Red X"),
       png_fixed(png_ptr, red_y, "cHRM Red Y"),
       png_fixed(png_ptr, green_x, "cHRM Green X"),
       png_fixed(png_ptr, green_y, "cHRM Green Y"),
       png_fixed(png_ptr, blue_x, "cHRM Blue X"),
       png_fixed(png_ptr, blue_y, "cHRM Blue Y"));
115 116 117
}

void PNGAPI
118
png_set_cHRM_XYZ(png_const_structrp png_ptr, png_inforp info_ptr, double red_X,
119 120 121 122
    double red_Y, double red_Z, double green_X, double green_Y, double green_Z,
    double blue_X, double blue_Y, double blue_Z)
{
   png_set_cHRM_XYZ_fixed(png_ptr, info_ptr,
kurenai's avatar
kurenai committed
123 124 125 126 127 128 129 130 131
       png_fixed(png_ptr, red_X, "cHRM Red X"),
       png_fixed(png_ptr, red_Y, "cHRM Red Y"),
       png_fixed(png_ptr, red_Z, "cHRM Red Z"),
       png_fixed(png_ptr, green_X, "cHRM Green X"),
       png_fixed(png_ptr, green_Y, "cHRM Green Y"),
       png_fixed(png_ptr, green_Z, "cHRM Green Z"),
       png_fixed(png_ptr, blue_X, "cHRM Blue X"),
       png_fixed(png_ptr, blue_Y, "cHRM Blue Y"),
       png_fixed(png_ptr, blue_Z, "cHRM Blue Z"));
132
}
133
#  endif /* FLOATING_POINT */
134

135
#endif /* cHRM */
136

137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
#ifdef PNG_eXIf_SUPPORTED
void PNGAPI
png_set_eXIf(png_const_structrp png_ptr, png_inforp info_ptr,
    const png_bytep eXIf_buf)
{
  png_warning(png_ptr, "png_set_eXIf does not work; use png_set_eXIf_1");
  PNG_UNUSED(info_ptr)
  PNG_UNUSED(eXIf_buf)
}

void PNGAPI
png_set_eXIf_1(png_const_structrp png_ptr, png_inforp info_ptr,
    const png_uint_32 num_exif, const png_bytep eXIf_buf)
{
   int i;

   png_debug1(1, "in %s storage function", "eXIf");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   if (info_ptr->exif)
   {
      png_free(png_ptr, info_ptr->exif);
      info_ptr->exif = NULL;
   }

   info_ptr->num_exif = num_exif;

   info_ptr->exif = png_voidcast(png_bytep, png_malloc_warn(png_ptr,
       info_ptr->num_exif));

   if (info_ptr->exif == NULL)
   {
      png_warning(png_ptr, "Insufficient memory for eXIf chunk data");
      return;
   }

   info_ptr->free_me |= PNG_FREE_EXIF;

   for (i = 0; i < (int) info_ptr->num_exif; i++)
      info_ptr->exif[i] = eXIf_buf[i];

   info_ptr->valid |= PNG_INFO_eXIf;
}
#endif /* eXIf */

184 185
#ifdef PNG_gAMA_SUPPORTED
void PNGFAPI
186 187
png_set_gAMA_fixed(png_const_structrp png_ptr, png_inforp info_ptr,
    png_fixed_point file_gamma)
188 189 190 191 192 193
{
   png_debug1(1, "in %s storage function", "gAMA");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

194 195
   png_colorspace_set_gamma(png_ptr, &info_ptr->colorspace, file_gamma);
   png_colorspace_sync_info(png_ptr, info_ptr);
196 197 198 199
}

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
200
png_set_gAMA(png_const_structrp png_ptr, png_inforp info_ptr, double file_gamma)
201 202 203 204 205 206 207 208 209
{
   png_set_gAMA_fixed(png_ptr, info_ptr, png_fixed(png_ptr, file_gamma,
       "png_set_gAMA"));
}
#  endif
#endif

#ifdef PNG_hIST_SUPPORTED
void PNGAPI
210 211
png_set_hIST(png_const_structrp png_ptr, png_inforp info_ptr,
    png_const_uint_16p hist)
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
{
   int i;

   png_debug1(1, "in %s storage function", "hIST");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   if (info_ptr->num_palette == 0 || info_ptr->num_palette
       > PNG_MAX_PALETTE_LENGTH)
   {
      png_warning(png_ptr,
          "Invalid palette size, hIST allocation skipped");

      return;
   }

   png_free_data(png_ptr, info_ptr, PNG_FREE_HIST, 0);

   /* Changed from info->num_palette to PNG_MAX_PALETTE_LENGTH in
    * version 1.2.1
    */
234 235
   info_ptr->hist = png_voidcast(png_uint_16p, png_malloc_warn(png_ptr,
       PNG_MAX_PALETTE_LENGTH * (sizeof (png_uint_16))));
236

237
   if (info_ptr->hist == NULL)
238 239
   {
      png_warning(png_ptr, "Insufficient memory for hIST chunk data");
240

241 242 243
      return;
   }

244 245
   info_ptr->free_me |= PNG_FREE_HIST;

246
   for (i = 0; i < info_ptr->num_palette; i++)
247
      info_ptr->hist[i] = hist[i];
248 249 250 251 252 253

   info_ptr->valid |= PNG_INFO_hIST;
}
#endif

void PNGAPI
254
png_set_IHDR(png_const_structrp png_ptr, png_inforp info_ptr,
255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
    png_uint_32 width, png_uint_32 height, int bit_depth,
    int color_type, int interlace_type, int compression_type,
    int filter_type)
{
   png_debug1(1, "in %s storage function", "IHDR");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   info_ptr->width = width;
   info_ptr->height = height;
   info_ptr->bit_depth = (png_byte)bit_depth;
   info_ptr->color_type = (png_byte)color_type;
   info_ptr->compression_type = (png_byte)compression_type;
   info_ptr->filter_type = (png_byte)filter_type;
   info_ptr->interlace_type = (png_byte)interlace_type;

   png_check_IHDR (png_ptr, info_ptr->width, info_ptr->height,
       info_ptr->bit_depth, info_ptr->color_type, info_ptr->interlace_type,
       info_ptr->compression_type, info_ptr->filter_type);

   if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
      info_ptr->channels = 1;

279
   else if ((info_ptr->color_type & PNG_COLOR_MASK_COLOR) != 0)
280 281 282 283 284
      info_ptr->channels = 3;

   else
      info_ptr->channels = 1;

285
   if ((info_ptr->color_type & PNG_COLOR_MASK_ALPHA) != 0)
286 287 288 289
      info_ptr->channels++;

   info_ptr->pixel_depth = (png_byte)(info_ptr->channels * info_ptr->bit_depth);

290
   info_ptr->rowbytes = PNG_ROWBYTES(info_ptr->pixel_depth, width);
291 292 293 294
}

#ifdef PNG_oFFs_SUPPORTED
void PNGAPI
295
png_set_oFFs(png_const_structrp png_ptr, png_inforp info_ptr,
296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
    png_int_32 offset_x, png_int_32 offset_y, int unit_type)
{
   png_debug1(1, "in %s storage function", "oFFs");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   info_ptr->x_offset = offset_x;
   info_ptr->y_offset = offset_y;
   info_ptr->offset_unit_type = (png_byte)unit_type;
   info_ptr->valid |= PNG_INFO_oFFs;
}
#endif

#ifdef PNG_pCAL_SUPPORTED
void PNGAPI
312
png_set_pCAL(png_const_structrp png_ptr, png_inforp info_ptr,
313 314 315 316 317 318 319 320
    png_const_charp purpose, png_int_32 X0, png_int_32 X1, int type,
    int nparams, png_const_charp units, png_charpp params)
{
   png_size_t length;
   int i;

   png_debug1(1, "in %s storage function", "pCAL");

321 322
   if (png_ptr == NULL || info_ptr == NULL || purpose == NULL || units == NULL
       || (nparams > 0 && params == NULL))
323 324
      return;

325
   length = strlen(purpose) + 1;
326 327 328 329 330 331 332
   png_debug1(3, "allocating purpose for info (%lu bytes)",
       (unsigned long)length);

   /* TODO: validate format of calibration name and unit name */

   /* Check that the type matches the specification. */
   if (type < 0 || type > 3)
333 334 335 336 337
   {
      png_chunk_report(png_ptr, "Invalid pCAL equation type",
            PNG_CHUNK_WRITE_ERROR);
      return;
   }
338

339
   if (nparams < 0 || nparams > 255)
340 341 342 343 344
   {
      png_chunk_report(png_ptr, "Invalid pCAL parameter count",
            PNG_CHUNK_WRITE_ERROR);
      return;
   }
345

346 347
   /* Validate params[nparams] */
   for (i=0; i<nparams; ++i)
348 349 350
   {
      if (params[i] == NULL ||
          !png_check_fp_string(params[i], strlen(params[i])))
351 352 353 354 355
      {
         png_chunk_report(png_ptr, "Invalid format for pCAL parameter",
               PNG_CHUNK_WRITE_ERROR);
         return;
      }
356
   }
357

358 359
   info_ptr->pcal_purpose = png_voidcast(png_charp,
       png_malloc_warn(png_ptr, length));
360 361 362

   if (info_ptr->pcal_purpose == NULL)
   {
363 364
      png_chunk_report(png_ptr, "Insufficient memory for pCAL purpose",
            PNG_CHUNK_WRITE_ERROR);
365 366 367
      return;
   }

368
   memcpy(info_ptr->pcal_purpose, purpose, length);
369 370 371 372 373 374 375

   png_debug(3, "storing X0, X1, type, and nparams in info");
   info_ptr->pcal_X0 = X0;
   info_ptr->pcal_X1 = X1;
   info_ptr->pcal_type = (png_byte)type;
   info_ptr->pcal_nparams = (png_byte)nparams;

376
   length = strlen(units) + 1;
377
   png_debug1(3, "allocating units for info (%lu bytes)",
kurenai's avatar
kurenai committed
378
       (unsigned long)length);
379

380
   info_ptr->pcal_units = png_voidcast(png_charp,
kurenai's avatar
kurenai committed
381
       png_malloc_warn(png_ptr, length));
382 383 384 385

   if (info_ptr->pcal_units == NULL)
   {
      png_warning(png_ptr, "Insufficient memory for pCAL units");
386

387 388 389
      return;
   }

390
   memcpy(info_ptr->pcal_units, units, length);
391

392
   info_ptr->pcal_params = png_voidcast(png_charpp, png_malloc_warn(png_ptr,
393
       (png_size_t)(((unsigned int)nparams + 1) * (sizeof (png_charp)))));
394 395 396 397

   if (info_ptr->pcal_params == NULL)
   {
      png_warning(png_ptr, "Insufficient memory for pCAL params");
398

399 400 401
      return;
   }

402 403
   memset(info_ptr->pcal_params, 0, ((unsigned int)nparams + 1) *
       (sizeof (png_charp)));
404 405 406

   for (i = 0; i < nparams; i++)
   {
407
      length = strlen(params[i]) + 1;
408 409 410 411 412 413 414 415
      png_debug2(3, "allocating parameter %d for info (%lu bytes)", i,
          (unsigned long)length);

      info_ptr->pcal_params[i] = (png_charp)png_malloc_warn(png_ptr, length);

      if (info_ptr->pcal_params[i] == NULL)
      {
         png_warning(png_ptr, "Insufficient memory for pCAL parameter");
416

417 418 419
         return;
      }

420
      memcpy(info_ptr->pcal_params[i], params[i], length);
421 422 423 424 425 426 427 428 429
   }

   info_ptr->valid |= PNG_INFO_pCAL;
   info_ptr->free_me |= PNG_FREE_PCAL;
}
#endif

#ifdef PNG_sCAL_SUPPORTED
void PNGAPI
430
png_set_sCAL_s(png_const_structrp png_ptr, png_inforp info_ptr,
431 432 433 434 435 436 437 438 439 440 441 442 443 444 445
    int unit, png_const_charp swidth, png_const_charp sheight)
{
   png_size_t lengthw = 0, lengthh = 0;

   png_debug1(1, "in %s storage function", "sCAL");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   /* Double check the unit (should never get here with an invalid
    * unit unless this is an API call.)
    */
   if (unit != 1 && unit != 2)
      png_error(png_ptr, "Invalid sCAL unit");

446
   if (swidth == NULL || (lengthw = strlen(swidth)) == 0 ||
447 448 449
       swidth[0] == 45 /* '-' */ || !png_check_fp_string(swidth, lengthw))
      png_error(png_ptr, "Invalid sCAL width");

450
   if (sheight == NULL || (lengthh = strlen(sheight)) == 0 ||
451 452 453 454 455 456 457 458 459
       sheight[0] == 45 /* '-' */ || !png_check_fp_string(sheight, lengthh))
      png_error(png_ptr, "Invalid sCAL height");

   info_ptr->scal_unit = (png_byte)unit;

   ++lengthw;

   png_debug1(3, "allocating unit for info (%u bytes)", (unsigned int)lengthw);

460
   info_ptr->scal_s_width = png_voidcast(png_charp,
kurenai's avatar
kurenai committed
461
       png_malloc_warn(png_ptr, lengthw));
462 463 464 465

   if (info_ptr->scal_s_width == NULL)
   {
      png_warning(png_ptr, "Memory allocation failed while processing sCAL");
466

467 468 469
      return;
   }

470
   memcpy(info_ptr->scal_s_width, swidth, lengthw);
471 472 473 474 475

   ++lengthh;

   png_debug1(3, "allocating unit for info (%u bytes)", (unsigned int)lengthh);

476
   info_ptr->scal_s_height = png_voidcast(png_charp,
kurenai's avatar
kurenai committed
477
       png_malloc_warn(png_ptr, lengthh));
478 479 480 481 482 483 484

   if (info_ptr->scal_s_height == NULL)
   {
      png_free (png_ptr, info_ptr->scal_s_width);
      info_ptr->scal_s_width = NULL;

      png_warning(png_ptr, "Memory allocation failed while processing sCAL");
485

486 487 488
      return;
   }

489
   memcpy(info_ptr->scal_s_height, sheight, lengthh);
490 491 492 493 494 495 496

   info_ptr->valid |= PNG_INFO_sCAL;
   info_ptr->free_me |= PNG_FREE_SCAL;
}

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
497 498
png_set_sCAL(png_const_structrp png_ptr, png_inforp info_ptr, int unit,
    double width, double height)
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
{
   png_debug1(1, "in %s storage function", "sCAL");

   /* Check the arguments. */
   if (width <= 0)
      png_warning(png_ptr, "Invalid sCAL width ignored");

   else if (height <= 0)
      png_warning(png_ptr, "Invalid sCAL height ignored");

   else
   {
      /* Convert 'width' and 'height' to ASCII. */
      char swidth[PNG_sCAL_MAX_DIGITS+1];
      char sheight[PNG_sCAL_MAX_DIGITS+1];

515
      png_ascii_from_fp(png_ptr, swidth, (sizeof swidth), width,
kurenai's avatar
kurenai committed
516
          PNG_sCAL_PRECISION);
517
      png_ascii_from_fp(png_ptr, sheight, (sizeof sheight), height,
kurenai's avatar
kurenai committed
518
          PNG_sCAL_PRECISION);
519 520 521 522 523 524 525 526

      png_set_sCAL_s(png_ptr, info_ptr, unit, swidth, sheight);
   }
}
#  endif

#  ifdef PNG_FIXED_POINT_SUPPORTED
void PNGAPI
527
png_set_sCAL_fixed(png_const_structrp png_ptr, png_inforp info_ptr, int unit,
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544
    png_fixed_point width, png_fixed_point height)
{
   png_debug1(1, "in %s storage function", "sCAL");

   /* Check the arguments. */
   if (width <= 0)
      png_warning(png_ptr, "Invalid sCAL width ignored");

   else if (height <= 0)
      png_warning(png_ptr, "Invalid sCAL height ignored");

   else
   {
      /* Convert 'width' and 'height' to ASCII. */
      char swidth[PNG_sCAL_MAX_DIGITS+1];
      char sheight[PNG_sCAL_MAX_DIGITS+1];

545 546
      png_ascii_from_fixed(png_ptr, swidth, (sizeof swidth), width);
      png_ascii_from_fixed(png_ptr, sheight, (sizeof sheight), height);
547 548 549 550 551 552 553 554 555

      png_set_sCAL_s(png_ptr, info_ptr, unit, swidth, sheight);
   }
}
#  endif
#endif

#ifdef PNG_pHYs_SUPPORTED
void PNGAPI
556
png_set_pHYs(png_const_structrp png_ptr, png_inforp info_ptr,
557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
    png_uint_32 res_x, png_uint_32 res_y, int unit_type)
{
   png_debug1(1, "in %s storage function", "pHYs");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

   info_ptr->x_pixels_per_unit = res_x;
   info_ptr->y_pixels_per_unit = res_y;
   info_ptr->phys_unit_type = (png_byte)unit_type;
   info_ptr->valid |= PNG_INFO_pHYs;
}
#endif

void PNGAPI
572
png_set_PLTE(png_structrp png_ptr, png_inforp info_ptr,
573 574 575
    png_const_colorp palette, int num_palette)
{

576 577
   png_uint_32 max_palette_length;

578 579 580 581 582
   png_debug1(1, "in %s storage function", "PLTE");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

kurenai's avatar
kurenai committed
583 584
   max_palette_length = (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE) ?
      (1 << info_ptr->bit_depth) : PNG_MAX_PALETTE_LENGTH;
585 586

   if (num_palette < 0 || num_palette > (int) max_palette_length)
587 588 589 590 591 592 593
   {
      if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
         png_error(png_ptr, "Invalid palette length");

      else
      {
         png_warning(png_ptr, "Invalid palette length");
594

595 596 597 598
         return;
      }
   }

599 600 601 602 603 604 605 606 607 608
   if ((num_palette > 0 && palette == NULL) ||
      (num_palette == 0
#        ifdef PNG_MNG_FEATURES_SUPPORTED
            && (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0
#        endif
      ))
   {
      png_error(png_ptr, "Invalid palette");
   }

609 610 611
   /* It may not actually be necessary to set png_ptr->palette here;
    * we do it for backward compatibility with the way the png_handle_tRNS
    * function used to do the allocation.
612 613 614
    *
    * 1.6.0: the above statement appears to be incorrect; something has to set
    * the palette inside png_struct on read.
615 616 617 618
    */
   png_free_data(png_ptr, info_ptr, PNG_FREE_PLTE, 0);

   /* Changed in libpng-1.2.1 to allocate PNG_MAX_PALETTE_LENGTH instead
619 620
    * of num_palette entries, in case of an invalid PNG file or incorrect
    * call to png_set_PLTE() with too-large sample values.
621
    */
622 623
   png_ptr->palette = png_voidcast(png_colorp, png_calloc(png_ptr,
       PNG_MAX_PALETTE_LENGTH * (sizeof (png_color))));
624

625
   if (num_palette > 0)
626 627
      memcpy(png_ptr->palette, palette, (unsigned int)num_palette *
          (sizeof (png_color)));
628 629 630 631 632 633 634 635 636 637
   info_ptr->palette = png_ptr->palette;
   info_ptr->num_palette = png_ptr->num_palette = (png_uint_16)num_palette;

   info_ptr->free_me |= PNG_FREE_PLTE;

   info_ptr->valid |= PNG_INFO_PLTE;
}

#ifdef PNG_sBIT_SUPPORTED
void PNGAPI
638
png_set_sBIT(png_const_structrp png_ptr, png_inforp info_ptr,
639 640 641 642
    png_const_color_8p sig_bit)
{
   png_debug1(1, "in %s storage function", "sBIT");

643
   if (png_ptr == NULL || info_ptr == NULL || sig_bit == NULL)
644 645
      return;

646
   info_ptr->sig_bit = *sig_bit;
647 648 649 650 651 652
   info_ptr->valid |= PNG_INFO_sBIT;
}
#endif

#ifdef PNG_sRGB_SUPPORTED
void PNGAPI
653
png_set_sRGB(png_const_structrp png_ptr, png_inforp info_ptr, int srgb_intent)
654 655 656 657 658 659
{
   png_debug1(1, "in %s storage function", "sRGB");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

660 661
   (void)png_colorspace_set_sRGB(png_ptr, &info_ptr->colorspace, srgb_intent);
   png_colorspace_sync_info(png_ptr, info_ptr);
662 663 664
}

void PNGAPI
665
png_set_sRGB_gAMA_and_cHRM(png_const_structrp png_ptr, png_inforp info_ptr,
666 667 668 669 670 671 672
    int srgb_intent)
{
   png_debug1(1, "in %s storage function", "sRGB_gAMA_and_cHRM");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

673 674 675 676 677 678 679
   if (png_colorspace_set_sRGB(png_ptr, &info_ptr->colorspace,
       srgb_intent) != 0)
   {
      /* This causes the gAMA and cHRM to be written too */
      info_ptr->colorspace.flags |=
         PNG_COLORSPACE_FROM_gAMA|PNG_COLORSPACE_FROM_cHRM;
   }
680

681
   png_colorspace_sync_info(png_ptr, info_ptr);
682 683 684 685 686 687
}
#endif /* sRGB */


#ifdef PNG_iCCP_SUPPORTED
void PNGAPI
688
png_set_iCCP(png_const_structrp png_ptr, png_inforp info_ptr,
689 690 691 692 693 694 695 696 697 698 699 700
    png_const_charp name, int compression_type,
    png_const_bytep profile, png_uint_32 proflen)
{
   png_charp new_iccp_name;
   png_bytep new_iccp_profile;
   png_size_t length;

   png_debug1(1, "in %s storage function", "iCCP");

   if (png_ptr == NULL || info_ptr == NULL || name == NULL || profile == NULL)
      return;

701 702 703 704 705 706 707 708 709 710 711
   if (compression_type != PNG_COMPRESSION_TYPE_BASE)
      png_app_error(png_ptr, "Invalid iCCP compression method");

   /* Set the colorspace first because this validates the profile; do not
    * override previously set app cHRM or gAMA here (because likely as not the
    * application knows better than libpng what the correct values are.)  Pass
    * the info_ptr color_type field to png_colorspace_set_ICC because in the
    * write case it has not yet been stored in png_ptr.
    */
   {
      int result = png_colorspace_set_ICC(png_ptr, &info_ptr->colorspace, name,
kurenai's avatar
kurenai committed
712
          proflen, profile, info_ptr->color_type);
713 714 715 716 717 718 719 720 721 722 723 724 725 726

      png_colorspace_sync_info(png_ptr, info_ptr);

      /* Don't do any of the copying if the profile was bad, or inconsistent. */
      if (result == 0)
         return;

      /* But do write the gAMA and cHRM chunks from the profile. */
      info_ptr->colorspace.flags |=
         PNG_COLORSPACE_FROM_gAMA|PNG_COLORSPACE_FROM_cHRM;
   }

   length = strlen(name)+1;
   new_iccp_name = png_voidcast(png_charp, png_malloc_warn(png_ptr, length));
727 728 729

   if (new_iccp_name == NULL)
   {
730 731
      png_benign_error(png_ptr, "Insufficient memory to process iCCP chunk");

732 733 734
      return;
   }

735 736
   memcpy(new_iccp_name, name, length);
   new_iccp_profile = png_voidcast(png_bytep,
kurenai's avatar
kurenai committed
737
       png_malloc_warn(png_ptr, proflen));
738 739 740

   if (new_iccp_profile == NULL)
   {
741 742
      png_free(png_ptr, new_iccp_name);
      png_benign_error(png_ptr,
743
          "Insufficient memory to process iCCP profile");
744

745 746 747
      return;
   }

748
   memcpy(new_iccp_profile, profile, proflen);
749 750 751 752 753 754 755 756 757 758 759 760 761

   png_free_data(png_ptr, info_ptr, PNG_FREE_ICCP, 0);

   info_ptr->iccp_proflen = proflen;
   info_ptr->iccp_name = new_iccp_name;
   info_ptr->iccp_profile = new_iccp_profile;
   info_ptr->free_me |= PNG_FREE_ICCP;
   info_ptr->valid |= PNG_INFO_iCCP;
}
#endif

#ifdef PNG_TEXT_SUPPORTED
void PNGAPI
762 763
png_set_text(png_const_structrp png_ptr, png_inforp info_ptr,
    png_const_textp text_ptr, int num_text)
764 765 766 767
{
   int ret;
   ret = png_set_text_2(png_ptr, info_ptr, text_ptr, num_text);

768
   if (ret != 0)
769 770 771 772
      png_error(png_ptr, "Insufficient memory to store text");
}

int /* PRIVATE */
773
png_set_text_2(png_const_structrp png_ptr, png_inforp info_ptr,
774 775 776 777
    png_const_textp text_ptr, int num_text)
{
   int i;

778
   png_debug1(1, "in %lx storage function", png_ptr == NULL ? 0xabadca11U :
779 780
      (unsigned long)png_ptr->chunk_name);

781
   if (png_ptr == NULL || info_ptr == NULL || num_text <= 0 || text_ptr == NULL)
782 783 784
      return(0);

   /* Make sure we have enough space in the "text" array in info_struct
785 786 787
    * to hold all of the incoming text_ptr objects.  This compare can't overflow
    * because max_text >= num_text (anyway, subtract of two positive integers
    * can't overflow in any case.)
788
    */
789
   if (num_text > info_ptr->max_text - info_ptr->num_text)
790
   {
791
      int old_num_text = info_ptr->num_text;
792 793
      int max_text;
      png_textp new_text = NULL;
794

795 796 797
      /* Calculate an appropriate max_text, checking for overflow. */
      max_text = old_num_text;
      if (num_text <= INT_MAX - max_text)
798
      {
799
         max_text += num_text;
800

801 802 803
         /* Round up to a multiple of 8 */
         if (max_text < INT_MAX-8)
            max_text = (max_text + 8) & ~0x7;
804

805 806 807 808 809 810 811
         else
            max_text = INT_MAX;

         /* Now allocate a new array and copy the old members in; this does all
          * the overflow checks.
          */
         new_text = png_voidcast(png_textp,png_realloc_array(png_ptr,
kurenai's avatar
kurenai committed
812 813
             info_ptr->text, old_num_text, max_text-old_num_text,
             sizeof *new_text));
814 815
      }

816
      if (new_text == NULL)
817
      {
818
         png_chunk_report(png_ptr, "too many text chunks",
kurenai's avatar
kurenai committed
819
             PNG_CHUNK_WRITE_ERROR);
820 821

         return 1;
822 823
      }

824 825 826 827 828 829 830 831
      png_free(png_ptr, info_ptr->text);

      info_ptr->text = new_text;
      info_ptr->free_me |= PNG_FREE_TEXT;
      info_ptr->max_text = max_text;
      /* num_text is adjusted below as the entries are copied in */

      png_debug1(3, "allocated %d entries for info_ptr->text", max_text);
832
   }
833

834 835
   for (i = 0; i < num_text; i++)
   {
836 837
      size_t text_length, key_len;
      size_t lang_len, lang_key_len;
838 839 840 841 842 843 844 845
      png_textp textp = &(info_ptr->text[info_ptr->num_text]);

      if (text_ptr[i].key == NULL)
          continue;

      if (text_ptr[i].compression < PNG_TEXT_COMPRESSION_NONE ||
          text_ptr[i].compression >= PNG_TEXT_COMPRESSION_LAST)
      {
846
         png_chunk_report(png_ptr, "text compression mode is out of range",
kurenai's avatar
kurenai committed
847
             PNG_CHUNK_WRITE_ERROR);
848 849 850
         continue;
      }

851
      key_len = strlen(text_ptr[i].key);
852 853 854 855 856 857 858 859 860 861 862 863 864

      if (text_ptr[i].compression <= 0)
      {
         lang_len = 0;
         lang_key_len = 0;
      }

      else
#  ifdef PNG_iTXt_SUPPORTED
      {
         /* Set iTXt data */

         if (text_ptr[i].lang != NULL)
865
            lang_len = strlen(text_ptr[i].lang);
866 867 868 869 870

         else
            lang_len = 0;

         if (text_ptr[i].lang_key != NULL)
871
            lang_key_len = strlen(text_ptr[i].lang_key);
872 873 874 875

         else
            lang_key_len = 0;
      }
876
#  else /* iTXt */
877
      {
878
         png_chunk_report(png_ptr, "iTXt chunk not supported",
kurenai's avatar
kurenai committed
879
             PNG_CHUNK_WRITE_ERROR);
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897
         continue;
      }
#  endif

      if (text_ptr[i].text == NULL || text_ptr[i].text[0] == '\0')
      {
         text_length = 0;
#  ifdef PNG_iTXt_SUPPORTED
         if (text_ptr[i].compression > 0)
            textp->compression = PNG_ITXT_COMPRESSION_NONE;

         else
#  endif
            textp->compression = PNG_TEXT_COMPRESSION_NONE;
      }

      else
      {
898
         text_length = strlen(text_ptr[i].text);
899 900 901
         textp->compression = text_ptr[i].compression;
      }

902 903
      textp->key = png_voidcast(png_charp,png_malloc_base(png_ptr,
          key_len + text_length + lang_len + lang_key_len + 4));
904 905

      if (textp->key == NULL)
906 907
      {
         png_chunk_report(png_ptr, "text chunk: out of memory",
kurenai's avatar
kurenai committed
908
             PNG_CHUNK_WRITE_ERROR);
909 910 911

         return 1;
      }
912 913 914 915 916 917

      png_debug2(2, "Allocated %lu bytes at %p in png_set_text",
          (unsigned long)(png_uint_32)
          (key_len + lang_len + lang_key_len + text_length + 4),
          textp->key);

918
      memcpy(textp->key, text_ptr[i].key, key_len);
919 920 921 922 923
      *(textp->key + key_len) = '\0';

      if (text_ptr[i].compression > 0)
      {
         textp->lang = textp->key + key_len + 1;
924
         memcpy(textp->lang, text_ptr[i].lang, lang_len);
925 926
         *(textp->lang + lang_len) = '\0';
         textp->lang_key = textp->lang + lang_len + 1;
927
         memcpy(textp->lang_key, text_ptr[i].lang_key, lang_key_len);
928 929 930 931 932 933 934 935 936 937 938
         *(textp->lang_key + lang_key_len) = '\0';
         textp->text = textp->lang_key + lang_key_len + 1;
      }

      else
      {
         textp->lang=NULL;
         textp->lang_key=NULL;
         textp->text = textp->key + key_len + 1;
      }

939 940
      if (text_length != 0)
         memcpy(textp->text, text_ptr[i].text, text_length);
941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960

      *(textp->text + text_length) = '\0';

#  ifdef PNG_iTXt_SUPPORTED
      if (textp->compression > 0)
      {
         textp->text_length = 0;
         textp->itxt_length = text_length;
      }

      else
#  endif
      {
         textp->text_length = text_length;
         textp->itxt_length = 0;
      }

      info_ptr->num_text++;
      png_debug1(3, "transferred text chunk %d", info_ptr->num_text);
   }
961

962 963 964 965 966 967
   return(0);
}
#endif

#ifdef PNG_tIME_SUPPORTED
void PNGAPI
968 969
png_set_tIME(png_const_structrp png_ptr, png_inforp info_ptr,
    png_const_timep mod_time)
970 971 972
{
   png_debug1(1, "in %s storage function", "tIME");

973 974
   if (png_ptr == NULL || info_ptr == NULL || mod_time == NULL ||
       (png_ptr->mode & PNG_WROTE_tIME) != 0)
975 976 977 978 979 980 981 982
      return;

   if (mod_time->month == 0   || mod_time->month > 12  ||
       mod_time->day   == 0   || mod_time->day   > 31  ||
       mod_time->hour  > 23   || mod_time->minute > 59 ||
       mod_time->second > 60)
   {
      png_warning(png_ptr, "Ignoring invalid time value");
983

984 985 986
      return;
   }

987
   info_ptr->mod_time = *mod_time;
988 989 990 991 992 993
   info_ptr->valid |= PNG_INFO_tIME;
}
#endif

#ifdef PNG_tRNS_SUPPORTED
void PNGAPI
994
png_set_tRNS(png_structrp png_ptr, png_inforp info_ptr,
995 996 997 998 999
    png_const_bytep trans_alpha, int num_trans, png_const_color_16p trans_color)
{
   png_debug1(1, "in %s storage function", "tRNS");

   if (png_ptr == NULL || info_ptr == NULL)
1000

1001 1002 1003 1004 1005 1006 1007
      return;

   if (trans_alpha != NULL)
   {
       /* It may not actually be necessary to set png_ptr->trans_alpha here;
        * we do it for backward compatibility with the way the png_handle_tRNS
        * function used to do the allocation.
1008 1009 1010 1011
        *
        * 1.6.0: The above statement is incorrect; png_handle_tRNS effectively
        * relies on png_set_tRNS storing the information in png_struct
        * (otherwise it won't be there for the code in pngrtran.c).
1012 1013 1014 1015 1016
        */

       png_free_data(png_ptr, info_ptr, PNG_FREE_TRNS, 0);

       if (num_trans > 0 && num_trans <= PNG_MAX_PALETTE_LENGTH)
kurenai's avatar
kurenai committed
1017 1018 1019 1020
       {
         /* Changed from num_trans to PNG_MAX_PALETTE_LENGTH in version 1.2.1 */
          info_ptr->trans_alpha = png_voidcast(png_bytep,
              png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH));
1021
          memcpy(info_ptr->trans_alpha, trans_alpha, (png_size_t)num_trans);
kurenai's avatar
kurenai committed
1022 1023
       }
       png_ptr->trans_alpha = info_ptr->trans_alpha;
1024 1025 1026 1027
   }

   if (trans_color != NULL)
   {
1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
#ifdef PNG_WARNINGS_SUPPORTED
      if (info_ptr->bit_depth < 16)
      {
         int sample_max = (1 << info_ptr->bit_depth) - 1;

         if ((info_ptr->color_type == PNG_COLOR_TYPE_GRAY &&
             trans_color->gray > sample_max) ||
             (info_ptr->color_type == PNG_COLOR_TYPE_RGB &&
             (trans_color->red > sample_max ||
             trans_color->green > sample_max ||
             trans_color->blue > sample_max)))
            png_warning(png_ptr,
kurenai's avatar
kurenai committed
1040
                "tRNS chunk has out-of-range samples for bit_depth");
1041 1042
      }
#endif
1043

1044
      info_ptr->trans_color = *trans_color;
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061

      if (num_trans == 0)
         num_trans = 1;
   }

   info_ptr->num_trans = (png_uint_16)num_trans;

   if (num_trans != 0)
   {
      info_ptr->valid |= PNG_INFO_tRNS;
      info_ptr->free_me |= PNG_FREE_TRNS;
   }
}
#endif

#ifdef PNG_sPLT_SUPPORTED
void PNGAPI
1062 1063
png_set_sPLT(png_const_structrp png_ptr,
    png_inforp info_ptr, png_const_sPLT_tp entries, int nentries)
1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074
/*
 *  entries        - array of png_sPLT_t structures
 *                   to be added to the list of palettes
 *                   in the info structure.
 *
 *  nentries       - number of palette structures to be
 *                   added.
 */
{
   png_sPLT_tp np;

1075
   if (png_ptr == NULL || info_ptr == NULL || nentries <= 0 || entries == NULL)
1076 1077
      return;

1078 1079 1080 1081
   /* Use the internal realloc function, which checks for all the possible
    * overflows.  Notice that the parameters are (int) and (size_t)
    */
   np = png_voidcast(png_sPLT_tp,png_realloc_array(png_ptr,
kurenai's avatar
kurenai committed
1082 1083
       info_ptr->splt_palettes, info_ptr->splt_palettes_num, nentries,
       sizeof *np));
1084 1085 1086

   if (np == NULL)
   {
1087 1088 1089
      /* Out of memory or too many chunks */
      png_chunk_report(png_ptr, "too many sPLT chunks", PNG_CHUNK_WRITE_ERROR);

1090 1091 1092 1093
      return;
   }

   png_free(png_ptr, info_ptr->splt_palettes);
1094 1095 1096 1097
   info_ptr->splt_palettes = np;
   info_ptr->free_me |= PNG_FREE_SPLT;

   np += info_ptr->splt_palettes_num;
1098

1099
   do
1100 1101 1102
   {
      png_size_t length;

1103 1104
      /* Skip invalid input entries */
      if (entries->name == NULL || entries->entries == NULL)
1105
      {
1106 1107 1108
         /* png_handle_sPLT doesn't do this, so this is an app error */
         png_app_error(png_ptr, "png_set_sPLT: invalid sPLT");
         /* Just skip the invalid entry */
1109 1110 1111
         continue;
      }

1112 1113 1114 1115 1116 1117 1118 1119 1120 1121
      np->depth = entries->depth;

      /* In the event of out-of-memory just return - there's no point keeping
       * on trying to add sPLT chunks.
       */
      length = strlen(entries->name) + 1;
      np->name = png_voidcast(png_charp, png_malloc_base(png_ptr, length));

      if (np->name == NULL)
         break;
1122

1123 1124 1125 1126 1127 1128 1129 1130 1131 1132
      memcpy(np->name, entries->name, length);

      /* IMPORTANT: we have memory now that won't get freed if something else
       * goes wrong; this code must free it.  png_malloc_array produces no
       * warnings; use a png_chunk_report (below) if there is an error.
       */
      np->entries = png_voidcast(png_sPLT_entryp, png_malloc_array(png_ptr,
          entries->nentries, sizeof (png_sPLT_entry)));

      if (np->entries == NULL)
1133
      {
1134 1135 1136
         png_free(png_ptr, np->name);
         np->name = NULL;
         break;
1137 1138
      }

1139 1140 1141 1142 1143
      np->nentries = entries->nentries;
      /* This multiply can't overflow because png_malloc_array has already
       * checked it when doing the allocation.
       */
      memcpy(np->entries, entries->entries,
1144
          (unsigned int)entries->nentries * sizeof (png_sPLT_entry));
1145 1146 1147 1148 1149 1150 1151

      /* Note that 'continue' skips the advance of the out pointer and out
       * count, so an invalid entry is not added.
       */
      info_ptr->valid |= PNG_INFO_sPLT;
      ++(info_ptr->splt_palettes_num);
      ++np;
1152
      ++entries;
1153
   }
1154
   while (--nentries);
1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165

   if (nentries > 0)
      png_chunk_report(png_ptr, "sPLT out of memory", PNG_CHUNK_WRITE_ERROR);
}
#endif /* sPLT */

#ifdef PNG_STORE_UNKNOWN_CHUNKS_SUPPORTED
static png_byte
check_location(png_const_structrp png_ptr, int location)
{
   location &= (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT);
1166

1167 1168 1169 1170 1171 1172 1173 1174
   /* New in 1.6.0; copy the location and check it.  This is an API
    * change; previously the app had to use the
    * png_set_unknown_chunk_location API below for each chunk.
    */
   if (location == 0 && (png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
   {
      /* Write struct, so unknown chunks come from the app */
      png_app_warning(png_ptr,
kurenai's avatar
kurenai committed
1175
          "png_set_unknown_chunks now expects a valid location");
1176 1177
      /* Use the old behavior */
      location = (png_byte)(png_ptr->mode &
kurenai's avatar
kurenai committed
1178
          (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT));
1179 1180
   }

1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196
   /* This need not be an internal error - if the app calls
    * png_set_unknown_chunks on a read pointer it must get the location right.
    */
   if (location == 0)
      png_error(png_ptr, "invalid location in png_set_unknown_chunks");

   /* Now reduce the location to the top-most set bit by removing each least
    * significant bit in turn.
    */
   while (location != (location & -location))
      location &= ~(location & -location);

   /* The cast is safe because 'location' is a bit mask and only the low four
    * bits are significant.
    */
   return (png_byte)location;
1197 1198 1199
}

void PNGAPI
1200
png_set_unknown_chunks(png_const_structrp png_ptr,
kurenai's avatar
kurenai committed
1201
    png_inforp info_ptr, png_const_unknown_chunkp unknowns, int num_unknowns)
1202 1203 1204
{
   png_unknown_chunkp np;

1205 1206
   if (png_ptr == NULL || info_ptr == NULL || num_unknowns <= 0 ||
       unknowns == NULL)
1207 1208
      return;

1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
   /* Check for the failure cases where support has been disabled at compile
    * time.  This code is hardly ever compiled - it's here because
    * STORE_UNKNOWN_CHUNKS is set by both read and write code (compiling in this
    * code) but may be meaningless if the read or write handling of unknown
    * chunks is not compiled in.
    */
#  if !defined(PNG_READ_UNKNOWN_CHUNKS_SUPPORTED) && \
      defined(PNG_READ_SUPPORTED)
      if ((png_ptr->mode & PNG_IS_READ_STRUCT) != 0)
      {
         png_app_error(png_ptr, "no unknown chunk support on read");

         return;
      }
#  endif
#  if !defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED) && \
      defined(PNG_WRITE_SUPPORTED)
      if ((png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
      {
         png_app_error(png_ptr, "no unknown chunk support on write");

         return;
      }
#  endif

   /* Prior to 1.6.0 this code used png_malloc_warn; however, this meant that
    * unknown critical chunks could be lost with just a warning resulting in
    * undefined behavior.  Now png_chunk_report is used to provide behavior
    * appropriate to read or write.
    */
   np = png_voidcast(png_unknown_chunkp, png_realloc_array(png_ptr,
kurenai's avatar
kurenai committed
1240 1241
       info_ptr->unknown_chunks, info_ptr->unknown_chunks_num, num_unknowns,
       sizeof *np));
1242 1243 1244

   if (np == NULL)
   {
1245
      png_chunk_report(png_ptr, "too many unknown chunks",
kurenai's avatar
kurenai committed
1246
          PNG_CHUNK_WRITE_ERROR);
1247

1248 1249 1250 1251
      return;
   }

   png_free(png_ptr, info_ptr->unknown_chunks);
1252 1253
   info_ptr->unknown_chunks = np; /* safe because it is initialized */
   info_ptr->free_me |= PNG_FREE_UNKN;
1254

1255
   np += info_ptr->unknown_chunks_num;
1256

1257 1258 1259 1260 1261 1262 1263 1264
   /* Increment unknown_chunks_num each time round the loop to protect the
    * just-allocated chunk data.
    */
   for (; num_unknowns > 0; --num_unknowns, ++unknowns)
   {
      memcpy(np->name, unknowns->name, (sizeof np->name));
      np->name[(sizeof np->name)-1] = '\0';
      np->location = check_location(png_ptr, unknowns->location);
1265

1266 1267 1268 1269 1270
      if (unknowns->size == 0)
      {
         np->data = NULL;
         np->size = 0;
      }
1271 1272 1273

      else
      {
1274
         np->data = png_voidcast(png_bytep,
kurenai's avatar
kurenai committed
1275
             png_malloc_base(png_ptr, unknowns->size));
1276

1277
         if (np->data == NULL)
1278
         {
1279
            png_chunk_report(png_ptr, "unknown chunk: out of memory",
kurenai's avatar
kurenai committed
1280
                PNG_CHUNK_WRITE_ERROR);
1281 1282
            /* But just skip storing the unknown chunk */
            continue;
1283 1284
         }

1285 1286
         memcpy(np->data, unknowns->data, unknowns->size);
         np->size = unknowns->size;
1287 1288
      }

1289 1290 1291 1292 1293 1294 1295
      /* These increments are skipped on out-of-memory for the data - the
       * unknown chunk entry gets overwritten if the png_chunk_report returns.
       * This is correct in the read case (the chunk is just dropped.)
       */
      ++np;
      ++(info_ptr->unknown_chunks_num);
   }
1296 1297 1298
}

void PNGAPI
1299
png_set_unknown_chunk_location(png_const_structrp png_ptr, png_inforp info_ptr,
1300 1301
    int chunk, int location)
{
1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313
   /* This API is pretty pointless in 1.6.0 because the location can be set
    * before the call to png_set_unknown_chunks.
    *
    * TODO: add a png_app_warning in 1.7
    */
   if (png_ptr != NULL && info_ptr != NULL && chunk >= 0 &&
      chunk < info_ptr->unknown_chunks_num)
   {
      if ((location & (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT)) == 0)
      {
         png_app_error(png_ptr, "invalid unknown chunk location");
         /* Fake out the pre 1.6.0 behavior: */
1314
         if (((unsigned int)location & PNG_HAVE_IDAT) != 0) /* undocumented! */
1315
            location = PNG_AFTER_IDAT;
1316

1317 1318 1319 1320 1321 1322 1323 1324 1325
         else
            location = PNG_HAVE_IHDR; /* also undocumented */
      }

      info_ptr->unknown_chunks[chunk].location =
         check_location(png_ptr, location);
   }
}
#endif /* STORE_UNKNOWN_CHUNKS */
1326 1327 1328

#ifdef PNG_MNG_FEATURES_SUPPORTED
png_uint_32 PNGAPI
1329
png_permit_mng_features (png_structrp png_ptr, png_uint_32 mng_features)
1330 1331 1332 1333
{
   png_debug(1, "in png_permit_mng_features");

   if (png_ptr == NULL)
1334
      return 0;
1335

1336
   png_ptr->mng_features_permitted = mng_features & PNG_ALL_MNG_FEATURES;
1337

1338
   return png_ptr->mng_features_permitted;
1339 1340 1341 1342
}
#endif

#ifdef PNG_HANDLE_AS_UNKNOWN_SUPPORTED
1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
static unsigned int
add_one_chunk(png_bytep list, unsigned int count, png_const_bytep add, int keep)
{
   unsigned int i;

   /* Utility function: update the 'keep' state of a chunk if it is already in
    * the list, otherwise add it to the list.
    */
   for (i=0; i<count; ++i, list += 5)
   {
      if (memcmp(list, add, 4) == 0)
      {
         list[4] = (png_byte)keep;

         return count;
      }
   }

   if (keep != PNG_HANDLE_CHUNK_AS_DEFAULT)
   {
      ++count;
      memcpy(list, add, 4);
      list[4] = (png_byte)keep;
   }

   return count;
}

1371
void PNGAPI
1372 1373
png_set_keep_unknown_chunks(png_structrp png_ptr, int keep,
    png_const_bytep chunk_list, int num_chunks_in)
1374
{
1375 1376 1377
   png_bytep new_list;
   unsigned int num_chunks, old_num_chunks;

1378 1379 1380
   if (png_ptr == NULL)
      return;

1381
   if (keep < 0 || keep >= PNG_HANDLE_CHUNK_LAST)
1382
   {
1383
      png_app_error(png_ptr, "png_set_keep_unknown_chunks: invalid keep");
1384

1385 1386
      return;
   }
1387

1388 1389 1390
   if (num_chunks_in <= 0)
   {
      png_ptr->unknown_default = keep;
1391

1392 1393 1394 1395
      /* '0' means just set the flags, so stop here */
      if (num_chunks_in == 0)
        return;
   }
1396

1397 1398 1399 1400 1401 1402 1403 1404
   if (num_chunks_in < 0)
   {
      /* Ignore all unknown chunks and all chunks recognized by
       * libpng except for IHDR, PLTE, tRNS, IDAT, and IEND
       */
      static PNG_CONST png_byte chunks_to_ignore[] = {
         98,  75,  71,  68, '\0',  /* bKGD */
         99,  72,  82,  77, '\0',  /* cHRM */
1405
        101,  88,  73, 102, '\0',  /* eXIf */
1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424
        103,  65,  77,  65, '\0',  /* gAMA */
        104,  73,  83,  84, '\0',  /* hIST */
        105,  67,  67,  80, '\0',  /* iCCP */
        105,  84,  88, 116, '\0',  /* iTXt */
        111,  70,  70, 115, '\0',  /* oFFs */
        112,  67,  65,  76, '\0',  /* pCAL */
        112,  72,  89, 115, '\0',  /* pHYs */
        115,  66,  73,  84, '\0',  /* sBIT */
        115,  67,  65,  76, '\0',  /* sCAL */
        115,  80,  76,  84, '\0',  /* sPLT */
        115,  84,  69,  82, '\0',  /* sTER */
        115,  82,  71,  66, '\0',  /* sRGB */
        116,  69,  88, 116, '\0',  /* tEXt */
        116,  73,  77,  69, '\0',  /* tIME */
        122,  84,  88, 116, '\0'   /* zTXt */
      };

      chunk_list = chunks_to_ignore;
      num_chunks = (unsigned int)/*SAFE*/(sizeof chunks_to_ignore)/5U;
1425 1426
   }

1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438
   else /* num_chunks_in > 0 */
   {
      if (chunk_list == NULL)
      {
         /* Prior to 1.6.0 this was silently ignored, now it is an app_error
          * which can be switched off.
          */
         png_app_error(png_ptr, "png_set_keep_unknown_chunks: no chunk list");

         return;
      }

1439
      num_chunks = (unsigned int)num_chunks_in;
1440
   }
1441 1442

   old_num_chunks = png_ptr->num_chunk_list;
1443 1444
   if (png_ptr->chunk_list == NULL)
      old_num_chunks = 0;
1445

1446 1447 1448
   /* Since num_chunks is always restricted to UINT_MAX/5 this can't overflow.
    */
   if (num_chunks + old_num_chunks > UINT_MAX/5)
1449
   {
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
      png_app_error(png_ptr, "png_set_keep_unknown_chunks: too many chunks");

      return;
   }

   /* If these chunks are being reset to the default then no more memory is
    * required because add_one_chunk above doesn't extend the list if the 'keep'
    * parameter is the default.
    */
   if (keep != 0)
   {
      new_list = png_voidcast(png_bytep, png_malloc(png_ptr,
          5 * (num_chunks + old_num_chunks)));

      if (old_num_chunks > 0)
         memcpy(new_list, png_ptr->chunk_list, 5*old_num_chunks);
1466 1467
   }

1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487
   else if (old_num_chunks > 0)
      new_list = png_ptr->chunk_list;

   else
      new_list = NULL;

   /* Add the new chunks together with each one's handling code.  If the chunk
    * already exists the code is updated, otherwise the chunk is added to the
    * end.  (In libpng 1.6.0 order no longer matters because this code enforces
    * the earlier convention that the last setting is the one that is used.)
    */
   if (new_list != NULL)
   {
      png_const_bytep inlist;
      png_bytep outlist;
      unsigned int i;

      for (i=0; i<num_chunks; ++i)
      {
         old_num_chunks = add_one_chunk(new_list, old_num_chunks,
kurenai's avatar
kurenai committed
1488
             chunk_list+5*i, keep);
1489
      }
1490

1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522
      /* Now remove any spurious 'default' entries. */
      num_chunks = 0;
      for (i=0, inlist=outlist=new_list; i<old_num_chunks; ++i, inlist += 5)
      {
         if (inlist[4])
         {
            if (outlist != inlist)
               memcpy(outlist, inlist, 5);
            outlist += 5;
            ++num_chunks;
         }
      }

      /* This means the application has removed all the specialized handling. */
      if (num_chunks == 0)
      {
         if (png_ptr->chunk_list != new_list)
            png_free(png_ptr, new_list);

         new_list = NULL;
      }
   }

   else
      num_chunks = 0;

   png_ptr->num_chunk_list = num_chunks;

   if (png_ptr->chunk_list != new_list)
   {
      if (png_ptr->chunk_list != NULL)
         png_free(png_ptr, png_ptr->chunk_list);
1523

1524 1525
      png_ptr->chunk_list = new_list;
   }
1526 1527 1528 1529 1530
}
#endif

#ifdef PNG_READ_USER_CHUNKS_SUPPORTED
void PNGAPI
1531
png_set_read_user_chunk_fn(png_structrp png_ptr, png_voidp user_chunk_ptr,
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545
    png_user_chunk_ptr read_user_chunk_fn)
{
   png_debug(1, "in png_set_read_user_chunk_fn");

   if (png_ptr == NULL)
      return;

   png_ptr->read_user_chunk_fn = read_user_chunk_fn;
   png_ptr->user_chunk_ptr = user_chunk_ptr;
}
#endif

#ifdef PNG_INFO_IMAGE_SUPPORTED
void PNGAPI
1546 1547
png_set_rows(png_const_structrp png_ptr, png_inforp info_ptr,
    png_bytepp row_pointers)
1548 1549 1550 1551 1552 1553
{
   png_debug1(1, "in %s storage function", "rows");

   if (png_ptr == NULL || info_ptr == NULL)
      return;

1554 1555
   if (info_ptr->row_pointers != NULL &&
       (info_ptr->row_pointers != row_pointers))
1556 1557 1558 1559
      png_free_data(png_ptr, info_ptr, PNG_FREE_ROWS, 0);

   info_ptr->row_pointers = row_pointers;

1560
   if (row_pointers != NULL)
1561 1562 1563 1564 1565
      info_ptr->valid |= PNG_INFO_IDAT;
}
#endif

void PNGAPI
1566
png_set_compression_buffer_size(png_structrp png_ptr, png_size_t size)
1567
{
kurenai's avatar
kurenai committed
1568 1569
   if (png_ptr == NULL)
      return;
1570

kurenai's avatar
kurenai committed
1571 1572
   if (size == 0 || size > PNG_UINT_31_MAX)
      png_error(png_ptr, "invalid compression buffer size");
1573

1574
#  ifdef PNG_SEQUENTIAL_READ_SUPPORTED
kurenai's avatar
kurenai committed
1575 1576 1577 1578 1579
   if ((png_ptr->mode & PNG_IS_READ_STRUCT) != 0)
   {
      png_ptr->IDAT_read_size = (png_uint_32)size; /* checked above */
      return;
   }
1580
#  endif
1581

1582
#  ifdef PNG_WRITE_SUPPORTED
kurenai's avatar
kurenai committed
1583 1584 1585
   if ((png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
   {
      if (png_ptr->zowner != 0)
1586
      {
kurenai's avatar
kurenai committed
1587 1588
         png_warning(png_ptr,
             "Compression buffer size cannot be changed because it is in use");
1589

kurenai's avatar
kurenai committed
1590 1591
         return;
      }
1592

1593
#ifndef __COVERITY__
kurenai's avatar
kurenai committed
1594 1595 1596 1597 1598 1599 1600 1601 1602
      /* Some compilers complain that this is always false.  However, it
       * can be true when integer overflow happens.
       */
      if (size > ZLIB_IO_MAX)
      {
         png_warning(png_ptr,
             "Compression buffer size limited to system maximum");
         size = ZLIB_IO_MAX; /* must fit */
      }
1603 1604
#endif

kurenai's avatar
kurenai committed
1605 1606 1607 1608 1609 1610 1611
      if (size < 6)
      {
         /* Deflate will potentially go into an infinite loop on a SYNC_FLUSH
          * if this is permitted.
          */
         png_warning(png_ptr,
             "Compression buffer size cannot be reduced below 6");
1612

kurenai's avatar
kurenai committed
1613 1614
         return;
      }
1615

kurenai's avatar
kurenai committed
1616 1617 1618 1619
      if (png_ptr->zbuffer_size != size)
      {
         png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list);
         png_ptr->zbuffer_size = (uInt)size;
1620
      }
kurenai's avatar
kurenai committed
1621
   }
1622
#  endif
1623 1624 1625
}

void PNGAPI
1626
png_set_invalid(png_const_structrp png_ptr, png_inforp info_ptr, int mask)
1627
{
1628
   if (png_ptr != NULL && info_ptr != NULL)
1629
      info_ptr->valid &= (unsigned int)(~mask);
1630 1631 1632 1633 1634 1635
}


#ifdef PNG_SET_USER_LIMITS_SUPPORTED
/* This function was added to libpng 1.2.6 */
void PNGAPI
1636
png_set_user_limits (png_structrp png_ptr, png_uint_32 user_width_max,
1637 1638 1639 1640
    png_uint_32 user_height_max)
{
   /* Images with dimensions larger than these limits will be
    * rejected by png_set_IHDR().  To accept any PNG datastream
kurenai's avatar
kurenai committed
1641
    * regardless of dimensions, set both limits to 0x7fffffff.
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651
    */
   if (png_ptr == NULL)
      return;

   png_ptr->user_width_max = user_width_max;
   png_ptr->user_height_max = user_height_max;
}

/* This function was added to libpng 1.4.0 */
void PNGAPI
1652
png_set_chunk_cache_max (png_structrp png_ptr, png_uint_32 user_chunk_cache_max)
1653
{
1654 1655
   if (png_ptr != NULL)
      png_ptr->user_chunk_cache_max = user_chunk_cache_max;
1656 1657 1658 1659
}

/* This function was added to libpng 1.4.1 */
void PNGAPI
1660
png_set_chunk_malloc_max (png_structrp png_ptr,
1661 1662
    png_alloc_size_t user_chunk_malloc_max)
{
1663
   if (png_ptr != NULL)
1664 1665
      png_ptr->user_chunk_malloc_max = user_chunk_malloc_max;
}
1666
#endif /* ?SET_USER_LIMITS */
1667 1668 1669 1670


#ifdef PNG_BENIGN_ERRORS_SUPPORTED
void PNGAPI
1671
png_set_benign_errors(png_structrp png_ptr, int allowed)
1672 1673 1674
{
   png_debug(1, "in png_set_benign_errors");

1675 1676 1677 1678 1679 1680 1681 1682 1683
   /* If allowed is 1, png_benign_error() is treated as a warning.
    *
    * If allowed is 0, png_benign_error() is treated as an error (which
    * is the default behavior if png_set_benign_errors() is not called).
    */

   if (allowed != 0)
      png_ptr->flags |= PNG_FLAG_BENIGN_ERRORS_WARN |
         PNG_FLAG_APP_WARNINGS_WARN | PNG_FLAG_APP_ERRORS_WARN;
1684 1685

   else
1686 1687
      png_ptr->flags &= ~(PNG_FLAG_BENIGN_ERRORS_WARN |
         PNG_FLAG_APP_WARNINGS_WARN | PNG_FLAG_APP_ERRORS_WARN);
1688
}
1689
#endif /* BENIGN_ERRORS */
1690 1691

#ifdef PNG_CHECK_FOR_INVALID_INDEX_SUPPORTED
1692 1693 1694 1695 1696 1697 1698 1699
   /* Whether to report invalid palette index; added at libng-1.5.10.
    * It is possible for an indexed (color-type==3) PNG file to contain
    * pixels with invalid (out-of-range) indexes if the PLTE chunk has
    * fewer entries than the image's bit-depth would allow. We recover
    * from this gracefully by filling any incomplete palette with zeros
    * (opaque black).  By default, when this occurs libpng will issue
    * a benign error.  This API can be used to override that behavior.
    */
1700
void PNGAPI
1701
png_set_check_for_invalid_index(png_structrp png_ptr, int allowed)
1702 1703 1704
{
   png_debug(1, "in png_set_check_for_invalid_index");

1705
   if (allowed > 0)
1706 1707 1708 1709 1710 1711
      png_ptr->num_palette_max = 0;

   else
      png_ptr->num_palette_max = -1;
}
#endif
kurenai's avatar
kurenai committed
1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747

#if defined(PNG_TEXT_SUPPORTED) || defined(PNG_pCAL_SUPPORTED) || \
    defined(PNG_iCCP_SUPPORTED) || defined(PNG_sPLT_SUPPORTED)
/* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
 * and if invalid, correct the keyword rather than discarding the entire
 * chunk.  The PNG 1.0 specification requires keywords 1-79 characters in
 * length, forbids leading or trailing whitespace, multiple internal spaces,
 * and the non-break space (0x80) from ISO 8859-1.  Returns keyword length.
 *
 * The 'new_key' buffer must be 80 characters in size (for the keyword plus a
 * trailing '\0').  If this routine returns 0 then there was no keyword, or a
 * valid one could not be generated, and the caller must png_error.
 */
png_uint_32 /* PRIVATE */
png_check_keyword(png_structrp png_ptr, png_const_charp key, png_bytep new_key)
{
#ifdef PNG_WARNINGS_SUPPORTED
   png_const_charp orig_key = key;
#endif
   png_uint_32 key_len = 0;
   int bad_character = 0;
   int space = 1;

   png_debug(1, "in png_check_keyword");

   if (key == NULL)
   {
      *new_key = 0;
      return 0;
   }

   while (*key && key_len < 79)
   {
      png_byte ch = (png_byte)*key++;

      if ((ch > 32 && ch <= 126) || (ch >= 161 /*&& ch <= 255*/))
1748 1749 1750
      {
         *new_key++ = ch; ++key_len; space = 0;
      }
kurenai's avatar
kurenai committed
1751 1752 1753 1754 1755 1756

      else if (space == 0)
      {
         /* A space or an invalid character when one wasn't seen immediately
          * before; output just a space.
          */
1757
         *new_key++ = 32; ++key_len; space = 1;
kurenai's avatar
kurenai committed
1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769

         /* If the character was not a space then it is invalid. */
         if (ch != 32)
            bad_character = ch;
      }

      else if (bad_character == 0)
         bad_character = ch; /* just skip it, record the first error */
   }

   if (key_len > 0 && space != 0) /* trailing space */
   {
1770
      --key_len; --new_key;
kurenai's avatar
kurenai committed
1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801
      if (bad_character == 0)
         bad_character = 32;
   }

   /* Terminate the keyword */
   *new_key = 0;

   if (key_len == 0)
      return 0;

#ifdef PNG_WARNINGS_SUPPORTED
   /* Try to only output one warning per keyword: */
   if (*key != 0) /* keyword too long */
      png_warning(png_ptr, "keyword truncated");

   else if (bad_character != 0)
   {
      PNG_WARNING_PARAMETERS(p)

      png_warning_parameter(p, 1, orig_key);
      png_warning_parameter_signed(p, 2, PNG_NUMBER_FORMAT_02x, bad_character);

      png_formatted_warning(png_ptr, p, "keyword \"@1\": bad character '0x@2'");
   }
#else /* !WARNINGS */
   PNG_UNUSED(png_ptr)
#endif /* !WARNINGS */

   return key_len;
}
#endif /* TEXT || pCAL || iCCP || sPLT */
1802
#endif /* READ || WRITE */