Merge branch 'upstream' into tizen
[platform/upstream/libpng.git] / pngset.c
1
2 /* pngset.c - storage of image information into info struct
3  *
4  * Copyright (c) 2018-2023 Cosmin Truta
5  * Copyright (c) 1998-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  * The functions here are used during reads to store data from the file
14  * into the info struct, and during writes to store application data
15  * into the info struct for writing into the file.  This abstracts the
16  * info struct and allows us to change the structure in the future.
17  */
18
19 #include "pngpriv.h"
20
21 #if defined(PNG_READ_SUPPORTED) || defined(PNG_WRITE_SUPPORTED)
22
23 #ifdef PNG_bKGD_SUPPORTED
24 void PNGAPI
25 png_set_bKGD(png_const_structrp png_ptr, png_inforp info_ptr,
26     png_const_color_16p background)
27 {
28    png_debug1(1, "in %s storage function", "bKGD");
29
30    if (png_ptr == NULL || info_ptr == NULL || background == NULL)
31       return;
32
33    info_ptr->background = *background;
34    info_ptr->valid |= PNG_INFO_bKGD;
35 }
36 #endif
37
38 #ifdef PNG_cHRM_SUPPORTED
39 void PNGFAPI
40 png_set_cHRM_fixed(png_const_structrp png_ptr, png_inforp info_ptr,
41     png_fixed_point white_x, png_fixed_point white_y, png_fixed_point red_x,
42     png_fixed_point red_y, png_fixed_point green_x, png_fixed_point green_y,
43     png_fixed_point blue_x, png_fixed_point blue_y)
44 {
45    png_xy xy;
46
47    png_debug1(1, "in %s storage function", "cHRM fixed");
48
49    if (png_ptr == NULL || info_ptr == NULL)
50       return;
51
52    xy.redx = red_x;
53    xy.redy = red_y;
54    xy.greenx = green_x;
55    xy.greeny = green_y;
56    xy.bluex = blue_x;
57    xy.bluey = blue_y;
58    xy.whitex = white_x;
59    xy.whitey = white_y;
60
61    if (png_colorspace_set_chromaticities(png_ptr, &info_ptr->colorspace, &xy,
62        2/* override with app values*/) != 0)
63       info_ptr->colorspace.flags |= PNG_COLORSPACE_FROM_cHRM;
64
65    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     png_fixed_point int_red_X, png_fixed_point int_red_Y,
71     png_fixed_point int_red_Z, png_fixed_point int_green_X,
72     png_fixed_point int_green_Y, png_fixed_point int_green_Z,
73     png_fixed_point int_blue_X, png_fixed_point int_blue_Y,
74     png_fixed_point int_blue_Z)
75 {
76    png_XYZ XYZ;
77
78    png_debug1(1, "in %s storage function", "cHRM XYZ fixed");
79
80    if (png_ptr == NULL || info_ptr == NULL)
81       return;
82
83    XYZ.red_X = int_red_X;
84    XYZ.red_Y = int_red_Y;
85    XYZ.red_Z = int_red_Z;
86    XYZ.green_X = int_green_X;
87    XYZ.green_Y = int_green_Y;
88    XYZ.green_Z = int_green_Z;
89    XYZ.blue_X = int_blue_X;
90    XYZ.blue_Y = int_blue_Y;
91    XYZ.blue_Z = int_blue_Z;
92
93    if (png_colorspace_set_endpoints(png_ptr, &info_ptr->colorspace,
94        &XYZ, 2) != 0)
95       info_ptr->colorspace.flags |= PNG_COLORSPACE_FROM_cHRM;
96
97    png_colorspace_sync_info(png_ptr, info_ptr);
98 }
99
100 #  ifdef PNG_FLOATING_POINT_SUPPORTED
101 void PNGAPI
102 png_set_cHRM(png_const_structrp png_ptr, png_inforp info_ptr,
103     double white_x, double white_y, double red_x, double red_y,
104     double green_x, double green_y, double blue_x, double blue_y)
105 {
106    png_set_cHRM_fixed(png_ptr, info_ptr,
107        png_fixed(png_ptr, white_x, "cHRM White X"),
108        png_fixed(png_ptr, white_y, "cHRM White Y"),
109        png_fixed(png_ptr, red_x, "cHRM Red X"),
110        png_fixed(png_ptr, red_y, "cHRM Red Y"),
111        png_fixed(png_ptr, green_x, "cHRM Green X"),
112        png_fixed(png_ptr, green_y, "cHRM Green Y"),
113        png_fixed(png_ptr, blue_x, "cHRM Blue X"),
114        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     double red_Y, double red_Z, double green_X, double green_Y, double green_Z,
120     double blue_X, double blue_Y, double blue_Z)
121 {
122    png_set_cHRM_XYZ_fixed(png_ptr, info_ptr,
123        png_fixed(png_ptr, red_X, "cHRM Red X"),
124        png_fixed(png_ptr, red_Y, "cHRM Red Y"),
125        png_fixed(png_ptr, red_Z, "cHRM Red Z"),
126        png_fixed(png_ptr, green_X, "cHRM Green X"),
127        png_fixed(png_ptr, green_Y, "cHRM Green Y"),
128        png_fixed(png_ptr, green_Z, "cHRM Green Z"),
129        png_fixed(png_ptr, blue_X, "cHRM Blue X"),
130        png_fixed(png_ptr, blue_Y, "cHRM Blue Y"),
131        png_fixed(png_ptr, blue_Z, "cHRM Blue Z"));
132 }
133 #  endif /* FLOATING_POINT */
134
135 #endif /* cHRM */
136
137 #ifdef PNG_eXIf_SUPPORTED
138 void PNGAPI
139 png_set_eXIf(png_const_structrp png_ptr, png_inforp info_ptr,
140     png_bytep exif)
141 {
142   png_warning(png_ptr, "png_set_eXIf does not work; use png_set_eXIf_1");
143   PNG_UNUSED(info_ptr)
144   PNG_UNUSED(exif)
145 }
146
147 void PNGAPI
148 png_set_eXIf_1(png_const_structrp png_ptr, png_inforp info_ptr,
149     png_uint_32 num_exif, png_bytep exif)
150 {
151    png_bytep new_exif;
152
153    png_debug1(1, "in %s storage function", "eXIf");
154
155    if (png_ptr == NULL || info_ptr == NULL ||
156        (png_ptr->mode & PNG_WROTE_eXIf) != 0)
157       return;
158
159    new_exif = png_voidcast(png_bytep, png_malloc_warn(png_ptr, num_exif));
160
161    if (new_exif == NULL)
162    {
163       png_warning(png_ptr, "Insufficient memory for eXIf chunk data");
164       return;
165    }
166
167    memcpy(new_exif, exif, (size_t)num_exif);
168
169    png_free_data(png_ptr, info_ptr, PNG_FREE_EXIF, 0);
170
171    info_ptr->num_exif = num_exif;
172    info_ptr->exif = new_exif;
173    info_ptr->free_me |= PNG_FREE_EXIF;
174    info_ptr->valid |= PNG_INFO_eXIf;
175 }
176 #endif /* eXIf */
177
178 #ifdef PNG_gAMA_SUPPORTED
179 void PNGFAPI
180 png_set_gAMA_fixed(png_const_structrp png_ptr, png_inforp info_ptr,
181     png_fixed_point file_gamma)
182 {
183    png_debug1(1, "in %s storage function", "gAMA");
184
185    if (png_ptr == NULL || info_ptr == NULL)
186       return;
187
188    png_colorspace_set_gamma(png_ptr, &info_ptr->colorspace, file_gamma);
189    png_colorspace_sync_info(png_ptr, info_ptr);
190 }
191
192 #  ifdef PNG_FLOATING_POINT_SUPPORTED
193 void PNGAPI
194 png_set_gAMA(png_const_structrp png_ptr, png_inforp info_ptr, double file_gamma)
195 {
196    png_set_gAMA_fixed(png_ptr, info_ptr, png_fixed(png_ptr, file_gamma,
197        "png_set_gAMA"));
198 }
199 #  endif
200 #endif
201
202 #ifdef PNG_hIST_SUPPORTED
203 void PNGAPI
204 png_set_hIST(png_const_structrp png_ptr, png_inforp info_ptr,
205     png_const_uint_16p hist)
206 {
207    int i;
208
209    png_debug1(1, "in %s storage function", "hIST");
210
211    if (png_ptr == NULL || info_ptr == NULL)
212       return;
213
214    if (info_ptr->num_palette == 0 || info_ptr->num_palette
215        > PNG_MAX_PALETTE_LENGTH)
216    {
217       png_warning(png_ptr,
218           "Invalid palette size, hIST allocation skipped");
219
220       return;
221    }
222
223    png_free_data(png_ptr, info_ptr, PNG_FREE_HIST, 0);
224
225    /* Changed from info->num_palette to PNG_MAX_PALETTE_LENGTH in
226     * version 1.2.1
227     */
228    info_ptr->hist = png_voidcast(png_uint_16p, png_malloc_warn(png_ptr,
229        PNG_MAX_PALETTE_LENGTH * (sizeof (png_uint_16))));
230
231    if (info_ptr->hist == NULL)
232    {
233       png_warning(png_ptr, "Insufficient memory for hIST chunk data");
234       return;
235    }
236
237    for (i = 0; i < info_ptr->num_palette; i++)
238       info_ptr->hist[i] = hist[i];
239
240    info_ptr->free_me |= PNG_FREE_HIST;
241    info_ptr->valid |= PNG_INFO_hIST;
242 }
243 #endif
244
245 void PNGAPI
246 png_set_IHDR(png_const_structrp png_ptr, png_inforp info_ptr,
247     png_uint_32 width, png_uint_32 height, int bit_depth,
248     int color_type, int interlace_type, int compression_type,
249     int filter_type)
250 {
251    png_debug1(1, "in %s storage function", "IHDR");
252
253    if (png_ptr == NULL || info_ptr == NULL)
254       return;
255
256    info_ptr->width = width;
257    info_ptr->height = height;
258    info_ptr->bit_depth = (png_byte)bit_depth;
259    info_ptr->color_type = (png_byte)color_type;
260    info_ptr->compression_type = (png_byte)compression_type;
261    info_ptr->filter_type = (png_byte)filter_type;
262    info_ptr->interlace_type = (png_byte)interlace_type;
263
264    png_check_IHDR (png_ptr, info_ptr->width, info_ptr->height,
265        info_ptr->bit_depth, info_ptr->color_type, info_ptr->interlace_type,
266        info_ptr->compression_type, info_ptr->filter_type);
267
268    if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
269       info_ptr->channels = 1;
270
271    else if ((info_ptr->color_type & PNG_COLOR_MASK_COLOR) != 0)
272       info_ptr->channels = 3;
273
274    else
275       info_ptr->channels = 1;
276
277    if ((info_ptr->color_type & PNG_COLOR_MASK_ALPHA) != 0)
278       info_ptr->channels++;
279
280    info_ptr->pixel_depth = (png_byte)(info_ptr->channels * info_ptr->bit_depth);
281
282    info_ptr->rowbytes = PNG_ROWBYTES(info_ptr->pixel_depth, width);
283
284 #ifdef PNG_APNG_SUPPORTED
285    /* for non-animated png. this may be overwritten from an acTL chunk later */
286    info_ptr->num_frames = 1;
287 #endif
288 }
289
290 #ifdef PNG_oFFs_SUPPORTED
291 void PNGAPI
292 png_set_oFFs(png_const_structrp png_ptr, png_inforp info_ptr,
293     png_int_32 offset_x, png_int_32 offset_y, int unit_type)
294 {
295    png_debug1(1, "in %s storage function", "oFFs");
296
297    if (png_ptr == NULL || info_ptr == NULL)
298       return;
299
300    info_ptr->x_offset = offset_x;
301    info_ptr->y_offset = offset_y;
302    info_ptr->offset_unit_type = (png_byte)unit_type;
303    info_ptr->valid |= PNG_INFO_oFFs;
304 }
305 #endif
306
307 #ifdef PNG_pCAL_SUPPORTED
308 void PNGAPI
309 png_set_pCAL(png_const_structrp png_ptr, png_inforp info_ptr,
310     png_const_charp purpose, png_int_32 X0, png_int_32 X1, int type,
311     int nparams, png_const_charp units, png_charpp params)
312 {
313    size_t length;
314    int i;
315
316    png_debug1(1, "in %s storage function", "pCAL");
317
318    if (png_ptr == NULL || info_ptr == NULL || purpose == NULL || units == NULL
319        || (nparams > 0 && params == NULL))
320       return;
321
322    length = strlen(purpose) + 1;
323    png_debug1(3, "allocating purpose for info (%lu bytes)",
324        (unsigned long)length);
325
326    /* TODO: validate format of calibration name and unit name */
327
328    /* Check that the type matches the specification. */
329    if (type < 0 || type > 3)
330    {
331       png_chunk_report(png_ptr, "Invalid pCAL equation type",
332             PNG_CHUNK_WRITE_ERROR);
333       return;
334    }
335
336    if (nparams < 0 || nparams > 255)
337    {
338       png_chunk_report(png_ptr, "Invalid pCAL parameter count",
339             PNG_CHUNK_WRITE_ERROR);
340       return;
341    }
342
343    /* Validate params[nparams] */
344    for (i=0; i<nparams; ++i)
345    {
346       if (params[i] == NULL ||
347           !png_check_fp_string(params[i], strlen(params[i])))
348       {
349          png_chunk_report(png_ptr, "Invalid format for pCAL parameter",
350                PNG_CHUNK_WRITE_ERROR);
351          return;
352       }
353    }
354
355    info_ptr->pcal_purpose = png_voidcast(png_charp,
356        png_malloc_warn(png_ptr, length));
357
358    if (info_ptr->pcal_purpose == NULL)
359    {
360       png_chunk_report(png_ptr, "Insufficient memory for pCAL purpose",
361             PNG_CHUNK_WRITE_ERROR);
362       return;
363    }
364
365    memcpy(info_ptr->pcal_purpose, purpose, length);
366
367    info_ptr->free_me |= PNG_FREE_PCAL;
368
369    png_debug(3, "storing X0, X1, type, and nparams in info");
370    info_ptr->pcal_X0 = X0;
371    info_ptr->pcal_X1 = X1;
372    info_ptr->pcal_type = (png_byte)type;
373    info_ptr->pcal_nparams = (png_byte)nparams;
374
375    length = strlen(units) + 1;
376    png_debug1(3, "allocating units for info (%lu bytes)",
377        (unsigned long)length);
378
379    info_ptr->pcal_units = png_voidcast(png_charp,
380        png_malloc_warn(png_ptr, length));
381
382    if (info_ptr->pcal_units == NULL)
383    {
384       png_warning(png_ptr, "Insufficient memory for pCAL units");
385       return;
386    }
387
388    memcpy(info_ptr->pcal_units, units, length);
389
390    info_ptr->pcal_params = png_voidcast(png_charpp, png_malloc_warn(png_ptr,
391        (size_t)(((unsigned int)nparams + 1) * (sizeof (png_charp)))));
392
393    if (info_ptr->pcal_params == NULL)
394    {
395       png_warning(png_ptr, "Insufficient memory for pCAL params");
396       return;
397    }
398
399    memset(info_ptr->pcal_params, 0, ((unsigned int)nparams + 1) *
400        (sizeof (png_charp)));
401
402    for (i = 0; i < nparams; i++)
403    {
404       length = strlen(params[i]) + 1;
405       png_debug2(3, "allocating parameter %d for info (%lu bytes)", i,
406           (unsigned long)length);
407
408       info_ptr->pcal_params[i] = (png_charp)png_malloc_warn(png_ptr, length);
409
410       if (info_ptr->pcal_params[i] == NULL)
411       {
412          png_warning(png_ptr, "Insufficient memory for pCAL parameter");
413          return;
414       }
415
416       memcpy(info_ptr->pcal_params[i], params[i], length);
417    }
418
419    info_ptr->valid |= PNG_INFO_pCAL;
420 }
421 #endif
422
423 #ifdef PNG_sCAL_SUPPORTED
424 void PNGAPI
425 png_set_sCAL_s(png_const_structrp png_ptr, png_inforp info_ptr,
426     int unit, png_const_charp swidth, png_const_charp sheight)
427 {
428    size_t lengthw = 0, lengthh = 0;
429
430    png_debug1(1, "in %s storage function", "sCAL");
431
432    if (png_ptr == NULL || info_ptr == NULL)
433       return;
434
435    /* Double check the unit (should never get here with an invalid
436     * unit unless this is an API call.)
437     */
438    if (unit != 1 && unit != 2)
439       png_error(png_ptr, "Invalid sCAL unit");
440
441    if (swidth == NULL || (lengthw = strlen(swidth)) == 0 ||
442        swidth[0] == 45 /* '-' */ || !png_check_fp_string(swidth, lengthw))
443       png_error(png_ptr, "Invalid sCAL width");
444
445    if (sheight == NULL || (lengthh = strlen(sheight)) == 0 ||
446        sheight[0] == 45 /* '-' */ || !png_check_fp_string(sheight, lengthh))
447       png_error(png_ptr, "Invalid sCAL height");
448
449    info_ptr->scal_unit = (png_byte)unit;
450
451    ++lengthw;
452
453    png_debug1(3, "allocating unit for info (%u bytes)", (unsigned int)lengthw);
454
455    info_ptr->scal_s_width = png_voidcast(png_charp,
456        png_malloc_warn(png_ptr, lengthw));
457
458    if (info_ptr->scal_s_width == NULL)
459    {
460       png_warning(png_ptr, "Memory allocation failed while processing sCAL");
461
462       return;
463    }
464
465    memcpy(info_ptr->scal_s_width, swidth, lengthw);
466
467    ++lengthh;
468
469    png_debug1(3, "allocating unit for info (%u bytes)", (unsigned int)lengthh);
470
471    info_ptr->scal_s_height = png_voidcast(png_charp,
472        png_malloc_warn(png_ptr, lengthh));
473
474    if (info_ptr->scal_s_height == NULL)
475    {
476       png_free(png_ptr, info_ptr->scal_s_width);
477       info_ptr->scal_s_width = NULL;
478
479       png_warning(png_ptr, "Memory allocation failed while processing sCAL");
480       return;
481    }
482
483    memcpy(info_ptr->scal_s_height, sheight, lengthh);
484
485    info_ptr->free_me |= PNG_FREE_SCAL;
486    info_ptr->valid |= PNG_INFO_sCAL;
487 }
488
489 #  ifdef PNG_FLOATING_POINT_SUPPORTED
490 void PNGAPI
491 png_set_sCAL(png_const_structrp png_ptr, png_inforp info_ptr, int unit,
492     double width, double height)
493 {
494    png_debug1(1, "in %s storage function", "sCAL");
495
496    /* Check the arguments. */
497    if (width <= 0)
498       png_warning(png_ptr, "Invalid sCAL width ignored");
499
500    else if (height <= 0)
501       png_warning(png_ptr, "Invalid sCAL height ignored");
502
503    else
504    {
505       /* Convert 'width' and 'height' to ASCII. */
506       char swidth[PNG_sCAL_MAX_DIGITS+1];
507       char sheight[PNG_sCAL_MAX_DIGITS+1];
508
509       png_ascii_from_fp(png_ptr, swidth, (sizeof swidth), width,
510           PNG_sCAL_PRECISION);
511       png_ascii_from_fp(png_ptr, sheight, (sizeof sheight), height,
512           PNG_sCAL_PRECISION);
513
514       png_set_sCAL_s(png_ptr, info_ptr, unit, swidth, sheight);
515    }
516 }
517 #  endif
518
519 #  ifdef PNG_FIXED_POINT_SUPPORTED
520 void PNGAPI
521 png_set_sCAL_fixed(png_const_structrp png_ptr, png_inforp info_ptr, int unit,
522     png_fixed_point width, png_fixed_point height)
523 {
524    png_debug1(1, "in %s storage function", "sCAL");
525
526    /* Check the arguments. */
527    if (width <= 0)
528       png_warning(png_ptr, "Invalid sCAL width ignored");
529
530    else if (height <= 0)
531       png_warning(png_ptr, "Invalid sCAL height ignored");
532
533    else
534    {
535       /* Convert 'width' and 'height' to ASCII. */
536       char swidth[PNG_sCAL_MAX_DIGITS+1];
537       char sheight[PNG_sCAL_MAX_DIGITS+1];
538
539       png_ascii_from_fixed(png_ptr, swidth, (sizeof swidth), width);
540       png_ascii_from_fixed(png_ptr, sheight, (sizeof sheight), height);
541
542       png_set_sCAL_s(png_ptr, info_ptr, unit, swidth, sheight);
543    }
544 }
545 #  endif
546 #endif
547
548 #ifdef PNG_pHYs_SUPPORTED
549 void PNGAPI
550 png_set_pHYs(png_const_structrp png_ptr, png_inforp info_ptr,
551     png_uint_32 res_x, png_uint_32 res_y, int unit_type)
552 {
553    png_debug1(1, "in %s storage function", "pHYs");
554
555    if (png_ptr == NULL || info_ptr == NULL)
556       return;
557
558    info_ptr->x_pixels_per_unit = res_x;
559    info_ptr->y_pixels_per_unit = res_y;
560    info_ptr->phys_unit_type = (png_byte)unit_type;
561    info_ptr->valid |= PNG_INFO_pHYs;
562 }
563 #endif
564
565 void PNGAPI
566 png_set_PLTE(png_structrp png_ptr, png_inforp info_ptr,
567     png_const_colorp palette, int num_palette)
568 {
569
570    png_uint_32 max_palette_length;
571
572    png_debug1(1, "in %s storage function", "PLTE");
573
574    if (png_ptr == NULL || info_ptr == NULL)
575       return;
576
577    max_palette_length = (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE) ?
578       (1 << info_ptr->bit_depth) : PNG_MAX_PALETTE_LENGTH;
579
580    if (num_palette < 0 || num_palette > (int) max_palette_length)
581    {
582       if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
583          png_error(png_ptr, "Invalid palette length");
584
585       else
586       {
587          png_warning(png_ptr, "Invalid palette length");
588
589          return;
590       }
591    }
592
593    if ((num_palette > 0 && palette == NULL) ||
594       (num_palette == 0
595 #        ifdef PNG_MNG_FEATURES_SUPPORTED
596             && (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0
597 #        endif
598       ))
599    {
600       png_error(png_ptr, "Invalid palette");
601    }
602
603    /* It may not actually be necessary to set png_ptr->palette here;
604     * we do it for backward compatibility with the way the png_handle_tRNS
605     * function used to do the allocation.
606     *
607     * 1.6.0: the above statement appears to be incorrect; something has to set
608     * the palette inside png_struct on read.
609     */
610    png_free_data(png_ptr, info_ptr, PNG_FREE_PLTE, 0);
611
612    /* Changed in libpng-1.2.1 to allocate PNG_MAX_PALETTE_LENGTH instead
613     * of num_palette entries, in case of an invalid PNG file or incorrect
614     * call to png_set_PLTE() with too-large sample values.
615     */
616    png_ptr->palette = png_voidcast(png_colorp, png_calloc(png_ptr,
617        PNG_MAX_PALETTE_LENGTH * (sizeof (png_color))));
618
619    if (num_palette > 0)
620       memcpy(png_ptr->palette, palette, (unsigned int)num_palette *
621           (sizeof (png_color)));
622
623    info_ptr->palette = png_ptr->palette;
624    info_ptr->num_palette = png_ptr->num_palette = (png_uint_16)num_palette;
625    info_ptr->free_me |= PNG_FREE_PLTE;
626    info_ptr->valid |= PNG_INFO_PLTE;
627 }
628
629 #ifdef PNG_sBIT_SUPPORTED
630 void PNGAPI
631 png_set_sBIT(png_const_structrp png_ptr, png_inforp info_ptr,
632     png_const_color_8p sig_bit)
633 {
634    png_debug1(1, "in %s storage function", "sBIT");
635
636    if (png_ptr == NULL || info_ptr == NULL || sig_bit == NULL)
637       return;
638
639    info_ptr->sig_bit = *sig_bit;
640    info_ptr->valid |= PNG_INFO_sBIT;
641 }
642 #endif
643
644 #ifdef PNG_sRGB_SUPPORTED
645 void PNGAPI
646 png_set_sRGB(png_const_structrp png_ptr, png_inforp info_ptr, int srgb_intent)
647 {
648    png_debug1(1, "in %s storage function", "sRGB");
649
650    if (png_ptr == NULL || info_ptr == NULL)
651       return;
652
653    (void)png_colorspace_set_sRGB(png_ptr, &info_ptr->colorspace, srgb_intent);
654    png_colorspace_sync_info(png_ptr, info_ptr);
655 }
656
657 void PNGAPI
658 png_set_sRGB_gAMA_and_cHRM(png_const_structrp png_ptr, png_inforp info_ptr,
659     int srgb_intent)
660 {
661    png_debug1(1, "in %s storage function", "sRGB_gAMA_and_cHRM");
662
663    if (png_ptr == NULL || info_ptr == NULL)
664       return;
665
666    if (png_colorspace_set_sRGB(png_ptr, &info_ptr->colorspace,
667        srgb_intent) != 0)
668    {
669       /* This causes the gAMA and cHRM to be written too */
670       info_ptr->colorspace.flags |=
671          PNG_COLORSPACE_FROM_gAMA|PNG_COLORSPACE_FROM_cHRM;
672    }
673
674    png_colorspace_sync_info(png_ptr, info_ptr);
675 }
676 #endif /* sRGB */
677
678
679 #ifdef PNG_iCCP_SUPPORTED
680 void PNGAPI
681 png_set_iCCP(png_const_structrp png_ptr, png_inforp info_ptr,
682     png_const_charp name, int compression_type,
683     png_const_bytep profile, png_uint_32 proflen)
684 {
685    png_charp new_iccp_name;
686    png_bytep new_iccp_profile;
687    size_t length;
688
689    png_debug1(1, "in %s storage function", "iCCP");
690
691    if (png_ptr == NULL || info_ptr == NULL || name == NULL || profile == NULL)
692       return;
693
694    if (compression_type != PNG_COMPRESSION_TYPE_BASE)
695       png_app_error(png_ptr, "Invalid iCCP compression method");
696
697    /* Set the colorspace first because this validates the profile; do not
698     * override previously set app cHRM or gAMA here (because likely as not the
699     * application knows better than libpng what the correct values are.)  Pass
700     * the info_ptr color_type field to png_colorspace_set_ICC because in the
701     * write case it has not yet been stored in png_ptr.
702     */
703    {
704       int result = png_colorspace_set_ICC(png_ptr, &info_ptr->colorspace, name,
705           proflen, profile, info_ptr->color_type);
706
707       png_colorspace_sync_info(png_ptr, info_ptr);
708
709       /* Don't do any of the copying if the profile was bad, or inconsistent. */
710       if (result == 0)
711          return;
712
713       /* But do write the gAMA and cHRM chunks from the profile. */
714       info_ptr->colorspace.flags |=
715          PNG_COLORSPACE_FROM_gAMA|PNG_COLORSPACE_FROM_cHRM;
716    }
717
718    length = strlen(name)+1;
719    new_iccp_name = png_voidcast(png_charp, png_malloc_warn(png_ptr, length));
720
721    if (new_iccp_name == NULL)
722    {
723       png_benign_error(png_ptr, "Insufficient memory to process iCCP chunk");
724
725       return;
726    }
727
728    memcpy(new_iccp_name, name, length);
729    new_iccp_profile = png_voidcast(png_bytep,
730        png_malloc_warn(png_ptr, proflen));
731
732    if (new_iccp_profile == NULL)
733    {
734       png_free(png_ptr, new_iccp_name);
735       png_benign_error(png_ptr,
736           "Insufficient memory to process iCCP profile");
737
738       return;
739    }
740
741    memcpy(new_iccp_profile, profile, proflen);
742
743    png_free_data(png_ptr, info_ptr, PNG_FREE_ICCP, 0);
744
745    info_ptr->iccp_proflen = proflen;
746    info_ptr->iccp_name = new_iccp_name;
747    info_ptr->iccp_profile = new_iccp_profile;
748    info_ptr->free_me |= PNG_FREE_ICCP;
749    info_ptr->valid |= PNG_INFO_iCCP;
750 }
751 #endif
752
753 #ifdef PNG_TEXT_SUPPORTED
754 void PNGAPI
755 png_set_text(png_const_structrp png_ptr, png_inforp info_ptr,
756     png_const_textp text_ptr, int num_text)
757 {
758    int ret;
759    ret = png_set_text_2(png_ptr, info_ptr, text_ptr, num_text);
760
761    if (ret != 0)
762       png_error(png_ptr, "Insufficient memory to store text");
763 }
764
765 int /* PRIVATE */
766 png_set_text_2(png_const_structrp png_ptr, png_inforp info_ptr,
767     png_const_textp text_ptr, int num_text)
768 {
769    int i;
770
771    png_debug1(1, "in %lx storage function", png_ptr == NULL ? 0xabadca11U :
772       (unsigned long)png_ptr->chunk_name);
773
774    if (png_ptr == NULL || info_ptr == NULL || num_text <= 0 || text_ptr == NULL)
775       return(0);
776
777    /* Make sure we have enough space in the "text" array in info_struct
778     * to hold all of the incoming text_ptr objects.  This compare can't overflow
779     * because max_text >= num_text (anyway, subtract of two positive integers
780     * can't overflow in any case.)
781     */
782    if (num_text > info_ptr->max_text - info_ptr->num_text)
783    {
784       int old_num_text = info_ptr->num_text;
785       int max_text;
786       png_textp new_text = NULL;
787
788       /* Calculate an appropriate max_text, checking for overflow. */
789       max_text = old_num_text;
790       if (num_text <= INT_MAX - max_text)
791       {
792          max_text += num_text;
793
794          /* Round up to a multiple of 8 */
795          if (max_text < INT_MAX-8)
796             max_text = (max_text + 8) & ~0x7;
797
798          else
799             max_text = INT_MAX;
800
801          /* Now allocate a new array and copy the old members in; this does all
802           * the overflow checks.
803           */
804          new_text = png_voidcast(png_textp,png_realloc_array(png_ptr,
805              info_ptr->text, old_num_text, max_text-old_num_text,
806              sizeof *new_text));
807       }
808
809       if (new_text == NULL)
810       {
811          png_chunk_report(png_ptr, "too many text chunks",
812              PNG_CHUNK_WRITE_ERROR);
813
814          return 1;
815       }
816
817       png_free(png_ptr, info_ptr->text);
818
819       info_ptr->text = new_text;
820       info_ptr->free_me |= PNG_FREE_TEXT;
821       info_ptr->max_text = max_text;
822       /* num_text is adjusted below as the entries are copied in */
823
824       png_debug1(3, "allocated %d entries for info_ptr->text", max_text);
825    }
826
827    for (i = 0; i < num_text; i++)
828    {
829       size_t text_length, key_len;
830       size_t lang_len, lang_key_len;
831       png_textp textp = &(info_ptr->text[info_ptr->num_text]);
832
833       if (text_ptr[i].key == NULL)
834           continue;
835
836       if (text_ptr[i].compression < PNG_TEXT_COMPRESSION_NONE ||
837           text_ptr[i].compression >= PNG_TEXT_COMPRESSION_LAST)
838       {
839          png_chunk_report(png_ptr, "text compression mode is out of range",
840              PNG_CHUNK_WRITE_ERROR);
841          continue;
842       }
843
844       key_len = strlen(text_ptr[i].key);
845
846       if (text_ptr[i].compression <= 0)
847       {
848          lang_len = 0;
849          lang_key_len = 0;
850       }
851
852       else
853 #  ifdef PNG_iTXt_SUPPORTED
854       {
855          /* Set iTXt data */
856
857          if (text_ptr[i].lang != NULL)
858             lang_len = strlen(text_ptr[i].lang);
859
860          else
861             lang_len = 0;
862
863          if (text_ptr[i].lang_key != NULL)
864             lang_key_len = strlen(text_ptr[i].lang_key);
865
866          else
867             lang_key_len = 0;
868       }
869 #  else /* iTXt */
870       {
871          png_chunk_report(png_ptr, "iTXt chunk not supported",
872              PNG_CHUNK_WRITE_ERROR);
873          continue;
874       }
875 #  endif
876
877       if (text_ptr[i].text == NULL || text_ptr[i].text[0] == '\0')
878       {
879          text_length = 0;
880 #  ifdef PNG_iTXt_SUPPORTED
881          if (text_ptr[i].compression > 0)
882             textp->compression = PNG_ITXT_COMPRESSION_NONE;
883
884          else
885 #  endif
886             textp->compression = PNG_TEXT_COMPRESSION_NONE;
887       }
888
889       else
890       {
891          text_length = strlen(text_ptr[i].text);
892          textp->compression = text_ptr[i].compression;
893       }
894
895       textp->key = png_voidcast(png_charp,png_malloc_base(png_ptr,
896           key_len + text_length + lang_len + lang_key_len + 4));
897
898       if (textp->key == NULL)
899       {
900          png_chunk_report(png_ptr, "text chunk: out of memory",
901              PNG_CHUNK_WRITE_ERROR);
902
903          return 1;
904       }
905
906       png_debug2(2, "Allocated %lu bytes at %p in png_set_text",
907           (unsigned long)(png_uint_32)
908           (key_len + lang_len + lang_key_len + text_length + 4),
909           textp->key);
910
911       memcpy(textp->key, text_ptr[i].key, key_len);
912       *(textp->key + key_len) = '\0';
913
914       if (text_ptr[i].compression > 0)
915       {
916          textp->lang = textp->key + key_len + 1;
917          memcpy(textp->lang, text_ptr[i].lang, lang_len);
918          *(textp->lang + lang_len) = '\0';
919          textp->lang_key = textp->lang + lang_len + 1;
920          memcpy(textp->lang_key, text_ptr[i].lang_key, lang_key_len);
921          *(textp->lang_key + lang_key_len) = '\0';
922          textp->text = textp->lang_key + lang_key_len + 1;
923       }
924
925       else
926       {
927          textp->lang=NULL;
928          textp->lang_key=NULL;
929          textp->text = textp->key + key_len + 1;
930       }
931
932       if (text_length != 0)
933          memcpy(textp->text, text_ptr[i].text, text_length);
934
935       *(textp->text + text_length) = '\0';
936
937 #  ifdef PNG_iTXt_SUPPORTED
938       if (textp->compression > 0)
939       {
940          textp->text_length = 0;
941          textp->itxt_length = text_length;
942       }
943
944       else
945 #  endif
946       {
947          textp->text_length = text_length;
948          textp->itxt_length = 0;
949       }
950
951       info_ptr->num_text++;
952       png_debug1(3, "transferred text chunk %d", info_ptr->num_text);
953    }
954
955    return(0);
956 }
957 #endif
958
959 #ifdef PNG_tIME_SUPPORTED
960 void PNGAPI
961 png_set_tIME(png_const_structrp png_ptr, png_inforp info_ptr,
962     png_const_timep mod_time)
963 {
964    png_debug1(1, "in %s storage function", "tIME");
965
966    if (png_ptr == NULL || info_ptr == NULL || mod_time == NULL ||
967        (png_ptr->mode & PNG_WROTE_tIME) != 0)
968       return;
969
970    if (mod_time->month == 0   || mod_time->month > 12  ||
971        mod_time->day   == 0   || mod_time->day   > 31  ||
972        mod_time->hour  > 23   || mod_time->minute > 59 ||
973        mod_time->second > 60)
974    {
975       png_warning(png_ptr, "Ignoring invalid time value");
976
977       return;
978    }
979
980    info_ptr->mod_time = *mod_time;
981    info_ptr->valid |= PNG_INFO_tIME;
982 }
983 #endif
984
985 #ifdef PNG_tRNS_SUPPORTED
986 void PNGAPI
987 png_set_tRNS(png_structrp png_ptr, png_inforp info_ptr,
988     png_const_bytep trans_alpha, int num_trans, png_const_color_16p trans_color)
989 {
990    png_debug1(1, "in %s storage function", "tRNS");
991
992    if (png_ptr == NULL || info_ptr == NULL)
993
994       return;
995
996    if (trans_alpha != NULL)
997    {
998        /* It may not actually be necessary to set png_ptr->trans_alpha here;
999         * we do it for backward compatibility with the way the png_handle_tRNS
1000         * function used to do the allocation.
1001         *
1002         * 1.6.0: The above statement is incorrect; png_handle_tRNS effectively
1003         * relies on png_set_tRNS storing the information in png_struct
1004         * (otherwise it won't be there for the code in pngrtran.c).
1005         */
1006
1007        png_free_data(png_ptr, info_ptr, PNG_FREE_TRNS, 0);
1008
1009        if (num_trans > 0 && num_trans <= PNG_MAX_PALETTE_LENGTH)
1010        {
1011          /* Changed from num_trans to PNG_MAX_PALETTE_LENGTH in version 1.2.1 */
1012           info_ptr->trans_alpha = png_voidcast(png_bytep,
1013               png_malloc(png_ptr, PNG_MAX_PALETTE_LENGTH));
1014           memcpy(info_ptr->trans_alpha, trans_alpha, (size_t)num_trans);
1015
1016           info_ptr->free_me |= PNG_FREE_TRNS;
1017           info_ptr->valid |= PNG_INFO_tRNS;
1018        }
1019        png_ptr->trans_alpha = info_ptr->trans_alpha;
1020    }
1021
1022    if (trans_color != NULL)
1023    {
1024 #ifdef PNG_WARNINGS_SUPPORTED
1025       if (info_ptr->bit_depth < 16)
1026       {
1027          int sample_max = (1 << info_ptr->bit_depth) - 1;
1028
1029          if ((info_ptr->color_type == PNG_COLOR_TYPE_GRAY &&
1030              trans_color->gray > sample_max) ||
1031              (info_ptr->color_type == PNG_COLOR_TYPE_RGB &&
1032              (trans_color->red > sample_max ||
1033              trans_color->green > sample_max ||
1034              trans_color->blue > sample_max)))
1035             png_warning(png_ptr,
1036                 "tRNS chunk has out-of-range samples for bit_depth");
1037       }
1038 #endif
1039
1040       info_ptr->trans_color = *trans_color;
1041
1042       if (num_trans == 0)
1043          num_trans = 1;
1044    }
1045
1046    info_ptr->num_trans = (png_uint_16)num_trans;
1047
1048    if (num_trans != 0)
1049    {
1050       info_ptr->free_me |= PNG_FREE_TRNS;
1051       info_ptr->valid |= PNG_INFO_tRNS;
1052    }
1053 }
1054 #endif
1055
1056 #ifdef PNG_sPLT_SUPPORTED
1057 void PNGAPI
1058 png_set_sPLT(png_const_structrp png_ptr,
1059     png_inforp info_ptr, png_const_sPLT_tp entries, int nentries)
1060 /*
1061  *  entries        - array of png_sPLT_t structures
1062  *                   to be added to the list of palettes
1063  *                   in the info structure.
1064  *
1065  *  nentries       - number of palette structures to be
1066  *                   added.
1067  */
1068 {
1069    png_sPLT_tp np;
1070
1071    if (png_ptr == NULL || info_ptr == NULL || nentries <= 0 || entries == NULL)
1072       return;
1073
1074    /* Use the internal realloc function, which checks for all the possible
1075     * overflows.  Notice that the parameters are (int) and (size_t)
1076     */
1077    np = png_voidcast(png_sPLT_tp,png_realloc_array(png_ptr,
1078        info_ptr->splt_palettes, info_ptr->splt_palettes_num, nentries,
1079        sizeof *np));
1080
1081    if (np == NULL)
1082    {
1083       /* Out of memory or too many chunks */
1084       png_chunk_report(png_ptr, "too many sPLT chunks", PNG_CHUNK_WRITE_ERROR);
1085       return;
1086    }
1087
1088    png_free(png_ptr, info_ptr->splt_palettes);
1089
1090    info_ptr->splt_palettes = np;
1091    info_ptr->free_me |= PNG_FREE_SPLT;
1092
1093    np += info_ptr->splt_palettes_num;
1094
1095    do
1096    {
1097       size_t length;
1098
1099       /* Skip invalid input entries */
1100       if (entries->name == NULL || entries->entries == NULL)
1101       {
1102          /* png_handle_sPLT doesn't do this, so this is an app error */
1103          png_app_error(png_ptr, "png_set_sPLT: invalid sPLT");
1104          /* Just skip the invalid entry */
1105          continue;
1106       }
1107
1108       np->depth = entries->depth;
1109
1110       /* In the event of out-of-memory just return - there's no point keeping
1111        * on trying to add sPLT chunks.
1112        */
1113       length = strlen(entries->name) + 1;
1114       np->name = png_voidcast(png_charp, png_malloc_base(png_ptr, length));
1115
1116       if (np->name == NULL)
1117          break;
1118
1119       memcpy(np->name, entries->name, length);
1120
1121       /* IMPORTANT: we have memory now that won't get freed if something else
1122        * goes wrong; this code must free it.  png_malloc_array produces no
1123        * warnings; use a png_chunk_report (below) if there is an error.
1124        */
1125       np->entries = png_voidcast(png_sPLT_entryp, png_malloc_array(png_ptr,
1126           entries->nentries, sizeof (png_sPLT_entry)));
1127
1128       if (np->entries == NULL)
1129       {
1130          png_free(png_ptr, np->name);
1131          np->name = NULL;
1132          break;
1133       }
1134
1135       np->nentries = entries->nentries;
1136       /* This multiply can't overflow because png_malloc_array has already
1137        * checked it when doing the allocation.
1138        */
1139       memcpy(np->entries, entries->entries,
1140           (unsigned int)entries->nentries * sizeof (png_sPLT_entry));
1141
1142       /* Note that 'continue' skips the advance of the out pointer and out
1143        * count, so an invalid entry is not added.
1144        */
1145       info_ptr->valid |= PNG_INFO_sPLT;
1146       ++(info_ptr->splt_palettes_num);
1147       ++np;
1148       ++entries;
1149    }
1150    while (--nentries);
1151
1152    if (nentries > 0)
1153       png_chunk_report(png_ptr, "sPLT out of memory", PNG_CHUNK_WRITE_ERROR);
1154 }
1155 #endif /* sPLT */
1156
1157 #ifdef PNG_APNG_SUPPORTED
1158 png_uint_32 PNGAPI
1159 png_set_acTL(png_structp png_ptr, png_infop info_ptr,
1160     png_uint_32 num_frames, png_uint_32 num_plays)
1161 {
1162     png_debug1(1, "in %s storage function", "acTL");
1163
1164     if (png_ptr == NULL || info_ptr == NULL)
1165     {
1166         png_warning(png_ptr,
1167                     "Call to png_set_acTL() with NULL png_ptr "
1168                     "or info_ptr ignored");
1169         return (0);
1170     }
1171     if (num_frames == 0)
1172     {
1173         png_warning(png_ptr,
1174                     "Ignoring attempt to set acTL with num_frames zero");
1175         return (0);
1176     }
1177     if (num_frames > PNG_UINT_31_MAX)
1178     {
1179         png_warning(png_ptr,
1180                     "Ignoring attempt to set acTL with num_frames > 2^31-1");
1181         return (0);
1182     }
1183     if (num_plays > PNG_UINT_31_MAX)
1184     {
1185         png_warning(png_ptr,
1186                     "Ignoring attempt to set acTL with num_plays "
1187                     "> 2^31-1");
1188         return (0);
1189     }
1190
1191     info_ptr->num_frames = num_frames;
1192     info_ptr->num_plays = num_plays;
1193
1194     info_ptr->valid |= PNG_INFO_acTL;
1195
1196     return (1);
1197 }
1198
1199 /* delay_num and delay_den can hold any 16-bit values including zero */
1200 png_uint_32 PNGAPI
1201 png_set_next_frame_fcTL(png_structp png_ptr, png_infop info_ptr,
1202     png_uint_32 width, png_uint_32 height,
1203     png_uint_32 x_offset, png_uint_32 y_offset,
1204     png_uint_16 delay_num, png_uint_16 delay_den,
1205     png_byte dispose_op, png_byte blend_op)
1206 {
1207     png_debug1(1, "in %s storage function", "fcTL");
1208
1209     if (png_ptr == NULL || info_ptr == NULL)
1210     {
1211         png_warning(png_ptr,
1212                     "Call to png_set_fcTL() with NULL png_ptr or info_ptr "
1213                     "ignored");
1214         return (0);
1215     }
1216
1217     png_ensure_fcTL_is_valid(png_ptr, width, height, x_offset, y_offset,
1218                              delay_num, delay_den, dispose_op, blend_op);
1219
1220     if (blend_op == PNG_BLEND_OP_OVER)
1221     {
1222         if (!(png_ptr->color_type & PNG_COLOR_MASK_ALPHA) &&
1223             !(png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)))
1224         {
1225           png_warning(png_ptr, "PNG_BLEND_OP_OVER is meaningless "
1226                                "and wasteful for opaque images, ignored");
1227           blend_op = PNG_BLEND_OP_SOURCE;
1228         }
1229     }
1230
1231     info_ptr->next_frame_width = width;
1232     info_ptr->next_frame_height = height;
1233     info_ptr->next_frame_x_offset = x_offset;
1234     info_ptr->next_frame_y_offset = y_offset;
1235     info_ptr->next_frame_delay_num = delay_num;
1236     info_ptr->next_frame_delay_den = delay_den;
1237     info_ptr->next_frame_dispose_op = dispose_op;
1238     info_ptr->next_frame_blend_op = blend_op;
1239
1240     info_ptr->valid |= PNG_INFO_fcTL;
1241
1242     return (1);
1243 }
1244
1245 void /* PRIVATE */
1246 png_ensure_fcTL_is_valid(png_structp png_ptr,
1247     png_uint_32 width, png_uint_32 height,
1248     png_uint_32 x_offset, png_uint_32 y_offset,
1249     png_uint_16 delay_num, png_uint_16 delay_den,
1250     png_byte dispose_op, png_byte blend_op)
1251 {
1252     if (width == 0 || width > PNG_UINT_31_MAX)
1253         png_error(png_ptr, "invalid width in fcTL (> 2^31-1)");
1254     if (height == 0 || height > PNG_UINT_31_MAX)
1255         png_error(png_ptr, "invalid height in fcTL (> 2^31-1)");
1256     if (x_offset > PNG_UINT_31_MAX)
1257         png_error(png_ptr, "invalid x_offset in fcTL (> 2^31-1)");
1258     if (y_offset > PNG_UINT_31_MAX)
1259         png_error(png_ptr, "invalid y_offset in fcTL (> 2^31-1)");
1260     if (width + x_offset > png_ptr->first_frame_width ||
1261         height + y_offset > png_ptr->first_frame_height)
1262         png_error(png_ptr, "dimensions of a frame are greater than"
1263                            "the ones in IHDR");
1264
1265     if (dispose_op != PNG_DISPOSE_OP_NONE &&
1266         dispose_op != PNG_DISPOSE_OP_BACKGROUND &&
1267         dispose_op != PNG_DISPOSE_OP_PREVIOUS)
1268         png_error(png_ptr, "invalid dispose_op in fcTL");
1269
1270     if (blend_op != PNG_BLEND_OP_SOURCE &&
1271         blend_op != PNG_BLEND_OP_OVER)
1272         png_error(png_ptr, "invalid blend_op in fcTL");
1273
1274     PNG_UNUSED(delay_num)
1275     PNG_UNUSED(delay_den)
1276 }
1277
1278 png_uint_32 PNGAPI
1279 png_set_first_frame_is_hidden(png_structp png_ptr, png_infop info_ptr,
1280                               png_byte is_hidden)
1281 {
1282     png_debug(1, "in png_first_frame_is_hidden()");
1283
1284     if (png_ptr == NULL)
1285         return 0;
1286
1287     if (is_hidden)
1288         png_ptr->apng_flags |= PNG_FIRST_FRAME_HIDDEN;
1289     else
1290         png_ptr->apng_flags &= ~PNG_FIRST_FRAME_HIDDEN;
1291
1292     PNG_UNUSED(info_ptr)
1293
1294     return 1;
1295 }
1296 #endif /* PNG_APNG_SUPPORTED */
1297
1298 #ifdef PNG_STORE_UNKNOWN_CHUNKS_SUPPORTED
1299 static png_byte
1300 check_location(png_const_structrp png_ptr, int location)
1301 {
1302    location &= (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT);
1303
1304    /* New in 1.6.0; copy the location and check it.  This is an API
1305     * change; previously the app had to use the
1306     * png_set_unknown_chunk_location API below for each chunk.
1307     */
1308    if (location == 0 && (png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
1309    {
1310       /* Write struct, so unknown chunks come from the app */
1311       png_app_warning(png_ptr,
1312           "png_set_unknown_chunks now expects a valid location");
1313       /* Use the old behavior */
1314       location = (png_byte)(png_ptr->mode &
1315           (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT));
1316    }
1317
1318    /* This need not be an internal error - if the app calls
1319     * png_set_unknown_chunks on a read pointer it must get the location right.
1320     */
1321    if (location == 0)
1322       png_error(png_ptr, "invalid location in png_set_unknown_chunks");
1323
1324    /* Now reduce the location to the top-most set bit by removing each least
1325     * significant bit in turn.
1326     */
1327    while (location != (location & -location))
1328       location &= ~(location & -location);
1329
1330    /* The cast is safe because 'location' is a bit mask and only the low four
1331     * bits are significant.
1332     */
1333    return (png_byte)location;
1334 }
1335
1336 void PNGAPI
1337 png_set_unknown_chunks(png_const_structrp png_ptr,
1338     png_inforp info_ptr, png_const_unknown_chunkp unknowns, int num_unknowns)
1339 {
1340    png_unknown_chunkp np;
1341
1342    if (png_ptr == NULL || info_ptr == NULL || num_unknowns <= 0 ||
1343        unknowns == NULL)
1344       return;
1345
1346    /* Check for the failure cases where support has been disabled at compile
1347     * time.  This code is hardly ever compiled - it's here because
1348     * STORE_UNKNOWN_CHUNKS is set by both read and write code (compiling in this
1349     * code) but may be meaningless if the read or write handling of unknown
1350     * chunks is not compiled in.
1351     */
1352 #  if !defined(PNG_READ_UNKNOWN_CHUNKS_SUPPORTED) && \
1353       defined(PNG_READ_SUPPORTED)
1354       if ((png_ptr->mode & PNG_IS_READ_STRUCT) != 0)
1355       {
1356          png_app_error(png_ptr, "no unknown chunk support on read");
1357
1358          return;
1359       }
1360 #  endif
1361 #  if !defined(PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED) && \
1362       defined(PNG_WRITE_SUPPORTED)
1363       if ((png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
1364       {
1365          png_app_error(png_ptr, "no unknown chunk support on write");
1366
1367          return;
1368       }
1369 #  endif
1370
1371    /* Prior to 1.6.0 this code used png_malloc_warn; however, this meant that
1372     * unknown critical chunks could be lost with just a warning resulting in
1373     * undefined behavior.  Now png_chunk_report is used to provide behavior
1374     * appropriate to read or write.
1375     */
1376    np = png_voidcast(png_unknown_chunkp, png_realloc_array(png_ptr,
1377        info_ptr->unknown_chunks, info_ptr->unknown_chunks_num, num_unknowns,
1378        sizeof *np));
1379
1380    if (np == NULL)
1381    {
1382       png_chunk_report(png_ptr, "too many unknown chunks",
1383           PNG_CHUNK_WRITE_ERROR);
1384       return;
1385    }
1386
1387    png_free(png_ptr, info_ptr->unknown_chunks);
1388
1389    info_ptr->unknown_chunks = np; /* safe because it is initialized */
1390    info_ptr->free_me |= PNG_FREE_UNKN;
1391
1392    np += info_ptr->unknown_chunks_num;
1393
1394    /* Increment unknown_chunks_num each time round the loop to protect the
1395     * just-allocated chunk data.
1396     */
1397    for (; num_unknowns > 0; --num_unknowns, ++unknowns)
1398    {
1399       memcpy(np->name, unknowns->name, (sizeof np->name));
1400       np->name[(sizeof np->name)-1] = '\0';
1401       np->location = check_location(png_ptr, unknowns->location);
1402
1403       if (unknowns->size == 0)
1404       {
1405          np->data = NULL;
1406          np->size = 0;
1407       }
1408
1409       else
1410       {
1411          np->data = png_voidcast(png_bytep,
1412              png_malloc_base(png_ptr, unknowns->size));
1413
1414          if (np->data == NULL)
1415          {
1416             png_chunk_report(png_ptr, "unknown chunk: out of memory",
1417                 PNG_CHUNK_WRITE_ERROR);
1418             /* But just skip storing the unknown chunk */
1419             continue;
1420          }
1421
1422          memcpy(np->data, unknowns->data, unknowns->size);
1423          np->size = unknowns->size;
1424       }
1425
1426       /* These increments are skipped on out-of-memory for the data - the
1427        * unknown chunk entry gets overwritten if the png_chunk_report returns.
1428        * This is correct in the read case (the chunk is just dropped.)
1429        */
1430       ++np;
1431       ++(info_ptr->unknown_chunks_num);
1432    }
1433 }
1434
1435 void PNGAPI
1436 png_set_unknown_chunk_location(png_const_structrp png_ptr, png_inforp info_ptr,
1437     int chunk, int location)
1438 {
1439    /* This API is pretty pointless in 1.6.0 because the location can be set
1440     * before the call to png_set_unknown_chunks.
1441     *
1442     * TODO: add a png_app_warning in 1.7
1443     */
1444    if (png_ptr != NULL && info_ptr != NULL && chunk >= 0 &&
1445       chunk < info_ptr->unknown_chunks_num)
1446    {
1447       if ((location & (PNG_HAVE_IHDR|PNG_HAVE_PLTE|PNG_AFTER_IDAT)) == 0)
1448       {
1449          png_app_error(png_ptr, "invalid unknown chunk location");
1450          /* Fake out the pre 1.6.0 behavior: */
1451          if (((unsigned int)location & PNG_HAVE_IDAT) != 0) /* undocumented! */
1452             location = PNG_AFTER_IDAT;
1453
1454          else
1455             location = PNG_HAVE_IHDR; /* also undocumented */
1456       }
1457
1458       info_ptr->unknown_chunks[chunk].location =
1459          check_location(png_ptr, location);
1460    }
1461 }
1462 #endif /* STORE_UNKNOWN_CHUNKS */
1463
1464 #ifdef PNG_MNG_FEATURES_SUPPORTED
1465 png_uint_32 PNGAPI
1466 png_permit_mng_features(png_structrp png_ptr, png_uint_32 mng_features)
1467 {
1468    png_debug(1, "in png_permit_mng_features");
1469
1470    if (png_ptr == NULL)
1471       return 0;
1472
1473    png_ptr->mng_features_permitted = mng_features & PNG_ALL_MNG_FEATURES;
1474
1475    return png_ptr->mng_features_permitted;
1476 }
1477 #endif
1478
1479 #ifdef PNG_HANDLE_AS_UNKNOWN_SUPPORTED
1480 static unsigned int
1481 add_one_chunk(png_bytep list, unsigned int count, png_const_bytep add, int keep)
1482 {
1483    unsigned int i;
1484
1485    /* Utility function: update the 'keep' state of a chunk if it is already in
1486     * the list, otherwise add it to the list.
1487     */
1488    for (i=0; i<count; ++i, list += 5)
1489    {
1490       if (memcmp(list, add, 4) == 0)
1491       {
1492          list[4] = (png_byte)keep;
1493
1494          return count;
1495       }
1496    }
1497
1498    if (keep != PNG_HANDLE_CHUNK_AS_DEFAULT)
1499    {
1500       ++count;
1501       memcpy(list, add, 4);
1502       list[4] = (png_byte)keep;
1503    }
1504
1505    return count;
1506 }
1507
1508 void PNGAPI
1509 png_set_keep_unknown_chunks(png_structrp png_ptr, int keep,
1510     png_const_bytep chunk_list, int num_chunks_in)
1511 {
1512    png_bytep new_list;
1513    unsigned int num_chunks, old_num_chunks;
1514
1515    if (png_ptr == NULL)
1516       return;
1517
1518    if (keep < 0 || keep >= PNG_HANDLE_CHUNK_LAST)
1519    {
1520       png_app_error(png_ptr, "png_set_keep_unknown_chunks: invalid keep");
1521
1522       return;
1523    }
1524
1525    if (num_chunks_in <= 0)
1526    {
1527       png_ptr->unknown_default = keep;
1528
1529       /* '0' means just set the flags, so stop here */
1530       if (num_chunks_in == 0)
1531         return;
1532    }
1533
1534    if (num_chunks_in < 0)
1535    {
1536       /* Ignore all unknown chunks and all chunks recognized by
1537        * libpng except for IHDR, PLTE, tRNS, IDAT, and IEND
1538        */
1539       static const png_byte chunks_to_ignore[] = {
1540          98,  75,  71,  68, '\0',  /* bKGD */
1541          99,  72,  82,  77, '\0',  /* cHRM */
1542         101,  88,  73, 102, '\0',  /* eXIf */
1543         103,  65,  77,  65, '\0',  /* gAMA */
1544         104,  73,  83,  84, '\0',  /* hIST */
1545         105,  67,  67,  80, '\0',  /* iCCP */
1546         105,  84,  88, 116, '\0',  /* iTXt */
1547         111,  70,  70, 115, '\0',  /* oFFs */
1548         112,  67,  65,  76, '\0',  /* pCAL */
1549         112,  72,  89, 115, '\0',  /* pHYs */
1550         115,  66,  73,  84, '\0',  /* sBIT */
1551         115,  67,  65,  76, '\0',  /* sCAL */
1552         115,  80,  76,  84, '\0',  /* sPLT */
1553         115,  84,  69,  82, '\0',  /* sTER */
1554         115,  82,  71,  66, '\0',  /* sRGB */
1555         116,  69,  88, 116, '\0',  /* tEXt */
1556         116,  73,  77,  69, '\0',  /* tIME */
1557         122,  84,  88, 116, '\0'   /* zTXt */
1558       };
1559
1560       chunk_list = chunks_to_ignore;
1561       num_chunks = (unsigned int)/*SAFE*/(sizeof chunks_to_ignore)/5U;
1562    }
1563
1564    else /* num_chunks_in > 0 */
1565    {
1566       if (chunk_list == NULL)
1567       {
1568          /* Prior to 1.6.0 this was silently ignored, now it is an app_error
1569           * which can be switched off.
1570           */
1571          png_app_error(png_ptr, "png_set_keep_unknown_chunks: no chunk list");
1572
1573          return;
1574       }
1575
1576       num_chunks = (unsigned int)num_chunks_in;
1577    }
1578
1579    old_num_chunks = png_ptr->num_chunk_list;
1580    if (png_ptr->chunk_list == NULL)
1581       old_num_chunks = 0;
1582
1583    /* Since num_chunks is always restricted to UINT_MAX/5 this can't overflow.
1584     */
1585    if (num_chunks + old_num_chunks > UINT_MAX/5)
1586    {
1587       png_app_error(png_ptr, "png_set_keep_unknown_chunks: too many chunks");
1588
1589       return;
1590    }
1591
1592    /* If these chunks are being reset to the default then no more memory is
1593     * required because add_one_chunk above doesn't extend the list if the 'keep'
1594     * parameter is the default.
1595     */
1596    if (keep != 0)
1597    {
1598       new_list = png_voidcast(png_bytep, png_malloc(png_ptr,
1599           5 * (num_chunks + old_num_chunks)));
1600
1601       if (old_num_chunks > 0)
1602          memcpy(new_list, png_ptr->chunk_list, 5*old_num_chunks);
1603    }
1604
1605    else if (old_num_chunks > 0)
1606       new_list = png_ptr->chunk_list;
1607
1608    else
1609       new_list = NULL;
1610
1611    /* Add the new chunks together with each one's handling code.  If the chunk
1612     * already exists the code is updated, otherwise the chunk is added to the
1613     * end.  (In libpng 1.6.0 order no longer matters because this code enforces
1614     * the earlier convention that the last setting is the one that is used.)
1615     */
1616    if (new_list != NULL)
1617    {
1618       png_const_bytep inlist;
1619       png_bytep outlist;
1620       unsigned int i;
1621
1622       for (i=0; i<num_chunks; ++i)
1623       {
1624          old_num_chunks = add_one_chunk(new_list, old_num_chunks,
1625              chunk_list+5*i, keep);
1626       }
1627
1628       /* Now remove any spurious 'default' entries. */
1629       num_chunks = 0;
1630       for (i=0, inlist=outlist=new_list; i<old_num_chunks; ++i, inlist += 5)
1631       {
1632          if (inlist[4])
1633          {
1634             if (outlist != inlist)
1635                memcpy(outlist, inlist, 5);
1636             outlist += 5;
1637             ++num_chunks;
1638          }
1639       }
1640
1641       /* This means the application has removed all the specialized handling. */
1642       if (num_chunks == 0)
1643       {
1644          if (png_ptr->chunk_list != new_list)
1645             png_free(png_ptr, new_list);
1646
1647          new_list = NULL;
1648       }
1649    }
1650
1651    else
1652       num_chunks = 0;
1653
1654    png_ptr->num_chunk_list = num_chunks;
1655
1656    if (png_ptr->chunk_list != new_list)
1657    {
1658       if (png_ptr->chunk_list != NULL)
1659          png_free(png_ptr, png_ptr->chunk_list);
1660
1661       png_ptr->chunk_list = new_list;
1662    }
1663 }
1664 #endif
1665
1666 #ifdef PNG_READ_USER_CHUNKS_SUPPORTED
1667 void PNGAPI
1668 png_set_read_user_chunk_fn(png_structrp png_ptr, png_voidp user_chunk_ptr,
1669     png_user_chunk_ptr read_user_chunk_fn)
1670 {
1671    png_debug(1, "in png_set_read_user_chunk_fn");
1672
1673    if (png_ptr == NULL)
1674       return;
1675
1676    png_ptr->read_user_chunk_fn = read_user_chunk_fn;
1677    png_ptr->user_chunk_ptr = user_chunk_ptr;
1678 }
1679 #endif
1680
1681 #ifdef PNG_INFO_IMAGE_SUPPORTED
1682 void PNGAPI
1683 png_set_rows(png_const_structrp png_ptr, png_inforp info_ptr,
1684     png_bytepp row_pointers)
1685 {
1686    png_debug1(1, "in %s storage function", "rows");
1687
1688    if (png_ptr == NULL || info_ptr == NULL)
1689       return;
1690
1691    if (info_ptr->row_pointers != NULL &&
1692        (info_ptr->row_pointers != row_pointers))
1693       png_free_data(png_ptr, info_ptr, PNG_FREE_ROWS, 0);
1694
1695    info_ptr->row_pointers = row_pointers;
1696
1697    if (row_pointers != NULL)
1698       info_ptr->valid |= PNG_INFO_IDAT;
1699 }
1700 #endif
1701
1702 void PNGAPI
1703 png_set_compression_buffer_size(png_structrp png_ptr, size_t size)
1704 {
1705    if (png_ptr == NULL)
1706       return;
1707
1708    if (size == 0 || size > PNG_UINT_31_MAX)
1709       png_error(png_ptr, "invalid compression buffer size");
1710
1711 #  ifdef PNG_SEQUENTIAL_READ_SUPPORTED
1712    if ((png_ptr->mode & PNG_IS_READ_STRUCT) != 0)
1713    {
1714       png_ptr->IDAT_read_size = (png_uint_32)size; /* checked above */
1715       return;
1716    }
1717 #  endif
1718
1719 #  ifdef PNG_WRITE_SUPPORTED
1720    if ((png_ptr->mode & PNG_IS_READ_STRUCT) == 0)
1721    {
1722       if (png_ptr->zowner != 0)
1723       {
1724          png_warning(png_ptr,
1725              "Compression buffer size cannot be changed because it is in use");
1726
1727          return;
1728       }
1729
1730 #ifndef __COVERITY__
1731       /* Some compilers complain that this is always false.  However, it
1732        * can be true when integer overflow happens.
1733        */
1734       if (size > ZLIB_IO_MAX)
1735       {
1736          png_warning(png_ptr,
1737              "Compression buffer size limited to system maximum");
1738          size = ZLIB_IO_MAX; /* must fit */
1739       }
1740 #endif
1741
1742       if (size < 6)
1743       {
1744          /* Deflate will potentially go into an infinite loop on a SYNC_FLUSH
1745           * if this is permitted.
1746           */
1747          png_warning(png_ptr,
1748              "Compression buffer size cannot be reduced below 6");
1749
1750          return;
1751       }
1752
1753       if (png_ptr->zbuffer_size != size)
1754       {
1755          png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list);
1756          png_ptr->zbuffer_size = (uInt)size;
1757       }
1758    }
1759 #  endif
1760 }
1761
1762 void PNGAPI
1763 png_set_invalid(png_const_structrp png_ptr, png_inforp info_ptr, int mask)
1764 {
1765    if (png_ptr != NULL && info_ptr != NULL)
1766       info_ptr->valid &= (unsigned int)(~mask);
1767 }
1768
1769
1770 #ifdef PNG_SET_USER_LIMITS_SUPPORTED
1771 /* This function was added to libpng 1.2.6 */
1772 void PNGAPI
1773 png_set_user_limits(png_structrp png_ptr, png_uint_32 user_width_max,
1774     png_uint_32 user_height_max)
1775 {
1776    /* Images with dimensions larger than these limits will be
1777     * rejected by png_set_IHDR().  To accept any PNG datastream
1778     * regardless of dimensions, set both limits to 0x7fffffff.
1779     */
1780    if (png_ptr == NULL)
1781       return;
1782
1783    png_ptr->user_width_max = user_width_max;
1784    png_ptr->user_height_max = user_height_max;
1785 }
1786
1787 /* This function was added to libpng 1.4.0 */
1788 void PNGAPI
1789 png_set_chunk_cache_max(png_structrp png_ptr, png_uint_32 user_chunk_cache_max)
1790 {
1791    if (png_ptr != NULL)
1792       png_ptr->user_chunk_cache_max = user_chunk_cache_max;
1793 }
1794
1795 /* This function was added to libpng 1.4.1 */
1796 void PNGAPI
1797 png_set_chunk_malloc_max(png_structrp png_ptr,
1798     png_alloc_size_t user_chunk_malloc_max)
1799 {
1800    if (png_ptr != NULL)
1801       png_ptr->user_chunk_malloc_max = user_chunk_malloc_max;
1802 }
1803 #endif /* ?SET_USER_LIMITS */
1804
1805
1806 #ifdef PNG_BENIGN_ERRORS_SUPPORTED
1807 void PNGAPI
1808 png_set_benign_errors(png_structrp png_ptr, int allowed)
1809 {
1810    png_debug(1, "in png_set_benign_errors");
1811
1812    /* If allowed is 1, png_benign_error() is treated as a warning.
1813     *
1814     * If allowed is 0, png_benign_error() is treated as an error (which
1815     * is the default behavior if png_set_benign_errors() is not called).
1816     */
1817
1818    if (allowed != 0)
1819       png_ptr->flags |= PNG_FLAG_BENIGN_ERRORS_WARN |
1820          PNG_FLAG_APP_WARNINGS_WARN | PNG_FLAG_APP_ERRORS_WARN;
1821
1822    else
1823       png_ptr->flags &= ~(PNG_FLAG_BENIGN_ERRORS_WARN |
1824          PNG_FLAG_APP_WARNINGS_WARN | PNG_FLAG_APP_ERRORS_WARN);
1825 }
1826 #endif /* BENIGN_ERRORS */
1827
1828 #ifdef PNG_CHECK_FOR_INVALID_INDEX_SUPPORTED
1829    /* Whether to report invalid palette index; added at libng-1.5.10.
1830     * It is possible for an indexed (color-type==3) PNG file to contain
1831     * pixels with invalid (out-of-range) indexes if the PLTE chunk has
1832     * fewer entries than the image's bit-depth would allow. We recover
1833     * from this gracefully by filling any incomplete palette with zeros
1834     * (opaque black).  By default, when this occurs libpng will issue
1835     * a benign error.  This API can be used to override that behavior.
1836     */
1837 void PNGAPI
1838 png_set_check_for_invalid_index(png_structrp png_ptr, int allowed)
1839 {
1840    png_debug(1, "in png_set_check_for_invalid_index");
1841
1842    if (allowed > 0)
1843       png_ptr->num_palette_max = 0;
1844
1845    else
1846       png_ptr->num_palette_max = -1;
1847 }
1848 #endif
1849
1850 #if defined(PNG_TEXT_SUPPORTED) || defined(PNG_pCAL_SUPPORTED) || \
1851     defined(PNG_iCCP_SUPPORTED) || defined(PNG_sPLT_SUPPORTED)
1852 /* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
1853  * and if invalid, correct the keyword rather than discarding the entire
1854  * chunk.  The PNG 1.0 specification requires keywords 1-79 characters in
1855  * length, forbids leading or trailing whitespace, multiple internal spaces,
1856  * and the non-break space (0x80) from ISO 8859-1.  Returns keyword length.
1857  *
1858  * The 'new_key' buffer must be 80 characters in size (for the keyword plus a
1859  * trailing '\0').  If this routine returns 0 then there was no keyword, or a
1860  * valid one could not be generated, and the caller must png_error.
1861  */
1862 png_uint_32 /* PRIVATE */
1863 png_check_keyword(png_structrp png_ptr, png_const_charp key, png_bytep new_key)
1864 {
1865 #ifdef PNG_WARNINGS_SUPPORTED
1866    png_const_charp orig_key = key;
1867 #endif
1868    png_uint_32 key_len = 0;
1869    int bad_character = 0;
1870    int space = 1;
1871
1872    png_debug(1, "in png_check_keyword");
1873
1874    if (key == NULL)
1875    {
1876       *new_key = 0;
1877       return 0;
1878    }
1879
1880    while (*key && key_len < 79)
1881    {
1882       png_byte ch = (png_byte)*key++;
1883
1884       if ((ch > 32 && ch <= 126) || (ch >= 161 /*&& ch <= 255*/))
1885       {
1886          *new_key++ = ch; ++key_len; space = 0;
1887       }
1888
1889       else if (space == 0)
1890       {
1891          /* A space or an invalid character when one wasn't seen immediately
1892           * before; output just a space.
1893           */
1894          *new_key++ = 32; ++key_len; space = 1;
1895
1896          /* If the character was not a space then it is invalid. */
1897          if (ch != 32)
1898             bad_character = ch;
1899       }
1900
1901       else if (bad_character == 0)
1902          bad_character = ch; /* just skip it, record the first error */
1903    }
1904
1905    if (key_len > 0 && space != 0) /* trailing space */
1906    {
1907       --key_len; --new_key;
1908       if (bad_character == 0)
1909          bad_character = 32;
1910    }
1911
1912    /* Terminate the keyword */
1913    *new_key = 0;
1914
1915    if (key_len == 0)
1916       return 0;
1917
1918 #ifdef PNG_WARNINGS_SUPPORTED
1919    /* Try to only output one warning per keyword: */
1920    if (*key != 0) /* keyword too long */
1921       png_warning(png_ptr, "keyword truncated");
1922
1923    else if (bad_character != 0)
1924    {
1925       PNG_WARNING_PARAMETERS(p)
1926
1927       png_warning_parameter(p, 1, orig_key);
1928       png_warning_parameter_signed(p, 2, PNG_NUMBER_FORMAT_02x, bad_character);
1929
1930       png_formatted_warning(png_ptr, p, "keyword \"@1\": bad character '0x@2'");
1931    }
1932 #else /* !WARNINGS */
1933    PNG_UNUSED(png_ptr)
1934 #endif /* !WARNINGS */
1935
1936    return key_len;
1937 }
1938 #endif /* TEXT || pCAL || iCCP || sPLT */
1939 #endif /* READ || WRITE */