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