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