Adjust proto_tree_add_[float|double]_format_value calls to use unit string
[metze/wireshark/wip.git] / epan / dissectors / file-png.c
1 /* file-png.c
2  *
3  * Routines for PNG (Portable Network Graphics) image file dissection
4  *
5  * Copyright 2006 Ronnie Sahlberg
6  *
7  * Wireshark - Network traffic analyzer
8  * By Gerald Combs <gerald@wireshark.org>
9  * Copyright 1998 Gerald Combs
10  *
11  * This program is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU General Public License
13  * as published by the Free Software Foundation; either version 2
14  * of the License, or (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24  */
25 /* See http://www.w3.org/TR/PNG for specification
26  */
27
28 #define NEW_PROTO_TREE_API
29
30 #include "config.h"
31
32 #include <epan/packet.h>
33 #include <epan/expert.h>
34
35
36 #define MAKE_TYPE_VAL(a, b, c, d)   ((a)<<24 | (b)<<16 | (c)<<8 | (d))
37
38 #define CHUNK_TYPE_IHDR   MAKE_TYPE_VAL('I', 'H', 'D', 'R')
39 #define CHUNK_TYPE_bKGD   MAKE_TYPE_VAL('b', 'K', 'G', 'D')
40 #define CHUNK_TYPE_gAMA   MAKE_TYPE_VAL('g', 'A', 'M', 'A')
41 #define CHUNK_TYPE_iCCP   MAKE_TYPE_VAL('i', 'C', 'C', 'P')
42 #define CHUNK_TYPE_cHRM   MAKE_TYPE_VAL('c', 'H', 'R', 'M')
43 #define CHUNK_TYPE_pHYs   MAKE_TYPE_VAL('p', 'H', 'Y', 's')
44 #define CHUNK_TYPE_iTXt   MAKE_TYPE_VAL('i', 'T', 'X', 't')
45 #define CHUNK_TYPE_tEXt   MAKE_TYPE_VAL('t', 'E', 'X', 't')
46 #define CHUNK_TYPE_sBIT   MAKE_TYPE_VAL('s', 'B', 'I', 'T')
47 #define CHUNK_TYPE_sRGB   MAKE_TYPE_VAL('s', 'R', 'G', 'B')
48 #define CHUNK_TYPE_tIME   MAKE_TYPE_VAL('t', 'I', 'M', 'E')
49 #define CHUNK_TYPE_IDAT   MAKE_TYPE_VAL('I', 'D', 'A', 'T')
50 #define CHUNK_TYPE_IEND   MAKE_TYPE_VAL('I', 'E', 'N', 'D')
51 #define CHUNK_TYPE_tRNS   MAKE_TYPE_VAL('t', 'R', 'N', 'S')
52 #define CHUNK_TYPE_PLTE   MAKE_TYPE_VAL('P', 'L', 'T', 'E')
53
54 static const value_string chunk_types[] = {
55     { CHUNK_TYPE_IHDR, "Image Header" },
56     { CHUNK_TYPE_bKGD, "Background colour" },
57     { CHUNK_TYPE_gAMA, "Image gamma" },
58     { CHUNK_TYPE_iCCP, "Embedded ICC profile" },
59     { CHUNK_TYPE_cHRM, "Primary chromaticities and white point" },
60     { CHUNK_TYPE_pHYs, "Physical pixel dimensions" },
61     { CHUNK_TYPE_iTXt, "International textual data" },
62     { CHUNK_TYPE_tEXt, "Textual data" },
63     { CHUNK_TYPE_sBIT, "Significant bits" },
64     { CHUNK_TYPE_sRGB, "Standard RGB colour space" },
65     { CHUNK_TYPE_tIME, "Image last-modification time" },
66     { CHUNK_TYPE_IDAT, "Image data chunk" },
67     { CHUNK_TYPE_IEND, "Image Trailer" },
68     { CHUNK_TYPE_tRNS, "Transparency" },
69     { CHUNK_TYPE_PLTE, "Palette" },
70     { 0, NULL }
71 };
72
73
74 void proto_register_png(void);
75 void proto_reg_handoff_png(void);
76
77 static header_field_info *hfi_png = NULL;
78
79 #define PNG_HFI_INIT HFI_INIT(proto_png)
80
81 static header_field_info hfi_png_signature PNG_HFI_INIT = {
82     "PNG Signature", "png.signature", FT_BYTES, BASE_NONE,
83     NULL, 0, NULL, HFILL };
84
85 static header_field_info hfi_png_chunk_data PNG_HFI_INIT = {
86     "Data", "png.chunk.data", FT_NONE, BASE_NONE,
87     NULL, 0, NULL, HFILL };
88
89 static header_field_info hfi_png_chunk_type_str PNG_HFI_INIT = {
90     "Chunk", "png.chunk.type", FT_STRING, BASE_NONE,
91     NULL, 0, NULL, HFILL };
92
93 static header_field_info hfi_png_chunk_len PNG_HFI_INIT = {
94     "Len", "png.chunk.len", FT_UINT32, BASE_DEC,
95     NULL, 0, NULL, HFILL };
96
97 static header_field_info hfi_png_chunk_crc PNG_HFI_INIT = {
98     "CRC", "png.chunk.crc", FT_UINT32, BASE_HEX,
99     NULL, 0, NULL, HFILL };
100
101 static const true_false_string png_chunk_anc = {
102     "This is an ANCILLARY chunk",
103     "This is a CRITICAL chunk"
104 };
105
106 static header_field_info hfi_png_chunk_flag_anc PNG_HFI_INIT = {
107     "Ancillary", "png.chunk.flag.ancillary", FT_BOOLEAN, 32,
108     TFS(&png_chunk_anc), 0x20000000, NULL, HFILL };
109
110 static const true_false_string png_chunk_priv = {
111     "This is a PRIVATE chunk",
112     "This is a PUBLIC chunk"
113 };
114
115 static header_field_info hfi_png_chunk_flag_priv PNG_HFI_INIT = {
116     "Private", "png.chunk.flag.private", FT_BOOLEAN, 32,
117     TFS(&png_chunk_priv), 0x00200000, NULL, HFILL };
118
119 static const true_false_string png_chunk_stc = {
120     "This chunk is SAFE TO COPY",
121     "This chunk is NOT safe to copy"
122 };
123
124 static header_field_info hfi_png_chunk_flag_stc PNG_HFI_INIT = {
125     "Safe To Copy", "png.chunk.flag.stc", FT_BOOLEAN, 32,
126     TFS(&png_chunk_stc), 0x00000020, NULL, HFILL };
127
128 static header_field_info hfi_png_ihdr_width PNG_HFI_INIT = {
129     "Width", "png.ihdr.width", FT_UINT32, BASE_DEC,
130     NULL, 0, NULL, HFILL };
131
132 static header_field_info hfi_png_ihdr_height PNG_HFI_INIT = {
133     "Height", "png.ihdr.height", FT_UINT32, BASE_DEC,
134     NULL, 0, NULL, HFILL };
135
136 static header_field_info hfi_png_ihdr_bitdepth PNG_HFI_INIT = {
137     "Bit Depth", "png.ihdr.bitdepth", FT_UINT8, BASE_DEC,
138     NULL, 0, NULL, HFILL };
139
140 static const value_string colour_type_vals[] = {
141     { 0,    "Greyscale"},
142     { 2,    "Truecolour"},
143     { 3,    "Indexed-colour"},
144     { 4,    "Greyscale with alpha"},
145     { 6,    "Truecolour with alpha"},
146     { 0, NULL }
147 };
148
149 static header_field_info hfi_png_ihdr_colour_type PNG_HFI_INIT = {
150     "Colour Type", "png.ihdr.colour_type", FT_UINT8, BASE_DEC,
151     VALS(colour_type_vals), 0, NULL, HFILL };
152
153 static const value_string compression_method_vals[] = {
154     { 0,    "Deflate"},
155     { 0, NULL }
156 };
157
158 static header_field_info hfi_png_ihdr_compression_method PNG_HFI_INIT = {
159     "Compression Method", "png.ihdr.compression_method", FT_UINT8, BASE_DEC,
160     VALS(compression_method_vals), 0, NULL, HFILL };
161
162 static const value_string filter_method_vals[] = {
163     { 0,    "Adaptive"},
164     { 0, NULL }
165 };
166
167 static header_field_info hfi_png_ihdr_filter_method PNG_HFI_INIT = {
168     "Filter Method", "png.ihdr.filter_method", FT_UINT8, BASE_DEC,
169     VALS(filter_method_vals), 0, NULL, HFILL };
170
171 static const value_string interlace_method_vals[] = {
172     { 0,    "No interlace"},
173     { 1,    "Adam7"},
174     { 0, NULL }
175 };
176
177 static header_field_info hfi_png_ihdr_interlace_method PNG_HFI_INIT = {
178     "Interlace Method", "png.ihdr.interlace_method", FT_UINT8, BASE_DEC,
179     VALS(interlace_method_vals), 0, NULL, HFILL };
180
181 static const value_string srgb_intent_vals[] = {
182     { 0, "Perceptual" },
183     { 1, "Relative colorimetric" },
184     { 2, "Saturation" },
185     { 3, "Absolute colorimetric" },
186     { 0, NULL }
187 };
188
189 static header_field_info hfi_png_srgb_intent PNG_HFI_INIT = {
190     "Intent", "png.srgb.intent", FT_UINT8, BASE_DEC,
191     VALS(srgb_intent_vals), 0, NULL, HFILL };
192
193 static header_field_info hfi_png_text_keyword PNG_HFI_INIT = {
194     "Keyword", "png.text.keyword", FT_STRING, STR_UNICODE,
195     NULL, 0, NULL, HFILL };
196
197 static header_field_info hfi_png_text_string PNG_HFI_INIT = {
198     "String", "png.text.string", FT_STRING, STR_UNICODE,
199     NULL, 0, NULL, HFILL };
200
201 static header_field_info hfi_png_time_year PNG_HFI_INIT = {
202     "Year", "png.time.year", FT_UINT16, BASE_DEC,
203     NULL, 0, NULL, HFILL };
204
205 static header_field_info hfi_png_time_month PNG_HFI_INIT = {
206     "Month", "png.time.month", FT_UINT8, BASE_DEC,
207     NULL, 0, NULL, HFILL };
208
209 static header_field_info hfi_png_time_day PNG_HFI_INIT = {
210     "Day", "png.time.day", FT_UINT8, BASE_DEC,
211     NULL, 0, NULL, HFILL };
212
213 static header_field_info hfi_png_time_hour PNG_HFI_INIT = {
214     "Hour", "png.time.hour", FT_UINT8, BASE_DEC,
215     NULL, 0, NULL, HFILL };
216
217 static header_field_info hfi_png_time_minute PNG_HFI_INIT = {
218     "Minute", "png.time.minute", FT_UINT8, BASE_DEC,
219     NULL, 0, NULL, HFILL };
220
221 static header_field_info hfi_png_time_second PNG_HFI_INIT = {
222     "Second", "png.time.second", FT_UINT8, BASE_DEC,
223     NULL, 0, NULL, HFILL };
224
225 static header_field_info hfi_png_phys_horiz PNG_HFI_INIT = {
226     "Horizontal pixels per unit", "png.phys.horiz", FT_UINT32, BASE_DEC,
227     NULL, 0, NULL, HFILL };
228
229 static header_field_info hfi_png_phys_vert PNG_HFI_INIT = {
230     "Vertical pixels per unit", "png.phys.vert", FT_UINT32, BASE_DEC,
231     NULL, 0, NULL, HFILL };
232
233 static const value_string phys_unit_vals[] = {
234     { 0,    "Unit is unknown"},
235     { 1,    "Unit is METRE"},
236     { 0, NULL }
237 };
238
239 static header_field_info hfi_png_phys_unit PNG_HFI_INIT = {
240     "Unit", "png.phys.unit", FT_UINT8, BASE_DEC,
241     VALS(phys_unit_vals), 0, NULL, HFILL };
242
243 static header_field_info hfi_png_bkgd_palette_index PNG_HFI_INIT = {
244     "Palette Index", "png.bkgd.palette_index", FT_UINT8, BASE_DEC,
245     NULL, 0, NULL, HFILL };
246
247 static header_field_info hfi_png_bkgd_greyscale PNG_HFI_INIT = {
248     "Greyscale", "png.bkgd.greyscale", FT_UINT16, BASE_HEX,
249     NULL, 0, NULL, HFILL };
250
251 static header_field_info hfi_png_bkgd_red PNG_HFI_INIT = {
252     "Red", "png.bkgd.red", FT_UINT16, BASE_HEX,
253     NULL, 0, NULL, HFILL };
254
255 static header_field_info hfi_png_bkgd_green PNG_HFI_INIT = {
256     "Green", "png.bkgd.green", FT_UINT16, BASE_HEX,
257     NULL, 0, NULL, HFILL };
258
259 static header_field_info hfi_png_bkgd_blue PNG_HFI_INIT = {
260     "Blue", "png.bkgd.blue", FT_UINT16, BASE_HEX,
261     NULL, 0, NULL, HFILL };
262
263 static header_field_info hfi_png_chrm_white_x PNG_HFI_INIT = {
264     "White X", "png.chrm.white.x", FT_FLOAT, BASE_NONE,
265     NULL, 0, NULL, HFILL };
266
267 static header_field_info hfi_png_chrm_white_y PNG_HFI_INIT = {
268     "White Y", "png.chrm.white.y", FT_FLOAT, BASE_NONE,
269     NULL, 0, NULL, HFILL };
270
271 static header_field_info hfi_png_chrm_red_x PNG_HFI_INIT = {
272     "Red X", "png.chrm.red.x", FT_FLOAT, BASE_NONE,
273     NULL, 0, NULL, HFILL };
274
275 static header_field_info hfi_png_chrm_red_y PNG_HFI_INIT = {
276     "Red Y", "png.chrm.red.y", FT_FLOAT, BASE_NONE,
277     NULL, 0, NULL, HFILL };
278
279 static header_field_info hfi_png_chrm_green_x PNG_HFI_INIT = {
280     "Green X", "png.chrm.green.x", FT_FLOAT, BASE_NONE,
281     NULL, 0, NULL, HFILL };
282
283 static header_field_info hfi_png_chrm_green_y PNG_HFI_INIT = {
284     "Green Y", "png.chrm.green.y", FT_FLOAT, BASE_NONE,
285     NULL, 0, NULL, HFILL };
286
287 static header_field_info hfi_png_chrm_blue_x PNG_HFI_INIT = {
288     "Blue X", "png.chrm.blue.x", FT_FLOAT, BASE_NONE,
289     NULL, 0, NULL, HFILL };
290
291 static header_field_info hfi_png_chrm_blue_y PNG_HFI_INIT = {
292     "Blue Y", "png.chrm.blue.y", FT_FLOAT, BASE_NONE,
293     NULL, 0, NULL, HFILL };
294
295 static header_field_info hfi_png_gama_gamma PNG_HFI_INIT = {
296     "Gamma", "png.gama.gamma", FT_FLOAT, BASE_NONE,
297     NULL, 0, NULL, HFILL };
298
299 static gint ett_png = -1;
300 static gint ett_png_chunk = -1;
301
302 static expert_field ei_png_chunk_too_large = EI_INIT;
303
304 static dissector_handle_t png_handle;
305
306 static void
307 dissect_png_ihdr(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
308 {
309     proto_tree_add_item(tree, &hfi_png_ihdr_width, tvb, 0, 4, ENC_BIG_ENDIAN);
310     proto_tree_add_item(tree, &hfi_png_ihdr_height, tvb, 4, 4, ENC_BIG_ENDIAN);
311     proto_tree_add_item(tree, &hfi_png_ihdr_bitdepth, tvb, 8, 1, ENC_BIG_ENDIAN);
312     proto_tree_add_item(tree, &hfi_png_ihdr_colour_type, tvb, 9, 1, ENC_BIG_ENDIAN);
313     proto_tree_add_item(tree, &hfi_png_ihdr_compression_method, tvb, 10, 1, ENC_BIG_ENDIAN);
314     proto_tree_add_item(tree, &hfi_png_ihdr_filter_method, tvb, 11, 1, ENC_BIG_ENDIAN);
315     proto_tree_add_item(tree, &hfi_png_ihdr_interlace_method, tvb, 12, 1, ENC_BIG_ENDIAN);
316
317 }
318
319 static void
320 dissect_png_srgb(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
321 {
322     proto_tree_add_item(tree, &hfi_png_srgb_intent,
323             tvb, 0, 1, ENC_BIG_ENDIAN);
324 }
325
326 static void
327 dissect_png_text(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
328 {
329     gint offset=0, nul_offset;
330
331     nul_offset = tvb_find_guint8(tvb, offset, tvb_captured_length_remaining(tvb, offset), 0);
332     /* nul_offset == 0 means empty keyword, this is not allowed by the png standard */
333     if (nul_offset<=0) {
334         /* XXX exception */
335         return;
336     }
337
338     proto_tree_add_item(tree, &hfi_png_text_keyword, tvb, offset, nul_offset, ENC_ISO_8859_1|ENC_NA);
339     offset = nul_offset+1; /* length of the key word + 0 character */
340
341     proto_tree_add_item(tree, &hfi_png_text_string, tvb, offset, tvb_captured_length_remaining(tvb, offset), ENC_ISO_8859_1|ENC_NA);
342
343 }
344
345 static void
346 dissect_png_time(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
347 {
348     proto_tree_add_item(tree, &hfi_png_time_year, tvb, 0, 2, ENC_BIG_ENDIAN);
349     proto_tree_add_item(tree, &hfi_png_time_month, tvb, 2, 1, ENC_BIG_ENDIAN);
350     proto_tree_add_item(tree, &hfi_png_time_day, tvb, 3, 1, ENC_BIG_ENDIAN);
351     proto_tree_add_item(tree, &hfi_png_time_hour, tvb, 4, 1, ENC_BIG_ENDIAN);
352     proto_tree_add_item(tree, &hfi_png_time_minute, tvb, 5, 1, ENC_BIG_ENDIAN);
353     proto_tree_add_item(tree, &hfi_png_time_second, tvb, 6, 1, ENC_BIG_ENDIAN);
354 }
355
356 static void
357 dissect_png_phys(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
358 {
359     proto_tree_add_item(tree, &hfi_png_phys_horiz, tvb, 0, 4, ENC_BIG_ENDIAN);
360     proto_tree_add_item(tree, &hfi_png_phys_vert, tvb, 4, 4, ENC_BIG_ENDIAN);
361     proto_tree_add_item(tree, &hfi_png_phys_unit, tvb, 8, 1, ENC_BIG_ENDIAN);
362 }
363
364 static void
365 dissect_png_bkgd(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
366 {
367     switch(tvb_reported_length(tvb)){
368         case 1: /* colour type 3 */
369             proto_tree_add_item(tree, &hfi_png_bkgd_palette_index, tvb, 0, 1, ENC_BIG_ENDIAN);
370             break;
371         case 2: /* colour type 0, 4 */
372             proto_tree_add_item(tree, &hfi_png_bkgd_greyscale, tvb, 0, 2, ENC_BIG_ENDIAN);
373             break;
374         case 6: /* colour type 2, 6 */
375             proto_tree_add_item(tree, &hfi_png_bkgd_red, tvb, 0, 2, ENC_BIG_ENDIAN);
376             proto_tree_add_item(tree, &hfi_png_bkgd_green, tvb, 2, 2, ENC_BIG_ENDIAN);
377             proto_tree_add_item(tree, &hfi_png_bkgd_blue, tvb, 4, 2, ENC_BIG_ENDIAN);
378             break;
379     }
380 }
381
382 static void
383 dissect_png_chrm(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
384 {
385     float  wx, wy, rx, ry, gx, gy, bx, by;
386     gint   offset = 0;
387
388     wx = tvb_get_ntohl(tvb, offset) / 100000.0f;
389     proto_tree_add_float(tree, &hfi_png_chrm_white_x,
390             tvb, offset, 4, wx);
391     offset += 4;
392
393     wy = tvb_get_ntohl(tvb, offset) / 100000.0f;
394     proto_tree_add_float(tree, &hfi_png_chrm_white_y,
395             tvb, offset, 4, wy);
396     offset += 4;
397
398     rx = tvb_get_ntohl(tvb, offset) / 100000.0f;
399     proto_tree_add_float(tree, &hfi_png_chrm_red_x,
400             tvb, offset, 4, rx);
401     offset += 4;
402
403     ry = tvb_get_ntohl(tvb, offset) / 100000.0f;
404     proto_tree_add_float(tree, &hfi_png_chrm_red_y,
405             tvb, offset, 4, ry);
406     offset += 4;
407
408     gx = tvb_get_ntohl(tvb, offset) / 100000.0f;
409     proto_tree_add_float(tree, &hfi_png_chrm_green_x,
410             tvb, offset, 4, gx);
411     offset += 4;
412
413     gy = tvb_get_ntohl(tvb, offset) / 100000.0f;
414     proto_tree_add_float(tree, &hfi_png_chrm_green_y,
415             tvb, offset, 4, gy);
416     offset += 4;
417
418     bx = tvb_get_ntohl(tvb, offset) / 100000.0f;
419     proto_tree_add_float(tree, &hfi_png_chrm_blue_x,
420             tvb, offset, 4, bx);
421     offset += 4;
422
423     by = tvb_get_ntohl(tvb, offset) / 100000.0f;
424     proto_tree_add_float(tree, &hfi_png_chrm_blue_y,
425             tvb, offset, 4, by);
426 }
427
428 static void
429 dissect_png_gama(tvbuff_t *tvb, packet_info *pinfo _U_, proto_tree *tree)
430 {
431     float  gamma;
432
433     gamma = tvb_get_ntohl(tvb, 0) / 100000.0f;
434     proto_tree_add_float(tree, &hfi_png_gama_gamma,
435             tvb, 0, 4, gamma);
436 }
437
438 static gint
439 dissect_png(tvbuff_t *tvb, packet_info *pinfo, proto_tree *parent_tree, void *data _U_)
440 {
441     proto_tree *tree;
442     proto_item *ti;
443     gint        offset=0;
444     /* http://libpng.org/pub/png/spec/1.2/PNG-Structure.html#PNG-file-signature */
445     static const guint8 magic[8] = { 137, 80, 78, 71, 13, 10, 26, 10 };
446
447     if (tvb_captured_length(tvb) < 20)
448         return 0;
449     if (tvb_memeql(tvb, 0, magic, sizeof(magic)) != 0)
450         return 0;
451
452     col_append_str(pinfo->cinfo, COL_INFO, " (PNG)");
453
454     ti=proto_tree_add_item(parent_tree, hfi_png, tvb, offset, -1, ENC_NA);
455     tree=proto_item_add_subtree(ti, ett_png);
456
457     proto_tree_add_item(tree, &hfi_png_signature, tvb, offset, 8, ENC_NA);
458     offset+=8;
459
460     while(tvb_reported_length_remaining(tvb, offset) > 0){
461         guint32     len_field;
462         proto_item *len_it;
463         proto_tree *chunk_tree;
464         guint32     type;
465         guint8     *type_str;
466         tvbuff_t   *chunk_tvb;
467
468         len_field = tvb_get_ntohl(tvb, offset);
469
470         type = tvb_get_ntohl(tvb, offset+4);
471         type_str = tvb_get_string_enc(wmem_packet_scope(),
472                 tvb, offset+4, 4, ENC_ASCII|ENC_NA);
473
474         /* 4 byte len field, 4 byte chunk type, 4 byte CRC */
475         chunk_tree = proto_tree_add_subtree_format(tree, tvb, offset, 4+4+len_field+4, ett_png_chunk, NULL,
476                 "%s (%s)", val_to_str_const(type, chunk_types, "unknown"), type_str);
477
478         len_it = proto_tree_add_item(chunk_tree, &hfi_png_chunk_len,
479                 tvb, offset, 4, ENC_BIG_ENDIAN);
480         offset+=4;
481         if (len_field > G_MAXINT) {
482             expert_add_info(pinfo, len_it, &ei_png_chunk_too_large);
483             return offset;
484         }
485
486         proto_tree_add_item(chunk_tree, &hfi_png_chunk_type_str,
487                 tvb, offset, 4, ENC_ASCII|ENC_NA);
488
489         proto_tree_add_item(chunk_tree, &hfi_png_chunk_flag_anc, tvb, offset, 4, ENC_BIG_ENDIAN);
490         proto_tree_add_item(chunk_tree, &hfi_png_chunk_flag_priv, tvb, offset, 4, ENC_BIG_ENDIAN);
491         proto_tree_add_item(chunk_tree, &hfi_png_chunk_flag_stc, tvb, offset, 4, ENC_BIG_ENDIAN);
492         offset+=4;
493
494         chunk_tvb=tvb_new_subset_length(tvb, offset, len_field);
495         switch (type) {
496             case CHUNK_TYPE_IHDR:
497                 dissect_png_ihdr(chunk_tvb, pinfo, chunk_tree);
498                 break;
499             case CHUNK_TYPE_bKGD:
500                 dissect_png_bkgd(chunk_tvb, pinfo, chunk_tree);
501                 break;
502             case CHUNK_TYPE_cHRM:
503                 dissect_png_chrm(chunk_tvb, pinfo, chunk_tree);
504                 break;
505             case CHUNK_TYPE_gAMA:
506                 dissect_png_gama(chunk_tvb, pinfo, chunk_tree);
507                 break;
508             case CHUNK_TYPE_pHYs:
509                 dissect_png_phys(chunk_tvb, pinfo, chunk_tree);
510                 break;
511             case CHUNK_TYPE_sRGB:
512                 dissect_png_srgb(chunk_tvb, pinfo, chunk_tree);
513                 break;
514             case CHUNK_TYPE_tEXt:
515                 dissect_png_text(chunk_tvb, pinfo, chunk_tree);
516                 break;
517             case CHUNK_TYPE_tIME:
518                 dissect_png_time(chunk_tvb, pinfo, chunk_tree);
519                 break;
520             default:
521                 if (len_field>0) {
522                     proto_tree_add_item(chunk_tree, &hfi_png_chunk_data,
523                             tvb, offset, len_field, ENC_NA);
524                 }
525                 break;
526         }
527         offset += len_field;
528
529         proto_tree_add_item(chunk_tree, &hfi_png_chunk_crc, tvb, offset, 4, ENC_BIG_ENDIAN);
530         offset+=4;
531     }
532     return offset;
533 }
534
535 void
536 proto_register_png(void)
537 {
538 #ifndef HAVE_HFI_SECTION_INIT
539     static header_field_info *hfi[] =
540     {
541         &hfi_png_signature,
542         &hfi_png_chunk_type_str,
543         &hfi_png_chunk_data,
544         &hfi_png_chunk_len,
545         &hfi_png_chunk_crc,
546         &hfi_png_chunk_flag_anc,
547         &hfi_png_chunk_flag_priv,
548         &hfi_png_chunk_flag_stc,
549         &hfi_png_ihdr_width,
550         &hfi_png_ihdr_height,
551         &hfi_png_ihdr_bitdepth,
552         &hfi_png_ihdr_colour_type,
553         &hfi_png_ihdr_compression_method,
554         &hfi_png_ihdr_filter_method,
555         &hfi_png_ihdr_interlace_method,
556         &hfi_png_srgb_intent,
557         &hfi_png_text_keyword,
558         &hfi_png_text_string,
559         &hfi_png_time_year,
560         &hfi_png_time_month,
561         &hfi_png_time_day,
562         &hfi_png_time_hour,
563         &hfi_png_time_minute,
564         &hfi_png_time_second,
565         &hfi_png_phys_horiz,
566         &hfi_png_phys_vert,
567         &hfi_png_phys_unit,
568         &hfi_png_bkgd_palette_index,
569         &hfi_png_bkgd_greyscale,
570         &hfi_png_bkgd_red,
571         &hfi_png_bkgd_green,
572         &hfi_png_bkgd_blue,
573         &hfi_png_chrm_white_x,
574         &hfi_png_chrm_white_y,
575         &hfi_png_chrm_red_x,
576         &hfi_png_chrm_red_y,
577         &hfi_png_chrm_green_x,
578         &hfi_png_chrm_green_y,
579         &hfi_png_chrm_blue_x,
580         &hfi_png_chrm_blue_y,
581         &hfi_png_gama_gamma
582     };
583 #endif
584
585     static gint *ett[] =
586     {
587         &ett_png,
588         &ett_png_chunk,
589     };
590
591     static ei_register_info ei[] = {
592         { &ei_png_chunk_too_large,
593             { "png.chunk_too_large", PI_PROTOCOL, PI_WARN,
594                 "chunk size too large, dissection of this chunk is not supported", EXPFILL }}
595     };
596     expert_module_t *expert_png;
597
598     int proto_png;
599
600     proto_png = proto_register_protocol("Portable Network Graphics","PNG","png");
601     hfi_png = proto_registrar_get_nth(proto_png);
602
603     proto_register_fields(proto_png, hfi, array_length(hfi));
604     proto_register_subtree_array(ett, array_length(ett));
605
606     expert_png = expert_register_protocol(proto_png);
607     expert_register_field_array(expert_png, ei, array_length(ei));
608
609     png_handle = register_dissector("png", dissect_png, proto_png);
610 }
611
612 static gboolean dissect_png_heur(tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree, void *data _U_)
613 {
614     return dissect_png(tvb, pinfo, tree, NULL) > 0;
615 }
616
617 void
618 proto_reg_handoff_png(void)
619 {
620     dissector_add_string("media_type", "image/png", png_handle);
621     heur_dissector_add("http", dissect_png_heur, "PNG file in HTTP", "png_http", hfi_png->id, HEURISTIC_ENABLE);
622     heur_dissector_add("wtap_file", dissect_png_heur, "PNG file in HTTP", "png_wtap", hfi_png->id, HEURISTIC_ENABLE);
623 }
624
625 /*
626  * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
627  *
628  * Local variables:
629  * c-basic-offset: 4
630  * tab-width: 8
631  * indent-tabs-mode: nil
632  * End:
633  *
634  * vi: set shiftwidth=4 tabstop=8 expandtab:
635  * :indentSize=4:tabSize=8:noTabs=true:
636  */