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