1 files changed, 272 insertions, 0 deletions
diff --git a/ext/metadata/metadatamuxjpeg.c b/ext/metadata/metadatamuxjpeg.c
new file mode 100644
index 00000000..15d9d19c
--- /dev/null
+++ b/ext/metadata/metadatamuxjpeg.c
@@ -0,0 +1,272 @@
+/*
+ * GStreamer
+ * Copyright 2007 Edgard Lima <edgard.lima@indt.org.br>
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ *
+ * Alternatively, the contents of this file may be used under the
+ * GNU Lesser General Public License Version 2.1 (the "LGPL"), in
+ * which case the following provisions apply instead of the ones
+ * mentioned above:
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#include "metadatamuxjpeg.h"
+
+#include <string.h>
+
+#include <libiptcdata/iptc-jpeg.h>
+
+static int
+metadatamux_jpeg_reading (JpegMuxData * jpeg_data, guint8 ** buf,
+    guint32 * bufsize, const guint32 offset, const guint8 * step_buf,
+    guint8 ** next_start, guint32 * next_size);
+
+#define READ(buf, size) ( (size)--, *((buf)++) )
+
+void
+metadatamux_jpeg_lazy_update (JpegMuxData * jpeg_data)
+{
+  gsize i;
+
+  for (i = 0; i < jpeg_data->inject_chunks->len; ++i) {
+    if (jpeg_data->inject_chunks->chunk[i].type == MD_CHUNK_EXIF &&
+        jpeg_data->inject_chunks->chunk[i].size > 0) {
+      /* zero size chunks should be removed before but we check anyway */
+      break;
+    }
+  }
+  if (i == jpeg_data->inject_chunks->len) {
+    /* EXIF not injected so not strip JFIF anymore */
+    metadata_chunk_array_clear (jpeg_data->strip_chunks);
+  }
+
+}
+
+void
+metadatamux_jpeg_init (JpegMuxData * jpeg_data, GstAdapter ** exif_adpt,
+    GstAdapter ** iptc_adpt, GstAdapter ** xmp_adpt,
+    MetadataChunkArray * strip_chunks, MetadataChunkArray * inject_chunks)
+{
+  jpeg_data->state = JPEG_MUX_NULL;
+
+  jpeg_data->strip_chunks = strip_chunks;
+  jpeg_data->inject_chunks = inject_chunks;
+
+}
+
+void
+metadatamux_jpeg_dispose (JpegMuxData * jpeg_data)
+{
+  jpeg_data->inject_chunks = NULL;
+  jpeg_data->strip_chunks = NULL;
+
+  jpeg_data->state = JPEG_MUX_NULL;
+}
+
+int
+metadatamux_jpeg_parse (JpegMuxData * jpeg_data, guint8 * buf,
+    guint32 * bufsize, const guint32 offset, guint8 ** next_start,
+    guint32 * next_size)
+{
+
+  int ret = 0;
+  guint8 mark[2] = { 0x00, 0x00 };
+  const guint8 *step_buf = buf;
+
+  *next_start = buf;
+
+  if (jpeg_data->state == JPEG_MUX_NULL) {
+
+    if (*bufsize < 2) {
+      *next_size = (buf - *next_start) + 2;
+      ret = 1;
+      goto done;
+    }
+
+    mark[0] = READ (buf, *bufsize);
+    mark[1] = READ (buf, *bufsize);
+
+    if (mark[0] != 0xFF || mark[1] != 0xD8) {
+      ret = -1;
+      goto done;
+    }
+
+    jpeg_data->state = JPEG_MUX_READING;
+
+  }
+
+  while (ret == 0) {
+    switch (jpeg_data->state) {
+      case JPEG_MUX_READING:
+        ret =
+            metadatamux_jpeg_reading (jpeg_data, &buf, bufsize,
+            offset, step_buf, next_start, next_size);
+        break;
+      case JPEG_MUX_DONE:
+        goto done;
+        break;
+      default:
+        ret = -1;
+        break;
+    }
+  }
+
+done:
+
+  return ret;
+
+}
+
+
+/* look for markers */
+static int
+metadatamux_jpeg_reading (JpegMuxData * jpeg_data, guint8 ** buf,
+    guint32 * bufsize, const guint32 offset, const guint8 * step_buf,
+    guint8 ** next_start, guint32 * next_size)
+{
+
+  int ret = -1;
+  guint8 mark[2] = { 0x00, 0x00 };
+  guint16 chunk_size = 0;
+  gint64 new_chunk_offset = 0;
+  MetadataChunk chunk;
+  gboolean jfif_found = FALSE;
+
+  static const char JfifHeader[] = "JFIF";
+  static const unsigned char ExifHeader[] =
+      { 0x45, 0x78, 0x69, 0x66, 0x00, 0x00 };
+  static const char IptcHeader[] = "Photoshop 3.0";
+  static const char XmpHeader[] = "http://ns.adobe.com/xap/1.0/";
+
+  *next_start = *buf;
+
+  if (*bufsize < 2) {
+    *next_size = (*buf - *next_start) + 2;
+    ret = 1;
+    goto done;
+  }
+
+  mark[0] = READ (*buf, *bufsize);
+  mark[1] = READ (*buf, *bufsize);
+
+  if (mark[0] == 0xFF) {
+    if (mark[1] == 0xD9) {      /* end of image */
+      ret = 0;
+      jpeg_data->state = JPEG_MUX_DONE;
+      goto done;
+    } else if (mark[1] == 0xDA) {       /* start of scan, lets not look behinf of this */
+      ret = 0;
+      jpeg_data->state = JPEG_MUX_DONE;
+      goto done;
+    }
+
+    if (*bufsize < 2) {
+      *next_size = (*buf - *next_start) + 2;
+      ret = 1;
+      goto done;
+    }
+
+    chunk_size = READ (*buf, *bufsize) << 8;
+    chunk_size += READ (*buf, *bufsize);
+
+    if (mark[1] == 0xE0) {      /* may be JFIF */
+
+      if (chunk_size >= 16) {
+        if (*bufsize < 14) {
+          *next_size = (*buf - *next_start) + 14;
+          ret = 1;
+          goto done;
+        }
+
+        if (0 == memcmp (JfifHeader, *buf, 5)) {
+          jfif_found = TRUE;
+          /* FIXME: should we fail if not find JFIF */
+          /* Yes, I think so */
+        }
+      }
+
+    }
+
+    new_chunk_offset = (*buf - step_buf) + offset - 4;  /* maker + size */
+
+    /* EXIF will always be in the begining */
+
+    memset (&chunk, 0x00, sizeof (MetadataChunk));
+    chunk.offset_orig = new_chunk_offset;
+    chunk.type = MD_CHUNK_EXIF;
+    metadata_chunk_array_append_sorted (jpeg_data->inject_chunks, &chunk);
+
+    if (jfif_found) {
+      /* remove JFIF chunk */
+      /* this acation can be canceled with lazy update if no Exif is add */
+
+      memset (&chunk, 0x00, sizeof (MetadataChunk));
+      chunk.offset_orig = new_chunk_offset;
+      chunk.size = chunk_size + 2;      /* chunk size plus app marker */
+      chunk.type = MD_CHUNK_UNKNOWN;
+
+      metadata_chunk_array_append_sorted (jpeg_data->strip_chunks, &chunk);
+
+      new_chunk_offset = chunk.offset_orig + chunk.size;
+    }
+
+    /* IPTC */
+
+    memset (&chunk, 0x00, sizeof (MetadataChunk));
+    chunk.offset_orig = new_chunk_offset;
+    chunk.type = MD_CHUNK_IPTC;
+    metadata_chunk_array_append_sorted (jpeg_data->inject_chunks, &chunk);
+
+    /* XMP */
+
+    memset (&chunk, 0x00, sizeof (MetadataChunk));
+    chunk.offset_orig = new_chunk_offset;
+    chunk.type = MD_CHUNK_XMP;
+    metadata_chunk_array_append_sorted (jpeg_data->inject_chunks, &chunk);
+
+    jpeg_data->state = JPEG_MUX_DONE;
+    ret = 0;
+
+  } else {
+    /* invalid JPEG chunk */
+    ret = -1;
+  }
+
+
+done:
+
+  return ret;
+
+
+}