Add a failsafe on the maximum number of Canon MakerNote subtags.
[platform/upstream/libexif.git] / libexif / canon / exif-mnote-data-canon.c
1 /* exif-mnote-data-canon.c
2  *
3  * Copyright (c) 2002, 2003 Lutz Mueller <lutz@users.sourceforge.net>
4  * Copyright (c) 2003 Matthieu Castet <mat-c@users.sourceforge.net>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
19  * Boston, MA  02110-1301  USA.
20  */
21
22 #include <config.h>
23 #include "exif-mnote-data-canon.h"
24
25 #include <stdlib.h>
26 #include <stdio.h>
27 #include <string.h>
28
29 #include <libexif/exif-byte-order.h>
30 #include <libexif/exif-utils.h>
31 #include <libexif/exif-data.h>
32
33 #define CHECKOVERFLOW(offset,datasize,structsize) (( offset >= datasize) || (structsize > datasize) || (offset > datasize - structsize ))
34
35 /* Total size limit to prevent abuse by DoS */
36 #define FAILSAFE_SIZE_MAX 1000000L
37
38 static void
39 exif_mnote_data_canon_clear (ExifMnoteDataCanon *n)
40 {
41         ExifMnoteData *d = (ExifMnoteData *) n;
42         unsigned int i;
43
44         if (!n) return;
45
46         if (n->entries) {
47                 for (i = 0; i < n->count; i++)
48                         if (n->entries[i].data) {
49                                 exif_mem_free (d->mem, n->entries[i].data);
50                                 n->entries[i].data = NULL;
51                         }
52                 exif_mem_free (d->mem, n->entries);
53                 n->entries = NULL;
54                 n->count = 0;
55         }
56 }
57
58 static void
59 exif_mnote_data_canon_free (ExifMnoteData *n)
60 {
61         if (!n) return;
62
63         exif_mnote_data_canon_clear ((ExifMnoteDataCanon *) n);
64 }
65
66 static void
67 exif_mnote_data_canon_get_tags (ExifMnoteDataCanon *dc, unsigned int n,
68                 unsigned int *m, unsigned int *s)
69 {
70         unsigned int from = 0, to;
71
72         if (!dc || !m) return;
73         for (*m = 0; *m < dc->count; (*m)++) {
74                 to = from + mnote_canon_entry_count_values (&dc->entries[*m]);
75                 if (to > n) {
76                         if (s) *s = n - from;
77                         break;
78                 }
79                 from = to;
80         }
81 }
82
83 static char *
84 exif_mnote_data_canon_get_value (ExifMnoteData *note, unsigned int n, char *val, unsigned int maxlen)
85 {
86         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) note;
87         unsigned int m, s;
88
89         if (!dc) return NULL;
90         exif_mnote_data_canon_get_tags (dc, n, &m, &s);
91         if (m >= dc->count) return NULL;
92         return mnote_canon_entry_get_value (&dc->entries[m], s, val, maxlen);
93 }
94
95 static void
96 exif_mnote_data_canon_set_byte_order (ExifMnoteData *d, ExifByteOrder o)
97 {
98         ExifByteOrder o_orig;
99         ExifMnoteDataCanon *n = (ExifMnoteDataCanon *) d;
100         unsigned int i;
101
102         if (!n) return;
103
104         o_orig = n->order;
105         n->order = o;
106         for (i = 0; i < n->count; i++) {
107                 if (n->entries[i].components && (n->entries[i].size/n->entries[i].components < exif_format_get_size (n->entries[i].format)))
108                         continue;
109                 n->entries[i].order = o;
110                 exif_array_set_byte_order (n->entries[i].format, n->entries[i].data,
111                                 n->entries[i].components, o_orig, o);
112         }
113 }
114
115 static void
116 exif_mnote_data_canon_set_offset (ExifMnoteData *n, unsigned int o)
117 {
118         if (n) ((ExifMnoteDataCanon *) n)->offset = o;
119 }
120
121 static void
122 exif_mnote_data_canon_save (ExifMnoteData *ne, 
123         unsigned char **buf, unsigned int *buf_size)
124 {
125         ExifMnoteDataCanon *n = (ExifMnoteDataCanon *) ne;
126         size_t i, o, s, doff;
127         unsigned char *t;
128         size_t ts;
129
130         if (!n || !buf || !buf_size) return;
131
132         /*
133          * Allocate enough memory for all entries and the number
134          * of entries.
135          */
136         *buf_size = 2 + n->count * 12 + 4;
137         *buf = exif_mem_alloc (ne->mem, sizeof (char) * *buf_size);
138         if (!*buf) {
139                 EXIF_LOG_NO_MEMORY(ne->log, "ExifMnoteCanon", *buf_size);
140                 return;
141         }
142
143         /* Save the number of entries */
144         exif_set_short (*buf, n->order, (ExifShort) n->count);
145         
146         /* Save each entry */
147         for (i = 0; i < n->count; i++) {
148                 o = 2 + i * 12;
149                 exif_set_short (*buf + o + 0, n->order, (ExifShort) n->entries[i].tag);
150                 exif_set_short (*buf + o + 2, n->order, (ExifShort) n->entries[i].format);
151                 exif_set_long  (*buf + o + 4, n->order,
152                                 n->entries[i].components);
153                 o += 8;
154                 s = exif_format_get_size (n->entries[i].format) *
155                                                 n->entries[i].components;
156                 if (s > 65536) {
157                         /* Corrupt data: EXIF data size is limited to the
158                          * maximum size of a JPEG segment (64 kb).
159                          */
160                         continue;
161                 }
162                 if (s > 4) {
163                         ts = *buf_size + s;
164
165                         /* Ensure even offsets. Set padding bytes to 0. */
166                         if (s & 1) ts += 1;
167                         t = exif_mem_realloc (ne->mem, *buf,
168                                                  sizeof (char) * ts);
169                         if (!t) {
170                                 EXIF_LOG_NO_MEMORY(ne->log, "ExifMnoteCanon", ts);
171                                 return;
172                         }
173                         *buf = t;
174                         *buf_size = ts;
175                         doff = *buf_size - s;
176                         if (s & 1) { doff--; *(*buf + *buf_size - 1) = '\0'; }
177                         exif_set_long (*buf + o, n->order, n->offset + doff);
178                 } else
179                         doff = o;
180
181                 /*
182                  * Write the data. Fill unneeded bytes with 0. Do not
183                  * crash if data is NULL.
184                  */
185                 if (!n->entries[i].data) memset (*buf + doff, 0, s);
186                 else memcpy (*buf + doff, n->entries[i].data, s);
187                 if (s < 4) memset (*buf + doff + s, 0, (4 - s));
188         }
189 }
190
191 /* XXX
192  * FIXME: exif_mnote_data_canon_load() may fail and there is no
193  *        semantics to express that.
194  *        See bug #1054323 for details, especially the comment by liblit
195  *        after it has supposedly been fixed:
196  *
197  *        https://sourceforge.net/tracker/?func=detail&aid=1054323&group_id=12272&atid=112272
198  *        Unfortunately, the "return" statements aren't commented at
199  *        all, so it isn't trivial to find out what is a normal
200  *        return, and what is a reaction to an error condition.
201  */
202
203 static void
204 exif_mnote_data_canon_load (ExifMnoteData *ne,
205         const unsigned char *buf, unsigned int buf_size)
206 {
207         ExifMnoteDataCanon *n = (ExifMnoteDataCanon *) ne;
208         ExifShort c;
209         size_t i, tcount, o, datao;
210         long failsafe_size = 0;
211
212         if (!n || !buf || !buf_size) {
213                 exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
214                           "ExifMnoteCanon", "Short MakerNote");
215                 return;
216         }
217         datao = 6 + n->offset;
218         if (CHECKOVERFLOW(datao, buf_size, 2)) {
219                 exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
220                           "ExifMnoteCanon", "Short MakerNote");
221                 return;
222         }
223
224         /* Read the number of tags */
225         c = exif_get_short (buf + datao, n->order);
226         datao += 2;
227
228         /* Remove any old entries */
229         exif_mnote_data_canon_clear (n);
230
231         /* Reserve enough space for all the possible MakerNote tags */
232         n->entries = exif_mem_alloc (ne->mem, sizeof (MnoteCanonEntry) * c);
233         if (!n->entries) {
234                 EXIF_LOG_NO_MEMORY(ne->log, "ExifMnoteCanon", sizeof (MnoteCanonEntry) * c);
235                 return;
236         }
237
238         /* Parse the entries */
239         tcount = 0;
240         for (i = c, o = datao; i; --i, o += 12) {
241                 size_t s;
242
243                 memset(&n->entries[tcount], 0, sizeof(MnoteCanonEntry));
244                 if (CHECKOVERFLOW(o,buf_size,12)) {
245                         exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
246                                 "ExifMnoteCanon", "Short MakerNote");
247                         break;
248                 }
249
250                 n->entries[tcount].tag        = exif_get_short (buf + o, n->order);
251                 n->entries[tcount].format     = exif_get_short (buf + o + 2, n->order);
252                 n->entries[tcount].components = exif_get_long (buf + o + 4, n->order);
253                 n->entries[tcount].order      = n->order;
254
255                 exif_log (ne->log, EXIF_LOG_CODE_DEBUG, "ExifMnoteCanon",
256                         "Loading entry 0x%x ('%s')...", n->entries[tcount].tag,
257                          mnote_canon_tag_get_name (n->entries[tcount].tag));
258
259                 /* Check if we overflow the multiplication. Use buf_size as the max size for integer overflow detection,
260                  * we will check the buffer sizes closer later. */
261                 if (    exif_format_get_size (n->entries[tcount].format) &&
262                         buf_size / exif_format_get_size (n->entries[tcount].format) < n->entries[tcount].components
263                 ) {
264                         exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
265                                   "ExifMnoteCanon", "Tag size overflow detected (%u * %lu)", exif_format_get_size (n->entries[tcount].format), n->entries[tcount].components);
266                         continue;
267                 }
268
269                 /*
270                  * Size? If bigger than 4 bytes, the actual data is not
271                  * in the entry but somewhere else (offset).
272                  */
273                 s = exif_format_get_size (n->entries[tcount].format) * 
274                                                                   n->entries[tcount].components;
275                 n->entries[tcount].size = s;
276                 if (!s) {
277                         exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
278                                   "ExifMnoteCanon",
279                                   "Invalid zero-length tag size");
280                         continue;
281
282                 } else {
283                         size_t dataofs = o + 8;
284                         if (s > 4) dataofs = exif_get_long (buf + dataofs, n->order) + 6;
285
286                         if (CHECKOVERFLOW(dataofs, buf_size, s)) {
287                                 exif_log (ne->log, EXIF_LOG_CODE_DEBUG,
288                                         "ExifMnoteCanon",
289                                         "Tag data past end of buffer (%u > %u)",
290                                         (unsigned)(dataofs + s), buf_size);
291                                 continue;
292                         }
293
294                         n->entries[tcount].data = exif_mem_alloc (ne->mem, s);
295                         if (!n->entries[tcount].data) {
296                                 EXIF_LOG_NO_MEMORY(ne->log, "ExifMnoteCanon", s);
297                                 continue;
298                         }
299                         memcpy (n->entries[tcount].data, buf + dataofs, s);
300                 }
301
302                 /* Track the size of decoded tag data. A malicious file could
303                  * be crafted to cause extremely large values here without
304                  * tripping any buffer range checks.  This is especially bad
305                  * with the libexif representation of Canon MakerNotes because
306                  * some arrays are turned into individual tags that the
307                  * application must loop around. */
308                 failsafe_size += mnote_canon_entry_count_values(&n->entries[tcount]);
309
310                 if (failsafe_size > FAILSAFE_SIZE_MAX) {
311                         /* Abort if the total size of the data in the tags extraordinarily large, */
312                         exif_mem_free (ne->mem, n->entries[tcount].data);
313                         exif_log (ne->log, EXIF_LOG_CODE_CORRUPT_DATA,
314                                           "ExifMnoteCanon", "Failsafe tag size overflow (%lu > %ld)",
315                                           failsafe_size, FAILSAFE_SIZE_MAX);
316                         break;
317                 }
318
319                 /* Tag was successfully parsed */
320                 ++tcount;
321         }
322         /* Store the count of successfully parsed tags */
323         n->count = tcount;
324 }
325
326 static unsigned int
327 exif_mnote_data_canon_count (ExifMnoteData *n)
328 {
329         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) n;
330         unsigned int i, c;
331
332         for (i = c = 0; dc && (i < dc->count); i++)
333                 c += mnote_canon_entry_count_values (&dc->entries[i]);
334         return c;
335 }
336
337 static unsigned int
338 exif_mnote_data_canon_get_id (ExifMnoteData *d, unsigned int i)
339 {
340         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) d;
341         unsigned int m;
342
343         if (!dc) return 0;
344         exif_mnote_data_canon_get_tags (dc, i, &m, NULL);
345         if (m >= dc->count) return 0;
346         return dc->entries[m].tag;
347 }
348
349 static const char *
350 exif_mnote_data_canon_get_name (ExifMnoteData *note, unsigned int i)
351 {
352         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) note;
353         unsigned int m, s;
354
355         if (!dc) return NULL;
356         exif_mnote_data_canon_get_tags (dc, i, &m, &s);
357         if (m >= dc->count) return NULL;
358         return mnote_canon_tag_get_name_sub (dc->entries[m].tag, s, dc->options);
359 }
360
361 static const char *
362 exif_mnote_data_canon_get_title (ExifMnoteData *note, unsigned int i)
363 {
364         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) note;
365         unsigned int m, s;
366
367         if (!dc) return NULL;
368         exif_mnote_data_canon_get_tags (dc, i, &m, &s);
369         if (m >= dc->count) return NULL;
370         return mnote_canon_tag_get_title_sub (dc->entries[m].tag, s, dc->options);
371 }
372
373 static const char *
374 exif_mnote_data_canon_get_description (ExifMnoteData *note, unsigned int i)
375 {
376         ExifMnoteDataCanon *dc = (ExifMnoteDataCanon *) note;
377         unsigned int m;
378
379         if (!dc) return NULL;
380         exif_mnote_data_canon_get_tags (dc, i, &m, NULL);
381         if (m >= dc->count) return NULL;
382         return mnote_canon_tag_get_description (dc->entries[m].tag);
383 }
384
385 int
386 exif_mnote_data_canon_identify (const ExifData *ed, const ExifEntry *e)
387 {
388         char value[8];
389
390         (void) e;  /* unused */
391         ExifEntry *em = exif_data_get_entry (ed, EXIF_TAG_MAKE);
392         if (!em) 
393                 return 0;
394         return !strcmp (exif_entry_get_value (em, value, sizeof (value)), "Canon");
395 }
396
397 ExifMnoteData *
398 exif_mnote_data_canon_new (ExifMem *mem, ExifDataOption o)
399 {
400         ExifMnoteData *d;
401         ExifMnoteDataCanon *dc;
402
403         if (!mem) return NULL;
404
405         d = exif_mem_alloc (mem, sizeof (ExifMnoteDataCanon));
406         if (!d)
407                 return NULL;
408
409         exif_mnote_data_construct (d, mem);
410
411         /* Set up function pointers */
412         d->methods.free            = exif_mnote_data_canon_free;
413         d->methods.set_byte_order  = exif_mnote_data_canon_set_byte_order;
414         d->methods.set_offset      = exif_mnote_data_canon_set_offset;
415         d->methods.load            = exif_mnote_data_canon_load;
416         d->methods.save            = exif_mnote_data_canon_save;
417         d->methods.count           = exif_mnote_data_canon_count;
418         d->methods.get_id          = exif_mnote_data_canon_get_id;
419         d->methods.get_name        = exif_mnote_data_canon_get_name;
420         d->methods.get_title       = exif_mnote_data_canon_get_title;
421         d->methods.get_description = exif_mnote_data_canon_get_description;
422         d->methods.get_value       = exif_mnote_data_canon_get_value;
423
424         dc = (ExifMnoteDataCanon*)d;
425         dc->options = o;
426         return d;
427 }