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