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_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 )
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 )
452 if( p_sys->i_pcr < i_pts )
454 p_sys->i_pcr = i_pts;
455 es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_pcr );
458 if( tk->fmt.i_codec == VLC_FOURCC( 'm', 'p', '4', 'a' ) )
460 int i_sub = (p_sys->buffer[1] >> 4)&0x0f;
461 uint8_t *p_sub = &p_sys->buffer[2+2*i_sub];
464 for( i = 0; i < i_sub; i++ )
466 int i_sub_size = GetWBE( &p_sys->buffer[2+i*2]);
467 block_t *p_block = block_New( p_demux, i_sub_size );
470 memcpy( p_block->p_buffer, p_sub, i_sub_size );
474 p_block->i_pts = ( i == 0 ? i_pts : 0 );
476 es_out_Send( p_demux->out, tk->p_es, p_block );
482 block_t *p_block = block_New( p_demux, i_size );
484 if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
486 uint8_t *src = p_sys->buffer;
487 uint8_t *dst = p_block->p_buffer;
490 while( dst < &p_block->p_buffer[i_size- 1])
500 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
503 p_block->i_pts = i_pts;
505 es_out_Send( p_demux->out, tk->p_es, p_block );
513 /*****************************************************************************
515 *****************************************************************************/
516 static int Control( demux_t *p_demux, int i_query, va_list args )
519 demux_sys_t *p_sys = p_demux->p_sys;
525 case DEMUX_GET_POSITION:
526 pf = (double*) va_arg( args, double* );
527 i64 = stream_Size( p_demux->s );
530 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
537 case DEMUX_SET_POSITION:
538 f = (double) va_arg( args, double );
539 i64 = stream_Size( p_demux->s );
541 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
543 return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
546 pi64 = (int64_t*)va_arg( args, int64_t * );
547 if( p_sys->i_mux_rate > 0 )
549 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
555 case DEMUX_GET_LENGTH:
556 pi64 = (int64_t*)va_arg( args, int64_t * );
557 if( p_sys->i_mux_rate > 0 )
559 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
574 /*****************************************************************************
576 *****************************************************************************/
577 static int HeaderRead( demux_t *p_demux )
579 demux_sys_t *p_sys = p_demux->p_sys;
580 uint8_t header[100]; /* FIXME */
589 /* Read the header */
590 if( stream_Read( p_demux->s, header, 10 ) < 10 )
594 i_id = VLC_FOURCC( header[0], header[1], header[2], header[3] );
595 i_size = GetDWBE( &header[4] );
596 i_version = GetWBE( &header[8] );
598 msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
599 (char*)&i_id, i_size, i_version );
605 i_skip = i_size - 10;
607 if( i_id == VLC_FOURCC('.','R','M','F') )
609 if( stream_Read( p_demux->s, header, 8 ) < 8 )
613 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
614 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
618 else if( i_id == VLC_FOURCC('P','R','O','P') )
622 if( stream_Read( p_demux->s, header, 40 ) < 40 )
626 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
627 GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
628 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
629 GetDWBE( &header[8] ), GetDWBE( &header[12] ) );
630 msg_Dbg( p_demux, " - packets count=%d", GetDWBE( &header[16] ) );
631 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE( &header[20] ) );
632 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE( &header[24] ) );
633 msg_Dbg( p_demux, " - index offset=%d", GetDWBE( &header[28] ) );
634 msg_Dbg( p_demux, " - data offset=%d", GetDWBE( &header[32] ) );
635 msg_Dbg( p_demux, " - num streams=%d", GetWBE( &header[36] ) );
636 i_flags = GetWBE( &header[38]);
637 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
639 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
640 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
641 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
644 else if( i_id == VLC_FOURCC('C','O','N','T') )
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, " - title=`%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, " - author=`%s'", psz );
675 stream_Read( p_demux->s, header, 2 );
676 if( ( i_len = GetWBE( header ) ) > 0 )
678 psz = malloc( i_len + 1 );
679 stream_Read( p_demux->s, psz, i_len );
682 msg_Dbg( p_demux, " - copyright=`%s'", psz );
688 stream_Read( p_demux->s, header, 2 );
689 if( ( i_len = GetWBE( header ) ) > 0 )
691 psz = malloc( i_len + 1 );
692 stream_Read( p_demux->s, psz, i_len );
695 msg_Dbg( p_demux, " - comment=`%s'", psz );
701 else if( i_id == VLC_FOURCC('M','D','P','R') )
707 if( stream_Read( p_demux->s, header, 30 ) < 30 )
711 i_num = GetWBE( header );
712 msg_Dbg( p_demux, " - id=0x%x", i_num );
713 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d", GetDWBE( &header[2] ), GetDWBE( &header[6] ) );
714 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d", GetDWBE( &header[10] ), GetDWBE( &header[14] ));
715 msg_Dbg( p_demux, " - start time=%d", GetDWBE( &header[18] ) );
716 msg_Dbg( p_demux, " - preroll=%d", GetDWBE( &header[22] ) );
717 msg_Dbg( p_demux, " - duration=%d", GetDWBE( &header[26] ));
721 stream_Read( p_demux->s, header, 1 );
722 if( ( i_len = header[0] ) > 0 )
724 psz = malloc( i_len + 1 );
725 stream_Read( p_demux->s, psz, i_len );
728 msg_Dbg( p_demux, " - name=`%s'", psz );
734 stream_Read( p_demux->s, header, 1 );
735 if( ( i_len = header[0] ) > 0 )
737 psz = malloc( i_len + 1 );
738 stream_Read( p_demux->s, psz, i_len );
741 msg_Dbg( p_demux, " - mime=`%s'", psz );
747 stream_Read( p_demux->s, header, 4 );
748 if( ( i_len = GetDWBE( header ) ) > 0 )
754 msg_Dbg( p_demux, " - specific data len=%d", i_len );
755 if( stream_Peek( p_demux->s, &p_peek, 34 ) >= 34 )
757 if( !strncmp( &p_peek[4], "VIDO", 4 ) )
759 es_format_Init( &fmt, VIDEO_ES,
760 VLC_FOURCC( p_peek[8], p_peek[9], p_peek[10], p_peek[11] ) );
761 fmt.video.i_width = GetWBE( &p_peek[12] );
762 fmt.video.i_height= GetWBE( &p_peek[14] );
765 fmt.p_extra = malloc( 8 );
766 ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
767 ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
769 msg_Dbg( p_demux, " - video 0x%08x 0x%08x",
770 ((uint32_t*)fmt.p_extra)[0],
771 ((uint32_t*)fmt.p_extra)[1] );
773 if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
774 GetDWBE( &p_peek[30] ) == 0x10003001 )
776 fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
779 msg_Dbg( p_demux, " - video %4.4s %dx%d",
781 fmt.video.i_width, fmt.video.i_height );
783 tk = malloc( sizeof( real_track_t ) );
788 tk->p_es = es_out_Add( p_demux->out, &fmt );
790 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
792 else if( !strncmp( p_peek, ".ra\xfd", 4 ) )
794 int i_version = GetWBE( &p_peek[4] );
795 uint8_t *p_extra = NULL;
796 msg_Dbg( p_demux, " - audio version=%d", i_version );
798 es_format_Init( &fmt, AUDIO_ES, 0 );
799 if( i_version == 4 && stream_Peek( p_demux->s, &p_peek, 56 ) >= 56 )
801 fmt.audio.i_channels = GetWBE( &p_peek[54] );
802 fmt.audio.i_rate = GetWBE( &p_peek[48] );
804 if( stream_Peek( p_demux->s, &p_peek, 57 ) >= 57 )
806 int i_extra = p_peek[56] + 1 + 4;
807 if( stream_Peek( p_demux->s, &p_peek, 57 + i_extra ) >= 57 + i_extra )
809 memcpy( &fmt.i_codec, &p_peek[57 + p_peek[56] + 1], 4 );
811 p_extra = &p_peek[57 + p_peek[56] + 1+ 4 + 3];
814 else if( i_version == 5 && stream_Peek( p_demux->s, &p_peek, 70 ) >= 70 )
816 memcpy( &fmt.i_codec, &p_peek[66], 4 );
817 fmt.audio.i_channels = GetWBE( &p_peek[60] );
818 fmt.audio.i_rate = GetWBE( &p_peek[54] );
820 p_extra = &p_peek[66+4+3+1];
822 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
824 fmt.audio.i_channels, fmt.audio.i_rate );
826 if( fmt.i_codec == VLC_FOURCC( 'd', 'n', 'e', 't' ) )
828 fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
830 else if( fmt.i_codec == VLC_FOURCC( 'r', 'a', 'a', 'c' ) ||
831 fmt.i_codec == VLC_FOURCC( 'r', 'a', 'c', 'p' ) )
833 int i_peek = p_extra - p_peek;
834 if( stream_Peek( p_demux->s, &p_peek, i_peek + 4 ) >= i_peek + 4 )
836 int i_extra = GetDWBE( &p_peek[i_peek] );
838 if( i_extra > 1 && i_peek + 4 + i_extra <= i_len &&
839 stream_Peek( p_demux->s, &p_peek, i_peek + 4 + i_extra ) >= i_peek + 4 + i_extra )
841 fmt.i_extra = i_extra - 1;
842 fmt.p_extra = malloc ( i_extra - 1 );
843 memcpy( fmt.p_extra, &p_peek[i_peek+4+1], i_extra - 1 );
845 msg_Dbg( p_demux, " - extra data=%d", i_extra );
848 for( i = 0; i < fmt.i_extra; i++ )
850 msg_Dbg( p_demux, " data[%d] = 0x%x", i,((uint8_t*)fmt.p_extra)[i] );
855 fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
858 if( fmt.i_codec != 0 )
860 tk = malloc( sizeof( real_track_t ) );
865 tk->p_es = es_out_Add( p_demux->out, &fmt );
867 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
871 stream_Read( p_demux->s, NULL, i_len );
877 else if( i_id == VLC_FOURCC('D','A','T','A') )
879 stream_Read( p_demux->s, header, 8 );
881 p_sys->i_data_offset = stream_Tell( p_demux->s ) - 10;
882 p_sys->i_data_size = i_size;
883 p_sys->i_data_packets_count = GetDWBE( header );
884 p_sys->i_data_packets = 0;
885 p_sys->i_data_offset_next = GetDWBE( &header[4] );
887 msg_Dbg( p_demux, " - packets count=%d next=%u",
888 p_sys->i_data_packets_count,
889 (uint32_t)p_sys->i_data_offset_next );
891 /* we have finished the header */
897 msg_Dbg( p_demux, "unknown chunk" );
904 stream_Read( p_demux->s, NULL, i_skip );
907 /* TODO read index if possible */