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