1 /*****************************************************************************
2 * real.c: Real demuxer.
3 *****************************************************************************
4 * Copyright (C) 2004 VideoLAN (Centrale Réseaux) and its contributors
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
27 #include <stdlib.h> /* malloc(), free() */
30 #include <vlc/input.h>
32 /*****************************************************************************
34 *****************************************************************************/
35 static int Open ( vlc_object_t * );
36 static void Close ( vlc_object_t * );
39 set_description( _("Real demuxer" ) );
40 set_capability( "demux2", 15 );
41 set_category( CAT_INPUT );
42 set_subcategory( SUBCAT_INPUT_DEMUX );
43 set_callbacks( Open, Close );
44 add_shortcut( "real" );
48 /*****************************************************************************
50 *****************************************************************************/
66 int64_t i_data_offset;
68 uint32_t i_data_packets_count;
69 uint32_t i_data_packets;
70 int64_t i_data_offset_next;
75 uint8_t buffer[65536];
80 static int Demux( demux_t *p_demux );
81 static int Control( demux_t *p_demux, int i_query, va_list args );
83 static int HeaderRead( demux_t *p_demux );
85 /*****************************************************************************
87 *****************************************************************************/
88 static int Open( vlc_object_t *p_this )
90 demux_t *p_demux = (demux_t*)p_this;
95 if( stream_Peek( p_demux->s, &p_peek, 10 ) < 10 ) return VLC_EGENERIC;
96 if( strncmp( p_peek, ".RMF", 4 ) ) return VLC_EGENERIC;
98 /* Fill p_demux field */
99 p_demux->pf_demux = Demux;
100 p_demux->pf_control = Control;
101 p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
102 p_sys->i_data_offset = 0;
108 /* Parse the headers */
109 if( HeaderRead( p_demux ) )
112 msg_Err( p_demux, "invalid header" );
113 for( i = 0; i < p_sys->i_track; i++ )
115 real_track_t *tk = p_sys->track[i];
119 es_out_Del( p_demux->out, tk->p_es );
123 if( p_sys->i_track > 0 )
125 free( p_sys->track );
134 /*****************************************************************************
136 *****************************************************************************/
137 static void Close( vlc_object_t *p_this )
139 demux_t *p_demux = (demux_t*)p_this;
140 demux_sys_t *p_sys = p_demux->p_sys;
143 for( i = 0; i < p_sys->i_track; i++ )
145 real_track_t *tk = p_sys->track[i];
149 block_Release( tk->p_frame );
154 if( p_sys->i_track > 0 )
156 free( p_sys->track );
163 /*****************************************************************************
165 *****************************************************************************/
166 static int Demux( demux_t *p_demux )
168 demux_sys_t *p_sys = p_demux->p_sys;
174 real_track_t *tk = NULL;
175 vlc_bool_t b_selected;
177 if( p_sys->i_data_packets >= p_sys->i_data_packets_count )
180 if( stream_Read( p_demux->s, header, 18 ) < 18 )
184 if( strncmp( header, "DATA", 4 ) )
188 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 18;
189 p_sys->i_data_size = GetDWBE( &header[4] );
190 p_sys->i_data_packets_count = GetDWBE( &header[10] );
191 p_sys->i_data_packets = 0;
192 p_sys->i_data_offset_next = GetDWBE( &header[14] );
194 msg_Dbg( p_demux, "entering new DATA packets=%d next=%u",
195 p_sys->i_data_packets_count,
196 (uint32_t)p_sys->i_data_offset_next );
199 if( stream_Read( p_demux->s, header, 12 ) < 12 )
203 i_size = GetWBE( &header[2] ) - 12;
204 i_id = GetWBE( &header[4] );
205 i_pts = 1000 * GetDWBE( &header[6] );
206 /* header[11] -> flags 0x02 -> keyframe */
207 msg_Dbg( p_demux, "packet %d size=%d id=%d pts=%u",
208 p_sys->i_data_packets, i_size, i_id, (uint32_t)(i_pts/1000) );
209 p_sys->i_data_packets++;
211 stream_Read( p_demux->s, p_sys->buffer, i_size );
213 for( i = 0; i < p_sys->i_track; i++ )
215 if( p_sys->track[i]->i_id == i_id )
217 tk = p_sys->track[i];
223 msg_Warn( p_demux, "unknown track id(0x%x)", i_id );
226 es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b_selected );
228 if( tk->fmt.i_cat == VIDEO_ES && b_selected )
230 uint8_t *p = p_sys->buffer;
232 while( p < &p_sys->buffer[i_size - 2] )
241 if( (h&0xc0) == 0x40 )
245 i_len = &p_sys->buffer[i_size] - p;
251 i_subseq = (*p++)&0x7f;
253 i_len = (p[0] << 8)|p[1]; p += 2;
254 if( (i_len&0xc000) == 0 )
257 i_len |= (p[0] << 8)|p[1]; p += 2;
265 i_offset = (p[0] << 8)|p[1]; p += 2;
266 if( (i_offset&0xc000) == 0 )
269 i_offset |= (p[0] << 8)|p[1]; p += 2;
270 i_offset &= 0x3fffffff;
278 i_copy = i_len - i_offset;
279 if( i_copy > &p_sys->buffer[i_size] - p )
281 i_copy = &p_sys->buffer[i_size] - p;
283 else if( i_copy < 0 )
288 msg_Dbg( p_demux, " - len=%d offset=%d size=%d subseq=%d seqnum=%d",
289 i_len, i_offset, i_copy, i_subseq, i_seqnum );
291 if( (h&0xc0) == 0x80 )
293 /* last fragment -> fixes */
295 i_offset = i_len - i_copy;
296 msg_Dbg( p_demux, "last fixing copy=%d offset=%d", i_copy, i_offset );
300 ( tk->p_frame->i_dts != i_pts ||
301 tk->i_frame != i_len ) )
303 msg_Dbg( p_demux, "sending size=%d", tk->p_frame->i_buffer );
305 if( p_sys->i_pcr < tk->p_frame->i_dts )
307 p_sys->i_pcr = tk->p_frame->i_dts;
309 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
311 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
317 if( (h&0xc0) != 0x80 && (h&0xc0) != 0x00 && tk->p_frame == NULL )
325 if( tk->p_frame == NULL )
327 msg_Dbg( p_demux, "new frame size=%d", i_len );
329 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
333 memset( &tk->p_frame->p_buffer[8], 0, i_len );
334 tk->p_frame->i_dts = i_pts;
335 tk->p_frame->i_pts = i_pts;
337 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
338 ((uint32_t*)tk->p_frame->p_buffer)[1] = 0; /* chunk counts */
341 if( i_offset < tk->i_frame)
343 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
345 msg_Dbg( p_demux, "copying new buffer n=%d offset=%d copy=%d", i_ck, i_offset, i_copy );
347 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
349 memcpy( &tk->p_frame->p_buffer[i_offset + 8],
355 if( (h&0xc0) != 0x80 )
363 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
365 if( (h&0xc0) == 0x80 )
369 i_offset = i_len - i_offset;
371 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
372 memcpy( &tk->p_frame->p_buffer[i_offset+ 8], p, i_copy );
375 if( p_sys->i_pcr < tk->p_frame->i_dts )
377 p_sys->i_pcr = tk->p_frame->i_dts;
378 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
380 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
388 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
389 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
393 if( (h&0xc0) != 0x00 )
398 if( ( p_frame = block_New( p_demux, i_copy + 8 + 8 ) ) == NULL )
402 p_frame->i_dts = i_pts;
403 p_frame->i_pts = i_pts;
405 ((uint32_t*)p_frame->p_buffer)[0] = i_copy;
406 ((uint32_t*)p_frame->p_buffer)[1] = 1;
407 ((uint32_t*)(p_frame->p_buffer+i_copy+8))[0] = 0;
408 memcpy( &p_frame->p_buffer[8], p, i_copy );
412 if( p_sys->i_pcr < p_frame->i_dts )
414 p_sys->i_pcr = p_frame->i_dts;
415 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
417 es_out_Send( p_demux->out, tk->p_es, p_frame );
423 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
427 memset( &tk->p_frame->p_buffer[8], 0, i_len );
428 tk->p_frame->i_dts = i_pts;
429 tk->p_frame->i_pts = i_pts;
431 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
432 ((uint32_t*)tk->p_frame->p_buffer)[1] = 1; /* chunk counts */
433 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[0] = i_offset;
434 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
441 else if( tk->fmt.i_cat == AUDIO_ES && b_selected )
444 if( p_sys->i_pcr < i_pts )
446 p_sys->i_pcr = i_pts;
447 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
450 if( tk->fmt.i_codec == VLC_FOURCC( 'm', 'p', '4', 'a' ) )
452 int i_sub = (p_sys->buffer[1] >> 4)&0x0f;
453 uint8_t *p_sub = &p_sys->buffer[2+2*i_sub];
456 for( i = 0; i < i_sub; i++ )
458 int i_sub_size = GetWBE( &p_sys->buffer[2+i*2]);
459 block_t *p_block = block_New( p_demux, i_sub_size );
462 memcpy( p_block->p_buffer, p_sub, i_sub_size );
466 p_block->i_pts = ( i == 0 ? i_pts : 0 );
468 es_out_Send( p_demux->out, tk->p_es, p_block );
474 block_t *p_block = block_New( p_demux, i_size );
476 if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
478 uint8_t *src = p_sys->buffer;
479 uint8_t *dst = p_block->p_buffer;
482 while( dst < &p_block->p_buffer[i_size- 1])
492 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
495 p_block->i_pts = i_pts;
497 es_out_Send( p_demux->out, tk->p_es, p_block );
505 /*****************************************************************************
507 *****************************************************************************/
508 static int Control( demux_t *p_demux, int i_query, va_list args )
511 demux_sys_t *p_sys = p_demux->p_sys;
517 case DEMUX_GET_POSITION:
518 pf = (double*) va_arg( args, double* );
519 i64 = stream_Size( p_demux->s );
522 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
529 case DEMUX_SET_POSITION:
530 f = (double) va_arg( args, double );
531 i64 = stream_Size( p_demux->s );
533 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
535 return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
538 pi64 = (int64_t*)va_arg( args, int64_t * );
539 if( p_sys->i_mux_rate > 0 )
541 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
547 case DEMUX_GET_LENGTH:
548 pi64 = (int64_t*)va_arg( args, int64_t * );
549 if( p_sys->i_mux_rate > 0 )
551 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
566 /*****************************************************************************
568 *****************************************************************************/
569 static int HeaderRead( demux_t *p_demux )
571 demux_sys_t *p_sys = p_demux->p_sys;
572 uint8_t header[100]; /* FIXME */
581 /* Read the header */
582 if( stream_Read( p_demux->s, header, 10 ) < 10 )
586 i_id = VLC_FOURCC( header[0], header[1], header[2], header[3] );
587 i_size = GetDWBE( &header[4] );
588 i_version = GetWBE( &header[8] );
590 msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
591 (char*)&i_id, i_size, i_version );
597 i_skip = i_size - 10;
599 if( i_id == VLC_FOURCC('.','R','M','F') )
601 if( stream_Read( p_demux->s, header, 8 ) < 8 )
605 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
606 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
610 else if( i_id == VLC_FOURCC('P','R','O','P') )
614 if( stream_Read( p_demux->s, header, 40 ) < 40 )
618 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
619 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
620 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
621 GetDWBE( &header[8] ), GetDWBE( &header[12] ) );
622 msg_Dbg( p_demux, " - packets count=%d", GetDWBE( &header[16] ) );
623 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE( &header[20] ) );
624 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE( &header[24] ) );
625 msg_Dbg( p_demux, " - index offset=%d", GetDWBE( &header[28] ) );
626 msg_Dbg( p_demux, " - data offset=%d", GetDWBE( &header[32] ) );
627 msg_Dbg( p_demux, " - num streams=%d", GetWBE( &header[36] ) );
628 i_flags = GetWBE( &header[38]);
629 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
631 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
632 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
633 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
636 else if( i_id == VLC_FOURCC('C','O','N','T') )
641 stream_Read( p_demux->s, header, 2 );
642 if( ( i_len = GetWBE( header ) ) > 0 )
644 psz = malloc( i_len + 1 );
645 stream_Read( p_demux->s, psz, i_len );
648 msg_Dbg( p_demux, " - title=`%s'", psz );
654 stream_Read( p_demux->s, header, 2 );
655 if( ( i_len = GetWBE( header ) ) > 0 )
657 psz = malloc( i_len + 1 );
658 stream_Read( p_demux->s, psz, i_len );
661 msg_Dbg( p_demux, " - author=`%s'", psz );
667 stream_Read( p_demux->s, header, 2 );
668 if( ( i_len = GetWBE( header ) ) > 0 )
670 psz = malloc( i_len + 1 );
671 stream_Read( p_demux->s, psz, i_len );
674 msg_Dbg( p_demux, " - copyright=`%s'", psz );
680 stream_Read( p_demux->s, header, 2 );
681 if( ( i_len = GetWBE( header ) ) > 0 )
683 psz = malloc( i_len + 1 );
684 stream_Read( p_demux->s, psz, i_len );
687 msg_Dbg( p_demux, " - comment=`%s'", psz );
693 else if( i_id == VLC_FOURCC('M','D','P','R') )
699 if( stream_Read( p_demux->s, header, 30 ) < 30 )
703 i_num = GetWBE( header );
704 msg_Dbg( p_demux, " - id=0x%x", i_num );
705 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d", GetDWBE( &header[2] ), GetDWBE( &header[6] ) );
706 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d", GetDWBE( &header[10] ), GetDWBE( &header[14] ));
707 msg_Dbg( p_demux, " - start time=%d", GetDWBE( &header[18] ) );
708 msg_Dbg( p_demux, " - preroll=%d", GetDWBE( &header[22] ) );
709 msg_Dbg( p_demux, " - duration=%d", GetDWBE( &header[26] ));
713 stream_Read( p_demux->s, header, 1 );
714 if( ( i_len = header[0] ) > 0 )
716 psz = malloc( i_len + 1 );
717 stream_Read( p_demux->s, psz, i_len );
720 msg_Dbg( p_demux, " - name=`%s'", psz );
726 stream_Read( p_demux->s, header, 1 );
727 if( ( i_len = header[0] ) > 0 )
729 psz = malloc( i_len + 1 );
730 stream_Read( p_demux->s, psz, i_len );
733 msg_Dbg( p_demux, " - mime=`%s'", psz );
739 stream_Read( p_demux->s, header, 4 );
740 if( ( i_len = GetDWBE( header ) ) > 0 )
746 msg_Dbg( p_demux, " - specific data len=%d", i_len );
747 if( stream_Peek( p_demux->s, &p_peek, 34 ) >= 34 )
749 if( !strncmp( &p_peek[4], "VIDO", 4 ) )
751 es_format_Init( &fmt, VIDEO_ES,
752 VLC_FOURCC( p_peek[8], p_peek[9], p_peek[10], p_peek[11] ) );
753 fmt.video.i_width = GetWBE( &p_peek[12] );
754 fmt.video.i_height= GetWBE( &p_peek[14] );
757 fmt.p_extra = malloc( 8 );
758 ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
759 ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
761 msg_Dbg( p_demux, " - video 0x%08x 0x%08x",
762 ((uint32_t*)fmt.p_extra)[0],
763 ((uint32_t*)fmt.p_extra)[1] );
765 if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
766 GetDWBE( &p_peek[30] ) == 0x10003001 )
768 fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
771 msg_Dbg( p_demux, " - video %4.4s %dx%d",
773 fmt.video.i_width, fmt.video.i_height );
775 tk = malloc( sizeof( real_track_t ) );
780 tk->p_es = es_out_Add( p_demux->out, &fmt );
782 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
784 else if( !strncmp( p_peek, ".ra\xfd", 4 ) )
786 int i_version = GetWBE( &p_peek[4] );
787 uint8_t *p_extra = NULL;
788 msg_Dbg( p_demux, " - audio version=%d", i_version );
790 es_format_Init( &fmt, AUDIO_ES, 0 );
791 if( i_version == 4 && stream_Peek( p_demux->s, &p_peek, 56 ) >= 56 )
793 fmt.audio.i_channels = GetWBE( &p_peek[54] );
794 fmt.audio.i_rate = GetWBE( &p_peek[48] );
796 if( stream_Peek( p_demux->s, &p_peek, 57 ) >= 57 )
798 int i_extra = p_peek[56] + 1 + 4;
799 if( stream_Peek( p_demux->s, &p_peek, 57 + i_extra ) >= 57 + i_extra )
801 memcpy( &fmt.i_codec, &p_peek[57 + p_peek[56] + 1], 4 );
803 p_extra = &p_peek[57 + p_peek[56] + 1+ 4 + 3];
806 else if( i_version == 5 && stream_Peek( p_demux->s, &p_peek, 70 ) >= 70 )
808 memcpy( &fmt.i_codec, &p_peek[66], 4 );
809 fmt.audio.i_channels = GetWBE( &p_peek[60] );
810 fmt.audio.i_rate = GetWBE( &p_peek[54] );
812 p_extra = &p_peek[66+4+3+1];
814 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
816 fmt.audio.i_channels, fmt.audio.i_rate );
818 if( fmt.i_codec == VLC_FOURCC( 'd', 'n', 'e', 't' ) )
820 fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
822 else if( fmt.i_codec == VLC_FOURCC( 'r', 'a', 'a', 'c' ) ||
823 fmt.i_codec == VLC_FOURCC( 'r', 'a', 'c', 'p' ) )
825 int i_peek = p_extra - p_peek;
826 if( stream_Peek( p_demux->s, &p_peek, i_peek + 4 ) >= i_peek + 4 )
828 int i_extra = GetDWBE( &p_peek[i_peek] );
830 if( i_extra > 1 && i_peek + 4 + i_extra <= i_len &&
831 stream_Peek( p_demux->s, &p_peek, i_peek + 4 + i_extra ) >= i_peek + 4 + i_extra )
833 fmt.i_extra = i_extra - 1;
834 fmt.p_extra = malloc ( i_extra - 1 );
835 memcpy( fmt.p_extra, &p_peek[i_peek+4+1], i_extra - 1 );
837 msg_Dbg( p_demux, " - extra data=%d", i_extra );
840 for( i = 0; i < fmt.i_extra; i++ )
842 msg_Dbg( p_demux, " data[%d] = 0x%x", i,((uint8_t*)fmt.p_extra)[i] );
847 fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
850 if( fmt.i_codec != 0 )
852 tk = malloc( sizeof( real_track_t ) );
857 tk->p_es = es_out_Add( p_demux->out, &fmt );
859 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
863 stream_Read( p_demux->s, NULL, i_len );
869 else if( i_id == VLC_FOURCC('D','A','T','A') )
871 stream_Read( p_demux->s, header, 8 );
873 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 10;
874 p_sys->i_data_size = i_size;
875 p_sys->i_data_packets_count = GetDWBE( header );
876 p_sys->i_data_packets = 0;
877 p_sys->i_data_offset_next = GetDWBE( &header[4] );
879 msg_Dbg( p_demux, " - packets count=%d next=%u",
880 p_sys->i_data_packets_count,
881 (uint32_t)p_sys->i_data_offset_next );
883 /* we have finished the header */
889 msg_Dbg( p_demux, "unknown chunk" );
896 stream_Read( p_demux->s, NULL, i_skip );
899 /* TODO read index if possible */