]> git.sesse.net Git - vlc/blob - modules/access/udp.c
Remove extra whitespace
[vlc] / modules / access / udp.c
1 /*****************************************************************************
2  * udp.c: raw UDP & RTP input module
3  *****************************************************************************
4  * Copyright (C) 2001-2005 the VideoLAN team
5  * Copyright (C) 2007 Remi Denis-Courmont
6  * $Id$
7  *
8  * Authors: Christophe Massiot <massiot@via.ecp.fr>
9  *          Tristan Leteurtre <tooney@via.ecp.fr>
10  *          Laurent Aimar <fenrir@via.ecp.fr>
11  *          Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
12  *          Remi Denis-Courmont
13  *
14  * Reviewed: 23 October 2003, Jean-Paul Saman <jpsaman _at_ videolan _dot_ org>
15  *
16  * This program is free software; you can redistribute it and/or modify
17  * it under the terms of the GNU General Public License as published by
18  * the Free Software Foundation; either version 2 of the License, or
19  * (at your option) any later version.
20  *
21  * This program is distributed in the hope that it will be useful,
22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
24  * GNU General Public License for more details.
25  *
26  * You should have received a copy of the GNU General Public License
27  * along with this program; if not, write to the Free Software
28  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
29  *****************************************************************************/
30
31 /*****************************************************************************
32  * Preamble
33  *****************************************************************************/
34 #include <stdlib.h>
35
36 #include <vlc/vlc.h>
37 #include <vlc_access.h>
38 #include <vlc_network.h>
39
40 /* Big pile of stuff missing form glibc 2.5 */
41 #if defined (HAVE_NETINET_UDPLITE_H)
42 # include <netinet/udplite.h>
43 #elif defined (__linux__)
44 # define UDPLITE_SEND_CSCOV     10
45 # define UDPLITE_RECV_CSCOV     11
46 #endif
47
48 #ifndef SOCK_DCCP /* provisional API */
49 # ifdef __linux__
50 #  define SOCK_DCCP 6
51 # endif
52 #endif
53
54 #ifndef IPPROTO_DCCP
55 # define IPPROTO_DCCP 33 /* IANA */
56 #endif
57
58 #ifndef IPPROTO_UDPLITE
59 # define IPPROTO_UDPLITE 136 /* from IANA */
60 #endif
61 #ifndef SOL_UDPLITE
62 # define SOL_UDPLITE IPPROTO_UDPLITE
63 #endif
64
65
66 /*****************************************************************************
67  * Module descriptor
68  *****************************************************************************/
69 #define CACHING_TEXT N_("Caching value in ms")
70 #define CACHING_LONGTEXT N_( \
71     "Caching value for UDP streams. This " \
72     "value should be set in milliseconds." )
73
74 #define AUTO_MTU_TEXT N_("Autodetection of MTU")
75 #define AUTO_MTU_LONGTEXT N_( \
76     "Automatically detect the line's MTU. This will increase the size if" \
77     " truncated packets are found" )
78
79 #define RTP_LATE_TEXT N_("RTP reordering timeout in ms")
80 #define RTP_LATE_LONGTEXT N_( \
81     "VLC reorders RTP packets. The input will wait for late packets at most "\
82     "the time specified here (in milliseconds)." )
83
84 static int  Open ( vlc_object_t * );
85 static void Close( vlc_object_t * );
86
87 vlc_module_begin();
88     set_shortname( _("UDP/RTP" ) );
89     set_description( _("UDP/RTP input") );
90     set_category( CAT_INPUT );
91     set_subcategory( SUBCAT_INPUT_ACCESS );
92
93     add_integer( "udp-caching", DEFAULT_PTS_DELAY / 1000, NULL, CACHING_TEXT,
94                  CACHING_LONGTEXT, VLC_TRUE );
95     add_integer( "rtp-late", 100, NULL, RTP_LATE_TEXT, RTP_LATE_LONGTEXT, VLC_TRUE );
96
97     add_bool( "udp-auto-mtu", 1, NULL,
98               AUTO_MTU_TEXT, AUTO_MTU_LONGTEXT, VLC_TRUE );
99
100     set_capability( "access2", 0 );
101     add_shortcut( "udp" );
102     add_shortcut( "udpstream" );
103     add_shortcut( "udp4" );
104     add_shortcut( "udp6" );
105     add_shortcut( "rtp" );
106     add_shortcut( "rtp4" );
107     add_shortcut( "rtp6" );
108     add_shortcut( "udplite" );
109     add_shortcut( "rtptcp" );
110     add_shortcut( "dccp" );
111
112     set_callbacks( Open, Close );
113 vlc_module_end();
114
115 /*****************************************************************************
116  * Local prototypes
117  *****************************************************************************/
118 #define RTP_HEADER_LEN 12
119
120 static block_t *BlockUDP( access_t * );
121 static block_t *BlockTCP( access_t * );
122 static block_t *BlockRTP( access_t * );
123 static block_t *BlockChoose( access_t * );
124 static int Control( access_t *, int, va_list );
125
126 struct access_sys_t
127 {
128     int fd;
129
130     int i_mtu;
131     vlc_bool_t b_auto_mtu;
132     vlc_bool_t b_framed_rtp;
133
134     /* reorder rtp packets when out-of-sequence */
135     uint16_t i_last_seqno;
136     mtime_t i_rtp_late;
137     block_t *p_list;
138     block_t *p_end;
139     block_t *p_partial_frame; /* Partial Framed RTP packet */
140 };
141
142 /*****************************************************************************
143  * Open: open the socket
144  *****************************************************************************/
145 static int Open( vlc_object_t *p_this )
146 {
147     access_t     *p_access = (access_t*)p_this;
148     access_sys_t *p_sys;
149
150     char *psz_name = strdup( p_access->psz_path );
151     char *psz_parser;
152     const char *psz_server_addr, *psz_bind_addr = "";
153     int  i_bind_port, i_server_port = 0;
154     int fam = AF_UNSPEC, proto = IPPROTO_UDP;
155
156     if (strlen (p_access->psz_access) >= 3)
157     {
158         switch (p_access->psz_access[3])
159         {
160             case '4':
161                 fam = AF_INET;
162                 break;
163
164             case '6':
165                 fam = AF_INET6;
166                 break;
167         }
168         if (strcmp (p_access->psz_access + 3, "lite") == 0)
169             proto = IPPROTO_UDPLITE;
170         else
171         if (strcmp (p_access->psz_access + 3, "tcp") == 0)
172             proto = IPPROTO_TCP;
173         else
174         if (strcmp (p_access->psz_access, "dccp") == 0)
175             proto = IPPROTO_DCCP;
176     }
177
178     i_bind_port = var_CreateGetInteger( p_access, "server-port" );
179
180     /* Parse psz_name syntax :
181      * [serveraddr[:serverport]][@[bindaddr]:[bindport]] */
182     psz_parser = strchr( psz_name, '@' );
183     if( psz_parser != NULL )
184     {
185         /* Found bind address and/or bind port */
186         *psz_parser++ = '\0';
187         psz_bind_addr = psz_parser;
188
189         if( *psz_parser == '[' )
190             /* skips bracket'd IPv6 address */
191             psz_parser = strchr( psz_parser, ']' );
192
193         if( psz_parser != NULL )
194         {
195             psz_parser = strchr( psz_parser, ':' );
196             if( psz_parser != NULL )
197             {
198                 *psz_parser++ = '\0';
199                 i_bind_port = atoi( psz_parser );
200             }
201         }
202     }
203
204     psz_server_addr = psz_name;
205     if( *psz_server_addr == '[' )
206         /* skips bracket'd IPv6 address */
207         psz_parser = strchr( psz_name, ']' );
208
209     if( psz_parser != NULL )
210     {
211         psz_parser = strchr( psz_parser, ':' );
212         if( psz_parser != NULL )
213         {
214             *psz_parser++ = '\0';
215             i_server_port = atoi( psz_parser );
216         }
217     }
218
219     msg_Dbg( p_access, "opening server=%s:%d local=%s:%d",
220              psz_server_addr, i_server_port, psz_bind_addr, i_bind_port );
221
222     /* Set up p_access */
223     access_InitFields( p_access );
224     ACCESS_SET_CALLBACKS( NULL, BlockChoose, Control, NULL );
225     p_access->info.b_prebuffered = VLC_FALSE;
226     MALLOC_ERR( p_access->p_sys, access_sys_t ); p_sys = p_access->p_sys;
227     memset (p_sys, 0, sizeof (*p_sys));
228
229     switch (proto)
230     {
231         case IPPROTO_UDP:
232         case IPPROTO_UDPLITE:
233             p_sys->fd = net_OpenDgram( p_access, psz_bind_addr, i_bind_port,
234                                        psz_server_addr, i_server_port, fam,
235                                        proto );
236             break;
237
238         case IPPROTO_TCP:
239             p_sys->fd = net_ConnectTCP( p_access, psz_server_addr, i_server_port );
240             p_sys->b_framed_rtp = VLC_TRUE;
241             break;
242
243         case IPPROTO_DCCP:
244 #ifdef SOCK_DCCP
245             p_sys->fd = net_Connect( p_access, psz_server_addr, i_server_port,
246                                      SOCK_DCCP, IPPROTO_DCCP );
247 #else
248             p_sys->fd = -1;
249             msg_Err( p_access, "DCCP support not compiled-in!" );
250 #endif
251             break;
252     }
253     free (psz_name);
254     if( p_sys->fd == -1 )
255     {
256         msg_Err( p_access, "cannot open socket" );
257         free( p_sys );
258         return VLC_EGENERIC;
259     }
260
261     shutdown( p_sys->fd, SHUT_WR );
262
263 #ifdef UDPLITE_RECV_CSCOV
264     if (proto == IPPROTO_UDPLITE)
265         /* UDP header: 8 bytes + RTP header: 12 bytes (or more) */
266         setsockopt (p_sys->fd, SOL_UDPLITE, UDPLITE_RECV_CSCOV,
267                     &(int){ 20 }, sizeof (int));
268 #endif
269
270     if (p_sys->b_framed_rtp)
271     {
272         /* We don't do autodetection and prebuffering in case of framing */
273         p_access->pf_block = BlockRTP;
274         p_sys->i_mtu = 65535;
275     }
276     else
277     {
278         /* FIXME */
279         p_sys->i_mtu = var_CreateGetInteger( p_access, "mtu" );
280         if( p_sys->i_mtu <= 1 )
281             p_sys->i_mtu  = 1500;   /* Avoid problem */
282
283         p_sys->b_auto_mtu = var_CreateGetBool( p_access, "udp-auto-mtu" );;
284     }
285
286     /* Update default_pts to a suitable value for udp access */
287     var_Create( p_access, "udp-caching", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
288
289     /* RTP reordering for out-of-sequence packets */
290     p_sys->i_rtp_late = var_CreateGetInteger( p_access, "rtp-late" ) * 1000;
291     p_sys->i_last_seqno = 0;
292     p_sys->p_list = NULL;
293     p_sys->p_end = NULL;
294     return VLC_SUCCESS;
295 }
296
297 /*****************************************************************************
298  * Close: free unused data structures
299  *****************************************************************************/
300 static void Close( vlc_object_t *p_this )
301 {
302     access_t     *p_access = (access_t*)p_this;
303     access_sys_t *p_sys = p_access->p_sys;
304
305     block_ChainRelease( p_sys->p_list );
306     net_Close( p_sys->fd );
307     free( p_sys );
308 }
309
310 /*****************************************************************************
311  * Control:
312  *****************************************************************************/
313 static int Control( access_t *p_access, int i_query, va_list args )
314 {
315     access_sys_t *p_sys = p_access->p_sys;
316     vlc_bool_t   *pb_bool;
317     int          *pi_int;
318     int64_t      *pi_64;
319
320     switch( i_query )
321     {
322         /* */
323         case ACCESS_CAN_SEEK:
324         case ACCESS_CAN_FASTSEEK:
325         case ACCESS_CAN_PAUSE:
326         case ACCESS_CAN_CONTROL_PACE:
327             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
328             *pb_bool = VLC_FALSE;
329             break;
330         /* */
331         case ACCESS_GET_MTU:
332             pi_int = (int*)va_arg( args, int * );
333             *pi_int = p_sys->i_mtu;
334             break;
335
336         case ACCESS_GET_PTS_DELAY:
337             pi_64 = (int64_t*)va_arg( args, int64_t * );
338             *pi_64 = var_GetInteger( p_access, "udp-caching" ) * 1000;
339             break;
340
341         /* */
342         case ACCESS_SET_PAUSE_STATE:
343         case ACCESS_GET_TITLE_INFO:
344         case ACCESS_SET_TITLE:
345         case ACCESS_SET_SEEKPOINT:
346         case ACCESS_SET_PRIVATE_ID_STATE:
347             return VLC_EGENERIC;
348
349         default:
350             msg_Warn( p_access, "unimplemented query in control" );
351             return VLC_EGENERIC;
352
353     }
354     return VLC_SUCCESS;
355 }
356
357 /*****************************************************************************
358  * BlockUDP:
359  *****************************************************************************/
360 static block_t *BlockUDP( access_t *p_access )
361 {
362     access_sys_t *p_sys = p_access->p_sys;
363     block_t      *p_block;
364
365     /* Read data */
366     p_block = block_New( p_access, p_sys->i_mtu );
367     p_block->i_buffer = net_Read( p_access, p_sys->fd, NULL,
368                                   p_block->p_buffer, p_sys->i_mtu,
369                                   VLC_FALSE );
370     if( p_block->i_buffer < 0 )
371     {
372         block_Release( p_block );
373         return NULL;
374     }
375
376     if( (p_block->i_buffer >= p_sys->i_mtu) && p_sys->b_auto_mtu &&
377         p_sys->i_mtu < 32767 )
378     {
379         /* Increase by 100% */
380         p_sys->i_mtu *= 2;
381         msg_Dbg( p_access, "increasing MTU to %d", p_sys->i_mtu );
382     }
383
384     return p_block;
385 }
386
387 /*****************************************************************************
388  * BlockTCP: Framed RTP/AVP packet reception for COMEDIA (see RFC4571)
389  *****************************************************************************/
390 static block_t *BlockTCP( access_t *p_access )
391 {
392     access_sys_t *p_sys = p_access->p_sys;
393     block_t      *p_block = p_sys->p_partial_frame;
394
395     if( p_access->info.b_eof )
396         return NULL;
397
398     if( p_block == NULL )
399     {
400         /* MTU should always be 65535 in this case */
401         p_sys->p_partial_frame = p_block = block_New( p_access, 65537 );
402         if (p_block == NULL)
403             return NULL;
404     }
405
406     /* Read RTP framing */
407     if (p_block->i_buffer < 2)
408     {
409         /* FIXME: not very efficient */
410         int i_read = net_Read( p_access, p_sys->fd, NULL,
411                                p_block->p_buffer + p_block->i_buffer,
412                                2 - p_block->i_buffer, VLC_FALSE );
413         if( i_read <= 0 )
414             goto error;
415
416         p_block->i_buffer += i_read;
417         if (p_block->i_buffer < 2)
418             return NULL;
419     }
420
421     uint16_t framelen = GetWLE( p_block->p_buffer );
422     /* Read RTP frame */
423     if( framelen > 0 )
424     {
425         int i_read = net_Read( p_access, p_sys->fd, NULL,
426                                p_block->p_buffer + p_block->i_buffer,
427                                2 + framelen - p_block->i_buffer, VLC_FALSE );
428         if( i_read <= 0 )
429             goto error;
430
431         p_block->i_buffer += i_read;
432     }
433
434     if( p_block->i_buffer < (2 + framelen) )
435         return NULL; // incomplete frame
436
437     /* Hide framing from RTP layer */
438     p_block->p_buffer += 2;
439     p_block->i_buffer -= 2;
440     p_sys->p_partial_frame = NULL;
441     return p_block;
442
443 error:
444     p_access->info.b_eof = VLC_TRUE;
445     block_Release( p_block );
446     p_sys->p_partial_frame = NULL;
447     return NULL;
448 }
449
450
451 /*
452  * rtp_ChainInsert - insert a p_block in the chain and
453  * look at the sequence numbers.
454  */
455 static inline vlc_bool_t rtp_ChainInsert( access_t *p_access, block_t *p_block )
456 {
457     access_sys_t *p_sys = (access_sys_t *) p_access->p_sys;
458     block_t *p_prev = NULL;
459     block_t *p = p_sys->p_end;
460     uint16_t i_new = (uint16_t) p_block->i_dts;
461     uint16_t i_tmp = 0;
462
463     if( !p_sys->p_list )
464     {
465         p_sys->p_list = p_block;
466         p_sys->p_end = p_block;
467         return VLC_TRUE;
468     }
469     /* walk through the queue from top down since the new packet is in 
470     most cases just appended to the end */
471
472     for( ;; )
473     {
474         i_tmp = i_new - (uint16_t) p->i_dts;
475
476         if( !i_tmp )   /* trash duplicate */
477             break;
478
479         if ( i_tmp < 32768 )
480         {   /* insert after this block ( i_new > p->i_dts ) */
481             p_block->p_next = p->p_next;
482             p->p_next = p_block;
483             p_block->p_prev = p;
484             if (p_prev)
485             {
486                 p_prev->p_prev = p_block;
487                 msg_Dbg(p_access, "RTP reordering: insert after %d, new %d", 
488                     (uint16_t) p->i_dts, i_new );
489             }
490             else 
491             {
492                 p_sys->p_end = p_block;
493             }
494             return VLC_TRUE;
495         }
496         if( p == p_sys->p_list )
497         {   /* we've reached bottom of chain */
498             i_tmp = p_sys->i_last_seqno - i_new;
499             if( !p_access->info.b_prebuffered || (i_tmp > 32767) )
500             {
501                 msg_Dbg(p_access, "RTP reordering: prepend %d before %d", 
502                         i_new, (uint16_t) p->i_dts );
503                 p_block->p_next = p;
504                 p->p_prev = p_block;
505                 p_sys->p_list = p_block;
506                 return VLC_TRUE;
507             }
508
509             if( !i_tmp )   /* trash duplicate */
510                 break;
511
512             /* reordering failed - append the packet to the end of queue */
513             msg_Dbg(p_access, "RTP: sequence changed (or buffer too small) "
514                 "new: %d, buffer %d...%d", i_new, (uint16_t) p->i_dts, 
515                 (uint16_t) p_sys->p_end->i_dts);
516             p_sys->p_end->p_next = p_block;
517             p_block->p_prev = p_sys->p_end;
518             p_sys->p_end = p_block;
519             return VLC_TRUE;
520         }
521         p_prev = p;
522         p = p->p_prev;
523     }
524     block_Release( p_block );
525     return VLC_FALSE;
526 }
527
528 /*****************************************************************************
529  * BlockParseRTP/BlockRTP:
530  *****************************************************************************/
531 static block_t *BlockParseRTP( access_t *p_access, block_t *p_block )
532 {
533     int      i_payload_type;
534     size_t   i_skip = RTP_HEADER_LEN;
535
536     if( p_block == NULL )
537         return NULL;
538
539     if( p_block->i_buffer < RTP_HEADER_LEN )
540     {
541         msg_Dbg( p_access, "short RTP packet received" );
542         goto trash;
543     }
544
545     /* Parse the header and make some verifications.
546      * See RFC 3550. */
547     // Version number:
548     if( ( p_block->p_buffer[0] >> 6 ) != 2)
549     {
550         msg_Dbg( p_access, "RTP version is %u instead of 2",
551                  p_block->p_buffer[0] >> 6 );
552         goto trash;
553     }
554     // Padding bit:
555     uint8_t pad = (p_block->p_buffer[0] & 0x20)
556                     ? p_block->p_buffer[p_block->i_buffer - 1] : 0;
557     // CSRC count:
558     i_skip += (p_block->p_buffer[0] & 0x0F) * 4;
559     // Extension header:
560     if (p_block->p_buffer[0] & 0x10) /* Extension header */
561     {
562         i_skip += 4;
563         if ((size_t)p_block->i_buffer < i_skip)
564             goto trash;
565
566         i_skip += 4 * GetWBE( p_block->p_buffer + i_skip - 2 );
567     }
568
569     i_payload_type    = p_block->p_buffer[1] & 0x7F;
570
571     /* Remember sequence number in i_dts */
572     p_block->i_pts = mdate();
573     p_block->i_dts = (mtime_t) GetWBE( p_block->p_buffer + 2 );
574
575     /* FIXME: use rtpmap */
576     switch( i_payload_type )
577     {
578         case 14: // MPA: MPEG Audio (RFC2250, §3.4)
579             i_skip += 4; // 32 bits RTP/MPA header
580             break;
581
582         case 32: // MPV: MPEG Video (RFC2250, §3.5)
583             i_skip += 4; // 32 bits RTP/MPV header
584             if( (size_t)p_block->i_buffer < i_skip )
585                 goto trash;
586             if( p_block->p_buffer[i_skip - 3] & 0x4 )
587             {
588                 /* MPEG2 Video extension header */
589                 /* TODO: shouldn't we skip this too ? */
590             }
591             break;
592
593         case 33: // MP2: MPEG TS (RFC2250, §2)
594             /* plain TS over RTP */
595             break;
596
597         default:
598             msg_Dbg( p_access, "unsupported RTP payload type: %u", i_payload_type );
599             goto trash;
600     }
601
602     if( (size_t)p_block->i_buffer < (i_skip + pad) )
603         goto trash;
604
605     /* Remove the RTP header */
606     p_block->i_buffer -= i_skip;
607     p_block->p_buffer += i_skip;
608
609     /* This is the place for deciphering and authentication */
610
611     /* Remove padding (at the end) */
612     p_block->i_buffer -= pad;
613
614 #if 0
615     /* Emulate packet loss */
616     if ( (i_sequence_number % 4000) == 0)
617     {
618         msg_Warn( p_access, "Emulating packet drop" );
619         block_Release( p_block );
620         return NULL;
621     }
622 #endif
623
624     return p_block;
625
626 trash:
627     block_Release( p_block );
628     return NULL;
629 }
630
631 static block_t *BlockPrebufferRTP( access_t *p_access, block_t *p_block )
632 {
633     access_sys_t *p_sys = p_access->p_sys;
634     mtime_t   i_first = mdate();
635     int       i_count = 0;
636     block_t   *p = p_block;
637
638     for( ;; )
639     {
640         mtime_t i_date = mdate();
641
642         if( p && rtp_ChainInsert( p_access, p ))
643             i_count++;
644
645         /* Require at least 2 packets in the buffer */
646         if( i_count > 2 && (i_date - i_first) > p_sys->i_rtp_late )
647             break;
648
649         p = BlockParseRTP( p_access, BlockUDP( p_access ) );
650         if( !p && (i_date - i_first) > p_sys->i_rtp_late ) 
651         {
652             msg_Err( p_access, "error in RTP prebuffering!" );
653             break;
654         }
655     }
656
657     msg_Dbg( p_access, "RTP: prebuffered %d packets", i_count - 1 );
658     p_access->info.b_prebuffered = VLC_TRUE;
659     p = p_sys->p_list;
660     p_sys->p_list = p_sys->p_list->p_next;
661     p_sys->i_last_seqno = (uint16_t) p->i_dts;
662     p->p_next = NULL;
663     return p;
664 }
665
666 static block_t *BlockRTP( access_t *p_access )
667 {
668     access_sys_t *p_sys = p_access->p_sys;
669     block_t *p;
670
671     while ( !p_sys->p_list ||
672              ( mdate() - p_sys->p_list->i_pts ) < p_sys->i_rtp_late )
673     {
674         p = BlockParseRTP( p_access,
675                            p_sys->b_framed_rtp ? BlockTCP( p_access )
676                                                : BlockUDP( p_access ) );
677         if ( !p )
678             return NULL;
679
680         rtp_ChainInsert( p_access, p );
681     }
682
683     p = p_sys->p_list;
684     p_sys->p_list = p_sys->p_list->p_next;
685     p_sys->i_last_seqno++;
686     if( p_sys->i_last_seqno != (uint16_t) p->i_dts )
687     {
688         msg_Dbg( p_access, "RTP: packet(s) lost, expected %d, got %d",
689                  p_sys->i_last_seqno, (uint16_t) p->i_dts );
690         p_sys->i_last_seqno = (uint16_t) p->i_dts;
691     }
692     p->p_next = NULL;
693     return p;
694 }
695
696 /*****************************************************************************
697  * BlockChoose: decide between RTP and UDP
698  *****************************************************************************/
699 static block_t *BlockChoose( access_t *p_access )
700 {
701     block_t *p_block;
702     int     i_rtp_version;
703     int     i_payload_type;
704
705     if( ( p_block = BlockUDP( p_access ) ) == NULL )
706         return NULL;
707
708     if( p_block->p_buffer[0] == 0x47 )
709     {
710         msg_Dbg( p_access, "detected TS over raw UDP" );
711         p_access->pf_block = BlockUDP;
712         p_access->info.b_prebuffered = VLC_TRUE;
713         return p_block;
714     }
715
716     if( p_block->i_buffer < RTP_HEADER_LEN )
717         return p_block;
718
719     /* Parse the header and make some verifications.
720      * See RFC 3550. */
721
722     i_rtp_version  = p_block->p_buffer[0] >> 6;
723     i_payload_type = ( p_block->p_buffer[1] & 0x7F );
724
725     if( i_rtp_version != 2 )
726     {
727         msg_Dbg( p_access, "no supported RTP header detected" );
728         p_access->pf_block = BlockUDP;
729         p_access->info.b_prebuffered = VLC_TRUE;
730         return p_block;
731     }
732
733     switch( i_payload_type )
734     {
735         case 33:
736             msg_Dbg( p_access, "detected MPEG2 TS over RTP" );
737             p_access->psz_demux = strdup( "ts" );
738             break;
739
740         case 14:
741             msg_Dbg( p_access, "detected MPEG Audio over RTP" );
742             p_access->psz_demux = strdup( "mpga" );
743             break;
744
745         case 32:
746             msg_Dbg( p_access, "detected MPEG Video over RTP" );
747             p_access->psz_demux = strdup( "mpgv" );
748             break;
749
750         default:
751             msg_Dbg( p_access, "no RTP header detected" );
752             p_access->pf_block = BlockUDP;
753             p_access->info.b_prebuffered = VLC_TRUE;
754             return p_block;
755     }
756
757     if( !BlockParseRTP( p_access, p_block )) return NULL;
758
759     p_access->pf_block = BlockRTP;
760
761     return BlockPrebufferRTP( p_access, p_block );
762 }