1 /*****************************************************************************
2 * real.c: Real demuxer.
3 *****************************************************************************
4 * Copyright (C) 2004 VideoLAN
5 * $Id: real.c,v 1.2 2004/01/07 15:31:31 fenrir Exp $
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;
95 if( stream_Peek( p_demux->s, &p_peek, 10 ) < 10 )
97 msg_Err( p_demux, "cannot peek" );
100 if( strncmp( p_peek, ".RMF", 4 ) )
102 msg_Warn( p_demux, "Real module discarded" );
106 /* Fill p_demux field */
107 p_demux->pf_demux = Demux;
108 p_demux->pf_control = Control;
109 p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
110 p_sys->i_data_offset = 0;
116 /* Parse the headers */
117 if( HeaderRead( p_demux ) )
120 msg_Err( p_demux, "Invalid header" );
121 for( i = 0; i < p_sys->i_track; i++ )
123 real_track_t *tk = p_sys->track[i];
127 es_out_Del( p_demux->out, tk->p_es );
131 if( p_sys->i_track > 0 )
133 free( p_sys->track );
142 /*****************************************************************************
144 *****************************************************************************/
145 static void Close( vlc_object_t *p_this )
147 demux_t *p_demux = (demux_t*)p_this;
148 demux_sys_t *p_sys = p_demux->p_sys;
151 for( i = 0; i < p_sys->i_track; i++ )
153 real_track_t *tk = p_sys->track[i];
157 block_Release( tk->p_frame );
162 if( p_sys->i_track > 0 )
164 free( p_sys->track );
171 /*****************************************************************************
173 *****************************************************************************/
174 static int Demux( demux_t *p_demux )
176 demux_sys_t *p_sys = p_demux->p_sys;
182 real_track_t *tk = NULL;
183 vlc_bool_t b_selected;
185 if( p_sys->i_data_packets >= p_sys->i_data_packets_count )
188 if( stream_Read( p_demux->s, header, 18 ) < 18 )
192 if( strncmp( header, "DATA", 4 ) )
196 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 18;
197 p_sys->i_data_size = GetDWBE( &header[4] );
198 p_sys->i_data_packets_count = GetDWBE( &header[10] );
199 p_sys->i_data_packets = 0;
200 p_sys->i_data_offset_next = GetDWBE( &header[14] );
202 msg_Dbg( p_demux, "entering new DATA packets=%d next=%u",
203 p_sys->i_data_packets_count,
204 (uint32_t)p_sys->i_data_offset_next );
207 if( stream_Read( p_demux->s, header, 12 ) < 12 )
211 i_size = GetWBE( &header[2] ) - 12;
212 i_id = GetWBE( &header[4] );
213 i_pts = 1000 * GetDWBE( &header[6] );
214 /* header[11] -> flags 0x02 -> keyframe */
215 msg_Dbg( p_demux, "packet %d size=%d id=%d pts=%u",
216 p_sys->i_data_packets, i_size, i_id, (uint32_t)(i_pts/1000) );
217 p_sys->i_data_packets++;
219 stream_Read( p_demux->s, p_sys->buffer, i_size );
221 for( i = 0; i < p_sys->i_track; i++ )
223 if( p_sys->track[i]->i_id == i_id )
225 tk = p_sys->track[i];
231 msg_Warn( p_demux, "unknown track id(0x%x)", i_id );
234 es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b_selected );
236 if( tk->fmt.i_cat == VIDEO_ES && b_selected )
238 uint8_t *p = p_sys->buffer;
240 while( p < &p_sys->buffer[i_size - 2] )
249 if( (h&0xc0) == 0x40 )
253 i_len = &p_sys->buffer[i_size] - p;
259 i_subseq = (*p++)&0x7f;
261 i_len = (p[0] << 8)|p[1]; p += 2;
262 if( (i_len&0xc000) == 0 )
265 i_len |= (p[0] << 8)|p[1]; p += 2;
273 i_offset = (p[0] << 8)|p[1]; p += 2;
274 if( (i_offset&0xc000) == 0 )
277 i_offset |= (p[0] << 8)|p[1]; p += 2;
278 i_offset &= 0x3fffffff;
286 i_copy = i_len - i_offset;
287 if( i_copy > &p_sys->buffer[i_size] - p )
289 i_copy = &p_sys->buffer[i_size] - p;
291 else if( i_copy < 0 )
296 msg_Dbg( p_demux, " - len=%d offset=%d size=%d subseq=%d seqnum=%d",
297 i_len, i_offset, i_copy, i_subseq, i_seqnum );
299 if( (h&0xc0) == 0x80 )
301 /* last fragment -> fixes */
303 i_offset = i_len - i_copy;
304 msg_Dbg( p_demux, "last fixing copy=%d offset=%d", i_copy, i_offset );
308 ( tk->p_frame->i_dts != i_pts ||
309 tk->i_frame != i_len ) )
311 msg_Dbg( p_demux, "sending size=%d", tk->p_frame->i_buffer );
313 if( p_sys->i_pcr < tk->p_frame->i_dts )
315 p_sys->i_pcr = tk->p_frame->i_dts;
317 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
319 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
325 if( (h&0xc0) != 0x80 && (h&0xc0) != 0x00 && tk->p_frame == NULL )
333 if( tk->p_frame == NULL )
335 msg_Dbg( p_demux, "new frame size=%d", i_len );
337 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
341 memset( &tk->p_frame->p_buffer[8], 0, i_len );
342 tk->p_frame->i_dts = i_pts;
343 tk->p_frame->i_pts = i_pts;
345 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
346 ((uint32_t*)tk->p_frame->p_buffer)[1] = 0; /* chunk counts */
349 if( i_offset < tk->i_frame)
351 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
353 msg_Dbg( p_demux, "copying new buffer n=%d offset=%d copy=%d", i_ck, i_offset, i_copy );
355 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
357 memcpy( &tk->p_frame->p_buffer[i_offset + 8],
363 if( (h&0xc0) != 0x80 )
371 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
373 if( (h&0xc0) == 0x80 )
377 i_offset = i_len - i_offset;
379 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
380 memcpy( &tk->p_frame->p_buffer[i_offset+ 8], p, i_copy );
383 if( p_sys->i_pcr < tk->p_frame->i_dts )
385 p_sys->i_pcr = tk->p_frame->i_dts;
386 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
388 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
396 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
397 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
401 if( (h&0xc0) != 0x00 )
406 if( ( p_frame = block_New( p_demux, i_copy + 8 + 8 ) ) == NULL )
410 p_frame->i_dts = i_pts;
411 p_frame->i_pts = i_pts;
413 ((uint32_t*)p_frame->p_buffer)[0] = i_copy;
414 ((uint32_t*)p_frame->p_buffer)[1] = 1;
415 ((uint32_t*)(p_frame->p_buffer+i_copy+8))[0] = 0;
416 memcpy( &p_frame->p_buffer[8], p, i_copy );
420 if( p_sys->i_pcr < p_frame->i_dts )
422 p_sys->i_pcr = p_frame->i_dts;
423 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
425 es_out_Send( p_demux->out, tk->p_es, p_frame );
431 if( ( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) == NULL )
435 memset( &tk->p_frame->p_buffer[8], 0, i_len );
436 tk->p_frame->i_dts = i_pts;
437 tk->p_frame->i_pts = i_pts;
439 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len; /* len */
440 ((uint32_t*)tk->p_frame->p_buffer)[1] = 1; /* chunk counts */
441 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[0] = i_offset;
442 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
449 else if( tk->fmt.i_cat == AUDIO_ES && b_selected )
451 block_t *p_block = block_New( p_demux, i_size );
453 if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
455 uint8_t *src = p_sys->buffer;
456 uint8_t *dst = p_block->p_buffer;
459 while( dst < &p_block->p_buffer[i_size- 1])
469 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
472 p_block->i_pts = i_pts;
474 if( p_sys->i_pcr < i_pts )
476 p_sys->i_pcr = i_pts;
478 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
480 es_out_Send( p_demux->out, tk->p_es, p_block );
487 /*****************************************************************************
489 *****************************************************************************/
490 static int Control( demux_t *p_demux, int i_query, va_list args )
492 demux_sys_t *p_sys = p_demux->p_sys;
499 case DEMUX_GET_POSITION:
500 pf = (double*) va_arg( args, double* );
501 i64 = stream_Size( p_demux->s );
504 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
511 case DEMUX_SET_POSITION:
512 f = (double) va_arg( args, double );
513 i64 = stream_Size( p_demux->s );
515 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
517 return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
520 pi64 = (int64_t*)va_arg( args, int64_t * );
521 if( p_sys->i_mux_rate > 0 )
523 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
529 case DEMUX_GET_LENGTH:
530 pi64 = (int64_t*)va_arg( args, int64_t * );
531 if( p_sys->i_mux_rate > 0 )
533 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
548 /*****************************************************************************
550 *****************************************************************************/
551 static int HeaderRead( demux_t *p_demux )
553 demux_sys_t *p_sys = p_demux->p_sys;
554 uint8_t header[100]; /* FIXME */
563 /* Read the header */
564 if( stream_Read( p_demux->s, header, 10 ) < 10 )
568 i_id = VLC_FOURCC( header[0], header[1], header[2], header[3] );
569 i_size = GetDWBE( &header[4] );
570 i_version = GetWBE( &header[8] );
572 msg_Dbg( p_demux, "Object %4.4s size=%d version=%d",
573 (char*)&i_id, i_size, i_version );
579 i_skip = i_size - 10;
581 if( i_id == VLC_FOURCC('.','R','M','F') )
583 if( stream_Read( p_demux->s, header, 8 ) < 8 )
587 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
588 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
592 else if( i_id == VLC_FOURCC('P','R','O','P') )
596 if( stream_Read( p_demux->s, header, 40 ) < 40 )
600 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
601 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
602 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
603 GetDWBE( &header[8] ), GetDWBE( &header[12] ) );
604 msg_Dbg( p_demux, " - packets count=%d", GetDWBE( &header[16] ) );
605 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE( &header[20] ) );
606 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE( &header[24] ) );
607 msg_Dbg( p_demux, " - index offset=%d", GetDWBE( &header[28] ) );
608 msg_Dbg( p_demux, " - data offset=%d", GetDWBE( &header[32] ) );
609 msg_Dbg( p_demux, " - num streams=%d", GetWBE( &header[36] ) );
610 i_flags = GetWBE( &header[38]);
611 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
613 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
614 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
615 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
618 else if( i_id == VLC_FOURCC('C','O','N','T') )
623 stream_Read( p_demux->s, header, 2 );
624 if( ( i_len = GetWBE( header ) ) > 0 )
626 psz = malloc( i_len + 1 );
627 stream_Read( p_demux->s, psz, i_len );
630 msg_Dbg( p_demux, " - title=`%s'", psz );
636 stream_Read( p_demux->s, header, 2 );
637 if( ( i_len = GetWBE( header ) ) > 0 )
639 psz = malloc( i_len + 1 );
640 stream_Read( p_demux->s, psz, i_len );
643 msg_Dbg( p_demux, " - author=`%s'", psz );
649 stream_Read( p_demux->s, header, 2 );
650 if( ( i_len = GetWBE( header ) ) > 0 )
652 psz = malloc( i_len + 1 );
653 stream_Read( p_demux->s, psz, i_len );
656 msg_Dbg( p_demux, " - copyright=`%s'", psz );
662 stream_Read( p_demux->s, header, 2 );
663 if( ( i_len = GetWBE( header ) ) > 0 )
665 psz = malloc( i_len + 1 );
666 stream_Read( p_demux->s, psz, i_len );
669 msg_Dbg( p_demux, " - comment=`%s'", psz );
675 else if( i_id == VLC_FOURCC('M','D','P','R') )
681 if( stream_Read( p_demux->s, header, 30 ) < 30 )
685 i_num = GetWBE( header );
686 msg_Dbg( p_demux, " - id=0x%x", i_num );
687 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d", GetDWBE( &header[2] ), GetDWBE( &header[6] ) );
688 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d", GetDWBE( &header[10] ), GetDWBE( &header[14] ));
689 msg_Dbg( p_demux, " - start time=%d", GetDWBE( &header[18] ) );
690 msg_Dbg( p_demux, " - preroll=%d", GetDWBE( &header[22] ) );
691 msg_Dbg( p_demux, " - duration=%d", GetDWBE( &header[26] ));
695 stream_Read( p_demux->s, header, 1 );
696 if( ( i_len = header[0] ) > 0 )
698 psz = malloc( i_len + 1 );
699 stream_Read( p_demux->s, psz, i_len );
702 msg_Dbg( p_demux, " - name=`%s'", psz );
708 stream_Read( p_demux->s, header, 1 );
709 if( ( i_len = header[0] ) > 0 )
711 psz = malloc( i_len + 1 );
712 stream_Read( p_demux->s, psz, i_len );
715 msg_Dbg( p_demux, " - mime=`%s'", psz );
721 stream_Read( p_demux->s, header, 4 );
722 if( ( i_len = GetDWBE( header ) ) > 0 )
728 msg_Dbg( p_demux, " - specific data len=%d", i_len );
729 if( stream_Peek( p_demux->s, &p_peek, 34 ) >= 34 )
731 if( !strncmp( &p_peek[4], "VIDO", 4 ) )
733 es_format_Init( &fmt, VIDEO_ES,
734 VLC_FOURCC( p_peek[8], p_peek[9], p_peek[10], p_peek[11] ) );
735 fmt.video.i_width = GetWBE( &p_peek[12] );
736 fmt.video.i_height= GetWBE( &p_peek[14] );
739 fmt.p_extra = malloc( 8 );
740 ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
741 ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
743 msg_Dbg( p_demux, " - video 0x%08x 0x%08x",
744 ((uint32_t*)fmt.p_extra)[0],
745 ((uint32_t*)fmt.p_extra)[1] );
747 if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
748 GetDWBE( &p_peek[30] ) == 0x10003001 )
750 fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
753 msg_Dbg( p_demux, " - video %4.4s %dx%d",
755 fmt.video.i_width, fmt.video.i_height );
757 tk = malloc( sizeof( real_track_t ) );
762 tk->p_es = es_out_Add( p_demux->out, &fmt );
764 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
766 else if( !strncmp( p_peek, ".ra\xfd", 4 ) )
768 int i_version = GetWBE( &p_peek[4] );
769 msg_Dbg( p_demux, " - audio version=%d", i_version );
771 es_format_Init( &fmt, AUDIO_ES, 0 );
772 if( i_version == 4 && stream_Peek( p_demux->s, &p_peek, 56 ) >= 56 )
774 fmt.audio.i_channels = GetWBE( &p_peek[54] );
775 fmt.audio.i_rate = GetWBE( &p_peek[48] );
777 if( stream_Peek( p_demux->s, &p_peek, 57 ) >= 57 )
779 int i_extra = p_peek[56] + 1 + 4;
780 if( stream_Peek( p_demux->s, &p_peek, 57 + i_extra ) >= 57 + i_extra )
782 memcpy( &fmt.i_codec, &p_peek[57 + p_peek[56] + 1], 4 );
786 else if( i_version == 5 && stream_Peek( p_demux->s, &p_peek, 70 ) >= 70 )
788 memcpy( &fmt.i_codec, &p_peek[66], 4 );
789 fmt.audio.i_channels = GetWBE( &p_peek[60] );
790 fmt.audio.i_rate = GetWBE( &p_peek[54] );
792 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
794 fmt.audio.i_channels, fmt.audio.i_rate );
796 if( !strncasecmp( (char*)&fmt.i_codec, "dnet", 4 ) )
798 fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
801 if( fmt.i_codec != 0 )
803 tk = malloc( sizeof( real_track_t ) );
808 tk->p_es = es_out_Add( p_demux->out, &fmt );
810 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
814 stream_Read( p_demux->s, NULL, i_len );
820 else if( i_id == VLC_FOURCC('D','A','T','A') )
822 stream_Read( p_demux->s, header, 8 );
824 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 10;
825 p_sys->i_data_size = i_size;
826 p_sys->i_data_packets_count = GetDWBE( header );
827 p_sys->i_data_packets = 0;
828 p_sys->i_data_offset_next = GetDWBE( &header[4] );
830 msg_Dbg( p_demux, " - packets count=%d next=%u",
831 p_sys->i_data_packets_count,
832 (uint32_t)p_sys->i_data_offset_next );
834 /* we have finished the header */
840 msg_Dbg( p_demux, "unknown chunk" );
847 stream_Read( p_demux->s, NULL, i_skip );
850 /* TODO read index if possible */