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