libexif/exif-data.c

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