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