Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(11)

Side by Side Diff: third_party/libpng/pngwutil.c

Issue 1509773004: CL for perf tryjob on android Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « third_party/libpng/pngwtran.c ('k') | tools/run-perf-test.cfg » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 1
2 /* pngwutil.c - utilities to write a PNG file 2 /* pngwutil.c - utilities to write a PNG file
3 * 3 *
4 * Last changed in libpng 1.2.54 [November 12, 2015] 4 * Last changed in libpng 1.6.19 [November 12, 2015]
5 * Copyright (c) 1998-2015 Glenn Randers-Pehrson 5 * Copyright (c) 1998-2015 Glenn Randers-Pehrson
6 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger) 6 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
7 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.) 7 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
8 * 8 *
9 * This code is released under the libpng license. 9 * This code is released under the libpng license.
10 * For conditions of distribution and use, see the disclaimer 10 * For conditions of distribution and use, see the disclaimer
11 * and license in png.h 11 * and license in png.h
12 */ 12 */
13 13
14 #define PNG_INTERNAL 14 #include "pngpriv.h"
15 #define PNG_NO_PEDANTIC_WARNINGS 15
16 #include "png.h"
17 #ifdef PNG_WRITE_SUPPORTED 16 #ifdef PNG_WRITE_SUPPORTED
18 17
18 #ifdef PNG_WRITE_INT_FUNCTIONS_SUPPORTED
19 /* Place a 32-bit number into a buffer in PNG byte order. We work 19 /* Place a 32-bit number into a buffer in PNG byte order. We work
20 * with unsigned numbers for convenience, although one supported 20 * with unsigned numbers for convenience, although one supported
21 * ancillary chunk uses signed (two's complement) numbers. 21 * ancillary chunk uses signed (two's complement) numbers.
22 */ 22 */
23 void PNGAPI 23 void PNGAPI
24 png_save_uint_32(png_bytep buf, png_uint_32 i) 24 png_save_uint_32(png_bytep buf, png_uint_32 i)
25 { 25 {
26 buf[0] = (png_byte)((i >> 24) & 0xff); 26 buf[0] = (png_byte)(i >> 24);
27 buf[1] = (png_byte)((i >> 16) & 0xff); 27 buf[1] = (png_byte)(i >> 16);
28 buf[2] = (png_byte)((i >> 8) & 0xff); 28 buf[2] = (png_byte)(i >> 8);
29 buf[3] = (png_byte)(i & 0xff); 29 buf[3] = (png_byte)(i );
30 }
31
32 /* The png_save_int_32 function assumes integers are stored in two's
33 * complement format. If this isn't the case, then this routine needs to
34 * be modified to write data in two's complement format.
35 */
36 void PNGAPI
37 png_save_int_32(png_bytep buf, png_int_32 i)
38 {
39 buf[0] = (png_byte)((i >> 24) & 0xff);
40 buf[1] = (png_byte)((i >> 16) & 0xff);
41 buf[2] = (png_byte)((i >> 8) & 0xff);
42 buf[3] = (png_byte)(i & 0xff);
43 } 30 }
44 31
45 /* Place a 16-bit number into a buffer in PNG byte order. 32 /* Place a 16-bit number into a buffer in PNG byte order.
46 * The parameter is declared unsigned int, not png_uint_16, 33 * The parameter is declared unsigned int, not png_uint_16,
47 * just to avoid potential problems on pre-ANSI C compilers. 34 * just to avoid potential problems on pre-ANSI C compilers.
48 */ 35 */
49 void PNGAPI 36 void PNGAPI
50 png_save_uint_16(png_bytep buf, unsigned int i) 37 png_save_uint_16(png_bytep buf, unsigned int i)
51 { 38 {
52 buf[0] = (png_byte)((i >> 8) & 0xff); 39 buf[0] = (png_byte)(i >> 8);
53 buf[1] = (png_byte)(i & 0xff); 40 buf[1] = (png_byte)(i );
54 } 41 }
42 #endif
55 43
56 /* Simple function to write the signature. If we have already written 44 /* Simple function to write the signature. If we have already written
57 * the magic bytes of the signature, or more likely, the PNG stream is 45 * the magic bytes of the signature, or more likely, the PNG stream is
58 * being embedded into another stream and doesn't need its own signature, 46 * being embedded into another stream and doesn't need its own signature,
59 * we should call png_set_sig_bytes() to tell libpng how many of the 47 * we should call png_set_sig_bytes() to tell libpng how many of the
60 * bytes have already been written. 48 * bytes have already been written.
61 */ 49 */
62 void /* PRIVATE */ 50 void PNGAPI
63 png_write_sig(png_structp png_ptr) 51 png_write_sig(png_structrp png_ptr)
64 { 52 {
65 png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10}; 53 png_byte png_signature[8] = {137, 80, 78, 71, 13, 10, 26, 10};
66 54
55 #ifdef PNG_IO_STATE_SUPPORTED
56 /* Inform the I/O callback that the signature is being written */
57 png_ptr->io_state = PNG_IO_WRITING | PNG_IO_SIGNATURE;
58 #endif
59
67 /* Write the rest of the 8 byte signature */ 60 /* Write the rest of the 8 byte signature */
68 png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes], 61 png_write_data(png_ptr, &png_signature[png_ptr->sig_bytes],
69 (png_size_t)(8 - png_ptr->sig_bytes)); 62 (png_size_t)(8 - png_ptr->sig_bytes));
63
70 if (png_ptr->sig_bytes < 3) 64 if (png_ptr->sig_bytes < 3)
71 png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE; 65 png_ptr->mode |= PNG_HAVE_PNG_SIGNATURE;
72 } 66 }
73 67
68 /* Write the start of a PNG chunk. The type is the chunk type.
69 * The total_length is the sum of the lengths of all the data you will be
70 * passing in png_write_chunk_data().
71 */
72 static void
73 png_write_chunk_header(png_structrp png_ptr, png_uint_32 chunk_name,
74 png_uint_32 length)
75 {
76 png_byte buf[8];
77
78 #if defined(PNG_DEBUG) && (PNG_DEBUG > 0)
79 PNG_CSTRING_FROM_CHUNK(buf, chunk_name);
80 png_debug2(0, "Writing %s chunk, length = %lu", buf, (unsigned long)length);
81 #endif
82
83 if (png_ptr == NULL)
84 return;
85
86 #ifdef PNG_IO_STATE_SUPPORTED
87 /* Inform the I/O callback that the chunk header is being written.
88 * PNG_IO_CHUNK_HDR requires a single I/O call.
89 */
90 png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_HDR;
91 #endif
92
93 /* Write the length and the chunk name */
94 png_save_uint_32(buf, length);
95 png_save_uint_32(buf + 4, chunk_name);
96 png_write_data(png_ptr, buf, 8);
97
98 /* Put the chunk name into png_ptr->chunk_name */
99 png_ptr->chunk_name = chunk_name;
100
101 /* Reset the crc and run it over the chunk name */
102 png_reset_crc(png_ptr);
103
104 png_calculate_crc(png_ptr, buf + 4, 4);
105
106 #ifdef PNG_IO_STATE_SUPPORTED
107 /* Inform the I/O callback that chunk data will (possibly) be written.
108 * PNG_IO_CHUNK_DATA does NOT require a specific number of I/O calls.
109 */
110 png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_DATA;
111 #endif
112 }
113
114 void PNGAPI
115 png_write_chunk_start(png_structrp png_ptr, png_const_bytep chunk_string,
116 png_uint_32 length)
117 {
118 png_write_chunk_header(png_ptr, PNG_CHUNK_FROM_STRING(chunk_string), length);
119 }
120
121 /* Write the data of a PNG chunk started with png_write_chunk_header().
122 * Note that multiple calls to this function are allowed, and that the
123 * sum of the lengths from these calls *must* add up to the total_length
124 * given to png_write_chunk_header().
125 */
126 void PNGAPI
127 png_write_chunk_data(png_structrp png_ptr, png_const_bytep data,
128 png_size_t length)
129 {
130 /* Write the data, and run the CRC over it */
131 if (png_ptr == NULL)
132 return;
133
134 if (data != NULL && length > 0)
135 {
136 png_write_data(png_ptr, data, length);
137
138 /* Update the CRC after writing the data,
139 * in case the user I/O routine alters it.
140 */
141 png_calculate_crc(png_ptr, data, length);
142 }
143 }
144
145 /* Finish a chunk started with png_write_chunk_header(). */
146 void PNGAPI
147 png_write_chunk_end(png_structrp png_ptr)
148 {
149 png_byte buf[4];
150
151 if (png_ptr == NULL) return;
152
153 #ifdef PNG_IO_STATE_SUPPORTED
154 /* Inform the I/O callback that the chunk CRC is being written.
155 * PNG_IO_CHUNK_CRC requires a single I/O function call.
156 */
157 png_ptr->io_state = PNG_IO_WRITING | PNG_IO_CHUNK_CRC;
158 #endif
159
160 /* Write the crc in a single operation */
161 png_save_uint_32(buf, png_ptr->crc);
162
163 png_write_data(png_ptr, buf, (png_size_t)4);
164 }
165
74 /* Write a PNG chunk all at once. The type is an array of ASCII characters 166 /* Write a PNG chunk all at once. The type is an array of ASCII characters
75 * representing the chunk name. The array must be at least 4 bytes in 167 * representing the chunk name. The array must be at least 4 bytes in
76 * length, and does not need to be null terminated. To be safe, pass the 168 * length, and does not need to be null terminated. To be safe, pass the
77 * pre-defined chunk names here, and if you need a new one, define it 169 * pre-defined chunk names here, and if you need a new one, define it
78 * where the others are defined. The length is the length of the data. 170 * where the others are defined. The length is the length of the data.
79 * All the data must be present. If that is not possible, use the 171 * All the data must be present. If that is not possible, use the
80 * png_write_chunk_start(), png_write_chunk_data(), and png_write_chunk_end() 172 * png_write_chunk_start(), png_write_chunk_data(), and png_write_chunk_end()
81 * functions instead. 173 * functions instead.
82 */ 174 */
83 void PNGAPI 175 static void
84 png_write_chunk(png_structp png_ptr, png_bytep chunk_name, 176 png_write_complete_chunk(png_structrp png_ptr, png_uint_32 chunk_name,
85 png_bytep data, png_size_t length) 177 png_const_bytep data, png_size_t length)
86 { 178 {
87 if (png_ptr == NULL) 179 if (png_ptr == NULL)
88 return; 180 return;
89 png_write_chunk_start(png_ptr, chunk_name, (png_uint_32)length); 181
90 png_write_chunk_data(png_ptr, data, (png_size_t)length); 182 /* On 64-bit architectures 'length' may not fit in a png_uint_32. */
183 if (length > PNG_UINT_31_MAX)
184 png_error(png_ptr, "length exceeds PNG maximum");
185
186 png_write_chunk_header(png_ptr, chunk_name, (png_uint_32)length);
187 png_write_chunk_data(png_ptr, data, length);
91 png_write_chunk_end(png_ptr); 188 png_write_chunk_end(png_ptr);
92 } 189 }
93 190
94 /* Write the start of a PNG chunk. The type is the chunk type. 191 /* This is the API that calls the internal function above. */
95 * The total_length is the sum of the lengths of all the data you will be 192 void PNGAPI
96 * passing in png_write_chunk_data(). 193 png_write_chunk(png_structrp png_ptr, png_const_bytep chunk_string,
194 png_const_bytep data, png_size_t length)
195 {
196 png_write_complete_chunk(png_ptr, PNG_CHUNK_FROM_STRING(chunk_string), data,
197 length);
198 }
199
200 /* This is used below to find the size of an image to pass to png_deflate_claim,
201 * so it only needs to be accurate if the size is less than 16384 bytes (the
202 * point at which a lower LZ window size can be used.)
97 */ 203 */
98 void PNGAPI 204 static png_alloc_size_t
99 png_write_chunk_start(png_structp png_ptr, png_bytep chunk_name, 205 png_image_size(png_structrp png_ptr)
100 png_uint_32 length) 206 {
101 { 207 /* Only return sizes up to the maximum of a png_uint_32; do this by limiting
102 png_byte buf[8]; 208 * the width and height used to 15 bits.
103 209 */
104 png_debug2(0, "Writing %s chunk, length = %lu", chunk_name, 210 png_uint_32 h = png_ptr->height;
105 (unsigned long)length); 211
106 212 if (png_ptr->rowbytes < 32768 && h < 32768)
107 if (png_ptr == NULL) 213 {
108 return; 214 if (png_ptr->interlaced != 0)
109 215 {
110 216 /* Interlacing makes the image larger because of the replication of
111 /* Write the length and the chunk name */ 217 * both the filter byte and the padding to a byte boundary.
112 png_save_uint_32(buf, length); 218 */
113 png_memcpy(buf + 4, chunk_name, 4); 219 png_uint_32 w = png_ptr->width;
114 png_write_data(png_ptr, buf, (png_size_t)8); 220 unsigned int pd = png_ptr->pixel_depth;
115 /* Put the chunk name into png_ptr->chunk_name */ 221 png_alloc_size_t cb_base;
116 png_memcpy(png_ptr->chunk_name, chunk_name, 4); 222 int pass;
117 /* Reset the crc and run it over the chunk name */ 223
118 png_reset_crc(png_ptr); 224 for (cb_base=0, pass=0; pass<=6; ++pass)
119 png_calculate_crc(png_ptr, chunk_name, (png_size_t)4); 225 {
120 } 226 png_uint_32 pw = PNG_PASS_COLS(w, pass);
121 227
122 /* Write the data of a PNG chunk started with png_write_chunk_start(). 228 if (pw > 0)
123 * Note that multiple calls to this function are allowed, and that the 229 cb_base += (PNG_ROWBYTES(pd, pw)+1) * PNG_PASS_ROWS(h, pass);
124 * sum of the lengths from these calls *must* add up to the total_length 230 }
125 * given to png_write_chunk_start(). 231
126 */ 232 return cb_base;
127 void PNGAPI 233 }
128 png_write_chunk_data(png_structp png_ptr, png_bytep data, png_size_t length) 234
129 { 235 else
130 /* Write the data, and run the CRC over it */ 236 return (png_ptr->rowbytes+1) * h;
131 if (png_ptr == NULL) 237 }
132 return; 238
133 if (data != NULL && length > 0) 239 else
134 { 240 return 0xffffffffU;
135 png_write_data(png_ptr, data, length); 241 }
136 /* Update the CRC after writing the data, 242
137 * in case that the user I/O routine alters it. 243 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
138 */ 244 /* This is the code to hack the first two bytes of the deflate stream (the
139 png_calculate_crc(png_ptr, data, length); 245 * deflate header) to correct the windowBits value to match the actual data
140 } 246 * size. Note that the second argument is the *uncompressed* size but the
141 } 247 * first argument is the *compressed* data (and it must be deflate
142 248 * compressed.)
143 /* Finish a chunk started with png_write_chunk_start(). */ 249 */
144 void PNGAPI 250 static void
145 png_write_chunk_end(png_structp png_ptr) 251 optimize_cmf(png_bytep data, png_alloc_size_t data_size)
146 { 252 {
147 png_byte buf[4]; 253 /* Optimize the CMF field in the zlib stream. The resultant zlib stream is
148 254 * still compliant to the stream specification.
149 if (png_ptr == NULL) return; 255 */
150 256 if (data_size <= 16384) /* else windowBits must be 15 */
151 /* Write the crc in a single operation */ 257 {
152 png_save_uint_32(buf, png_ptr->crc); 258 unsigned int z_cmf = data[0]; /* zlib compression method and flags */
153 259
154 png_write_data(png_ptr, buf, (png_size_t)4); 260 if ((z_cmf & 0x0f) == 8 && (z_cmf & 0xf0) <= 0x70)
155 } 261 {
156 262 unsigned int z_cinfo;
157 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_iCCP_SUPPORTED) 263 unsigned int half_z_window_size;
264
265 z_cinfo = z_cmf >> 4;
266 half_z_window_size = 1U << (z_cinfo + 7);
267
268 if (data_size <= half_z_window_size) /* else no change */
269 {
270 unsigned int tmp;
271
272 do
273 {
274 half_z_window_size >>= 1;
275 --z_cinfo;
276 }
277 while (z_cinfo > 0 && data_size <= half_z_window_size);
278
279 z_cmf = (z_cmf & 0x0f) | (z_cinfo << 4);
280
281 data[0] = (png_byte)z_cmf;
282 tmp = data[1] & 0xe0;
283 tmp += 0x1f - ((z_cmf << 8) + tmp) % 0x1f;
284 data[1] = (png_byte)tmp;
285 }
286 }
287 }
288 }
289 #endif /* WRITE_OPTIMIZE_CMF */
290
291 /* Initialize the compressor for the appropriate type of compression. */
292 static int
293 png_deflate_claim(png_structrp png_ptr, png_uint_32 owner,
294 png_alloc_size_t data_size)
295 {
296 if (png_ptr->zowner != 0)
297 {
298 #if defined(PNG_WARNINGS_SUPPORTED) || defined(PNG_ERROR_TEXT_SUPPORTED)
299 char msg[64];
300
301 PNG_STRING_FROM_CHUNK(msg, owner);
302 msg[4] = ':';
303 msg[5] = ' ';
304 PNG_STRING_FROM_CHUNK(msg+6, png_ptr->zowner);
305 /* So the message that results is "<chunk> using zstream"; this is an
306 * internal error, but is very useful for debugging. i18n requirements
307 * are minimal.
308 */
309 (void)png_safecat(msg, (sizeof msg), 10, " using zstream");
310 #endif
311 #if PNG_RELEASE_BUILD
312 png_warning(png_ptr, msg);
313
314 /* Attempt sane error recovery */
315 if (png_ptr->zowner == png_IDAT) /* don't steal from IDAT */
316 {
317 png_ptr->zstream.msg = PNGZ_MSG_CAST("in use by IDAT");
318 return Z_STREAM_ERROR;
319 }
320
321 png_ptr->zowner = 0;
322 #else
323 png_error(png_ptr, msg);
324 #endif
325 }
326
327 {
328 int level = png_ptr->zlib_level;
329 int method = png_ptr->zlib_method;
330 int windowBits = png_ptr->zlib_window_bits;
331 int memLevel = png_ptr->zlib_mem_level;
332 int strategy; /* set below */
333 int ret; /* zlib return code */
334
335 if (owner == png_IDAT)
336 {
337 if ((png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_STRATEGY) != 0)
338 strategy = png_ptr->zlib_strategy;
339
340 else if (png_ptr->do_filter != PNG_FILTER_NONE)
341 strategy = PNG_Z_DEFAULT_STRATEGY;
342
343 else
344 strategy = PNG_Z_DEFAULT_NOFILTER_STRATEGY;
345 }
346
347 else
348 {
349 #ifdef PNG_WRITE_CUSTOMIZE_ZTXT_COMPRESSION_SUPPORTED
350 level = png_ptr->zlib_text_level;
351 method = png_ptr->zlib_text_method;
352 windowBits = png_ptr->zlib_text_window_bits;
353 memLevel = png_ptr->zlib_text_mem_level;
354 strategy = png_ptr->zlib_text_strategy;
355 #else
356 /* If customization is not supported the values all come from the
357 * IDAT values except for the strategy, which is fixed to the
358 * default. (This is the pre-1.6.0 behavior too, although it was
359 * implemented in a very different way.)
360 */
361 strategy = Z_DEFAULT_STRATEGY;
362 #endif
363 }
364
365 /* Adjust 'windowBits' down if larger than 'data_size'; to stop this
366 * happening just pass 32768 as the data_size parameter. Notice that zlib
367 * requires an extra 262 bytes in the window in addition to the data to be
368 * able to see the whole of the data, so if data_size+262 takes us to the
369 * next windowBits size we need to fix up the value later. (Because even
370 * though deflate needs the extra window, inflate does not!)
371 */
372 if (data_size <= 16384)
373 {
374 /* IMPLEMENTATION NOTE: this 'half_window_size' stuff is only here to
375 * work round a Microsoft Visual C misbehavior which, contrary to C-90,
376 * widens the result of the following shift to 64-bits if (and,
377 * apparently, only if) it is used in a test.
378 */
379 unsigned int half_window_size = 1U << (windowBits-1);
380
381 while (data_size + 262 <= half_window_size)
382 {
383 half_window_size >>= 1;
384 --windowBits;
385 }
386 }
387
388 /* Check against the previous initialized values, if any. */
389 if ((png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED) != 0 &&
390 (png_ptr->zlib_set_level != level ||
391 png_ptr->zlib_set_method != method ||
392 png_ptr->zlib_set_window_bits != windowBits ||
393 png_ptr->zlib_set_mem_level != memLevel ||
394 png_ptr->zlib_set_strategy != strategy))
395 {
396 if (deflateEnd(&png_ptr->zstream) != Z_OK)
397 png_warning(png_ptr, "deflateEnd failed (ignored)");
398
399 png_ptr->flags &= ~PNG_FLAG_ZSTREAM_INITIALIZED;
400 }
401
402 /* For safety clear out the input and output pointers (currently zlib
403 * doesn't use them on Init, but it might in the future).
404 */
405 png_ptr->zstream.next_in = NULL;
406 png_ptr->zstream.avail_in = 0;
407 png_ptr->zstream.next_out = NULL;
408 png_ptr->zstream.avail_out = 0;
409
410 /* Now initialize if required, setting the new parameters, otherwise just
411 * to a simple reset to the previous parameters.
412 */
413 if ((png_ptr->flags & PNG_FLAG_ZSTREAM_INITIALIZED) != 0)
414 ret = deflateReset(&png_ptr->zstream);
415
416 else
417 {
418 ret = deflateInit2(&png_ptr->zstream, level, method, windowBits,
419 memLevel, strategy);
420
421 if (ret == Z_OK)
422 png_ptr->flags |= PNG_FLAG_ZSTREAM_INITIALIZED;
423 }
424
425 /* The return code is from either deflateReset or deflateInit2; they have
426 * pretty much the same set of error codes.
427 */
428 if (ret == Z_OK)
429 png_ptr->zowner = owner;
430
431 else
432 png_zstream_error(png_ptr, ret);
433
434 return ret;
435 }
436 }
437
438 /* Clean up (or trim) a linked list of compression buffers. */
439 void /* PRIVATE */
440 png_free_buffer_list(png_structrp png_ptr, png_compression_bufferp *listp)
441 {
442 png_compression_bufferp list = *listp;
443
444 if (list != NULL)
445 {
446 *listp = NULL;
447
448 do
449 {
450 png_compression_bufferp next = list->next;
451
452 png_free(png_ptr, list);
453 list = next;
454 }
455 while (list != NULL);
456 }
457 }
458
459 #ifdef PNG_WRITE_COMPRESSED_TEXT_SUPPORTED
158 /* This pair of functions encapsulates the operation of (a) compressing a 460 /* This pair of functions encapsulates the operation of (a) compressing a
159 * text string, and (b) issuing it later as a series of chunk data writes. 461 * text string, and (b) issuing it later as a series of chunk data writes.
160 * The compression_state structure is shared context for these functions 462 * The compression_state structure is shared context for these functions
161 * set up by the caller in order to make the whole mess thread-safe. 463 * set up by the caller to allow access to the relevant local variables.
464 *
465 * compression_buffer (new in 1.6.0) is just a linked list of zbuffer_size
466 * temporary buffers. From 1.6.0 it is retained in png_struct so that it will
467 * be correctly freed in the event of a write error (previous implementations
468 * just leaked memory.)
162 */ 469 */
163
164 typedef struct 470 typedef struct
165 { 471 {
166 char *input; /* The uncompressed input data */ 472 png_const_bytep input; /* The uncompressed input data */
167 int input_len; /* Its length */ 473 png_alloc_size_t input_len; /* Its length */
168 int num_output_ptr; /* Number of output pointers used */ 474 png_uint_32 output_len; /* Final compressed length */
169 int max_output_ptr; /* Size of output_ptr */ 475 png_byte output[1024]; /* First block of output */
170 png_charpp output_ptr; /* Array of pointers to output */
171 } compression_state; 476 } compression_state;
172 477
173 /* Compress given text into storage in the png_ptr structure */ 478 static void
174 static int /* PRIVATE */ 479 png_text_compress_init(compression_state *comp, png_const_bytep input,
175 png_text_compress(png_structp png_ptr, 480 png_alloc_size_t input_len)
176 png_charp text, png_size_t text_len, int compression, 481 {
177 compression_state *comp) 482 comp->input = input;
483 comp->input_len = input_len;
484 comp->output_len = 0;
485 }
486
487 /* Compress the data in the compression state input */
488 static int
489 png_text_compress(png_structrp png_ptr, png_uint_32 chunk_name,
490 compression_state *comp, png_uint_32 prefix_len)
178 { 491 {
179 int ret; 492 int ret;
180 493
181 comp->num_output_ptr = 0; 494 /* To find the length of the output it is necessary to first compress the
182 comp->max_output_ptr = 0; 495 * input. The result is buffered rather than using the two-pass algorithm
183 comp->output_ptr = NULL; 496 * that is used on the inflate side; deflate is assumed to be slower and a
184 comp->input = NULL; 497 * PNG writer is assumed to have more memory available than a PNG reader.
185 comp->input_len = 0;
186
187 /* We may just want to pass the text right through */
188 if (compression == PNG_TEXT_COMPRESSION_NONE)
189 {
190 comp->input = text;
191 comp->input_len = text_len;
192 return((int)text_len);
193 }
194
195 if (compression >= PNG_TEXT_COMPRESSION_LAST)
196 {
197 #if defined(PNG_STDIO_SUPPORTED) && !defined(_WIN32_WCE)
198 char msg[50];
199 png_snprintf(msg, 50, "Unknown compression type %d", compression);
200 png_warning(png_ptr, msg);
201 #else
202 png_warning(png_ptr, "Unknown compression type");
203 #endif
204 }
205
206 /* We can't write the chunk until we find out how much data we have,
207 * which means we need to run the compressor first and save the
208 * output. This shouldn't be a problem, as the vast majority of
209 * comments should be reasonable, but we will set up an array of
210 * malloc'd pointers to be sure.
211 * 498 *
212 * If we knew the application was well behaved, we could simplify this 499 * IMPLEMENTATION NOTE: the zlib API deflateBound() can be used to find an
213 * greatly by assuming we can always malloc an output buffer large 500 * upper limit on the output size, but it is always bigger than the input
214 * enough to hold the compressed text ((1001 * text_len / 1000) + 12) 501 * size so it is likely to be more efficient to use this linked-list
215 * and malloc this directly. The only time this would be a bad idea is 502 * approach.
216 * if we can't malloc more than 64K and we have 64K of random input
217 * data, or if the input string is incredibly large (although this
218 * wouldn't cause a failure, just a slowdown due to swapping).
219 */ 503 */
220 504 ret = png_deflate_claim(png_ptr, chunk_name, comp->input_len);
221 /* Set up the compression buffers */ 505
222 png_ptr->zstream.avail_in = (uInt)text_len; 506 if (ret != Z_OK)
223 png_ptr->zstream.next_in = (Bytef *)text; 507 return ret;
224 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size; 508
225 png_ptr->zstream.next_out = (Bytef *)png_ptr->zbuf; 509 /* Set up the compression buffers, we need a loop here to avoid overflowing a
226 510 * uInt. Use ZLIB_IO_MAX to limit the input. The output is always limited
227 /* This is the same compression loop as in png_write_row() */ 511 * by the output buffer size, so there is no need to check that. Since this
228 do 512 * is ANSI-C we know that an 'int', hence a uInt, is always at least 16 bits
229 { 513 * in size.
230 /* Compress the data */ 514 */
231 ret = deflate(&png_ptr->zstream, Z_NO_FLUSH); 515 {
232 if (ret != Z_OK) 516 png_compression_bufferp *end = &png_ptr->zbuffer_list;
233 { 517 png_alloc_size_t input_len = comp->input_len; /* may be zero! */
234 /* Error */ 518 png_uint_32 output_len;
235 if (png_ptr->zstream.msg != NULL) 519
236 png_error(png_ptr, png_ptr->zstream.msg); 520 /* zlib updates these for us: */
237 else 521 png_ptr->zstream.next_in = PNGZ_INPUT_CAST(comp->input);
238 png_error(png_ptr, "zlib error"); 522 png_ptr->zstream.avail_in = 0; /* Set below */
239 } 523 png_ptr->zstream.next_out = comp->output;
240 /* Check to see if we need more room */ 524 png_ptr->zstream.avail_out = (sizeof comp->output);
241 if (!(png_ptr->zstream.avail_out)) 525
242 { 526 output_len = png_ptr->zstream.avail_out;
243 /* Make sure the output array has room */ 527
244 if (comp->num_output_ptr >= comp->max_output_ptr) 528 do
529 {
530 uInt avail_in = ZLIB_IO_MAX;
531
532 if (avail_in > input_len)
533 avail_in = (uInt)input_len;
534
535 input_len -= avail_in;
536
537 png_ptr->zstream.avail_in = avail_in;
538
539 if (png_ptr->zstream.avail_out == 0)
245 { 540 {
246 int old_max; 541 png_compression_buffer *next;
247 542
248 old_max = comp->max_output_ptr; 543 /* Chunk data is limited to 2^31 bytes in length, so the prefix
249 comp->max_output_ptr = comp->num_output_ptr + 4; 544 * length must be counted here.
250 if (comp->output_ptr != NULL) 545 */
546 if (output_len + prefix_len > PNG_UINT_31_MAX)
251 { 547 {
252 png_charpp old_ptr; 548 ret = Z_MEM_ERROR;
253 549 break;
254 old_ptr = comp->output_ptr;
255 comp->output_ptr = (png_charpp)png_malloc(png_ptr,
256 (png_uint_32)
257 (comp->max_output_ptr * png_sizeof(png_charp)));
258 png_memcpy(comp->output_ptr, old_ptr, old_max
259 * png_sizeof(png_charp));
260 png_free(png_ptr, old_ptr);
261 } 550 }
262 else 551
263 comp->output_ptr = (png_charpp)png_malloc(png_ptr, 552 /* Need a new (malloc'ed) buffer, but there may be one present
264 (png_uint_32) 553 * already.
265 (comp->max_output_ptr * png_sizeof(png_charp))); 554 */
555 next = *end;
556 if (next == NULL)
557 {
558 next = png_voidcast(png_compression_bufferp, png_malloc_base
559 (png_ptr, PNG_COMPRESSION_BUFFER_SIZE(png_ptr)));
560
561 if (next == NULL)
562 {
563 ret = Z_MEM_ERROR;
564 break;
565 }
566
567 /* Link in this buffer (so that it will be freed later) */
568 next->next = NULL;
569 *end = next;
570 }
571
572 png_ptr->zstream.next_out = next->output;
573 png_ptr->zstream.avail_out = png_ptr->zbuffer_size;
574 output_len += png_ptr->zstream.avail_out;
575
576 /* Move 'end' to the next buffer pointer. */
577 end = &next->next;
266 } 578 }
267 579
268 /* Save the data */ 580 /* Compress the data */
269 comp->output_ptr[comp->num_output_ptr] = 581 ret = deflate(&png_ptr->zstream,
270 (png_charp)png_malloc(png_ptr, 582 input_len > 0 ? Z_NO_FLUSH : Z_FINISH);
271 (png_uint_32)png_ptr->zbuf_size); 583
272 png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf, 584 /* Claw back input data that was not consumed (because avail_in is
273 png_ptr->zbuf_size); 585 * reset above every time round the loop).
274 comp->num_output_ptr++; 586 */
275 587 input_len += png_ptr->zstream.avail_in;
276 /* and reset the buffer */ 588 png_ptr->zstream.avail_in = 0; /* safety */
277 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size; 589 }
278 png_ptr->zstream.next_out = png_ptr->zbuf; 590 while (ret == Z_OK);
279 } 591
280 /* Continue until we don't have any more to compress */ 592 /* There may be some space left in the last output buffer. This needs to
281 } while (png_ptr->zstream.avail_in); 593 * be subtracted from output_len.
282 594 */
283 /* Finish the compression */ 595 output_len -= png_ptr->zstream.avail_out;
284 do 596 png_ptr->zstream.avail_out = 0; /* safety */
285 { 597 comp->output_len = output_len;
286 /* Tell zlib we are finished */ 598
287 ret = deflate(&png_ptr->zstream, Z_FINISH); 599 /* Now double check the output length, put in a custom message if it is
288 600 * too long. Otherwise ensure the z_stream::msg pointer is set to
289 if (ret == Z_OK) 601 * something.
290 { 602 */
291 /* Check to see if we need more room */ 603 if (output_len + prefix_len >= PNG_UINT_31_MAX)
292 if (!(png_ptr->zstream.avail_out)) 604 {
293 { 605 png_ptr->zstream.msg = PNGZ_MSG_CAST("compressed data too long");
294 /* Check to make sure our output array has room */ 606 ret = Z_MEM_ERROR;
295 if (comp->num_output_ptr >= comp->max_output_ptr) 607 }
296 { 608
297 int old_max; 609 else
298 610 png_zstream_error(png_ptr, ret);
299 old_max = comp->max_output_ptr; 611
300 comp->max_output_ptr = comp->num_output_ptr + 4; 612 /* Reset zlib for another zTXt/iTXt or image data */
301 if (comp->output_ptr != NULL) 613 png_ptr->zowner = 0;
302 { 614
303 png_charpp old_ptr; 615 /* The only success case is Z_STREAM_END, input_len must be 0; if not this
304 616 * is an internal error.
305 old_ptr = comp->output_ptr; 617 */
306 /* This could be optimized to realloc() */ 618 if (ret == Z_STREAM_END && input_len == 0)
307 comp->output_ptr = (png_charpp)png_malloc(png_ptr, 619 {
308 (png_uint_32)(comp->max_output_ptr * 620 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
309 png_sizeof(png_charp))); 621 /* Fix up the deflate header, if required */
310 png_memcpy(comp->output_ptr, old_ptr, 622 optimize_cmf(comp->output, comp->input_len);
311 old_max * png_sizeof(png_charp)); 623 #endif
312 png_free(png_ptr, old_ptr); 624 /* But Z_OK is returned, not Z_STREAM_END; this allows the claim
313 } 625 * function above to return Z_STREAM_END on an error (though it never
314 else 626 * does in the current versions of zlib.)
315 comp->output_ptr = (png_charpp)png_malloc(png_ptr, 627 */
316 (png_uint_32)(comp->max_output_ptr * 628 return Z_OK;
317 png_sizeof(png_charp))); 629 }
318 } 630
319 631 else
320 /* Save the data */ 632 return ret;
321 comp->output_ptr[comp->num_output_ptr] = 633 }
322 (png_charp)png_malloc(png_ptr,
323 (png_uint_32)png_ptr->zbuf_size);
324 png_memcpy(comp->output_ptr[comp->num_output_ptr], png_ptr->zbuf,
325 png_ptr->zbuf_size);
326 comp->num_output_ptr++;
327
328 /* and reset the buffer pointers */
329 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
330 png_ptr->zstream.next_out = png_ptr->zbuf;
331 }
332 }
333 else if (ret != Z_STREAM_END)
334 {
335 /* We got an error */
336 if (png_ptr->zstream.msg != NULL)
337 png_error(png_ptr, png_ptr->zstream.msg);
338 else
339 png_error(png_ptr, "zlib error");
340 }
341 } while (ret != Z_STREAM_END);
342
343 /* Text length is number of buffers plus last buffer */
344 text_len = png_ptr->zbuf_size * comp->num_output_ptr;
345 if (png_ptr->zstream.avail_out < png_ptr->zbuf_size)
346 text_len += png_ptr->zbuf_size - (png_size_t)png_ptr->zstream.avail_out;
347
348 return((int)text_len);
349 } 634 }
350 635
351 /* Ship the compressed text out via chunk writes */ 636 /* Ship the compressed text out via chunk writes */
352 static void /* PRIVATE */ 637 static void
353 png_write_compressed_data_out(png_structp png_ptr, compression_state *comp) 638 png_write_compressed_data_out(png_structrp png_ptr, compression_state *comp)
354 { 639 {
355 int i; 640 png_uint_32 output_len = comp->output_len;
356 641 png_const_bytep output = comp->output;
357 /* Handle the no-compression case */ 642 png_uint_32 avail = (sizeof comp->output);
358 if (comp->input) 643 png_compression_buffer *next = png_ptr->zbuffer_list;
359 { 644
360 png_write_chunk_data(png_ptr, (png_bytep)comp->input, 645 for (;;)
361 (png_size_t)comp->input_len); 646 {
362 return; 647 if (avail > output_len)
363 } 648 avail = output_len;
364 649
365 /* Write saved output buffers, if any */ 650 png_write_chunk_data(png_ptr, output, avail);
366 for (i = 0; i < comp->num_output_ptr; i++) 651
367 { 652 output_len -= avail;
368 png_write_chunk_data(png_ptr, (png_bytep)comp->output_ptr[i], 653
369 (png_size_t)png_ptr->zbuf_size); 654 if (output_len == 0 || next == NULL)
370 png_free(png_ptr, comp->output_ptr[i]); 655 break;
371 comp->output_ptr[i]=NULL; 656
372 } 657 avail = png_ptr->zbuffer_size;
373 if (comp->max_output_ptr != 0) 658 output = next->output;
374 png_free(png_ptr, comp->output_ptr); 659 next = next->next;
375 comp->output_ptr=NULL; 660 }
376 /* Write anything left in zbuf */ 661
377 if (png_ptr->zstream.avail_out < (png_uint_32)png_ptr->zbuf_size) 662 /* This is an internal error; 'next' must have been NULL! */
378 png_write_chunk_data(png_ptr, png_ptr->zbuf, 663 if (output_len > 0)
379 (png_size_t)(png_ptr->zbuf_size - png_ptr->zstream.avail_out)); 664 png_error(png_ptr, "error writing ancillary chunked compressed data");
380 665 }
381 /* Reset zlib for another zTXt/iTXt or image data */ 666 #endif /* WRITE_COMPRESSED_TEXT */
382 deflateReset(&png_ptr->zstream); 667
383 png_ptr->zstream.data_type = Z_BINARY; 668 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_pCAL_SUPPORTED) || \
384 } 669 defined(PNG_WRITE_iCCP_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
385 #endif 670 /* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
671 * and if invalid, correct the keyword rather than discarding the entire
672 * chunk. The PNG 1.0 specification requires keywords 1-79 characters in
673 * length, forbids leading or trailing whitespace, multiple internal spaces,
674 * and the non-break space (0x80) from ISO 8859-1. Returns keyword length.
675 *
676 * The 'new_key' buffer must be 80 characters in size (for the keyword plus a
677 * trailing '\0'). If this routine returns 0 then there was no keyword, or a
678 * valid one could not be generated, and the caller must png_error.
679 */
680 static png_uint_32
681 png_check_keyword(png_structrp png_ptr, png_const_charp key, png_bytep new_key)
682 {
683 png_const_charp orig_key = key;
684 png_uint_32 key_len = 0;
685 int bad_character = 0;
686 int space = 1;
687
688 png_debug(1, "in png_check_keyword");
689
690 if (key == NULL)
691 {
692 *new_key = 0;
693 return 0;
694 }
695
696 while (*key && key_len < 79)
697 {
698 png_byte ch = (png_byte)*key++;
699
700 if ((ch > 32 && ch <= 126) || (ch >= 161 /*&& ch <= 255*/))
701 *new_key++ = ch, ++key_len, space = 0;
702
703 else if (space == 0)
704 {
705 /* A space or an invalid character when one wasn't seen immediately
706 * before; output just a space.
707 */
708 *new_key++ = 32, ++key_len, space = 1;
709
710 /* If the character was not a space then it is invalid. */
711 if (ch != 32)
712 bad_character = ch;
713 }
714
715 else if (bad_character == 0)
716 bad_character = ch; /* just skip it, record the first error */
717 }
718
719 if (key_len > 0 && space != 0) /* trailing space */
720 {
721 --key_len, --new_key;
722 if (bad_character == 0)
723 bad_character = 32;
724 }
725
726 /* Terminate the keyword */
727 *new_key = 0;
728
729 if (key_len == 0)
730 return 0;
731
732 #ifdef PNG_WARNINGS_SUPPORTED
733 /* Try to only output one warning per keyword: */
734 if (*key != 0) /* keyword too long */
735 png_warning(png_ptr, "keyword truncated");
736
737 else if (bad_character != 0)
738 {
739 PNG_WARNING_PARAMETERS(p)
740
741 png_warning_parameter(p, 1, orig_key);
742 png_warning_parameter_signed(p, 2, PNG_NUMBER_FORMAT_02x, bad_character);
743
744 png_formatted_warning(png_ptr, p, "keyword \"@1\": bad character '0x@2'");
745 }
746 #endif /* WARNINGS */
747
748 return key_len;
749 }
750 #endif /* WRITE_TEXT || WRITE_pCAL || WRITE_iCCP || WRITE_sPLT */
386 751
387 /* Write the IHDR chunk, and update the png_struct with the necessary 752 /* Write the IHDR chunk, and update the png_struct with the necessary
388 * information. Note that the rest of this code depends upon this 753 * information. Note that the rest of this code depends upon this
389 * information being correct. 754 * information being correct.
390 */ 755 */
391 void /* PRIVATE */ 756 void /* PRIVATE */
392 png_write_IHDR(png_structp png_ptr, png_uint_32 width, png_uint_32 height, 757 png_write_IHDR(png_structrp png_ptr, png_uint_32 width, png_uint_32 height,
393 int bit_depth, int color_type, int compression_type, int filter_type, 758 int bit_depth, int color_type, int compression_type, int filter_type,
394 int interlace_type) 759 int interlace_type)
395 { 760 {
396 #ifdef PNG_USE_LOCAL_ARRAYS
397 PNG_IHDR;
398 #endif
399 int ret;
400
401 png_byte buf[13]; /* Buffer to store the IHDR info */ 761 png_byte buf[13]; /* Buffer to store the IHDR info */
402 762
403 png_debug(1, "in png_write_IHDR"); 763 png_debug(1, "in png_write_IHDR");
404 764
405 /* Check that we have valid input data from the application info */ 765 /* Check that we have valid input data from the application info */
406 switch (color_type) 766 switch (color_type)
407 { 767 {
408 case PNG_COLOR_TYPE_GRAY: 768 case PNG_COLOR_TYPE_GRAY:
409 switch (bit_depth) 769 switch (bit_depth)
410 { 770 {
411 case 1: 771 case 1:
412 case 2: 772 case 2:
413 case 4: 773 case 4:
414 case 8: 774 case 8:
415 case 16: png_ptr->channels = 1; break; 775 #ifdef PNG_WRITE_16BIT_SUPPORTED
416 default: png_error(png_ptr, 776 case 16:
417 "Invalid bit depth for grayscale image"); 777 #endif
778 png_ptr->channels = 1; break;
779
780 default:
781 png_error(png_ptr,
782 "Invalid bit depth for grayscale image");
418 } 783 }
419 break; 784 break;
785
420 case PNG_COLOR_TYPE_RGB: 786 case PNG_COLOR_TYPE_RGB:
787 #ifdef PNG_WRITE_16BIT_SUPPORTED
421 if (bit_depth != 8 && bit_depth != 16) 788 if (bit_depth != 8 && bit_depth != 16)
789 #else
790 if (bit_depth != 8)
791 #endif
422 png_error(png_ptr, "Invalid bit depth for RGB image"); 792 png_error(png_ptr, "Invalid bit depth for RGB image");
793
423 png_ptr->channels = 3; 794 png_ptr->channels = 3;
424 break; 795 break;
796
425 case PNG_COLOR_TYPE_PALETTE: 797 case PNG_COLOR_TYPE_PALETTE:
426 switch (bit_depth) 798 switch (bit_depth)
427 { 799 {
428 case 1: 800 case 1:
429 case 2: 801 case 2:
430 case 4: 802 case 4:
431 case 8: png_ptr->channels = 1; break; 803 case 8:
432 default: png_error(png_ptr, "Invalid bit depth for paletted image"); 804 png_ptr->channels = 1;
805 break;
806
807 default:
808 png_error(png_ptr, "Invalid bit depth for paletted image");
433 } 809 }
434 break; 810 break;
811
435 case PNG_COLOR_TYPE_GRAY_ALPHA: 812 case PNG_COLOR_TYPE_GRAY_ALPHA:
436 if (bit_depth != 8 && bit_depth != 16) 813 if (bit_depth != 8 && bit_depth != 16)
437 png_error(png_ptr, "Invalid bit depth for grayscale+alpha image"); 814 png_error(png_ptr, "Invalid bit depth for grayscale+alpha image");
815
438 png_ptr->channels = 2; 816 png_ptr->channels = 2;
439 break; 817 break;
818
440 case PNG_COLOR_TYPE_RGB_ALPHA: 819 case PNG_COLOR_TYPE_RGB_ALPHA:
820 #ifdef PNG_WRITE_16BIT_SUPPORTED
441 if (bit_depth != 8 && bit_depth != 16) 821 if (bit_depth != 8 && bit_depth != 16)
822 #else
823 if (bit_depth != 8)
824 #endif
442 png_error(png_ptr, "Invalid bit depth for RGBA image"); 825 png_error(png_ptr, "Invalid bit depth for RGBA image");
826
443 png_ptr->channels = 4; 827 png_ptr->channels = 4;
444 break; 828 break;
829
445 default: 830 default:
446 png_error(png_ptr, "Invalid image color type specified"); 831 png_error(png_ptr, "Invalid image color type specified");
447 } 832 }
448 833
449 if (compression_type != PNG_COMPRESSION_TYPE_BASE) 834 if (compression_type != PNG_COMPRESSION_TYPE_BASE)
450 { 835 {
451 png_warning(png_ptr, "Invalid compression type specified"); 836 png_warning(png_ptr, "Invalid compression type specified");
452 compression_type = PNG_COMPRESSION_TYPE_BASE; 837 compression_type = PNG_COMPRESSION_TYPE_BASE;
453 } 838 }
454 839
455 /* Write filter_method 64 (intrapixel differencing) only if 840 /* Write filter_method 64 (intrapixel differencing) only if
456 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and 841 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and
457 * 2. Libpng did not write a PNG signature (this filter_method is only 842 * 2. Libpng did not write a PNG signature (this filter_method is only
458 * used in PNG datastreams that are embedded in MNG datastreams) and 843 * used in PNG datastreams that are embedded in MNG datastreams) and
459 * 3. The application called png_permit_mng_features with a mask that 844 * 3. The application called png_permit_mng_features with a mask that
460 * included PNG_FLAG_MNG_FILTER_64 and 845 * included PNG_FLAG_MNG_FILTER_64 and
461 * 4. The filter_method is 64 and 846 * 4. The filter_method is 64 and
462 * 5. The color_type is RGB or RGBA 847 * 5. The color_type is RGB or RGBA
463 */ 848 */
464 if ( 849 if (
465 #ifdef PNG_MNG_FEATURES_SUPPORTED 850 #ifdef PNG_MNG_FEATURES_SUPPORTED
466 !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) && 851 !((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) != 0 &&
467 ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE) == 0) && 852 ((png_ptr->mode & PNG_HAVE_PNG_SIGNATURE) == 0) &&
468 (color_type == PNG_COLOR_TYPE_RGB || 853 (color_type == PNG_COLOR_TYPE_RGB ||
469 color_type == PNG_COLOR_TYPE_RGB_ALPHA) && 854 color_type == PNG_COLOR_TYPE_RGB_ALPHA) &&
470 (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) && 855 (filter_type == PNG_INTRAPIXEL_DIFFERENCING)) &&
471 #endif 856 #endif
472 filter_type != PNG_FILTER_TYPE_BASE) 857 filter_type != PNG_FILTER_TYPE_BASE)
473 { 858 {
474 png_warning(png_ptr, "Invalid filter type specified"); 859 png_warning(png_ptr, "Invalid filter type specified");
475 filter_type = PNG_FILTER_TYPE_BASE; 860 filter_type = PNG_FILTER_TYPE_BASE;
476 } 861 }
477 862
478 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 863 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
479 if (interlace_type != PNG_INTERLACE_NONE && 864 if (interlace_type != PNG_INTERLACE_NONE &&
480 interlace_type != PNG_INTERLACE_ADAM7) 865 interlace_type != PNG_INTERLACE_ADAM7)
481 { 866 {
482 png_warning(png_ptr, "Invalid interlace type specified"); 867 png_warning(png_ptr, "Invalid interlace type specified");
483 interlace_type = PNG_INTERLACE_ADAM7; 868 interlace_type = PNG_INTERLACE_ADAM7;
484 } 869 }
485 #else 870 #else
486 interlace_type=PNG_INTERLACE_NONE; 871 interlace_type=PNG_INTERLACE_NONE;
487 #endif 872 #endif
488 873
489 /* Save the relevent information */ 874 /* Save the relevant information */
490 png_ptr->bit_depth = (png_byte)bit_depth; 875 png_ptr->bit_depth = (png_byte)bit_depth;
491 png_ptr->color_type = (png_byte)color_type; 876 png_ptr->color_type = (png_byte)color_type;
492 png_ptr->interlaced = (png_byte)interlace_type; 877 png_ptr->interlaced = (png_byte)interlace_type;
493 #ifdef PNG_MNG_FEATURES_SUPPORTED 878 #ifdef PNG_MNG_FEATURES_SUPPORTED
494 png_ptr->filter_type = (png_byte)filter_type; 879 png_ptr->filter_type = (png_byte)filter_type;
495 #endif 880 #endif
496 png_ptr->compression_type = (png_byte)compression_type; 881 png_ptr->compression_type = (png_byte)compression_type;
497 png_ptr->width = width; 882 png_ptr->width = width;
498 png_ptr->height = height; 883 png_ptr->height = height;
499 884
500 png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels); 885 png_ptr->pixel_depth = (png_byte)(bit_depth * png_ptr->channels);
501 png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width); 886 png_ptr->rowbytes = PNG_ROWBYTES(png_ptr->pixel_depth, width);
502 /* Set the usr info, so any transformations can modify it */ 887 /* Set the usr info, so any transformations can modify it */
503 png_ptr->usr_width = png_ptr->width; 888 png_ptr->usr_width = png_ptr->width;
504 png_ptr->usr_bit_depth = png_ptr->bit_depth; 889 png_ptr->usr_bit_depth = png_ptr->bit_depth;
505 png_ptr->usr_channels = png_ptr->channels; 890 png_ptr->usr_channels = png_ptr->channels;
506 891
507 /* Pack the header information into the buffer */ 892 /* Pack the header information into the buffer */
508 png_save_uint_32(buf, width); 893 png_save_uint_32(buf, width);
509 png_save_uint_32(buf + 4, height); 894 png_save_uint_32(buf + 4, height);
510 buf[8] = (png_byte)bit_depth; 895 buf[8] = (png_byte)bit_depth;
511 buf[9] = (png_byte)color_type; 896 buf[9] = (png_byte)color_type;
512 buf[10] = (png_byte)compression_type; 897 buf[10] = (png_byte)compression_type;
513 buf[11] = (png_byte)filter_type; 898 buf[11] = (png_byte)filter_type;
514 buf[12] = (png_byte)interlace_type; 899 buf[12] = (png_byte)interlace_type;
515 900
516 /* Write the chunk */ 901 /* Write the chunk */
517 png_write_chunk(png_ptr, (png_bytep)png_IHDR, buf, (png_size_t)13); 902 png_write_complete_chunk(png_ptr, png_IHDR, buf, (png_size_t)13);
518 903
519 /* Initialize zlib with PNG info */ 904 if ((png_ptr->do_filter) == PNG_NO_FILTERS)
520 png_ptr->zstream.zalloc = png_zalloc;
521 png_ptr->zstream.zfree = png_zfree;
522 png_ptr->zstream.opaque = (voidpf)png_ptr;
523 if (!(png_ptr->do_filter))
524 { 905 {
525 if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE || 906 if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE ||
526 png_ptr->bit_depth < 8) 907 png_ptr->bit_depth < 8)
527 png_ptr->do_filter = PNG_FILTER_NONE; 908 png_ptr->do_filter = PNG_FILTER_NONE;
909
528 else 910 else
529 png_ptr->do_filter = PNG_ALL_FILTERS; 911 png_ptr->do_filter = PNG_ALL_FILTERS;
530 } 912 }
531 if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_STRATEGY))
532 {
533 if (png_ptr->do_filter != PNG_FILTER_NONE)
534 png_ptr->zlib_strategy = Z_FILTERED;
535 else
536 png_ptr->zlib_strategy = Z_DEFAULT_STRATEGY;
537 }
538 if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_LEVEL))
539 png_ptr->zlib_level = Z_DEFAULT_COMPRESSION;
540 if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL))
541 png_ptr->zlib_mem_level = 8;
542 if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS))
543 png_ptr->zlib_window_bits = 15;
544 if (!(png_ptr->flags & PNG_FLAG_ZLIB_CUSTOM_METHOD))
545 png_ptr->zlib_method = 8;
546 ret = deflateInit2(&png_ptr->zstream, png_ptr->zlib_level,
547 png_ptr->zlib_method, png_ptr->zlib_window_bits,
548 png_ptr->zlib_mem_level, png_ptr->zlib_strategy);
549 if (ret != Z_OK)
550 {
551 if (ret == Z_VERSION_ERROR) png_error(png_ptr,
552 "zlib failed to initialize compressor -- version error");
553 if (ret == Z_STREAM_ERROR) png_error(png_ptr,
554 "zlib failed to initialize compressor -- stream error");
555 if (ret == Z_MEM_ERROR) png_error(png_ptr,
556 "zlib failed to initialize compressor -- mem error");
557 png_error(png_ptr, "zlib failed to initialize compressor");
558 }
559 png_ptr->zstream.next_out = png_ptr->zbuf;
560 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
561 /* libpng is not interested in zstream.data_type */
562 /* Set it to a predefined value, to avoid its evaluation inside zlib */
563 png_ptr->zstream.data_type = Z_BINARY;
564 913
565 png_ptr->mode = PNG_HAVE_IHDR; 914 png_ptr->mode = PNG_HAVE_IHDR; /* not READY_FOR_ZTXT */
566 } 915 }
567 916
568 /* Write the palette. We are careful not to trust png_color to be in the 917 /* Write the palette. We are careful not to trust png_color to be in the
569 * correct order for PNG, so people can redefine it to any convenient 918 * correct order for PNG, so people can redefine it to any convenient
570 * structure. 919 * structure.
571 */ 920 */
572 void /* PRIVATE */ 921 void /* PRIVATE */
573 png_write_PLTE(png_structp png_ptr, png_colorp palette, png_uint_32 num_pal) 922 png_write_PLTE(png_structrp png_ptr, png_const_colorp palette,
923 png_uint_32 num_pal)
574 { 924 {
575 #ifdef PNG_USE_LOCAL_ARRAYS
576 PNG_PLTE;
577 #endif
578 png_uint_32 max_palette_length, i; 925 png_uint_32 max_palette_length, i;
579 png_colorp pal_ptr; 926 png_const_colorp pal_ptr;
580 png_byte buf[3]; 927 png_byte buf[3];
581 928
582 png_debug(1, "in png_write_PLTE"); 929 png_debug(1, "in png_write_PLTE");
583 930
584 max_palette_length = (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE) ? 931 max_palette_length = (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE) ?
585 (1 << png_ptr->bit_depth) : PNG_MAX_PALETTE_LENGTH; 932 (1 << png_ptr->bit_depth) : PNG_MAX_PALETTE_LENGTH;
586 933
587 if (( 934 if ((
588 #ifdef PNG_MNG_FEATURES_SUPPORTED 935 #ifdef PNG_MNG_FEATURES_SUPPORTED
589 !(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) && 936 (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0 &&
590 #endif 937 #endif
591 num_pal == 0) || num_pal > max_palette_length) 938 num_pal == 0) || num_pal > max_palette_length)
592 { 939 {
593 if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE) 940 if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
594 { 941 {
595 png_error(png_ptr, "Invalid number of colors in palette"); 942 png_error(png_ptr, "Invalid number of colors in palette");
596 } 943 }
597 else 944
598 { 945 else
599 png_warning(png_ptr, "Invalid number of colors in palette"); 946 {
600 return; 947 png_warning(png_ptr, "Invalid number of colors in palette");
601 } 948 return;
949 }
602 } 950 }
603 951
604 if (!(png_ptr->color_type&PNG_COLOR_MASK_COLOR)) 952 if ((png_ptr->color_type & PNG_COLOR_MASK_COLOR) == 0)
605 { 953 {
606 png_warning(png_ptr, 954 png_warning(png_ptr,
607 "Ignoring request to write a PLTE chunk in grayscale PNG"); 955 "Ignoring request to write a PLTE chunk in grayscale PNG");
956
608 return; 957 return;
609 } 958 }
610 959
611 png_ptr->num_palette = (png_uint_16)num_pal; 960 png_ptr->num_palette = (png_uint_16)num_pal;
612 png_debug1(3, "num_palette = %d", png_ptr->num_palette); 961 png_debug1(3, "num_palette = %d", png_ptr->num_palette);
613 962
614 png_write_chunk_start(png_ptr, (png_bytep)png_PLTE, 963 png_write_chunk_header(png_ptr, png_PLTE, (png_uint_32)(num_pal * 3));
615 (png_uint_32)(num_pal * 3));
616 #ifdef PNG_POINTER_INDEXING_SUPPORTED 964 #ifdef PNG_POINTER_INDEXING_SUPPORTED
965
617 for (i = 0, pal_ptr = palette; i < num_pal; i++, pal_ptr++) 966 for (i = 0, pal_ptr = palette; i < num_pal; i++, pal_ptr++)
618 { 967 {
619 buf[0] = pal_ptr->red; 968 buf[0] = pal_ptr->red;
620 buf[1] = pal_ptr->green; 969 buf[1] = pal_ptr->green;
621 buf[2] = pal_ptr->blue; 970 buf[2] = pal_ptr->blue;
622 png_write_chunk_data(png_ptr, buf, (png_size_t)3); 971 png_write_chunk_data(png_ptr, buf, (png_size_t)3);
623 } 972 }
973
624 #else 974 #else
625 /* This is a little slower but some buggy compilers need to do this 975 /* This is a little slower but some buggy compilers need to do this
626 * instead 976 * instead
627 */ 977 */
628 pal_ptr=palette; 978 pal_ptr=palette;
979
629 for (i = 0; i < num_pal; i++) 980 for (i = 0; i < num_pal; i++)
630 { 981 {
631 buf[0] = pal_ptr[i].red; 982 buf[0] = pal_ptr[i].red;
632 buf[1] = pal_ptr[i].green; 983 buf[1] = pal_ptr[i].green;
633 buf[2] = pal_ptr[i].blue; 984 buf[2] = pal_ptr[i].blue;
634 png_write_chunk_data(png_ptr, buf, (png_size_t)3); 985 png_write_chunk_data(png_ptr, buf, (png_size_t)3);
635 } 986 }
987
636 #endif 988 #endif
637 png_write_chunk_end(png_ptr); 989 png_write_chunk_end(png_ptr);
638 png_ptr->mode |= PNG_HAVE_PLTE; 990 png_ptr->mode |= PNG_HAVE_PLTE;
639 } 991 }
640 992
641 /* Write an IDAT chunk */ 993 /* This is similar to png_text_compress, above, except that it does not require
642 void /* PRIVATE */ 994 * all of the data at once and, instead of buffering the compressed result,
643 png_write_IDAT(png_structp png_ptr, png_bytep data, png_size_t length) 995 * writes it as IDAT chunks. Unlike png_text_compress it *can* png_error out
644 { 996 * because it calls the write interface. As a result it does its own error
645 #ifdef PNG_USE_LOCAL_ARRAYS 997 * reporting and does not return an error code. In the event of error it will
646 PNG_IDAT; 998 * just call png_error. The input data length may exceed 32-bits. The 'flush'
647 #endif 999 * parameter is exactly the same as that to deflate, with the following
648 1000 * meanings:
649 png_debug(1, "in png_write_IDAT"); 1001 *
650 1002 * Z_NO_FLUSH: normal incremental output of compressed data
651 /* Optimize the CMF field in the zlib stream. */ 1003 * Z_SYNC_FLUSH: do a SYNC_FLUSH, used by png_write_flush
652 /* This hack of the zlib stream is compliant to the stream specification. */ 1004 * Z_FINISH: this is the end of the input, do a Z_FINISH and clean up
653 if (!(png_ptr->mode & PNG_HAVE_IDAT) && 1005 *
654 png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE) 1006 * The routine manages the acquire and release of the png_ptr->zstream by
1007 * checking and (at the end) clearing png_ptr->zowner; it does some sanity
1008 * checks on the 'mode' flags while doing this.
1009 */
1010 void /* PRIVATE */
1011 png_compress_IDAT(png_structrp png_ptr, png_const_bytep input,
1012 png_alloc_size_t input_len, int flush)
1013 {
1014 if (png_ptr->zowner != png_IDAT)
655 { 1015 {
656 unsigned int z_cmf = data[0]; /* zlib compression method and flags */ 1016 /* First time. Ensure we have a temporary buffer for compression and
657 if ((z_cmf & 0x0f) == 8 && (z_cmf & 0xf0) <= 0x70) 1017 * trim the buffer list if it has more than one entry to free memory.
658 { 1018 * If 'WRITE_COMPRESSED_TEXT' is not set the list will never have been
659 /* Avoid memory underflows and multiplication overflows. 1019 * created at this point, but the check here is quick and safe.
660 * 1020 */
661 * The conditions below are practically always satisfied; 1021 if (png_ptr->zbuffer_list == NULL)
662 * however, they still must be checked. 1022 {
663 */ 1023 png_ptr->zbuffer_list = png_voidcast(png_compression_bufferp,
664 if (length >= 2 && 1024 png_malloc(png_ptr, PNG_COMPRESSION_BUFFER_SIZE(png_ptr)));
665 png_ptr->height < 16384 && png_ptr->width < 16384) 1025 png_ptr->zbuffer_list->next = NULL;
1026 }
1027
1028 else
1029 png_free_buffer_list(png_ptr, &png_ptr->zbuffer_list->next);
1030
1031 /* It is a terminal error if we can't claim the zstream. */
1032 if (png_deflate_claim(png_ptr, png_IDAT, png_image_size(png_ptr)) != Z_OK)
1033 png_error(png_ptr, png_ptr->zstream.msg);
1034
1035 /* The output state is maintained in png_ptr->zstream, so it must be
1036 * initialized here after the claim.
1037 */
1038 png_ptr->zstream.next_out = png_ptr->zbuffer_list->output;
1039 png_ptr->zstream.avail_out = png_ptr->zbuffer_size;
1040 }
1041
1042 /* Now loop reading and writing until all the input is consumed or an error
1043 * terminates the operation. The _out values are maintained across calls to
1044 * this function, but the input must be reset each time.
1045 */
1046 png_ptr->zstream.next_in = PNGZ_INPUT_CAST(input);
1047 png_ptr->zstream.avail_in = 0; /* set below */
1048 for (;;)
1049 {
1050 int ret;
1051
1052 /* INPUT: from the row data */
1053 uInt avail = ZLIB_IO_MAX;
1054
1055 if (avail > input_len)
1056 avail = (uInt)input_len; /* safe because of the check */
1057
1058 png_ptr->zstream.avail_in = avail;
1059 input_len -= avail;
1060
1061 ret = deflate(&png_ptr->zstream, input_len > 0 ? Z_NO_FLUSH : flush);
1062
1063 /* Include as-yet unconsumed input */
1064 input_len += png_ptr->zstream.avail_in;
1065 png_ptr->zstream.avail_in = 0;
1066
1067 /* OUTPUT: write complete IDAT chunks when avail_out drops to zero. Note
1068 * that these two zstream fields are preserved across the calls, therefore
1069 * there is no need to set these up on entry to the loop.
1070 */
1071 if (png_ptr->zstream.avail_out == 0)
1072 {
1073 png_bytep data = png_ptr->zbuffer_list->output;
1074 uInt size = png_ptr->zbuffer_size;
1075
1076 /* Write an IDAT containing the data then reset the buffer. The
1077 * first IDAT may need deflate header optimization.
1078 */
1079 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
1080 if ((png_ptr->mode & PNG_HAVE_IDAT) == 0 &&
1081 png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
1082 optimize_cmf(data, png_image_size(png_ptr));
1083 #endif
1084
1085 png_write_complete_chunk(png_ptr, png_IDAT, data, size);
1086 png_ptr->mode |= PNG_HAVE_IDAT;
1087
1088 png_ptr->zstream.next_out = data;
1089 png_ptr->zstream.avail_out = size;
1090
1091 /* For SYNC_FLUSH or FINISH it is essential to keep calling zlib with
1092 * the same flush parameter until it has finished output, for NO_FLUSH
1093 * it doesn't matter.
1094 */
1095 if (ret == Z_OK && flush != Z_NO_FLUSH)
1096 continue;
1097 }
1098
1099 /* The order of these checks doesn't matter much; it just affects which
1100 * possible error might be detected if multiple things go wrong at once.
1101 */
1102 if (ret == Z_OK) /* most likely return code! */
1103 {
1104 /* If all the input has been consumed then just return. If Z_FINISH
1105 * was used as the flush parameter something has gone wrong if we get
1106 * here.
1107 */
1108 if (input_len == 0)
666 { 1109 {
667 png_uint_32 uncompressed_idat_size = png_ptr->height * 1110 if (flush == Z_FINISH)
668 ((png_ptr->width * 1111 png_error(png_ptr, "Z_OK on Z_FINISH with output space");
669 png_ptr->channels * png_ptr->bit_depth + 15) >> 3); 1112
670 unsigned int z_cinfo = z_cmf >> 4; 1113 return;
671 unsigned int half_z_window_size = 1 << (z_cinfo + 7);
672 while (uncompressed_idat_size <= half_z_window_size &&
673 half_z_window_size >= 256)
674 {
675 z_cinfo--;
676 half_z_window_size >>= 1;
677 }
678 z_cmf = (z_cmf & 0x0f) | (z_cinfo << 4);
679 if (data[0] != (png_byte)z_cmf)
680 {
681 data[0] = (png_byte)z_cmf;
682 data[1] &= 0xe0;
683 data[1] += (png_byte)(0x1f - ((z_cmf << 8) + data[1]) % 0x1f);
684 }
685 } 1114 }
686 } 1115 }
1116
1117 else if (ret == Z_STREAM_END && flush == Z_FINISH)
1118 {
1119 /* This is the end of the IDAT data; any pending output must be
1120 * flushed. For small PNG files we may still be at the beginning.
1121 */
1122 png_bytep data = png_ptr->zbuffer_list->output;
1123 uInt size = png_ptr->zbuffer_size - png_ptr->zstream.avail_out;
1124
1125 #ifdef PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
1126 if ((png_ptr->mode & PNG_HAVE_IDAT) == 0 &&
1127 png_ptr->compression_type == PNG_COMPRESSION_TYPE_BASE)
1128 optimize_cmf(data, png_image_size(png_ptr));
1129 #endif
1130
1131 png_write_complete_chunk(png_ptr, png_IDAT, data, size);
1132 png_ptr->zstream.avail_out = 0;
1133 png_ptr->zstream.next_out = NULL;
1134 png_ptr->mode |= PNG_HAVE_IDAT | PNG_AFTER_IDAT;
1135
1136 png_ptr->zowner = 0; /* Release the stream */
1137 return;
1138 }
1139
687 else 1140 else
688 png_error(png_ptr, 1141 {
689 "Invalid zlib compression method or flags in IDAT"); 1142 /* This is an error condition. */
690 } 1143 png_zstream_error(png_ptr, ret);
691 1144 png_error(png_ptr, png_ptr->zstream.msg);
692 png_write_chunk(png_ptr, (png_bytep)png_IDAT, data, length); 1145 }
693 png_ptr->mode |= PNG_HAVE_IDAT; 1146 }
694 } 1147 }
695 1148
696 /* Write an IEND chunk */ 1149 /* Write an IEND chunk */
697 void /* PRIVATE */ 1150 void /* PRIVATE */
698 png_write_IEND(png_structp png_ptr) 1151 png_write_IEND(png_structrp png_ptr)
699 { 1152 {
700 #ifdef PNG_USE_LOCAL_ARRAYS
701 PNG_IEND;
702 #endif
703
704 png_debug(1, "in png_write_IEND"); 1153 png_debug(1, "in png_write_IEND");
705 1154
706 png_write_chunk(png_ptr, (png_bytep)png_IEND, png_bytep_NULL, 1155 png_write_complete_chunk(png_ptr, png_IEND, NULL, (png_size_t)0);
707 (png_size_t)0);
708 png_ptr->mode |= PNG_HAVE_IEND; 1156 png_ptr->mode |= PNG_HAVE_IEND;
709 } 1157 }
710 1158
711 #ifdef PNG_WRITE_gAMA_SUPPORTED 1159 #ifdef PNG_WRITE_gAMA_SUPPORTED
712 /* Write a gAMA chunk */ 1160 /* Write a gAMA chunk */
713 #ifdef PNG_FLOATING_POINT_SUPPORTED 1161 void /* PRIVATE */
714 void /* PRIVATE */ 1162 png_write_gAMA_fixed(png_structrp png_ptr, png_fixed_point file_gamma)
715 png_write_gAMA(png_structp png_ptr, double file_gamma) 1163 {
716 {
717 #ifdef PNG_USE_LOCAL_ARRAYS
718 PNG_gAMA;
719 #endif
720 png_uint_32 igamma;
721 png_byte buf[4];
722
723 png_debug(1, "in png_write_gAMA");
724
725 /* file_gamma is saved in 1/100,000ths */
726 igamma = (png_uint_32)(file_gamma * 100000.0 + 0.5);
727 png_save_uint_32(buf, igamma);
728 png_write_chunk(png_ptr, (png_bytep)png_gAMA, buf, (png_size_t)4);
729 }
730 #endif
731 #ifdef PNG_FIXED_POINT_SUPPORTED
732 void /* PRIVATE */
733 png_write_gAMA_fixed(png_structp png_ptr, png_fixed_point file_gamma)
734 {
735 #ifdef PNG_USE_LOCAL_ARRAYS
736 PNG_gAMA;
737 #endif
738 png_byte buf[4]; 1164 png_byte buf[4];
739 1165
740 png_debug(1, "in png_write_gAMA"); 1166 png_debug(1, "in png_write_gAMA");
741 1167
742 /* file_gamma is saved in 1/100,000ths */ 1168 /* file_gamma is saved in 1/100,000ths */
743 png_save_uint_32(buf, (png_uint_32)file_gamma); 1169 png_save_uint_32(buf, (png_uint_32)file_gamma);
744 png_write_chunk(png_ptr, (png_bytep)png_gAMA, buf, (png_size_t)4); 1170 png_write_complete_chunk(png_ptr, png_gAMA, buf, (png_size_t)4);
745 } 1171 }
746 #endif
747 #endif 1172 #endif
748 1173
749 #ifdef PNG_WRITE_sRGB_SUPPORTED 1174 #ifdef PNG_WRITE_sRGB_SUPPORTED
750 /* Write a sRGB chunk */ 1175 /* Write a sRGB chunk */
751 void /* PRIVATE */ 1176 void /* PRIVATE */
752 png_write_sRGB(png_structp png_ptr, int srgb_intent) 1177 png_write_sRGB(png_structrp png_ptr, int srgb_intent)
753 { 1178 {
754 #ifdef PNG_USE_LOCAL_ARRAYS
755 PNG_sRGB;
756 #endif
757 png_byte buf[1]; 1179 png_byte buf[1];
758 1180
759 png_debug(1, "in png_write_sRGB"); 1181 png_debug(1, "in png_write_sRGB");
760 1182
761 if (srgb_intent >= PNG_sRGB_INTENT_LAST) 1183 if (srgb_intent >= PNG_sRGB_INTENT_LAST)
762 png_warning(png_ptr, 1184 png_warning(png_ptr,
763 "Invalid sRGB rendering intent specified"); 1185 "Invalid sRGB rendering intent specified");
1186
764 buf[0]=(png_byte)srgb_intent; 1187 buf[0]=(png_byte)srgb_intent;
765 png_write_chunk(png_ptr, (png_bytep)png_sRGB, buf, (png_size_t)1); 1188 png_write_complete_chunk(png_ptr, png_sRGB, buf, (png_size_t)1);
766 } 1189 }
767 #endif 1190 #endif
768 1191
769 #ifdef PNG_WRITE_iCCP_SUPPORTED 1192 #ifdef PNG_WRITE_iCCP_SUPPORTED
770 /* Write an iCCP chunk */ 1193 /* Write an iCCP chunk */
771 void /* PRIVATE */ 1194 void /* PRIVATE */
772 png_write_iCCP(png_structp png_ptr, png_charp name, int compression_type, 1195 png_write_iCCP(png_structrp png_ptr, png_const_charp name,
773 png_charp profile, int profile_len) 1196 png_const_bytep profile)
774 { 1197 {
775 #ifdef PNG_USE_LOCAL_ARRAYS 1198 png_uint_32 name_len;
776 PNG_iCCP; 1199 png_uint_32 profile_len;
777 #endif 1200 png_byte new_name[81]; /* 1 byte for the compression byte */
778 png_size_t name_len;
779 png_charp new_name;
780 compression_state comp; 1201 compression_state comp;
781 int embedded_profile_len = 0; 1202 png_uint_32 temp;
782 1203
783 png_debug(1, "in png_write_iCCP"); 1204 png_debug(1, "in png_write_iCCP");
784 1205
785 comp.num_output_ptr = 0; 1206 /* These are all internal problems: the profile should have been checked
786 comp.max_output_ptr = 0; 1207 * before when it was stored.
787 comp.output_ptr = NULL; 1208 */
788 comp.input = NULL;
789 comp.input_len = 0;
790
791 if ((name_len = png_check_keyword(png_ptr, name,
792 &new_name)) == 0)
793 return;
794
795 if (compression_type != PNG_COMPRESSION_TYPE_BASE)
796 png_warning(png_ptr, "Unknown compression type in iCCP chunk");
797
798 if (profile == NULL) 1209 if (profile == NULL)
799 profile_len = 0; 1210 png_error(png_ptr, "No profile for iCCP chunk"); /* internal error */
800 1211
801 if (profile_len > 3) 1212 profile_len = png_get_uint_32(profile);
802 embedded_profile_len = 1213
803 ((*( (png_bytep)profile ))<<24) | 1214 if (profile_len < 132)
804 ((*( (png_bytep)profile + 1))<<16) | 1215 png_error(png_ptr, "ICC profile too short");
805 ((*( (png_bytep)profile + 2))<< 8) | 1216
806 ((*( (png_bytep)profile + 3)) ); 1217 temp = (png_uint_32) (*(profile+8));
807 1218 if (temp > 3 && (profile_len & 0x03))
808 if (embedded_profile_len < 0) 1219 png_error(png_ptr, "ICC profile length invalid (not a multiple of 4)");
1220
809 { 1221 {
810 png_warning(png_ptr, 1222 png_uint_32 embedded_profile_len = png_get_uint_32(profile);
811 "Embedded profile length in iCCP chunk is negative"); 1223
812 png_free(png_ptr, new_name); 1224 if (profile_len != embedded_profile_len)
813 return; 1225 png_error(png_ptr, "Profile length does not match profile");
814 } 1226 }
815 1227
816 if (profile_len < embedded_profile_len) 1228 name_len = png_check_keyword(png_ptr, name, new_name);
817 { 1229
818 png_warning(png_ptr, 1230 if (name_len == 0)
819 "Embedded profile length too large in iCCP chunk"); 1231 png_error(png_ptr, "iCCP: invalid keyword");
820 png_free(png_ptr, new_name); 1232
821 return; 1233 new_name[++name_len] = PNG_COMPRESSION_TYPE_BASE;
822 }
823
824 if (profile_len > embedded_profile_len)
825 {
826 png_warning(png_ptr,
827 "Truncating profile to actual length in iCCP chunk");
828 profile_len = embedded_profile_len;
829 }
830
831 if (profile_len)
832 profile_len = png_text_compress(png_ptr, profile,
833 (png_size_t)profile_len, PNG_COMPRESSION_TYPE_BASE, &comp);
834 1234
835 /* Make sure we include the NULL after the name and the compression type */ 1235 /* Make sure we include the NULL after the name and the compression type */
836 png_write_chunk_start(png_ptr, (png_bytep)png_iCCP, 1236 ++name_len;
837 (png_uint_32)(name_len + profile_len + 2)); 1237
838 new_name[name_len + 1] = 0x00; 1238 png_text_compress_init(&comp, profile, profile_len);
839 png_write_chunk_data(png_ptr, (png_bytep)new_name, 1239
840 (png_size_t)(name_len + 2)); 1240 /* Allow for keyword terminator and compression byte */
841 1241 if (png_text_compress(png_ptr, png_iCCP, &comp, name_len) != Z_OK)
842 if (profile_len) 1242 png_error(png_ptr, png_ptr->zstream.msg);
843 png_write_compressed_data_out(png_ptr, &comp); 1243
1244 png_write_chunk_header(png_ptr, png_iCCP, name_len + comp.output_len);
1245
1246 png_write_chunk_data(png_ptr, new_name, name_len);
1247
1248 png_write_compressed_data_out(png_ptr, &comp);
844 1249
845 png_write_chunk_end(png_ptr); 1250 png_write_chunk_end(png_ptr);
846 png_free(png_ptr, new_name);
847 } 1251 }
848 #endif 1252 #endif
849 1253
850 #ifdef PNG_WRITE_sPLT_SUPPORTED 1254 #ifdef PNG_WRITE_sPLT_SUPPORTED
851 /* Write a sPLT chunk */ 1255 /* Write a sPLT chunk */
852 void /* PRIVATE */ 1256 void /* PRIVATE */
853 png_write_sPLT(png_structp png_ptr, png_sPLT_tp spalette) 1257 png_write_sPLT(png_structrp png_ptr, png_const_sPLT_tp spalette)
854 { 1258 {
855 #ifdef PNG_USE_LOCAL_ARRAYS 1259 png_uint_32 name_len;
856 PNG_sPLT; 1260 png_byte new_name[80];
857 #endif
858 png_size_t name_len;
859 png_charp new_name;
860 png_byte entrybuf[10]; 1261 png_byte entrybuf[10];
861 int entry_size = (spalette->depth == 8 ? 6 : 10); 1262 png_size_t entry_size = (spalette->depth == 8 ? 6 : 10);
862 int palette_size = entry_size * spalette->nentries; 1263 png_size_t palette_size = entry_size * spalette->nentries;
863 png_sPLT_entryp ep; 1264 png_sPLT_entryp ep;
864 #ifndef PNG_POINTER_INDEXING_SUPPORTED 1265 #ifndef PNG_POINTER_INDEXING_SUPPORTED
865 int i; 1266 int i;
866 #endif 1267 #endif
867 1268
868 png_debug(1, "in png_write_sPLT"); 1269 png_debug(1, "in png_write_sPLT");
869 1270
870 if ((name_len = png_check_keyword(png_ptr,spalette->name, &new_name))==0) 1271 name_len = png_check_keyword(png_ptr, spalette->name, new_name);
871 return; 1272
1273 if (name_len == 0)
1274 png_error(png_ptr, "sPLT: invalid keyword");
872 1275
873 /* Make sure we include the NULL after the name */ 1276 /* Make sure we include the NULL after the name */
874 png_write_chunk_start(png_ptr, (png_bytep)png_sPLT, 1277 png_write_chunk_header(png_ptr, png_sPLT,
875 (png_uint_32)(name_len + 2 + palette_size)); 1278 (png_uint_32)(name_len + 2 + palette_size));
1279
876 png_write_chunk_data(png_ptr, (png_bytep)new_name, 1280 png_write_chunk_data(png_ptr, (png_bytep)new_name,
877 (png_size_t)(name_len + 1)); 1281 (png_size_t)(name_len + 1));
878 png_write_chunk_data(png_ptr, (png_bytep)&spalette->depth, (png_size_t)1); 1282
1283 png_write_chunk_data(png_ptr, &spalette->depth, (png_size_t)1);
879 1284
880 /* Loop through each palette entry, writing appropriately */ 1285 /* Loop through each palette entry, writing appropriately */
881 #ifdef PNG_POINTER_INDEXING_SUPPORTED 1286 #ifdef PNG_POINTER_INDEXING_SUPPORTED
882 for (ep = spalette->entries; ep<spalette->entries + spalette->nentries; ep++) 1287 for (ep = spalette->entries; ep<spalette->entries + spalette->nentries; ep++)
883 { 1288 {
884 if (spalette->depth == 8) 1289 if (spalette->depth == 8)
885 { 1290 {
886 entrybuf[0] = (png_byte)ep->red; 1291 entrybuf[0] = (png_byte)ep->red;
887 entrybuf[1] = (png_byte)ep->green; 1292 entrybuf[1] = (png_byte)ep->green;
888 entrybuf[2] = (png_byte)ep->blue; 1293 entrybuf[2] = (png_byte)ep->blue;
889 entrybuf[3] = (png_byte)ep->alpha; 1294 entrybuf[3] = (png_byte)ep->alpha;
890 png_save_uint_16(entrybuf + 4, ep->frequency); 1295 png_save_uint_16(entrybuf + 4, ep->frequency);
891 } 1296 }
1297
892 else 1298 else
893 { 1299 {
894 png_save_uint_16(entrybuf + 0, ep->red); 1300 png_save_uint_16(entrybuf + 0, ep->red);
895 png_save_uint_16(entrybuf + 2, ep->green); 1301 png_save_uint_16(entrybuf + 2, ep->green);
896 png_save_uint_16(entrybuf + 4, ep->blue); 1302 png_save_uint_16(entrybuf + 4, ep->blue);
897 png_save_uint_16(entrybuf + 6, ep->alpha); 1303 png_save_uint_16(entrybuf + 6, ep->alpha);
898 png_save_uint_16(entrybuf + 8, ep->frequency); 1304 png_save_uint_16(entrybuf + 8, ep->frequency);
899 } 1305 }
900 png_write_chunk_data(png_ptr, entrybuf, (png_size_t)entry_size); 1306
1307 png_write_chunk_data(png_ptr, entrybuf, entry_size);
901 } 1308 }
902 #else 1309 #else
903 ep=spalette->entries; 1310 ep=spalette->entries;
904 for (i=0; i>spalette->nentries; i++) 1311 for (i = 0; i>spalette->nentries; i++)
905 { 1312 {
906 if (spalette->depth == 8) 1313 if (spalette->depth == 8)
907 { 1314 {
908 entrybuf[0] = (png_byte)ep[i].red; 1315 entrybuf[0] = (png_byte)ep[i].red;
909 entrybuf[1] = (png_byte)ep[i].green; 1316 entrybuf[1] = (png_byte)ep[i].green;
910 entrybuf[2] = (png_byte)ep[i].blue; 1317 entrybuf[2] = (png_byte)ep[i].blue;
911 entrybuf[3] = (png_byte)ep[i].alpha; 1318 entrybuf[3] = (png_byte)ep[i].alpha;
912 png_save_uint_16(entrybuf + 4, ep[i].frequency); 1319 png_save_uint_16(entrybuf + 4, ep[i].frequency);
913 } 1320 }
1321
914 else 1322 else
915 { 1323 {
916 png_save_uint_16(entrybuf + 0, ep[i].red); 1324 png_save_uint_16(entrybuf + 0, ep[i].red);
917 png_save_uint_16(entrybuf + 2, ep[i].green); 1325 png_save_uint_16(entrybuf + 2, ep[i].green);
918 png_save_uint_16(entrybuf + 4, ep[i].blue); 1326 png_save_uint_16(entrybuf + 4, ep[i].blue);
919 png_save_uint_16(entrybuf + 6, ep[i].alpha); 1327 png_save_uint_16(entrybuf + 6, ep[i].alpha);
920 png_save_uint_16(entrybuf + 8, ep[i].frequency); 1328 png_save_uint_16(entrybuf + 8, ep[i].frequency);
921 } 1329 }
922 png_write_chunk_data(png_ptr, entrybuf, (png_size_t)entry_size); 1330
1331 png_write_chunk_data(png_ptr, entrybuf, entry_size);
923 } 1332 }
924 #endif 1333 #endif
925 1334
926 png_write_chunk_end(png_ptr); 1335 png_write_chunk_end(png_ptr);
927 png_free(png_ptr, new_name);
928 } 1336 }
929 #endif 1337 #endif
930 1338
931 #ifdef PNG_WRITE_sBIT_SUPPORTED 1339 #ifdef PNG_WRITE_sBIT_SUPPORTED
932 /* Write the sBIT chunk */ 1340 /* Write the sBIT chunk */
933 void /* PRIVATE */ 1341 void /* PRIVATE */
934 png_write_sBIT(png_structp png_ptr, png_color_8p sbit, int color_type) 1342 png_write_sBIT(png_structrp png_ptr, png_const_color_8p sbit, int color_type)
935 { 1343 {
936 #ifdef PNG_USE_LOCAL_ARRAYS
937 PNG_sBIT;
938 #endif
939 png_byte buf[4]; 1344 png_byte buf[4];
940 png_size_t size; 1345 png_size_t size;
941 1346
942 png_debug(1, "in png_write_sBIT"); 1347 png_debug(1, "in png_write_sBIT");
943 1348
944 /* Make sure we don't depend upon the order of PNG_COLOR_8 */ 1349 /* Make sure we don't depend upon the order of PNG_COLOR_8 */
945 if (color_type & PNG_COLOR_MASK_COLOR) 1350 if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
946 { 1351 {
947 png_byte maxbits; 1352 png_byte maxbits;
948 1353
949 maxbits = (png_byte)(color_type==PNG_COLOR_TYPE_PALETTE ? 8 : 1354 maxbits = (png_byte)(color_type==PNG_COLOR_TYPE_PALETTE ? 8 :
950 png_ptr->usr_bit_depth); 1355 png_ptr->usr_bit_depth);
1356
951 if (sbit->red == 0 || sbit->red > maxbits || 1357 if (sbit->red == 0 || sbit->red > maxbits ||
952 sbit->green == 0 || sbit->green > maxbits || 1358 sbit->green == 0 || sbit->green > maxbits ||
953 sbit->blue == 0 || sbit->blue > maxbits) 1359 sbit->blue == 0 || sbit->blue > maxbits)
954 { 1360 {
955 png_warning(png_ptr, "Invalid sBIT depth specified"); 1361 png_warning(png_ptr, "Invalid sBIT depth specified");
956 return; 1362 return;
957 } 1363 }
1364
958 buf[0] = sbit->red; 1365 buf[0] = sbit->red;
959 buf[1] = sbit->green; 1366 buf[1] = sbit->green;
960 buf[2] = sbit->blue; 1367 buf[2] = sbit->blue;
961 size = 3; 1368 size = 3;
962 } 1369 }
1370
963 else 1371 else
964 { 1372 {
965 if (sbit->gray == 0 || sbit->gray > png_ptr->usr_bit_depth) 1373 if (sbit->gray == 0 || sbit->gray > png_ptr->usr_bit_depth)
966 { 1374 {
967 png_warning(png_ptr, "Invalid sBIT depth specified"); 1375 png_warning(png_ptr, "Invalid sBIT depth specified");
968 return; 1376 return;
969 } 1377 }
1378
970 buf[0] = sbit->gray; 1379 buf[0] = sbit->gray;
971 size = 1; 1380 size = 1;
972 } 1381 }
973 1382
974 if (color_type & PNG_COLOR_MASK_ALPHA) 1383 if ((color_type & PNG_COLOR_MASK_ALPHA) != 0)
975 { 1384 {
976 if (sbit->alpha == 0 || sbit->alpha > png_ptr->usr_bit_depth) 1385 if (sbit->alpha == 0 || sbit->alpha > png_ptr->usr_bit_depth)
977 { 1386 {
978 png_warning(png_ptr, "Invalid sBIT depth specified"); 1387 png_warning(png_ptr, "Invalid sBIT depth specified");
979 return; 1388 return;
980 } 1389 }
1390
981 buf[size++] = sbit->alpha; 1391 buf[size++] = sbit->alpha;
982 } 1392 }
983 1393
984 png_write_chunk(png_ptr, (png_bytep)png_sBIT, buf, size); 1394 png_write_complete_chunk(png_ptr, png_sBIT, buf, size);
985 } 1395 }
986 #endif 1396 #endif
987 1397
988 #ifdef PNG_WRITE_cHRM_SUPPORTED 1398 #ifdef PNG_WRITE_cHRM_SUPPORTED
989 /* Write the cHRM chunk */ 1399 /* Write the cHRM chunk */
990 #ifdef PNG_FLOATING_POINT_SUPPORTED
991 void /* PRIVATE */ 1400 void /* PRIVATE */
992 png_write_cHRM(png_structp png_ptr, double white_x, double white_y, 1401 png_write_cHRM_fixed(png_structrp png_ptr, const png_xy *xy)
993 double red_x, double red_y, double green_x, double green_y,
994 double blue_x, double blue_y)
995 { 1402 {
996 #ifdef PNG_USE_LOCAL_ARRAYS
997 PNG_cHRM;
998 #endif
999 png_byte buf[32];
1000
1001 png_fixed_point int_white_x, int_white_y, int_red_x, int_red_y,
1002 int_green_x, int_green_y, int_blue_x, int_blue_y;
1003
1004 png_debug(1, "in png_write_cHRM");
1005
1006 int_white_x = (png_uint_32)(white_x * 100000.0 + 0.5);
1007 int_white_y = (png_uint_32)(white_y * 100000.0 + 0.5);
1008 int_red_x = (png_uint_32)(red_x * 100000.0 + 0.5);
1009 int_red_y = (png_uint_32)(red_y * 100000.0 + 0.5);
1010 int_green_x = (png_uint_32)(green_x * 100000.0 + 0.5);
1011 int_green_y = (png_uint_32)(green_y * 100000.0 + 0.5);
1012 int_blue_x = (png_uint_32)(blue_x * 100000.0 + 0.5);
1013 int_blue_y = (png_uint_32)(blue_y * 100000.0 + 0.5);
1014
1015 #ifdef PNG_CHECK_cHRM_SUPPORTED
1016 if (png_check_cHRM_fixed(png_ptr, int_white_x, int_white_y,
1017 int_red_x, int_red_y, int_green_x, int_green_y, int_blue_x, int_blue_y))
1018 #endif
1019 {
1020 /* Each value is saved in 1/100,000ths */
1021
1022 png_save_uint_32(buf, int_white_x);
1023 png_save_uint_32(buf + 4, int_white_y);
1024
1025 png_save_uint_32(buf + 8, int_red_x);
1026 png_save_uint_32(buf + 12, int_red_y);
1027
1028 png_save_uint_32(buf + 16, int_green_x);
1029 png_save_uint_32(buf + 20, int_green_y);
1030
1031 png_save_uint_32(buf + 24, int_blue_x);
1032 png_save_uint_32(buf + 28, int_blue_y);
1033
1034 png_write_chunk(png_ptr, (png_bytep)png_cHRM, buf, (png_size_t)32);
1035 }
1036 }
1037 #endif
1038 #ifdef PNG_FIXED_POINT_SUPPORTED
1039 void /* PRIVATE */
1040 png_write_cHRM_fixed(png_structp png_ptr, png_fixed_point white_x,
1041 png_fixed_point white_y, png_fixed_point red_x, png_fixed_point red_y,
1042 png_fixed_point green_x, png_fixed_point green_y, png_fixed_point blue_x,
1043 png_fixed_point blue_y)
1044 {
1045 #ifdef PNG_USE_LOCAL_ARRAYS
1046 PNG_cHRM;
1047 #endif
1048 png_byte buf[32]; 1403 png_byte buf[32];
1049 1404
1050 png_debug(1, "in png_write_cHRM"); 1405 png_debug(1, "in png_write_cHRM");
1051 1406
1052 /* Each value is saved in 1/100,000ths */ 1407 /* Each value is saved in 1/100,000ths */
1053 #ifdef PNG_CHECK_cHRM_SUPPORTED 1408 png_save_int_32(buf, xy->whitex);
1054 if (png_check_cHRM_fixed(png_ptr, white_x, white_y, red_x, red_y, 1409 png_save_int_32(buf + 4, xy->whitey);
1055 green_x, green_y, blue_x, blue_y))
1056 #endif
1057 {
1058 png_save_uint_32(buf, (png_uint_32)white_x);
1059 png_save_uint_32(buf + 4, (png_uint_32)white_y);
1060 1410
1061 png_save_uint_32(buf + 8, (png_uint_32)red_x); 1411 png_save_int_32(buf + 8, xy->redx);
1062 png_save_uint_32(buf + 12, (png_uint_32)red_y); 1412 png_save_int_32(buf + 12, xy->redy);
1063 1413
1064 png_save_uint_32(buf + 16, (png_uint_32)green_x); 1414 png_save_int_32(buf + 16, xy->greenx);
1065 png_save_uint_32(buf + 20, (png_uint_32)green_y); 1415 png_save_int_32(buf + 20, xy->greeny);
1066 1416
1067 png_save_uint_32(buf + 24, (png_uint_32)blue_x); 1417 png_save_int_32(buf + 24, xy->bluex);
1068 png_save_uint_32(buf + 28, (png_uint_32)blue_y); 1418 png_save_int_32(buf + 28, xy->bluey);
1069 1419
1070 png_write_chunk(png_ptr, (png_bytep)png_cHRM, buf, (png_size_t)32); 1420 png_write_complete_chunk(png_ptr, png_cHRM, buf, 32);
1071 }
1072 } 1421 }
1073 #endif 1422 #endif
1074 #endif
1075 1423
1076 #ifdef PNG_WRITE_tRNS_SUPPORTED 1424 #ifdef PNG_WRITE_tRNS_SUPPORTED
1077 /* Write the tRNS chunk */ 1425 /* Write the tRNS chunk */
1078 void /* PRIVATE */ 1426 void /* PRIVATE */
1079 png_write_tRNS(png_structp png_ptr, png_bytep trans, png_color_16p tran, 1427 png_write_tRNS(png_structrp png_ptr, png_const_bytep trans_alpha,
1080 int num_trans, int color_type) 1428 png_const_color_16p tran, int num_trans, int color_type)
1081 { 1429 {
1082 #ifdef PNG_USE_LOCAL_ARRAYS
1083 PNG_tRNS;
1084 #endif
1085 png_byte buf[6]; 1430 png_byte buf[6];
1086 1431
1087 png_debug(1, "in png_write_tRNS"); 1432 png_debug(1, "in png_write_tRNS");
1088 1433
1089 if (color_type == PNG_COLOR_TYPE_PALETTE) 1434 if (color_type == PNG_COLOR_TYPE_PALETTE)
1090 { 1435 {
1091 if (num_trans <= 0 || num_trans > (int)png_ptr->num_palette) 1436 if (num_trans <= 0 || num_trans > (int)png_ptr->num_palette)
1092 { 1437 {
1093 png_warning(png_ptr, "Invalid number of transparent colors specified"); 1438 png_app_warning(png_ptr,
1439 "Invalid number of transparent colors specified");
1094 return; 1440 return;
1095 } 1441 }
1442
1096 /* Write the chunk out as it is */ 1443 /* Write the chunk out as it is */
1097 png_write_chunk(png_ptr, (png_bytep)png_tRNS, trans, 1444 png_write_complete_chunk(png_ptr, png_tRNS, trans_alpha,
1098 (png_size_t)num_trans); 1445 (png_size_t)num_trans);
1099 } 1446 }
1447
1100 else if (color_type == PNG_COLOR_TYPE_GRAY) 1448 else if (color_type == PNG_COLOR_TYPE_GRAY)
1101 { 1449 {
1102 /* One 16 bit value */ 1450 /* One 16-bit value */
1103 if (tran->gray >= (1 << png_ptr->bit_depth)) 1451 if (tran->gray >= (1 << png_ptr->bit_depth))
1104 { 1452 {
1105 png_warning(png_ptr, 1453 png_app_warning(png_ptr,
1106 "Ignoring attempt to write tRNS chunk out-of-range for bit_depth"); 1454 "Ignoring attempt to write tRNS chunk out-of-range for bit_depth");
1455
1107 return; 1456 return;
1108 } 1457 }
1458
1109 png_save_uint_16(buf, tran->gray); 1459 png_save_uint_16(buf, tran->gray);
1110 png_write_chunk(png_ptr, (png_bytep)png_tRNS, buf, (png_size_t)2); 1460 png_write_complete_chunk(png_ptr, png_tRNS, buf, (png_size_t)2);
1111 } 1461 }
1462
1112 else if (color_type == PNG_COLOR_TYPE_RGB) 1463 else if (color_type == PNG_COLOR_TYPE_RGB)
1113 { 1464 {
1114 /* Three 16 bit values */ 1465 /* Three 16-bit values */
1115 png_save_uint_16(buf, tran->red); 1466 png_save_uint_16(buf, tran->red);
1116 png_save_uint_16(buf + 2, tran->green); 1467 png_save_uint_16(buf + 2, tran->green);
1117 png_save_uint_16(buf + 4, tran->blue); 1468 png_save_uint_16(buf + 4, tran->blue);
1118 if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4])) 1469 #ifdef PNG_WRITE_16BIT_SUPPORTED
1470 if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]) != 0)
1471 #else
1472 if ((buf[0] | buf[2] | buf[4]) != 0)
1473 #endif
1119 { 1474 {
1120 png_warning(png_ptr, 1475 png_app_warning(png_ptr,
1121 "Ignoring attempt to write 16-bit tRNS chunk when bit_depth is 8"); 1476 "Ignoring attempt to write 16-bit tRNS chunk when bit_depth is 8");
1122 return; 1477 return;
1123 } 1478 }
1124 png_write_chunk(png_ptr, (png_bytep)png_tRNS, buf, (png_size_t)6); 1479
1480 png_write_complete_chunk(png_ptr, png_tRNS, buf, (png_size_t)6);
1125 } 1481 }
1482
1126 else 1483 else
1127 { 1484 {
1128 png_warning(png_ptr, "Can't write tRNS with an alpha channel"); 1485 png_app_warning(png_ptr, "Can't write tRNS with an alpha channel");
1129 } 1486 }
1130 } 1487 }
1131 #endif 1488 #endif
1132 1489
1133 #ifdef PNG_WRITE_bKGD_SUPPORTED 1490 #ifdef PNG_WRITE_bKGD_SUPPORTED
1134 /* Write the background chunk */ 1491 /* Write the background chunk */
1135 void /* PRIVATE */ 1492 void /* PRIVATE */
1136 png_write_bKGD(png_structp png_ptr, png_color_16p back, int color_type) 1493 png_write_bKGD(png_structrp png_ptr, png_const_color_16p back, int color_type)
1137 { 1494 {
1138 #ifdef PNG_USE_LOCAL_ARRAYS
1139 PNG_bKGD;
1140 #endif
1141 png_byte buf[6]; 1495 png_byte buf[6];
1142 1496
1143 png_debug(1, "in png_write_bKGD"); 1497 png_debug(1, "in png_write_bKGD");
1144 1498
1145 if (color_type == PNG_COLOR_TYPE_PALETTE) 1499 if (color_type == PNG_COLOR_TYPE_PALETTE)
1146 { 1500 {
1147 if ( 1501 if (
1148 #ifdef PNG_MNG_FEATURES_SUPPORTED 1502 #ifdef PNG_MNG_FEATURES_SUPPORTED
1149 (png_ptr->num_palette || 1503 (png_ptr->num_palette != 0 ||
1150 (!(png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE))) && 1504 (png_ptr->mng_features_permitted & PNG_FLAG_MNG_EMPTY_PLTE) == 0) &&
1151 #endif 1505 #endif
1152 back->index >= png_ptr->num_palette) 1506 back->index >= png_ptr->num_palette)
1153 { 1507 {
1154 png_warning(png_ptr, "Invalid background palette index"); 1508 png_warning(png_ptr, "Invalid background palette index");
1155 return; 1509 return;
1156 } 1510 }
1511
1157 buf[0] = back->index; 1512 buf[0] = back->index;
1158 png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)1); 1513 png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)1);
1159 } 1514 }
1160 else if (color_type & PNG_COLOR_MASK_COLOR) 1515
1516 else if ((color_type & PNG_COLOR_MASK_COLOR) != 0)
1161 { 1517 {
1162 png_save_uint_16(buf, back->red); 1518 png_save_uint_16(buf, back->red);
1163 png_save_uint_16(buf + 2, back->green); 1519 png_save_uint_16(buf + 2, back->green);
1164 png_save_uint_16(buf + 4, back->blue); 1520 png_save_uint_16(buf + 4, back->blue);
1165 if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4])) 1521 #ifdef PNG_WRITE_16BIT_SUPPORTED
1522 if (png_ptr->bit_depth == 8 && (buf[0] | buf[2] | buf[4]) != 0)
1523 #else
1524 if ((buf[0] | buf[2] | buf[4]) != 0)
1525 #endif
1166 { 1526 {
1167 png_warning(png_ptr, 1527 png_warning(png_ptr,
1168 "Ignoring attempt to write 16-bit bKGD chunk when bit_depth is 8"); 1528 "Ignoring attempt to write 16-bit bKGD chunk when bit_depth is 8");
1529
1169 return; 1530 return;
1170 } 1531 }
1171 png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)6); 1532
1533 png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)6);
1172 } 1534 }
1535
1173 else 1536 else
1174 { 1537 {
1175 if (back->gray >= (1 << png_ptr->bit_depth)) 1538 if (back->gray >= (1 << png_ptr->bit_depth))
1176 { 1539 {
1177 png_warning(png_ptr, 1540 png_warning(png_ptr,
1178 "Ignoring attempt to write bKGD chunk out-of-range for bit_depth"); 1541 "Ignoring attempt to write bKGD chunk out-of-range for bit_depth");
1542
1179 return; 1543 return;
1180 } 1544 }
1545
1181 png_save_uint_16(buf, back->gray); 1546 png_save_uint_16(buf, back->gray);
1182 png_write_chunk(png_ptr, (png_bytep)png_bKGD, buf, (png_size_t)2); 1547 png_write_complete_chunk(png_ptr, png_bKGD, buf, (png_size_t)2);
1183 } 1548 }
1184 } 1549 }
1185 #endif 1550 #endif
1186 1551
1187 #ifdef PNG_WRITE_hIST_SUPPORTED 1552 #ifdef PNG_WRITE_hIST_SUPPORTED
1188 /* Write the histogram */ 1553 /* Write the histogram */
1189 void /* PRIVATE */ 1554 void /* PRIVATE */
1190 png_write_hIST(png_structp png_ptr, png_uint_16p hist, int num_hist) 1555 png_write_hIST(png_structrp png_ptr, png_const_uint_16p hist, int num_hist)
1191 { 1556 {
1192 #ifdef PNG_USE_LOCAL_ARRAYS
1193 PNG_hIST;
1194 #endif
1195 int i; 1557 int i;
1196 png_byte buf[3]; 1558 png_byte buf[3];
1197 1559
1198 png_debug(1, "in png_write_hIST"); 1560 png_debug(1, "in png_write_hIST");
1199 1561
1200 if (num_hist > (int)png_ptr->num_palette) 1562 if (num_hist > (int)png_ptr->num_palette)
1201 { 1563 {
1202 png_debug2(3, "num_hist = %d, num_palette = %d", num_hist, 1564 png_debug2(3, "num_hist = %d, num_palette = %d", num_hist,
1203 png_ptr->num_palette); 1565 png_ptr->num_palette);
1566
1204 png_warning(png_ptr, "Invalid number of histogram entries specified"); 1567 png_warning(png_ptr, "Invalid number of histogram entries specified");
1205 return; 1568 return;
1206 } 1569 }
1207 1570
1208 png_write_chunk_start(png_ptr, (png_bytep)png_hIST, 1571 png_write_chunk_header(png_ptr, png_hIST, (png_uint_32)(num_hist * 2));
1209 (png_uint_32)(num_hist * 2)); 1572
1210 for (i = 0; i < num_hist; i++) 1573 for (i = 0; i < num_hist; i++)
1211 { 1574 {
1212 png_save_uint_16(buf, hist[i]); 1575 png_save_uint_16(buf, hist[i]);
1213 png_write_chunk_data(png_ptr, buf, (png_size_t)2); 1576 png_write_chunk_data(png_ptr, buf, (png_size_t)2);
1214 } 1577 }
1578
1215 png_write_chunk_end(png_ptr); 1579 png_write_chunk_end(png_ptr);
1216 } 1580 }
1217 #endif 1581 #endif
1218 1582
1219 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_pCAL_SUPPORTED) || \
1220 defined(PNG_WRITE_iCCP_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED)
1221 /* Check that the tEXt or zTXt keyword is valid per PNG 1.0 specification,
1222 * and if invalid, correct the keyword rather than discarding the entire
1223 * chunk. The PNG 1.0 specification requires keywords 1-79 characters in
1224 * length, forbids leading or trailing whitespace, multiple internal spaces,
1225 * and the non-break space (0x80) from ISO 8859-1. Returns keyword length.
1226 *
1227 * The new_key is allocated to hold the corrected keyword and must be freed
1228 * by the calling routine. This avoids problems with trying to write to
1229 * static keywords without having to have duplicate copies of the strings.
1230 */
1231 png_size_t /* PRIVATE */
1232 png_check_keyword(png_structp png_ptr, png_charp key, png_charpp new_key)
1233 {
1234 png_size_t key_len;
1235 png_charp kp, dp;
1236 int kflag;
1237 int kwarn=0;
1238
1239 png_debug(1, "in png_check_keyword");
1240
1241 *new_key = NULL;
1242
1243 if (key == NULL || (key_len = png_strlen(key)) == 0)
1244 {
1245 png_warning(png_ptr, "zero length keyword");
1246 return ((png_size_t)0);
1247 }
1248
1249 png_debug1(2, "Keyword to be checked is '%s'", key);
1250
1251 *new_key = (png_charp)png_malloc_warn(png_ptr, (png_uint_32)(key_len + 2));
1252 if (*new_key == NULL)
1253 {
1254 png_warning(png_ptr, "Out of memory while procesing keyword");
1255 return ((png_size_t)0);
1256 }
1257
1258 /* Replace non-printing characters with a blank and print a warning */
1259 for (kp = key, dp = *new_key; *kp != '\0'; kp++, dp++)
1260 {
1261 if ((png_byte)*kp < 0x20 ||
1262 ((png_byte)*kp > 0x7E && (png_byte)*kp < 0xA1))
1263 {
1264 #if defined(PNG_STDIO_SUPPORTED) && !defined(_WIN32_WCE)
1265 char msg[40];
1266
1267 png_snprintf(msg, 40,
1268 "invalid keyword character 0x%02X", (png_byte)*kp);
1269 png_warning(png_ptr, msg);
1270 #else
1271 png_warning(png_ptr, "invalid character in keyword");
1272 #endif
1273 *dp = ' ';
1274 }
1275 else
1276 {
1277 *dp = *kp;
1278 }
1279 }
1280 *dp = '\0';
1281
1282 /* Remove any trailing white space. */
1283 kp = *new_key + key_len - 1;
1284 if (*kp == ' ')
1285 {
1286 png_warning(png_ptr, "trailing spaces removed from keyword");
1287
1288 while (*kp == ' ')
1289 {
1290 *(kp--) = '\0';
1291 key_len--;
1292 }
1293 }
1294
1295 /* Remove any leading white space. */
1296 kp = *new_key;
1297 if (*kp == ' ')
1298 {
1299 png_warning(png_ptr, "leading spaces removed from keyword");
1300
1301 while (*kp == ' ')
1302 {
1303 kp++;
1304 key_len--;
1305 }
1306 }
1307
1308 png_debug1(2, "Checking for multiple internal spaces in '%s'", kp);
1309
1310 /* Remove multiple internal spaces. */
1311 for (kflag = 0, dp = *new_key; *kp != '\0'; kp++)
1312 {
1313 if (*kp == ' ' && kflag == 0)
1314 {
1315 *(dp++) = *kp;
1316 kflag = 1;
1317 }
1318 else if (*kp == ' ')
1319 {
1320 key_len--;
1321 kwarn=1;
1322 }
1323 else
1324 {
1325 *(dp++) = *kp;
1326 kflag = 0;
1327 }
1328 }
1329 *dp = '\0';
1330 if (kwarn)
1331 png_warning(png_ptr, "extra interior spaces removed from keyword");
1332
1333 if (key_len == 0)
1334 {
1335 png_free(png_ptr, *new_key);
1336 *new_key=NULL;
1337 png_warning(png_ptr, "Zero length keyword");
1338 }
1339
1340 if (key_len > 79)
1341 {
1342 png_warning(png_ptr, "keyword length must be 1 - 79 characters");
1343 (*new_key)[79] = '\0';
1344 key_len = 79;
1345 }
1346
1347 return (key_len);
1348 }
1349 #endif
1350
1351 #ifdef PNG_WRITE_tEXt_SUPPORTED 1583 #ifdef PNG_WRITE_tEXt_SUPPORTED
1352 /* Write a tEXt chunk */ 1584 /* Write a tEXt chunk */
1353 void /* PRIVATE */ 1585 void /* PRIVATE */
1354 png_write_tEXt(png_structp png_ptr, png_charp key, png_charp text, 1586 png_write_tEXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
1355 png_size_t text_len) 1587 png_size_t text_len)
1356 { 1588 {
1357 #ifdef PNG_USE_LOCAL_ARRAYS 1589 png_uint_32 key_len;
1358 PNG_tEXt; 1590 png_byte new_key[80];
1359 #endif
1360 png_size_t key_len;
1361 png_charp new_key;
1362 1591
1363 png_debug(1, "in png_write_tEXt"); 1592 png_debug(1, "in png_write_tEXt");
1364 1593
1365 if ((key_len = png_check_keyword(png_ptr, key, &new_key))==0) 1594 key_len = png_check_keyword(png_ptr, key, new_key);
1366 return; 1595
1596 if (key_len == 0)
1597 png_error(png_ptr, "tEXt: invalid keyword");
1367 1598
1368 if (text == NULL || *text == '\0') 1599 if (text == NULL || *text == '\0')
1369 text_len = 0; 1600 text_len = 0;
1601
1370 else 1602 else
1371 text_len = png_strlen(text); 1603 text_len = strlen(text);
1604
1605 if (text_len > PNG_UINT_31_MAX - (key_len+1))
1606 png_error(png_ptr, "tEXt: text too long");
1372 1607
1373 /* Make sure we include the 0 after the key */ 1608 /* Make sure we include the 0 after the key */
1374 png_write_chunk_start(png_ptr, (png_bytep)png_tEXt, 1609 png_write_chunk_header(png_ptr, png_tEXt,
1375 (png_uint_32)(key_len + text_len + 1)); 1610 (png_uint_32)/*checked above*/(key_len + text_len + 1));
1376 /* 1611 /*
1377 * We leave it to the application to meet PNG-1.0 requirements on the 1612 * We leave it to the application to meet PNG-1.0 requirements on the
1378 * contents of the text. PNG-1.0 through PNG-1.2 discourage the use of 1613 * contents of the text. PNG-1.0 through PNG-1.2 discourage the use of
1379 * any non-Latin-1 characters except for NEWLINE. ISO PNG will forbid them. 1614 * any non-Latin-1 characters except for NEWLINE. ISO PNG will forbid them.
1380 * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG. 1615 * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1381 */ 1616 */
1382 png_write_chunk_data(png_ptr, (png_bytep)new_key, 1617 png_write_chunk_data(png_ptr, new_key, key_len + 1);
1383 (png_size_t)(key_len + 1)); 1618
1384 if (text_len) 1619 if (text_len != 0)
1385 png_write_chunk_data(png_ptr, (png_bytep)text, (png_size_t)text_len); 1620 png_write_chunk_data(png_ptr, (png_const_bytep)text, text_len);
1386 1621
1387 png_write_chunk_end(png_ptr); 1622 png_write_chunk_end(png_ptr);
1388 png_free(png_ptr, new_key);
1389 } 1623 }
1390 #endif 1624 #endif
1391 1625
1392 #ifdef PNG_WRITE_zTXt_SUPPORTED 1626 #ifdef PNG_WRITE_zTXt_SUPPORTED
1393 /* Write a compressed text chunk */ 1627 /* Write a compressed text chunk */
1394 void /* PRIVATE */ 1628 void /* PRIVATE */
1395 png_write_zTXt(png_structp png_ptr, png_charp key, png_charp text, 1629 png_write_zTXt(png_structrp png_ptr, png_const_charp key, png_const_charp text,
1396 png_size_t text_len, int compression) 1630 int compression)
1397 { 1631 {
1398 #ifdef PNG_USE_LOCAL_ARRAYS 1632 png_uint_32 key_len;
1399 PNG_zTXt; 1633 png_byte new_key[81];
1400 #endif
1401 png_size_t key_len;
1402 char buf[1];
1403 png_charp new_key;
1404 compression_state comp; 1634 compression_state comp;
1405 1635
1406 png_debug(1, "in png_write_zTXt"); 1636 png_debug(1, "in png_write_zTXt");
1407 1637
1408 comp.num_output_ptr = 0; 1638 if (compression == PNG_TEXT_COMPRESSION_NONE)
1409 comp.max_output_ptr = 0;
1410 comp.output_ptr = NULL;
1411 comp.input = NULL;
1412 comp.input_len = 0;
1413
1414 if ((key_len = png_check_keyword(png_ptr, key, &new_key))==0)
1415 { 1639 {
1416 png_free(png_ptr, new_key); 1640 png_write_tEXt(png_ptr, key, text, 0);
1417 return; 1641 return;
1418 } 1642 }
1419 1643
1420 if (text == NULL || *text == '\0' || compression==PNG_TEXT_COMPRESSION_NONE) 1644 if (compression != PNG_TEXT_COMPRESSION_zTXt)
1421 { 1645 png_error(png_ptr, "zTXt: invalid compression type");
1422 png_write_tEXt(png_ptr, new_key, text, (png_size_t)0);
1423 png_free(png_ptr, new_key);
1424 return;
1425 }
1426 1646
1427 text_len = png_strlen(text); 1647 key_len = png_check_keyword(png_ptr, key, new_key);
1648
1649 if (key_len == 0)
1650 png_error(png_ptr, "zTXt: invalid keyword");
1651
1652 /* Add the compression method and 1 for the keyword separator. */
1653 new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
1654 ++key_len;
1428 1655
1429 /* Compute the compressed data; do it now for the length */ 1656 /* Compute the compressed data; do it now for the length */
1430 text_len = png_text_compress(png_ptr, text, text_len, compression, 1657 png_text_compress_init(&comp, (png_const_bytep)text,
1431 &comp); 1658 text == NULL ? 0 : strlen(text));
1659
1660 if (png_text_compress(png_ptr, png_zTXt, &comp, key_len) != Z_OK)
1661 png_error(png_ptr, png_ptr->zstream.msg);
1432 1662
1433 /* Write start of chunk */ 1663 /* Write start of chunk */
1434 png_write_chunk_start(png_ptr, (png_bytep)png_zTXt, 1664 png_write_chunk_header(png_ptr, png_zTXt, key_len + comp.output_len);
1435 (png_uint_32)(key_len+text_len + 2)); 1665
1436 /* Write key */ 1666 /* Write key */
1437 png_write_chunk_data(png_ptr, (png_bytep)new_key, 1667 png_write_chunk_data(png_ptr, new_key, key_len);
1438 (png_size_t)(key_len + 1));
1439 png_free(png_ptr, new_key);
1440 1668
1441 buf[0] = (png_byte)compression;
1442 /* Write compression */
1443 png_write_chunk_data(png_ptr, (png_bytep)buf, (png_size_t)1);
1444 /* Write the compressed data */ 1669 /* Write the compressed data */
1445 png_write_compressed_data_out(png_ptr, &comp); 1670 png_write_compressed_data_out(png_ptr, &comp);
1446 1671
1447 /* Close the chunk */ 1672 /* Close the chunk */
1448 png_write_chunk_end(png_ptr); 1673 png_write_chunk_end(png_ptr);
1449 } 1674 }
1450 #endif 1675 #endif
1451 1676
1452 #ifdef PNG_WRITE_iTXt_SUPPORTED 1677 #ifdef PNG_WRITE_iTXt_SUPPORTED
1453 /* Write an iTXt chunk */ 1678 /* Write an iTXt chunk */
1454 void /* PRIVATE */ 1679 void /* PRIVATE */
1455 png_write_iTXt(png_structp png_ptr, int compression, png_charp key, 1680 png_write_iTXt(png_structrp png_ptr, int compression, png_const_charp key,
1456 png_charp lang, png_charp lang_key, png_charp text) 1681 png_const_charp lang, png_const_charp lang_key, png_const_charp text)
1457 { 1682 {
1458 #ifdef PNG_USE_LOCAL_ARRAYS 1683 png_uint_32 key_len, prefix_len;
1459 PNG_iTXt; 1684 png_size_t lang_len, lang_key_len;
1460 #endif 1685 png_byte new_key[82];
1461 png_size_t lang_len, key_len, lang_key_len, text_len;
1462 png_charp new_lang;
1463 png_charp new_key = NULL;
1464 png_byte cbuf[2];
1465 compression_state comp; 1686 compression_state comp;
1466 1687
1467 png_debug(1, "in png_write_iTXt"); 1688 png_debug(1, "in png_write_iTXt");
1468 1689
1469 comp.num_output_ptr = 0; 1690 key_len = png_check_keyword(png_ptr, key, new_key);
1470 comp.max_output_ptr = 0;
1471 comp.output_ptr = NULL;
1472 comp.input = NULL;
1473 1691
1474 if ((key_len = png_check_keyword(png_ptr, key, &new_key))==0) 1692 if (key_len == 0)
1475 return; 1693 png_error(png_ptr, "iTXt: invalid keyword");
1476 1694
1477 if ((lang_len = png_check_keyword(png_ptr, lang, &new_lang))==0) 1695 /* Set the compression flag */
1696 switch (compression)
1478 { 1697 {
1479 png_warning(png_ptr, "Empty language field in iTXt chunk"); 1698 case PNG_ITXT_COMPRESSION_NONE:
1480 new_lang = NULL; 1699 case PNG_TEXT_COMPRESSION_NONE:
1481 lang_len = 0; 1700 compression = new_key[++key_len] = 0; /* no compression */
1701 break;
1702
1703 case PNG_TEXT_COMPRESSION_zTXt:
1704 case PNG_ITXT_COMPRESSION_zTXt:
1705 compression = new_key[++key_len] = 1; /* compressed */
1706 break;
1707
1708 default:
1709 png_error(png_ptr, "iTXt: invalid compression");
1482 } 1710 }
1483 1711
1484 if (lang_key == NULL) 1712 new_key[++key_len] = PNG_COMPRESSION_TYPE_BASE;
1485 lang_key_len = 0; 1713 ++key_len; /* for the keywod separator */
1486 else
1487 lang_key_len = png_strlen(lang_key);
1488
1489 if (text == NULL)
1490 text_len = 0;
1491 else
1492 text_len = png_strlen(text);
1493
1494 /* Compute the compressed data; do it now for the length */
1495 text_len = png_text_compress(png_ptr, text, text_len, compression-2,
1496 &comp);
1497
1498
1499 /* Make sure we include the compression flag, the compression byte,
1500 * and the NULs after the key, lang, and lang_key parts */
1501
1502 png_write_chunk_start(png_ptr, (png_bytep)png_iTXt,
1503 (png_uint_32)(
1504 5 /* comp byte, comp flag, terminators for key, lang and lang_key */
1505 + key_len
1506 + lang_len
1507 + lang_key_len
1508 + text_len));
1509 1714
1510 /* We leave it to the application to meet PNG-1.0 requirements on the 1715 /* We leave it to the application to meet PNG-1.0 requirements on the
1511 * contents of the text. PNG-1.0 through PNG-1.2 discourage the use of 1716 * contents of the text. PNG-1.0 through PNG-1.2 discourage the use of
1512 * any non-Latin-1 characters except for NEWLINE. ISO PNG will forbid them. 1717 * any non-Latin-1 characters except for NEWLINE. ISO PNG, however,
1718 * specifies that the text is UTF-8 and this really doesn't require any
1719 * checking.
1720 *
1513 * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG. 1721 * The NUL character is forbidden by PNG-1.0 through PNG-1.2 and ISO PNG.
1722 *
1723 * TODO: validate the language tag correctly (see the spec.)
1514 */ 1724 */
1515 png_write_chunk_data(png_ptr, (png_bytep)new_key, 1725 if (lang == NULL) lang = ""; /* empty language is valid */
1516 (png_size_t)(key_len + 1)); 1726 lang_len = strlen(lang)+1;
1727 if (lang_key == NULL) lang_key = ""; /* may be empty */
1728 lang_key_len = strlen(lang_key)+1;
1729 if (text == NULL) text = ""; /* may be empty */
1517 1730
1518 /* Set the compression flag */ 1731 prefix_len = key_len;
1519 if (compression == PNG_ITXT_COMPRESSION_NONE || \ 1732 if (lang_len > PNG_UINT_31_MAX-prefix_len)
1520 compression == PNG_TEXT_COMPRESSION_NONE) 1733 prefix_len = PNG_UINT_31_MAX;
1521 cbuf[0] = 0; 1734 else
1522 else /* compression == PNG_ITXT_COMPRESSION_zTXt */ 1735 prefix_len = (png_uint_32)(prefix_len + lang_len);
1523 cbuf[0] = 1;
1524 /* Set the compression method */
1525 cbuf[1] = 0;
1526 png_write_chunk_data(png_ptr, cbuf, (png_size_t)2);
1527 1736
1528 cbuf[0] = 0; 1737 if (lang_key_len > PNG_UINT_31_MAX-prefix_len)
1529 png_write_chunk_data(png_ptr, (new_lang ? (png_bytep)new_lang : cbuf), 1738 prefix_len = PNG_UINT_31_MAX;
1530 (png_size_t)(lang_len + 1)); 1739 else
1531 png_write_chunk_data(png_ptr, (lang_key ? (png_bytep)lang_key : cbuf), 1740 prefix_len = (png_uint_32)(prefix_len + lang_key_len);
1532 (png_size_t)(lang_key_len + 1)); 1741
1533 png_write_compressed_data_out(png_ptr, &comp); 1742 png_text_compress_init(&comp, (png_const_bytep)text, strlen(text));
1743
1744 if (compression != 0)
1745 {
1746 if (png_text_compress(png_ptr, png_iTXt, &comp, prefix_len) != Z_OK)
1747 png_error(png_ptr, png_ptr->zstream.msg);
1748 }
1749
1750 else
1751 {
1752 if (comp.input_len > PNG_UINT_31_MAX-prefix_len)
1753 png_error(png_ptr, "iTXt: uncompressed text too long");
1754
1755 /* So the string will fit in a chunk: */
1756 comp.output_len = (png_uint_32)/*SAFE*/comp.input_len;
1757 }
1758
1759 png_write_chunk_header(png_ptr, png_iTXt, comp.output_len + prefix_len);
1760
1761 png_write_chunk_data(png_ptr, new_key, key_len);
1762
1763 png_write_chunk_data(png_ptr, (png_const_bytep)lang, lang_len);
1764
1765 png_write_chunk_data(png_ptr, (png_const_bytep)lang_key, lang_key_len);
1766
1767 if (compression != 0)
1768 png_write_compressed_data_out(png_ptr, &comp);
1769
1770 else
1771 png_write_chunk_data(png_ptr, (png_const_bytep)text, comp.output_len);
1534 1772
1535 png_write_chunk_end(png_ptr); 1773 png_write_chunk_end(png_ptr);
1536 png_free(png_ptr, new_key);
1537 png_free(png_ptr, new_lang);
1538 } 1774 }
1539 #endif 1775 #endif
1540 1776
1541 #ifdef PNG_WRITE_oFFs_SUPPORTED 1777 #ifdef PNG_WRITE_oFFs_SUPPORTED
1542 /* Write the oFFs chunk */ 1778 /* Write the oFFs chunk */
1543 void /* PRIVATE */ 1779 void /* PRIVATE */
1544 png_write_oFFs(png_structp png_ptr, png_int_32 x_offset, png_int_32 y_offset, 1780 png_write_oFFs(png_structrp png_ptr, png_int_32 x_offset, png_int_32 y_offset,
1545 int unit_type) 1781 int unit_type)
1546 { 1782 {
1547 #ifdef PNG_USE_LOCAL_ARRAYS
1548 PNG_oFFs;
1549 #endif
1550 png_byte buf[9]; 1783 png_byte buf[9];
1551 1784
1552 png_debug(1, "in png_write_oFFs"); 1785 png_debug(1, "in png_write_oFFs");
1553 1786
1554 if (unit_type >= PNG_OFFSET_LAST) 1787 if (unit_type >= PNG_OFFSET_LAST)
1555 png_warning(png_ptr, "Unrecognized unit type for oFFs chunk"); 1788 png_warning(png_ptr, "Unrecognized unit type for oFFs chunk");
1556 1789
1557 png_save_int_32(buf, x_offset); 1790 png_save_int_32(buf, x_offset);
1558 png_save_int_32(buf + 4, y_offset); 1791 png_save_int_32(buf + 4, y_offset);
1559 buf[8] = (png_byte)unit_type; 1792 buf[8] = (png_byte)unit_type;
1560 1793
1561 png_write_chunk(png_ptr, (png_bytep)png_oFFs, buf, (png_size_t)9); 1794 png_write_complete_chunk(png_ptr, png_oFFs, buf, (png_size_t)9);
1562 } 1795 }
1563 #endif 1796 #endif
1564 #ifdef PNG_WRITE_pCAL_SUPPORTED 1797 #ifdef PNG_WRITE_pCAL_SUPPORTED
1565 /* Write the pCAL chunk (described in the PNG extensions document) */ 1798 /* Write the pCAL chunk (described in the PNG extensions document) */
1566 void /* PRIVATE */ 1799 void /* PRIVATE */
1567 png_write_pCAL(png_structp png_ptr, png_charp purpose, png_int_32 X0, 1800 png_write_pCAL(png_structrp png_ptr, png_charp purpose, png_int_32 X0,
1568 png_int_32 X1, int type, int nparams, png_charp units, png_charpp params) 1801 png_int_32 X1, int type, int nparams, png_const_charp units,
1802 png_charpp params)
1569 { 1803 {
1570 #ifdef PNG_USE_LOCAL_ARRAYS 1804 png_uint_32 purpose_len;
1571 PNG_pCAL; 1805 png_size_t units_len, total_len;
1572 #endif 1806 png_size_tp params_len;
1573 png_size_t purpose_len, units_len, total_len;
1574 png_uint_32p params_len;
1575 png_byte buf[10]; 1807 png_byte buf[10];
1576 png_charp new_purpose; 1808 png_byte new_purpose[80];
1577 int i; 1809 int i;
1578 1810
1579 png_debug1(1, "in png_write_pCAL (%d parameters)", nparams); 1811 png_debug1(1, "in png_write_pCAL (%d parameters)", nparams);
1580 1812
1581 if (type >= PNG_EQUATION_LAST) 1813 if (type >= PNG_EQUATION_LAST)
1582 png_warning(png_ptr, "Unrecognized equation type for pCAL chunk"); 1814 png_error(png_ptr, "Unrecognized equation type for pCAL chunk");
1583 1815
1584 purpose_len = png_check_keyword(png_ptr, purpose, &new_purpose) + 1; 1816 purpose_len = png_check_keyword(png_ptr, purpose, new_purpose);
1817
1818 if (purpose_len == 0)
1819 png_error(png_ptr, "pCAL: invalid keyword");
1820
1821 ++purpose_len; /* terminator */
1822
1585 png_debug1(3, "pCAL purpose length = %d", (int)purpose_len); 1823 png_debug1(3, "pCAL purpose length = %d", (int)purpose_len);
1586 units_len = png_strlen(units) + (nparams == 0 ? 0 : 1); 1824 units_len = strlen(units) + (nparams == 0 ? 0 : 1);
1587 png_debug1(3, "pCAL units length = %d", (int)units_len); 1825 png_debug1(3, "pCAL units length = %d", (int)units_len);
1588 total_len = purpose_len + units_len + 10; 1826 total_len = purpose_len + units_len + 10;
1589 1827
1590 params_len = (png_uint_32p)png_malloc(png_ptr, 1828 params_len = (png_size_tp)png_malloc(png_ptr,
1591 (png_uint_32)(nparams * png_sizeof(png_uint_32))); 1829 (png_alloc_size_t)(nparams * (sizeof (png_size_t))));
1592 1830
1593 /* Find the length of each parameter, making sure we don't count the 1831 /* Find the length of each parameter, making sure we don't count the
1594 null terminator for the last parameter. */ 1832 * null terminator for the last parameter.
1833 */
1595 for (i = 0; i < nparams; i++) 1834 for (i = 0; i < nparams; i++)
1596 { 1835 {
1597 params_len[i] = png_strlen(params[i]) + (i == nparams - 1 ? 0 : 1); 1836 params_len[i] = strlen(params[i]) + (i == nparams - 1 ? 0 : 1);
1598 png_debug2(3, "pCAL parameter %d length = %lu", i, 1837 png_debug2(3, "pCAL parameter %d length = %lu", i,
1599 (unsigned long) params_len[i]); 1838 (unsigned long)params_len[i]);
1600 total_len += (png_size_t)params_len[i]; 1839 total_len += params_len[i];
1601 } 1840 }
1602 1841
1603 png_debug1(3, "pCAL total length = %d", (int)total_len); 1842 png_debug1(3, "pCAL total length = %d", (int)total_len);
1604 png_write_chunk_start(png_ptr, (png_bytep)png_pCAL, (png_uint_32)total_len); 1843 png_write_chunk_header(png_ptr, png_pCAL, (png_uint_32)total_len);
1605 png_write_chunk_data(png_ptr, (png_bytep)new_purpose, 1844 png_write_chunk_data(png_ptr, new_purpose, purpose_len);
1606 (png_size_t)purpose_len);
1607 png_save_int_32(buf, X0); 1845 png_save_int_32(buf, X0);
1608 png_save_int_32(buf + 4, X1); 1846 png_save_int_32(buf + 4, X1);
1609 buf[8] = (png_byte)type; 1847 buf[8] = (png_byte)type;
1610 buf[9] = (png_byte)nparams; 1848 buf[9] = (png_byte)nparams;
1611 png_write_chunk_data(png_ptr, buf, (png_size_t)10); 1849 png_write_chunk_data(png_ptr, buf, (png_size_t)10);
1612 png_write_chunk_data(png_ptr, (png_bytep)units, (png_size_t)units_len); 1850 png_write_chunk_data(png_ptr, (png_const_bytep)units, (png_size_t)units_len);
1613
1614 png_free(png_ptr, new_purpose);
1615 1851
1616 for (i = 0; i < nparams; i++) 1852 for (i = 0; i < nparams; i++)
1617 { 1853 {
1618 png_write_chunk_data(png_ptr, (png_bytep)params[i], 1854 png_write_chunk_data(png_ptr, (png_const_bytep)params[i], params_len[i]);
1619 (png_size_t)params_len[i]);
1620 } 1855 }
1621 1856
1622 png_free(png_ptr, params_len); 1857 png_free(png_ptr, params_len);
1623 png_write_chunk_end(png_ptr); 1858 png_write_chunk_end(png_ptr);
1624 } 1859 }
1625 #endif 1860 #endif
1626 1861
1627 #ifdef PNG_WRITE_sCAL_SUPPORTED 1862 #ifdef PNG_WRITE_sCAL_SUPPORTED
1628 /* Write the sCAL chunk */ 1863 /* Write the sCAL chunk */
1629 #if defined(PNG_FLOATING_POINT_SUPPORTED) && defined(PNG_STDIO_SUPPORTED)
1630 void /* PRIVATE */ 1864 void /* PRIVATE */
1631 png_write_sCAL(png_structp png_ptr, int unit, double width, double height) 1865 png_write_sCAL_s(png_structrp png_ptr, int unit, png_const_charp width,
1866 png_const_charp height)
1632 { 1867 {
1633 #ifdef PNG_USE_LOCAL_ARRAYS
1634 PNG_sCAL;
1635 #endif
1636 char buf[64];
1637 png_size_t total_len;
1638
1639 png_debug(1, "in png_write_sCAL");
1640
1641 buf[0] = (char)unit;
1642 #ifdef _WIN32_WCE
1643 /* sprintf() function is not supported on WindowsCE */
1644 {
1645 wchar_t wc_buf[32];
1646 size_t wc_len;
1647 swprintf(wc_buf, TEXT("%12.12e"), width);
1648 wc_len = wcslen(wc_buf);
1649 WideCharToMultiByte(CP_ACP, 0, wc_buf, -1, buf + 1, wc_len, NULL,
1650 NULL);
1651 total_len = wc_len + 2;
1652 swprintf(wc_buf, TEXT("%12.12e"), height);
1653 wc_len = wcslen(wc_buf);
1654 WideCharToMultiByte(CP_ACP, 0, wc_buf, -1, buf + total_len, wc_len,
1655 NULL, NULL);
1656 total_len += wc_len;
1657 }
1658 #else
1659 png_snprintf(buf + 1, 63, "%12.12e", width);
1660 total_len = 1 + png_strlen(buf + 1) + 1;
1661 png_snprintf(buf + total_len, 64-total_len, "%12.12e", height);
1662 total_len += png_strlen(buf + total_len);
1663 #endif
1664
1665 png_debug1(3, "sCAL total length = %u", (unsigned int)total_len);
1666 png_write_chunk(png_ptr, (png_bytep)png_sCAL, (png_bytep)buf, total_len);
1667 }
1668 #else
1669 #ifdef PNG_FIXED_POINT_SUPPORTED
1670 void /* PRIVATE */
1671 png_write_sCAL_s(png_structp png_ptr, int unit, png_charp width,
1672 png_charp height)
1673 {
1674 #ifdef PNG_USE_LOCAL_ARRAYS
1675 PNG_sCAL;
1676 #endif
1677 png_byte buf[64]; 1868 png_byte buf[64];
1678 png_size_t wlen, hlen, total_len; 1869 png_size_t wlen, hlen, total_len;
1679 1870
1680 png_debug(1, "in png_write_sCAL_s"); 1871 png_debug(1, "in png_write_sCAL_s");
1681 1872
1682 wlen = png_strlen(width); 1873 wlen = strlen(width);
1683 hlen = png_strlen(height); 1874 hlen = strlen(height);
1684 total_len = wlen + hlen + 2; 1875 total_len = wlen + hlen + 2;
1876
1685 if (total_len > 64) 1877 if (total_len > 64)
1686 { 1878 {
1687 png_warning(png_ptr, "Can't write sCAL (buffer too small)"); 1879 png_warning(png_ptr, "Can't write sCAL (buffer too small)");
1688 return; 1880 return;
1689 } 1881 }
1690 1882
1691 buf[0] = (png_byte)unit; 1883 buf[0] = (png_byte)unit;
1692 png_memcpy(buf + 1, width, wlen + 1); /* Append the '\0' here */ 1884 memcpy(buf + 1, width, wlen + 1); /* Append the '\0' here */
1693 png_memcpy(buf + wlen + 2, height, hlen); /* Do NOT append the '\0' here */ 1885 memcpy(buf + wlen + 2, height, hlen); /* Do NOT append the '\0' here */
1694 1886
1695 png_debug1(3, "sCAL total length = %u", (unsigned int)total_len); 1887 png_debug1(3, "sCAL total length = %u", (unsigned int)total_len);
1696 png_write_chunk(png_ptr, (png_bytep)png_sCAL, buf, total_len); 1888 png_write_complete_chunk(png_ptr, png_sCAL, buf, total_len);
1697 } 1889 }
1698 #endif 1890 #endif
1699 #endif
1700 #endif
1701 1891
1702 #ifdef PNG_WRITE_pHYs_SUPPORTED 1892 #ifdef PNG_WRITE_pHYs_SUPPORTED
1703 /* Write the pHYs chunk */ 1893 /* Write the pHYs chunk */
1704 void /* PRIVATE */ 1894 void /* PRIVATE */
1705 png_write_pHYs(png_structp png_ptr, png_uint_32 x_pixels_per_unit, 1895 png_write_pHYs(png_structrp png_ptr, png_uint_32 x_pixels_per_unit,
1706 png_uint_32 y_pixels_per_unit, 1896 png_uint_32 y_pixels_per_unit,
1707 int unit_type) 1897 int unit_type)
1708 { 1898 {
1709 #ifdef PNG_USE_LOCAL_ARRAYS
1710 PNG_pHYs;
1711 #endif
1712 png_byte buf[9]; 1899 png_byte buf[9];
1713 1900
1714 png_debug(1, "in png_write_pHYs"); 1901 png_debug(1, "in png_write_pHYs");
1715 1902
1716 if (unit_type >= PNG_RESOLUTION_LAST) 1903 if (unit_type >= PNG_RESOLUTION_LAST)
1717 png_warning(png_ptr, "Unrecognized unit type for pHYs chunk"); 1904 png_warning(png_ptr, "Unrecognized unit type for pHYs chunk");
1718 1905
1719 png_save_uint_32(buf, x_pixels_per_unit); 1906 png_save_uint_32(buf, x_pixels_per_unit);
1720 png_save_uint_32(buf + 4, y_pixels_per_unit); 1907 png_save_uint_32(buf + 4, y_pixels_per_unit);
1721 buf[8] = (png_byte)unit_type; 1908 buf[8] = (png_byte)unit_type;
1722 1909
1723 png_write_chunk(png_ptr, (png_bytep)png_pHYs, buf, (png_size_t)9); 1910 png_write_complete_chunk(png_ptr, png_pHYs, buf, (png_size_t)9);
1724 } 1911 }
1725 #endif 1912 #endif
1726 1913
1727 #ifdef PNG_WRITE_tIME_SUPPORTED 1914 #ifdef PNG_WRITE_tIME_SUPPORTED
1728 /* Write the tIME chunk. Use either png_convert_from_struct_tm() 1915 /* Write the tIME chunk. Use either png_convert_from_struct_tm()
1729 * or png_convert_from_time_t(), or fill in the structure yourself. 1916 * or png_convert_from_time_t(), or fill in the structure yourself.
1730 */ 1917 */
1731 void /* PRIVATE */ 1918 void /* PRIVATE */
1732 png_write_tIME(png_structp png_ptr, png_timep mod_time) 1919 png_write_tIME(png_structrp png_ptr, png_const_timep mod_time)
1733 { 1920 {
1734 #ifdef PNG_USE_LOCAL_ARRAYS
1735 PNG_tIME;
1736 #endif
1737 png_byte buf[7]; 1921 png_byte buf[7];
1738 1922
1739 png_debug(1, "in png_write_tIME"); 1923 png_debug(1, "in png_write_tIME");
1740 1924
1741 if (mod_time->month > 12 || mod_time->month < 1 || 1925 if (mod_time->month > 12 || mod_time->month < 1 ||
1742 mod_time->day > 31 || mod_time->day < 1 || 1926 mod_time->day > 31 || mod_time->day < 1 ||
1743 mod_time->hour > 23 || mod_time->second > 60) 1927 mod_time->hour > 23 || mod_time->second > 60)
1744 { 1928 {
1745 png_warning(png_ptr, "Invalid time specified for tIME chunk"); 1929 png_warning(png_ptr, "Invalid time specified for tIME chunk");
1746 return; 1930 return;
1747 } 1931 }
1748 1932
1749 png_save_uint_16(buf, mod_time->year); 1933 png_save_uint_16(buf, mod_time->year);
1750 buf[2] = mod_time->month; 1934 buf[2] = mod_time->month;
1751 buf[3] = mod_time->day; 1935 buf[3] = mod_time->day;
1752 buf[4] = mod_time->hour; 1936 buf[4] = mod_time->hour;
1753 buf[5] = mod_time->minute; 1937 buf[5] = mod_time->minute;
1754 buf[6] = mod_time->second; 1938 buf[6] = mod_time->second;
1755 1939
1756 png_write_chunk(png_ptr, (png_bytep)png_tIME, buf, (png_size_t)7); 1940 png_write_complete_chunk(png_ptr, png_tIME, buf, (png_size_t)7);
1757 } 1941 }
1758 #endif 1942 #endif
1759 1943
1760 /* Initializes the row writing capability of libpng */ 1944 /* Initializes the row writing capability of libpng */
1761 void /* PRIVATE */ 1945 void /* PRIVATE */
1762 png_write_start_row(png_structp png_ptr) 1946 png_write_start_row(png_structrp png_ptr)
1763 { 1947 {
1764 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 1948 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1765 #ifndef PNG_USE_GLOBAL_ARRAYS
1766 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */ 1949 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1767 1950
1768 /* Start of interlace block */ 1951 /* Start of interlace block */
1769 int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0}; 1952 static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1770 1953
1771 /* Offset to next interlace block */ 1954 /* Offset to next interlace block */
1772 int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1}; 1955 static PNG_CONST png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1773 1956
1774 /* Start of interlace block in the y direction */ 1957 /* Start of interlace block in the y direction */
1775 int png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1}; 1958 static PNG_CONST png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
1776 1959
1777 /* Offset to next interlace block in the y direction */ 1960 /* Offset to next interlace block in the y direction */
1778 int png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2}; 1961 static PNG_CONST png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
1779 #endif
1780 #endif 1962 #endif
1781 1963
1782 png_size_t buf_size; 1964 png_alloc_size_t buf_size;
1965 int usr_pixel_depth;
1966
1967 #ifdef PNG_WRITE_FILTER_SUPPORTED
1968 png_byte filters;
1969 #endif
1783 1970
1784 png_debug(1, "in png_write_start_row"); 1971 png_debug(1, "in png_write_start_row");
1785 1972
1786 buf_size = (png_size_t)(PNG_ROWBYTES( 1973 usr_pixel_depth = png_ptr->usr_channels * png_ptr->usr_bit_depth;
1787 png_ptr->usr_channels*png_ptr->usr_bit_depth, png_ptr->width) + 1); 1974 buf_size = PNG_ROWBYTES(usr_pixel_depth, png_ptr->width) + 1;
1975
1976 /* 1.5.6: added to allow checking in the row write code. */
1977 png_ptr->transformed_pixel_depth = png_ptr->pixel_depth;
1978 png_ptr->maximum_pixel_depth = (png_byte)usr_pixel_depth;
1788 1979
1789 /* Set up row buffer */ 1980 /* Set up row buffer */
1790 png_ptr->row_buf = (png_bytep)png_malloc(png_ptr, 1981 png_ptr->row_buf = png_voidcast(png_bytep, png_malloc(png_ptr, buf_size));
1791 (png_uint_32)buf_size); 1982
1792 png_ptr->row_buf[0] = PNG_FILTER_VALUE_NONE; 1983 png_ptr->row_buf[0] = PNG_FILTER_VALUE_NONE;
1793 1984
1794 #ifdef PNG_WRITE_FILTER_SUPPORTED 1985 #ifdef PNG_WRITE_FILTER_SUPPORTED
1795 /* Set up filtering buffer, if using this filter */ 1986 filters = png_ptr->do_filter;
1796 if (png_ptr->do_filter & PNG_FILTER_SUB) 1987
1988 if (png_ptr->height == 1)
1989 filters &= 0xff & ~(PNG_FILTER_UP|PNG_FILTER_AVG|PNG_FILTER_PAETH);
1990
1991 if (png_ptr->width == 1)
1992 filters &= 0xff & ~(PNG_FILTER_SUB|PNG_FILTER_AVG|PNG_FILTER_PAETH);
1993
1994 if (filters == 0)
1995 filters = PNG_FILTER_NONE;
1996
1997 png_ptr->do_filter = filters;
1998
1999 if (((filters & (PNG_FILTER_SUB | PNG_FILTER_UP | PNG_FILTER_AVG |
2000 PNG_FILTER_PAETH)) != 0) && png_ptr->try_row == NULL)
1797 { 2001 {
1798 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr, 2002 int num_filters = 0;
1799 (png_uint_32)(png_ptr->rowbytes + 1)); 2003
1800 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB; 2004 png_ptr->try_row = png_voidcast(png_bytep, png_malloc(png_ptr, buf_size));
2005
2006 if (filters & PNG_FILTER_SUB)
2007 num_filters++;
2008
2009 if (filters & PNG_FILTER_UP)
2010 num_filters++;
2011
2012 if (filters & PNG_FILTER_AVG)
2013 num_filters++;
2014
2015 if (filters & PNG_FILTER_PAETH)
2016 num_filters++;
2017
2018 if (num_filters > 1)
2019 png_ptr->tst_row = png_voidcast(png_bytep, png_malloc(png_ptr,
2020 buf_size));
1801 } 2021 }
1802 2022
1803 /* We only need to keep the previous row if we are using one of these. */ 2023 /* We only need to keep the previous row if we are using one of the following
1804 if (png_ptr->do_filter & (PNG_FILTER_AVG | PNG_FILTER_UP | PNG_FILTER_PAETH)) 2024 * filters.
1805 { 2025 */
1806 /* Set up previous row buffer */ 2026 if ((filters & (PNG_FILTER_AVG | PNG_FILTER_UP | PNG_FILTER_PAETH)) != 0)
1807 png_ptr->prev_row = (png_bytep)png_calloc(png_ptr, 2027 png_ptr->prev_row = png_voidcast(png_bytep,
1808 (png_uint_32)buf_size); 2028 png_calloc(png_ptr, buf_size));
1809 2029 #endif /* WRITE_FILTER */
1810 if (png_ptr->do_filter & PNG_FILTER_UP)
1811 {
1812 png_ptr->up_row = (png_bytep)png_malloc(png_ptr,
1813 (png_uint_32)(png_ptr->rowbytes + 1));
1814 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP;
1815 }
1816
1817 if (png_ptr->do_filter & PNG_FILTER_AVG)
1818 {
1819 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr,
1820 (png_uint_32)(png_ptr->rowbytes + 1));
1821 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG;
1822 }
1823
1824 if (png_ptr->do_filter & PNG_FILTER_PAETH)
1825 {
1826 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr,
1827 (png_uint_32)(png_ptr->rowbytes + 1));
1828 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH;
1829 }
1830 }
1831 #endif /* PNG_WRITE_FILTER_SUPPORTED */
1832 2030
1833 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 2031 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1834 /* If interlaced, we need to set up width and height of pass */ 2032 /* If interlaced, we need to set up width and height of pass */
1835 if (png_ptr->interlaced) 2033 if (png_ptr->interlaced != 0)
1836 { 2034 {
1837 if (!(png_ptr->transformations & PNG_INTERLACE)) 2035 if ((png_ptr->transformations & PNG_INTERLACE) == 0)
1838 { 2036 {
1839 png_ptr->num_rows = (png_ptr->height + png_pass_yinc[0] - 1 - 2037 png_ptr->num_rows = (png_ptr->height + png_pass_yinc[0] - 1 -
1840 png_pass_ystart[0]) / png_pass_yinc[0]; 2038 png_pass_ystart[0]) / png_pass_yinc[0];
2039
1841 png_ptr->usr_width = (png_ptr->width + png_pass_inc[0] - 1 - 2040 png_ptr->usr_width = (png_ptr->width + png_pass_inc[0] - 1 -
1842 png_pass_start[0]) / png_pass_inc[0]; 2041 png_pass_start[0]) / png_pass_inc[0];
1843 } 2042 }
2043
1844 else 2044 else
1845 { 2045 {
1846 png_ptr->num_rows = png_ptr->height; 2046 png_ptr->num_rows = png_ptr->height;
1847 png_ptr->usr_width = png_ptr->width; 2047 png_ptr->usr_width = png_ptr->width;
1848 } 2048 }
1849 } 2049 }
2050
1850 else 2051 else
1851 #endif 2052 #endif
1852 { 2053 {
1853 png_ptr->num_rows = png_ptr->height; 2054 png_ptr->num_rows = png_ptr->height;
1854 png_ptr->usr_width = png_ptr->width; 2055 png_ptr->usr_width = png_ptr->width;
1855 } 2056 }
1856 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
1857 png_ptr->zstream.next_out = png_ptr->zbuf;
1858 } 2057 }
1859 2058
1860 /* Internal use only. Called when finished processing a row of data. */ 2059 /* Internal use only. Called when finished processing a row of data. */
1861 void /* PRIVATE */ 2060 void /* PRIVATE */
1862 png_write_finish_row(png_structp png_ptr) 2061 png_write_finish_row(png_structrp png_ptr)
1863 { 2062 {
1864 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 2063 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1865 #ifndef PNG_USE_GLOBAL_ARRAYS
1866 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */ 2064 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1867 2065
1868 /* Start of interlace block */ 2066 /* Start of interlace block */
1869 int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0}; 2067 static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1870 2068
1871 /* Offset to next interlace block */ 2069 /* Offset to next interlace block */
1872 int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1}; 2070 static PNG_CONST png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1873 2071
1874 /* Start of interlace block in the y direction */ 2072 /* Start of interlace block in the y direction */
1875 int png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1}; 2073 static PNG_CONST png_byte png_pass_ystart[7] = {0, 0, 4, 0, 2, 0, 1};
1876 2074
1877 /* Offset to next interlace block in the y direction */ 2075 /* Offset to next interlace block in the y direction */
1878 int png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2}; 2076 static PNG_CONST png_byte png_pass_yinc[7] = {8, 8, 8, 4, 4, 2, 2};
1879 #endif 2077 #endif
1880 #endif
1881
1882 int ret;
1883 2078
1884 png_debug(1, "in png_write_finish_row"); 2079 png_debug(1, "in png_write_finish_row");
1885 2080
1886 /* Next row */ 2081 /* Next row */
1887 png_ptr->row_number++; 2082 png_ptr->row_number++;
1888 2083
1889 /* See if we are done */ 2084 /* See if we are done */
1890 if (png_ptr->row_number < png_ptr->num_rows) 2085 if (png_ptr->row_number < png_ptr->num_rows)
1891 return; 2086 return;
1892 2087
1893 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 2088 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1894 /* If interlaced, go to next pass */ 2089 /* If interlaced, go to next pass */
1895 if (png_ptr->interlaced) 2090 if (png_ptr->interlaced != 0)
1896 { 2091 {
1897 png_ptr->row_number = 0; 2092 png_ptr->row_number = 0;
1898 if (png_ptr->transformations & PNG_INTERLACE) 2093 if ((png_ptr->transformations & PNG_INTERLACE) != 0)
1899 { 2094 {
1900 png_ptr->pass++; 2095 png_ptr->pass++;
1901 } 2096 }
2097
1902 else 2098 else
1903 { 2099 {
1904 /* Loop until we find a non-zero width or height pass */ 2100 /* Loop until we find a non-zero width or height pass */
1905 do 2101 do
1906 { 2102 {
1907 png_ptr->pass++; 2103 png_ptr->pass++;
2104
1908 if (png_ptr->pass >= 7) 2105 if (png_ptr->pass >= 7)
1909 break; 2106 break;
2107
1910 png_ptr->usr_width = (png_ptr->width + 2108 png_ptr->usr_width = (png_ptr->width +
1911 png_pass_inc[png_ptr->pass] - 1 - 2109 png_pass_inc[png_ptr->pass] - 1 -
1912 png_pass_start[png_ptr->pass]) / 2110 png_pass_start[png_ptr->pass]) /
1913 png_pass_inc[png_ptr->pass]; 2111 png_pass_inc[png_ptr->pass];
2112
1914 png_ptr->num_rows = (png_ptr->height + 2113 png_ptr->num_rows = (png_ptr->height +
1915 png_pass_yinc[png_ptr->pass] - 1 - 2114 png_pass_yinc[png_ptr->pass] - 1 -
1916 png_pass_ystart[png_ptr->pass]) / 2115 png_pass_ystart[png_ptr->pass]) /
1917 png_pass_yinc[png_ptr->pass]; 2116 png_pass_yinc[png_ptr->pass];
1918 if (png_ptr->transformations & PNG_INTERLACE) 2117
2118 if ((png_ptr->transformations & PNG_INTERLACE) != 0)
1919 break; 2119 break;
2120
1920 } while (png_ptr->usr_width == 0 || png_ptr->num_rows == 0); 2121 } while (png_ptr->usr_width == 0 || png_ptr->num_rows == 0);
1921 2122
1922 } 2123 }
1923 2124
1924 /* Reset the row above the image for the next pass */ 2125 /* Reset the row above the image for the next pass */
1925 if (png_ptr->pass < 7) 2126 if (png_ptr->pass < 7)
1926 { 2127 {
1927 if (png_ptr->prev_row != NULL) 2128 if (png_ptr->prev_row != NULL)
1928 png_memset(png_ptr->prev_row, 0, 2129 memset(png_ptr->prev_row, 0,
1929 (png_size_t)(PNG_ROWBYTES(png_ptr->usr_channels* 2130 (png_size_t)(PNG_ROWBYTES(png_ptr->usr_channels*
1930 png_ptr->usr_bit_depth, png_ptr->width)) + 1); 2131 png_ptr->usr_bit_depth, png_ptr->width)) + 1);
2132
1931 return; 2133 return;
1932 } 2134 }
1933 } 2135 }
1934 #endif 2136 #endif
1935 2137
1936 /* If we get here, we've just written the last row, so we need 2138 /* If we get here, we've just written the last row, so we need
1937 to flush the compressor */ 2139 to flush the compressor */
1938 do 2140 png_compress_IDAT(png_ptr, NULL, 0, Z_FINISH);
1939 {
1940 /* Tell the compressor we are done */
1941 ret = deflate(&png_ptr->zstream, Z_FINISH);
1942 /* Check for an error */
1943 if (ret == Z_OK)
1944 {
1945 /* Check to see if we need more room */
1946 if (!(png_ptr->zstream.avail_out))
1947 {
1948 png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size);
1949 png_ptr->zstream.next_out = png_ptr->zbuf;
1950 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
1951 }
1952 }
1953 else if (ret != Z_STREAM_END)
1954 {
1955 if (png_ptr->zstream.msg != NULL)
1956 png_error(png_ptr, png_ptr->zstream.msg);
1957 else
1958 png_error(png_ptr, "zlib error");
1959 }
1960 } while (ret != Z_STREAM_END);
1961
1962 /* Write any extra space */
1963 if (png_ptr->zstream.avail_out < png_ptr->zbuf_size)
1964 {
1965 png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size -
1966 png_ptr->zstream.avail_out);
1967 }
1968
1969 deflateReset(&png_ptr->zstream);
1970 png_ptr->zstream.data_type = Z_BINARY;
1971 } 2141 }
1972 2142
1973 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 2143 #ifdef PNG_WRITE_INTERLACING_SUPPORTED
1974 /* Pick out the correct pixels for the interlace pass. 2144 /* Pick out the correct pixels for the interlace pass.
1975 * The basic idea here is to go through the row with a source 2145 * The basic idea here is to go through the row with a source
1976 * pointer and a destination pointer (sp and dp), and copy the 2146 * pointer and a destination pointer (sp and dp), and copy the
1977 * correct pixels for the pass. As the row gets compacted, 2147 * correct pixels for the pass. As the row gets compacted,
1978 * sp will always be >= dp, so we should never overwrite anything. 2148 * sp will always be >= dp, so we should never overwrite anything.
1979 * See the default: case for the easiest code to understand. 2149 * See the default: case for the easiest code to understand.
1980 */ 2150 */
1981 void /* PRIVATE */ 2151 void /* PRIVATE */
1982 png_do_write_interlace(png_row_infop row_info, png_bytep row, int pass) 2152 png_do_write_interlace(png_row_infop row_info, png_bytep row, int pass)
1983 { 2153 {
1984 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */ 2154 /* Arrays to facilitate easy interlacing - use pass (0 - 6) as index */
1985 2155
1986 #ifndef PNG_USE_GLOBAL_ARRAYS
1987 /* Start of interlace block */ 2156 /* Start of interlace block */
1988 int png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0}; 2157 static PNG_CONST png_byte png_pass_start[7] = {0, 4, 0, 2, 0, 1, 0};
1989 2158
1990 /* Offset to next interlace block */ 2159 /* Offset to next interlace block */
1991 int png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1}; 2160 static PNG_CONST png_byte png_pass_inc[7] = {8, 8, 4, 4, 2, 2, 1};
1992 #endif
1993 2161
1994 png_debug(1, "in png_do_write_interlace"); 2162 png_debug(1, "in png_do_write_interlace");
1995 2163
1996 /* We don't have to do anything on the last pass (6) */ 2164 /* We don't have to do anything on the last pass (6) */
1997 #ifdef PNG_USELESS_TESTS_SUPPORTED
1998 if (row != NULL && row_info != NULL && pass < 6)
1999 #else
2000 if (pass < 6) 2165 if (pass < 6)
2001 #endif
2002 { 2166 {
2003 /* Each pixel depth is handled separately */ 2167 /* Each pixel depth is handled separately */
2004 switch (row_info->pixel_depth) 2168 switch (row_info->pixel_depth)
2005 { 2169 {
2006 case 1: 2170 case 1:
2007 { 2171 {
2008 png_bytep sp; 2172 png_bytep sp;
2009 png_bytep dp; 2173 png_bytep dp;
2010 int shift; 2174 unsigned int shift;
2011 int d; 2175 int d;
2012 int value; 2176 int value;
2013 png_uint_32 i; 2177 png_uint_32 i;
2014 png_uint_32 row_width = row_info->width; 2178 png_uint_32 row_width = row_info->width;
2015 2179
2016 dp = row; 2180 dp = row;
2017 d = 0; 2181 d = 0;
2018 shift = 7; 2182 shift = 7;
2183
2019 for (i = png_pass_start[pass]; i < row_width; 2184 for (i = png_pass_start[pass]; i < row_width;
2020 i += png_pass_inc[pass]) 2185 i += png_pass_inc[pass])
2021 { 2186 {
2022 sp = row + (png_size_t)(i >> 3); 2187 sp = row + (png_size_t)(i >> 3);
2023 value = (int)(*sp >> (7 - (int)(i & 0x07))) & 0x01; 2188 value = (int)(*sp >> (7 - (int)(i & 0x07))) & 0x01;
2024 d |= (value << shift); 2189 d |= (value << shift);
2025 2190
2026 if (shift == 0) 2191 if (shift == 0)
2027 { 2192 {
2028 shift = 7; 2193 shift = 7;
2029 *dp++ = (png_byte)d; 2194 *dp++ = (png_byte)d;
2030 d = 0; 2195 d = 0;
2031 } 2196 }
2197
2032 else 2198 else
2033 shift--; 2199 shift--;
2034 2200
2035 } 2201 }
2036 if (shift != 7) 2202 if (shift != 7)
2037 *dp = (png_byte)d; 2203 *dp = (png_byte)d;
2204
2038 break; 2205 break;
2039 } 2206 }
2207
2040 case 2: 2208 case 2:
2041 { 2209 {
2042 png_bytep sp; 2210 png_bytep sp;
2043 png_bytep dp; 2211 png_bytep dp;
2044 int shift; 2212 unsigned int shift;
2045 int d; 2213 int d;
2046 int value; 2214 int value;
2047 png_uint_32 i; 2215 png_uint_32 i;
2048 png_uint_32 row_width = row_info->width; 2216 png_uint_32 row_width = row_info->width;
2049 2217
2050 dp = row; 2218 dp = row;
2051 shift = 6; 2219 shift = 6;
2052 d = 0; 2220 d = 0;
2221
2053 for (i = png_pass_start[pass]; i < row_width; 2222 for (i = png_pass_start[pass]; i < row_width;
2054 i += png_pass_inc[pass]) 2223 i += png_pass_inc[pass])
2055 { 2224 {
2056 sp = row + (png_size_t)(i >> 2); 2225 sp = row + (png_size_t)(i >> 2);
2057 value = (*sp >> ((3 - (int)(i & 0x03)) << 1)) & 0x03; 2226 value = (*sp >> ((3 - (int)(i & 0x03)) << 1)) & 0x03;
2058 d |= (value << shift); 2227 d |= (value << shift);
2059 2228
2060 if (shift == 0) 2229 if (shift == 0)
2061 { 2230 {
2062 shift = 6; 2231 shift = 6;
2063 *dp++ = (png_byte)d; 2232 *dp++ = (png_byte)d;
2064 d = 0; 2233 d = 0;
2065 } 2234 }
2235
2066 else 2236 else
2067 shift -= 2; 2237 shift -= 2;
2068 } 2238 }
2069 if (shift != 6) 2239 if (shift != 6)
2070 *dp = (png_byte)d; 2240 *dp = (png_byte)d;
2241
2071 break; 2242 break;
2072 } 2243 }
2244
2073 case 4: 2245 case 4:
2074 { 2246 {
2075 png_bytep sp; 2247 png_bytep sp;
2076 png_bytep dp; 2248 png_bytep dp;
2077 int shift; 2249 unsigned int shift;
2078 int d; 2250 int d;
2079 int value; 2251 int value;
2080 png_uint_32 i; 2252 png_uint_32 i;
2081 png_uint_32 row_width = row_info->width; 2253 png_uint_32 row_width = row_info->width;
2082 2254
2083 dp = row; 2255 dp = row;
2084 shift = 4; 2256 shift = 4;
2085 d = 0; 2257 d = 0;
2086 for (i = png_pass_start[pass]; i < row_width; 2258 for (i = png_pass_start[pass]; i < row_width;
2087 i += png_pass_inc[pass]) 2259 i += png_pass_inc[pass])
2088 { 2260 {
2089 sp = row + (png_size_t)(i >> 1); 2261 sp = row + (png_size_t)(i >> 1);
2090 value = (*sp >> ((1 - (int)(i & 0x01)) << 2)) & 0x0f; 2262 value = (*sp >> ((1 - (int)(i & 0x01)) << 2)) & 0x0f;
2091 d |= (value << shift); 2263 d |= (value << shift);
2092 2264
2093 if (shift == 0) 2265 if (shift == 0)
2094 { 2266 {
2095 shift = 4; 2267 shift = 4;
2096 *dp++ = (png_byte)d; 2268 *dp++ = (png_byte)d;
2097 d = 0; 2269 d = 0;
2098 } 2270 }
2271
2099 else 2272 else
2100 shift -= 4; 2273 shift -= 4;
2101 } 2274 }
2102 if (shift != 4) 2275 if (shift != 4)
2103 *dp = (png_byte)d; 2276 *dp = (png_byte)d;
2277
2104 break; 2278 break;
2105 } 2279 }
2280
2106 default: 2281 default:
2107 { 2282 {
2108 png_bytep sp; 2283 png_bytep sp;
2109 png_bytep dp; 2284 png_bytep dp;
2110 png_uint_32 i; 2285 png_uint_32 i;
2111 png_uint_32 row_width = row_info->width; 2286 png_uint_32 row_width = row_info->width;
2112 png_size_t pixel_bytes; 2287 png_size_t pixel_bytes;
2113 2288
2114 /* Start at the beginning */ 2289 /* Start at the beginning */
2115 dp = row; 2290 dp = row;
2291
2116 /* Find out how many bytes each pixel takes up */ 2292 /* Find out how many bytes each pixel takes up */
2117 pixel_bytes = (row_info->pixel_depth >> 3); 2293 pixel_bytes = (row_info->pixel_depth >> 3);
2118 /* Loop through the row, only looking at the pixels that 2294
2119 matter */ 2295 /* Loop through the row, only looking at the pixels that matter */
2120 for (i = png_pass_start[pass]; i < row_width; 2296 for (i = png_pass_start[pass]; i < row_width;
2121 i += png_pass_inc[pass]) 2297 i += png_pass_inc[pass])
2122 { 2298 {
2123 /* Find out where the original pixel is */ 2299 /* Find out where the original pixel is */
2124 sp = row + (png_size_t)i * pixel_bytes; 2300 sp = row + (png_size_t)i * pixel_bytes;
2301
2125 /* Move the pixel */ 2302 /* Move the pixel */
2126 if (dp != sp) 2303 if (dp != sp)
2127 png_memcpy(dp, sp, pixel_bytes); 2304 memcpy(dp, sp, pixel_bytes);
2305
2128 /* Next pixel */ 2306 /* Next pixel */
2129 dp += pixel_bytes; 2307 dp += pixel_bytes;
2130 } 2308 }
2131 break; 2309 break;
2132 } 2310 }
2133 } 2311 }
2134 /* Set new row width */ 2312 /* Set new row width */
2135 row_info->width = (row_info->width + 2313 row_info->width = (row_info->width +
2136 png_pass_inc[pass] - 1 - 2314 png_pass_inc[pass] - 1 -
2137 png_pass_start[pass]) / 2315 png_pass_start[pass]) /
2138 png_pass_inc[pass]; 2316 png_pass_inc[pass];
2139 row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, 2317
2140 row_info->width); 2318 row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth,
2319 row_info->width);
2141 } 2320 }
2142 } 2321 }
2143 #endif 2322 #endif
2144 2323
2324
2145 /* This filters the row, chooses which filter to use, if it has not already 2325 /* This filters the row, chooses which filter to use, if it has not already
2146 * been specified by the application, and then writes the row out with the 2326 * been specified by the application, and then writes the row out with the
2147 * chosen filter. 2327 * chosen filter.
2148 */ 2328 */
2149 #define PNG_MAXSUM (((png_uint_32)(-1)) >> 1) 2329 static void /* PRIVATE */
2150 #define PNG_HISHIFT 10 2330 png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
2151 #define PNG_LOMASK ((png_uint_32)0xffffL) 2331 png_size_t row_bytes);
2152 #define PNG_HIMASK ((png_uint_32)(~PNG_LOMASK >> PNG_HISHIFT)) 2332
2333 #ifdef PNG_WRITE_FILTER_SUPPORTED
2334 static png_size_t /* PRIVATE */
2335 png_setup_sub_row(png_structrp png_ptr, const png_uint_32 bpp,
2336 const png_size_t row_bytes, const png_size_t lmins)
2337 {
2338 png_bytep rp, dp, lp;
2339 png_size_t i;
2340 png_size_t sum = 0;
2341 int v;
2342
2343 png_ptr->try_row[0] = PNG_FILTER_VALUE_SUB;
2344
2345 for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1; i < bpp;
2346 i++, rp++, dp++)
2347 {
2348 v = *dp = *rp;
2349 sum += (v < 128) ? v : 256 - v;
2350 }
2351
2352 for (lp = png_ptr->row_buf + 1; i < row_bytes;
2353 i++, rp++, lp++, dp++)
2354 {
2355 v = *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2356 sum += (v < 128) ? v : 256 - v;
2357
2358 if (sum > lmins) /* We are already worse, don't continue. */
2359 break;
2360 }
2361
2362 return (sum);
2363 }
2364
2365 static png_size_t /* PRIVATE */
2366 png_setup_up_row(png_structrp png_ptr, const png_size_t row_bytes,
2367 const png_size_t lmins)
2368 {
2369 png_bytep rp, dp, pp;
2370 png_size_t i;
2371 png_size_t sum = 0;
2372 int v;
2373
2374 png_ptr->try_row[0] = PNG_FILTER_VALUE_UP;
2375
2376 for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2377 pp = png_ptr->prev_row + 1; i < row_bytes;
2378 i++, rp++, pp++, dp++)
2379 {
2380 v = *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
2381 sum += (v < 128) ? v : 256 - v;
2382
2383 if (sum > lmins) /* We are already worse, don't continue. */
2384 break;
2385 }
2386
2387 return (sum);
2388 }
2389
2390 static png_size_t /* PRIVATE */
2391 png_setup_avg_row(png_structrp png_ptr, const png_uint_32 bpp,
2392 const png_size_t row_bytes, const png_size_t lmins)
2393 {
2394 png_bytep rp, dp, pp, lp;
2395 png_uint_32 i;
2396 png_size_t sum = 0;
2397 int v;
2398
2399 png_ptr->try_row[0] = PNG_FILTER_VALUE_AVG;
2400
2401 for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2402 pp = png_ptr->prev_row + 1; i < bpp; i++)
2403 {
2404 v = *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2405
2406 sum += (v < 128) ? v : 256 - v;
2407 }
2408
2409 for (lp = png_ptr->row_buf + 1; i < row_bytes; i++)
2410 {
2411 v = *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
2412 & 0xff);
2413
2414 sum += (v < 128) ? v : 256 - v;
2415
2416 if (sum > lmins) /* We are already worse, don't continue. */
2417 break;
2418 }
2419
2420 return (sum);
2421 }
2422
2423 static png_size_t /* PRIVATE */
2424 png_setup_paeth_row(png_structrp png_ptr, const png_uint_32 bpp,
2425 const png_size_t row_bytes, const png_size_t lmins)
2426 {
2427 png_bytep rp, dp, pp, cp, lp;
2428 png_size_t i;
2429 png_size_t sum = 0;
2430 int v;
2431
2432 png_ptr->try_row[0] = PNG_FILTER_VALUE_PAETH;
2433
2434 for (i = 0, rp = png_ptr->row_buf + 1, dp = png_ptr->try_row + 1,
2435 pp = png_ptr->prev_row + 1; i < bpp; i++)
2436 {
2437 v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2438
2439 sum += (v < 128) ? v : 256 - v;
2440 }
2441
2442 for (lp = png_ptr->row_buf + 1, cp = png_ptr->prev_row + 1; i < row_bytes;
2443 i++)
2444 {
2445 int a, b, c, pa, pb, pc, p;
2446
2447 b = *pp++;
2448 c = *cp++;
2449 a = *lp++;
2450
2451 p = b - c;
2452 pc = a - c;
2453
2454 #ifdef PNG_USE_ABS
2455 pa = abs(p);
2456 pb = abs(pc);
2457 pc = abs(p + pc);
2458 #else
2459 pa = p < 0 ? -p : p;
2460 pb = pc < 0 ? -pc : pc;
2461 pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2462 #endif
2463
2464 p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2465
2466 v = *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2467
2468 sum += (v < 128) ? v : 256 - v;
2469
2470 if (sum > lmins) /* We are already worse, don't continue. */
2471 break;
2472 }
2473
2474 return (sum);
2475 }
2476 #endif /* WRITE_FILTER */
2477
2153 void /* PRIVATE */ 2478 void /* PRIVATE */
2154 png_write_find_filter(png_structp png_ptr, png_row_infop row_info) 2479 png_write_find_filter(png_structrp png_ptr, png_row_infop row_info)
2155 { 2480 {
2481 #ifndef PNG_WRITE_FILTER_SUPPORTED
2482 png_write_filtered_row(png_ptr, png_ptr->row_buf, row_info->rowbytes+1);
2483 #else
2484 png_byte filter_to_do = png_ptr->do_filter;
2485 png_bytep row_buf;
2156 png_bytep best_row; 2486 png_bytep best_row;
2157 #ifdef PNG_WRITE_FILTER_SUPPORTED 2487 png_uint_32 bpp;
2158 png_bytep prev_row, row_buf; 2488 png_size_t mins;
2159 png_uint_32 mins, bpp; 2489 png_size_t row_bytes = row_info->rowbytes;
2160 png_byte filter_to_do = png_ptr->do_filter;
2161 png_uint_32 row_bytes = row_info->rowbytes;
2162 2490
2163 png_debug(1, "in png_write_find_filter"); 2491 png_debug(1, "in png_write_find_filter");
2164 2492
2165 /* Find out how many bytes offset each pixel is */ 2493 /* Find out how many bytes offset each pixel is */
2166 bpp = (row_info->pixel_depth + 7) >> 3; 2494 bpp = (row_info->pixel_depth + 7) >> 3;
2167 2495
2168 prev_row = png_ptr->prev_row; 2496 row_buf = png_ptr->row_buf;
2169 #endif 2497 mins = PNG_SIZE_MAX - 256/* so we can detect potential overflow of the
2170 best_row = png_ptr->row_buf; 2498 running sum */;
2171 #ifdef PNG_WRITE_FILTER_SUPPORTED
2172 row_buf = best_row;
2173 mins = PNG_MAXSUM;
2174 2499
2175 /* The prediction method we use is to find which method provides the 2500 /* The prediction method we use is to find which method provides the
2176 * smallest value when summing the absolute values of the distances 2501 * smallest value when summing the absolute values of the distances
2177 * from zero, using anything >= 128 as negative numbers. This is known 2502 * from zero, using anything >= 128 as negative numbers. This is known
2178 * as the "minimum sum of absolute differences" heuristic. Other 2503 * as the "minimum sum of absolute differences" heuristic. Other
2179 * heuristics are the "weighted minimum sum of absolute differences" 2504 * heuristics are the "weighted minimum sum of absolute differences"
2180 * (experimental and can in theory improve compression), and the "zlib 2505 * (experimental and can in theory improve compression), and the "zlib
2181 * predictive" method (not implemented yet), which does test compressions 2506 * predictive" method (not implemented yet), which does test compressions
2182 * of lines using different filter methods, and then chooses the 2507 * of lines using different filter methods, and then chooses the
2183 * (series of) filter(s) that give minimum compressed data size (VERY 2508 * (series of) filter(s) that give minimum compressed data size (VERY
2184 * computationally expensive). 2509 * computationally expensive).
2185 * 2510 *
2186 * GRR 980525: consider also 2511 * GRR 980525: consider also
2512 *
2187 * (1) minimum sum of absolute differences from running average (i.e., 2513 * (1) minimum sum of absolute differences from running average (i.e.,
2188 * keep running sum of non-absolute differences & count of bytes) 2514 * keep running sum of non-absolute differences & count of bytes)
2189 * [track dispersion, too? restart average if dispersion too large?] 2515 * [track dispersion, too? restart average if dispersion too large?]
2516 *
2190 * (1b) minimum sum of absolute differences from sliding average, probably 2517 * (1b) minimum sum of absolute differences from sliding average, probably
2191 * with window size <= deflate window (usually 32K) 2518 * with window size <= deflate window (usually 32K)
2519 *
2192 * (2) minimum sum of squared differences from zero or running average 2520 * (2) minimum sum of squared differences from zero or running average
2193 * (i.e., ~ root-mean-square approach) 2521 * (i.e., ~ root-mean-square approach)
2194 */ 2522 */
2195 2523
2196 2524
2197 /* We don't need to test the 'no filter' case if this is the only filter 2525 /* We don't need to test the 'no filter' case if this is the only filter
2198 * that has been chosen, as it doesn't actually do anything to the data. 2526 * that has been chosen, as it doesn't actually do anything to the data.
2199 */ 2527 */
2200 if ((filter_to_do & PNG_FILTER_NONE) && 2528 best_row = png_ptr->row_buf;
2201 filter_to_do != PNG_FILTER_NONE) 2529
2530
2531 if ((filter_to_do & PNG_FILTER_NONE) != 0 && filter_to_do != PNG_FILTER_NONE)
2202 { 2532 {
2203 png_bytep rp; 2533 png_bytep rp;
2204 png_uint_32 sum = 0; 2534 png_size_t sum = 0;
2205 png_uint_32 i; 2535 png_size_t i;
2206 int v; 2536 int v;
2207 2537
2208 for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++) 2538 if (PNG_SIZE_MAX/128 <= row_bytes)
2209 { 2539 {
2210 v = *rp; 2540 for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++)
2211 sum += (v < 128) ? v : 256 - v; 2541 {
2542 /* Check for overflow */
2543 if (sum > PNG_SIZE_MAX/128 - 256)
2544 break;
2545
2546 v = *rp;
2547 sum += (v < 128) ? v : 256 - v;
2548 }
2549 }
2550 else /* Overflow is not possible */
2551 {
2552 for (i = 0, rp = row_buf + 1; i < row_bytes; i++, rp++)
2553 {
2554 v = *rp;
2555 sum += (v < 128) ? v : 256 - v;
2556 }
2212 } 2557 }
2213 2558
2214 mins = sum; 2559 mins = sum;
2215 } 2560 }
2216 2561
2217 /* Sub filter */ 2562 /* Sub filter */
2218 if (filter_to_do == PNG_FILTER_SUB) 2563 if (filter_to_do == PNG_FILTER_SUB)
2219 /* It's the only filter so no testing is needed */ 2564 /* It's the only filter so no testing is needed */
2220 { 2565 {
2221 png_bytep rp, lp, dp; 2566 (void) png_setup_sub_row(png_ptr, bpp, row_bytes, mins);
2222 png_uint_32 i; 2567 best_row = png_ptr->try_row;
2223 for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp;
2224 i++, rp++, dp++)
2225 {
2226 *dp = *rp;
2227 }
2228 for (lp = row_buf + 1; i < row_bytes;
2229 i++, rp++, lp++, dp++)
2230 {
2231 *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2232 }
2233 best_row = png_ptr->sub_row;
2234 } 2568 }
2235 2569
2236 else if (filter_to_do & PNG_FILTER_SUB) 2570 else if ((filter_to_do & PNG_FILTER_SUB) != 0)
2237 { 2571 {
2238 png_bytep rp, dp, lp; 2572 png_size_t sum;
2239 png_uint_32 sum = 0, lmins = mins; 2573 png_size_t lmins = mins;
2240 png_uint_32 i;
2241 int v;
2242 2574
2243 for (i = 0, rp = row_buf + 1, dp = png_ptr->sub_row + 1; i < bpp; 2575 sum = png_setup_sub_row(png_ptr, bpp, row_bytes, lmins);
2244 i++, rp++, dp++)
2245 {
2246 v = *dp = *rp;
2247
2248 sum += (v < 128) ? v : 256 - v;
2249 }
2250 for (lp = row_buf + 1; i < row_bytes;
2251 i++, rp++, lp++, dp++)
2252 {
2253 v = *dp = (png_byte)(((int)*rp - (int)*lp) & 0xff);
2254
2255 sum += (v < 128) ? v : 256 - v;
2256
2257 if (sum > lmins) /* We are already worse, don't continue. */
2258 break;
2259 }
2260 2576
2261 if (sum < mins) 2577 if (sum < mins)
2262 { 2578 {
2263 mins = sum; 2579 mins = sum;
2264 best_row = png_ptr->sub_row; 2580 best_row = png_ptr->try_row;
2581 if (png_ptr->tst_row != NULL)
2582 {
2583 png_ptr->try_row = png_ptr->tst_row;
2584 png_ptr->tst_row = best_row;
2585 }
2265 } 2586 }
2266 } 2587 }
2267 2588
2268 /* Up filter */ 2589 /* Up filter */
2269 if (filter_to_do == PNG_FILTER_UP) 2590 if (filter_to_do == PNG_FILTER_UP)
2270 { 2591 {
2271 png_bytep rp, dp, pp; 2592 (void) png_setup_up_row(png_ptr, row_bytes, mins);
2272 png_uint_32 i; 2593 best_row = png_ptr->try_row;
2273
2274 for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1,
2275 pp = prev_row + 1; i < row_bytes;
2276 i++, rp++, pp++, dp++)
2277 {
2278 *dp = (png_byte)(((int)*rp - (int)*pp) & 0xff);
2279 }
2280 best_row = png_ptr->up_row;
2281 } 2594 }
2282 2595
2283 else if (filter_to_do & PNG_FILTER_UP) 2596 else if ((filter_to_do & PNG_FILTER_UP) != 0)
2284 { 2597 {
2285 png_bytep rp, dp, pp; 2598 png_size_t sum;
2286 png_uint_32 sum = 0, lmins = mins; 2599 png_size_t lmins = mins;
2287 png_uint_32 i;
2288 int v;
2289 2600
2290 for (i = 0, rp = row_buf + 1, dp = png_ptr->up_row + 1, 2601 sum = png_setup_up_row(png_ptr, row_bytes, lmins);
2291 pp = prev_row + 1; i < row_bytes; i++)
2292 {
2293 v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2294
2295 sum += (v < 128) ? v : 256 - v;
2296
2297 if (sum > lmins) /* We are already worse, don't continue. */
2298 break;
2299 }
2300 2602
2301 if (sum < mins) 2603 if (sum < mins)
2302 { 2604 {
2303 mins = sum; 2605 mins = sum;
2304 best_row = png_ptr->up_row; 2606 best_row = png_ptr->try_row;
2607 if (png_ptr->tst_row != NULL)
2608 {
2609 png_ptr->try_row = png_ptr->tst_row;
2610 png_ptr->tst_row = best_row;
2611 }
2305 } 2612 }
2306 } 2613 }
2307 2614
2308 /* Avg filter */ 2615 /* Avg filter */
2309 if (filter_to_do == PNG_FILTER_AVG) 2616 if (filter_to_do == PNG_FILTER_AVG)
2310 { 2617 {
2311 png_bytep rp, dp, pp, lp; 2618 (void) png_setup_avg_row(png_ptr, bpp, row_bytes, mins);
2312 png_uint_32 i; 2619 best_row = png_ptr->try_row;
2313 for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1,
2314 pp = prev_row + 1; i < bpp; i++)
2315 {
2316 *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2317 }
2318 for (lp = row_buf + 1; i < row_bytes; i++)
2319 {
2320 *dp++ = (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2))
2321 & 0xff);
2322 }
2323 best_row = png_ptr->avg_row;
2324 } 2620 }
2325 2621
2326 else if (filter_to_do & PNG_FILTER_AVG) 2622 else if ((filter_to_do & PNG_FILTER_AVG) != 0)
2327 { 2623 {
2328 png_bytep rp, dp, pp, lp; 2624 png_size_t sum;
2329 png_uint_32 sum = 0, lmins = mins; 2625 png_size_t lmins = mins;
2330 png_uint_32 i;
2331 int v;
2332 2626
2333 for (i = 0, rp = row_buf + 1, dp = png_ptr->avg_row + 1, 2627 sum= png_setup_avg_row(png_ptr, bpp, row_bytes, lmins);
2334 pp = prev_row + 1; i < bpp; i++)
2335 {
2336 v = *dp++ = (png_byte)(((int)*rp++ - ((int)*pp++ / 2)) & 0xff);
2337
2338 sum += (v < 128) ? v : 256 - v;
2339 }
2340 for (lp = row_buf + 1; i < row_bytes; i++)
2341 {
2342 v = *dp++ =
2343 (png_byte)(((int)*rp++ - (((int)*pp++ + (int)*lp++) / 2)) & 0xff);
2344
2345 sum += (v < 128) ? v : 256 - v;
2346
2347 if (sum > lmins) /* We are already worse, don't continue. */
2348 break;
2349 }
2350 2628
2351 if (sum < mins) 2629 if (sum < mins)
2352 { 2630 {
2353 mins = sum; 2631 mins = sum;
2354 best_row = png_ptr->avg_row; 2632 best_row = png_ptr->try_row;
2633 if (png_ptr->tst_row != NULL)
2634 {
2635 png_ptr->try_row = png_ptr->tst_row;
2636 png_ptr->tst_row = best_row;
2637 }
2355 } 2638 }
2356 } 2639 }
2357 2640
2358 /* Paeth filter */ 2641 /* Paeth filter */
2359 if (filter_to_do == PNG_FILTER_PAETH) 2642 if ((filter_to_do == PNG_FILTER_PAETH) != 0)
2360 { 2643 {
2361 png_bytep rp, dp, pp, cp, lp; 2644 (void) png_setup_paeth_row(png_ptr, bpp, row_bytes, mins);
2362 png_uint_32 i; 2645 best_row = png_ptr->try_row;
2363 for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1,
2364 pp = prev_row + 1; i < bpp; i++)
2365 {
2366 *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2367 }
2368
2369 for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
2370 {
2371 int a, b, c, pa, pb, pc, p;
2372
2373 b = *pp++;
2374 c = *cp++;
2375 a = *lp++;
2376
2377 p = b - c;
2378 pc = a - c;
2379
2380 #ifdef PNG_USE_ABS
2381 pa = abs(p);
2382 pb = abs(pc);
2383 pc = abs(p + pc);
2384 #else
2385 pa = p < 0 ? -p : p;
2386 pb = pc < 0 ? -pc : pc;
2387 pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2388 #endif
2389
2390 p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2391
2392 *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2393 }
2394 best_row = png_ptr->paeth_row;
2395 } 2646 }
2396 2647
2397 else if (filter_to_do & PNG_FILTER_PAETH) 2648 else if ((filter_to_do & PNG_FILTER_PAETH) != 0)
2398 { 2649 {
2399 png_bytep rp, dp, pp, cp, lp; 2650 png_size_t sum;
2400 png_uint_32 sum = 0, lmins = mins; 2651 png_size_t lmins = mins;
2401 png_uint_32 i;
2402 int v;
2403 2652
2404 for (i = 0, rp = row_buf + 1, dp = png_ptr->paeth_row + 1, 2653 sum = png_setup_paeth_row(png_ptr, bpp, row_bytes, lmins);
2405 pp = prev_row + 1; i < bpp; i++)
2406 {
2407 v = *dp++ = (png_byte)(((int)*rp++ - (int)*pp++) & 0xff);
2408
2409 sum += (v < 128) ? v : 256 - v;
2410 }
2411
2412 for (lp = row_buf + 1, cp = prev_row + 1; i < row_bytes; i++)
2413 {
2414 int a, b, c, pa, pb, pc, p;
2415
2416 b = *pp++;
2417 c = *cp++;
2418 a = *lp++;
2419
2420 #ifndef PNG_SLOW_PAETH
2421 p = b - c;
2422 pc = a - c;
2423 #ifdef PNG_USE_ABS
2424 pa = abs(p);
2425 pb = abs(pc);
2426 pc = abs(p + pc);
2427 #else
2428 pa = p < 0 ? -p : p;
2429 pb = pc < 0 ? -pc : pc;
2430 pc = (p + pc) < 0 ? -(p + pc) : p + pc;
2431 #endif
2432 p = (pa <= pb && pa <=pc) ? a : (pb <= pc) ? b : c;
2433 #else /* PNG_SLOW_PAETH */
2434 p = a + b - c;
2435 pa = abs(p - a);
2436 pb = abs(p - b);
2437 pc = abs(p - c);
2438 if (pa <= pb && pa <= pc)
2439 p = a;
2440 else if (pb <= pc)
2441 p = b;
2442 else
2443 p = c;
2444 #endif /* PNG_SLOW_PAETH */
2445
2446 v = *dp++ = (png_byte)(((int)*rp++ - p) & 0xff);
2447
2448 sum += (v < 128) ? v : 256 - v;
2449
2450 if (sum > lmins) /* We are already worse, don't continue. */
2451 break;
2452 }
2453 2654
2454 if (sum < mins) 2655 if (sum < mins)
2455 { 2656 {
2456 best_row = png_ptr->paeth_row; 2657 best_row = png_ptr->try_row;
2658 if (png_ptr->tst_row != NULL)
2659 {
2660 png_ptr->try_row = png_ptr->tst_row;
2661 png_ptr->tst_row = best_row;
2662 }
2457 } 2663 }
2458 } 2664 }
2459 #endif /* PNG_WRITE_FILTER_SUPPORTED */ 2665
2460 /* Do the actual writing of the filtered row data from the chosen filter. */ 2666 /* Do the actual writing of the filtered row data from the chosen filter. */
2667 png_write_filtered_row(png_ptr, best_row, row_info->rowbytes+1);
2461 2668
2462 png_write_filtered_row(png_ptr, best_row); 2669 #endif /* WRITE_FILTER */
2463 } 2670 }
2464 2671
2465 2672
2466 /* Do the actual writing of a previously filtered row. */ 2673 /* Do the actual writing of a previously filtered row. */
2467 void /* PRIVATE */ 2674 static void
2468 png_write_filtered_row(png_structp png_ptr, png_bytep filtered_row) 2675 png_write_filtered_row(png_structrp png_ptr, png_bytep filtered_row,
2676 png_size_t full_row_length/*includes filter byte*/)
2469 { 2677 {
2470 png_debug(1, "in png_write_filtered_row"); 2678 png_debug(1, "in png_write_filtered_row");
2471 2679
2472 png_debug1(2, "filter = %d", filtered_row[0]); 2680 png_debug1(2, "filter = %d", filtered_row[0]);
2473 /* Set up the zlib input buffer */
2474 2681
2475 png_ptr->zstream.next_in = filtered_row; 2682 png_compress_IDAT(png_ptr, filtered_row, full_row_length, Z_NO_FLUSH);
2476 png_ptr->zstream.avail_in = (uInt)png_ptr->row_info.rowbytes + 1;
2477 /* Repeat until we have compressed all the data */
2478 do
2479 {
2480 int ret; /* Return of zlib */
2481 2683
2482 /* Compress the data */ 2684 #ifdef PNG_WRITE_FILTER_SUPPORTED
2483 ret = deflate(&png_ptr->zstream, Z_NO_FLUSH);
2484 /* Check for compression errors */
2485 if (ret != Z_OK)
2486 {
2487 if (png_ptr->zstream.msg != NULL)
2488 png_error(png_ptr, png_ptr->zstream.msg);
2489 else
2490 png_error(png_ptr, "zlib error");
2491 }
2492
2493 /* See if it is time to write another IDAT */
2494 if (!(png_ptr->zstream.avail_out))
2495 {
2496 /* Write the IDAT and reset the zlib output buffer */
2497 png_write_IDAT(png_ptr, png_ptr->zbuf, png_ptr->zbuf_size);
2498 png_ptr->zstream.next_out = png_ptr->zbuf;
2499 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size;
2500 }
2501 /* Repeat until all data has been compressed */
2502 } while (png_ptr->zstream.avail_in);
2503
2504 /* Swap the current and previous rows */ 2685 /* Swap the current and previous rows */
2505 if (png_ptr->prev_row != NULL) 2686 if (png_ptr->prev_row != NULL)
2506 { 2687 {
2507 png_bytep tptr; 2688 png_bytep tptr;
2508 2689
2509 tptr = png_ptr->prev_row; 2690 tptr = png_ptr->prev_row;
2510 png_ptr->prev_row = png_ptr->row_buf; 2691 png_ptr->prev_row = png_ptr->row_buf;
2511 png_ptr->row_buf = tptr; 2692 png_ptr->row_buf = tptr;
2512 } 2693 }
2694 #endif /* WRITE_FILTER */
2513 2695
2514 /* Finish row - updates counters and flushes zlib if last row */ 2696 /* Finish row - updates counters and flushes zlib if last row */
2515 png_write_finish_row(png_ptr); 2697 png_write_finish_row(png_ptr);
2516 2698
2517 #ifdef PNG_WRITE_FLUSH_SUPPORTED 2699 #ifdef PNG_WRITE_FLUSH_SUPPORTED
2518 png_ptr->flush_rows++; 2700 png_ptr->flush_rows++;
2519 2701
2520 if (png_ptr->flush_dist > 0 && 2702 if (png_ptr->flush_dist > 0 &&
2521 png_ptr->flush_rows >= png_ptr->flush_dist) 2703 png_ptr->flush_rows >= png_ptr->flush_dist)
2522 { 2704 {
2523 png_write_flush(png_ptr); 2705 png_write_flush(png_ptr);
2524 } 2706 }
2525 #endif 2707 #endif /* WRITE_FLUSH */
2526 } 2708 }
2527 #endif /* PNG_WRITE_SUPPORTED */ 2709 #endif /* WRITE */
OLDNEW
« no previous file with comments | « third_party/libpng/pngwtran.c ('k') | tools/run-perf-test.cfg » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698