When loading an entry fails, do not add it to the IFD.
[platform/upstream/libexif.git] / libexif / exif-data.c
1 /* exif-data.c
2  *
3  * Copyright © 2001 Lutz Müller <lutz@users.sourceforge.net>
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the
17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18  * Boston, MA 02111-1307, USA.
19  */
20
21 #include <config.h>
22
23 #include <libexif/exif-mnote-data.h>
24 #include <libexif/exif-data.h>
25 #include <libexif/exif-ifd.h>
26 #include <libexif/exif-mnote-data-priv.h>
27 #include <libexif/exif-utils.h>
28 #include <libexif/exif-loader.h>
29 #include <libexif/exif-log.h>
30 #include <libexif/i18n.h>
31
32 #include <libexif/olympus/exif-mnote-data-olympus.h>
33 #include <libexif/canon/exif-mnote-data-canon.h>
34 #include <libexif/pentax/exif-mnote-data-pentax.h>
35
36 #include <stdlib.h>
37 #include <stdio.h>
38 #include <string.h>
39
40 #undef MAX
41 #define MAX(a, b)  (((a) > (b)) ? (a) : (b))
42
43 #if defined(__WATCOMC__) || defined(_MSC_VER)
44 #      define strncasecmp strnicmp
45 #endif
46
47 #undef JPEG_MARKER_SOI
48 #define JPEG_MARKER_SOI  0xd8
49 #undef JPEG_MARKER_APP0
50 #define JPEG_MARKER_APP0 0xe0
51 #undef JPEG_MARKER_APP1
52 #define JPEG_MARKER_APP1 0xe1
53
54 static const unsigned char ExifHeader[] = {0x45, 0x78, 0x69, 0x66, 0x00, 0x00};
55
56 struct _ExifDataPrivate
57 {
58         ExifByteOrder order;
59
60         ExifMnoteData *md;
61
62         ExifLog *log;
63         ExifMem *mem;
64
65         unsigned int ref_count;
66
67         /* Temporarily used while loading data */
68         unsigned int offset_mnote;
69
70         ExifDataOption options;
71         ExifDataType data_type;
72 };
73
74 static void *
75 exif_data_alloc (ExifData *data, unsigned int i)
76 {
77         void *d;
78
79         if (!data || !i) 
80                 return NULL;
81
82         d = exif_mem_alloc (data->priv->mem, i);
83         if (d) 
84                 return d;
85
86         EXIF_LOG_NO_MEMORY (data->priv->log, "ExifData", i);
87         return NULL;
88 }
89
90 ExifMnoteData *
91 exif_data_get_mnote_data (ExifData *d)
92 {
93         return (d && d->priv) ? d->priv->md : NULL;
94 }
95
96 ExifData *
97 exif_data_new (void)
98 {
99         ExifMem *mem = exif_mem_new_default ();
100         ExifData *d = exif_data_new_mem (mem);
101
102         exif_mem_unref (mem);
103
104         return d;
105 }
106
107 ExifData *
108 exif_data_new_mem (ExifMem *mem)
109 {
110         ExifData *data;
111         unsigned int i;
112
113         if (!mem) 
114                 return NULL;
115
116         data = exif_mem_alloc (mem, sizeof (ExifData));
117         if (!data) 
118                 return (NULL);
119         data->priv = exif_mem_alloc (mem, sizeof (ExifDataPrivate));
120         if (!data->priv) { 
121                 exif_mem_free (mem, data); 
122                 return (NULL); 
123         }
124         data->priv->ref_count = 1;
125
126         data->priv->mem = mem;
127         exif_mem_ref (mem);
128
129         for (i = 0; i < EXIF_IFD_COUNT; i++) {
130                 data->ifd[i] = exif_content_new_mem (data->priv->mem);
131                 if (!data->ifd[i]) {
132                         exif_data_free (data);
133                         return (NULL);
134                 }
135                 data->ifd[i]->parent = data;
136         }
137
138         /* Default options */
139         exif_data_set_option (data, EXIF_DATA_OPTION_IGNORE_UNKNOWN_TAGS);
140         exif_data_set_option (data, EXIF_DATA_OPTION_FOLLOW_SPECIFICATION);
141
142         /* Default data type: none */
143         exif_data_set_data_type (data, EXIF_DATA_TYPE_COUNT);
144
145         return (data);
146 }
147
148 ExifData *
149 exif_data_new_from_data (const unsigned char *data, unsigned int size)
150 {
151         ExifData *edata;
152
153         edata = exif_data_new ();
154         exif_data_load_data (edata, data, size);
155         return (edata);
156 }
157
158 static int
159 exif_data_load_data_entry (ExifData *data, ExifEntry *entry,
160                            const unsigned char *d,
161                            unsigned int size, unsigned int offset)
162 {
163         unsigned int s, doff;
164
165         entry->tag        = exif_get_short (d + offset + 0, data->priv->order);
166         entry->format     = exif_get_short (d + offset + 2, data->priv->order);
167         entry->components = exif_get_long  (d + offset + 4, data->priv->order);
168
169         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
170                   "Loading entry 0x%x ('%s')...", entry->tag,
171                   exif_tag_get_name (entry->tag));
172
173         /*
174          * Size? If bigger than 4 bytes, the actual data is not
175          * in the entry but somewhere else (offset).
176          */
177         s = exif_format_get_size (entry->format) * entry->components;
178         if (!s)
179                 return 0;
180         if (s > 4)
181                 doff = exif_get_long (d + offset + 8, data->priv->order);
182         else
183                 doff = offset + 8;
184
185         /* Sanity check */
186         if (size < doff + s)
187                 return 0;
188
189         entry->data = exif_data_alloc (data, s);
190         if (entry->data) {
191                 entry->size = s;
192                 memcpy (entry->data, d + doff, s);
193         }
194
195         /* If this is the MakerNote, remember the offset */
196         if (entry->tag == EXIF_TAG_MAKER_NOTE) {
197                 if (entry->size > 6) exif_log (data->priv->log,
198                                                EXIF_LOG_CODE_DEBUG, "ExifData",
199                                                "MakerNote found (%02x %02x %02x %02x "
200                                                "%02x %02x %02x...).",
201                                                entry->data[0], entry->data[1], entry->data[2],
202                                                entry->data[3], entry->data[4], entry->data[5],
203                                                entry->data[6]);
204                 data->priv->offset_mnote = doff;
205         }
206         return 1;
207 }
208
209 static void
210 exif_data_save_data_entry (ExifData *data, ExifEntry *e,
211                            unsigned char **d, unsigned int *ds,
212                            unsigned int offset)
213 {
214         unsigned int doff, s;
215
216         if (!data || !data->priv) 
217                 return;
218
219         /*
220          * Each entry is 12 bytes long. The memory for the entry has
221          * already been allocated.
222          */
223         exif_set_short (*d + 6 + offset + 0,
224                         data->priv->order, (ExifShort) e->tag);
225         exif_set_short (*d + 6 + offset + 2,
226                         data->priv->order, (ExifShort) e->format);
227
228 #ifndef EXIF_DONT_CHANGE_MAKER_NOTE
229         /* If this is the maker note tag, update it. */
230         if ((e->tag == EXIF_TAG_MAKER_NOTE) && data->priv->md) {
231                 exif_mem_free (data->priv->mem, e->data);
232                 e->data = NULL;
233                 e->size = 0;
234                 exif_mnote_data_set_offset (data->priv->md, *ds - 6);
235                 exif_mnote_data_save (data->priv->md, &e->data, &e->size);
236                 e->components = e->size;
237         }
238 #endif
239
240         exif_set_long  (*d + 6 + offset + 4,
241                         data->priv->order, e->components);
242
243         /*
244          * Size? If bigger than 4 bytes, the actual data is not in
245          * the entry but somewhere else.
246          */
247         s = exif_format_get_size (e->format) * e->components;
248         if (s > 4) {
249                 doff = *ds - 6;
250                 *ds += s;
251
252                 /*
253                  * According to the TIFF specification,
254                  * the offset must be an even number. If we need to introduce
255                  * a padding byte, we set it to 0.
256                  */
257                 if (s & 1) 
258                         (*ds)++;
259                 *d = exif_mem_realloc (data->priv->mem, *d, *ds);
260                 if (!*d) {
261                         EXIF_LOG_NO_MEMORY (data->priv->log, "ExifData", *ds);
262                         return;
263                 }
264                 exif_set_long (*d + 6 + offset + 8, data->priv->order, doff);
265                 if (s & 1) 
266                         *(*d + *ds - 1) = '\0';
267
268         } else
269                 doff = offset + 8;
270
271         /* Write the data. Fill unneeded bytes with 0. */
272         memcpy (*d + 6 + doff, e->data, s);
273         if (s < 4) 
274                 memset (*d + 6 + doff + s, 0, (4 - s));
275 }
276
277 static void
278 exif_data_load_data_thumbnail (ExifData *data, const unsigned char *d,
279                                unsigned int ds, ExifLong offset, ExifLong size)
280 {
281         if (ds < offset + size) {
282                 exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
283                           "Bogus thumbnail offset and size: %i < %i + %i.",
284                           (int) ds, (int) offset, (int) size);
285                 return;
286         }
287         if (data->data) 
288                 exif_mem_free (data->priv->mem, data->data);
289         data->size = size;
290         data->data = exif_data_alloc (data, data->size);
291         if (!data->data) 
292                 return;
293         memcpy (data->data, d + offset, data->size);
294 }
295
296 #undef CHECK_REC
297 #define CHECK_REC(i)                                    \
298 if ((i) == ifd) {                               \
299         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, \
300                 "ExifData", "Recursive entry in IFD "   \
301                 "'%s' detected. Skipping...",           \
302                 exif_ifd_get_name (i));                 \
303         break;                                          \
304 }                                                       \
305 if (data->ifd[(i)]->count) {                            \
306         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, \
307                 "ExifData", "Attemt to load IFD "       \
308                 "'%s' multiple times detected. "        \
309                 "Skipping...",                          \
310                 exif_ifd_get_name (i));                 \
311         break;                                          \
312 }
313
314 static void
315 exif_data_load_data_content (ExifData *data, ExifIfd ifd,
316                              const unsigned char *d,
317                              unsigned int ds, unsigned int offset, unsigned int recursion_depth)
318 {
319         ExifLong o, thumbnail_offset = 0, thumbnail_length = 0;
320         ExifShort n;
321         ExifEntry *entry;
322         unsigned int i;
323         ExifTag tag;
324
325         if (!data || !data->priv) 
326                 return;
327         if ((ifd < 0) || (ifd >= EXIF_IFD_COUNT)) 
328                 return;
329
330         if (recursion_depth > 150) {
331                 exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA, "ExifData",
332                           "Deep recursion detected!");
333                 return;
334         }
335
336         /* Read the number of entries */
337         if (offset >= ds - 1) 
338                 return;
339         n = exif_get_short (d + offset, data->priv->order);
340         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
341                   "Loading %i entries...", n);
342         offset += 2;
343
344         /* Check if we have enough data. */
345         if (offset + 12 * n > ds) 
346                 n = (ds - offset) / 12;
347
348         for (i = 0; i < n; i++) {
349
350                 tag = exif_get_short (d + offset + 12 * i, data->priv->order);
351                 switch (tag) {
352                 case EXIF_TAG_EXIF_IFD_POINTER:
353                 case EXIF_TAG_GPS_INFO_IFD_POINTER:
354                 case EXIF_TAG_INTEROPERABILITY_IFD_POINTER:
355                 case EXIF_TAG_JPEG_INTERCHANGE_FORMAT_LENGTH:
356                 case EXIF_TAG_JPEG_INTERCHANGE_FORMAT:
357                         o = exif_get_long (d + offset + 12 * i + 8,
358                                            data->priv->order);
359                         switch (tag) {
360                         case EXIF_TAG_EXIF_IFD_POINTER:
361                                 CHECK_REC (EXIF_IFD_EXIF);
362                                 exif_data_load_data_content (data, EXIF_IFD_EXIF, d, ds, o, recursion_depth + 1);
363                                 break;
364                         case EXIF_TAG_GPS_INFO_IFD_POINTER:
365                                 CHECK_REC (EXIF_IFD_GPS);
366                                 exif_data_load_data_content (data, EXIF_IFD_GPS, d, ds, o, recursion_depth + 1);
367                                 break;
368                         case EXIF_TAG_INTEROPERABILITY_IFD_POINTER:
369                                 CHECK_REC (EXIF_IFD_INTEROPERABILITY);
370                                 exif_data_load_data_content (data, EXIF_IFD_INTEROPERABILITY, d, ds, o, recursion_depth + 1);
371                                 break;
372                         case EXIF_TAG_JPEG_INTERCHANGE_FORMAT:
373                                 thumbnail_offset = o;
374                                 if (thumbnail_offset && thumbnail_length)
375                                         exif_data_load_data_thumbnail (data, d,
376                                                                        ds, thumbnail_offset,
377                                                                        thumbnail_length);
378                                 break;
379                         case EXIF_TAG_JPEG_INTERCHANGE_FORMAT_LENGTH:
380                                 thumbnail_length = o;
381                                 if (thumbnail_offset && thumbnail_length)
382                                         exif_data_load_data_thumbnail (data, d,
383                                                                        ds, thumbnail_offset,
384                                                                        thumbnail_length);
385                                 break;
386                         default:
387                                 return;
388                         }
389                         break;
390                 default:
391
392                         /*
393                          * If we don't know the tag, don't fail. It could be that new 
394                          * versions of the standard have defined additional tags. Note that
395                          * 0 is a valid tag in the GPS IFD.
396                          */
397                         if (!exif_tag_get_name_in_ifd (tag, ifd)) {
398
399                                 /*
400                                  * Special case: Tag and format 0. That's against specification.
401                                  * At least up to 2.2. But Photoshop writes it anyways.
402                                  */
403                                 if (!memcmp (d + offset + 12 * i, "\0\0\0\0", 4)) {
404                                         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
405                                                   "Skipping empty entry at position %i in '%s'.", i, 
406                                                   exif_ifd_get_name (ifd));
407                                         break;
408                                 }
409                                 exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
410                                           "Unknown tag 0x%04x (entry %i in '%s'). Please report this tag "
411                                           "to <libexif-devel@lists.sourceforge.net>.", tag, i,
412                                           exif_ifd_get_name (ifd));
413                                 if (data->priv->options & EXIF_DATA_OPTION_IGNORE_UNKNOWN_TAGS)
414                                         break;
415                         }
416                         entry = exif_entry_new_mem (data->priv->mem);
417                         if (exif_data_load_data_entry (data, entry, d, ds,
418                                                    offset + 12 * i))
419                                 exif_content_add_entry (data->ifd[ifd], entry);
420                         exif_entry_unref (entry);
421                         break;
422                 }
423         }
424 }
425
426 static int
427 cmp_func (const unsigned char *p1, const unsigned char *p2, ExifByteOrder o)
428 {
429         ExifShort tag1 = exif_get_short (p1, o);
430         ExifShort tag2 = exif_get_short (p2, o);
431
432         return (tag1 < tag2) ? -1 : (tag1 > tag2) ? 1 : 0;
433 }
434
435 static int
436 cmp_func_intel (const void *elem1, const void *elem2)
437 {
438         return cmp_func ((const unsigned char *) elem1,
439                          (const unsigned char *) elem2, EXIF_BYTE_ORDER_INTEL);
440 }
441
442 static int
443 cmp_func_motorola (const void *elem1, const void *elem2)
444 {
445         return cmp_func ((const unsigned char *) elem1,
446                          (const unsigned char *) elem2, EXIF_BYTE_ORDER_MOTOROLA);
447 }
448
449 static void
450 exif_data_save_data_content (ExifData *data, ExifContent *ifd,
451                              unsigned char **d, unsigned int *ds,
452                              unsigned int offset)
453 {
454         unsigned int j, n_ptr = 0, n_thumb = 0;
455         ExifIfd i;
456
457         if (!data || !data->priv || !ifd || !d || !ds) 
458                 return;
459
460         for (i = 0; i < EXIF_IFD_COUNT; i++)
461                 if (ifd == data->ifd[i])
462                         break;
463         if (i == EXIF_IFD_COUNT)
464                 return;
465
466         /*
467          * Check if we need some extra entries for pointers or the thumbnail.
468          */
469         switch (i) {
470         case EXIF_IFD_0:
471
472                 /*
473                  * The pointer to IFD_EXIF is in IFD_0. The pointer to
474                  * IFD_INTEROPERABILITY is in IFD_EXIF.
475                  */
476                 if (data->ifd[EXIF_IFD_EXIF]->count ||
477                     data->ifd[EXIF_IFD_INTEROPERABILITY]->count)
478                         n_ptr++;
479
480                 /* The pointer to IFD_GPS is in IFD_0. */
481                 if (data->ifd[EXIF_IFD_GPS]->count)
482                         n_ptr++;
483
484                 break;
485         case EXIF_IFD_1:
486                 if (data->size)
487                         n_thumb = 2;
488                 break;
489         case EXIF_IFD_EXIF:
490                 if (data->ifd[EXIF_IFD_INTEROPERABILITY]->count)
491                         n_ptr++;
492         default:
493                 break;
494         }
495
496         /*
497          * Allocate enough memory for all entries
498          * and the number of entries.
499          */
500         *ds += (2 + (ifd->count + n_ptr + n_thumb) * 12 + 4);
501         *d = exif_mem_realloc (data->priv->mem, *d, *ds);
502         if (!*d) {
503                 EXIF_LOG_NO_MEMORY (data->priv->log, "ExifData", *ds);
504                 return;
505         }
506
507         /* Save the number of entries */
508         exif_set_short (*d + 6 + offset, data->priv->order,
509                         (ExifShort) (ifd->count + n_ptr + n_thumb));
510         offset += 2;
511
512         /* Save each entry */
513         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
514                   "Saving %i entries (IFD '%s', offset: %i)...",
515                   ifd->count, exif_ifd_get_name (i), offset);
516         for (j = 0; j < ifd->count; j++)
517                 exif_data_save_data_entry (data, ifd->entries[j], d, ds, offset + 12 * j);
518
519         offset += 12 * ifd->count;
520
521         /* Now save special entries. */
522         switch (i) {
523         case EXIF_IFD_0:
524
525                 /*
526                  * The pointer to IFD_EXIF is in IFD_0.
527                  * However, the pointer to IFD_INTEROPERABILITY is in IFD_EXIF,
528                  * therefore, if IFD_INTEROPERABILITY is not empty, we need
529                  * IFD_EXIF even if latter is empty.
530                  */
531                 if (data->ifd[EXIF_IFD_EXIF]->count ||
532                     data->ifd[EXIF_IFD_INTEROPERABILITY]->count) {
533                         exif_set_short (*d + 6 + offset + 0, data->priv->order,
534                                         EXIF_TAG_EXIF_IFD_POINTER);
535                         exif_set_short (*d + 6 + offset + 2, data->priv->order,
536                                         EXIF_FORMAT_LONG);
537                         exif_set_long  (*d + 6 + offset + 4, data->priv->order,
538                                         1);
539                         exif_set_long  (*d + 6 + offset + 8, data->priv->order,
540                                         *ds - 6);
541                         exif_data_save_data_content (data,
542                                                      data->ifd[EXIF_IFD_EXIF], d, ds, *ds - 6);
543                         offset += 12;
544                 }
545
546                 /* The pointer to IFD_GPS is in IFD_0, too. */
547                 if (data->ifd[EXIF_IFD_GPS]->count) {
548                         exif_set_short (*d + 6 + offset + 0, data->priv->order,
549                                         EXIF_TAG_GPS_INFO_IFD_POINTER);
550                         exif_set_short (*d + 6 + offset + 2, data->priv->order,
551                                         EXIF_FORMAT_LONG);
552                         exif_set_long  (*d + 6 + offset + 4, data->priv->order,
553                                         1);
554                         exif_set_long  (*d + 6 + offset + 8, data->priv->order,
555                                         *ds - 6);
556                         exif_data_save_data_content (data,
557                                                      data->ifd[EXIF_IFD_GPS], d, ds, *ds - 6);
558                         offset += 12;
559                 }
560
561                 break;
562         case EXIF_IFD_EXIF:
563
564                 /*
565                  * The pointer to IFD_INTEROPERABILITY is in IFD_EXIF.
566                  * See note above.
567                  */
568                 if (data->ifd[EXIF_IFD_INTEROPERABILITY]->count) {
569                         exif_set_short (*d + 6 + offset + 0, data->priv->order,
570                                         EXIF_TAG_INTEROPERABILITY_IFD_POINTER);
571                         exif_set_short (*d + 6 + offset + 2, data->priv->order,
572                                         EXIF_FORMAT_LONG);
573                         exif_set_long  (*d + 6 + offset + 4, data->priv->order,
574                                         1);
575                         exif_set_long  (*d + 6 + offset + 8, data->priv->order,
576                                         *ds - 6);
577                         exif_data_save_data_content (data,
578                                                      data->ifd[EXIF_IFD_INTEROPERABILITY], d, ds,
579                                                      *ds - 6);
580                         offset += 12;
581                 }
582
583                 break;
584         case EXIF_IFD_1:
585
586                 /*
587                  * Information about the thumbnail (if any) is saved in
588                  * IFD_1.
589                  */
590                 if (data->size) {
591
592                         /* EXIF_TAG_JPEG_INTERCHANGE_FORMAT */
593                         exif_set_short (*d + 6 + offset + 0, data->priv->order,
594                                         EXIF_TAG_JPEG_INTERCHANGE_FORMAT);
595                         exif_set_short (*d + 6 + offset + 2, data->priv->order,
596                                         EXIF_FORMAT_LONG);
597                         exif_set_long  (*d + 6 + offset + 4, data->priv->order,
598                                         1);
599                         exif_set_long  (*d + 6 + offset + 8, data->priv->order,
600                                         *ds - 6);
601                         *ds += data->size;
602                         *d = exif_mem_realloc (data->priv->mem, *d, *ds);
603                         if (!*d) {
604                                 EXIF_LOG_NO_MEMORY (data->priv->log, "ExifData",
605                                                     *ds);
606                                 return;
607                         }
608                         memcpy (*d + *ds - data->size, data->data, data->size);
609                         offset += 12;
610
611                         /* EXIF_TAG_JPEG_INTERCHANGE_FORMAT_LENGTH */
612                         exif_set_short (*d + 6 + offset + 0, data->priv->order,
613                                         EXIF_TAG_JPEG_INTERCHANGE_FORMAT_LENGTH);
614                         exif_set_short (*d + 6 + offset + 2, data->priv->order,
615                                         EXIF_FORMAT_LONG);
616                         exif_set_long  (*d + 6 + offset + 4, data->priv->order,
617                                         1);
618                         exif_set_long  (*d + 6 + offset + 8, data->priv->order,
619                                         data->size);
620                         offset += 12;
621                 }
622
623                 break;
624         default:
625                 break;
626         }
627
628         /* Sort the directory according to TIFF specification */
629         qsort (*d + 6 + offset - (ifd->count + n_ptr + n_thumb) * 12,
630                (ifd->count + n_ptr + n_thumb), 12,
631                (data->priv->order == EXIF_BYTE_ORDER_INTEL) ? cmp_func_intel : cmp_func_motorola);
632
633         /* Correctly terminate the directory */
634         if (i == EXIF_IFD_0 && (data->ifd[EXIF_IFD_1]->count ||
635                                 data->size)) {
636
637                 /*
638                  * We are saving IFD 0. Tell where IFD 1 starts and save
639                  * IFD 1.
640                  */
641                 exif_set_long (*d + 6 + offset, data->priv->order, *ds - 6);
642                 exif_data_save_data_content (data, data->ifd[EXIF_IFD_1], d, ds,
643                                              *ds - 6);
644         } else
645                 exif_set_long (*d + 6 + offset, data->priv->order, 0);
646 }
647
648 typedef enum {
649         EXIF_DATA_TYPE_MAKER_NOTE_NONE          = 0,
650         EXIF_DATA_TYPE_MAKER_NOTE_CANON         = 1,
651         EXIF_DATA_TYPE_MAKER_NOTE_OLYMPUS       = 2,
652         EXIF_DATA_TYPE_MAKER_NOTE_PENTAX        = 3
653 } ExifDataTypeMakerNote;
654
655 static ExifDataTypeMakerNote
656 exif_data_get_type_maker_note (ExifData *d)
657 {
658         ExifEntry *e, *em;
659         char value[1024];
660
661         if (!d) 
662                 return EXIF_DATA_TYPE_MAKER_NOTE_NONE;
663         
664         e = exif_data_get_entry (d, EXIF_TAG_MAKER_NOTE);
665         if (!e) 
666                 return EXIF_DATA_TYPE_MAKER_NOTE_NONE;
667
668         /* Olympus & Nikon */
669         if ((e->size >= 5) && (!memcmp (e->data, "OLYMP", 5) ||
670                                !memcmp (e->data, "Nikon", 5)))
671                 return EXIF_DATA_TYPE_MAKER_NOTE_OLYMPUS;
672
673         em = exif_data_get_entry (d, EXIF_TAG_MAKE);
674         if (!em) 
675                 return EXIF_DATA_TYPE_MAKER_NOTE_NONE;
676
677         /* Canon */
678         if (!strcmp (exif_entry_get_value (em, value, sizeof (value)), "Canon"))
679                 return EXIF_DATA_TYPE_MAKER_NOTE_CANON;
680
681         /* Pentax & some variant of Nikon */
682         if ((e->size >= 2) && (e->data[0] == 0x00) && (e->data[1] == 0x1b)) {
683                 if (!strncasecmp (
684                             exif_entry_get_value (em, value, sizeof(value)),
685                             "Nikon", 5))
686                         return EXIF_DATA_TYPE_MAKER_NOTE_OLYMPUS;
687                 else
688                         return EXIF_DATA_TYPE_MAKER_NOTE_PENTAX;
689         }
690
691         return EXIF_DATA_TYPE_MAKER_NOTE_NONE;
692 }
693
694 #define LOG_TOO_SMALL \
695 exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA, "ExifData", \
696                 _("Size of data too small to allow for EXIF data."));
697
698 void
699 exif_data_load_data (ExifData *data, const unsigned char *d_orig,
700                      unsigned int ds_orig)
701 {
702         unsigned int l;
703         ExifLong offset;
704         ExifShort n;
705         const unsigned char *d = d_orig;
706         unsigned int ds = ds_orig, len;
707
708         if (!data || !data->priv || !d || !ds) 
709                 return;
710
711         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
712                   "Parsing %i byte(s) EXIF data...\n", ds);
713
714         /*
715          * It can be that the data starts with the EXIF header. If it does
716          * not, search the EXIF marker.
717          */
718         if (ds < 6) {
719                 LOG_TOO_SMALL;
720                 return;
721         }
722         if (!memcmp (d, ExifHeader, 6)) {
723                 exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
724                           "Found EXIF header.");
725         } else {
726                 while (1) {
727                         while ((d[0] == 0xff) && ds) {
728                                 d++;
729                                 ds--;
730                         }
731
732                         /* JPEG_MARKER_SOI */
733                         if (d[0] == JPEG_MARKER_SOI) {
734                                 d++;
735                                 ds--;
736                                 continue;
737                         }
738
739                         /* JPEG_MARKER_APP0 */
740                         if (d[0] == JPEG_MARKER_APP0) {
741                                 d++;
742                                 ds--;
743                                 l = (d[0] << 8) | d[1];
744                                 if (l > ds)
745                                         return;
746                                 d += l;
747                                 ds -= l;
748                                 continue;
749                         }
750
751                         /* JPEG_MARKER_APP1 */
752                         if (d[0] == JPEG_MARKER_APP1)
753                                 break;
754
755                         /* Unknown marker or data. Give up. */
756                         exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA,
757                                   "ExifData", _("EXIF marker not found."));
758                         return;
759                 }
760                 d++;
761                 ds--;
762                 if (ds < 2) {
763                         LOG_TOO_SMALL;
764                         return;
765                 }
766                 len = (d[0] << 8) | d[1];
767                 exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
768                           "We have to deal with %i byte(s) of EXIF data.",
769                           len);
770                 d += 2;
771                 ds -= 2;
772         }
773
774         /*
775          * Verify the exif header
776          * (offset 2, length 6).
777          */
778         if (ds < 6) {
779                 LOG_TOO_SMALL;
780                 return;
781         }
782         if (memcmp (d, ExifHeader, 6)) {
783                 exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA,
784                           "ExifData", _("EXIF header not found."));
785                 return;
786         }
787
788         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
789                   "Found EXIF header.");
790
791         /* Byte order (offset 6, length 2) */
792         if (ds < 14)
793                 return;
794         if (!memcmp (d + 6, "II", 2))
795                 data->priv->order = EXIF_BYTE_ORDER_INTEL;
796         else if (!memcmp (d + 6, "MM", 2))
797                 data->priv->order = EXIF_BYTE_ORDER_MOTOROLA;
798         else {
799                 exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA,
800                           "ExifData", _("Unknown encoding."));
801                 return;
802         }
803
804         /* Fixed value */
805         if (exif_get_short (d + 8, data->priv->order) != 0x002a)
806                 return;
807
808         /* IFD 0 offset */
809         offset = exif_get_long (d + 10, data->priv->order);
810         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData", 
811                   "IFD 0 at %i.", (int) offset);
812
813         /* Parse the actual exif data (usually offset 14 from start) */
814         exif_data_load_data_content (data, EXIF_IFD_0, d + 6, ds - 6, offset, 0);
815
816         /* IFD 1 offset */
817         if (offset + 6 + 2 > ds) {
818                 return;
819         }
820         n = exif_get_short (d + 6 + offset, data->priv->order);
821         if (offset + 6 + 2 + 12 * n + 4 > ds) {
822                 return;
823         }
824         offset = exif_get_long (d + 6 + offset + 2 + 12 * n, data->priv->order);
825         if (offset) {
826                 exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
827                           "IFD 1 at %i.", (int) offset);
828
829                 /* Sanity check. */
830                 if (offset > ds - 6) {
831                         exif_log (data->priv->log, EXIF_LOG_CODE_CORRUPT_DATA,
832                                   "ExifData", "Bogus offset.");
833                         return;
834                 }
835
836                 exif_data_load_data_content (data, EXIF_IFD_1, d + 6, ds - 6, offset, 0);
837         }
838
839         /*
840          * If we got an EXIF_TAG_MAKER_NOTE, try to interpret it. Some
841          * cameras use pointers in the maker note tag that point to the
842          * space between IFDs. Here is the only place where we have access
843          * to that data.
844          */
845         switch (exif_data_get_type_maker_note (data)) {
846         case EXIF_DATA_TYPE_MAKER_NOTE_OLYMPUS:
847                 data->priv->md = exif_mnote_data_olympus_new (data->priv->mem);
848                 break;
849         case EXIF_DATA_TYPE_MAKER_NOTE_PENTAX:
850                 data->priv->md = exif_mnote_data_pentax_new (data->priv->mem);
851                 break;
852         case EXIF_DATA_TYPE_MAKER_NOTE_CANON:
853                 data->priv->md = exif_mnote_data_canon_new (data->priv->mem);
854                 break;
855         default:
856                 break;
857         }
858
859         /* 
860          * If we are able to interpret the maker note, do so.
861          */
862         if (data->priv->md) {
863                 exif_mnote_data_log (data->priv->md, data->priv->log);
864                 exif_mnote_data_set_byte_order (data->priv->md,
865                                                 data->priv->order);
866                 exif_mnote_data_set_offset (data->priv->md,
867                                             data->priv->offset_mnote);
868                 exif_mnote_data_load (data->priv->md, d, ds);
869         }
870
871         if (data->priv->options & EXIF_DATA_OPTION_FOLLOW_SPECIFICATION)
872                 exif_data_fix (data);
873 }
874
875 void
876 exif_data_save_data (ExifData *data, unsigned char **d, unsigned int *ds)
877 {
878         if (!data || !d || !ds)
879                 return;
880
881         /* Header */
882         *ds = 14;
883         *d = exif_data_alloc (data, *ds);
884         if (!*d) 
885                 return;
886         memcpy (*d, ExifHeader, 6);
887
888         /* Order (offset 6) */
889         if (data->priv->order == EXIF_BYTE_ORDER_INTEL) {
890                 memcpy (*d + 6, "II", 2);
891         } else {
892                 memcpy (*d + 6, "MM", 2);
893         }
894
895         /* Fixed value (2 bytes, offset 8) */
896         exif_set_short (*d + 8, data->priv->order, 0x002a);
897
898         /*
899          * IFD 0 offset (4 bytes, offset 10).
900          * We will start 8 bytes after the
901          * EXIF header (2 bytes for order, another 2 for the test, and
902          * 4 bytes for the IFD 0 offset make 8 bytes together).
903          */
904         exif_set_long (*d + 10, data->priv->order, 8);
905
906         /* Now save IFD 0. IFD 1 will be saved automatically. */
907         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
908                   "Saving IFDs...");
909         exif_data_save_data_content (data, data->ifd[EXIF_IFD_0], d, ds,
910                                      *ds - 6);
911         exif_log (data->priv->log, EXIF_LOG_CODE_DEBUG, "ExifData",
912                   "Saved %i byte(s) EXIF data.", *ds);
913 }
914
915 ExifData *
916 exif_data_new_from_file (const char *path)
917 {
918         ExifData *edata;
919         ExifLoader *loader;
920
921         loader = exif_loader_new ();
922         exif_loader_write_file (loader, path);
923         edata = exif_loader_get_data (loader);
924         exif_loader_unref (loader);
925
926         return (edata);
927 }
928
929 void
930 exif_data_ref (ExifData *data)
931 {
932         if (!data)
933                 return;
934
935         data->priv->ref_count++;
936 }
937
938 void
939 exif_data_unref (ExifData *data)
940 {
941         if (!data) 
942                 return;
943
944         data->priv->ref_count--;
945         if (!data->priv->ref_count) 
946                 exif_data_free (data);
947 }
948
949 void
950 exif_data_free (ExifData *data)
951 {
952         unsigned int i;
953         ExifMem *mem = (data && data->priv) ? data->priv->mem : NULL;
954
955         if (!data) 
956                 return;
957
958         for (i = 0; i < EXIF_IFD_COUNT; i++) {
959                 if (data->ifd[i]) {
960                         exif_content_unref (data->ifd[i]);
961                         data->ifd[i] = NULL;
962                 }
963         }
964
965         if (data->data) {
966                 exif_mem_free (mem, data->data);
967                 data->data = NULL;
968         }
969
970         if (data->priv) {
971                 if (data->priv->log) {
972                         exif_log_unref (data->priv->log);
973                         data->priv->log = NULL;
974                 }
975                 if (data->priv->md) {
976                         exif_mnote_data_unref (data->priv->md);
977                         data->priv->md = NULL;
978                 }
979                 exif_mem_free (mem, data->priv);
980                 exif_mem_free (mem, data);
981         }
982
983         exif_mem_unref (mem);
984 }
985
986 void
987 exif_data_dump (ExifData *data)
988 {
989         unsigned int i;
990
991         if (!data)
992                 return;
993
994         for (i = 0; i < EXIF_IFD_COUNT; i++) {
995                 if (data->ifd[i] && data->ifd[i]->count) {
996                         printf ("Dumping IFD '%s'...\n",
997                                 exif_ifd_get_name (i));
998                         exif_content_dump (data->ifd[i], 0);
999                 }
1000         }
1001
1002         if (data->data) {
1003                 printf ("%i byte(s) thumbnail data available.", data->size);
1004                 if (data->size >= 4) {
1005                         printf ("0x%02x 0x%02x ... 0x%02x 0x%02x\n",
1006                                 data->data[0], data->data[1],
1007                                 data->data[data->size - 2],
1008                                 data->data[data->size - 1]);
1009                 }
1010         }
1011 }
1012
1013 ExifByteOrder
1014 exif_data_get_byte_order (ExifData *data)
1015 {
1016         if (!data)
1017                 return (0);
1018
1019         return (data->priv->order);
1020 }
1021
1022 void
1023 exif_data_foreach_content (ExifData *data, ExifDataForeachContentFunc func,
1024                            void *user_data)
1025 {
1026         unsigned int i;
1027
1028         if (!data || !func)
1029                 return;
1030
1031         for (i = 0; i < EXIF_IFD_COUNT; i++)
1032                 func (data->ifd[i], user_data);
1033 }
1034
1035 typedef struct _ByteOrderChangeData ByteOrderChangeData;
1036 struct _ByteOrderChangeData {
1037         ExifByteOrder old, new;
1038 };
1039
1040 static void
1041 entry_set_byte_order (ExifEntry *e, void *data)
1042 {
1043         ByteOrderChangeData *d = data;
1044
1045         if (!e)
1046                 return;
1047
1048         exif_array_set_byte_order (e->format, e->data, e->components, d->old, d->new);
1049 }
1050
1051 static void
1052 content_set_byte_order (ExifContent *content, void *data)
1053 {
1054         exif_content_foreach_entry (content, entry_set_byte_order, data);
1055 }
1056
1057 void
1058 exif_data_set_byte_order (ExifData *data, ExifByteOrder order)
1059 {
1060         ByteOrderChangeData d;
1061
1062         if (!data || (order == data->priv->order))
1063                 return;
1064
1065         d.old = data->priv->order;
1066         d.new = order;
1067         exif_data_foreach_content (data, content_set_byte_order, &d);
1068         data->priv->order = order;
1069         if (data->priv->md)
1070                 exif_mnote_data_set_byte_order (data->priv->md, order);
1071 }
1072
1073 void
1074 exif_data_log (ExifData *data, ExifLog *log)
1075 {
1076         unsigned int i;
1077
1078         if (!data || !data->priv) 
1079                 return;
1080         exif_log_unref (data->priv->log);
1081         data->priv->log = log;
1082         exif_log_ref (log);
1083
1084         for (i = 0; i < EXIF_IFD_COUNT; i++)
1085                 exif_content_log (data->ifd[i], log);
1086 }
1087
1088 /* Used internally within libexif */
1089 ExifLog *exif_data_get_log (ExifData *);
1090 ExifLog *
1091 exif_data_get_log (ExifData *data)
1092 {
1093         if (!data || !data->priv) 
1094                 return NULL;
1095         return data->priv->log;
1096 }
1097
1098 static struct {
1099         ExifDataOption option;
1100         const char *name;
1101         const char *description;
1102 } exif_data_option[] = {
1103         {EXIF_DATA_OPTION_IGNORE_UNKNOWN_TAGS, N_("Ignore unknown tags"),
1104          N_("Ignore unknown tags when loading EXIF data.")},
1105         {EXIF_DATA_OPTION_FOLLOW_SPECIFICATION, N_("Follow specification"),
1106          N_("Add, correct and remove entries to get EXIF data that follows "
1107             "the specification.")},
1108         {0, NULL, NULL}
1109 };
1110
1111 const char *
1112 exif_data_option_get_name (ExifDataOption o)
1113 {
1114         unsigned int i;
1115
1116         for (i = 0; exif_data_option[i].name; i++)
1117                 if (exif_data_option[i].option == o) 
1118                         break;
1119         return _(exif_data_option[i].name);
1120 }
1121
1122 const char *
1123 exif_data_option_get_description (ExifDataOption o)
1124 {
1125         unsigned int i;
1126
1127         for (i = 0; exif_data_option[i].description; i++)
1128                 if (exif_data_option[i].option == o) 
1129                         break;
1130         return _(exif_data_option[i].description);
1131 }
1132
1133 void
1134 exif_data_set_option (ExifData *d, ExifDataOption o)
1135 {
1136         if (!d) 
1137                 return;
1138
1139         d->priv->options |= o;
1140 }
1141
1142 void
1143 exif_data_unset_option (ExifData *d, ExifDataOption o)
1144 {
1145         if (!d) 
1146                 return;
1147
1148         d->priv->options &= ~o;
1149 }
1150
1151 static void
1152 fix_func (ExifContent *c, void *data)
1153 {
1154         switch (exif_content_get_ifd (c)) {
1155         case EXIF_IFD_1:
1156                 if (c->parent->data)
1157                         exif_content_fix (c);
1158                 else {
1159                         exif_log (c->parent->priv->log, EXIF_LOG_CODE_DEBUG, "exif-data",
1160                                   "No thumbnail but entries on thumbnail. These entries have been "
1161                                   "removed.");
1162                         while (c->count)
1163                                 exif_content_remove_entry (c, c->entries[c->count - 1]);
1164                 }
1165                 break;
1166         default:
1167                 exif_content_fix (c);
1168         }
1169 }
1170
1171 void
1172 exif_data_fix (ExifData *d)
1173 {
1174         exif_data_foreach_content (d, fix_func, NULL);
1175 }
1176
1177 void
1178 exif_data_set_data_type (ExifData *d, ExifDataType dt)
1179 {
1180         if (!d || !d->priv) 
1181                 return;
1182
1183         d->priv->data_type = dt;
1184 }
1185
1186 ExifDataType
1187 exif_data_get_data_type (ExifData *d)
1188 {
1189         return (d && d->priv) ? d->priv->data_type : EXIF_DATA_TYPE_COUNT;
1190 }