libexif/exif-data.c

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