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