1 /*****************************************************************************
2 * real.c: Real demuxer.
3 *****************************************************************************
4 * Copyright (C) 2004 VideoLAN
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_callbacks( Open, Close );
42 add_shortcut( "real" );
46 /*****************************************************************************
48 *****************************************************************************/
64 int64_t i_data_offset;
66 uint32_t i_data_packets_count;
67 uint32_t i_data_packets;
68 int64_t i_data_offset_next;
73 uint8_t buffer[65536];
78 static int Demux( demux_t *p_demux );
79 static int Control( demux_t *p_demux, int i_query, va_list args );
81 static int HeaderRead( demux_t *p_demux );
83 /*****************************************************************************
85 *****************************************************************************/
86 static int Open( vlc_object_t *p_this )
88 demux_t *p_demux = (demux_t*)p_this;
93 if( stream_Peek( p_demux->s, &p_peek, 10 ) < 10 )
95 msg_Err( p_demux, "cannot peek" );
98 if( strncmp( p_peek, ".RMF", 4 ) )
100 msg_Warn( p_demux, "Real module discarded" );
104 /* Fill p_demux field */
105 p_demux->pf_demux = Demux;
106 p_demux->pf_control = Control;
107 p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
108 p_sys->i_data_offset = 0;
114 /* Parse the headers */
115 if( HeaderRead( p_demux ) )
118 msg_Err( p_demux, "invalid header" );
119 for( i = 0; i < p_sys->i_track; i++ )
121 real_track_t *tk = p_sys->track[i];
125 es_out_Del( p_demux->out, tk->p_es );
129 if( p_sys->i_track > 0 )
131 free( p_sys->track );
140 /*****************************************************************************
142 *****************************************************************************/
143 static void Close( vlc_object_t *p_this )
145 demux_t *p_demux = (demux_t*)p_this;
146 demux_sys_t *p_sys = p_demux->p_sys;
149 for( i = 0; i < p_sys->i_track; i++ )
151 real_track_t *tk = p_sys->track[i];
155 block_Release( tk->p_frame );
160 if( p_sys->i_track > 0 )
162 free( p_sys->track );
169 /*****************************************************************************
171 *****************************************************************************/
172 static int Demux( demux_t *p_demux )
174 demux_sys_t *p_sys = p_demux->p_sys;
180 real_track_t *tk = NULL;
181 vlc_bool_t b_selected;
183 if( p_sys->i_data_packets >= p_sys->i_data_packets_count )
186 if( stream_Read( p_demux->s, header, 18 ) < 18 )
190 if( strncmp( header, "DATA", 4 ) )
194 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 18;
195 p_sys->i_data_size = GetDWBE( &header[4] );
196 p_sys->i_data_packets_count = GetDWBE( &header[10] );
197 p_sys->i_data_packets = 0;
198 p_sys->i_data_offset_next = GetDWBE( &header[14] );
200 msg_Dbg( p_demux, "entering new DATA packets=%d next=%u",
201 p_sys->i_data_packets_count,
202 (uint32_t)p_sys->i_data_offset_next );
205 if( stream_Read( p_demux->s, header, 12 ) < 12 )
209 i_size = GetWBE( &header[2] ) - 12;
210 i_id = GetWBE( &header[4] );
211 i_pts = 1000 * GetDWBE( &header[6] );
212 /* header[11] -> flags 0x02 -> keyframe */
213 msg_Dbg( p_demux, "packet %d size=%d id=%d pts=%u",
214 p_sys->i_data_packets, i_size, i_id, (uint32_t)(i_pts/1000) );
215 p_sys->i_data_packets++;
217 stream_Read( p_demux->s, p_sys->buffer, i_size );
219 for( i = 0; i < p_sys->i_track; i++ )
221 if( p_sys->track[i]->i_id == i_id )
223 tk = p_sys->track[i];
229 msg_Warn( p_demux, "unknown track id(0x%x)", i_id );
232 es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b_selected );
234 if( tk->fmt.i_cat == VIDEO_ES && b_selected )
236 uint8_t *p = p_sys->buffer;
238 while( p < &p_sys->buffer[i_size - 2] )
247 if( (h&0xc0) == 0x40 )
251 i_len = &p_sys->buffer[i_size] - p;
257 i_subseq = (*p++)&0x7f;
259 i_len = (p[0] << 8)|p[1]; p += 2;
260 if( (i_len&0xc000) == 0 )
263 i_len |= (p[0] << 8)|p[1]; p += 2;
271 i_offset = (p[0] << 8)|p[1]; p += 2;
272 if( (i_offset&0xc000) == 0 )
275 i_offset |= (p[0] << 8)|p[1]; p += 2;
276 i_offset &= 0x3fffffff;
284 i_copy = i_len - i_offset;
285 if( i_copy > &p_sys->buffer[i_size] - p )
287 i_copy = &p_sys->buffer[i_size] - p;
289 else if( i_copy < 0 )
294 msg_Dbg( p_demux, " - len=%d offset=%d size=%d subseq=%d seqnum=%d",
295 i_len, i_offset, i_copy, i_subseq, i_seqnum );
297 if( (h&0xc0) == 0x80 )
299 /* last fragment -> fixes */
301 i_offset = i_len - i_copy;
302 msg_Dbg( p_demux, "last fixing copy=%d offset=%d", i_copy, i_offset );
306 ( tk->p_frame->i_dts != i_pts ||
307 tk->i_frame != i_len ) )
309 msg_Dbg( p_demux, "sending size=%d", tk->p_frame->i_buffer );
311 if( p_sys->i_pcr < tk->p_frame->i_dts )
313 p_sys->i_pcr = tk->p_frame->i_dts;
315 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
317 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
323 if( (h&0xc0) != 0x80 && (h&0xc0) != 0x00 && tk->p_frame == NULL )
331 if( tk->p_frame == NULL )
333 msg_Dbg( p_demux, "new frame size=%d", i_len );
335 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
339 memset( &tk->p_frame->p_buffer[8], 0, i_len );
340 tk->p_frame->i_dts = i_pts;
341 tk->p_frame->i_pts = i_pts;
343 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
344 ((uint32_t*)tk->p_frame->p_buffer)[1] = 0; /* chunk counts */
347 if( i_offset < tk->i_frame)
349 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
351 msg_Dbg( p_demux, "copying new buffer n=%d offset=%d copy=%d", i_ck, i_offset, i_copy );
353 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
355 memcpy( &tk->p_frame->p_buffer[i_offset + 8],
361 if( (h&0xc0) != 0x80 )
369 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
371 if( (h&0xc0) == 0x80 )
375 i_offset = i_len - i_offset;
377 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
378 memcpy( &tk->p_frame->p_buffer[i_offset+ 8], p, i_copy );
381 if( p_sys->i_pcr < tk->p_frame->i_dts )
383 p_sys->i_pcr = tk->p_frame->i_dts;
384 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
386 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
394 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
395 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
399 if( (h&0xc0) != 0x00 )
404 if( ( p_frame = block_New( p_demux, i_copy + 8 + 8 ) ) == NULL )
408 p_frame->i_dts = i_pts;
409 p_frame->i_pts = i_pts;
411 ((uint32_t*)p_frame->p_buffer)[0] = i_copy;
412 ((uint32_t*)p_frame->p_buffer)[1] = 1;
413 ((uint32_t*)(p_frame->p_buffer+i_copy+8))[0] = 0;
414 memcpy( &p_frame->p_buffer[8], p, i_copy );
418 if( p_sys->i_pcr < p_frame->i_dts )
420 p_sys->i_pcr = p_frame->i_dts;
421 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
423 es_out_Send( p_demux->out, tk->p_es, p_frame );
429 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
433 memset( &tk->p_frame->p_buffer[8], 0, i_len );
434 tk->p_frame->i_dts = i_pts;
435 tk->p_frame->i_pts = i_pts;
437 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
438 ((uint32_t*)tk->p_frame->p_buffer)[1] = 1; /* chunk counts */
439 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[0] = i_offset;
440 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
447 else if( tk->fmt.i_cat == AUDIO_ES && b_selected )
450 if( p_sys->i_pcr < i_pts )
452 p_sys->i_pcr = i_pts;
453 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
456 if( tk->fmt.i_codec == VLC_FOURCC( 'm', 'p', '4', 'a' ) )
458 int i_sub = (p_sys->buffer[1] >> 4)&0x0f;
459 uint8_t *p_sub = &p_sys->buffer[2+2*i_sub];
462 for( i = 0; i < i_sub; i++ )
464 int i_sub_size = GetWBE( &p_sys->buffer[2+i*2]);
465 block_t *p_block = block_New( p_demux, i_sub_size );
468 memcpy( p_block->p_buffer, p_sub, i_sub_size );
472 p_block->i_pts = ( i == 0 ? i_pts : 0 );
474 es_out_Send( p_demux->out, tk->p_es, p_block );
480 block_t *p_block = block_New( p_demux, i_size );
482 if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
484 uint8_t *src = p_sys->buffer;
485 uint8_t *dst = p_block->p_buffer;
488 while( dst < &p_block->p_buffer[i_size- 1])
498 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
501 p_block->i_pts = i_pts;
503 es_out_Send( p_demux->out, tk->p_es, p_block );
511 /*****************************************************************************
513 *****************************************************************************/
514 static int Control( demux_t *p_demux, int i_query, va_list args )
517 demux_sys_t *p_sys = p_demux->p_sys;
523 case DEMUX_GET_POSITION:
524 pf = (double*) va_arg( args, double* );
525 i64 = stream_Size( p_demux->s );
528 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
535 case DEMUX_SET_POSITION:
536 f = (double) va_arg( args, double );
537 i64 = stream_Size( p_demux->s );
539 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
541 return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
544 pi64 = (int64_t*)va_arg( args, int64_t * );
545 if( p_sys->i_mux_rate > 0 )
547 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
553 case DEMUX_GET_LENGTH:
554 pi64 = (int64_t*)va_arg( args, int64_t * );
555 if( p_sys->i_mux_rate > 0 )
557 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
572 /*****************************************************************************
574 *****************************************************************************/
575 static int HeaderRead( demux_t *p_demux )
577 demux_sys_t *p_sys = p_demux->p_sys;
578 uint8_t header[100]; /* FIXME */
587 /* Read the header */
588 if( stream_Read( p_demux->s, header, 10 ) < 10 )
592 i_id = VLC_FOURCC( header[0], header[1], header[2], header[3] );
593 i_size = GetDWBE( &header[4] );
594 i_version = GetWBE( &header[8] );
596 msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
597 (char*)&i_id, i_size, i_version );
603 i_skip = i_size - 10;
605 if( i_id == VLC_FOURCC('.','R','M','F') )
607 if( stream_Read( p_demux->s, header, 8 ) < 8 )
611 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
612 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
616 else if( i_id == VLC_FOURCC('P','R','O','P') )
620 if( stream_Read( p_demux->s, header, 40 ) < 40 )
624 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
625 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
626 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
627 GetDWBE( &header[8] ), GetDWBE( &header[12] ) );
628 msg_Dbg( p_demux, " - packets count=%d", GetDWBE( &header[16] ) );
629 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE( &header[20] ) );
630 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE( &header[24] ) );
631 msg_Dbg( p_demux, " - index offset=%d", GetDWBE( &header[28] ) );
632 msg_Dbg( p_demux, " - data offset=%d", GetDWBE( &header[32] ) );
633 msg_Dbg( p_demux, " - num streams=%d", GetWBE( &header[36] ) );
634 i_flags = GetWBE( &header[38]);
635 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
637 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
638 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
639 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
642 else if( i_id == VLC_FOURCC('C','O','N','T') )
647 stream_Read( p_demux->s, header, 2 );
648 if( ( i_len = GetWBE( header ) ) > 0 )
650 psz = malloc( i_len + 1 );
651 stream_Read( p_demux->s, psz, i_len );
654 msg_Dbg( p_demux, " - title=`%s'", psz );
660 stream_Read( p_demux->s, header, 2 );
661 if( ( i_len = GetWBE( header ) ) > 0 )
663 psz = malloc( i_len + 1 );
664 stream_Read( p_demux->s, psz, i_len );
667 msg_Dbg( p_demux, " - author=`%s'", psz );
673 stream_Read( p_demux->s, header, 2 );
674 if( ( i_len = GetWBE( header ) ) > 0 )
676 psz = malloc( i_len + 1 );
677 stream_Read( p_demux->s, psz, i_len );
680 msg_Dbg( p_demux, " - copyright=`%s'", psz );
686 stream_Read( p_demux->s, header, 2 );
687 if( ( i_len = GetWBE( header ) ) > 0 )
689 psz = malloc( i_len + 1 );
690 stream_Read( p_demux->s, psz, i_len );
693 msg_Dbg( p_demux, " - comment=`%s'", psz );
699 else if( i_id == VLC_FOURCC('M','D','P','R') )
705 if( stream_Read( p_demux->s, header, 30 ) < 30 )
709 i_num = GetWBE( header );
710 msg_Dbg( p_demux, " - id=0x%x", i_num );
711 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d", GetDWBE( &header[2] ), GetDWBE( &header[6] ) );
712 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d", GetDWBE( &header[10] ), GetDWBE( &header[14] ));
713 msg_Dbg( p_demux, " - start time=%d", GetDWBE( &header[18] ) );
714 msg_Dbg( p_demux, " - preroll=%d", GetDWBE( &header[22] ) );
715 msg_Dbg( p_demux, " - duration=%d", GetDWBE( &header[26] ));
719 stream_Read( p_demux->s, header, 1 );
720 if( ( i_len = header[0] ) > 0 )
722 psz = malloc( i_len + 1 );
723 stream_Read( p_demux->s, psz, i_len );
726 msg_Dbg( p_demux, " - name=`%s'", psz );
732 stream_Read( p_demux->s, header, 1 );
733 if( ( i_len = header[0] ) > 0 )
735 psz = malloc( i_len + 1 );
736 stream_Read( p_demux->s, psz, i_len );
739 msg_Dbg( p_demux, " - mime=`%s'", psz );
745 stream_Read( p_demux->s, header, 4 );
746 if( ( i_len = GetDWBE( header ) ) > 0 )
752 msg_Dbg( p_demux, " - specific data len=%d", i_len );
753 if( stream_Peek( p_demux->s, &p_peek, 34 ) >= 34 )
755 if( !strncmp( &p_peek[4], "VIDO", 4 ) )
757 es_format_Init( &fmt, VIDEO_ES,
758 VLC_FOURCC( p_peek[8], p_peek[9], p_peek[10], p_peek[11] ) );
759 fmt.video.i_width = GetWBE( &p_peek[12] );
760 fmt.video.i_height= GetWBE( &p_peek[14] );
763 fmt.p_extra = malloc( 8 );
764 ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
765 ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
767 msg_Dbg( p_demux, " - video 0x%08x 0x%08x",
768 ((uint32_t*)fmt.p_extra)[0],
769 ((uint32_t*)fmt.p_extra)[1] );
771 if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
772 GetDWBE( &p_peek[30] ) == 0x10003001 )
774 fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
777 msg_Dbg( p_demux, " - video %4.4s %dx%d",
779 fmt.video.i_width, fmt.video.i_height );
781 tk = malloc( sizeof( real_track_t ) );
786 tk->p_es = es_out_Add( p_demux->out, &fmt );
788 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
790 else if( !strncmp( p_peek, ".ra\xfd", 4 ) )
792 int i_version = GetWBE( &p_peek[4] );
793 uint8_t *p_extra = NULL;
794 msg_Dbg( p_demux, " - audio version=%d", i_version );
796 es_format_Init( &fmt, AUDIO_ES, 0 );
797 if( i_version == 4 && stream_Peek( p_demux->s, &p_peek, 56 ) >= 56 )
799 fmt.audio.i_channels = GetWBE( &p_peek[54] );
800 fmt.audio.i_rate = GetWBE( &p_peek[48] );
802 if( stream_Peek( p_demux->s, &p_peek, 57 ) >= 57 )
804 int i_extra = p_peek[56] + 1 + 4;
805 if( stream_Peek( p_demux->s, &p_peek, 57 + i_extra ) >= 57 + i_extra )
807 memcpy( &fmt.i_codec, &p_peek[57 + p_peek[56] + 1], 4 );
809 p_extra = &p_peek[57 + p_peek[56] + 1+ 4 + 3];
812 else if( i_version == 5 && stream_Peek( p_demux->s, &p_peek, 70 ) >= 70 )
814 memcpy( &fmt.i_codec, &p_peek[66], 4 );
815 fmt.audio.i_channels = GetWBE( &p_peek[60] );
816 fmt.audio.i_rate = GetWBE( &p_peek[54] );
818 p_extra = &p_peek[66+4+3+1];
820 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
822 fmt.audio.i_channels, fmt.audio.i_rate );
824 if( fmt.i_codec == VLC_FOURCC( 'd', 'n', 'e', 't' ) )
826 fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
828 else if( fmt.i_codec == VLC_FOURCC( 'r', 'a', 'a', 'c' ) ||
829 fmt.i_codec == VLC_FOURCC( 'r', 'a', 'c', 'p' ) )
831 int i_peek = p_extra - p_peek;
832 if( stream_Peek( p_demux->s, &p_peek, i_peek + 4 ) >= i_peek + 4 )
834 int i_extra = GetDWBE( &p_peek[i_peek] );
836 if( i_extra > 1 && i_peek + 4 + i_extra <= i_len &&
837 stream_Peek( p_demux->s, &p_peek, i_peek + 4 + i_extra ) >= i_peek + 4 + i_extra )
839 fmt.i_extra = i_extra - 1;
840 fmt.p_extra = malloc ( i_extra - 1 );
841 memcpy( fmt.p_extra, &p_peek[i_peek+4+1], i_extra - 1 );
843 msg_Dbg( p_demux, " - extra data=%d", i_extra );
846 for( i = 0; i < fmt.i_extra; i++ )
848 msg_Dbg( p_demux, " data[%d] = 0x%x", i,((uint8_t*)fmt.p_extra)[i] );
853 fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
856 if( fmt.i_codec != 0 )
858 tk = malloc( sizeof( real_track_t ) );
863 tk->p_es = es_out_Add( p_demux->out, &fmt );
865 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
869 stream_Read( p_demux->s, NULL, i_len );
875 else if( i_id == VLC_FOURCC('D','A','T','A') )
877 stream_Read( p_demux->s, header, 8 );
879 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 10;
880 p_sys->i_data_size = i_size;
881 p_sys->i_data_packets_count = GetDWBE( header );
882 p_sys->i_data_packets = 0;
883 p_sys->i_data_offset_next = GetDWBE( &header[4] );
885 msg_Dbg( p_demux, " - packets count=%d next=%u",
886 p_sys->i_data_packets_count,
887 (uint32_t)p_sys->i_data_offset_next );
889 /* we have finished the header */
895 msg_Dbg( p_demux, "unknown chunk" );
902 stream_Read( p_demux->s, NULL, i_skip );
905 /* TODO read index if possible */