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