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