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", 1);
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 )
449 block_t *p_block = block_New( p_demux, i_size );
451 if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
453 uint8_t *src = p_sys->buffer;
454 uint8_t *dst = p_block->p_buffer;
457 while( dst < &p_block->p_buffer[i_size- 1])
467 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
470 p_block->i_pts = i_pts;
472 if( p_sys->i_pcr < i_pts )
474 p_sys->i_pcr = i_pts;
476 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
478 es_out_Send( p_demux->out, tk->p_es, p_block );
485 /*****************************************************************************
487 *****************************************************************************/
488 static int Control( demux_t *p_demux, int i_query, va_list args )
491 demux_sys_t *p_sys = p_demux->p_sys;
497 case DEMUX_GET_POSITION:
498 pf = (double*) va_arg( args, double* );
499 i64 = stream_Size( p_demux->s );
502 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
509 case DEMUX_SET_POSITION:
510 f = (double) va_arg( args, double );
511 i64 = stream_Size( p_demux->s );
513 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
515 return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
518 pi64 = (int64_t*)va_arg( args, int64_t * );
519 if( p_sys->i_mux_rate > 0 )
521 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
527 case DEMUX_GET_LENGTH:
528 pi64 = (int64_t*)va_arg( args, int64_t * );
529 if( p_sys->i_mux_rate > 0 )
531 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
546 /*****************************************************************************
548 *****************************************************************************/
549 static int HeaderRead( demux_t *p_demux )
551 demux_sys_t *p_sys = p_demux->p_sys;
552 uint8_t header[100]; /* FIXME */
561 /* Read the header */
562 if( stream_Read( p_demux->s, header, 10 ) < 10 )
566 i_id = VLC_FOURCC( header[0], header[1], header[2], header[3] );
567 i_size = GetDWBE( &header[4] );
568 i_version = GetWBE( &header[8] );
570 msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
571 (char*)&i_id, i_size, i_version );
577 i_skip = i_size - 10;
579 if( i_id == VLC_FOURCC('.','R','M','F') )
581 if( stream_Read( p_demux->s, header, 8 ) < 8 )
585 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
586 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
590 else if( i_id == VLC_FOURCC('P','R','O','P') )
594 if( stream_Read( p_demux->s, header, 40 ) < 40 )
598 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
599 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
600 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
601 GetDWBE( &header[8] ), GetDWBE( &header[12] ) );
602 msg_Dbg( p_demux, " - packets count=%d", GetDWBE( &header[16] ) );
603 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE( &header[20] ) );
604 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE( &header[24] ) );
605 msg_Dbg( p_demux, " - index offset=%d", GetDWBE( &header[28] ) );
606 msg_Dbg( p_demux, " - data offset=%d", GetDWBE( &header[32] ) );
607 msg_Dbg( p_demux, " - num streams=%d", GetWBE( &header[36] ) );
608 i_flags = GetWBE( &header[38]);
609 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
611 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
612 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
613 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
616 else if( i_id == VLC_FOURCC('C','O','N','T') )
621 stream_Read( p_demux->s, header, 2 );
622 if( ( i_len = GetWBE( header ) ) > 0 )
624 psz = malloc( i_len + 1 );
625 stream_Read( p_demux->s, psz, i_len );
628 msg_Dbg( p_demux, " - title=`%s'", psz );
634 stream_Read( p_demux->s, header, 2 );
635 if( ( i_len = GetWBE( header ) ) > 0 )
637 psz = malloc( i_len + 1 );
638 stream_Read( p_demux->s, psz, i_len );
641 msg_Dbg( p_demux, " - author=`%s'", psz );
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, " - copyright=`%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, " - comment=`%s'", psz );
673 else if( i_id == VLC_FOURCC('M','D','P','R') )
679 if( stream_Read( p_demux->s, header, 30 ) < 30 )
683 i_num = GetWBE( header );
684 msg_Dbg( p_demux, " - id=0x%x", i_num );
685 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d", GetDWBE( &header[2] ), GetDWBE( &header[6] ) );
686 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d", GetDWBE( &header[10] ), GetDWBE( &header[14] ));
687 msg_Dbg( p_demux, " - start time=%d", GetDWBE( &header[18] ) );
688 msg_Dbg( p_demux, " - preroll=%d", GetDWBE( &header[22] ) );
689 msg_Dbg( p_demux, " - duration=%d", GetDWBE( &header[26] ));
693 stream_Read( p_demux->s, header, 1 );
694 if( ( i_len = header[0] ) > 0 )
696 psz = malloc( i_len + 1 );
697 stream_Read( p_demux->s, psz, i_len );
700 msg_Dbg( p_demux, " - name=`%s'", psz );
706 stream_Read( p_demux->s, header, 1 );
707 if( ( i_len = header[0] ) > 0 )
709 psz = malloc( i_len + 1 );
710 stream_Read( p_demux->s, psz, i_len );
713 msg_Dbg( p_demux, " - mime=`%s'", psz );
719 stream_Read( p_demux->s, header, 4 );
720 if( ( i_len = GetDWBE( header ) ) > 0 )
726 msg_Dbg( p_demux, " - specific data len=%d", i_len );
727 if( stream_Peek( p_demux->s, &p_peek, 34 ) >= 34 )
729 if( !strncmp( &p_peek[4], "VIDO", 4 ) )
731 es_format_Init( &fmt, VIDEO_ES,
732 VLC_FOURCC( p_peek[8], p_peek[9], p_peek[10], p_peek[11] ) );
733 fmt.video.i_width = GetWBE( &p_peek[12] );
734 fmt.video.i_height= GetWBE( &p_peek[14] );
737 fmt.p_extra = malloc( 8 );
738 ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
739 ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
741 msg_Dbg( p_demux, " - video 0x%08x 0x%08x",
742 ((uint32_t*)fmt.p_extra)[0],
743 ((uint32_t*)fmt.p_extra)[1] );
745 if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
746 GetDWBE( &p_peek[30] ) == 0x10003001 )
748 fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
751 msg_Dbg( p_demux, " - video %4.4s %dx%d",
753 fmt.video.i_width, fmt.video.i_height );
755 tk = malloc( sizeof( real_track_t ) );
760 tk->p_es = es_out_Add( p_demux->out, &fmt );
762 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
764 else if( !strncmp( p_peek, ".ra\xfd", 4 ) )
766 int i_version = GetWBE( &p_peek[4] );
767 msg_Dbg( p_demux, " - audio version=%d", i_version );
769 es_format_Init( &fmt, AUDIO_ES, 0 );
770 if( i_version == 4 && stream_Peek( p_demux->s, &p_peek, 56 ) >= 56 )
772 fmt.audio.i_channels = GetWBE( &p_peek[54] );
773 fmt.audio.i_rate = GetWBE( &p_peek[48] );
775 if( stream_Peek( p_demux->s, &p_peek, 57 ) >= 57 )
777 int i_extra = p_peek[56] + 1 + 4;
778 if( stream_Peek( p_demux->s, &p_peek, 57 + i_extra ) >= 57 + i_extra )
780 memcpy( &fmt.i_codec, &p_peek[57 + p_peek[56] + 1], 4 );
784 else if( i_version == 5 && stream_Peek( p_demux->s, &p_peek, 70 ) >= 70 )
786 memcpy( &fmt.i_codec, &p_peek[66], 4 );
787 fmt.audio.i_channels = GetWBE( &p_peek[60] );
788 fmt.audio.i_rate = GetWBE( &p_peek[54] );
790 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
792 fmt.audio.i_channels, fmt.audio.i_rate );
794 if( !strncasecmp( (char*)&fmt.i_codec, "dnet", 4 ) )
796 fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
799 if( fmt.i_codec != 0 )
801 tk = malloc( sizeof( real_track_t ) );
806 tk->p_es = es_out_Add( p_demux->out, &fmt );
808 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
812 stream_Read( p_demux->s, NULL, i_len );
818 else if( i_id == VLC_FOURCC('D','A','T','A') )
820 stream_Read( p_demux->s, header, 8 );
822 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 10;
823 p_sys->i_data_size = i_size;
824 p_sys->i_data_packets_count = GetDWBE( header );
825 p_sys->i_data_packets = 0;
826 p_sys->i_data_offset_next = GetDWBE( &header[4] );
828 msg_Dbg( p_demux, " - packets count=%d next=%u",
829 p_sys->i_data_packets_count,
830 (uint32_t)p_sys->i_data_offset_next );
832 /* we have finished the header */
838 msg_Dbg( p_demux, "unknown chunk" );
845 stream_Read( p_demux->s, NULL, i_skip );
848 /* TODO read index if possible */