gst/typefind/gsttypefindfunctions.c

   1 /* GStreamer
   2  * Copyright (C) 2003 Benjamin Otte <in7y118@public.uni-hamburg.de>
   3  * Copyright (C) 2005-2009 Tim-Philipp Müller <tim centricular net>
   4  * Copyright (C) 2009 Sebastian Dröge <sebastian.droege@collabora.co.uk>
   5  *
   6  * gsttypefindfunctions.c: collection of various typefind functions
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Library General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Library General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Library General Public
  19  * License along with this library; if not, write to the
  20  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23
  24 #ifdef HAVE_CONFIG_H
  25 #include "config.h"
  26 #endif
  27
  28 #include <glib.h>
  29 #include <glib/gprintf.h>
  30
  31 /* don't want to add gio xdgmime typefinder if gio was disabled via configure */
  32 #ifdef HAVE_GIO
  33 #include <gio/gio.h>
  34 #define USE_GIO
  35 #endif
  36
  37 #include <gst/gst.h>
  38
  39 #include <stdio.h>
  40 #include <string.h>
  41 #include <ctype.h>
  42
  43 #include <gst/pbutils/pbutils.h>
  44 #include <gst/base/gstbytereader.h>
  45
  46 GST_DEBUG_CATEGORY_STATIC (type_find_debug);
  47 #define GST_CAT_DEFAULT type_find_debug
  48
  49 /* so our code stays ready for 0.11 */
  50 #define gst_type_find_peek(tf,off,len) \
  51     ((const guint8 *)gst_type_find_peek((tf),(off),(len)))
  52
  53 /* DataScanCtx: helper for typefind functions that scan through data
  54  * step-by-step, to avoid doing a peek at each and every offset */
  55
  56 #define DATA_SCAN_CTX_CHUNK_SIZE 4096
  57
  58 typedef struct
  59 {
  60   guint64 offset;
  61   const guint8 *data;
  62   gint size;
  63 } DataScanCtx;
  64
  65 static inline void
  66 data_scan_ctx_advance (GstTypeFind * tf, DataScanCtx * c, guint bytes_to_skip)
  67 {
  68   c->offset += bytes_to_skip;
  69   if (G_LIKELY (c->size > bytes_to_skip)) {
  70     c->size -= bytes_to_skip;
  71     c->data += bytes_to_skip;
  72   } else {
  73     c->data += c->size;
  74     c->size = 0;
  75   }
  76 }
  77
  78 static inline gboolean
  79 data_scan_ctx_ensure_data (GstTypeFind * tf, DataScanCtx * c, gint min_len)
  80 {
  81   const guint8 *data;
  82   guint64 len;
  83   guint chunk_len = MAX (DATA_SCAN_CTX_CHUNK_SIZE, min_len);
  84
  85   if (G_LIKELY (c->size >= min_len))
  86     return TRUE;
  87
  88   data = gst_type_find_peek (tf, c->offset, chunk_len);
  89   if (G_LIKELY (data != NULL)) {
  90     c->data = data;
  91     c->size = chunk_len;
  92     return TRUE;
  93   }
  94
  95   /* if there's less than our chunk size, try to get as much as we can, but
  96    * always at least min_len bytes (we might be typefinding the first buffer
  97    * of the stream and not have as much data available as we'd like) */
  98   len = gst_type_find_get_length (tf);
  99   if (len > 0) {
 100     len = CLAMP (len - c->offset, min_len, chunk_len);
 101   } else {
 102     len = min_len;
 103   }
 104
 105   data = gst_type_find_peek (tf, c->offset, len);
 106   if (data != NULL) {
 107     c->data = data;
 108     c->size = len;
 109     return TRUE;
 110   }
 111
 112   return FALSE;
 113 }
 114
 115 static inline gboolean
 116 data_scan_ctx_memcmp (GstTypeFind * tf, DataScanCtx * c, guint offset,
 117     const gchar * data, guint len)
 118 {
 119   if (!data_scan_ctx_ensure_data (tf, c, offset + len))
 120     return FALSE;
 121
 122   return (memcmp (c->data + offset, data, len) == 0);
 123 }
 124
 125 /*** text/plain ***/
 126 static gboolean xml_check_first_element (GstTypeFind * tf,
 127     const gchar * element, guint elen, gboolean strict);
 128 static gboolean sdp_check_header (GstTypeFind * tf);
 129
 130 static GstStaticCaps utf8_caps = GST_STATIC_CAPS ("text/plain");
 131
 132 #define UTF8_CAPS gst_static_caps_get(&utf8_caps)
 133
 134 static gboolean
 135 utf8_type_find_have_valid_utf8_at_offset (GstTypeFind * tf, guint64 offset,
 136     GstTypeFindProbability * prob)
 137 {
 138   const guint8 *data;
 139
 140   /* randomly decided values */
 141   guint min_size = 16;          /* minimum size  */
 142   guint size = 32 * 1024;       /* starting size */
 143   guint probability = 95;       /* starting probability */
 144   guint step = 10;              /* how much we reduce probability in each
 145                                  * iteration */
 146
 147   while (probability > step && size > min_size) {
 148     data = gst_type_find_peek (tf, offset, size);
 149     if (data) {
 150       gchar *end;
 151       gchar *start = (gchar *) data;
 152
 153       if (g_utf8_validate (start, size, (const gchar **) &end) || (end - start + 4 > size)) {   /* allow last char to be cut off */
 154         *prob = probability;
 155         return TRUE;
 156       }
 157       *prob = 0;
 158       return FALSE;
 159     }
 160     size /= 2;
 161     probability -= step;
 162   }
 163   *prob = 0;
 164   return FALSE;
 165 }
 166
 167 static void
 168 utf8_type_find (GstTypeFind * tf, gpointer unused)
 169 {
 170   GstTypeFindProbability start_prob, mid_prob;
 171   guint64 length;
 172
 173   /* leave xml to the xml typefinders */
 174   if (xml_check_first_element (tf, "", 0, TRUE))
 175     return;
 176
 177   /* leave sdp to the sdp typefinders */
 178   if (sdp_check_header (tf))
 179     return;
 180
 181   /* check beginning of stream */
 182   if (!utf8_type_find_have_valid_utf8_at_offset (tf, 0, &start_prob))
 183     return;
 184
 185   GST_LOG ("start is plain text with probability of %u", start_prob);
 186
 187   /* POSSIBLE is the highest probability we ever return if we can't
 188    * probe into the middle of the file and don't know its length */
 189
 190   length = gst_type_find_get_length (tf);
 191   if (length == 0 || length == (guint64) - 1) {
 192     gst_type_find_suggest (tf, MIN (start_prob, GST_TYPE_FIND_POSSIBLE),
 193         UTF8_CAPS);
 194     return;
 195   }
 196
 197   if (length < 64 * 1024) {
 198     gst_type_find_suggest (tf, start_prob, UTF8_CAPS);
 199     return;
 200   }
 201
 202   /* check middle of stream */
 203   if (!utf8_type_find_have_valid_utf8_at_offset (tf, length / 2, &mid_prob))
 204     return;
 205
 206   GST_LOG ("middle is plain text with probability of %u", mid_prob);
 207   gst_type_find_suggest (tf, (start_prob + mid_prob) / 2, UTF8_CAPS);
 208 }
 209
 210 /*** text/utf-16 and text/utf-32} ***/
 211 /* While UTF-8 is unicode too, using text/plain for UTF-16 and UTF-32
 212    is going to break stuff. */
 213
 214 typedef struct
 215 {
 216   size_t bomlen;
 217   const char *const bom;
 218     gboolean (*checker) (const guint8 *, gint, gint);
 219   int boost;
 220   int endianness;
 221 } GstUnicodeTester;
 222
 223 static gboolean
 224 check_utf16 (const guint8 * data, gint len, gint endianness)
 225 {
 226   GstByteReader br;
 227   guint16 high, low;
 228
 229   low = high = 0;
 230
 231   if (len & 1)
 232     return FALSE;
 233
 234   gst_byte_reader_init (&br, data, len);
 235   while (len >= 2) {
 236     /* test first for a single 16 bit value in the BMP */
 237     if (endianness == G_BIG_ENDIAN)
 238       high = gst_byte_reader_get_uint16_be_unchecked (&br);
 239     else
 240       high = gst_byte_reader_get_uint16_le_unchecked (&br);
 241     if (high >= 0xD800 && high <= 0xDBFF) {
 242       /* start of a surrogate pair */
 243       if (len < 4)
 244         return FALSE;
 245       len -= 2;
 246       if (endianness == G_BIG_ENDIAN)
 247         low = gst_byte_reader_get_uint16_be_unchecked (&br);
 248       else
 249         low = gst_byte_reader_get_uint16_le_unchecked (&br);
 250       if (low >= 0xDC00 && low <= 0xDFFF) {
 251         /* second half of the surrogate pair */
 252       } else
 253         return FALSE;
 254     } else {
 255       if (high >= 0xDC00 && high <= 0xDFFF)
 256         return FALSE;
 257     }
 258     len -= 2;
 259   }
 260   return TRUE;
 261 }
 262
 263 static gboolean
 264 check_utf32 (const guint8 * data, gint len, gint endianness)
 265 {
 266   if (len & 3)
 267     return FALSE;
 268   while (len > 3) {
 269     guint32 v;
 270     if (endianness == G_BIG_ENDIAN)
 271       v = (data[0] << 24) | (data[1] << 16) | (data[2] << 8) | data[3];
 272     else
 273       v = (data[3] << 24) | (data[2] << 16) | (data[1] << 8) | data[0];
 274     if (v >= 0x10FFFF)
 275       return FALSE;
 276     data += 4;
 277     len -= 4;
 278   }
 279   return TRUE;
 280 }
 281
 282 static void
 283 unicode_type_find (GstTypeFind * tf, const GstUnicodeTester * tester,
 284     guint n_tester, const char *media_type, gboolean require_bom)
 285 {
 286   size_t n;
 287   gint len = 4;
 288   const guint8 *data = gst_type_find_peek (tf, 0, len);
 289   int prob = -1;
 290   const gint max_scan_size = 256 * 1024;
 291   int endianness = 0;
 292
 293   if (!data) {
 294     len = 2;
 295     data = gst_type_find_peek (tf, 0, len);
 296     if (!data)
 297       return;
 298   }
 299
 300   /* find a large enough size that works */
 301   while (len < max_scan_size) {
 302     size_t newlen = len << 1;
 303     const guint8 *newdata = gst_type_find_peek (tf, 0, newlen);
 304     if (!newdata)
 305       break;
 306     len = newlen;
 307     data = newdata;
 308   }
 309
 310   for (n = 0; n < n_tester; ++n) {
 311     int bom_boost = 0, tmpprob;
 312     if (len >= tester[n].bomlen) {
 313       if (!memcmp (data, tester[n].bom, tester[n].bomlen))
 314         bom_boost = tester[n].boost;
 315     }
 316     if (require_bom && bom_boost == 0)
 317       continue;
 318     if (!(*tester[n].checker) (data, len, tester[n].endianness))
 319       continue;
 320     tmpprob = GST_TYPE_FIND_POSSIBLE - 20 + bom_boost;
 321     if (tmpprob > prob) {
 322       prob = tmpprob;
 323       endianness = tester[n].endianness;
 324     }
 325   }
 326
 327   if (prob > 0) {
 328     GST_DEBUG ("This is valid %s %s", media_type,
 329         endianness == G_BIG_ENDIAN ? "be" : "le");
 330     gst_type_find_suggest_simple (tf, prob, media_type,
 331         "endianness", G_TYPE_INT, endianness, NULL);
 332   }
 333 }
 334
 335 static GstStaticCaps utf16_caps = GST_STATIC_CAPS ("text/utf-16");
 336
 337 #define UTF16_CAPS gst_static_caps_get(&utf16_caps)
 338
 339 static void
 340 utf16_type_find (GstTypeFind * tf, gpointer unused)
 341 {
 342   static const GstUnicodeTester utf16tester[2] = {
 343     {2, "\xff\xfe", check_utf16, 10, G_LITTLE_ENDIAN},
 344     {2, "\xfe\xff", check_utf16, 20, G_BIG_ENDIAN},
 345   };
 346   unicode_type_find (tf, utf16tester, G_N_ELEMENTS (utf16tester),
 347       "text/utf-16", TRUE);
 348 }
 349
 350 static GstStaticCaps utf32_caps = GST_STATIC_CAPS ("text/utf-32");
 351
 352 #define UTF32_CAPS gst_static_caps_get(&utf32_caps)
 353
 354 static void
 355 utf32_type_find (GstTypeFind * tf, gpointer unused)
 356 {
 357   static const GstUnicodeTester utf32tester[2] = {
 358     {4, "\xff\xfe\x00\x00", check_utf32, 10, G_LITTLE_ENDIAN},
 359     {4, "\x00\x00\xfe\xff", check_utf32, 20, G_BIG_ENDIAN}
 360   };
 361   unicode_type_find (tf, utf32tester, G_N_ELEMENTS (utf32tester),
 362       "text/utf-32", TRUE);
 363 }
 364
 365 /*** text/uri-list ***/
 366
 367 static GstStaticCaps uri_caps = GST_STATIC_CAPS ("text/uri-list");
 368
 369 #define URI_CAPS (gst_static_caps_get(&uri_caps))
 370 #define BUFFER_SIZE 16          /* If the string is < 16 bytes we're screwed */
 371 #define INC_BUFFER {                                                    \
 372   pos++;                                                                \
 373   if (pos == BUFFER_SIZE) {                                             \
 374     pos = 0;                                                            \
 375     offset += BUFFER_SIZE;                                              \
 376     data = gst_type_find_peek (tf, offset, BUFFER_SIZE);                \
 377     if (data == NULL) return;                                           \
 378   } else {                                                              \
 379     data++;                                                             \
 380   }                                                                     \
 381 }
 382 static void
 383 uri_type_find (GstTypeFind * tf, gpointer unused)
 384 {
 385   const guint8 *data = gst_type_find_peek (tf, 0, BUFFER_SIZE);
 386   guint pos = 0;
 387   guint offset = 0;
 388
 389   if (data) {
 390     /* Search for # comment lines */
 391     while (*data == '#') {
 392       /* Goto end of line */
 393       while (*data != '\n') {
 394         INC_BUFFER;
 395       }
 396
 397       INC_BUFFER;
 398     }
 399
 400     if (!g_ascii_isalpha (*data)) {
 401       /* Had a non alpha char - can't be uri-list */
 402       return;
 403     }
 404
 405     INC_BUFFER;
 406
 407     while (g_ascii_isalnum (*data)) {
 408       INC_BUFFER;
 409     }
 410
 411     if (*data != ':') {
 412       /* First non alpha char is not a : */
 413       return;
 414     }
 415
 416     /* Get the next 2 bytes as well */
 417     data = gst_type_find_peek (tf, offset + pos, 3);
 418     if (data == NULL)
 419       return;
 420
 421     if (data[1] != '/' && data[2] != '/') {
 422       return;
 423     }
 424
 425     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, URI_CAPS);
 426   }
 427 }
 428
 429 /*** application/x-hls ***/
 430
 431 static GstStaticCaps hls_caps = GST_STATIC_CAPS ("application/x-hls");
 432 #define HLS_CAPS (gst_static_caps_get(&hls_caps))
 433
 434 /* See http://tools.ietf.org/html/draft-pantos-http-live-streaming-05 */
 435 static void
 436 hls_type_find (GstTypeFind * tf, gpointer unused)
 437 {
 438   DataScanCtx c = { 0, NULL, 0 };
 439
 440   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 7)))
 441     return;
 442
 443   if (memcmp (c.data, "#EXTM3U", 7))
 444     return;
 445
 446   data_scan_ctx_advance (tf, &c, 7);
 447
 448   /* Check only the first 256 bytes */
 449   while (c.offset < 256) {
 450     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 21)))
 451       return;
 452
 453     /* Search for # comment lines */
 454     if (c.data[0] == '#' && (memcmp (c.data, "#EXT-X-TARGETDURATION", 21) == 0
 455             || memcmp (c.data, "#EXT-X-STREAM-INF", 17) == 0)) {
 456       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, HLS_CAPS);
 457       return;
 458     }
 459
 460     data_scan_ctx_advance (tf, &c, 1);
 461   }
 462 }
 463
 464
 465 /*** application/xml **********************************************************/
 466
 467 #define XML_BUFFER_SIZE 16
 468 #define XML_INC_BUFFER {                                                \
 469   pos++;                                                                \
 470   if (pos == XML_BUFFER_SIZE) {                                         \
 471     pos = 0;                                                            \
 472     offset += XML_BUFFER_SIZE;                                          \
 473     data = gst_type_find_peek (tf, offset, XML_BUFFER_SIZE);            \
 474     if (data == NULL) return FALSE;                                     \
 475   } else {                                                              \
 476     data++;                                                             \
 477   }                                                                     \
 478 }
 479
 480 static gboolean
 481 xml_check_first_element (GstTypeFind * tf, const gchar * element, guint elen,
 482     gboolean strict)
 483 {
 484   gboolean got_xmldec;
 485   const guint8 *data;
 486   guint offset = 0;
 487   guint pos = 0;
 488
 489   data = gst_type_find_peek (tf, 0, XML_BUFFER_SIZE);
 490   if (!data)
 491     return FALSE;
 492
 493   /* look for the XMLDec
 494    * see XML spec 2.8, Prolog and Document Type Declaration
 495    * http://www.w3.org/TR/2004/REC-xml-20040204/#sec-prolog-dtd */
 496   got_xmldec = (memcmp (data, "<?xml", 5) == 0);
 497
 498   if (strict && !got_xmldec)
 499     return FALSE;
 500
 501   /* skip XMLDec in any case if we've got one */
 502   if (got_xmldec) {
 503     pos += 5;
 504     data += 5;
 505   }
 506
 507   /* look for the first element, it has to be the requested element. Bail
 508    * out if it is not within the first 4kB. */
 509   while (data && (offset + pos) < 4096) {
 510     while (*data != '<' && (offset + pos) < 4096) {
 511       XML_INC_BUFFER;
 512     }
 513
 514     XML_INC_BUFFER;
 515     if (!g_ascii_isalpha (*data)) {
 516       /* if not alphabetic, it's a PI or an element / attribute declaration
 517        * like <?xxx or <!xxx */
 518       XML_INC_BUFFER;
 519       continue;
 520     }
 521
 522     /* the first normal element, check if it's the one asked for */
 523     data = gst_type_find_peek (tf, offset + pos, elen + 1);
 524     return (data && element && strncmp ((char *) data, element, elen) == 0);
 525   }
 526
 527   return FALSE;
 528 }
 529
 530 static GstStaticCaps generic_xml_caps = GST_STATIC_CAPS ("application/xml");
 531
 532 #define GENERIC_XML_CAPS (gst_static_caps_get(&generic_xml_caps))
 533 static void
 534 xml_type_find (GstTypeFind * tf, gpointer unused)
 535 {
 536   if (xml_check_first_element (tf, "", 0, TRUE)) {
 537     gst_type_find_suggest (tf, GST_TYPE_FIND_MINIMUM, GENERIC_XML_CAPS);
 538   }
 539 }
 540
 541 /*** application/sdp *********************************************************/
 542
 543 static GstStaticCaps sdp_caps = GST_STATIC_CAPS ("application/sdp");
 544
 545 #define SDP_CAPS (gst_static_caps_get(&sdp_caps))
 546 static gboolean
 547 sdp_check_header (GstTypeFind * tf)
 548 {
 549   const guint8 *data;
 550
 551   data = gst_type_find_peek (tf, 0, 5);
 552   if (!data)
 553     return FALSE;
 554
 555   /* sdp must start with v=0[\r]\n */
 556   if (memcmp (data, "v=0", 3))
 557     return FALSE;
 558
 559   if (data[3] == '\r' && data[4] == '\n')
 560     return TRUE;
 561   if (data[3] == '\n')
 562     return TRUE;
 563
 564   return FALSE;
 565 }
 566
 567 static void
 568 sdp_type_find (GstTypeFind * tf, gpointer unused)
 569 {
 570   if (sdp_check_header (tf))
 571     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SDP_CAPS);
 572 }
 573
 574 /*** application/smil *********************************************************/
 575
 576 static GstStaticCaps smil_caps = GST_STATIC_CAPS ("application/smil");
 577
 578 #define SMIL_CAPS (gst_static_caps_get(&smil_caps))
 579 static void
 580 smil_type_find (GstTypeFind * tf, gpointer unused)
 581 {
 582   if (xml_check_first_element (tf, "smil", 4, FALSE)) {
 583     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SMIL_CAPS);
 584   }
 585 }
 586
 587 /*** text/html ***/
 588
 589 static GstStaticCaps html_caps = GST_STATIC_CAPS ("text/html");
 590
 591 #define HTML_CAPS gst_static_caps_get (&html_caps)
 592
 593 static void
 594 html_type_find (GstTypeFind * tf, gpointer unused)
 595 {
 596   const gchar *d, *data;
 597
 598   data = (const gchar *) gst_type_find_peek (tf, 0, 16);
 599   if (!data)
 600     return;
 601
 602   if (!g_ascii_strncasecmp (data, "<!DOCTYPE HTML", 14)) {
 603     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, HTML_CAPS);
 604   } else if (xml_check_first_element (tf, "html", 4, FALSE)) {
 605     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, HTML_CAPS);
 606   } else if ((d = memchr (data, '<', 16))) {
 607     data = (const gchar *) gst_type_find_peek (tf, d - data, 6);
 608     if (data && g_ascii_strncasecmp (data, "<html>", 6) == 0) {
 609       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, HTML_CAPS);
 610     }
 611   }
 612 }
 613
 614 /*** audio/midi ***/
 615
 616 static GstStaticCaps mid_caps = GST_STATIC_CAPS ("audio/midi");
 617
 618 #define MID_CAPS gst_static_caps_get(&mid_caps)
 619 static void
 620 mid_type_find (GstTypeFind * tf, gpointer unused)
 621 {
 622   const guint8 *data = gst_type_find_peek (tf, 0, 4);
 623
 624   /* http://jedi.ks.uiuc.edu/~johns/links/music/midifile.html */
 625   if (data && data[0] == 'M' && data[1] == 'T' && data[2] == 'h'
 626       && data[3] == 'd')
 627     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MID_CAPS);
 628 }
 629
 630 /*** audio/mobile-xmf ***/
 631
 632 static GstStaticCaps mxmf_caps = GST_STATIC_CAPS ("audio/mobile-xmf");
 633
 634 #define MXMF_CAPS gst_static_caps_get(&mxmf_caps)
 635 static void
 636 mxmf_type_find (GstTypeFind * tf, gpointer unused)
 637 {
 638   const guint8 *data = NULL;
 639
 640   /* Search FileId "XMF_" 4 bytes */
 641   data = gst_type_find_peek (tf, 0, 4);
 642   if (data && data[0] == 'X' && data[1] == 'M' && data[2] == 'F'
 643       && data[3] == '_') {
 644     /* Search Format version "2.00" 4 bytes */
 645     data = gst_type_find_peek (tf, 4, 4);
 646     if (data && data[0] == '2' && data[1] == '.' && data[2] == '0'
 647         && data[3] == '0') {
 648       /* Search TypeId 2     1 byte */
 649       data = gst_type_find_peek (tf, 11, 1);
 650       if (data && data[0] == 2) {
 651         gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MXMF_CAPS);
 652       }
 653     }
 654   }
 655 }
 656
 657
 658 /*** video/x-fli ***/
 659
 660 static GstStaticCaps flx_caps = GST_STATIC_CAPS ("video/x-fli");
 661
 662 #define FLX_CAPS gst_static_caps_get(&flx_caps)
 663 static void
 664 flx_type_find (GstTypeFind * tf, gpointer unused)
 665 {
 666   const guint8 *data = gst_type_find_peek (tf, 0, 134);
 667
 668   if (data) {
 669     /* check magic and the frame type of the first frame */
 670     if ((data[4] == 0x11 || data[4] == 0x12 ||
 671             data[4] == 0x30 || data[4] == 0x44) &&
 672         data[5] == 0xaf &&
 673         ((data[132] == 0x00 || data[132] == 0xfa) && data[133] == 0xf1)) {
 674       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, FLX_CAPS);
 675     }
 676     return;
 677   }
 678   data = gst_type_find_peek (tf, 0, 6);
 679   if (data) {
 680     /* check magic only */
 681     if ((data[4] == 0x11 || data[4] == 0x12 ||
 682             data[4] == 0x30 || data[4] == 0x44) && data[5] == 0xaf) {
 683       gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, FLX_CAPS);
 684     }
 685     return;
 686   }
 687 }
 688
 689 /*** application/x-id3 ***/
 690
 691 static GstStaticCaps id3_caps = GST_STATIC_CAPS ("application/x-id3");
 692
 693 #define ID3_CAPS gst_static_caps_get(&id3_caps)
 694 static void
 695 id3v2_type_find (GstTypeFind * tf, gpointer unused)
 696 {
 697   const guint8 *data = gst_type_find_peek (tf, 0, 10);
 698
 699   if (data && memcmp (data, "ID3", 3) == 0 &&
 700       data[3] != 0xFF && data[4] != 0xFF &&
 701       (data[6] & 0x80) == 0 && (data[7] & 0x80) == 0 &&
 702       (data[8] & 0x80) == 0 && (data[9] & 0x80) == 0) {
 703     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, ID3_CAPS);
 704   }
 705 }
 706
 707 static void
 708 id3v1_type_find (GstTypeFind * tf, gpointer unused)
 709 {
 710   const guint8 *data = gst_type_find_peek (tf, -128, 3);
 711
 712   if (data && memcmp (data, "TAG", 3) == 0) {
 713     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, ID3_CAPS);
 714   }
 715 }
 716
 717 /*** application/x-ape ***/
 718
 719 static GstStaticCaps apetag_caps = GST_STATIC_CAPS ("application/x-apetag");
 720
 721 #define APETAG_CAPS gst_static_caps_get(&apetag_caps)
 722 static void
 723 apetag_type_find (GstTypeFind * tf, gpointer unused)
 724 {
 725   const guint8 *data;
 726
 727   /* APEv1/2 at start of file */
 728   data = gst_type_find_peek (tf, 0, 8);
 729   if (data && !memcmp (data, "APETAGEX", 8)) {
 730     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, APETAG_CAPS);
 731     return;
 732   }
 733
 734   /* APEv1/2 at end of file */
 735   data = gst_type_find_peek (tf, -32, 8);
 736   if (data && !memcmp (data, "APETAGEX", 8)) {
 737     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, APETAG_CAPS);
 738     return;
 739   }
 740 }
 741
 742 /*** audio/x-ttafile ***/
 743
 744 static GstStaticCaps tta_caps = GST_STATIC_CAPS ("audio/x-ttafile");
 745
 746 #define TTA_CAPS gst_static_caps_get(&tta_caps)
 747 static void
 748 tta_type_find (GstTypeFind * tf, gpointer unused)
 749 {
 750   const guint8 *data = gst_type_find_peek (tf, 0, 3);
 751
 752   if (data) {
 753     if (memcmp (data, "TTA", 3) == 0) {
 754       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, TTA_CAPS);
 755       return;
 756     }
 757   }
 758 }
 759
 760 /*** audio/x-flac ***/
 761 static GstStaticCaps flac_caps = GST_STATIC_CAPS ("audio/x-flac");
 762
 763 #define FLAC_CAPS (gst_static_caps_get(&flac_caps))
 764
 765 static void
 766 flac_type_find (GstTypeFind * tf, gpointer unused)
 767 {
 768   DataScanCtx c = { 0, NULL, 0 };
 769
 770   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 4)))
 771     return;
 772
 773   /* standard flac (also old/broken flac-in-ogg with an initial 4-byte marker
 774    * packet and without the usual packet framing) */
 775   if (memcmp (c.data, "fLaC", 4) == 0) {
 776     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, FLAC_CAPS);
 777     return;
 778   }
 779
 780   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 6)))
 781     return;
 782
 783   /* flac-in-ogg, see http://flac.sourceforge.net/ogg_mapping.html */
 784   if (memcmp (c.data, "\177FLAC\001", 6) == 0) {
 785     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, FLAC_CAPS);
 786     return;
 787   }
 788
 789 /* disabled because it happily typefinds /dev/urandom as audio/x-flac, and
 790  * because I yet have to see header-less flac in the wild */
 791 #if 0
 792   /* flac without headers (subset format) */
 793   /* 64K should be enough */
 794   while (c.offset < (64 * 1024)) {
 795     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 4)))
 796       break;
 797
 798     /* look for frame header,
 799      * http://flac.sourceforge.net/format.html#frame_header
 800      */
 801     if (c.data[0] == 0xff && (c.data[1] >> 2) == 0x3e) {
 802       /* bit 15 in the header must be 0 */
 803       if (((c.data[1] >> 1) & 0x01) == 0x01)
 804         goto advance;
 805
 806       /* blocksize must be != 0x00 */
 807       if ((c.data[2] >> 4) == 0x00)
 808         goto advance;
 809
 810       /* samplerate must be != 0x0f */
 811       if ((c.data[2] & 0x0f) == 0x0f)
 812         goto advance;
 813       /* also 0 is invalid, as it means get the info from the header and we
 814        * don't have headers if we are here */
 815       if ((c.data[2] & 0x0f) == 0x00)
 816         goto advance;
 817
 818       /* channel assignment must be < 11 */
 819       if ((c.data[3] >> 4) >= 11)
 820         goto advance;
 821
 822       /* sample size must be != 0x07 and != 0x05 */
 823       if (((c.data[3] >> 1) & 0x07) == 0x07)
 824         goto advance;
 825       if (((c.data[3] >> 1) & 0x07) == 0x05)
 826         goto advance;
 827       /* also 0 is invalid, as it means get the info from the header and we
 828        * don't have headers if we are here */
 829       if (((c.data[3] >> 1) & 0x07) == 0x00)
 830         goto advance;
 831
 832       /* next bit must be 0 */
 833       if ((c.data[3] & 0x01) == 0x01)
 834         goto advance;
 835
 836       /* FIXME: shouldn't we include the crc check ? */
 837
 838       GST_DEBUG ("Found flac without headers at %d", (gint) c.offset);
 839       gst_type_find_suggest (tf, GST_TYPE_FIND_POSSIBLE, FLAC_CAPS);
 840       return;
 841     }
 842   advance:
 843     data_scan_ctx_advance (tf, &c, 1);
 844   }
 845 #endif
 846 }
 847
 848 /* TODO: we could probably make a generic function for this.. */
 849 static gint
 850 aac_type_find_scan_loas_frames_ep (GstTypeFind * tf, DataScanCtx * scan_ctx,
 851     gint max_frames)
 852 {
 853   DataScanCtx c = *scan_ctx;
 854   guint16 snc;
 855   guint len;
 856   gint count = 0;
 857
 858   do {
 859     if (!data_scan_ctx_ensure_data (tf, &c, 5))
 860       break;
 861
 862     /* EPAudioSyncStream */
 863     len = ((c.data[2] & 0x0f) << 9) | (c.data[3] << 1) |
 864         ((c.data[4] & 0x80) >> 7);
 865
 866     if (len == 0 || !data_scan_ctx_ensure_data (tf, &c, len + 2)) {
 867       GST_DEBUG ("Wrong sync or next frame not within reach, len=%u", len);
 868       break;
 869     }
 870
 871     /* check length of frame  */
 872     snc = GST_READ_UINT16_BE (c.data + len);
 873     if (snc != 0x4de1) {
 874       GST_DEBUG ("No sync found at 0x%" G_GINT64_MODIFIER "x", c.offset + len);
 875       break;
 876     }
 877
 878     ++count;
 879
 880     GST_DEBUG ("Found LOAS syncword #%d at offset 0x%" G_GINT64_MODIFIER "x, "
 881         "framelen %u", count, c.offset, len);
 882
 883     data_scan_ctx_advance (tf, &c, len);
 884   } while (count < max_frames && (c.offset - scan_ctx->offset) < 64 * 1024);
 885
 886   GST_DEBUG ("found %d consecutive frames", count);
 887   return count;
 888 }
 889
 890 static gint
 891 aac_type_find_scan_loas_frames (GstTypeFind * tf, DataScanCtx * scan_ctx,
 892     gint max_frames)
 893 {
 894   DataScanCtx c = *scan_ctx;
 895   guint16 snc;
 896   guint len;
 897   gint count = 0;
 898
 899   do {
 900     if (!data_scan_ctx_ensure_data (tf, &c, 3))
 901       break;
 902
 903     /* AudioSyncStream */
 904     len = ((c.data[1] & 0x1f) << 8) | c.data[2];
 905     /* add size of sync stream header */
 906     len += 3;
 907
 908     if (len == 0 || !data_scan_ctx_ensure_data (tf, &c, len)) {
 909       GST_DEBUG ("Wrong sync or next frame not within reach, len=%u", len);
 910       break;
 911     }
 912
 913     /* check length of frame  */
 914     snc = GST_READ_UINT16_BE (c.data + len);
 915     if ((snc & 0xffe0) != 0x56e0) {
 916       GST_DEBUG ("No sync found at 0x%" G_GINT64_MODIFIER "x", c.offset + len);
 917       break;
 918     }
 919
 920     ++count;
 921
 922     GST_DEBUG ("Found LOAS syncword #%d at offset 0x%" G_GINT64_MODIFIER "x, "
 923         "framelen %u", count, c.offset, len);
 924
 925     data_scan_ctx_advance (tf, &c, len);
 926   } while (count < max_frames && (c.offset - scan_ctx->offset) < 64 * 1024);
 927
 928   GST_DEBUG ("found %d consecutive frames", count);
 929   return count;
 930 }
 931
 932 /*** audio/mpeg version 2, 4 ***/
 933
 934 static GstStaticCaps aac_caps = GST_STATIC_CAPS ("audio/mpeg, "
 935     "mpegversion = (int) { 2, 4 }, framed = (bool) false");
 936 #define AAC_CAPS (gst_static_caps_get(&aac_caps))
 937 #define AAC_AMOUNT (4096)
 938 static void
 939 aac_type_find (GstTypeFind * tf, gpointer unused)
 940 {
 941   DataScanCtx c = { 0, NULL, 0 };
 942   GstTypeFindProbability best_probability = GST_TYPE_FIND_NONE;
 943   GstCaps *best_caps = NULL;
 944   guint best_count = 0;
 945
 946   while (c.offset < AAC_AMOUNT) {
 947     guint snc, len;
 948
 949     /* detect adts header or adif header.
 950      * The ADIF header is 4 bytes, that should be OK. The ADTS header, on
 951      * the other hand, is 14 bits only, so we require one valid frame with
 952      * again a valid syncpoint on the next one (28 bits) for certainty. We
 953      * require 4 kB, which is quite a lot, since frames are generally 200-400
 954      * bytes.
 955      * LOAS has 2 possible syncwords, which are 11 bits and 16 bits long.
 956      * The following stream syntax depends on which one is found.
 957      */
 958     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 6)))
 959       break;
 960
 961     snc = GST_READ_UINT16_BE (c.data);
 962     if (G_UNLIKELY ((snc & 0xfff6) == 0xfff0)) {
 963       /* ADTS header - find frame length */
 964       GST_DEBUG ("Found one ADTS syncpoint at offset 0x%" G_GINT64_MODIFIER
 965           "x, tracing next...", c.offset);
 966       len = ((c.data[3] & 0x03) << 11) |
 967           (c.data[4] << 3) | ((c.data[5] & 0xe0) >> 5);
 968
 969       if (len == 0 || !data_scan_ctx_ensure_data (tf, &c, len + 2)) {
 970         GST_DEBUG ("Wrong sync or next frame not within reach, len=%u", len);
 971         goto next;
 972       }
 973
 974       /* check if there's a second ADTS frame */
 975       snc = GST_READ_UINT16_BE (c.data + len);
 976       if ((snc & 0xfff6) == 0xfff0) {
 977         GstCaps *caps;
 978         guint mpegversion, sample_freq_idx, channel_config, profile_idx, rate;
 979         guint8 audio_config[2];
 980
 981         mpegversion = (c.data[1] & 0x08) ? 2 : 4;
 982         profile_idx = c.data[2] >> 6;
 983         sample_freq_idx = ((c.data[2] & 0x3c) >> 2);
 984         channel_config = ((c.data[2] & 0x01) << 2) + (c.data[3] >> 6);
 985
 986         GST_DEBUG ("Found second ADTS-%d syncpoint at offset 0x%"
 987             G_GINT64_MODIFIER "x, framelen %u", mpegversion, c.offset, len);
 988
 989         /* 0xd and 0xe are reserved. 0xf means the sample frequency is directly
 990          * specified in the header, but that's not allowed for ADTS */
 991         if (sample_freq_idx > 0xc) {
 992           GST_DEBUG ("Unexpected sample frequency index %d or wrong sync",
 993               sample_freq_idx);
 994           goto next;
 995         }
 996
 997         rate = gst_codec_utils_aac_get_sample_rate_from_index (sample_freq_idx);
 998         GST_LOG ("ADTS: profile=%u, rate=%u", profile_idx, rate);
 999
1000         /* The ADTS frame header is slightly different from the
1001          * AudioSpecificConfig defined for the MPEG-4 container, so we just
1002          * construct enough of it for getting the level here. */
1003         /* ADTS counts profiles from 0 instead of 1 to save bits */
1004         audio_config[0] = (profile_idx + 1) << 3;
1005         audio_config[0] |= (sample_freq_idx >> 1) & 0x7;
1006         audio_config[1] = (sample_freq_idx & 0x1) << 7;
1007         audio_config[1] |= (channel_config & 0xf) << 3;
1008
1009         caps = gst_caps_new_simple ("audio/mpeg",
1010             "framed", G_TYPE_BOOLEAN, FALSE,
1011             "mpegversion", G_TYPE_INT, mpegversion,
1012             "stream-format", G_TYPE_STRING, "adts", NULL);
1013
1014         gst_codec_utils_aac_caps_set_level_and_profile (caps, audio_config, 2);
1015
1016         /* add rate and number of channels if we can */
1017         if (channel_config != 0 && channel_config <= 7) {
1018           const guint channels_map[] = { 0, 1, 2, 3, 4, 5, 6, 8 };
1019
1020           gst_caps_set_simple (caps, "channels", G_TYPE_INT,
1021               channels_map[channel_config], "rate", G_TYPE_INT, rate, NULL);
1022         }
1023
1024         gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, caps);
1025         gst_caps_unref (caps);
1026         break;
1027       }
1028
1029       GST_DEBUG ("No next frame found... (should have been at 0x%x)", len);
1030     } else if (G_UNLIKELY ((snc & 0xffe0) == 0x56e0 || snc == 0x4de1)) {
1031       gint count;
1032
1033       /* LOAS frame */
1034       GST_INFO ("Possible LOAS syncword at offset 0x%" G_GINT64_MODIFIER
1035           "x, scanning for more frames...", c.offset);
1036
1037       if (snc == 0x4de1)
1038         count = aac_type_find_scan_loas_frames_ep (tf, &c, 20);
1039       else
1040         count = aac_type_find_scan_loas_frames (tf, &c, 20);
1041
1042       if (count >= 3 && count > best_count) {
1043         gst_caps_replace (&best_caps, NULL);
1044         best_caps = gst_caps_new_simple ("audio/mpeg",
1045             "framed", G_TYPE_BOOLEAN, FALSE,
1046             "mpegversion", G_TYPE_INT, 4,
1047             "stream-format", G_TYPE_STRING, "loas", NULL);
1048         best_count = count;
1049         best_probability = GST_TYPE_FIND_POSSIBLE - 10 + count * 3;
1050         if (best_probability >= GST_TYPE_FIND_LIKELY)
1051           break;
1052       }
1053     } else if (!memcmp (c.data, "ADIF", 4)) {
1054       /* ADIF header */
1055       gst_type_find_suggest_simple (tf, GST_TYPE_FIND_LIKELY, "audio/mpeg",
1056           "framed", G_TYPE_BOOLEAN, FALSE, "mpegversion", G_TYPE_INT, 4,
1057           "stream-format", G_TYPE_STRING, "adif", NULL);
1058       break;
1059     }
1060
1061   next:
1062
1063     data_scan_ctx_advance (tf, &c, 1);
1064   }
1065
1066   if (best_probability > GST_TYPE_FIND_NONE) {
1067     gst_type_find_suggest (tf, best_probability, best_caps);
1068     gst_caps_unref (best_caps);
1069   }
1070 }
1071
1072 /*** audio/mpeg version 1 ***/
1073
1074 /*
1075  * The chance that random data is identified as a valid mp3 header is 63 / 2^18
1076  * (0.024%) per try. This makes the function for calculating false positives
1077  *   1 - (1 - ((63 / 2 ^18) ^ GST_MP3_TYPEFIND_MIN_HEADERS)) ^ buffersize)
1078  * This has the following probabilities of false positives:
1079  * datasize               MIN_HEADERS
1080  * (bytes)      1       2       3       4
1081  * 4096         62.6%    0.02%   0%      0%
1082  * 16384        98%      0.09%   0%      0%
1083  * 1 MiB       100%      5.88%   0%      0%
1084  * 1 GiB       100%    100%      1.44%   0%
1085  * 1 TiB       100%    100%    100%      0.35%
1086  * This means that the current choice (3 headers by most of the time 4096 byte
1087  * buffers is pretty safe for now.
1088  *
1089  * The max. size of each frame is 1440 bytes, which means that for N frames to
1090  * be detected, we need 1440 * GST_MP3_TYPEFIND_MIN_HEADERS + 3 bytes of data.
1091  * Assuming we step into the stream right after the frame header, this
1092  * means we need 1440 * (GST_MP3_TYPEFIND_MIN_HEADERS + 1) - 1 + 3 bytes
1093  * of data (5762) to always detect any mp3.
1094  */
1095
1096 static const guint mp3types_bitrates[2][3][16] =
1097     { {{0, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 352, 384, 416, 448,},
1098     {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384,},
1099     {0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320,}},
1100 {{0, 32, 48, 56, 64, 80, 96, 112, 128, 144, 160, 176, 192, 224, 256,},
1101     {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160,},
1102     {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160,}},
1103 };
1104
1105 static const guint mp3types_freqs[3][3] = { {11025, 12000, 8000},
1106 {22050, 24000, 16000},
1107 {44100, 48000, 32000}
1108 };
1109
1110 static inline guint
1111 mp3_type_frame_length_from_header (guint32 header, guint * put_layer,
1112     guint * put_channels, guint * put_bitrate, guint * put_samplerate,
1113     gboolean * may_be_free_format, gint possible_free_framelen)
1114 {
1115   guint bitrate, layer, length, mode, samplerate, version, channels;
1116
1117   if ((header & 0xffe00000) != 0xffe00000)
1118     return 0;
1119
1120   /* we don't need extension, copyright, original or
1121    * emphasis for the frame length */
1122   header >>= 6;
1123
1124   /* mode */
1125   mode = header & 0x3;
1126   header >>= 3;
1127
1128   /* padding */
1129   length = header & 0x1;
1130   header >>= 1;
1131
1132   /* sampling frequency */
1133   samplerate = header & 0x3;
1134   if (samplerate == 3)
1135     return 0;
1136   header >>= 2;
1137
1138   /* bitrate index */
1139   bitrate = header & 0xF;
1140   if (bitrate == 0 && possible_free_framelen == -1) {
1141     GST_LOG ("Possibly a free format mp3 - signaling");
1142     *may_be_free_format = TRUE;
1143   }
1144   if (bitrate == 15 || (bitrate == 0 && possible_free_framelen == -1))
1145     return 0;
1146
1147   /* ignore error correction, too */
1148   header >>= 5;
1149
1150   /* layer */
1151   layer = 4 - (header & 0x3);
1152   if (layer == 4)
1153     return 0;
1154   header >>= 2;
1155
1156   /* version 0=MPEG2.5; 2=MPEG2; 3=MPEG1 */
1157   version = header & 0x3;
1158   if (version == 1)
1159     return 0;
1160
1161   /* lookup */
1162   channels = (mode == 3) ? 1 : 2;
1163   samplerate = mp3types_freqs[version > 0 ? version - 1 : 0][samplerate];
1164   if (bitrate == 0) {
1165     if (layer == 1) {
1166       length *= 4;
1167       length += possible_free_framelen;
1168       bitrate = length * samplerate / 48000;
1169     } else {
1170       length += possible_free_framelen;
1171       bitrate = length * samplerate /
1172           ((layer == 3 && version != 3) ? 72000 : 144000);
1173     }
1174   } else {
1175     /* calculating */
1176     bitrate = mp3types_bitrates[version == 3 ? 0 : 1][layer - 1][bitrate];
1177     if (layer == 1) {
1178       length = ((12000 * bitrate / samplerate) + length) * 4;
1179     } else {
1180       length += ((layer == 3
1181               && version != 3) ? 72000 : 144000) * bitrate / samplerate;
1182     }
1183   }
1184
1185   GST_LOG ("mp3typefind: calculated mp3 frame length of %u bytes", length);
1186   GST_LOG
1187       ("mp3typefind: samplerate = %u - bitrate = %u - layer = %u - version = %u"
1188       " - channels = %u", samplerate, bitrate, layer, version, channels);
1189
1190   if (put_layer)
1191     *put_layer = layer;
1192   if (put_channels)
1193     *put_channels = channels;
1194   if (put_bitrate)
1195     *put_bitrate = bitrate;
1196   if (put_samplerate)
1197     *put_samplerate = samplerate;
1198
1199   return length;
1200 }
1201
1202
1203 static GstStaticCaps mp3_caps = GST_STATIC_CAPS ("audio/mpeg, "
1204     "mpegversion = (int) 1, layer = (int) [ 1, 3 ]");
1205 #define MP3_CAPS (gst_static_caps_get(&mp3_caps))
1206 /*
1207  * random values for typefinding
1208  * if no more data is available, we will return a probability of
1209  * (found_headers/TRY_HEADERS) * (MAXIMUM * (TRY_SYNC - bytes_skipped)
1210  *        / TRY_SYNC)
1211  * if found_headers >= MIN_HEADERS
1212  */
1213 #define GST_MP3_TYPEFIND_MIN_HEADERS (2)
1214 #define GST_MP3_TYPEFIND_TRY_HEADERS (5)
1215 #define GST_MP3_TYPEFIND_TRY_SYNC (GST_TYPE_FIND_MAXIMUM * 100) /* 10kB */
1216 #define GST_MP3_TYPEFIND_SYNC_SIZE (2048)
1217 #define GST_MP3_WRONG_HEADER (10)
1218
1219 static void
1220 mp3_type_find_at_offset (GstTypeFind * tf, guint64 start_off,
1221     guint * found_layer, GstTypeFindProbability * found_prob)
1222 {
1223   const guint8 *data = NULL;
1224   const guint8 *data_end = NULL;
1225   guint size;
1226   guint64 skipped;
1227   gint last_free_offset = -1;
1228   gint last_free_framelen = -1;
1229   gboolean headerstart = TRUE;
1230
1231   *found_layer = 0;
1232   *found_prob = 0;
1233
1234   size = 0;
1235   skipped = 0;
1236   while (skipped < GST_MP3_TYPEFIND_TRY_SYNC) {
1237     if (size <= 0) {
1238       size = GST_MP3_TYPEFIND_SYNC_SIZE * 2;
1239       do {
1240         size /= 2;
1241         data = gst_type_find_peek (tf, skipped + start_off, size);
1242       } while (size > 10 && !data);
1243       if (!data)
1244         break;
1245       data_end = data + size;
1246     }
1247     if (*data == 0xFF) {
1248       const guint8 *head_data = NULL;
1249       guint layer = 0, bitrate, samplerate, channels;
1250       guint found = 0;          /* number of valid headers found */
1251       guint64 offset = skipped;
1252       gboolean changed = FALSE;
1253
1254       while (found < GST_MP3_TYPEFIND_TRY_HEADERS) {
1255         guint32 head;
1256         guint length;
1257         guint prev_layer = 0;
1258         guint prev_channels = 0, prev_samplerate = 0;
1259         gboolean free = FALSE;
1260
1261         if ((gint64) (offset - skipped + 4) >= 0 &&
1262             data + offset - skipped + 4 < data_end) {
1263           head_data = data + offset - skipped;
1264         } else {
1265           head_data = gst_type_find_peek (tf, offset + start_off, 4);
1266         }
1267         if (!head_data)
1268           break;
1269         head = GST_READ_UINT32_BE (head_data);
1270         if (!(length = mp3_type_frame_length_from_header (head, &layer,
1271                     &channels, &bitrate, &samplerate, &free,
1272                     last_free_framelen))) {
1273           if (free) {
1274             if (last_free_offset == -1)
1275               last_free_offset = offset;
1276             else {
1277               last_free_framelen = offset - last_free_offset;
1278               offset = last_free_offset;
1279               continue;
1280             }
1281           } else {
1282             last_free_framelen = -1;
1283           }
1284
1285           /* Mark the fact that we didn't find a valid header at the beginning */
1286           if (found == 0)
1287             headerstart = FALSE;
1288
1289           GST_LOG ("%d. header at offset %" G_GUINT64_FORMAT
1290               " (0x%" G_GINT64_MODIFIER "x) was not an mp3 header "
1291               "(possibly-free: %s)", found + 1, start_off + offset,
1292               start_off + offset, free ? "yes" : "no");
1293           break;
1294         }
1295         if ((prev_layer && prev_layer != layer) ||
1296             /* (prev_bitrate && prev_bitrate != bitrate) || <-- VBR */
1297             (prev_samplerate && prev_samplerate != samplerate) ||
1298             (prev_channels && prev_channels != channels)) {
1299           /* this means an invalid property, or a change, which might mean
1300            * that this is not a mp3 but just a random bytestream. It could
1301            * be a freaking funky encoded mp3 though. We'll just not count
1302            * this header*/
1303           if (prev_layer)
1304             changed = TRUE;
1305           prev_layer = layer;
1306           prev_channels = channels;
1307           prev_samplerate = samplerate;
1308         } else {
1309           found++;
1310           GST_LOG ("found %d. header at offset %" G_GUINT64_FORMAT " (0x%"
1311               G_GINT64_MODIFIER "X)", found, start_off + offset,
1312               start_off + offset);
1313         }
1314         offset += length;
1315       }
1316       g_assert (found <= GST_MP3_TYPEFIND_TRY_HEADERS);
1317       if (head_data == NULL &&
1318           gst_type_find_peek (tf, offset + start_off - 1, 1) == NULL)
1319         /* Incomplete last frame - don't count it. */
1320         found--;
1321       if (found == GST_MP3_TYPEFIND_TRY_HEADERS ||
1322           (found >= GST_MP3_TYPEFIND_MIN_HEADERS && head_data == NULL)) {
1323         /* we can make a valid guess */
1324         guint probability = found * GST_TYPE_FIND_MAXIMUM *
1325             (GST_MP3_TYPEFIND_TRY_SYNC - skipped) /
1326             GST_MP3_TYPEFIND_TRY_HEADERS / GST_MP3_TYPEFIND_TRY_SYNC;
1327
1328         if (!headerstart
1329             && probability > (GST_TYPE_FIND_MINIMUM + GST_MP3_WRONG_HEADER))
1330           probability -= GST_MP3_WRONG_HEADER;
1331         if (probability < GST_TYPE_FIND_MINIMUM)
1332           probability = GST_TYPE_FIND_MINIMUM;
1333         if (start_off > 0)
1334           probability /= 2;
1335         if (!changed)
1336           probability = (probability + GST_TYPE_FIND_MAXIMUM) / 2;
1337
1338         GST_INFO
1339             ("audio/mpeg calculated %u  =  %u  *  %u / %u  *  (%u - %"
1340             G_GUINT64_FORMAT ") / %u", probability, GST_TYPE_FIND_MAXIMUM,
1341             found, GST_MP3_TYPEFIND_TRY_HEADERS, GST_MP3_TYPEFIND_TRY_SYNC,
1342             (guint64) skipped, GST_MP3_TYPEFIND_TRY_SYNC);
1343         /* make sure we're not id3 tagged */
1344         head_data = gst_type_find_peek (tf, -128, 3);
1345         if (head_data && (memcmp (head_data, "TAG", 3) == 0)) {
1346           probability = 0;
1347         }
1348         g_assert (probability <= GST_TYPE_FIND_MAXIMUM);
1349
1350         *found_prob = probability;
1351         if (probability > 0)
1352           *found_layer = layer;
1353         return;
1354       }
1355     }
1356     data++;
1357     skipped++;
1358     size--;
1359   }
1360 }
1361
1362 static void
1363 mp3_type_find (GstTypeFind * tf, gpointer unused)
1364 {
1365   GstTypeFindProbability prob, mid_prob;
1366   const guint8 *data;
1367   guint layer, mid_layer;
1368   guint64 length;
1369
1370   mp3_type_find_at_offset (tf, 0, &layer, &prob);
1371   length = gst_type_find_get_length (tf);
1372
1373   if (length == 0 || length == (guint64) - 1) {
1374     if (prob != 0)
1375       goto suggest;
1376     return;
1377   }
1378
1379   /* if we're pretty certain already, skip the additional check */
1380   if (prob >= GST_TYPE_FIND_LIKELY)
1381     goto suggest;
1382
1383   mp3_type_find_at_offset (tf, length / 2, &mid_layer, &mid_prob);
1384
1385   if (mid_prob > 0) {
1386     if (prob == 0) {
1387       GST_LOG ("detected audio/mpeg only in the middle (p=%u)", mid_prob);
1388       layer = mid_layer;
1389       prob = mid_prob;
1390       goto suggest;
1391     }
1392
1393     if (layer != mid_layer) {
1394       GST_WARNING ("audio/mpeg layer discrepancy: %u vs. %u", layer, mid_layer);
1395       return;                   /* FIXME: or should we just go with the one in the middle? */
1396     }
1397
1398     /* detected mpeg audio both in middle of the file and at the start */
1399     prob = (prob + mid_prob) / 2;
1400     goto suggest;
1401   }
1402
1403   /* let's see if there's a valid header right at the start */
1404   data = gst_type_find_peek (tf, 0, 4); /* use min. frame size? */
1405   if (data && mp3_type_frame_length_from_header (GST_READ_UINT32_BE (data),
1406           &layer, NULL, NULL, NULL, NULL, 0) != 0) {
1407     if (prob == 0)
1408       prob = GST_TYPE_FIND_POSSIBLE - 10;
1409     else
1410       prob = MAX (GST_TYPE_FIND_POSSIBLE - 10, prob + 10);
1411   }
1412
1413   if (prob > 0)
1414     goto suggest;
1415
1416   return;
1417
1418 suggest:
1419   {
1420     g_return_if_fail (layer >= 1 && layer <= 3);
1421
1422     gst_type_find_suggest_simple (tf, prob, "audio/mpeg",
1423         "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, layer,
1424         "parsed", G_TYPE_BOOLEAN, FALSE, NULL);
1425   }
1426 }
1427
1428 /*** audio/x-musepack ***/
1429
1430 static GstStaticCaps musepack_caps =
1431 GST_STATIC_CAPS ("audio/x-musepack, streamversion= (int) { 7, 8 }");
1432
1433 #define MUSEPACK_CAPS (gst_static_caps_get(&musepack_caps))
1434 static void
1435 musepack_type_find (GstTypeFind * tf, gpointer unused)
1436 {
1437   const guint8 *data = gst_type_find_peek (tf, 0, 4);
1438   GstTypeFindProbability prop = GST_TYPE_FIND_MINIMUM;
1439   gint streamversion = -1;
1440
1441   if (data && memcmp (data, "MP+", 3) == 0) {
1442     streamversion = 7;
1443     if ((data[3] & 0x7f) == 7) {
1444       prop = GST_TYPE_FIND_MAXIMUM;
1445     } else {
1446       prop = GST_TYPE_FIND_LIKELY + 10;
1447     }
1448   } else if (data && memcmp (data, "MPCK", 4) == 0) {
1449     streamversion = 8;
1450     prop = GST_TYPE_FIND_MAXIMUM;
1451   }
1452
1453   if (streamversion != -1) {
1454     gst_type_find_suggest_simple (tf, prop, "audio/x-musepack",
1455         "streamversion", G_TYPE_INT, streamversion, NULL);
1456   }
1457 }
1458
1459 /*** audio/x-ac3 ***/
1460 /* FIXME 0.11: should be audio/ac3, but isn't for backwards compatibility */
1461 static GstStaticCaps ac3_caps = GST_STATIC_CAPS ("audio/x-ac3");
1462
1463 #define AC3_CAPS (gst_static_caps_get(&ac3_caps))
1464
1465 static GstStaticCaps eac3_caps = GST_STATIC_CAPS ("audio/x-eac3");
1466
1467 #define EAC3_CAPS (gst_static_caps_get(&eac3_caps))
1468
1469 struct ac3_frmsize
1470 {
1471   unsigned short bit_rate;
1472   unsigned short frm_size[3];
1473 };
1474
1475 static const struct ac3_frmsize ac3_frmsizecod_tbl[] = {
1476   {32, {64, 69, 96}},
1477   {32, {64, 70, 96}},
1478   {40, {80, 87, 120}},
1479   {40, {80, 88, 120}},
1480   {48, {96, 104, 144}},
1481   {48, {96, 105, 144}},
1482   {56, {112, 121, 168}},
1483   {56, {112, 122, 168}},
1484   {64, {128, 139, 192}},
1485   {64, {128, 140, 192}},
1486   {80, {160, 174, 240}},
1487   {80, {160, 175, 240}},
1488   {96, {192, 208, 288}},
1489   {96, {192, 209, 288}},
1490   {112, {224, 243, 336}},
1491   {112, {224, 244, 336}},
1492   {128, {256, 278, 384}},
1493   {128, {256, 279, 384}},
1494   {160, {320, 348, 480}},
1495   {160, {320, 349, 480}},
1496   {192, {384, 417, 576}},
1497   {192, {384, 418, 576}},
1498   {224, {448, 487, 672}},
1499   {224, {448, 488, 672}},
1500   {256, {512, 557, 768}},
1501   {256, {512, 558, 768}},
1502   {320, {640, 696, 960}},
1503   {320, {640, 697, 960}},
1504   {384, {768, 835, 1152}},
1505   {384, {768, 836, 1152}},
1506   {448, {896, 975, 1344}},
1507   {448, {896, 976, 1344}},
1508   {512, {1024, 1114, 1536}},
1509   {512, {1024, 1115, 1536}},
1510   {576, {1152, 1253, 1728}},
1511   {576, {1152, 1254, 1728}},
1512   {640, {1280, 1393, 1920}},
1513   {640, {1280, 1394, 1920}}
1514 };
1515
1516 static void
1517 ac3_type_find (GstTypeFind * tf, gpointer unused)
1518 {
1519   DataScanCtx c = { 0, NULL, 0 };
1520
1521   /* Search for an ac3 frame; not necessarily right at the start, but give it
1522    * a lower probability if not found right at the start. Check that the
1523    * frame is followed by a second frame at the expected offset.
1524    * We could also check the two ac3 CRCs, but we don't do that right now */
1525   while (c.offset < 1024) {
1526     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 5)))
1527       break;
1528
1529     if (c.data[0] == 0x0b && c.data[1] == 0x77) {
1530       guint bsid = c.data[5] >> 3;
1531
1532       if (bsid <= 8) {
1533         /* ac3 */
1534         guint fscod = c.data[4] >> 6;
1535         guint frmsizecod = c.data[4] & 0x3f;
1536
1537         if (fscod < 3 && frmsizecod < 38) {
1538           DataScanCtx c_next = c;
1539           guint frame_size;
1540
1541           frame_size = ac3_frmsizecod_tbl[frmsizecod].frm_size[fscod];
1542           GST_LOG ("possible AC3 frame sync at offset %"
1543               G_GUINT64_FORMAT ", size=%u", c.offset, frame_size);
1544           if (data_scan_ctx_ensure_data (tf, &c_next, (frame_size * 2) + 5)) {
1545             data_scan_ctx_advance (tf, &c_next, frame_size * 2);
1546
1547             if (c_next.data[0] == 0x0b && c_next.data[1] == 0x77) {
1548               fscod = c_next.data[4] >> 6;
1549               frmsizecod = c_next.data[4] & 0x3f;
1550
1551               if (fscod < 3 && frmsizecod < 38) {
1552                 GstTypeFindProbability prob;
1553
1554                 GST_LOG ("found second AC3 frame (size=%u), looks good",
1555                     ac3_frmsizecod_tbl[frmsizecod].frm_size[fscod]);
1556                 if (c.offset == 0)
1557                   prob = GST_TYPE_FIND_MAXIMUM;
1558                 else
1559                   prob = GST_TYPE_FIND_NEARLY_CERTAIN;
1560
1561                 gst_type_find_suggest (tf, prob, AC3_CAPS);
1562                 return;
1563               }
1564             } else {
1565               GST_LOG ("no second AC3 frame found, false sync");
1566             }
1567           }
1568         }
1569       } else if (bsid <= 16 && bsid > 10) {
1570         /* eac3 */
1571         DataScanCtx c_next = c;
1572         guint frame_size;
1573
1574         frame_size = (((c.data[2] & 0x07) << 8) + c.data[3]) + 1;
1575         GST_LOG ("possible E-AC3 frame sync at offset %"
1576             G_GUINT64_FORMAT ", size=%u", c.offset, frame_size);
1577         if (data_scan_ctx_ensure_data (tf, &c_next, (frame_size * 2) + 5)) {
1578           data_scan_ctx_advance (tf, &c_next, frame_size * 2);
1579
1580           if (c_next.data[0] == 0x0b && c_next.data[1] == 0x77) {
1581             GstTypeFindProbability prob;
1582
1583             GST_LOG ("found second E-AC3 frame, looks good");
1584             if (c.offset == 0)
1585               prob = GST_TYPE_FIND_MAXIMUM;
1586             else
1587               prob = GST_TYPE_FIND_NEARLY_CERTAIN;
1588
1589             gst_type_find_suggest (tf, prob, EAC3_CAPS);
1590             return;
1591           } else {
1592             GST_LOG ("no second E-AC3 frame found, false sync");
1593           }
1594         }
1595       } else {
1596         GST_LOG ("invalid AC3 BSID: %u", bsid);
1597       }
1598     }
1599     data_scan_ctx_advance (tf, &c, 1);
1600   }
1601 }
1602
1603 /*** audio/x-dts ***/
1604 static GstStaticCaps dts_caps = GST_STATIC_CAPS ("audio/x-dts");
1605 #define DTS_CAPS (gst_static_caps_get (&dts_caps))
1606 #define DTS_MIN_FRAMESIZE 96
1607 #define DTS_MAX_FRAMESIZE 18725 /* 16384*16/14 */
1608
1609 static gboolean
1610 dts_parse_frame_header (DataScanCtx * c, guint * frame_size,
1611     guint * sample_rate, guint * channels, guint * depth, guint * endianness)
1612 {
1613   static const int sample_rates[16] = { 0, 8000, 16000, 32000, 0, 0, 11025,
1614     22050, 44100, 0, 0, 12000, 24000, 48000, 96000, 192000
1615   };
1616   static const guint8 channels_table[16] = { 1, 2, 2, 2, 2, 3, 3, 4, 4, 5,
1617     6, 6, 6, 7, 8, 8
1618   };
1619   guint16 hdr[8];
1620   guint32 marker;
1621   guint num_blocks, chans, lfe, i;
1622
1623   marker = GST_READ_UINT32_BE (c->data);
1624
1625   /* raw big endian or 14-bit big endian */
1626   if (marker == 0x7FFE8001 || marker == 0x1FFFE800) {
1627     *endianness = G_BIG_ENDIAN;
1628     for (i = 0; i < G_N_ELEMENTS (hdr); ++i)
1629       hdr[i] = GST_READ_UINT16_BE (c->data + (i * sizeof (guint16)));
1630   } else
1631     /* raw little endian or 14-bit little endian */
1632   if (marker == 0xFE7F0180 || marker == 0xFF1F00E8) {
1633     *endianness = G_LITTLE_ENDIAN;
1634     for (i = 0; i < G_N_ELEMENTS (hdr); ++i)
1635       hdr[i] = GST_READ_UINT16_LE (c->data + (i * sizeof (guint16)));
1636   } else {
1637     return FALSE;
1638   }
1639
1640   GST_LOG ("dts sync marker 0x%08x at offset %u", marker, (guint) c->offset);
1641
1642   /* 14-bit mode */
1643   if (marker == 0x1FFFE800 || marker == 0xFF1F00E8) {
1644     if ((hdr[2] & 0xFFF0) != 0x07F0)
1645       return FALSE;
1646     /* discard top 2 bits (2 void), shift in 2 */
1647     hdr[0] = (hdr[0] << 2) | ((hdr[1] >> 12) & 0x0003);
1648     /* discard top 4 bits (2 void, 2 shifted into hdr[0]), shift in 4 etc. */
1649     hdr[1] = (hdr[1] << 4) | ((hdr[2] >> 10) & 0x000F);
1650     hdr[2] = (hdr[2] << 6) | ((hdr[3] >> 8) & 0x003F);
1651     hdr[3] = (hdr[3] << 8) | ((hdr[4] >> 6) & 0x00FF);
1652     hdr[4] = (hdr[4] << 10) | ((hdr[5] >> 4) & 0x03FF);
1653     hdr[5] = (hdr[5] << 12) | ((hdr[6] >> 2) & 0x0FFF);
1654     hdr[6] = (hdr[6] << 14) | ((hdr[7] >> 0) & 0x3FFF);
1655     g_assert (hdr[0] == 0x7FFE && hdr[1] == 0x8001);
1656     *depth = 14;
1657   } else {
1658     *depth = 16;
1659   }
1660
1661   GST_LOG ("frame header: %04x%04x%04x%04x", hdr[2], hdr[3], hdr[4], hdr[5]);
1662
1663   num_blocks = (hdr[2] >> 2) & 0x7F;
1664   *frame_size = (((hdr[2] & 0x03) << 12) | (hdr[3] >> 4)) + 1;
1665   chans = ((hdr[3] & 0x0F) << 2) | (hdr[4] >> 14);
1666   *sample_rate = sample_rates[(hdr[4] >> 10) & 0x0F];
1667   lfe = (hdr[5] >> 9) & 0x03;
1668
1669   if (num_blocks < 5 || *frame_size < 96 || *sample_rate == 0)
1670     return FALSE;
1671
1672   if (marker == 0x1FFFE800 || marker == 0xFF1F00E8)
1673     *frame_size = (*frame_size * 16) / 14;      /* FIXME: round up? */
1674
1675   if (chans < G_N_ELEMENTS (channels_table))
1676     *channels = channels_table[chans] + ((lfe) ? 1 : 0);
1677   else
1678     *channels = 0;
1679
1680   return TRUE;
1681 }
1682
1683 static void
1684 dts_type_find (GstTypeFind * tf, gpointer unused)
1685 {
1686   DataScanCtx c = { 0, NULL, 0 };
1687
1688   /* Search for an dts frame; not necessarily right at the start, but give it
1689    * a lower probability if not found right at the start. Check that the
1690    * frame is followed by a second frame at the expected offset. */
1691   while (c.offset <= DTS_MAX_FRAMESIZE) {
1692     guint frame_size = 0, rate = 0, chans = 0, depth = 0, endianness = 0;
1693
1694     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, DTS_MIN_FRAMESIZE)))
1695       return;
1696
1697     if (G_UNLIKELY (dts_parse_frame_header (&c, &frame_size, &rate, &chans,
1698                 &depth, &endianness))) {
1699       GstTypeFindProbability prob;
1700       DataScanCtx next_c;
1701
1702       prob = (c.offset == 0) ? GST_TYPE_FIND_LIKELY : GST_TYPE_FIND_POSSIBLE;
1703
1704       /* check for second frame sync */
1705       next_c = c;
1706       data_scan_ctx_advance (tf, &next_c, frame_size);
1707       if (data_scan_ctx_ensure_data (tf, &next_c, 4)) {
1708         GST_LOG ("frame size: %u 0x%04x", frame_size, frame_size);
1709         GST_MEMDUMP ("second frame sync", next_c.data, 4);
1710         if (GST_READ_UINT32_BE (c.data) == GST_READ_UINT32_BE (next_c.data))
1711           prob = GST_TYPE_FIND_MAXIMUM;
1712       }
1713
1714       if (chans > 0) {
1715         gst_type_find_suggest_simple (tf, prob, "audio/x-dts",
1716             "rate", G_TYPE_INT, rate, "channels", G_TYPE_INT, chans,
1717             "depth", G_TYPE_INT, depth, "endianness", G_TYPE_INT, endianness,
1718             "framed", G_TYPE_BOOLEAN, FALSE, NULL);
1719       } else {
1720         gst_type_find_suggest_simple (tf, prob, "audio/x-dts",
1721             "rate", G_TYPE_INT, rate, "depth", G_TYPE_INT, depth,
1722             "endianness", G_TYPE_INT, endianness,
1723             "framed", G_TYPE_BOOLEAN, FALSE, NULL);
1724       }
1725
1726       return;
1727     }
1728
1729     data_scan_ctx_advance (tf, &c, 1);
1730   }
1731 }
1732
1733 /*** gsm ***/
1734
1735 /* can only be detected by using the extension, in which case we use the default
1736  * GSM properties */
1737 static GstStaticCaps gsm_caps =
1738 GST_STATIC_CAPS ("audio/x-gsm, rate=8000, channels=1");
1739
1740 #define GSM_CAPS (gst_static_caps_get(&gsm_caps))
1741
1742 /*** wavpack ***/
1743
1744 static GstStaticCaps wavpack_caps =
1745 GST_STATIC_CAPS ("audio/x-wavpack, framed = (boolean) false");
1746
1747 #define WAVPACK_CAPS (gst_static_caps_get(&wavpack_caps))
1748
1749 static GstStaticCaps wavpack_correction_caps =
1750 GST_STATIC_CAPS ("audio/x-wavpack-correction, framed = (boolean) false");
1751
1752 #define WAVPACK_CORRECTION_CAPS (gst_static_caps_get(&wavpack_correction_caps))
1753
1754 static void
1755 wavpack_type_find (GstTypeFind * tf, gpointer unused)
1756 {
1757   GstTypeFindProbability base_prob = GST_TYPE_FIND_POSSIBLE;
1758   guint64 offset;
1759   guint32 blocksize;
1760   const guint8 *data;
1761   guint count_wv, count_wvc;
1762
1763   data = gst_type_find_peek (tf, 0, 32);
1764   if (!data)
1765     return;
1766
1767   if (data[0] != 'w' || data[1] != 'v' || data[2] != 'p' || data[3] != 'k')
1768     return;
1769
1770   /* Note: wavpack blocks can be fairly large (easily 60-110k), possibly
1771    * larger than the max. limits imposed by certain typefinding elements
1772    * like id3demux or apedemux, so typefinding is most likely only going to
1773    * work in pull-mode */
1774   blocksize = GST_READ_UINT32_LE (data + 4);
1775   GST_LOG ("wavpack header, blocksize=0x%04x", blocksize);
1776   count_wv = 0;
1777   count_wvc = 0;
1778   offset = 32;
1779   while (offset < 8 + blocksize) {
1780     guint32 sublen;
1781
1782     /* get chunk header */
1783     GST_LOG ("peeking at chunk at offset 0x%04x", (guint) offset);
1784     data = gst_type_find_peek (tf, offset, 4);
1785     if (data == NULL)
1786       break;
1787     sublen = ((guint32) data[1]) << 1;
1788     if (data[0] & 0x80) {
1789       sublen |= (((guint32) data[2]) << 9) | (((guint32) data[3]) << 17);
1790       sublen += 1 + 3;          /* id + length */
1791     } else {
1792       sublen += 1 + 1;          /* id + length */
1793     }
1794     if (offset + sublen > 8 + blocksize) {
1795       GST_LOG ("chunk length too big (%u > %" G_GUINT64_FORMAT ")", sublen,
1796           blocksize - offset);
1797       break;
1798     }
1799     if ((data[0] & 0x20) == 0) {
1800       switch (data[0] & 0x0f) {
1801         case 0xa:              /* ID_WV_BITSTREAM  */
1802         case 0xc:              /* ID_WVX_BITSTREAM */
1803           ++count_wv;
1804           break;
1805         case 0xb:              /* ID_WVC_BITSTREAM */
1806           ++count_wvc;
1807           break;
1808         default:
1809           break;
1810       }
1811       if (count_wv >= 5 || count_wvc >= 5)
1812         break;
1813     }
1814     offset += sublen;
1815   }
1816
1817   /* check for second block header */
1818   data = gst_type_find_peek (tf, 8 + blocksize, 4);
1819   if (data != NULL && memcmp (data, "wvpk", 4) == 0) {
1820     GST_DEBUG ("found second block sync");
1821     base_prob = GST_TYPE_FIND_LIKELY;
1822   }
1823
1824   GST_DEBUG ("wvc=%d, wv=%d", count_wvc, count_wv);
1825
1826   if (count_wvc > 0 && count_wvc > count_wv) {
1827     gst_type_find_suggest (tf,
1828         MIN (base_prob + 5 * count_wvc, GST_TYPE_FIND_NEARLY_CERTAIN),
1829         WAVPACK_CORRECTION_CAPS);
1830   } else if (count_wv > 0) {
1831     gst_type_find_suggest (tf,
1832         MIN (base_prob + 5 * count_wv, GST_TYPE_FIND_NEARLY_CERTAIN),
1833         WAVPACK_CAPS);
1834   }
1835 }
1836
1837 /*** application/postscrip ***/
1838 static GstStaticCaps postscript_caps =
1839 GST_STATIC_CAPS ("application/postscript");
1840
1841 #define POSTSCRIPT_CAPS (gst_static_caps_get(&postscript_caps))
1842
1843 static void
1844 postscript_type_find (GstTypeFind * tf, gpointer unused)
1845 {
1846   const guint8 *data = gst_type_find_peek (tf, 0, 3);
1847   if (!data)
1848     return;
1849
1850   if (data[0] == 0x04)
1851     data++;
1852   if (data[0] == '%' && data[1] == '!')
1853     gst_type_find_suggest (tf, GST_TYPE_FIND_POSSIBLE, POSTSCRIPT_CAPS);
1854
1855 }
1856
1857 /*** image/svg+xml ***/
1858 static GstStaticCaps svg_caps = GST_STATIC_CAPS ("image/svg+xml");
1859
1860 #define SVG_CAPS (gst_static_caps_get(&svg_caps))
1861
1862 static void
1863 svg_type_find (GstTypeFind * tf, gpointer unused)
1864 {
1865   static const gchar svg_doctype[] = "!DOCTYPE svg";
1866   static const gchar svg_tag[] = "<svg";
1867   DataScanCtx c = { 0, NULL, 0 };
1868
1869   while (c.offset <= 1024) {
1870     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 12)))
1871       break;
1872
1873     if (memcmp (svg_doctype, c.data, 12) == 0) {
1874       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SVG_CAPS);
1875       return;
1876     } else if (memcmp (svg_tag, c.data, 4) == 0) {
1877       gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, SVG_CAPS);
1878       return;
1879     }
1880     data_scan_ctx_advance (tf, &c, 1);
1881   }
1882 }
1883
1884 /*** multipart/x-mixed-replace mimestream ***/
1885
1886 static GstStaticCaps multipart_caps =
1887 GST_STATIC_CAPS ("multipart/x-mixed-replace");
1888 #define MULTIPART_CAPS gst_static_caps_get(&multipart_caps)
1889
1890 /* multipart/x-mixed replace is:
1891  *   <maybe some whitespace>--<some ascii chars>[\r]\n
1892  *   <more ascii chars>[\r]\nContent-type:<more ascii>[\r]\n */
1893 static void
1894 multipart_type_find (GstTypeFind * tf, gpointer unused)
1895 {
1896   const guint8 *data;
1897   const guint8 *x;
1898
1899 #define MULTIPART_MAX_BOUNDARY_OFFSET 16
1900   data = gst_type_find_peek (tf, 0, MULTIPART_MAX_BOUNDARY_OFFSET);
1901   if (!data)
1902     return;
1903
1904   for (x = data;
1905       x - data < MULTIPART_MAX_BOUNDARY_OFFSET - 2 && g_ascii_isspace (*x);
1906       x++);
1907   if (x[0] != '-' || x[1] != '-')
1908     return;
1909
1910   /* Could be okay, peek what should be enough for a complete header */
1911 #define MULTIPART_MAX_HEADER_SIZE 256
1912   data = gst_type_find_peek (tf, 0, MULTIPART_MAX_HEADER_SIZE);
1913   if (!data)
1914     return;
1915
1916   for (x = data; x - data < MULTIPART_MAX_HEADER_SIZE - 14; x++) {
1917     if (!isascii (*x)) {
1918       return;
1919     }
1920     if (*x == '\n' &&
1921         !g_ascii_strncasecmp ("content-type:", (gchar *) x + 1, 13)) {
1922       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MULTIPART_CAPS);
1923       return;
1924     }
1925   }
1926 }
1927
1928 /*** video/mpeg systemstream ***/
1929 static GstStaticCaps mpeg_sys_caps = GST_STATIC_CAPS ("video/mpeg, "
1930     "systemstream = (boolean) true, mpegversion = (int) [ 1, 2 ]");
1931
1932 #define MPEG_SYS_CAPS gst_static_caps_get(&mpeg_sys_caps)
1933 #define IS_MPEG_HEADER(data) (G_UNLIKELY((((guint8 *)(data))[0] == 0x00) &&  \
1934                                          (((guint8 *)(data))[1] == 0x00) &&  \
1935                                          (((guint8 *)(data))[2] == 0x01)))
1936
1937 #define IS_MPEG_PACK_CODE(b) ((b) == 0xBA)
1938 #define IS_MPEG_SYS_CODE(b) ((b) == 0xBB)
1939 #define IS_MPEG_PACK_HEADER(data)       (IS_MPEG_HEADER (data) &&            \
1940                                          IS_MPEG_PACK_CODE (((guint8 *)(data))[3]))
1941
1942 #define IS_MPEG_PES_CODE(b) (((b) & 0xF0) == 0xE0 || ((b) & 0xF0) == 0xC0 || \
1943                              (b) >= 0xBD)
1944 #define IS_MPEG_PES_HEADER(data)        (IS_MPEG_HEADER (data) &&            \
1945                                          IS_MPEG_PES_CODE (((guint8 *)(data))[3]))
1946
1947 #define MPEG2_MAX_PROBE_LENGTH (128 * 1024)     /* 128kB should be 64 packs of the
1948                                                  * most common 2kB pack size. */
1949
1950 #define MPEG2_MIN_SYS_HEADERS 2
1951 #define MPEG2_MAX_SYS_HEADERS 5
1952
1953 static gboolean
1954 mpeg_sys_is_valid_pack (GstTypeFind * tf, const guint8 * data, guint len,
1955     guint * pack_size)
1956 {
1957   /* Check the pack header @ offset for validity, assuming that the 4 byte header
1958    * itself has already been checked. */
1959   guint8 stuff_len;
1960
1961   if (len < 12)
1962     return FALSE;
1963
1964   /* Check marker bits */
1965   if ((data[4] & 0xC4) == 0x44) {
1966     /* MPEG-2 PACK */
1967     if (len < 14)
1968       return FALSE;
1969
1970     if ((data[6] & 0x04) != 0x04 ||
1971         (data[8] & 0x04) != 0x04 ||
1972         (data[9] & 0x01) != 0x01 || (data[12] & 0x03) != 0x03)
1973       return FALSE;
1974
1975     stuff_len = data[13] & 0x07;
1976
1977     /* Check the following header bytes, if we can */
1978     if ((14 + stuff_len + 4) <= len) {
1979       if (!IS_MPEG_HEADER (data + 14 + stuff_len))
1980         return FALSE;
1981     }
1982     if (pack_size)
1983       *pack_size = 14 + stuff_len;
1984     return TRUE;
1985   } else if ((data[4] & 0xF1) == 0x21) {
1986     /* MPEG-1 PACK */
1987     if ((data[6] & 0x01) != 0x01 ||
1988         (data[8] & 0x01) != 0x01 ||
1989         (data[9] & 0x80) != 0x80 || (data[11] & 0x01) != 0x01)
1990       return FALSE;
1991
1992     /* Check the following header bytes, if we can */
1993     if ((12 + 4) <= len) {
1994       if (!IS_MPEG_HEADER (data + 12))
1995         return FALSE;
1996     }
1997     if (pack_size)
1998       *pack_size = 12;
1999     return TRUE;
2000   }
2001
2002   return FALSE;
2003 }
2004
2005 static gboolean
2006 mpeg_sys_is_valid_pes (GstTypeFind * tf, const guint8 * data, guint len,
2007     guint * pack_size)
2008 {
2009   guint pes_packet_len;
2010
2011   /* Check the PES header at the given position, assuming the header code itself
2012    * was already checked */
2013   if (len < 6)
2014     return FALSE;
2015
2016   /* For MPEG Program streams, unbounded PES is not allowed, so we must have a
2017    * valid length present */
2018   pes_packet_len = GST_READ_UINT16_BE (data + 4);
2019   if (pes_packet_len == 0)
2020     return FALSE;
2021
2022   /* Check the following header, if we can */
2023   if (6 + pes_packet_len + 4 <= len) {
2024     if (!IS_MPEG_HEADER (data + 6 + pes_packet_len))
2025       return FALSE;
2026   }
2027
2028   if (pack_size)
2029     *pack_size = 6 + pes_packet_len;
2030   return TRUE;
2031 }
2032
2033 static gboolean
2034 mpeg_sys_is_valid_sys (GstTypeFind * tf, const guint8 * data, guint len,
2035     guint * pack_size)
2036 {
2037   guint sys_hdr_len;
2038
2039   /* Check the System header at the given position, assuming the header code itself
2040    * was already checked */
2041   if (len < 6)
2042     return FALSE;
2043   sys_hdr_len = GST_READ_UINT16_BE (data + 4);
2044   if (sys_hdr_len < 6)
2045     return FALSE;
2046
2047   /* Check the following header, if we can */
2048   if (6 + sys_hdr_len + 4 <= len) {
2049     if (!IS_MPEG_HEADER (data + 6 + sys_hdr_len))
2050       return FALSE;
2051   }
2052
2053   if (pack_size)
2054     *pack_size = 6 + sys_hdr_len;
2055
2056   return TRUE;
2057 }
2058
2059 /* calculation of possibility to identify random data as mpeg systemstream:
2060  * bits that must match in header detection:            32 (or more)
2061  * chance that random data is identifed:                1/2^32
2062  * chance that MPEG2_MIN_PACK_HEADERS headers are identified:
2063  *       1/2^(32*MPEG2_MIN_PACK_HEADERS)
2064  * chance that this happens in MPEG2_MAX_PROBE_LENGTH bytes:
2065  *       1-(1+1/2^(32*MPEG2_MIN_PACK_HEADERS)^MPEG2_MAX_PROBE_LENGTH)
2066  * for current values:
2067  *       1-(1+1/2^(32*4)^101024)
2068  *       = <some_number>
2069  * Since we also check marker bits and pes packet lengths, this probability is a
2070  * very coarse upper bound.
2071  */
2072 static void
2073 mpeg_sys_type_find (GstTypeFind * tf, gpointer unused)
2074 {
2075   const guint8 *data, *data0, *first_sync, *end;
2076   gint mpegversion = 0;
2077   guint pack_headers = 0;
2078   guint pes_headers = 0;
2079   guint pack_size;
2080   guint since_last_sync = 0;
2081   guint32 sync_word = 0xffffffff;
2082
2083   G_STMT_START {
2084     gint len;
2085
2086     len = MPEG2_MAX_PROBE_LENGTH;
2087     do {
2088       len = len / 2;
2089       data = gst_type_find_peek (tf, 0, 5 + len);
2090     } while (data == NULL && len >= 32);
2091
2092     if (!data)
2093       return;
2094
2095     end = data + len;
2096   }
2097   G_STMT_END;
2098
2099   data0 = data;
2100   first_sync = NULL;
2101
2102   while (data < end) {
2103     sync_word <<= 8;
2104     if (sync_word == 0x00000100) {
2105       /* Found potential sync word */
2106       if (first_sync == NULL)
2107         first_sync = data - 3;
2108
2109       if (since_last_sync > 4) {
2110         /* If more than 4 bytes since the last sync word, reset our counters,
2111          * as we're only interested in counting contiguous packets */
2112         pes_headers = pack_headers = 0;
2113       }
2114       pack_size = 0;
2115
2116       if (IS_MPEG_PACK_CODE (data[0])) {
2117         if ((data[1] & 0xC0) == 0x40) {
2118           /* MPEG-2 */
2119           mpegversion = 2;
2120         } else if ((data[1] & 0xF0) == 0x20) {
2121           mpegversion = 1;
2122         }
2123         if (mpegversion != 0 &&
2124             mpeg_sys_is_valid_pack (tf, data - 3, end - data + 3, &pack_size)) {
2125           pack_headers++;
2126         }
2127       } else if (IS_MPEG_PES_CODE (data[0])) {
2128         /* PES stream */
2129         if (mpeg_sys_is_valid_pes (tf, data - 3, end - data + 3, &pack_size)) {
2130           pes_headers++;
2131           if (mpegversion == 0)
2132             mpegversion = 2;
2133         }
2134       } else if (IS_MPEG_SYS_CODE (data[0])) {
2135         if (mpeg_sys_is_valid_sys (tf, data - 3, end - data + 3, &pack_size)) {
2136           pack_headers++;
2137         }
2138       }
2139
2140       /* If we found a packet with a known size, skip the bytes in it and loop
2141        * around to check the next packet. */
2142       if (pack_size != 0) {
2143         data += pack_size - 3;
2144         sync_word = 0xffffffff;
2145         since_last_sync = 0;
2146         continue;
2147       }
2148     }
2149
2150     sync_word |= data[0];
2151     since_last_sync++;
2152     data++;
2153
2154     /* If we have found MAX headers, and *some* were pes headers (pack headers
2155      * are optional in an mpeg system stream) then return our high-probability
2156      * result */
2157     if (pes_headers > 0 && (pack_headers + pes_headers) > MPEG2_MAX_SYS_HEADERS)
2158       goto suggest;
2159   }
2160
2161   /* If we at least saw MIN headers, and *some* were pes headers (pack headers
2162    * are optional in an mpeg system stream) then return a lower-probability
2163    * result */
2164   if (pes_headers > 0 && (pack_headers + pes_headers) > MPEG2_MIN_SYS_HEADERS)
2165     goto suggest;
2166
2167   return;
2168 suggest:
2169   {
2170     guint prob;
2171
2172     prob = GST_TYPE_FIND_POSSIBLE + (10 * (pack_headers + pes_headers));
2173     prob = MIN (prob, GST_TYPE_FIND_MAXIMUM);
2174
2175     /* lower probability if the first packet wasn't right at the start */
2176     if (data0 != first_sync && prob >= 10)
2177       prob -= 10;
2178
2179     GST_LOG ("Suggesting MPEG %d system stream, %d packs, %d pes, prob %u%%\n",
2180         mpegversion, pack_headers, pes_headers, prob);
2181
2182     gst_type_find_suggest_simple (tf, prob, "video/mpeg",
2183         "systemstream", G_TYPE_BOOLEAN, TRUE,
2184         "mpegversion", G_TYPE_INT, mpegversion, NULL);
2185   }
2186 };
2187
2188 /*** video/mpegts Transport Stream ***/
2189 static GstStaticCaps mpegts_caps = GST_STATIC_CAPS ("video/mpegts, "
2190     "systemstream = (boolean) true, packetsize = (int) [ 188, 208 ]");
2191 #define MPEGTS_CAPS gst_static_caps_get(&mpegts_caps)
2192
2193 #define GST_MPEGTS_TYPEFIND_MIN_HEADERS 4
2194 #define GST_MPEGTS_TYPEFIND_MAX_HEADERS 10
2195 #define GST_MPEGTS_MAX_PACKET_SIZE 208
2196 #define GST_MPEGTS_TYPEFIND_SYNC_SIZE \
2197             (GST_MPEGTS_TYPEFIND_MIN_HEADERS * GST_MPEGTS_MAX_PACKET_SIZE)
2198 #define GST_MPEGTS_TYPEFIND_MAX_SYNC \
2199             (GST_MPEGTS_TYPEFIND_MAX_HEADERS * GST_MPEGTS_MAX_PACKET_SIZE)
2200 #define GST_MPEGTS_TYPEFIND_SCAN_LENGTH \
2201             (GST_MPEGTS_TYPEFIND_MAX_SYNC * 4)
2202
2203 #define MPEGTS_HDR_SIZE 4
2204 /* Check for sync byte, error_indicator == 0 and packet has payload */
2205 #define IS_MPEGTS_HEADER(data) (((data)[0] == 0x47) && \
2206                                 (((data)[1] & 0x80) == 0x00) && \
2207                                 (((data)[3] & 0x30) != 0x00))
2208
2209 /* Helper function to search ahead at intervals of packet_size for mpegts
2210  * headers */
2211 static gint
2212 mpeg_ts_probe_headers (GstTypeFind * tf, guint64 offset, gint packet_size)
2213 {
2214   /* We always enter this function having found at least one header already */
2215   gint found = 1;
2216   const guint8 *data = NULL;
2217
2218   GST_LOG ("looking for mpeg-ts packets of size %u", packet_size);
2219   while (found < GST_MPEGTS_TYPEFIND_MAX_HEADERS) {
2220     offset += packet_size;
2221
2222     data = gst_type_find_peek (tf, offset, MPEGTS_HDR_SIZE);
2223     if (data == NULL || !IS_MPEGTS_HEADER (data))
2224       return found;
2225
2226     found++;
2227     GST_LOG ("mpeg-ts sync #%2d at offset %" G_GUINT64_FORMAT, found, offset);
2228   }
2229
2230   return found;
2231 }
2232
2233 /* Try and detect at least 4 packets in at most 10 packets worth of
2234  * data. Need to try several possible packet sizes */
2235 static void
2236 mpeg_ts_type_find (GstTypeFind * tf, gpointer unused)
2237 {
2238   /* TS packet sizes to test: normal, DVHS packet size and
2239    * FEC with 16 or 20 byte codes packet size. */
2240   const gint pack_sizes[] = { 188, 192, 204, 208 };
2241   const guint8 *data = NULL;
2242   guint size = 0;
2243   guint64 skipped = 0;
2244
2245   while (skipped < GST_MPEGTS_TYPEFIND_SCAN_LENGTH) {
2246     if (size < MPEGTS_HDR_SIZE) {
2247       data = gst_type_find_peek (tf, skipped, GST_MPEGTS_TYPEFIND_SYNC_SIZE);
2248       if (!data)
2249         break;
2250       size = GST_MPEGTS_TYPEFIND_SYNC_SIZE;
2251     }
2252
2253     /* Have at least MPEGTS_HDR_SIZE bytes at this point */
2254     if (IS_MPEGTS_HEADER (data)) {
2255       gint p;
2256
2257       GST_LOG ("possible mpeg-ts sync at offset %" G_GUINT64_FORMAT, skipped);
2258
2259       for (p = 0; p < G_N_ELEMENTS (pack_sizes); p++) {
2260         gint found;
2261
2262         /* Probe ahead at size pack_sizes[p] */
2263         found = mpeg_ts_probe_headers (tf, skipped, pack_sizes[p]);
2264         if (found >= GST_MPEGTS_TYPEFIND_MIN_HEADERS) {
2265           gint probability;
2266
2267           /* found at least 4 headers. 10 headers = MAXIMUM probability.
2268            * Arbitrarily, I assigned 10% probability for each header we
2269            * found, 40% -> 100% */
2270           probability = MIN (10 * found, GST_TYPE_FIND_MAXIMUM);
2271
2272           gst_type_find_suggest_simple (tf, probability, "video/mpegts",
2273               "systemstream", G_TYPE_BOOLEAN, TRUE,
2274               "packetsize", G_TYPE_INT, pack_sizes[p], NULL);
2275           return;
2276         }
2277       }
2278     }
2279     data++;
2280     skipped++;
2281     size--;
2282   }
2283 }
2284
2285 #define GST_MPEGVID_TYPEFIND_TRY_PICTURES 6
2286 #define GST_MPEGVID_TYPEFIND_TRY_SYNC (100 * 1024)      /* 100 kB */
2287
2288 /* Scan ahead a maximum of max_extra_offset bytes until the next IS_MPEG_HEADER
2289  * offset.  After the call, offset will be after the 0x000001, i.e. at the 4th
2290  * byte of the MPEG header.  Returns TRUE if a header was found, FALSE if not.
2291  */
2292 static gboolean
2293 mpeg_find_next_header (GstTypeFind * tf, DataScanCtx * c,
2294     guint64 max_extra_offset)
2295 {
2296   guint64 extra_offset;
2297
2298   for (extra_offset = 0; extra_offset <= max_extra_offset; ++extra_offset) {
2299     if (!data_scan_ctx_ensure_data (tf, c, 4))
2300       return FALSE;
2301     if (IS_MPEG_HEADER (c->data)) {
2302       data_scan_ctx_advance (tf, c, 3);
2303       return TRUE;
2304     }
2305     data_scan_ctx_advance (tf, c, 1);
2306   }
2307   return FALSE;
2308 }
2309
2310 /*** video/mpeg MPEG-4 elementary video stream ***/
2311
2312 static GstStaticCaps mpeg4_video_caps = GST_STATIC_CAPS ("video/mpeg, "
2313     "systemstream=(boolean)false, mpegversion=4, parsed=(boolean)false");
2314 #define MPEG4_VIDEO_CAPS gst_static_caps_get(&mpeg4_video_caps)
2315
2316 /*
2317  * This typefind is based on the elementary video header defined in
2318  * http://xhelmboyx.tripod.com/formats/mpeg-layout.txt
2319  * In addition, it allows the visual object sequence header to be
2320  * absent, and even the VOS header to be absent.  In the latter case,
2321  * a number of VOPs have to be present.
2322  */
2323 static void
2324 mpeg4_video_type_find (GstTypeFind * tf, gpointer unused)
2325 {
2326   DataScanCtx c = { 0, NULL, 0 };
2327   gboolean seen_vios_at_0 = FALSE;
2328   gboolean seen_vios = FALSE;
2329   gboolean seen_vos = FALSE;
2330   gboolean seen_vol = FALSE;
2331   guint num_vop_headers = 0;
2332   guint8 sc;
2333
2334   while (c.offset < GST_MPEGVID_TYPEFIND_TRY_SYNC) {
2335     if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES)
2336       break;
2337
2338     if (!mpeg_find_next_header (tf, &c,
2339             GST_MPEGVID_TYPEFIND_TRY_SYNC - c.offset))
2340       break;
2341
2342     sc = c.data[0];
2343
2344     /* visual_object_sequence_start_code */
2345     if (sc == 0xB0) {
2346       if (seen_vios)
2347         break;                  /* Terminate at second vios */
2348       if (c.offset == 0)
2349         seen_vios_at_0 = TRUE;
2350       seen_vios = TRUE;
2351       data_scan_ctx_advance (tf, &c, 2);
2352       if (!mpeg_find_next_header (tf, &c, 0))
2353         break;
2354
2355       sc = c.data[0];
2356
2357       /* Optional metadata */
2358       if (sc == 0xB2)
2359         if (!mpeg_find_next_header (tf, &c, 24))
2360           break;
2361     }
2362
2363     /* visual_object_start_code (consider it optional) */
2364     if (sc == 0xB5) {
2365       data_scan_ctx_advance (tf, &c, 2);
2366       /* may contain ID marker and YUV clamping */
2367       if (!mpeg_find_next_header (tf, &c, 7))
2368         break;
2369
2370       sc = c.data[0];
2371     }
2372
2373     /* video_object_start_code */
2374     if (sc <= 0x1F) {
2375       if (seen_vos)
2376         break;                  /* Terminate at second vos */
2377       seen_vos = TRUE;
2378       data_scan_ctx_advance (tf, &c, 2);
2379       continue;
2380     }
2381
2382     /* video_object_layer_start_code */
2383     if (sc >= 0x20 && sc <= 0x2F) {
2384       seen_vol = TRUE;
2385       data_scan_ctx_advance (tf, &c, 5);
2386       continue;
2387     }
2388
2389     /* video_object_plane_start_code */
2390     if (sc == 0xB6) {
2391       num_vop_headers++;
2392       data_scan_ctx_advance (tf, &c, 2);
2393       continue;
2394     }
2395
2396     /* Unknown start code. */
2397   }
2398
2399   if (num_vop_headers > 0 || seen_vol) {
2400     GstTypeFindProbability probability = 0;
2401
2402     GST_LOG ("Found %d pictures, vios: %d, vos:%d, vol:%d", num_vop_headers,
2403         seen_vios, seen_vos, seen_vol);
2404
2405     if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_vios_at_0
2406         && seen_vos && seen_vol)
2407       probability = GST_TYPE_FIND_MAXIMUM - 1;
2408     else if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_vios
2409         && seen_vos && seen_vol)
2410       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 1;
2411     else if (seen_vios_at_0 && seen_vos && seen_vol)
2412       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 6;
2413     else if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_vos
2414         && seen_vol)
2415       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 6;
2416     else if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_vol)
2417       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 9;
2418     else if (num_vop_headers >= GST_MPEGVID_TYPEFIND_TRY_PICTURES)
2419       probability = GST_TYPE_FIND_LIKELY - 1;
2420     else if (num_vop_headers > 2 && seen_vios && seen_vos && seen_vol)
2421       probability = GST_TYPE_FIND_LIKELY - 9;
2422     else if (seen_vios && seen_vos && seen_vol)
2423       probability = GST_TYPE_FIND_LIKELY - 20;
2424     else if (num_vop_headers > 0 && seen_vos && seen_vol)
2425       probability = GST_TYPE_FIND_POSSIBLE;
2426     else if (num_vop_headers > 0)
2427       probability = GST_TYPE_FIND_POSSIBLE - 10;
2428     else if (seen_vos && seen_vol)
2429       probability = GST_TYPE_FIND_POSSIBLE - 20;
2430
2431     gst_type_find_suggest (tf, probability, MPEG4_VIDEO_CAPS);
2432   }
2433 }
2434
2435 /*** video/x-h263 H263 video stream ***/
2436 static GstStaticCaps h263_video_caps = GST_STATIC_CAPS ("video/x-h263");
2437
2438 #define H263_VIDEO_CAPS gst_static_caps_get(&h263_video_caps)
2439
2440 #define H263_MAX_PROBE_LENGTH (128 * 1024)
2441
2442 static void
2443 h263_video_type_find (GstTypeFind * tf, gpointer unused)
2444 {
2445   DataScanCtx c = { 0, NULL, 0 };
2446   guint64 data = 0;
2447   guint64 psc = 0;
2448   guint8 tr = 0;
2449   guint format;
2450   guint good = 0;
2451   guint bad = 0;
2452
2453   while (c.offset < H263_MAX_PROBE_LENGTH) {
2454     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 4)))
2455       break;
2456
2457     /* Find the picture start code */
2458     data = (data << 8) + c.data[0];
2459     psc = data & G_GUINT64_CONSTANT (0xfffffc0000);
2460     if (psc == 0x800000) {
2461       /* Found PSC */
2462       /* TR */
2463       tr = (data & 0x3fc) >> 2;
2464       /* Source Format */
2465       format = tr & 0x07;
2466
2467       /* Now that we have a Valid PSC, check if we also have a valid PTYPE and
2468          the Source Format, which should range between 1 and 5 */
2469       if (((tr >> 6) == 0x2) && (format > 0 && format < 6))
2470         good++;
2471       else
2472         bad++;
2473
2474       /* FIXME: maybe bail out early if we get mostly bad syncs ? */
2475     }
2476
2477     data_scan_ctx_advance (tf, &c, 1);
2478   }
2479
2480   if (good > 0 && bad == 0)
2481     gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, H263_VIDEO_CAPS);
2482   else if (good > 2 * bad)
2483     gst_type_find_suggest (tf, GST_TYPE_FIND_POSSIBLE, H263_VIDEO_CAPS);
2484
2485   return;
2486 }
2487
2488 /*** video/x-h264 H264 elementary video stream ***/
2489
2490 static GstStaticCaps h264_video_caps =
2491 GST_STATIC_CAPS ("video/x-h264,stream-format=byte-stream");
2492
2493 #define H264_VIDEO_CAPS gst_static_caps_get(&h264_video_caps)
2494
2495 #define H264_MAX_PROBE_LENGTH (128 * 1024)      /* 128kB for HD should be enough. */
2496
2497 static void
2498 h264_video_type_find (GstTypeFind * tf, gpointer unused)
2499 {
2500   DataScanCtx c = { 0, NULL, 0 };
2501
2502   /* Stream consists of: a series of sync codes (00 00 00 01) followed
2503    * by NALs
2504    */
2505   gboolean seen_idr = FALSE;
2506   gboolean seen_sps = FALSE;
2507   gboolean seen_pps = FALSE;
2508   int nut, ref;
2509   int good = 0;
2510   int bad = 0;
2511
2512   while (c.offset < H264_MAX_PROBE_LENGTH) {
2513     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 4)))
2514       break;
2515
2516     if (IS_MPEG_HEADER (c.data)) {
2517       nut = c.data[3] & 0x9f;   /* forbiden_zero_bit | nal_unit_type */
2518       ref = c.data[3] & 0x60;   /* nal_ref_idc */
2519
2520       /* if forbidden bit is different to 0 won't be h264 */
2521       if (nut > 0x1f) {
2522         bad++;
2523         break;
2524       }
2525
2526       /* collect statistics about the NAL types */
2527       if ((nut >= 1 && nut <= 13) || nut == 19) {
2528         if ((nut == 5 && ref == 0) ||
2529             ((nut == 6 || (nut >= 9 && nut <= 12)) && ref != 0)) {
2530           bad++;
2531         } else {
2532           if (nut == 7)
2533             seen_sps = TRUE;
2534           else if (nut == 8)
2535             seen_pps = TRUE;
2536           else if (nut == 5)
2537             seen_idr = TRUE;
2538
2539           good++;
2540         }
2541       } else if (nut >= 14 && nut <= 33) {
2542         /* reserved */
2543         /* Theoretically these are good, since if they exist in the
2544            stream it merely means that a newer backwards-compatible
2545            h.264 stream.  But we should be identifying that separately. */
2546         bad++;
2547       } else {
2548         /* unspecified, application specific */
2549         /* don't consider these bad */
2550       }
2551
2552       GST_LOG ("good:%d, bad:%d, pps:%d, sps:%d, idr:%d", good, bad, seen_pps,
2553           seen_sps, seen_idr);
2554
2555       if (seen_sps && seen_pps && seen_idr && good >= 10 && bad < 4) {
2556         gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, H264_VIDEO_CAPS);
2557         return;
2558       }
2559
2560       data_scan_ctx_advance (tf, &c, 4);
2561     }
2562     data_scan_ctx_advance (tf, &c, 1);
2563   }
2564
2565   GST_LOG ("good:%d, bad:%d, pps:%d, sps:%d, idr:%d", good, bad, seen_pps,
2566       seen_sps, seen_idr);
2567
2568   if (good >= 2 && bad == 0) {
2569     gst_type_find_suggest (tf, GST_TYPE_FIND_POSSIBLE, H264_VIDEO_CAPS);
2570   }
2571 }
2572
2573 /*** video/mpeg video stream ***/
2574
2575 static GstStaticCaps mpeg_video_caps = GST_STATIC_CAPS ("video/mpeg, "
2576     "systemstream = (boolean) false");
2577 #define MPEG_VIDEO_CAPS gst_static_caps_get(&mpeg_video_caps)
2578
2579 /*
2580  * Idea is the same as MPEG system stream typefinding: We check each
2581  * byte of the stream to see if - from that point on - the stream
2582  * matches a predefined set of marker bits as defined in the MPEG
2583  * video specs.
2584  *
2585  * I'm sure someone will do a chance calculation here too.
2586  */
2587
2588 static void
2589 mpeg_video_stream_type_find (GstTypeFind * tf, gpointer unused)
2590 {
2591   DataScanCtx c = { 0, NULL, 0 };
2592   gboolean seen_seq_at_0 = FALSE;
2593   gboolean seen_seq = FALSE;
2594   gboolean seen_gop = FALSE;
2595   guint64 last_pic_offset = 0;
2596   guint num_pic_headers = 0;
2597   gint found = 0;
2598
2599   while (c.offset < GST_MPEGVID_TYPEFIND_TRY_SYNC) {
2600     if (found >= GST_MPEGVID_TYPEFIND_TRY_PICTURES)
2601       break;
2602
2603     if (!data_scan_ctx_ensure_data (tf, &c, 5))
2604       break;
2605
2606     if (!IS_MPEG_HEADER (c.data))
2607       goto next;
2608
2609     /* a pack header indicates that this isn't an elementary stream */
2610     if (c.data[3] == 0xBA && mpeg_sys_is_valid_pack (tf, c.data, c.size, NULL))
2611       return;
2612
2613     /* do we have a sequence header? */
2614     if (c.data[3] == 0xB3) {
2615       seen_seq_at_0 = seen_seq_at_0 || (c.offset == 0);
2616       seen_seq = TRUE;
2617       data_scan_ctx_advance (tf, &c, 4 + 8);
2618       continue;
2619     }
2620
2621     /* or a GOP header */
2622     if (c.data[3] == 0xB8) {
2623       seen_gop = TRUE;
2624       data_scan_ctx_advance (tf, &c, 8);
2625       continue;
2626     }
2627
2628     /* but what we'd really like to see is a picture header */
2629     if (c.data[3] == 0x00) {
2630       ++num_pic_headers;
2631       last_pic_offset = c.offset;
2632       data_scan_ctx_advance (tf, &c, 8);
2633       continue;
2634     }
2635
2636     /* ... each followed by a slice header with slice_vertical_pos=1 that's
2637      * not too far away from the previously seen picture header. */
2638     if (c.data[3] == 0x01 && num_pic_headers > found &&
2639         (c.offset - last_pic_offset) >= 4 &&
2640         (c.offset - last_pic_offset) <= 64) {
2641       data_scan_ctx_advance (tf, &c, 4);
2642       found += 1;
2643       continue;
2644     }
2645
2646   next:
2647
2648     data_scan_ctx_advance (tf, &c, 1);
2649   }
2650
2651   if (found > 0 || seen_seq) {
2652     GstTypeFindProbability probability = 0;
2653
2654     GST_LOG ("Found %d pictures, seq:%d, gop:%d", found, seen_seq, seen_gop);
2655
2656     if (found >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_seq && seen_gop)
2657       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 1;
2658     else if (found >= GST_MPEGVID_TYPEFIND_TRY_PICTURES && seen_seq)
2659       probability = GST_TYPE_FIND_NEARLY_CERTAIN - 9;
2660     else if (found >= GST_MPEGVID_TYPEFIND_TRY_PICTURES)
2661       probability = GST_TYPE_FIND_LIKELY;
2662     else if (seen_seq_at_0 && seen_gop && found > 2)
2663       probability = GST_TYPE_FIND_LIKELY - 10;
2664     else if (seen_seq && seen_gop && found > 2)
2665       probability = GST_TYPE_FIND_LIKELY - 20;
2666     else if (seen_seq_at_0 && found > 0)
2667       probability = GST_TYPE_FIND_POSSIBLE;
2668     else if (seen_seq && found > 0)
2669       probability = GST_TYPE_FIND_POSSIBLE - 5;
2670     else if (found > 0)
2671       probability = GST_TYPE_FIND_POSSIBLE - 10;
2672     else if (seen_seq)
2673       probability = GST_TYPE_FIND_POSSIBLE - 20;
2674
2675     gst_type_find_suggest_simple (tf, probability, "video/mpeg",
2676         "systemstream", G_TYPE_BOOLEAN, FALSE,
2677         "mpegversion", G_TYPE_INT, 1, "parsed", G_TYPE_BOOLEAN, FALSE, NULL);
2678   }
2679 }
2680
2681 /*** audio/x-aiff ***/
2682
2683 static GstStaticCaps aiff_caps = GST_STATIC_CAPS ("audio/x-aiff");
2684
2685 #define AIFF_CAPS gst_static_caps_get(&aiff_caps)
2686 static void
2687 aiff_type_find (GstTypeFind * tf, gpointer unused)
2688 {
2689   const guint8 *data = gst_type_find_peek (tf, 0, 4);
2690
2691   if (data && memcmp (data, "FORM", 4) == 0) {
2692     data += 8;
2693     if (memcmp (data, "AIFF", 4) == 0 || memcmp (data, "AIFC", 4) == 0)
2694       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, AIFF_CAPS);
2695   }
2696 }
2697
2698 /*** audio/x-svx ***/
2699
2700 static GstStaticCaps svx_caps = GST_STATIC_CAPS ("audio/x-svx");
2701
2702 #define SVX_CAPS gst_static_caps_get(&svx_caps)
2703 static void
2704 svx_type_find (GstTypeFind * tf, gpointer unused)
2705 {
2706   const guint8 *data = gst_type_find_peek (tf, 0, 4);
2707
2708   if (data && memcmp (data, "FORM", 4) == 0) {
2709     data += 8;
2710     if (memcmp (data, "8SVX", 4) == 0 || memcmp (data, "16SV", 4) == 0)
2711       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SVX_CAPS);
2712   }
2713 }
2714
2715 /*** audio/x-shorten ***/
2716
2717 static GstStaticCaps shn_caps = GST_STATIC_CAPS ("audio/x-shorten");
2718
2719 #define SHN_CAPS gst_static_caps_get(&shn_caps)
2720 static void
2721 shn_type_find (GstTypeFind * tf, gpointer unused)
2722 {
2723   const guint8 *data = gst_type_find_peek (tf, 0, 4);
2724
2725   if (data && memcmp (data, "ajkg", 4) == 0) {
2726     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SHN_CAPS);
2727   }
2728   data = gst_type_find_peek (tf, -8, 8);
2729   if (data && memcmp (data, "SHNAMPSK", 8) == 0) {
2730     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SHN_CAPS);
2731   }
2732 }
2733
2734 /*** application/x-ape ***/
2735
2736 static GstStaticCaps ape_caps = GST_STATIC_CAPS ("application/x-ape");
2737
2738 #define APE_CAPS gst_static_caps_get(&ape_caps)
2739 static void
2740 ape_type_find (GstTypeFind * tf, gpointer unused)
2741 {
2742   const guint8 *data = gst_type_find_peek (tf, 0, 4);
2743
2744   if (data && memcmp (data, "MAC ", 4) == 0) {
2745     gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY + 10, APE_CAPS);
2746   }
2747 }
2748
2749 /*** ISO FORMATS ***/
2750
2751 /*** audio/x-m4a ***/
2752
2753 static GstStaticCaps m4a_caps = GST_STATIC_CAPS ("audio/x-m4a");
2754
2755 #define M4A_CAPS (gst_static_caps_get(&m4a_caps))
2756 static void
2757 m4a_type_find (GstTypeFind * tf, gpointer unused)
2758 {
2759   const guint8 *data = gst_type_find_peek (tf, 4, 8);
2760
2761   if (data && (memcmp (data, "ftypM4A ", 8) == 0)) {
2762     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, M4A_CAPS);
2763   }
2764 }
2765
2766 /*** application/x-3gp ***/
2767
2768 /* The Q is there because variables can't start with a number. */
2769 static GstStaticCaps q3gp_caps = GST_STATIC_CAPS ("application/x-3gp");
2770 #define Q3GP_CAPS (gst_static_caps_get(&q3gp_caps))
2771
2772 static const gchar *
2773 q3gp_type_find_get_profile (const guint8 * data)
2774 {
2775   switch (GST_MAKE_FOURCC (data[0], data[1], data[2], 0)) {
2776     case GST_MAKE_FOURCC ('3', 'g', 'g', 0):
2777       return "general";
2778     case GST_MAKE_FOURCC ('3', 'g', 'p', 0):
2779       return "basic";
2780     case GST_MAKE_FOURCC ('3', 'g', 's', 0):
2781       return "streaming-server";
2782     case GST_MAKE_FOURCC ('3', 'g', 'r', 0):
2783       return "progressive-download";
2784     default:
2785       break;
2786   }
2787   return NULL;
2788 }
2789
2790 static void
2791 q3gp_type_find (GstTypeFind * tf, gpointer unused)
2792 {
2793   const gchar *profile;
2794   guint32 ftyp_size = 0;
2795   gint offset = 0;
2796   const guint8 *data = NULL;
2797
2798   if ((data = gst_type_find_peek (tf, 0, 12)) == NULL) {
2799     return;
2800   }
2801
2802   data += 4;
2803   if (memcmp (data, "ftyp", 4) != 0) {
2804     return;
2805   }
2806
2807   /* check major brand */
2808   data += 4;
2809   if ((profile = q3gp_type_find_get_profile (data))) {
2810     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM,
2811         "application/x-3gp", "profile", G_TYPE_STRING, profile, NULL);
2812     return;
2813   }
2814
2815   /* check compatible brands */
2816   if ((data = gst_type_find_peek (tf, 0, 4)) != NULL) {
2817     ftyp_size = GST_READ_UINT32_BE (data);
2818   }
2819   for (offset = 16; offset < ftyp_size; offset += 4) {
2820     if ((data = gst_type_find_peek (tf, offset, 3)) == NULL) {
2821       break;
2822     }
2823     if ((profile = q3gp_type_find_get_profile (data))) {
2824       gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM,
2825           "application/x-3gp", "profile", G_TYPE_STRING, profile, NULL);
2826       return;
2827     }
2828   }
2829
2830   return;
2831
2832 }
2833
2834 /*** video/mj2 and image/jp2 ***/
2835 static GstStaticCaps mj2_caps = GST_STATIC_CAPS ("video/mj2");
2836
2837 #define MJ2_CAPS gst_static_caps_get(&mj2_caps)
2838
2839 static GstStaticCaps jp2_caps = GST_STATIC_CAPS ("image/jp2");
2840
2841 #define JP2_CAPS gst_static_caps_get(&jp2_caps)
2842
2843 static void
2844 jp2_type_find (GstTypeFind * tf, gpointer unused)
2845 {
2846   const guint8 *data;
2847
2848   data = gst_type_find_peek (tf, 0, 24);
2849   if (!data)
2850     return;
2851
2852   /* jp2 signature */
2853   if (memcmp (data, "\000\000\000\014jP  \015\012\207\012", 12) != 0)
2854     return;
2855
2856   /* check ftyp box */
2857   data += 12;
2858   if (memcmp (data + 4, "ftyp", 4) == 0) {
2859     if (memcmp (data + 8, "jp2 ", 4) == 0)
2860       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, JP2_CAPS);
2861     else if (memcmp (data + 8, "mjp2", 4) == 0)
2862       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MJ2_CAPS);
2863   }
2864 }
2865
2866 /*** video/quicktime ***/
2867
2868 static GstStaticCaps qt_caps = GST_STATIC_CAPS ("video/quicktime");
2869
2870 #define QT_CAPS gst_static_caps_get(&qt_caps)
2871 #define STRNCMP(x,y,z) (strncmp ((char*)(x), (char*)(y), z))
2872
2873 /* FIXME 0.11: go through http://www.ftyps.com/ */
2874 static void
2875 qt_type_find (GstTypeFind * tf, gpointer unused)
2876 {
2877   const guint8 *data;
2878   guint tip = 0;
2879   guint64 offset = 0;
2880   guint64 size;
2881   const gchar *variant = NULL;
2882
2883   while ((data = gst_type_find_peek (tf, offset, 12)) != NULL) {
2884     guint64 new_offset;
2885
2886     if (STRNCMP (&data[4], "ftypqt  ", 8) == 0) {
2887       tip = GST_TYPE_FIND_MAXIMUM;
2888       break;
2889     }
2890
2891     if (STRNCMP (&data[4], "ftypisom", 8) == 0 ||
2892         STRNCMP (&data[4], "ftypavc1", 8) == 0 ||
2893         STRNCMP (&data[4], "ftypmp42", 8) == 0) {
2894       tip = GST_TYPE_FIND_MAXIMUM;
2895       variant = "iso";
2896       break;
2897     }
2898
2899     if (STRNCMP (&data[4], "ftypisml", 8) == 0) {
2900       tip = GST_TYPE_FIND_MAXIMUM;
2901       variant = "iso-fragmented";
2902       break;
2903     }
2904
2905     /* box/atom types that are in common with ISO base media file format */
2906     if (STRNCMP (&data[4], "moov", 4) == 0 ||
2907         STRNCMP (&data[4], "mdat", 4) == 0 ||
2908         STRNCMP (&data[4], "ftyp", 4) == 0 ||
2909         STRNCMP (&data[4], "free", 4) == 0 ||
2910         STRNCMP (&data[4], "uuid", 4) == 0 ||
2911         STRNCMP (&data[4], "skip", 4) == 0) {
2912       if (tip == 0) {
2913         tip = GST_TYPE_FIND_LIKELY;
2914       } else {
2915         tip = GST_TYPE_FIND_NEARLY_CERTAIN;
2916       }
2917     }
2918     /* other box/atom types, apparently quicktime specific */
2919     else if (STRNCMP (&data[4], "pnot", 4) == 0 ||
2920         STRNCMP (&data[4], "PICT", 4) == 0 ||
2921         STRNCMP (&data[4], "wide", 4) == 0 ||
2922         STRNCMP (&data[4], "prfl", 4) == 0) {
2923       tip = GST_TYPE_FIND_MAXIMUM;
2924       break;
2925     } else {
2926       tip = 0;
2927       break;
2928     }
2929
2930     size = GST_READ_UINT32_BE (data);
2931     /* check compatible brands rather than ever expaning major brands above */
2932     if ((STRNCMP (&data[4], "ftyp", 4) == 0) && (size >= 16)) {
2933       new_offset = offset + 12;
2934       while (new_offset + 4 <= offset + size) {
2935         data = gst_type_find_peek (tf, new_offset, 4);
2936         if (data == NULL)
2937           goto done;
2938         if (STRNCMP (&data[4], "isom", 4) == 0 ||
2939             STRNCMP (&data[4], "avc1", 4) == 0 ||
2940             STRNCMP (&data[4], "mp41", 4) == 0 ||
2941             STRNCMP (&data[4], "mp42", 4) == 0) {
2942           tip = GST_TYPE_FIND_MAXIMUM;
2943           variant = "iso";
2944           goto done;
2945         }
2946         new_offset += 4;
2947       }
2948     }
2949     if (size == 1) {
2950       const guint8 *sizedata;
2951
2952       sizedata = gst_type_find_peek (tf, offset + 8, 8);
2953       if (sizedata == NULL)
2954         break;
2955
2956       size = GST_READ_UINT64_BE (sizedata);
2957     } else {
2958       if (size < 8)
2959         break;
2960     }
2961     new_offset = offset + size;
2962     if (new_offset <= offset)
2963       break;
2964     offset = new_offset;
2965   }
2966
2967 done:
2968   if (tip > 0) {
2969     if (variant) {
2970       GstCaps *caps = gst_caps_copy (QT_CAPS);
2971
2972       gst_caps_set_simple (caps, "variant", G_TYPE_STRING, variant, NULL);
2973       gst_type_find_suggest (tf, tip, caps);
2974       gst_caps_unref (caps);
2975     } else {
2976       gst_type_find_suggest (tf, tip, QT_CAPS);
2977     }
2978   }
2979 };
2980
2981
2982 /*** image/x-quicktime ***/
2983
2984 static GstStaticCaps qtif_caps = GST_STATIC_CAPS ("image/x-quicktime");
2985
2986 #define QTIF_CAPS gst_static_caps_get(&qtif_caps)
2987
2988 /* how many atoms we check before we give up */
2989 #define QTIF_MAXROUNDS 25
2990
2991 static void
2992 qtif_type_find (GstTypeFind * tf, gpointer unused)
2993 {
2994   const guint8 *data;
2995   gboolean found_idsc = FALSE;
2996   gboolean found_idat = FALSE;
2997   guint64 offset = 0;
2998   guint rounds = 0;
2999
3000   while ((data = gst_type_find_peek (tf, offset, 8)) != NULL) {
3001     guint64 size;
3002
3003     size = GST_READ_UINT32_BE (data);
3004     if (size == 1) {
3005       const guint8 *sizedata;
3006
3007       sizedata = gst_type_find_peek (tf, offset + 8, 8);
3008       if (sizedata == NULL)
3009         break;
3010
3011       size = GST_READ_UINT64_BE (sizedata);
3012     }
3013     if (size < 8)
3014       break;
3015
3016     if (STRNCMP (data + 4, "idsc", 4) == 0)
3017       found_idsc = TRUE;
3018     if (STRNCMP (data + 4, "idat", 4) == 0)
3019       found_idat = TRUE;
3020
3021     if (found_idsc && found_idat) {
3022       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, QTIF_CAPS);
3023       return;
3024     }
3025
3026     offset += size;
3027     if (++rounds > QTIF_MAXROUNDS)
3028       break;
3029   }
3030
3031   if (found_idsc || found_idat) {
3032     gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, QTIF_CAPS);
3033     return;
3034   }
3035 };
3036
3037 /*** audio/x-mod ***/
3038
3039 static GstStaticCaps mod_caps = GST_STATIC_CAPS ("audio/x-mod");
3040
3041 #define MOD_CAPS gst_static_caps_get(&mod_caps)
3042 /* FIXME: M15 CheckType to do */
3043 static void
3044 mod_type_find (GstTypeFind * tf, gpointer unused)
3045 {
3046   const guint8 *data;
3047
3048   /* MOD */
3049   if ((data = gst_type_find_peek (tf, 1080, 4)) != NULL) {
3050     /* Protracker and variants */
3051     if ((memcmp (data, "M.K.", 4) == 0) || (memcmp (data, "M!K!", 4) == 0) ||
3052         /* Star Tracker */
3053         (memcmp (data, "FLT", 3) == 0 && isdigit (data[3])) ||
3054         (memcmp (data, "EXO", 3) == 0 && isdigit (data[3])) ||
3055         /* Oktalyzer (Amiga) */
3056         (memcmp (data, "OKTA", 4) == 0) ||
3057         /* Oktalyser (Atari) */
3058         (memcmp (data, "CD81", 4) == 0) ||
3059         /* Fasttracker */
3060         (memcmp (data + 1, "CHN", 3) == 0 && isdigit (data[0])) ||
3061         /* Fasttracker or Taketracker */
3062         (memcmp (data + 2, "CH", 2) == 0 && isdigit (data[0])
3063             && isdigit (data[1])) || (memcmp (data + 2, "CN", 2) == 0
3064             && isdigit (data[0]) && isdigit (data[1]))) {
3065       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3066       return;
3067     }
3068   }
3069   /* XM */
3070   if ((data = gst_type_find_peek (tf, 0, 38)) != NULL) {
3071     if (memcmp (data, "Extended Module: ", 17) == 0 && data[37] == 0x1A) {
3072       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3073       return;
3074     }
3075   }
3076   /* OKT */
3077   if (data || (data = gst_type_find_peek (tf, 0, 8)) != NULL) {
3078     if (memcmp (data, "OKTASONG", 8) == 0) {
3079       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3080       return;
3081     }
3082   }
3083   if (data || (data = gst_type_find_peek (tf, 0, 4)) != NULL) {
3084     /* 669 */
3085     if ((memcmp (data, "if", 2) == 0) || (memcmp (data, "JN", 2) == 0)) {
3086       gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, MOD_CAPS);
3087       return;
3088     }
3089     /* AMF */
3090     if ((memcmp (data, "AMF", 3) == 0 && data[3] > 10 && data[3] < 14) ||
3091         /* IT */
3092         (memcmp (data, "IMPM", 4) == 0) ||
3093         /* MED */
3094         (memcmp (data, "MMD0", 4) == 0) || (memcmp (data, "MMD1", 4) == 0) ||
3095         /* MTM */
3096         (memcmp (data, "MTM", 3) == 0)) {
3097       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3098       return;
3099     }
3100     /* DSM */
3101     if (memcmp (data, "RIFF", 4) == 0) {
3102       const guint8 *data2 = gst_type_find_peek (tf, 8, 4);
3103
3104       if (data2) {
3105         if (memcmp (data2, "DSMF", 4) == 0) {
3106           gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3107           return;
3108         }
3109       }
3110     }
3111     /* FAM */
3112     if (memcmp (data, "FAM\xFE", 4) == 0) {
3113       const guint8 *data2 = gst_type_find_peek (tf, 44, 3);
3114
3115       if (data2) {
3116         if (memcmp (data2, "compare", 3) == 0) {
3117           gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3118           return;
3119         }
3120       } else {
3121         gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, MOD_CAPS);
3122         return;
3123       }
3124     }
3125     /* GDM */
3126     if (memcmp (data, "GDM\xFE", 4) == 0) {
3127       const guint8 *data2 = gst_type_find_peek (tf, 71, 4);
3128
3129       if (data2) {
3130         if (memcmp (data2, "GMFS", 4) == 0) {
3131           gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3132           return;
3133         }
3134       } else {
3135         gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, MOD_CAPS);
3136         return;
3137       }
3138     }
3139   }
3140   /* IMF */
3141   if ((data = gst_type_find_peek (tf, 60, 4)) != NULL) {
3142     if (memcmp (data, "IM10", 4) == 0) {
3143       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3144       return;
3145     }
3146   }
3147   /* S3M */
3148   if ((data = gst_type_find_peek (tf, 44, 4)) != NULL) {
3149     if (memcmp (data, "SCRM", 4) == 0) {
3150       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3151       return;
3152     }
3153   }
3154   /* STM */
3155   if ((data = gst_type_find_peek (tf, 20, 8)) != NULL) {
3156     if (g_ascii_strncasecmp ((gchar *) data, "!Scream!", 8) == 0 ||
3157         g_ascii_strncasecmp ((gchar *) data, "BMOD2STM", 8) == 0) {
3158       const guint8 *id, *stmtype;
3159
3160       if ((id = gst_type_find_peek (tf, 28, 1)) == NULL)
3161         return;
3162       if ((stmtype = gst_type_find_peek (tf, 29, 1)) == NULL)
3163         return;
3164       if (*id == 0x1A && *stmtype == 2)
3165         gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3166       return;
3167     }
3168   }
3169   /* AMF */
3170   if ((data = gst_type_find_peek (tf, 0, 19)) != NULL) {
3171     if (memcmp (data, "ASYLUM Music Format", 19) == 0) {
3172       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MOD_CAPS);
3173       return;
3174     }
3175   }
3176 }
3177
3178 /*** application/x-shockwave-flash ***/
3179
3180 static GstStaticCaps swf_caps =
3181 GST_STATIC_CAPS ("application/x-shockwave-flash");
3182 #define SWF_CAPS (gst_static_caps_get(&swf_caps))
3183 static void
3184 swf_type_find (GstTypeFind * tf, gpointer unused)
3185 {
3186   const guint8 *data = gst_type_find_peek (tf, 0, 4);
3187
3188   if (data && (data[0] == 'F' || data[0] == 'C') &&
3189       data[1] == 'W' && data[2] == 'S') {
3190     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SWF_CAPS);
3191   }
3192 }
3193
3194 /*** image/jpeg ***/
3195
3196 #define JPEG_MARKER_IS_START_OF_FRAME(x) \
3197     ((x)>=0xc0 && (x) <= 0xcf && (x)!=0xc4 && (x)!=0xc8 && (x)!=0xcc)
3198
3199 static GstStaticCaps jpeg_caps = GST_STATIC_CAPS ("image/jpeg");
3200
3201 #define JPEG_CAPS (gst_static_caps_get(&jpeg_caps))
3202 static void
3203 jpeg_type_find (GstTypeFind * tf, gpointer unused)
3204 {
3205   GstTypeFindProbability prob = GST_TYPE_FIND_POSSIBLE;
3206   DataScanCtx c = { 0, NULL, 0 };
3207   GstCaps *caps;
3208   guint num_markers;
3209
3210   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 2)))
3211     return;
3212
3213   if (c.data[0] != 0xff || c.data[1] != 0xd8)
3214     return;
3215
3216   num_markers = 1;
3217   data_scan_ctx_advance (tf, &c, 2);
3218
3219   caps = gst_caps_copy (JPEG_CAPS);
3220
3221   while (data_scan_ctx_ensure_data (tf, &c, 4) && c.offset < (200 * 1024)) {
3222     guint16 len;
3223     guint8 marker;
3224
3225     if (c.data[0] != 0xff)
3226       break;
3227
3228     marker = c.data[1];
3229     if (G_UNLIKELY (marker == 0xff)) {
3230       data_scan_ctx_advance (tf, &c, 1);
3231       continue;
3232     }
3233
3234     data_scan_ctx_advance (tf, &c, 2);
3235
3236     /* we assume all markers we'll see before SOF have a payload length; if
3237      * that's not the case we'll just detect a false sync and bail out, but
3238      * still report POSSIBLE probability */
3239     len = GST_READ_UINT16_BE (c.data);
3240
3241     GST_LOG ("possible JPEG marker 0x%02x (@0x%04x), segment length %u",
3242         marker, (guint) c.offset, len);
3243
3244     if (!data_scan_ctx_ensure_data (tf, &c, len))
3245       break;
3246
3247     if (marker == 0xc4 ||       /* DEFINE_HUFFMAN_TABLES          */
3248         marker == 0xcc ||       /* DEFINE_ARITHMETIC_CONDITIONING */
3249         marker == 0xdb ||       /* DEFINE_QUANTIZATION_TABLES     */
3250         marker == 0xdd ||       /* DEFINE_RESTART_INTERVAL        */
3251         marker == 0xfe) {       /* COMMENT                        */
3252       data_scan_ctx_advance (tf, &c, len);
3253       ++num_markers;
3254     } else if (marker == 0xe0 && len >= (2 + 4) &&      /* APP0 */
3255         data_scan_ctx_memcmp (tf, &c, 2, "JFIF", 4)) {
3256       GST_LOG ("found JFIF tag");
3257       prob = GST_TYPE_FIND_MAXIMUM;
3258       data_scan_ctx_advance (tf, &c, len);
3259       ++num_markers;
3260       /* we continue until we find a start of frame marker */
3261     } else if (marker == 0xe1 && len >= (2 + 4) &&      /* APP1 */
3262         data_scan_ctx_memcmp (tf, &c, 2, "Exif", 4)) {
3263       GST_LOG ("found Exif tag");
3264       prob = GST_TYPE_FIND_MAXIMUM;
3265       data_scan_ctx_advance (tf, &c, len);
3266       ++num_markers;
3267       /* we continue until we find a start of frame marker */
3268     } else if (marker >= 0xe0 && marker <= 0xef) {      /* APPn */
3269       data_scan_ctx_advance (tf, &c, len);
3270       ++num_markers;
3271     } else if (JPEG_MARKER_IS_START_OF_FRAME (marker) && len >= (2 + 8)) {
3272       int h, w;
3273
3274       h = GST_READ_UINT16_BE (c.data + 2 + 1);
3275       w = GST_READ_UINT16_BE (c.data + 2 + 1 + 2);
3276       if (h == 0 || w == 0) {
3277         GST_WARNING ("bad width %u and/or height %u in SOF header", w, h);
3278         break;
3279       }
3280
3281       GST_LOG ("SOF at offset %" G_GUINT64_FORMAT ", num_markers=%d, "
3282           "WxH=%dx%d", c.offset - 2, num_markers, w, h);
3283
3284       if (num_markers >= 5 || prob == GST_TYPE_FIND_MAXIMUM)
3285         prob = GST_TYPE_FIND_MAXIMUM;
3286       else
3287         prob = GST_TYPE_FIND_LIKELY;
3288
3289       gst_caps_set_simple (caps, "width", G_TYPE_INT, w,
3290           "height", G_TYPE_INT, h, "sof-marker", G_TYPE_INT, marker & 0xf,
3291           NULL);
3292
3293       break;
3294     } else {
3295       GST_WARNING ("bad length or unexpected JPEG marker 0xff 0x%02x", marker);
3296       break;
3297     }
3298   }
3299
3300   gst_type_find_suggest (tf, prob, caps);
3301   gst_caps_unref (caps);
3302 }
3303
3304 /*** image/bmp ***/
3305
3306 static GstStaticCaps bmp_caps = GST_STATIC_CAPS ("image/bmp");
3307
3308 #define BMP_CAPS (gst_static_caps_get(&bmp_caps))
3309 static void
3310 bmp_type_find (GstTypeFind * tf, gpointer unused)
3311 {
3312   DataScanCtx c = { 0, NULL, 0 };
3313   guint32 struct_size, w, h, planes, bpp;
3314
3315   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 54)))
3316     return;
3317
3318   if (c.data[0] != 'B' || c.data[1] != 'M')
3319     return;
3320
3321   /* skip marker + size */
3322   data_scan_ctx_advance (tf, &c, 2 + 4);
3323
3324   /* reserved, must be 0 */
3325   if (c.data[0] != 0 || c.data[1] != 0 || c.data[2] != 0 || c.data[3] != 0)
3326     return;
3327
3328   data_scan_ctx_advance (tf, &c, 2 + 2);
3329
3330   /* offset to start of image data in bytes (check for sanity) */
3331   GST_LOG ("offset=%u", GST_READ_UINT32_LE (c.data));
3332   if (GST_READ_UINT32_LE (c.data) > (10 * 1024 * 1024))
3333     return;
3334
3335   struct_size = GST_READ_UINT32_LE (c.data + 4);
3336   GST_LOG ("struct_size=%u", struct_size);
3337
3338   data_scan_ctx_advance (tf, &c, 4 + 4);
3339
3340   if (struct_size == 0x0C) {
3341     w = GST_READ_UINT16_LE (c.data);
3342     h = GST_READ_UINT16_LE (c.data + 2);
3343     planes = GST_READ_UINT16_LE (c.data + 2 + 2);
3344     bpp = GST_READ_UINT16_LE (c.data + 2 + 2 + 2);
3345   } else if (struct_size == 40 || struct_size == 64 || struct_size == 108
3346       || struct_size == 124 || struct_size == 0xF0) {
3347     w = GST_READ_UINT32_LE (c.data);
3348     h = GST_READ_UINT32_LE (c.data + 4);
3349     planes = GST_READ_UINT16_LE (c.data + 4 + 4);
3350     bpp = GST_READ_UINT16_LE (c.data + 4 + 4 + 2);
3351   } else {
3352     return;
3353   }
3354
3355   /* image sizes sanity check */
3356   GST_LOG ("w=%u, h=%u, planes=%u, bpp=%u", w, h, planes, bpp);
3357   if (w == 0 || w > 0xfffff || h == 0 || h > 0xfffff || planes != 1 ||
3358       (bpp != 1 && bpp != 4 && bpp != 8 && bpp != 16 && bpp != 24 && bpp != 32))
3359     return;
3360
3361   gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM, "image/bmp",
3362       "width", G_TYPE_INT, w, "height", G_TYPE_INT, h, "bpp", G_TYPE_INT, bpp,
3363       NULL);
3364 }
3365
3366 /*** image/tiff ***/
3367 static GstStaticCaps tiff_caps = GST_STATIC_CAPS ("image/tiff, "
3368     "endianness = (int) { BIG_ENDIAN, LITTLE_ENDIAN }");
3369 #define TIFF_CAPS (gst_static_caps_get(&tiff_caps))
3370 static GstStaticCaps tiff_be_caps = GST_STATIC_CAPS ("image/tiff, "
3371     "endianness = (int) BIG_ENDIAN");
3372 #define TIFF_BE_CAPS (gst_static_caps_get(&tiff_be_caps))
3373 static GstStaticCaps tiff_le_caps = GST_STATIC_CAPS ("image/tiff, "
3374     "endianness = (int) LITTLE_ENDIAN");
3375 #define TIFF_LE_CAPS (gst_static_caps_get(&tiff_le_caps))
3376 static void
3377 tiff_type_find (GstTypeFind * tf, gpointer ununsed)
3378 {
3379   const guint8 *data = gst_type_find_peek (tf, 0, 8);
3380   guint8 le_header[4] = { 0x49, 0x49, 0x2A, 0x00 };
3381   guint8 be_header[4] = { 0x4D, 0x4D, 0x00, 0x2A };
3382
3383   if (data) {
3384     if (memcmp (data, le_header, 4) == 0) {
3385       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, TIFF_LE_CAPS);
3386     } else if (memcmp (data, be_header, 4) == 0) {
3387       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, TIFF_BE_CAPS);
3388     }
3389   }
3390 }
3391
3392 /*** PNM ***/
3393
3394 static GstStaticCaps pnm_caps = GST_STATIC_CAPS ("image/x-portable-bitmap; "
3395     "image/x-portable-graymap; image/x-portable-pixmap; "
3396     "image/x-portable-anymap");
3397
3398 #define PNM_CAPS (gst_static_caps_get(&pnm_caps))
3399
3400 #define IS_PNM_WHITESPACE(c) \
3401     ((c) == ' ' || (c) == '\r' || (c) == '\n' || (c) == 't')
3402
3403 static void
3404 pnm_type_find (GstTypeFind * tf, gpointer ununsed)
3405 {
3406   const gchar *media_type = NULL;
3407   DataScanCtx c = { 0, NULL, 0 };
3408   guint h = 0, w = 0;
3409
3410   if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 16)))
3411     return;
3412
3413   /* see http://en.wikipedia.org/wiki/Netpbm_format */
3414   if (c.data[0] != 'P' || c.data[1] < '1' || c.data[1] > '7' ||
3415       !IS_PNM_WHITESPACE (c.data[2]) ||
3416       (c.data[3] != '#' && c.data[3] < '0' && c.data[3] > '9'))
3417     return;
3418
3419   switch (c.data[1]) {
3420     case '1':
3421       media_type = "image/x-portable-bitmap";   /* ASCII */
3422       break;
3423     case '2':
3424       media_type = "image/x-portable-graymap";  /* ASCII */
3425       break;
3426     case '3':
3427       media_type = "image/x-portable-pixmap";   /* ASCII */
3428       break;
3429     case '4':
3430       media_type = "image/x-portable-bitmap";   /* Raw */
3431       break;
3432     case '5':
3433       media_type = "image/x-portable-graymap";  /* Raw */
3434       break;
3435     case '6':
3436       media_type = "image/x-portable-pixmap";   /* Raw */
3437       break;
3438     case '7':
3439       media_type = "image/x-portable-anymap";
3440       break;
3441     default:
3442       g_return_if_reached ();
3443   }
3444
3445   /* try to extract width and height as well */
3446   if (c.data[1] != '7') {
3447     gchar s[64] = { 0, }
3448     , sep1, sep2;
3449
3450     /* need to skip any comment lines first */
3451     data_scan_ctx_advance (tf, &c, 3);
3452     while (c.data[0] == '#') {  /* we know there's still data left */
3453       data_scan_ctx_advance (tf, &c, 1);
3454       while (c.data[0] != '\n' && c.data[0] != '\r') {
3455         if (!data_scan_ctx_ensure_data (tf, &c, 4))
3456           return;
3457         data_scan_ctx_advance (tf, &c, 1);
3458       }
3459       data_scan_ctx_advance (tf, &c, 1);
3460       GST_LOG ("skipped comment line in PNM header");
3461     }
3462
3463     if (!data_scan_ctx_ensure_data (tf, &c, 32) &&
3464         !data_scan_ctx_ensure_data (tf, &c, 4)) {
3465       return;
3466     }
3467
3468     /* need to NUL-terminate data for sscanf */
3469     memcpy (s, c.data, MIN (sizeof (s) - 1, c.size));
3470     if (sscanf (s, "%u%c%u%c", &w, &sep1, &h, &sep2) == 4 &&
3471         IS_PNM_WHITESPACE (sep1) && IS_PNM_WHITESPACE (sep2) &&
3472         w > 0 && w < G_MAXINT && h > 0 && h < G_MAXINT) {
3473       GST_LOG ("extracted PNM width and height: %dx%d", w, h);
3474     } else {
3475       w = 0;
3476       h = 0;
3477     }
3478   } else {
3479     /* FIXME: extract width + height for anymaps too */
3480   }
3481
3482   if (w > 0 && h > 0) {
3483     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM, media_type,
3484         "width", G_TYPE_INT, w, "height", G_TYPE_INT, h, NULL);
3485   } else {
3486     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_LIKELY, media_type, NULL);
3487   }
3488 }
3489
3490 static GstStaticCaps sds_caps = GST_STATIC_CAPS ("audio/x-sds");
3491
3492 #define SDS_CAPS (gst_static_caps_get(&sds_caps))
3493 static void
3494 sds_type_find (GstTypeFind * tf, gpointer ununsed)
3495 {
3496   const guint8 *data = gst_type_find_peek (tf, 0, 4);
3497   guint8 mask[4] = { 0xFF, 0xFF, 0x80, 0xFF };
3498   guint8 match[4] = { 0xF0, 0x7E, 0, 0x01 };
3499   gint x;
3500
3501   if (data) {
3502     for (x = 0; x < 4; x++) {
3503       if ((data[x] & mask[x]) != match[x]) {
3504         return;
3505       }
3506     }
3507     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SDS_CAPS);
3508   }
3509 }
3510
3511 static GstStaticCaps ircam_caps = GST_STATIC_CAPS ("audio/x-ircam");
3512
3513 #define IRCAM_CAPS (gst_static_caps_get(&ircam_caps))
3514 static void
3515 ircam_type_find (GstTypeFind * tf, gpointer ununsed)
3516 {
3517   const guint8 *data = gst_type_find_peek (tf, 0, 4);
3518   guint8 mask[4] = { 0xFF, 0xFF, 0xF8, 0xFF };
3519   guint8 match[4] = { 0x64, 0xA3, 0x00, 0x00 };
3520   gint x;
3521   gboolean matched = TRUE;
3522
3523   if (!data) {
3524     return;
3525   }
3526   for (x = 0; x < 4; x++) {
3527     if ((data[x] & mask[x]) != match[x]) {
3528       matched = FALSE;
3529     }
3530   }
3531   if (matched) {
3532     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, IRCAM_CAPS);
3533     return;
3534   }
3535   /* now try the reverse version */
3536   matched = TRUE;
3537   for (x = 0; x < 4; x++) {
3538     if ((data[x] & mask[3 - x]) != match[3 - x]) {
3539       matched = FALSE;
3540     }
3541   }
3542 }
3543
3544 /*** Matroska/WebM ***/
3545
3546 #define EBML_HEADER           0x1A45DFA3
3547 #define EBML_VERSION          0x4286
3548 #define EBML_DOCTYPE          0x4282
3549 #define EBML_DOCTYPE_VERSION  0x4287
3550 #define MATROSKA_SEGMENT      0x18538067
3551 #define MATROSKA_CLUSTER      0x1F43B675
3552 #define MATROSKA_TRACKS       0x1654AE6B
3553 #define MATROSKA_TRACK_ENTRY  0xAE
3554 #define MATROSKA_TRACK_TYPE   0x83
3555 #define MATROSKA_STEREO_MODE  0x53B8
3556
3557 #define EBML_MAX_LEN (2 * 1024 * 1024)
3558
3559 typedef enum
3560 {
3561   EBML_DOCTYPE_UNKNOWN = 0,
3562   EBML_DOCTYPE_MATROSKA,
3563   EBML_DOCTYPE_WEBM
3564 } GstEbmlDocType;
3565
3566 typedef struct
3567 {
3568   GstEbmlDocType doctype;
3569   guint audio;
3570   guint video;
3571   guint other;
3572   guint video_stereo;
3573   guint chunks;
3574   guint tracks_ok;              /* if we've seen and fully parsed the TRACKS element */
3575 } GstMatroskaInfo;
3576
3577 static inline guint
3578 ebml_read_chunk_header (GstTypeFind * tf, DataScanCtx * c, guint max_size,
3579     guint32 * id, guint64 * size)
3580 {
3581   guint64 mask;
3582   guint msbit_set, i, len, id_len;
3583
3584   if (c->size < 12 || max_size < 1)
3585     return 0;
3586
3587   /* element ID */
3588   *id = c->data[0];
3589   if ((c->data[0] & 0x80) == 0x80) {
3590     id_len = 1;
3591   } else if ((c->data[0] & 0xC0) == 0x40) {
3592     id_len = 2;
3593   } else if ((c->data[0] & 0xE0) == 0x20) {
3594     id_len = 3;
3595   } else if ((c->data[0] & 0xF0) == 0x10) {
3596     id_len = 4;
3597   } else {
3598     return 0;
3599   }
3600
3601   if (max_size < id_len)
3602     return 0;
3603
3604   for (i = 1; i < id_len; ++i) {
3605     *id = (*id << 8) | c->data[i];
3606   }
3607
3608   data_scan_ctx_advance (tf, c, id_len);
3609   max_size -= id_len;
3610
3611   /* size */
3612   if (max_size < 1 || c->data[0] == 0)
3613     return 0;
3614
3615   msbit_set = g_bit_nth_msf (c->data[0], 8);
3616   mask = ((1 << msbit_set) - 1);
3617   *size = c->data[0] & mask;
3618   len = 7 - msbit_set;
3619
3620   if (max_size < 1 + len)
3621     return 0;
3622   for (i = 0; i < len; ++i) {
3623     mask = (mask << 8) | 0xff;
3624     *size = (*size << 8) | c->data[1 + i];
3625   }
3626
3627   data_scan_ctx_advance (tf, c, 1 + len);
3628
3629   /* undefined/unknown size? (all bits 1) */
3630   if (*size == mask) {
3631     /* allow unknown size for SEGMENT chunk, bail out otherwise */
3632     if (*id == MATROSKA_SEGMENT)
3633       *size = G_MAXUINT64;
3634     else
3635       return 0;
3636   }
3637
3638   return id_len + (1 + len);
3639 }
3640
3641 static gboolean
3642 ebml_parse_chunk (GstTypeFind * tf, DataScanCtx * ctx, guint32 chunk_id,
3643     guint chunk_size, GstMatroskaInfo * info, guint depth)
3644 {                               /* FIXME: make sure input size is clipped to 32 bit */
3645   static const gchar SPACES[] = "                ";
3646   DataScanCtx c = *ctx;
3647   guint64 element_size;
3648   guint32 id, hdr_len;
3649
3650   if (depth >= 8)               /* keep SPACES large enough for depth */
3651     return FALSE;
3652
3653   while (chunk_size > 0) {
3654     if (c.offset > EBML_MAX_LEN || !data_scan_ctx_ensure_data (tf, &c, 64))
3655       return FALSE;
3656
3657     hdr_len = ebml_read_chunk_header (tf, &c, chunk_size, &id, &element_size);
3658     if (hdr_len == 0)
3659       return FALSE;
3660
3661     g_assert (hdr_len <= chunk_size);
3662     chunk_size -= hdr_len;
3663
3664     if (element_size > chunk_size)
3665       return FALSE;
3666
3667     GST_DEBUG ("%s %08x, size %" G_GUINT64_FORMAT " / %" G_GUINT64_FORMAT,
3668         SPACES + sizeof (SPACES) - 1 - (2 * depth), id, element_size,
3669         hdr_len + element_size);
3670
3671     if (!data_scan_ctx_ensure_data (tf, &c, element_size)) {
3672       GST_DEBUG ("not enough data");
3673       return FALSE;
3674     }
3675
3676     switch (id) {
3677       case EBML_DOCTYPE:
3678         if (element_size >= 8 && memcmp (c.data, "matroska", 8) == 0)
3679           info->doctype = EBML_DOCTYPE_MATROSKA;
3680         else if (element_size >= 4 && memcmp (c.data, "webm", 4) == 0)
3681           info->doctype = EBML_DOCTYPE_WEBM;
3682         break;
3683       case MATROSKA_SEGMENT:
3684         GST_LOG ("parsing segment");
3685         ebml_parse_chunk (tf, &c, id, element_size, info, depth + 1);
3686         GST_LOG ("parsed segment, done");
3687         return FALSE;
3688       case MATROSKA_TRACKS:
3689         GST_LOG ("parsing tracks");
3690         info->tracks_ok =
3691             ebml_parse_chunk (tf, &c, id, element_size, info, depth + 1);
3692         GST_LOG ("parsed tracks: %s, done (after %" G_GUINT64_FORMAT " bytes)",
3693             info->tracks_ok ? "ok" : "FAIL", c.offset + element_size);
3694         return FALSE;
3695       case MATROSKA_TRACK_ENTRY:
3696         GST_LOG ("parsing track entry");
3697         if (!ebml_parse_chunk (tf, &c, id, element_size, info, depth + 1))
3698           return FALSE;
3699         break;
3700       case MATROSKA_TRACK_TYPE:{
3701         guint type = 0, i;
3702
3703         /* is supposed to always be 1-byte, but not everyone's following that */
3704         for (i = 0; i < element_size; ++i)
3705           type = (type << 8) | c.data[i];
3706
3707         GST_DEBUG ("%s   track type %u",
3708             SPACES + sizeof (SPACES) - 1 - (2 * depth), type);
3709
3710         if (type == 1)
3711           ++info->video;
3712         else if (c.data[0] == 2)
3713           ++info->audio;
3714         else
3715           ++info->other;
3716         break;
3717       }
3718       case MATROSKA_STEREO_MODE:
3719         ++info->video_stereo;
3720         break;
3721       case MATROSKA_CLUSTER:
3722         GST_WARNING ("cluster, bailing out (should've found tracks by now)");
3723         return FALSE;
3724       default:
3725         break;
3726     }
3727     data_scan_ctx_advance (tf, &c, element_size);
3728     chunk_size -= element_size;
3729     ++info->chunks;
3730   }
3731
3732   return TRUE;
3733 }
3734
3735 static GstStaticCaps matroska_caps = GST_STATIC_CAPS ("video/x-matroska");
3736
3737 #define MATROSKA_CAPS (gst_static_caps_get(&matroska_caps))
3738 static void
3739 matroska_type_find (GstTypeFind * tf, gpointer ununsed)
3740 {
3741   GstTypeFindProbability prob;
3742   GstMatroskaInfo info = { 0, };
3743   const gchar *type_name;
3744   DataScanCtx c = { 0, NULL, 0 };
3745   gboolean is_audio;
3746   guint64 size;
3747   guint32 id, hdr_len;
3748
3749   if (!data_scan_ctx_ensure_data (tf, &c, 64))
3750     return;
3751
3752   if (GST_READ_UINT32_BE (c.data) != EBML_HEADER)
3753     return;
3754
3755   while (c.offset < EBML_MAX_LEN && data_scan_ctx_ensure_data (tf, &c, 64)) {
3756     hdr_len = ebml_read_chunk_header (tf, &c, c.size, &id, &size);
3757     if (hdr_len == 0)
3758       return;
3759
3760     GST_INFO ("=== top-level chunk %08x, size %" G_GUINT64_FORMAT
3761         " / %" G_GUINT64_FORMAT, id, size, size + hdr_len);
3762
3763     if (!ebml_parse_chunk (tf, &c, id, size, &info, 0))
3764       break;
3765     data_scan_ctx_advance (tf, &c, size);
3766     GST_INFO ("=== done with chunk %08x", id);
3767     if (id == MATROSKA_SEGMENT)
3768       break;
3769   }
3770
3771   GST_INFO ("audio=%u video=%u other=%u chunks=%u doctype=%d all_tracks=%d",
3772       info.audio, info.video, info.other, info.chunks, info.doctype,
3773       info.tracks_ok);
3774
3775   /* perhaps we should bail out if tracks_ok is FALSE and wait for more data?
3776    * (we would need new API to signal this properly and prevent other
3777    * typefinders from taking over the decision then) */
3778   is_audio = (info.audio > 0 && info.video == 0 && info.other == 0);
3779
3780   if (info.doctype == EBML_DOCTYPE_WEBM) {
3781     type_name = (is_audio) ? "audio/webm" : "video/webm";
3782   } else if (info.video > 0 && info.video_stereo) {
3783     type_name = "video/x-matroska-3d";
3784   } else {
3785     type_name = (is_audio) ? "audio/x-matroska" : "video/x-matroska";
3786   }
3787
3788   if (info.doctype == EBML_DOCTYPE_UNKNOWN)
3789     prob = GST_TYPE_FIND_LIKELY;
3790   else
3791     prob = GST_TYPE_FIND_MAXIMUM;
3792
3793   gst_type_find_suggest_simple (tf, prob, type_name, NULL);
3794 }
3795
3796 /*** application/mxf ***/
3797 static GstStaticCaps mxf_caps = GST_STATIC_CAPS ("application/mxf");
3798
3799 #define MXF_MAX_PROBE_LENGTH (1024 * 64)
3800 #define MXF_CAPS (gst_static_caps_get(&mxf_caps))
3801
3802 /*
3803  * MXF files start with a header partition pack key of 16 bytes which is defined
3804  * at SMPTE-377M 6.1. Before this there can be up to 64K of run-in which _must_
3805  * not contain the partition pack key.
3806  */
3807 static void
3808 mxf_type_find (GstTypeFind * tf, gpointer ununsed)
3809 {
3810   static const guint8 partition_pack_key[] =
3811       { 0x06, 0x0e, 0x2b, 0x34, 0x02, 0x05, 0x01, 0x01, 0x0d, 0x01, 0x02, 0x01,
3812     0x01
3813   };
3814   DataScanCtx c = { 0, NULL, 0 };
3815
3816   while (c.offset <= MXF_MAX_PROBE_LENGTH) {
3817     guint i;
3818     if (G_UNLIKELY (!data_scan_ctx_ensure_data (tf, &c, 1024)))
3819       break;
3820
3821     /* look over in chunks of 1kbytes to avoid too much overhead */
3822
3823     for (i = 0; i < 1024 - 16; i++) {
3824       /* Check first byte before calling more expensive memcmp function */
3825       if (G_UNLIKELY (c.data[i] == 0x06
3826               && memcmp (c.data + i, partition_pack_key, 13) == 0)) {
3827         /* Header partition pack? */
3828         if (c.data[i + 13] != 0x02)
3829           goto advance;
3830
3831         /* Partition status */
3832         if (c.data[i + 14] >= 0x05)
3833           goto advance;
3834
3835         /* Reserved, must be 0x00 */
3836         if (c.data[i + 15] != 0x00)
3837           goto advance;
3838
3839         gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, MXF_CAPS);
3840         return;
3841       }
3842     }
3843
3844   advance:
3845     data_scan_ctx_advance (tf, &c, 1024 - 16);
3846   }
3847 }
3848
3849 /*** video/x-dv ***/
3850
3851 static GstStaticCaps dv_caps = GST_STATIC_CAPS ("video/x-dv, "
3852     "systemstream = (boolean) true");
3853 #define DV_CAPS (gst_static_caps_get(&dv_caps))
3854 static void
3855 dv_type_find (GstTypeFind * tf, gpointer private)
3856 {
3857   const guint8 *data;
3858
3859   data = gst_type_find_peek (tf, 0, 5);
3860
3861   /* check for DIF  and DV flag */
3862   if (data && (data[0] == 0x1f) && (data[1] == 0x07) && (data[2] == 0x00)) {
3863     const gchar *format;
3864
3865     if (data[3] & 0x80) {
3866       format = "PAL";
3867     } else {
3868       format = "NTSC";
3869     }
3870
3871     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM, "video/x-dv",
3872         "systemstream", G_TYPE_BOOLEAN, TRUE,
3873         "format", G_TYPE_STRING, format, NULL);
3874   }
3875 }
3876
3877
3878 /*** Ogg variants ***/
3879 static GstStaticCaps ogg_caps =
3880     GST_STATIC_CAPS ("application/ogg;video/ogg;audio/ogg;application/kate");
3881
3882 #define OGG_CAPS (gst_static_caps_get(&ogg_caps))
3883
3884 typedef enum
3885 {
3886   OGG_AUDIO = 0,
3887   OGG_VIDEO,
3888   OGG_KATE,
3889   OGG_OTHER,
3890   OGG_SKELETON,
3891   OGG_ANNODEX,
3892   OGG_NUM
3893 } GstOggStreamType;
3894
3895 static void
3896 ogganx_type_find (GstTypeFind * tf, gpointer private)
3897 {
3898   const gchar *media_type;
3899   DataScanCtx c = { 0, NULL, 0 };
3900   guint ogg_syncs = 0;
3901   guint hdr_count[OGG_NUM] = { 0, };
3902   static const struct
3903   {
3904     const gchar marker[10];
3905     guint8 marker_size;
3906     GstOggStreamType stream_type;
3907   } markers[] = {
3908     {
3909     "\001vorbis", 7, OGG_AUDIO}, {
3910     "\200theora", 7, OGG_VIDEO}, {
3911     "fLaC", 4, OGG_AUDIO}, {
3912     "\177FLAC", 5, OGG_AUDIO}, {
3913     "Speex", 5, OGG_AUDIO}, {
3914     "CMML\0\0\0\0", 8, OGG_OTHER}, {
3915     "PCM     ", 8, OGG_AUDIO}, {
3916     "Annodex", 7, OGG_ANNODEX}, {
3917     "fishead", 7, OGG_SKELETON}, {
3918     "AnxData", 7, OGG_ANNODEX}, {
3919     "CELT    ", 8, OGG_AUDIO}, {
3920     "\200kate\0\0\0", 8, OGG_KATE}, {
3921     "BBCD\0", 5, OGG_VIDEO}, {
3922     "OVP80\1\1", 7, OGG_VIDEO}, {
3923     "OpusHead", 8, OGG_AUDIO}, {
3924     "\001audio\0\0\0", 9, OGG_AUDIO}, {
3925     "\001video\0\0\0", 9, OGG_VIDEO}, {
3926     "\001text\0\0\0", 9, OGG_OTHER}
3927   };
3928
3929   while (c.offset < 4096 && data_scan_ctx_ensure_data (tf, &c, 64)) {
3930     guint size, i;
3931
3932     if (memcmp (c.data, "OggS", 5) != 0)
3933       break;
3934
3935     ++ogg_syncs;
3936
3937     /* check if BOS */
3938     if (c.data[5] != 0x02)
3939       break;
3940
3941     /* headers should only have one segment */
3942     if (c.data[26] != 1)
3943       break;
3944
3945     size = c.data[27];
3946     if (size < 8)
3947       break;
3948
3949     data_scan_ctx_advance (tf, &c, 28);
3950
3951     if (!data_scan_ctx_ensure_data (tf, &c, MAX (size, 8)))
3952       break;
3953
3954     for (i = 0; i < G_N_ELEMENTS (markers); ++i) {
3955       if (memcmp (c.data, markers[i].marker, markers[i].marker_size) == 0) {
3956         ++hdr_count[markers[i].stream_type];
3957         break;
3958       }
3959     }
3960
3961     if (i == G_N_ELEMENTS (markers)) {
3962       GST_MEMDUMP ("unknown Ogg stream marker", c.data, size);
3963       ++hdr_count[OGG_OTHER];
3964     }
3965
3966     data_scan_ctx_advance (tf, &c, size);
3967   }
3968
3969   if (ogg_syncs == 0)
3970     return;
3971
3972   /* We don't bother with annodex types. FIXME: what about XSPF? */
3973   if (hdr_count[OGG_VIDEO] > 0) {
3974     media_type = "video/ogg";
3975   } else if (hdr_count[OGG_AUDIO] > 0) {
3976     media_type = "audio/ogg";
3977   } else if (hdr_count[OGG_KATE] > 0 && hdr_count[OGG_OTHER] == 0) {
3978     media_type = "application/kate";
3979   } else {
3980     media_type = "application/ogg";
3981   }
3982
3983   GST_INFO ("found %s (audio:%u, video:%u, annodex:%u, skeleton:%u, other:%u)",
3984       media_type, hdr_count[OGG_AUDIO], hdr_count[OGG_VIDEO],
3985       hdr_count[OGG_ANNODEX], hdr_count[OGG_SKELETON], hdr_count[OGG_OTHER]);
3986
3987   gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM, media_type, NULL);
3988 }
3989
3990 /*** audio/x-vorbis ***/
3991 static GstStaticCaps vorbis_caps = GST_STATIC_CAPS ("audio/x-vorbis");
3992
3993 #define VORBIS_CAPS (gst_static_caps_get(&vorbis_caps))
3994 static void
3995 vorbis_type_find (GstTypeFind * tf, gpointer private)
3996 {
3997   const guint8 *data = gst_type_find_peek (tf, 0, 30);
3998
3999   if (data) {
4000     guint blocksize_0;
4001     guint blocksize_1;
4002
4003     /* 1 byte packet type (identification=0x01)
4004        6 byte string "vorbis"
4005        4 byte vorbis version */
4006     if (memcmp (data, "\001vorbis\000\000\000\000", 11) != 0)
4007       return;
4008     data += 11;
4009     /* 1 byte channels must be != 0 */
4010     if (data[0] == 0)
4011       return;
4012     data++;
4013     /* 4 byte samplerate must be != 0 */
4014     if (GST_READ_UINT32_LE (data) == 0)
4015       return;
4016     data += 16;
4017     /* blocksize checks */
4018     blocksize_0 = data[0] & 0x0F;
4019     blocksize_1 = (data[0] & 0xF0) >> 4;
4020     if (blocksize_0 > blocksize_1)
4021       return;
4022     if (blocksize_0 < 6 || blocksize_0 > 13)
4023       return;
4024     if (blocksize_1 < 6 || blocksize_1 > 13)
4025       return;
4026     data++;
4027     /* framing bit */
4028     if ((data[0] & 0x01) != 1)
4029       return;
4030     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, VORBIS_CAPS);
4031   }
4032 }
4033
4034 /*** video/x-theora ***/
4035
4036 static GstStaticCaps theora_caps = GST_STATIC_CAPS ("video/x-theora");
4037
4038 #define THEORA_CAPS (gst_static_caps_get(&theora_caps))
4039 static void
4040 theora_type_find (GstTypeFind * tf, gpointer private)
4041 {
4042   const guint8 *data = gst_type_find_peek (tf, 0, 7);   //42);
4043
4044   if (data) {
4045     if (data[0] != 0x80)
4046       return;
4047     if (memcmp (&data[1], "theora", 6) != 0)
4048       return;
4049     /* FIXME: make this more reliable when specs are out */
4050
4051     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, THEORA_CAPS);
4052   }
4053 }
4054
4055 /*** kate ***/
4056 static void
4057 kate_type_find (GstTypeFind * tf, gpointer private)
4058 {
4059   const guint8 *data = gst_type_find_peek (tf, 0, 64);
4060   gchar category[16] = { 0, };
4061
4062   if (G_UNLIKELY (data == NULL))
4063     return;
4064
4065   /* see: http://wiki.xiph.org/index.php/OggKate#Format_specification */
4066   if (G_LIKELY (memcmp (data, "\200kate\0\0\0", 8) != 0))
4067     return;
4068
4069   /* make sure we always have a NUL-terminated string */
4070   memcpy (category, data + 48, 15);
4071   GST_LOG ("kate category: %s", category);
4072   /* canonical categories for subtitles: subtitles, spu-subtitles, SUB, K-SPU */
4073   if (strcmp (category, "subtitles") == 0 || strcmp (category, "SUB") == 0 ||
4074       strcmp (category, "spu-subtitles") == 0 ||
4075       strcmp (category, "K-SPU") == 0) {
4076     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM,
4077         "subtitle/x-kate", NULL);
4078   } else {
4079     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM,
4080         "application/x-kate", NULL);
4081   }
4082 }
4083
4084 /*** application/x-ogm-video or audio***/
4085
4086 static GstStaticCaps ogmvideo_caps =
4087 GST_STATIC_CAPS ("application/x-ogm-video");
4088 #define OGMVIDEO_CAPS (gst_static_caps_get(&ogmvideo_caps))
4089 static void
4090 ogmvideo_type_find (GstTypeFind * tf, gpointer private)
4091 {
4092   const guint8 *data = gst_type_find_peek (tf, 0, 9);
4093
4094   if (data) {
4095     if (memcmp (data, "\001video\000\000\000", 9) != 0)
4096       return;
4097     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, OGMVIDEO_CAPS);
4098   }
4099 }
4100
4101 static GstStaticCaps ogmaudio_caps =
4102 GST_STATIC_CAPS ("application/x-ogm-audio");
4103 #define OGMAUDIO_CAPS (gst_static_caps_get(&ogmaudio_caps))
4104 static void
4105 ogmaudio_type_find (GstTypeFind * tf, gpointer private)
4106 {
4107   const guint8 *data = gst_type_find_peek (tf, 0, 9);
4108
4109   if (data) {
4110     if (memcmp (data, "\001audio\000\000\000", 9) != 0)
4111       return;
4112     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, OGMAUDIO_CAPS);
4113   }
4114 }
4115
4116 static GstStaticCaps ogmtext_caps = GST_STATIC_CAPS ("application/x-ogm-text");
4117
4118 #define OGMTEXT_CAPS (gst_static_caps_get(&ogmtext_caps))
4119 static void
4120 ogmtext_type_find (GstTypeFind * tf, gpointer private)
4121 {
4122   const guint8 *data = gst_type_find_peek (tf, 0, 9);
4123
4124   if (data) {
4125     if (memcmp (data, "\001text\000\000\000\000", 9) != 0)
4126       return;
4127     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, OGMTEXT_CAPS);
4128   }
4129 }
4130
4131 /*** audio/x-speex ***/
4132
4133 static GstStaticCaps speex_caps = GST_STATIC_CAPS ("audio/x-speex");
4134
4135 #define SPEEX_CAPS (gst_static_caps_get(&speex_caps))
4136 static void
4137 speex_type_find (GstTypeFind * tf, gpointer private)
4138 {
4139   const guint8 *data = gst_type_find_peek (tf, 0, 80);
4140
4141   if (data) {
4142     /* 8 byte string "Speex   "
4143        24 byte speex version string + int */
4144     if (memcmp (data, "Speex   ", 8) != 0)
4145       return;
4146     data += 32;
4147
4148     /* 4 byte header size >= 80 */
4149     if (GST_READ_UINT32_LE (data) < 80)
4150       return;
4151     data += 4;
4152
4153     /* 4 byte sample rate <= 48000 */
4154     if (GST_READ_UINT32_LE (data) > 48000)
4155       return;
4156     data += 4;
4157
4158     /* currently there are only 3 speex modes. */
4159     if (GST_READ_UINT32_LE (data) > 3)
4160       return;
4161     data += 12;
4162
4163     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, SPEEX_CAPS);
4164   }
4165 }
4166
4167 /*** audio/x-celt ***/
4168
4169 static GstStaticCaps celt_caps = GST_STATIC_CAPS ("audio/x-celt");
4170
4171 #define CELT_CAPS (gst_static_caps_get(&celt_caps))
4172 static void
4173 celt_type_find (GstTypeFind * tf, gpointer private)
4174 {
4175   const guint8 *data = gst_type_find_peek (tf, 0, 8);
4176
4177   if (data) {
4178     /* 8 byte string "CELT   " */
4179     if (memcmp (data, "CELT    ", 8) != 0)
4180       return;
4181
4182     /* TODO: Check other values of the CELT header */
4183     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, CELT_CAPS);
4184   }
4185 }
4186
4187 /*** application/x-ogg-skeleton ***/
4188 static GstStaticCaps ogg_skeleton_caps =
4189 GST_STATIC_CAPS ("application/x-ogg-skeleton, parsed=(boolean)FALSE");
4190 #define OGG_SKELETON_CAPS (gst_static_caps_get(&ogg_skeleton_caps))
4191 static void
4192 oggskel_type_find (GstTypeFind * tf, gpointer private)
4193 {
4194   const guint8 *data = gst_type_find_peek (tf, 0, 12);
4195
4196   if (data) {
4197     /* 8 byte string "fishead\0" for the ogg skeleton stream */
4198     if (memcmp (data, "fishead\0", 8) != 0)
4199       return;
4200     data += 8;
4201
4202     /* Require that the header contains version 3.0 */
4203     if (GST_READ_UINT16_LE (data) != 3)
4204       return;
4205     data += 2;
4206     if (GST_READ_UINT16_LE (data) != 0)
4207       return;
4208
4209     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, OGG_SKELETON_CAPS);
4210   }
4211 }
4212
4213 static GstStaticCaps cmml_caps = GST_STATIC_CAPS ("text/x-cmml");
4214
4215 #define CMML_CAPS (gst_static_caps_get(&cmml_caps))
4216 static void
4217 cmml_type_find (GstTypeFind * tf, gpointer private)
4218 {
4219   /* Header is 12 bytes minimum (though we don't check the minor version */
4220   const guint8 *data = gst_type_find_peek (tf, 0, 12);
4221
4222   if (data) {
4223
4224     /* 8 byte string "CMML\0\0\0\0" for the magic number */
4225     if (memcmp (data, "CMML\0\0\0\0", 8) != 0)
4226       return;
4227     data += 8;
4228
4229     /* Require that the header contains at least version 2.0 */
4230     if (GST_READ_UINT16_LE (data) < 2)
4231       return;
4232
4233     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, CMML_CAPS);
4234   }
4235 }
4236
4237 /*** application/x-tar ***/
4238
4239 static GstStaticCaps tar_caps = GST_STATIC_CAPS ("application/x-tar");
4240
4241 #define TAR_CAPS (gst_static_caps_get(&tar_caps))
4242 #define OLDGNU_MAGIC "ustar  "  /* 7 chars and a NUL */
4243 #define NEWGNU_MAGIC "ustar"    /* 5 chars and a NUL */
4244 static void
4245 tar_type_find (GstTypeFind * tf, gpointer unused)
4246 {
4247   const guint8 *data = gst_type_find_peek (tf, 257, 8);
4248
4249   /* of course we are not certain, but we don't want other typefind funcs
4250    * to detect formats of files within the tar archive, e.g. mp3s */
4251   if (data) {
4252     if (memcmp (data, OLDGNU_MAGIC, 8) == 0) {  /* sic */
4253       gst_type_find_suggest (tf, GST_TYPE_FIND_NEARLY_CERTAIN, TAR_CAPS);
4254     } else if (memcmp (data, NEWGNU_MAGIC, 6) == 0 &&   /* sic */
4255         g_ascii_isdigit (data[6]) && g_ascii_isdigit (data[7])) {
4256       gst_type_find_suggest (tf, GST_TYPE_FIND_NEARLY_CERTAIN, TAR_CAPS);
4257     }
4258   }
4259 }
4260
4261 /*** application/x-ar ***/
4262
4263 static GstStaticCaps ar_caps = GST_STATIC_CAPS ("application/x-ar");
4264
4265 #define AR_CAPS (gst_static_caps_get(&ar_caps))
4266 static void
4267 ar_type_find (GstTypeFind * tf, gpointer unused)
4268 {
4269   const guint8 *data = gst_type_find_peek (tf, 0, 24);
4270
4271   if (data && memcmp (data, "!<arch>", 7) == 0) {
4272     gint i;
4273
4274     for (i = 7; i < 24; ++i) {
4275       if (!g_ascii_isprint (data[i]) && data[i] != '\n') {
4276         gst_type_find_suggest (tf, GST_TYPE_FIND_POSSIBLE, AR_CAPS);
4277       }
4278     }
4279
4280     gst_type_find_suggest (tf, GST_TYPE_FIND_NEARLY_CERTAIN, AR_CAPS);
4281   }
4282 }
4283
4284 /*** audio/x-au ***/
4285
4286 /* NOTE: we cannot replace this function with TYPE_FIND_REGISTER_START_WITH,
4287  * as it is only possible to register one typefind factory per 'name'
4288  * (which is in this case the caps), and the first one would be replaced by
4289  * the second one. */
4290 static GstStaticCaps au_caps = GST_STATIC_CAPS ("audio/x-au");
4291
4292 #define AU_CAPS (gst_static_caps_get(&au_caps))
4293 static void
4294 au_type_find (GstTypeFind * tf, gpointer unused)
4295 {
4296   const guint8 *data = gst_type_find_peek (tf, 0, 4);
4297
4298   if (data) {
4299     if (memcmp (data, ".snd", 4) == 0 || memcmp (data, "dns.", 4) == 0) {
4300       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, AU_CAPS);
4301     }
4302   }
4303 }
4304
4305
4306 /*** video/x-nuv ***/
4307
4308 /* NOTE: we cannot replace this function with TYPE_FIND_REGISTER_START_WITH,
4309  * as it is only possible to register one typefind factory per 'name'
4310  * (which is in this case the caps), and the first one would be replaced by
4311  * the second one. */
4312 static GstStaticCaps nuv_caps = GST_STATIC_CAPS ("video/x-nuv");
4313
4314 #define NUV_CAPS (gst_static_caps_get(&nuv_caps))
4315 static void
4316 nuv_type_find (GstTypeFind * tf, gpointer unused)
4317 {
4318   const guint8 *data = gst_type_find_peek (tf, 0, 11);
4319
4320   if (data) {
4321     if (memcmp (data, "MythTVVideo", 11) == 0
4322         || memcmp (data, "NuppelVideo", 11) == 0) {
4323       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, NUV_CAPS);
4324     }
4325   }
4326 }
4327
4328 /*** audio/x-paris ***/
4329 /* NOTE: do not replace this function with two TYPE_FIND_REGISTER_START_WITH */
4330 static GstStaticCaps paris_caps = GST_STATIC_CAPS ("audio/x-paris");
4331
4332 #define PARIS_CAPS (gst_static_caps_get(&paris_caps))
4333 static void
4334 paris_type_find (GstTypeFind * tf, gpointer unused)
4335 {
4336   const guint8 *data = gst_type_find_peek (tf, 0, 4);
4337
4338   if (data) {
4339     if (memcmp (data, " paf", 4) == 0 || memcmp (data, "fap ", 4) == 0) {
4340       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, PARIS_CAPS);
4341     }
4342   }
4343 }
4344
4345 /*** audio/x-sbc ***/
4346 static GstStaticCaps sbc_caps = GST_STATIC_CAPS ("audio/x-sbc");
4347 #define SBC_CAPS (gst_static_caps_get(&sbc_caps))
4348
4349 static gsize
4350 sbc_check_header (const guint8 * data, gsize len, guint * rate,
4351     guint * channels)
4352 {
4353   static const guint16 sbc_rates[4] = { 16000, 32000, 44100, 48000 };
4354   static const guint8 sbc_blocks[4] = { 4, 8, 12, 16 };
4355   guint n_blocks, ch_mode, n_subbands, bitpool;
4356
4357   if (data[0] != 0x9C || len < 4)
4358     return 0;
4359
4360   n_blocks = sbc_blocks[(data[1] >> 4) & 0x03];
4361   ch_mode = (data[1] >> 2) & 0x03;
4362   n_subbands = (data[1] & 0x01) ? 8 : 4;
4363   bitpool = data[2];
4364   if (bitpool < 2)
4365     return 0;
4366
4367   *rate = sbc_rates[(data[1] >> 6) & 0x03];
4368   *channels = (ch_mode == 0) ? 1 : 2;
4369
4370   if (ch_mode == 0)
4371     return 4 + (n_subbands * 1) / 2 + (n_blocks * 1 * bitpool) / 8;
4372   else if (ch_mode == 1)
4373     return 4 + (n_subbands * 2) / 2 + (n_blocks * 2 * bitpool) / 8;
4374   else if (ch_mode == 2)
4375     return 4 + (n_subbands * 2) / 2 + (n_blocks * bitpool) / 8;
4376   else if (ch_mode == 3)
4377     return 4 + (n_subbands * 2) / 2 + (n_subbands + n_blocks * bitpool) / 8;
4378
4379   return 0;
4380 }
4381
4382 static void
4383 sbc_type_find (GstTypeFind * tf, gpointer unused)
4384 {
4385   const guint8 *data;
4386   gsize frame_len;
4387   guint i, rate, channels, offset = 0;
4388
4389   for (i = 0; i < 10; ++i) {
4390     data = gst_type_find_peek (tf, offset, 8);
4391     if (data == NULL)
4392       return;
4393
4394     frame_len = sbc_check_header (data, 8, &rate, &channels);
4395     if (frame_len == 0)
4396       return;
4397
4398     offset += frame_len;
4399   }
4400   gst_type_find_suggest_simple (tf, GST_TYPE_FIND_POSSIBLE, "audio/x-sbc",
4401       "rate", G_TYPE_INT, rate, "channels", G_TYPE_INT, channels,
4402       "parsed", G_TYPE_BOOLEAN, FALSE, NULL);
4403 }
4404
4405 /*** audio/iLBC-sh ***/
4406 /* NOTE: do not replace this function with two TYPE_FIND_REGISTER_START_WITH */
4407 static GstStaticCaps ilbc_caps = GST_STATIC_CAPS ("audio/iLBC-sh");
4408
4409 #define ILBC_CAPS (gst_static_caps_get(&ilbc_caps))
4410 static void
4411 ilbc_type_find (GstTypeFind * tf, gpointer unused)
4412 {
4413   const guint8 *data = gst_type_find_peek (tf, 0, 8);
4414
4415   if (data) {
4416     if (memcmp (data, "#!iLBC30", 8) == 0 || memcmp (data, "#!iLBC20", 8) == 0) {
4417       gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, ILBC_CAPS);
4418     }
4419   }
4420 }
4421
4422 /*** application/x-ms-dos-executable ***/
4423
4424 static GstStaticCaps msdos_caps =
4425 GST_STATIC_CAPS ("application/x-ms-dos-executable");
4426 #define MSDOS_CAPS (gst_static_caps_get(&msdos_caps))
4427 /* see http://www.madchat.org/vxdevl/papers/winsys/pefile/pefile.htm */
4428 static void
4429 msdos_type_find (GstTypeFind * tf, gpointer unused)
4430 {
4431   const guint8 *data = gst_type_find_peek (tf, 0, 64);
4432
4433   if (data && data[0] == 'M' && data[1] == 'Z' &&
4434       GST_READ_UINT16_LE (data + 8) == 4) {
4435     guint32 pe_offset = GST_READ_UINT32_LE (data + 60);
4436
4437     data = gst_type_find_peek (tf, pe_offset, 2);
4438     if (data && data[0] == 'P' && data[1] == 'E') {
4439       gst_type_find_suggest (tf, GST_TYPE_FIND_NEARLY_CERTAIN, MSDOS_CAPS);
4440     }
4441   }
4442 }
4443
4444 /*** application/x-mmsh ***/
4445
4446 static GstStaticCaps mmsh_caps = GST_STATIC_CAPS ("application/x-mmsh");
4447
4448 #define MMSH_CAPS gst_static_caps_get(&mmsh_caps)
4449
4450 /* This is to recognise mssh-over-http */
4451 static void
4452 mmsh_type_find (GstTypeFind * tf, gpointer unused)
4453 {
4454   static const guint8 asf_marker[16] = { 0x30, 0x26, 0xb2, 0x75, 0x8e, 0x66,
4455     0xcf, 0x11, 0xa6, 0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c
4456   };
4457
4458   const guint8 *data;
4459
4460   data = gst_type_find_peek (tf, 0, 2 + 2 + 4 + 2 + 2 + 16);
4461   if (data && data[0] == 0x24 && data[1] == 0x48 &&
4462       GST_READ_UINT16_LE (data + 2) > 2 + 2 + 4 + 2 + 2 + 16 &&
4463       memcmp (data + 2 + 2 + 4 + 2 + 2, asf_marker, 16) == 0) {
4464     gst_type_find_suggest (tf, GST_TYPE_FIND_LIKELY, MMSH_CAPS);
4465   }
4466 }
4467
4468 /*** video/x-dirac ***/
4469
4470 /* NOTE: we cannot replace this function with TYPE_FIND_REGISTER_START_WITH,
4471  * as it is only possible to register one typefind factory per 'name'
4472  * (which is in this case the caps), and the first one would be replaced by
4473  * the second one. */
4474 static GstStaticCaps dirac_caps = GST_STATIC_CAPS ("video/x-dirac");
4475
4476 #define DIRAC_CAPS (gst_static_caps_get(&dirac_caps))
4477 static void
4478 dirac_type_find (GstTypeFind * tf, gpointer unused)
4479 {
4480   const guint8 *data = gst_type_find_peek (tf, 0, 8);
4481
4482   if (data) {
4483     if (memcmp (data, "BBCD", 4) == 0 || memcmp (data, "KW-DIRAC", 8) == 0) {
4484       gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, DIRAC_CAPS);
4485     }
4486   }
4487 }
4488
4489 /*** video/vivo ***/
4490
4491 static GstStaticCaps vivo_caps = GST_STATIC_CAPS ("video/vivo");
4492
4493 #define VIVO_CAPS gst_static_caps_get(&vivo_caps)
4494
4495 static void
4496 vivo_type_find (GstTypeFind * tf, gpointer unused)
4497 {
4498   static const guint8 vivo_marker[] = { 'V', 'e', 'r', 's', 'i', 'o', 'n',
4499     ':', 'V', 'i', 'v', 'o', '/'
4500   };
4501   const guint8 *data;
4502   guint hdr_len, pos;
4503
4504   data = gst_type_find_peek (tf, 0, 1024);
4505   if (data == NULL || data[0] != 0x00)
4506     return;
4507
4508   if ((data[1] & 0x80)) {
4509     if ((data[2] & 0x80))
4510       return;
4511     hdr_len = ((guint) (data[1] & 0x7f)) << 7;
4512     hdr_len += data[2];
4513     if (hdr_len > 2048)
4514       return;
4515     pos = 3;
4516   } else {
4517     hdr_len = data[1];
4518     pos = 2;
4519   }
4520
4521   /* 1008 = 1022 - strlen ("Version:Vivo/") - 1 */
4522   while (pos < 1008 && data[pos] == '\r' && data[pos + 1] == '\n')
4523     pos += 2;
4524
4525   if (memcmp (data + pos, vivo_marker, sizeof (vivo_marker)) == 0) {
4526     gst_type_find_suggest (tf, GST_TYPE_FIND_MAXIMUM, VIVO_CAPS);
4527   }
4528 }
4529
4530 /*** XDG MIME typefinder (to avoid false positives mostly) ***/
4531
4532 #ifdef USE_GIO
4533 static void
4534 xdgmime_typefind (GstTypeFind * find, gpointer user_data)
4535 {
4536   gchar *mimetype;
4537   gsize length = 16384;
4538   guint64 tf_length;
4539   const guint8 *data;
4540   gchar *tmp;
4541
4542   if ((tf_length = gst_type_find_get_length (find)) > 0)
4543     length = MIN (length, tf_length);
4544
4545   if ((data = gst_type_find_peek (find, 0, length)) == NULL)
4546     return;
4547
4548   tmp = g_content_type_guess (NULL, data, length, NULL);
4549   if (tmp == NULL || g_content_type_is_unknown (tmp)) {
4550     g_free (tmp);
4551     return;
4552   }
4553
4554   mimetype = g_content_type_get_mime_type (tmp);
4555   g_free (tmp);
4556
4557   if (mimetype == NULL)
4558     return;
4559
4560   GST_DEBUG ("Got mimetype '%s'", mimetype);
4561
4562   /* Ignore audio/video types:
4563    *  - our own typefinders in -base are likely to be better at this
4564    *    (and if they're not, we really want to fix them, that's why we don't
4565    *    report xdg-detected audio/video types at all, not even with a low
4566    *    probability)
4567    *  - we want to detect GStreamer media types and not MIME types
4568    *  - the purpose of this xdg mime finder is mainly to prevent false
4569    *    positives of non-media formats, not to typefind audio/video formats */
4570   if (g_str_has_prefix (mimetype, "audio/") ||
4571       g_str_has_prefix (mimetype, "video/")) {
4572     GST_LOG ("Ignoring audio/video mime type");
4573     g_free (mimetype);
4574     return;
4575   }
4576
4577   /* Again, we mainly want the xdg typefinding to prevent false-positives on
4578    * non-media formats, so suggest the type with a probability that trumps
4579    * uncertain results of our typefinders, but not more than that. */
4580   GST_LOG ("Suggesting '%s' with probability POSSIBLE", mimetype);
4581   gst_type_find_suggest_simple (find, GST_TYPE_FIND_POSSIBLE, mimetype, NULL);
4582   g_free (mimetype);
4583 }
4584 #endif /* USE_GIO */
4585
4586 /*** Windows icon typefinder (to avoid false positives mostly) ***/
4587
4588 static void
4589 windows_icon_typefind (GstTypeFind * find, gpointer user_data)
4590 {
4591   const guint8 *data;
4592   gint64 datalen;
4593   guint16 type, nimages;
4594   gint32 size, offset;
4595
4596   datalen = gst_type_find_get_length (find);
4597   if ((data = gst_type_find_peek (find, 0, 6)) == NULL)
4598     return;
4599
4600   /* header - simple and not enough to rely on it alone */
4601   if (GST_READ_UINT16_LE (data) != 0)
4602     return;
4603   type = GST_READ_UINT16_LE (data + 2);
4604   if (type != 1 && type != 2)
4605     return;
4606   nimages = GST_READ_UINT16_LE (data + 4);
4607   if (nimages == 0)             /* we can assume we can't have an empty image file ? */
4608     return;
4609
4610   /* first image */
4611   if (data[6 + 3] != 0)
4612     return;
4613   if (type == 1) {
4614     guint16 planes = GST_READ_UINT16_LE (data + 6 + 4);
4615     if (planes > 1)
4616       return;
4617   }
4618   size = GST_READ_UINT32_LE (data + 6 + 8);
4619   offset = GST_READ_UINT32_LE (data + 6 + 12);
4620   if (offset < 0 || size <= 0 || size >= datalen || offset >= datalen
4621       || size + offset > datalen)
4622     return;
4623
4624   gst_type_find_suggest_simple (find, GST_TYPE_FIND_NEARLY_CERTAIN,
4625       "image/x-icon", NULL);
4626 }
4627
4628 /*** WAP WBMP typefinder ***/
4629
4630 static void
4631 wbmp_typefind (GstTypeFind * find, gpointer user_data)
4632 {
4633   const guint8 *data;
4634   gint64 datalen;
4635   guint w, h, size;
4636
4637   /* http://en.wikipedia.org/wiki/Wireless_Application_Protocol_Bitmap_Format */
4638   datalen = gst_type_find_get_length (find);
4639   if (datalen == 0)
4640     return;
4641
4642   data = gst_type_find_peek (find, 0, 5);
4643   if (data == NULL)
4644     return;
4645
4646   /* want 0x00 0x00 at start */
4647   if (*data++ != 0 || *data++ != 0)
4648     return;
4649
4650   /* min header size */
4651   size = 4;
4652
4653   /* let's assume max width/height is 65536 */
4654   w = *data++;
4655   if ((w & 0x80)) {
4656     w = (w << 8) | *data++;
4657     if ((w & 0x80))
4658       return;
4659     ++size;
4660     data = gst_type_find_peek (find, 4, 2);
4661     if (data == NULL)
4662       return;
4663   }
4664   h = *data++;
4665   if ((h & 0x80)) {
4666     h = (h << 8) | *data++;
4667     if ((h & 0x80))
4668       return;
4669     ++size;
4670   }
4671
4672   if (w == 0 || h == 0)
4673     return;
4674
4675   /* now add bitmap size */
4676   size += h * (GST_ROUND_UP_8 (w) / 8);
4677
4678   if (datalen == size) {
4679     gst_type_find_suggest_simple (find, GST_TYPE_FIND_POSSIBLE - 10,
4680         "image/vnd.wap.wbmp", NULL);
4681   }
4682 }
4683
4684 /*** DEGAS Atari images (also to avoid false positives, see #625129) ***/
4685 static void
4686 degas_type_find (GstTypeFind * tf, gpointer private)
4687 {
4688   /* No magic, but it should have a fixed size and a few invalid values */
4689   /* http://www.fileformat.info/format/atari/spec/6ecf9f6eb5be494284a47feb8a214687/view.htm */
4690   gint64 len;
4691   const guint8 *data;
4692   guint16 resolution;
4693   int n;
4694
4695   len = gst_type_find_get_length (tf);
4696   if (len < 34)                 /* smallest header of the lot */
4697     return;
4698   data = gst_type_find_peek (tf, 0, 4);
4699   if (G_UNLIKELY (data == NULL))
4700     return;
4701   resolution = GST_READ_UINT16_BE (data);
4702   if (len == 32034) {
4703     /* could be DEGAS */
4704     if (resolution <= 2)
4705       gst_type_find_suggest_simple (tf, GST_TYPE_FIND_POSSIBLE + 5,
4706           "image/x-degas", NULL);
4707   } else if (len == 32066) {
4708     /* could be DEGAS Elite */
4709     if (resolution <= 2) {
4710       data = gst_type_find_peek (tf, len - 16, 8);
4711       if (G_UNLIKELY (data == NULL))
4712         return;
4713       for (n = 0; n < 4; n++) {
4714         if (GST_READ_UINT16_BE (data + n * 2) > 2)
4715           return;
4716       }
4717       gst_type_find_suggest_simple (tf, GST_TYPE_FIND_POSSIBLE + 5,
4718           "image/x-degas", NULL);
4719     }
4720   } else if (len >= 66 && len < 32066) {
4721     /* could be compressed DEGAS Elite, but it's compressed and so we can't rely on size,
4722        it does have 4 16 bytes values near the end that are 0-2 though. */
4723     if ((resolution & 0x8000) && (resolution & 0x7fff) <= 2) {
4724       data = gst_type_find_peek (tf, len - 16, 8);
4725       if (G_UNLIKELY (data == NULL))
4726         return;
4727       for (n = 0; n < 4; n++) {
4728         if (GST_READ_UINT16_BE (data + n * 2) > 2)
4729           return;
4730       }
4731       gst_type_find_suggest_simple (tf, GST_TYPE_FIND_POSSIBLE + 5,
4732           "image/x-degas", NULL);
4733     }
4734   }
4735 }
4736
4737 /*** DVD ISO images (looks like H.264, see #674069) ***/
4738 static void
4739 dvdiso_type_find (GstTypeFind * tf, gpointer private)
4740 {
4741   /* 0x8000 bytes of zeros, then "\001CD001" */
4742   gint64 len;
4743   const guint8 *data;
4744
4745   len = gst_type_find_get_length (tf);
4746   if (len < 0x8006)
4747     return;
4748   data = gst_type_find_peek (tf, 0, 0x8006);
4749   if (G_UNLIKELY (data == NULL))
4750     return;
4751   for (len = 0; len < 0x8000; len++)
4752     if (data[len])
4753       return;
4754   /* Can the '1' be anything else ? My three samples all have '1'. */
4755   if (memcmp (data + 0x8000, "\001CD001", 6))
4756     return;
4757
4758   /* May need more inspection, we may be able to demux some of them */
4759   gst_type_find_suggest_simple (tf, GST_TYPE_FIND_LIKELY,
4760       "application/octet-stream", NULL);
4761 }
4762
4763 /* SSA/ASS subtitles
4764  *
4765  * http://en.wikipedia.org/wiki/SubStation_Alpha
4766  * http://matroska.org/technical/specs/subtitles/ssa.html
4767  */
4768 static void
4769 ssa_type_find (GstTypeFind * tf, gpointer private)
4770 {
4771   const gchar *start, *end, *ver_str, *media_type = NULL;
4772   const guint8 *data;
4773   gchar *str, *script_type, *p = NULL;
4774   gint64 len;
4775
4776   data = gst_type_find_peek (tf, 0, 32);
4777
4778   if (data == NULL)
4779     return;
4780
4781   /* there might be a BOM at the beginning */
4782   if (memcmp (data, "[Script Info]", 13) != 0 &&
4783       memcmp (data + 2, "[Script Info]", 13) != 0 &&
4784       memcmp (data + 3, "[Script Info]", 13) != 0 &&
4785       memcmp (data + 4, "[Script Info]", 13) != 0) {
4786     return;
4787   }
4788
4789   /* now check if we have SSA or ASS */
4790   len = gst_type_find_get_length (tf);
4791   if (len > 8192)
4792     len = 8192;
4793
4794   data = gst_type_find_peek (tf, 0, len);
4795   if (data == NULL)
4796     return;
4797
4798   /* skip BOM */
4799   start = (gchar *) memchr (data, '[', 5);
4800   g_assert (start);
4801   len -= (start - (gchar *) data);
4802
4803   /* ignore anything non-UTF8 for now, in future we might at least allow
4804    * other UTF variants that are clearly prefixed with the appropriate BOM */
4805   if (!g_utf8_validate (start, len, &end) && (len - (end - start)) > 6) {
4806     GST_FIXME ("non-UTF8 SSA/ASS file");
4807     return;
4808   }
4809
4810   /* something at start,  but not a UTF-8 BOM? */
4811   if (data[0] != '[' && (data[0] != 0xEF || data[1] != 0xBB || data[2] != 0xBF))
4812     return;
4813
4814   /* ignore any partial UTF-8 characters at the end */
4815   len = end - start;
4816
4817   /* create a NUL-terminated string so it's easier to process it safely */
4818   str = g_strndup (start, len - 1);
4819   script_type = strstr (str, "ScriptType:");
4820   if (script_type != NULL) {
4821     gdouble version;
4822
4823     ver_str = script_type + 11;
4824     while (*ver_str == ' ' || *ver_str == 'v' || *ver_str == 'V')
4825       ++ver_str;
4826     version = g_ascii_strtod (ver_str, &p);
4827     if (version == 4.0 && p != NULL && *p == '+')
4828       media_type = "application/x-ass";
4829     else if (version >= 1.0 && version <= 4.0)
4830       media_type = "application/x-ssa";
4831   }
4832
4833   if (media_type == NULL) {
4834     if (strstr (str, "[v4+ Styles]") || strstr (str, "[V4+ Styles]"))
4835       media_type = "application/x-ass";
4836     else if (strstr (str, "[v4 Styles]") || strstr (str, "[V4 Styles]"))
4837       media_type = "application/x-ssa";
4838   }
4839
4840   if (media_type != NULL) {
4841     gst_type_find_suggest_simple (tf, GST_TYPE_FIND_MAXIMUM,
4842         media_type, "parsed", G_TYPE_BOOLEAN, FALSE, NULL);
4843   } else {
4844     GST_WARNING ("could not detect SSA/ASS variant");
4845   }
4846
4847   g_free (str);
4848 }
4849
4850 /*** generic typefind for streams that have some data at a specific position***/
4851 typedef struct
4852 {
4853   const guint8 *data;
4854   guint size;
4855   guint probability;
4856   GstCaps *caps;
4857 }
4858 GstTypeFindData;
4859
4860 static void
4861 start_with_type_find (GstTypeFind * tf, gpointer private)
4862 {
4863   GstTypeFindData *start_with = (GstTypeFindData *) private;
4864   const guint8 *data;
4865
4866   GST_LOG ("trying to find mime type %s with the first %u bytes of data",
4867       gst_structure_get_name (gst_caps_get_structure (start_with->caps, 0)),
4868       start_with->size);
4869   data = gst_type_find_peek (tf, 0, start_with->size);
4870   if (data && memcmp (data, start_with->data, start_with->size) == 0) {
4871     gst_type_find_suggest (tf, start_with->probability, start_with->caps);
4872   }
4873 }
4874
4875 static void
4876 sw_data_destroy (GstTypeFindData * sw_data)
4877 {
4878   if (G_LIKELY (sw_data->caps != NULL))
4879     gst_caps_unref (sw_data->caps);
4880   g_free (sw_data);
4881 }
4882
4883 #define TYPE_FIND_REGISTER_START_WITH(plugin,name,rank,ext,_data,_size,_probability)\
4884 G_BEGIN_DECLS{                                                          \
4885   GstTypeFindData *sw_data = g_new (GstTypeFindData, 1);                \
4886   sw_data->data = (const guint8 *)_data;                                \
4887   sw_data->size = _size;                                                \
4888   sw_data->probability = _probability;                                  \
4889   sw_data->caps = gst_caps_new_empty_simple (name);                     \
4890   if (!gst_type_find_register (plugin, name, rank, start_with_type_find,\
4891                      ext, sw_data->caps, sw_data,                       \
4892                      (GDestroyNotify) (sw_data_destroy))) {             \
4893     gst_caps_unref (sw_data->caps);                                     \
4894     g_free (sw_data);                                                   \
4895   }                                                                     \
4896 }G_END_DECLS
4897
4898 /*** same for riff types ***/
4899
4900 static void
4901 riff_type_find (GstTypeFind * tf, gpointer private)
4902 {
4903   GstTypeFindData *riff_data = (GstTypeFindData *) private;
4904   const guint8 *data = gst_type_find_peek (tf, 0, 12);
4905
4906   if (data && (memcmp (data, "RIFF", 4) == 0 || memcmp (data, "AVF0", 4) == 0)) {
4907     data += 8;
4908     if (memcmp (data, riff_data->data, 4) == 0)
4909       gst_type_find_suggest (tf, riff_data->probability, riff_data->caps);
4910   }
4911 }
4912
4913 #define TYPE_FIND_REGISTER_RIFF(plugin,name,rank,ext,_data)             \
4914 G_BEGIN_DECLS{                                                          \
4915   GstTypeFindData *sw_data = g_new (GstTypeFindData, 1);                \
4916   sw_data->data = (gpointer)_data;                                      \
4917   sw_data->size = 4;                                                    \
4918   sw_data->probability = GST_TYPE_FIND_MAXIMUM;                         \
4919   sw_data->caps = gst_caps_new_empty_simple (name);                     \
4920   if (!gst_type_find_register (plugin, name, rank, riff_type_find,      \
4921                       ext, sw_data->caps, sw_data,                      \
4922                       (GDestroyNotify) (sw_data_destroy))) {            \
4923     gst_caps_unref (sw_data->caps);                                     \
4924     g_free (sw_data);                                                   \
4925   }                                                                     \
4926 }G_END_DECLS
4927
4928
4929 /*** plugin initialization ***/
4930
4931 #define TYPE_FIND_REGISTER(plugin,name,rank,func,ext,caps,priv,notify) \
4932 G_BEGIN_DECLS{\
4933   if (!gst_type_find_register (plugin, name, rank, func, ext, caps, priv, notify))\
4934     return FALSE; \
4935 }G_END_DECLS
4936
4937
4938 static gboolean
4939 plugin_init (GstPlugin * plugin)
4940 {
4941   /* can't initialize this via a struct as caps can't be statically initialized */
4942
4943   GST_DEBUG_CATEGORY_INIT (type_find_debug, "typefindfunctions",
4944       GST_DEBUG_FG_GREEN | GST_DEBUG_BG_RED, "generic type find functions");
4945
4946   /* note: asx/wax/wmx are XML files, asf doesn't handle them */
4947   /* must use strings, macros don't accept initializers */
4948   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-ms-asf", GST_RANK_SECONDARY,
4949       "asf,wm,wma,wmv",
4950       "\060\046\262\165\216\146\317\021\246\331\000\252\000\142\316\154", 16,
4951       GST_TYPE_FIND_MAXIMUM);
4952   TYPE_FIND_REGISTER (plugin, "audio/x-musepack", GST_RANK_PRIMARY,
4953       musepack_type_find, "mpc,mpp,mp+", MUSEPACK_CAPS, NULL, NULL);
4954   TYPE_FIND_REGISTER (plugin, "audio/x-au", GST_RANK_MARGINAL,
4955       au_type_find, "au,snd", AU_CAPS, NULL, NULL);
4956   TYPE_FIND_REGISTER_RIFF (plugin, "video/x-msvideo", GST_RANK_PRIMARY,
4957       "avi", "AVI ");
4958   TYPE_FIND_REGISTER_RIFF (plugin, "audio/qcelp", GST_RANK_PRIMARY,
4959       "qcp", "QLCM");
4960   TYPE_FIND_REGISTER_RIFF (plugin, "video/x-cdxa", GST_RANK_PRIMARY,
4961       "dat", "CDXA");
4962   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-vcd", GST_RANK_PRIMARY,
4963       "dat", "\000\377\377\377\377\377\377\377\377\377\377\000", 12,
4964       GST_TYPE_FIND_MAXIMUM);
4965   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-imelody", GST_RANK_PRIMARY,
4966       "imy,ime,imelody", "BEGIN:IMELODY", 13, GST_TYPE_FIND_MAXIMUM);
4967 #if 0
4968   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-smoke", GST_RANK_PRIMARY,
4969       NULL, "\x80smoke\x00\x01\x00", 6, GST_TYPE_FIND_MAXIMUM);
4970 #endif
4971   TYPE_FIND_REGISTER (plugin, "audio/midi", GST_RANK_PRIMARY, mid_type_find,
4972       "mid,midi", MID_CAPS, NULL, NULL);
4973   TYPE_FIND_REGISTER_RIFF (plugin, "audio/riff-midi", GST_RANK_PRIMARY,
4974       "mid,midi", "RMID");
4975   TYPE_FIND_REGISTER (plugin, "audio/mobile-xmf", GST_RANK_PRIMARY,
4976       mxmf_type_find, "mxmf", MXMF_CAPS, NULL, NULL);
4977   TYPE_FIND_REGISTER (plugin, "video/x-fli", GST_RANK_MARGINAL, flx_type_find,
4978       "flc,fli", FLX_CAPS, NULL, NULL);
4979   TYPE_FIND_REGISTER (plugin, "application/x-id3v2", GST_RANK_PRIMARY + 103,
4980       id3v2_type_find, "mp3,mp2,mp1,mpga,ogg,flac,tta", ID3_CAPS, NULL, NULL);
4981   TYPE_FIND_REGISTER (plugin, "application/x-id3v1", GST_RANK_PRIMARY + 101,
4982       id3v1_type_find, "mp3,mp2,mp1,mpga,ogg,flac,tta", ID3_CAPS, NULL, NULL);
4983   TYPE_FIND_REGISTER (plugin, "application/x-apetag", GST_RANK_PRIMARY + 102,
4984       apetag_type_find, "mp3,ape,mpc,wv", APETAG_CAPS, NULL, NULL);
4985   TYPE_FIND_REGISTER (plugin, "audio/x-ttafile", GST_RANK_PRIMARY,
4986       tta_type_find, "tta", TTA_CAPS, NULL, NULL);
4987   TYPE_FIND_REGISTER (plugin, "audio/x-mod", GST_RANK_SECONDARY, mod_type_find,
4988       "669,amf,dsm,gdm,far,imf,it,med,mod,mtm,okt,sam,s3m,stm,stx,ult,xm",
4989       MOD_CAPS, NULL, NULL);
4990   TYPE_FIND_REGISTER (plugin, "audio/mpeg", GST_RANK_PRIMARY, mp3_type_find,
4991       "mp3,mp2,mp1,mpga", MP3_CAPS, NULL, NULL);
4992   TYPE_FIND_REGISTER (plugin, "audio/x-ac3", GST_RANK_PRIMARY, ac3_type_find,
4993       "ac3,eac3", AC3_CAPS, NULL, NULL);
4994   TYPE_FIND_REGISTER (plugin, "audio/x-dts", GST_RANK_SECONDARY, dts_type_find,
4995       "dts", DTS_CAPS, NULL, NULL);
4996   TYPE_FIND_REGISTER (plugin, "audio/x-gsm", GST_RANK_PRIMARY, NULL, "gsm",
4997       GSM_CAPS, NULL, NULL);
4998   TYPE_FIND_REGISTER (plugin, "video/mpeg-sys", GST_RANK_PRIMARY,
4999       mpeg_sys_type_find, "mpe,mpeg,mpg", MPEG_SYS_CAPS, NULL, NULL);
5000   TYPE_FIND_REGISTER (plugin, "video/mpegts", GST_RANK_PRIMARY,
5001       mpeg_ts_type_find, "ts,mts", MPEGTS_CAPS, NULL, NULL);
5002   TYPE_FIND_REGISTER (plugin, "application/ogg", GST_RANK_PRIMARY,
5003       ogganx_type_find, "ogg,oga,ogv,ogm,ogx,spx,anx,axa,axv", OGG_CAPS,
5004       NULL, NULL);
5005   TYPE_FIND_REGISTER (plugin, "video/mpeg-elementary", GST_RANK_MARGINAL,
5006       mpeg_video_stream_type_find, "mpv,mpeg,mpg", MPEG_VIDEO_CAPS, NULL, NULL);
5007   TYPE_FIND_REGISTER (plugin, "video/mpeg4", GST_RANK_PRIMARY,
5008       mpeg4_video_type_find, "m4v", MPEG_VIDEO_CAPS, NULL, NULL);
5009   TYPE_FIND_REGISTER (plugin, "video/x-h263", GST_RANK_SECONDARY,
5010       h263_video_type_find, "h263,263", H263_VIDEO_CAPS, NULL, NULL);
5011   TYPE_FIND_REGISTER (plugin, "video/x-h264", GST_RANK_PRIMARY,
5012       h264_video_type_find, "h264,x264,264", H264_VIDEO_CAPS, NULL, NULL);
5013   TYPE_FIND_REGISTER (plugin, "video/x-nuv", GST_RANK_SECONDARY, nuv_type_find,
5014       "nuv", NUV_CAPS, NULL, NULL);
5015
5016   /* ISO formats */
5017   TYPE_FIND_REGISTER (plugin, "audio/x-m4a", GST_RANK_PRIMARY, m4a_type_find,
5018       "m4a", M4A_CAPS, NULL, NULL);
5019   TYPE_FIND_REGISTER (plugin, "application/x-3gp", GST_RANK_PRIMARY,
5020       q3gp_type_find, "3gp", Q3GP_CAPS, NULL, NULL);
5021   TYPE_FIND_REGISTER (plugin, "video/quicktime", GST_RANK_SECONDARY,
5022       qt_type_find, "mov", QT_CAPS, NULL, NULL);
5023   TYPE_FIND_REGISTER (plugin, "image/x-quicktime", GST_RANK_SECONDARY,
5024       qtif_type_find, "qif,qtif,qti", QTIF_CAPS, NULL, NULL);
5025   TYPE_FIND_REGISTER (plugin, "image/jp2", GST_RANK_PRIMARY,
5026       jp2_type_find, "jp2", JP2_CAPS, NULL, NULL);
5027   TYPE_FIND_REGISTER (plugin, "video/mj2", GST_RANK_PRIMARY,
5028       jp2_type_find, "mj2", MJ2_CAPS, NULL, NULL);
5029
5030   TYPE_FIND_REGISTER (plugin, "text/html", GST_RANK_SECONDARY, html_type_find,
5031       "htm,html", HTML_CAPS, NULL, NULL);
5032   TYPE_FIND_REGISTER_START_WITH (plugin, "application/vnd.rn-realmedia",
5033       GST_RANK_SECONDARY, "ra,ram,rm,rmvb", ".RMF", 4, GST_TYPE_FIND_MAXIMUM);
5034   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-pn-realaudio",
5035       GST_RANK_SECONDARY, "ra,ram,rm,rmvb", ".ra\375", 4,
5036       GST_TYPE_FIND_MAXIMUM);
5037   TYPE_FIND_REGISTER (plugin, "application/x-shockwave-flash",
5038       GST_RANK_SECONDARY, swf_type_find, "swf,swfl", SWF_CAPS, NULL, NULL);
5039   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-flv", GST_RANK_SECONDARY,
5040       "flv", "FLV", 3, GST_TYPE_FIND_MAXIMUM);
5041   TYPE_FIND_REGISTER (plugin, "text/plain", GST_RANK_MARGINAL, utf8_type_find,
5042       "txt", UTF8_CAPS, NULL, NULL);
5043   TYPE_FIND_REGISTER (plugin, "text/utf-16", GST_RANK_MARGINAL, utf16_type_find,
5044       "txt", UTF16_CAPS, NULL, NULL);
5045   TYPE_FIND_REGISTER (plugin, "text/utf-32", GST_RANK_MARGINAL, utf32_type_find,
5046       "txt", UTF32_CAPS, NULL, NULL);
5047   TYPE_FIND_REGISTER (plugin, "text/uri-list", GST_RANK_MARGINAL, uri_type_find,
5048       "ram", URI_CAPS, NULL, NULL);
5049   TYPE_FIND_REGISTER (plugin, "application/x-hls", GST_RANK_MARGINAL,
5050       hls_type_find, "m3u8", HLS_CAPS, NULL, NULL);
5051   TYPE_FIND_REGISTER (plugin, "application/sdp", GST_RANK_SECONDARY,
5052       sdp_type_find, "sdp", SDP_CAPS, NULL, NULL);
5053   TYPE_FIND_REGISTER (plugin, "application/smil", GST_RANK_SECONDARY,
5054       smil_type_find, "smil", SMIL_CAPS, NULL, NULL);
5055   TYPE_FIND_REGISTER (plugin, "application/xml", GST_RANK_MARGINAL,
5056       xml_type_find, "xml", GENERIC_XML_CAPS, NULL, NULL);
5057   TYPE_FIND_REGISTER_RIFF (plugin, "audio/x-wav", GST_RANK_PRIMARY, "wav",
5058       "WAVE");
5059   TYPE_FIND_REGISTER (plugin, "audio/x-aiff", GST_RANK_SECONDARY,
5060       aiff_type_find, "aiff,aif,aifc", AIFF_CAPS, NULL, NULL);
5061   TYPE_FIND_REGISTER (plugin, "audio/x-svx", GST_RANK_SECONDARY, svx_type_find,
5062       "iff,svx", SVX_CAPS, NULL, NULL);
5063   TYPE_FIND_REGISTER (plugin, "audio/x-paris", GST_RANK_SECONDARY,
5064       paris_type_find, "paf", PARIS_CAPS, NULL, NULL);
5065   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-nist", GST_RANK_SECONDARY,
5066       "nist", "NIST", 4, GST_TYPE_FIND_MAXIMUM);
5067   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-voc", GST_RANK_SECONDARY,
5068       "voc", "Creative", 8, GST_TYPE_FIND_MAXIMUM);
5069   TYPE_FIND_REGISTER (plugin, "audio/x-sds", GST_RANK_SECONDARY, sds_type_find,
5070       "sds", SDS_CAPS, NULL, NULL);
5071   TYPE_FIND_REGISTER (plugin, "audio/x-ircam", GST_RANK_SECONDARY,
5072       ircam_type_find, "sf", IRCAM_CAPS, NULL, NULL);
5073   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-w64", GST_RANK_SECONDARY,
5074       "w64", "riff", 4, GST_TYPE_FIND_MAXIMUM);
5075   TYPE_FIND_REGISTER (plugin, "audio/x-shorten", GST_RANK_SECONDARY,
5076       shn_type_find, "shn", SHN_CAPS, NULL, NULL);
5077   TYPE_FIND_REGISTER (plugin, "application/x-ape", GST_RANK_SECONDARY,
5078       ape_type_find, "ape", APE_CAPS, NULL, NULL);
5079   TYPE_FIND_REGISTER (plugin, "image/jpeg", GST_RANK_PRIMARY + 15,
5080       jpeg_type_find, "jpg,jpe,jpeg", JPEG_CAPS, NULL, NULL);
5081   TYPE_FIND_REGISTER_START_WITH (plugin, "image/gif", GST_RANK_PRIMARY, "gif",
5082       "GIF8", 4, GST_TYPE_FIND_MAXIMUM);
5083   TYPE_FIND_REGISTER_START_WITH (plugin, "image/png", GST_RANK_PRIMARY + 14,
5084       "png", "\211PNG\015\012\032\012", 8, GST_TYPE_FIND_MAXIMUM);
5085   TYPE_FIND_REGISTER (plugin, "image/bmp", GST_RANK_PRIMARY, bmp_type_find,
5086       "bmp", BMP_CAPS, NULL, NULL);
5087   TYPE_FIND_REGISTER (plugin, "image/tiff", GST_RANK_PRIMARY, tiff_type_find,
5088       "tif,tiff", TIFF_CAPS, NULL, NULL);
5089   TYPE_FIND_REGISTER (plugin, "image/x-portable-pixmap", GST_RANK_SECONDARY,
5090       pnm_type_find, "pnm,ppm,pgm,pbm", PNM_CAPS, NULL, NULL);
5091   TYPE_FIND_REGISTER (plugin, "video/x-matroska", GST_RANK_PRIMARY,
5092       matroska_type_find, "mkv,mka,mk3d,webm", MATROSKA_CAPS, NULL, NULL);
5093   TYPE_FIND_REGISTER (plugin, "application/mxf", GST_RANK_PRIMARY,
5094       mxf_type_find, "mxf", MXF_CAPS, NULL, NULL);
5095   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-mve", GST_RANK_SECONDARY,
5096       "mve", "Interplay MVE File\032\000\032\000\000\001\063\021", 26,
5097       GST_TYPE_FIND_MAXIMUM);
5098   TYPE_FIND_REGISTER (plugin, "video/x-dv", GST_RANK_SECONDARY, dv_type_find,
5099       "dv,dif", DV_CAPS, NULL, NULL);
5100   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-amr-nb-sh", GST_RANK_PRIMARY,
5101       "amr", "#!AMR", 5, GST_TYPE_FIND_LIKELY);
5102   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-amr-wb-sh", GST_RANK_PRIMARY,
5103       "amr", "#!AMR-WB", 7, GST_TYPE_FIND_MAXIMUM);
5104   TYPE_FIND_REGISTER (plugin, "audio/iLBC-sh", GST_RANK_PRIMARY, ilbc_type_find,
5105       "ilbc", ILBC_CAPS, NULL, NULL);
5106   TYPE_FIND_REGISTER (plugin, "audio/x-sbc", GST_RANK_MARGINAL, sbc_type_find,
5107       "sbc", SBC_CAPS, NULL, NULL);
5108   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-sid", GST_RANK_MARGINAL,
5109       "sid", "PSID", 4, GST_TYPE_FIND_MAXIMUM);
5110   TYPE_FIND_REGISTER_START_WITH (plugin, "image/x-xcf", GST_RANK_SECONDARY,
5111       "xcf", "gimp xcf", 8, GST_TYPE_FIND_MAXIMUM);
5112   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-mng", GST_RANK_SECONDARY,
5113       "mng", "\212MNG\015\012\032\012", 8, GST_TYPE_FIND_MAXIMUM);
5114   TYPE_FIND_REGISTER_START_WITH (plugin, "image/x-jng", GST_RANK_SECONDARY,
5115       "jng", "\213JNG\015\012\032\012", 8, GST_TYPE_FIND_MAXIMUM);
5116   TYPE_FIND_REGISTER_START_WITH (plugin, "image/x-xpixmap", GST_RANK_SECONDARY,
5117       "xpm", "/* XPM */", 9, GST_TYPE_FIND_MAXIMUM);
5118   TYPE_FIND_REGISTER_START_WITH (plugin, "image/x-sun-raster",
5119       GST_RANK_SECONDARY, "ras", "\131\246\152\225", 4, GST_TYPE_FIND_MAXIMUM);
5120   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-bzip",
5121       GST_RANK_SECONDARY, "bz2", "BZh", 3, GST_TYPE_FIND_LIKELY);
5122   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-gzip",
5123       GST_RANK_SECONDARY, "gz", "\037\213", 2, GST_TYPE_FIND_LIKELY);
5124   TYPE_FIND_REGISTER_START_WITH (plugin, "application/zip", GST_RANK_SECONDARY,
5125       "zip", "PK\003\004", 4, GST_TYPE_FIND_LIKELY);
5126   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-compress",
5127       GST_RANK_SECONDARY, "Z", "\037\235", 2, GST_TYPE_FIND_LIKELY);
5128   TYPE_FIND_REGISTER (plugin, "subtitle/x-kate", GST_RANK_MARGINAL,
5129       kate_type_find, NULL, NULL, NULL, NULL);
5130   TYPE_FIND_REGISTER (plugin, "audio/x-flac", GST_RANK_PRIMARY, flac_type_find,
5131       "flac", FLAC_CAPS, NULL, NULL);
5132   TYPE_FIND_REGISTER (plugin, "audio/x-vorbis", GST_RANK_PRIMARY,
5133       vorbis_type_find, NULL, VORBIS_CAPS, NULL, NULL);
5134   TYPE_FIND_REGISTER (plugin, "video/x-theora", GST_RANK_PRIMARY,
5135       theora_type_find, NULL, THEORA_CAPS, NULL, NULL);
5136   TYPE_FIND_REGISTER (plugin, "application/x-ogm-video", GST_RANK_PRIMARY,
5137       ogmvideo_type_find, NULL, OGMVIDEO_CAPS, NULL, NULL);
5138   TYPE_FIND_REGISTER (plugin, "application/x-ogm-audio", GST_RANK_PRIMARY,
5139       ogmaudio_type_find, NULL, OGMAUDIO_CAPS, NULL, NULL);
5140   TYPE_FIND_REGISTER (plugin, "application/x-ogm-text", GST_RANK_PRIMARY,
5141       ogmtext_type_find, NULL, OGMTEXT_CAPS, NULL, NULL);
5142   TYPE_FIND_REGISTER (plugin, "audio/x-speex", GST_RANK_PRIMARY,
5143       speex_type_find, NULL, SPEEX_CAPS, NULL, NULL);
5144   TYPE_FIND_REGISTER (plugin, "audio/x-celt", GST_RANK_PRIMARY, celt_type_find,
5145       NULL, CELT_CAPS, NULL, NULL);
5146   TYPE_FIND_REGISTER (plugin, "application/x-ogg-skeleton", GST_RANK_PRIMARY,
5147       oggskel_type_find, NULL, OGG_SKELETON_CAPS, NULL, NULL);
5148   TYPE_FIND_REGISTER (plugin, "text/x-cmml", GST_RANK_PRIMARY, cmml_type_find,
5149       NULL, CMML_CAPS, NULL, NULL);
5150   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-executable",
5151       GST_RANK_MARGINAL, NULL, "\177ELF", 4, GST_TYPE_FIND_MAXIMUM);
5152   TYPE_FIND_REGISTER (plugin, "audio/aac", GST_RANK_SECONDARY, aac_type_find,
5153       "aac,adts,adif,loas", AAC_CAPS, NULL, NULL);
5154   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-spc", GST_RANK_SECONDARY,
5155       "spc", "SNES-SPC700 Sound File Data", 27, GST_TYPE_FIND_MAXIMUM);
5156   TYPE_FIND_REGISTER (plugin, "audio/x-wavpack", GST_RANK_SECONDARY,
5157       wavpack_type_find, "wv,wvp", WAVPACK_CAPS, NULL, NULL);
5158   TYPE_FIND_REGISTER (plugin, "audio/x-wavpack-correction", GST_RANK_SECONDARY,
5159       wavpack_type_find, "wvc", WAVPACK_CORRECTION_CAPS, NULL, NULL);
5160   TYPE_FIND_REGISTER (plugin, "application/postscript", GST_RANK_SECONDARY,
5161       postscript_type_find, "ps", POSTSCRIPT_CAPS, NULL, NULL);
5162   TYPE_FIND_REGISTER (plugin, "image/svg+xml", GST_RANK_SECONDARY,
5163       svg_type_find, "svg", SVG_CAPS, NULL, NULL);
5164   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-rar",
5165       GST_RANK_SECONDARY, "rar", "Rar!", 4, GST_TYPE_FIND_LIKELY);
5166   TYPE_FIND_REGISTER (plugin, "application/x-tar", GST_RANK_SECONDARY,
5167       tar_type_find, "tar", TAR_CAPS, NULL, NULL);
5168   TYPE_FIND_REGISTER (plugin, "application/x-ar", GST_RANK_SECONDARY,
5169       ar_type_find, "a", AR_CAPS, NULL, NULL);
5170   TYPE_FIND_REGISTER (plugin, "application/x-ms-dos-executable",
5171       GST_RANK_SECONDARY, msdos_type_find, "dll,exe,ocx,sys,scr,msstyles,cpl",
5172       MSDOS_CAPS, NULL, NULL);
5173   TYPE_FIND_REGISTER (plugin, "video/x-dirac", GST_RANK_PRIMARY,
5174       dirac_type_find, NULL, DIRAC_CAPS, NULL, NULL);
5175   TYPE_FIND_REGISTER (plugin, "multipart/x-mixed-replace", GST_RANK_SECONDARY,
5176       multipart_type_find, NULL, MULTIPART_CAPS, NULL, NULL);
5177   TYPE_FIND_REGISTER (plugin, "application/x-mmsh", GST_RANK_SECONDARY,
5178       mmsh_type_find, NULL, MMSH_CAPS, NULL, NULL);
5179   TYPE_FIND_REGISTER (plugin, "video/vivo", GST_RANK_SECONDARY, vivo_type_find,
5180       "viv", VIVO_CAPS, NULL, NULL);
5181   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-nsf", GST_RANK_SECONDARY,
5182       "nsf", "NESM\x1a", 5, GST_TYPE_FIND_MAXIMUM);
5183   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-gym", GST_RANK_SECONDARY,
5184       "gym", "GYMX", 4, GST_TYPE_FIND_MAXIMUM);
5185   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-ay", GST_RANK_SECONDARY, "ay",
5186       "ZXAYEMUL", 8, GST_TYPE_FIND_MAXIMUM);
5187   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-gbs", GST_RANK_SECONDARY,
5188       "gbs", "GBS\x01", 4, GST_TYPE_FIND_MAXIMUM);
5189   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-vgm", GST_RANK_SECONDARY,
5190       "vgm", "Vgm\x20", 4, GST_TYPE_FIND_MAXIMUM);
5191   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-sap", GST_RANK_SECONDARY,
5192       "sap", "SAP\x0d\x0a" "AUTHOR\x20", 12, GST_TYPE_FIND_MAXIMUM);
5193   TYPE_FIND_REGISTER_START_WITH (plugin, "video/x-ivf", GST_RANK_SECONDARY,
5194       "ivf", "DKIF", 4, GST_TYPE_FIND_NEARLY_CERTAIN);
5195   TYPE_FIND_REGISTER_START_WITH (plugin, "audio/x-kss", GST_RANK_SECONDARY,
5196       "kss", "KSSX\0", 5, GST_TYPE_FIND_MAXIMUM);
5197   TYPE_FIND_REGISTER_START_WITH (plugin, "application/pdf", GST_RANK_SECONDARY,
5198       "pdf", "%PDF-", 5, GST_TYPE_FIND_LIKELY);
5199   TYPE_FIND_REGISTER_START_WITH (plugin, "application/msword",
5200       GST_RANK_SECONDARY, "doc", "\320\317\021\340\241\261\032\341", 8,
5201       GST_TYPE_FIND_LIKELY);
5202   /* Mac OS X .DS_Store files tend to be taken for video/mpeg */
5203   TYPE_FIND_REGISTER_START_WITH (plugin, "application/octet-stream",
5204       GST_RANK_SECONDARY, "DS_Store", "\000\000\000\001Bud1", 8,
5205       GST_TYPE_FIND_LIKELY);
5206   TYPE_FIND_REGISTER_START_WITH (plugin, "image/vnd.adobe.photoshop",
5207       GST_RANK_SECONDARY, "psd", "8BPS\000\001\000\000\000\000", 10,
5208       GST_TYPE_FIND_LIKELY);
5209   TYPE_FIND_REGISTER (plugin, "image/vnd.wap.wbmp", GST_RANK_MARGINAL,
5210       wbmp_typefind, NULL, NULL, NULL, NULL);
5211   TYPE_FIND_REGISTER_START_WITH (plugin, "application/x-yuv4mpeg",
5212       GST_RANK_SECONDARY, "y4m", "YUV4MPEG2 ", 10, GST_TYPE_FIND_LIKELY);
5213   TYPE_FIND_REGISTER (plugin, "image/x-icon", GST_RANK_MARGINAL,
5214       windows_icon_typefind, NULL, NULL, NULL, NULL);
5215
5216 #ifdef USE_GIO
5217   TYPE_FIND_REGISTER (plugin, "xdgmime-base", GST_RANK_MARGINAL,
5218       xdgmime_typefind, NULL, NULL, NULL, NULL);
5219 #endif
5220
5221   TYPE_FIND_REGISTER (plugin, "image/x-degas", GST_RANK_MARGINAL,
5222       degas_type_find, NULL, NULL, NULL, NULL);
5223   TYPE_FIND_REGISTER (plugin, "application/octet-stream", GST_RANK_MARGINAL,
5224       dvdiso_type_find, NULL, NULL, NULL, NULL);
5225
5226   TYPE_FIND_REGISTER (plugin, "application/x-ssa", GST_RANK_SECONDARY,
5227       ssa_type_find, "ssa,ass", NULL, NULL, NULL);
5228
5229   return TRUE;
5230 }
5231
5232 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
5233     GST_VERSION_MINOR,
5234     typefindfunctions,
5235     "default typefind functions",
5236     plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)