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