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