?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Fix MODULES_LIST for smbc renaming
[vlc.git] / modules / demux / real.c
bloba9463595e12c708cc6c3da66c1002e3dd9e38512
1 /*****************************************************************************
2 * real.c: Real demuxer.
3 *****************************************************************************
4 * Copyright (C) 2004, 2006-2007 the VideoLAN team
5 * $Id$
6 *
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 *
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.
13 *
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.
18 *
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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
23
24 /**
25 * Status of this demuxer:
26 * Real Media format
27 * -----------------
28 *
29 * version v3 w/ 14_4/lpcJ is ok.
30 * version v4/5: - atrac3 is ok.
31 * - cook is ok.
32 * - raac, racp are ok.
33 * - dnet is twisted "The byte order of the data is reversed
34 * from standard AC3" but ok
35 * - 28_8 is ok.
36 * - sipr is ok.
37 * - ralf is unsupported, but hardly any sample exist.
38 * - mp3 is unsupported, one sample exists...
39 *
40 * Real Audio Only
41 * ---------------
42 * v3 and v4/5 headers are parsed.
43 * Doesn't work yet...
44 */
45
46 /*****************************************************************************
47 * Preamble
48 *****************************************************************************/
49
50 #ifdef HAVE_CONFIG_H
51 # include "config.h"
52 #endif
53
54 #define VLC_MODULE_LICENSE VLC_LICENSE_GPL_2_PLUS
55 #include <vlc_common.h>
56 #include <vlc_plugin.h>
57
58 #include <vlc_demux.h>
59 #include <vlc_charset.h>
60 #include <vlc_meta.h>
61
62 #include <assert.h>
63
64 /*****************************************************************************
65 * Module descriptor
66 *****************************************************************************/
67 static int Open ( vlc_object_t * );
68 static void Close ( vlc_object_t * );
69
70 vlc_module_begin ()
71 set_description( N_("Real demuxer" ) )
72 set_capability( "demux", 0 )
73 set_category( CAT_INPUT )
74 set_subcategory( SUBCAT_INPUT_DEMUX )
75 set_callbacks( Open, Close )
76 add_shortcut( "real", "rm" )
77 vlc_module_end ()
78
79 /*****************************************************************************
80 * Local prototypes
81 *****************************************************************************/
82
83 typedef struct
84 {
85 int i_id;
86 es_format_t fmt;
87
88 es_out_id_t *p_es;
89
90 unsigned i_frame_size;
91
92 int i_frame_num;
93 unsigned i_frame_pos;
94 int i_frame_slice;
95 int i_frame_slice_count;
96 block_t *p_frame;
97
98 int i_subpacket_h;
99 int i_subpacket_size;
100 int i_coded_frame_size;
101
102 int i_subpacket;
103 int i_subpackets;
104 block_t **p_subpackets;
105 vlc_tick_t *p_subpackets_timecode;
106 int i_out_subpacket;
107
108 block_t *p_sipr_packet;
109 int i_sipr_subpacket_count;
110 vlc_tick_t i_last_dts;
111 } real_track_t;
112
113 typedef struct
114 {
115 uint32_t i_file_offset;
116 uint32_t i_time_offset;
117 uint32_t i_frame_index;
118 } real_index_t;
119
120 typedef struct
121 {
122 int64_t i_data_offset;
123 int64_t i_data_size;
124 uint32_t i_data_packets_count;
125 uint32_t i_data_packets;
126 int64_t i_data_offset_next;
127
128 bool b_real_audio;
129
130 int64_t i_our_duration;
131
132 char* psz_title;
133 char* psz_artist;
134 char* psz_copyright;
135 char* psz_description;
136
137 int i_track;
138 real_track_t **track;
139
140 size_t i_buffer;
141 uint8_t buffer[65536];
142
143 int64_t i_pcr;
144
145 int64_t i_index_offset;
146 bool b_seek;
147 real_index_t *p_index;
148 } demux_sys_t;
149
150 static const unsigned char i_subpacket_size_sipr[4] = { 29, 19, 37, 20 };
151
152 static int Demux( demux_t * );
153 static int Control( demux_t *, int i_query, va_list args );
154
155
156 static void DemuxVideo( demux_t *, real_track_t *tk, vlc_tick_t i_dts, unsigned i_flags );
157 static void DemuxAudio( demux_t *, real_track_t *tk, vlc_tick_t i_pts, unsigned i_flags );
158
159 static int ControlSeekByte( demux_t *, int64_t i_bytes );
160 static int ControlSeekTime( demux_t *, vlc_tick_t i_time );
161
162 static int HeaderRead( demux_t *p_demux );
163 static int CodecParse( demux_t *p_demux, int i_len, int i_num );
164
165 static void RVoid( const uint8_t **pp_data, int *pi_data, int i_size );
166 static int RLength( const uint8_t **pp_data, int *pi_data );
167 static uint8_t R8( const uint8_t **pp_data, int *pi_data );
168 static uint16_t R16( const uint8_t **pp_data, int *pi_data );
169 static uint32_t R32( const uint8_t **pp_data, int *pi_data );
170 static void SiprPacketReorder(uint8_t *buf, int sub_packet_h, int framesize);
171
172 /*****************************************************************************
173 * Open
174 *****************************************************************************/
175 static int Open( vlc_object_t *p_this )
176 {
177 demux_t *p_demux = (demux_t*)p_this;
178 demux_sys_t *p_sys;
179
180 const uint8_t *p_peek;
181 bool b_real_audio = false;
182
183 if( vlc_stream_Peek( p_demux->s, &p_peek, 10 ) < 10 )
184 return VLC_EGENERIC;
185
186 /* Real Audio */
187 if( !memcmp( p_peek, ".ra", 3 ) )
188 {
189 msg_Err( p_demux, ".ra files unsupported" );
190 b_real_audio = true;
191 }
192 /* Real Media Format */
193 else if( memcmp( p_peek, ".RMF", 4 ) )
194 {
195 return VLC_EGENERIC;
196 }
197
198 /* Fill p_demux field */
199 p_demux->pf_demux = Demux;
200 p_demux->pf_control = Control;
201
202 p_demux->p_sys = p_sys = calloc( 1, sizeof( *p_sys ) );
203 if( !p_sys )
204 return VLC_ENOMEM;
205
206 p_sys->i_data_offset = 0;
207 p_sys->i_track = 0;
208 p_sys->track = NULL;
209 p_sys->i_pcr = VLC_TICK_INVALID;
210
211 p_sys->b_seek = false;
212 p_sys->b_real_audio = b_real_audio;
213
214 /* Parse the headers */
215 /* Real Audio files */
216 if( b_real_audio )
217 {
218 CodecParse( p_demux, 32, 0 ); /* At least 32 */
219 return VLC_EGENERIC; /* We don't know how to read
220 correctly the data yet */
221 }
222 /* RMF files */
223 else if( HeaderRead( p_demux ) )
224 {
225 msg_Err( p_demux, "invalid header" );
226 Close( p_this );
227 return VLC_EGENERIC;
228 }
229
230 return VLC_SUCCESS;
231 }
232
233 /*****************************************************************************
234 * Close
235 *****************************************************************************/
236 static void Close( vlc_object_t *p_this )
237 {
238 demux_t *p_demux = (demux_t*)p_this;
239 demux_sys_t *p_sys = p_demux->p_sys;
240
241 for( int i = 0; i < p_sys->i_track; i++ )
242 {
243 real_track_t *tk = p_sys->track[i];
244
245 es_format_Clean( &tk->fmt );
246
247 if( tk->p_frame )
248 block_Release( tk->p_frame );
249
250 for( int j = 0; j < tk->i_subpackets; j++ )
251 {
252 if( tk->p_subpackets[ j ] )
253 block_Release( tk->p_subpackets[ j ] );
254 }
255 free( tk->p_subpackets );
256 free( tk->p_subpackets_timecode );
257 if( tk->p_sipr_packet )
258 block_Release( tk->p_sipr_packet );
259 free( tk );
260 }
261 if( p_sys->i_track > 0 )
262 free( p_sys->track );
263
264 free( p_sys->psz_title );
265 free( p_sys->psz_artist );
266 free( p_sys->psz_copyright );
267 free( p_sys->psz_description );
268 free( p_sys->p_index );
269
270 free( p_sys );
271 }
272
273
274 /*****************************************************************************
275 * Demux:
276 *****************************************************************************/
277 static int Demux( demux_t *p_demux )
278 {
279 demux_sys_t *p_sys = p_demux->p_sys;
280 uint8_t header[18];
281
282 if( p_sys->i_data_packets >= p_sys->i_data_packets_count &&
283 p_sys->i_data_packets_count )
284 {
285 if( vlc_stream_Read( p_demux->s, header, 18 ) < 18 )
286 return VLC_DEMUXER_EOF;
287
288 if( memcmp( header, "DATA", 4 ) )
289 return VLC_DEMUXER_EOF;
290
291 p_sys->i_data_offset = vlc_stream_Tell( p_demux->s ) - 18;
292 p_sys->i_data_size = GetDWBE( &header[4] );
293 p_sys->i_data_packets_count = GetDWBE( &header[10] );
294 p_sys->i_data_packets = 0;
295 p_sys->i_data_offset_next = GetDWBE( &header[14] );
296
297 msg_Dbg( p_demux, "entering new DATA packets=%d next=%u",
298 p_sys->i_data_packets_count,
299 (unsigned int)p_sys->i_data_offset_next );
300 }
301
302 /* Read Packet Header */
303 if( vlc_stream_Read( p_demux->s, header, 12 ) < 12 )
304 return VLC_DEMUXER_EOF;
305 //const int i_version = GetWBE( &header[0] );
306 const size_t i_size = GetWBE( &header[2] ) - 12;
307 const int i_id = GetWBE( &header[4] );
308 const vlc_tick_t i_pts = VLC_TICK_0 + VLC_TICK_FROM_MS(GetDWBE( &header[6] ));
309 const int i_flags= header[11]; /* flags 0x02 -> keyframe */
310
311 p_sys->i_data_packets++;
312 if( i_size > sizeof(p_sys->buffer) )
313 {
314 msg_Err( p_demux, "Got a NUKK size to read. (Invalid format?)" );
315 return VLC_DEMUXER_SUCCESS;
316 }
317
318 p_sys->i_buffer = vlc_stream_Read( p_demux->s, p_sys->buffer, i_size );
319 if( p_sys->i_buffer < i_size )
320 return VLC_DEMUXER_EOF;
321
322 real_track_t *tk = NULL;
323 for( int i = 0; i < p_sys->i_track; i++ )
324 {
325 if( p_sys->track[i]->i_id == i_id )
326 tk = p_sys->track[i];
327 }
328
329 if( !tk )
330 {
331 msg_Warn( p_demux, "unknown track id(0x%x)", i_id );
332 return VLC_DEMUXER_SUCCESS;
333 }
334
335 if( tk->fmt.i_cat == VIDEO_ES )
336 {
337 DemuxVideo( p_demux, tk, i_pts, i_flags );
338 }
339 else
340 {
341 assert( tk->fmt.i_cat == AUDIO_ES );
342 DemuxAudio( p_demux, tk, i_pts, i_flags );
343 }
344
345 /* Update PCR */
346 vlc_tick_t i_pcr = VLC_TICK_INVALID;
347 for( int i = 0; i < p_sys->i_track; i++ )
348 {
349 tk = p_sys->track[i];
350
351 if( i_pcr == VLC_TICK_INVALID || ( tk->i_last_dts != VLC_TICK_INVALID && tk->i_last_dts < i_pcr ) )
352 i_pcr = tk->i_last_dts;
353 }
354 if( i_pcr != VLC_TICK_INVALID && i_pcr != p_sys->i_pcr )
355 {
356 p_sys->i_pcr = i_pcr;
357 es_out_SetPCR( p_demux->out, p_sys->i_pcr );
358 }
359 return VLC_DEMUXER_SUCCESS;
360 }
361
362 /*****************************************************************************
363 * Control:
364 *****************************************************************************/
365 static int Control( demux_t *p_demux, int i_query, va_list args )
366 {
367 demux_sys_t *p_sys = p_demux->p_sys;
368 double f, *pf;
369 int64_t i64;
370 int64_t *pi64;
371
372 switch( i_query )
373 {
374 case DEMUX_CAN_SEEK:
375 {
376 bool *b = va_arg( args, bool * );
377
378 if( stream_Size( p_demux->s ) == 0 )
379 *b = true; /* FIXME: kludge for Real-dialectal RTSP */
380 else
381 if( p_sys->p_index == NULL )
382 *b = false;
383 else
384 if( vlc_stream_Control( p_demux->s, STREAM_CAN_SEEK, &b ) )
385 *b = false;
386
387 return VLC_SUCCESS;
388 }
389
390 case DEMUX_GET_POSITION:
391 pf = va_arg( args, double * );
392
393 /* read stream size maybe failed in rtsp streaming,
394 so use duration to determin the position at first */
395 if( p_sys->i_our_duration > 0 )
396 {
397 if( p_sys->i_pcr != VLC_TICK_INVALID )
398 *pf = (double)p_sys->i_pcr / 1000.0 / p_sys->i_our_duration;
399 else
400 *pf = 0.0;
401 return VLC_SUCCESS;
402 }
403
404 i64 = stream_Size( p_demux->s );
405 if( i64 > 0 )
406 *pf = (double)1.0*vlc_stream_Tell( p_demux->s ) / (double)i64;
407 else
408 *pf = 0.0;
409 return VLC_SUCCESS;
410
411 case DEMUX_GET_TIME:
412 pi64 = va_arg( args, int64_t * );
413
414 if( p_sys->i_our_duration > 0 )
415 {
416 *pi64 = p_sys->i_pcr != VLC_TICK_INVALID ? p_sys->i_pcr : 0;
417 return VLC_SUCCESS;
418 }
419
420 /* same as GET_POSTION */
421 i64 = stream_Size( p_demux->s );
422 if( p_sys->i_our_duration > 0 && i64 > 0 )
423 {
424 *pi64 = (int64_t)( 1000.0 * p_sys->i_our_duration * vlc_stream_Tell( p_demux->s ) / i64 );
425 return VLC_SUCCESS;
426 }
427
428 *pi64 = 0;
429 return VLC_EGENERIC;
430
431 case DEMUX_SET_POSITION:
432 f = va_arg( args, double );
433 i64 = (int64_t) ( stream_Size( p_demux->s ) * f );
434
435 if( !p_sys->p_index && i64 != 0 )
436 {
437 /* TODO seek */
438 msg_Err( p_demux,"Seek No Index Real File failed!" );
439 return VLC_EGENERIC; // no index!
440 }
441 else if( i64 == 0 )
442 {
443 /* it is a rtsp stream , it is specials in access/rtsp/... */
444 msg_Dbg(p_demux, "Seek in real rtsp stream!");
445 p_sys->i_pcr = VLC_TICK_0 + INT64_C(1000) * ( p_sys->i_our_duration * f );
446 p_sys->b_seek = true;
447 return vlc_stream_Seek( p_demux->s, p_sys->i_pcr - VLC_TICK_0 );
448 }
449 return ControlSeekByte( p_demux, i64 );
450
451 case DEMUX_SET_TIME:
452 if( !p_sys->p_index )
453 return VLC_EGENERIC;
454
455 i64 = va_arg( args, int64_t );
456 return ControlSeekTime( p_demux, i64 );
457
458 case DEMUX_GET_LENGTH:
459 pi64 = va_arg( args, int64_t * );
460
461 if( p_sys->i_our_duration <= 0 )
462 {
463 *pi64 = 0;
464 return VLC_EGENERIC;
465 }
466
467 /* our stored duration is in ms, so... */
468 *pi64 = INT64_C(1000) * p_sys->i_our_duration;
469 return VLC_SUCCESS;
470
471 case DEMUX_GET_META:
472 {
473 vlc_meta_t *p_meta = va_arg( args, vlc_meta_t * );
474
475 /* the core will crash if we provide NULL strings, so check
476 * every string first */
477 if( p_sys->psz_title )
478 vlc_meta_SetTitle( p_meta, p_sys->psz_title );
479 if( p_sys->psz_artist )
480 vlc_meta_SetArtist( p_meta, p_sys->psz_artist );
481 if( p_sys->psz_copyright )
482 vlc_meta_SetCopyright( p_meta, p_sys->psz_copyright );
483 if( p_sys->psz_description )
484 vlc_meta_SetDescription( p_meta, p_sys->psz_description );
485 return VLC_SUCCESS;
486 }
487
488 case DEMUX_CAN_PAUSE:
489 case DEMUX_SET_PAUSE_STATE:
490 case DEMUX_CAN_CONTROL_PACE:
491 case DEMUX_GET_PTS_DELAY:
492 return demux_vaControlHelper( p_demux->s, p_sys->i_data_offset,
493 p_sys->i_data_size, 0, 1, i_query, args );
494
495 case DEMUX_GET_FPS:
496 default:
497 return VLC_EGENERIC;
498 }
499 return VLC_EGENERIC;
500 }
501
502 /*****************************************************************************
503 * Helpers: demux
504 *****************************************************************************/
505 static void CheckPcr( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_dts )
506 {
507 demux_sys_t *p_sys = p_demux->p_sys;
508
509 if( i_dts != VLC_TICK_INVALID )
510 tk->i_last_dts = i_dts;
511
512 if( p_sys->i_pcr != VLC_TICK_INVALID || i_dts == VLC_TICK_INVALID )
513 return;
514
515 p_sys->i_pcr = i_dts;
516 es_out_SetPCR( p_demux->out, p_sys->i_pcr );
517 }
518
519 static void DemuxVideo( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_dts, unsigned i_flags )
520 {
521 demux_sys_t *p_sys = p_demux->p_sys;
522
523 const uint8_t *p_data = p_sys->buffer;
524 int i_data = p_sys->i_buffer;
525
526 while( i_data > 1 )
527 {
528 uint8_t i_hdr = R8( &p_data, &i_data );
529 uint8_t i_type = i_hdr >> 6;
530
531 uint8_t i_seq;
532 int i_len;
533 int i_pos;
534 int i_frame_num;
535
536 if( i_type == 1 )
537 {
538 R8( &p_data, &i_data );
539 i_len = i_data;
540 i_pos = 0;
541 i_frame_num = -1;
542 i_seq = 1;
543 i_hdr &= ~0x3f;
544 }
545 else if( i_type == 3 )
546 {
547 i_len = RLength( &p_data, &i_data );
548 i_pos = RLength( &p_data, &i_data );
549 i_frame_num = R8( &p_data, &i_data );
550 i_seq = 1;
551 i_hdr &= ~0x3f;
552 }
553 else
554 {
555 assert( i_type == 0 || i_type == 2 );
556 i_seq = R8( &p_data, &i_data );
557 i_len = RLength( &p_data, &i_data );
558
559 i_pos = RLength( &p_data, &i_data );
560 i_frame_num = R8( &p_data, &i_data );
561 }
562
563 if( (i_seq & 0x7f) == 1 || tk->i_frame_num != i_frame_num )
564 {
565 tk->i_frame_slice = 0;
566 tk->i_frame_slice_count = 2 * (i_hdr & 0x3f) + 1;
567 tk->i_frame_pos = 2*4 * tk->i_frame_slice_count + 1;
568 tk->i_frame_size = i_len + 2*4 * tk->i_frame_slice_count + 1;
569 tk->i_frame_num = i_frame_num;
570
571 if( tk->p_frame )
572 block_Release( tk->p_frame );
573
574 tk->p_frame = block_Alloc( tk->i_frame_size );
575 if( !tk->p_frame )
576 {
577 tk->i_frame_size = 0;
578 return;
579 }
580
581 tk->p_frame->i_dts = i_dts;
582 tk->p_frame->i_pts = VLC_TICK_INVALID;
583 if( i_flags & 0x02 )
584 tk->p_frame->i_flags |= BLOCK_FLAG_TYPE_I;
585
586 i_dts = VLC_TICK_INVALID;
587 }
588
589 int i_frame_data;
590 if( i_type == 3 )
591 {
592 i_frame_data = i_len;
593 }
594 else
595 {
596 i_frame_data = i_data;
597 if( i_type == 2 && i_frame_data > i_pos )
598 i_frame_data = i_pos;
599 }
600 if( i_frame_data > i_data )
601 break;
602
603 /* */
604 tk->i_frame_slice++;
605 if( tk->i_frame_slice > tk->i_frame_slice_count || !tk->p_frame )
606 break;
607
608 /* */
609 SetDWLE( &tk->p_frame->p_buffer[2*4*(tk->i_frame_slice-1) + 1 + 0], 1 );
610 SetDWLE( &tk->p_frame->p_buffer[2*4*(tk->i_frame_slice-1) + 1 + 4], tk->i_frame_pos - (2*4 * tk->i_frame_slice_count + 1) );
611
612 if( tk->i_frame_pos + i_frame_data > tk->i_frame_size )
613 break;
614
615 memcpy( &tk->p_frame->p_buffer[tk->i_frame_pos], p_data, i_frame_data );
616 RVoid( &p_data, &i_data, i_frame_data );
617 tk->i_frame_pos += i_frame_data;
618
619 if( i_type != 0 || tk->i_frame_pos >= tk->i_frame_size )
620 {
621 /* Fix the buffer once the real number of slice is known */
622 tk->p_frame->p_buffer[0] = tk->i_frame_slice - 1;
623 tk->p_frame->i_buffer = tk->i_frame_pos - 2*4*( tk->i_frame_slice_count - tk->i_frame_slice );
624
625 memmove( &tk->p_frame->p_buffer[1+2*4*tk->i_frame_slice ],
626 &tk->p_frame->p_buffer[1+2*4*tk->i_frame_slice_count],
627 tk->i_frame_pos - (2*4*tk->i_frame_slice_count + 1) );
628
629 /* Send it */
630 CheckPcr( p_demux, tk, tk->p_frame->i_dts );
631 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
632
633 tk->i_frame_size = 0;
634 tk->p_frame = NULL;
635 }
636 }
637 }
638
639 static void DemuxAudioMethod1( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_pts, unsigned int i_flags )
640 {
641 demux_sys_t *p_sys = p_demux->p_sys;
642 uint8_t *p_buf = p_sys->buffer;
643
644 /* Sanity check */
645 if( (i_flags & 2) || p_sys->b_seek )
646 {
647 tk->i_subpacket = 0;
648 tk->i_out_subpacket = 0;
649 p_sys->b_seek = false;
650 }
651
652 if( tk->fmt.i_codec == VLC_CODEC_COOK ||
653 tk->fmt.i_codec == VLC_CODEC_ATRAC3 )
654 {
655 const int i_num = tk->i_frame_size / tk->i_subpacket_size;
656 const int y = tk->i_subpacket / ( tk->i_frame_size / tk->i_subpacket_size );
657
658 for( int i = 0; i < i_num; i++ )
659 {
660 int i_index = tk->i_subpacket_h * i +
661 ((tk->i_subpacket_h + 1) / 2) * (y&1) + (y>>1);
662 if( i_index >= tk->i_subpackets )
663 return;
664
665 block_t *p_block = block_Alloc( tk->i_subpacket_size );
666 if( !p_block )
667 return;
668 if( &p_buf[tk->i_subpacket_size] > &p_sys->buffer[p_sys->i_buffer] )
669 return;
670
671 memcpy( p_block->p_buffer, p_buf, tk->i_subpacket_size );
672 p_block->i_dts =
673 p_block->i_pts = VLC_TICK_INVALID;
674
675 p_buf += tk->i_subpacket_size;
676
677 if( tk->p_subpackets[i_index] != NULL )
678 {
679 msg_Dbg(p_demux, "p_subpackets[ %d ] not null!", i_index );
680 block_Release( tk->p_subpackets[i_index] );
681 }
682
683 tk->p_subpackets[i_index] = p_block;
684 if( tk->i_subpacket == 0 )
685 tk->p_subpackets_timecode[0] = i_pts;
686 tk->i_subpacket++;
687 }
688 }
689 else
690 {
691 const int y = tk->i_subpacket / (tk->i_subpacket_h / 2);
692 assert( tk->fmt.i_codec == VLC_CODEC_RA_288 );
693
694 for( int i = 0; i < tk->i_subpacket_h / 2; i++ )
695 {
696 int i_index = (i * 2 * tk->i_frame_size / tk->i_coded_frame_size) + y;
697 if( i_index >= tk->i_subpackets )
698 return;
699
700 block_t *p_block = block_Alloc( tk->i_coded_frame_size);
701 if( !p_block )
702 return;
703 if( &p_buf[tk->i_coded_frame_size] > &p_sys->buffer[p_sys->i_buffer] )
704 return;
705
706 memcpy( p_block->p_buffer, p_buf, tk->i_coded_frame_size );
707 p_block->i_dts =
708 p_block->i_pts = i_index == 0 ? i_pts : VLC_TICK_INVALID;
709
710 p_buf += tk->i_coded_frame_size;
711
712 if( tk->p_subpackets[i_index] != NULL )
713 {
714 msg_Dbg(p_demux, "p_subpackets[ %d ] not null!", i_index );
715 block_Release( tk->p_subpackets[i_index] );
716 }
717
718 tk->p_subpackets[i_index] = p_block;
719 tk->i_subpacket++;
720 }
721 }
722
723 while( tk->i_out_subpacket != tk->i_subpackets &&
724 tk->p_subpackets[tk->i_out_subpacket] )
725 {
726 block_t *p_block = tk->p_subpackets[tk->i_out_subpacket];
727 tk->p_subpackets[tk->i_out_subpacket] = NULL;
728
729 if( tk->p_subpackets_timecode[tk->i_out_subpacket] )
730 {
731 p_block->i_dts =
732 p_block->i_pts = tk->p_subpackets_timecode[tk->i_out_subpacket];
733
734 tk->p_subpackets_timecode[tk->i_out_subpacket] = 0;
735 }
736 tk->i_out_subpacket++;
737
738 CheckPcr( p_demux, tk, p_block->i_pts );
739 es_out_Send( p_demux->out, tk->p_es, p_block );
740 }
741
742 if( tk->i_subpacket == tk->i_subpackets &&
743 tk->i_out_subpacket != tk->i_subpackets )
744 {
745 msg_Warn( p_demux, "i_subpacket != i_out_subpacket, "
746 "this shouldn't happen" );
747 }
748
749 if( tk->i_subpacket == tk->i_subpackets )
750 {
751 tk->i_subpacket = 0;
752 tk->i_out_subpacket = 0;
753 }
754 }
755
756 static void DemuxAudioMethod2( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_pts )
757 {
758 demux_sys_t *p_sys = p_demux->p_sys;
759
760 if( p_sys->i_buffer < 2 )
761 return;
762
763 unsigned i_sub = (p_sys->buffer[1] >> 4)&0x0f;
764 if( p_sys->i_buffer < 2+2*i_sub )
765 return;
766
767 uint8_t *p_sub = &p_sys->buffer[2+2*i_sub];
768
769 for( unsigned i = 0; i < i_sub; i++ )
770 {
771 const int i_sub_size = GetWBE( &p_sys->buffer[2+i*2] );
772 block_t *p_block = block_Alloc( i_sub_size );
773 if( !p_block )
774 break;
775
776 if( &p_sub[i_sub_size] > &p_sys->buffer[p_sys->i_buffer] )
777 break;
778
779 memcpy( p_block->p_buffer, p_sub, i_sub_size );
780 p_sub += i_sub_size;
781
782 p_block->i_dts =
783 p_block->i_pts = i == 0 ? i_pts : VLC_TICK_INVALID;
784
785 CheckPcr( p_demux, tk, p_block->i_pts );
786 es_out_Send( p_demux->out, tk->p_es, p_block );
787 }
788 }
789 static void DemuxAudioMethod3( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_pts )
790 {
791 demux_sys_t *p_sys = p_demux->p_sys;
792
793 if( p_sys->i_buffer <= 0 )
794 return;
795
796 block_t *p_block = block_Alloc( p_sys->i_buffer );
797 if( !p_block )
798 return;
799
800 if( tk->fmt.i_codec == VLC_CODEC_A52 )
801 {
802 uint8_t *p_src = p_sys->buffer;
803 uint8_t *p_dst = p_block->p_buffer;
804
805 /* byte swap data */
806 while( p_dst < &p_block->p_buffer[p_sys->i_buffer - 1])
807 {
808 *p_dst++ = p_src[1];
809 *p_dst++ = p_src[0];
810
811 p_src += 2;
812 }
813 }
814 else
815 {
816 memcpy( p_block->p_buffer, p_sys->buffer, p_sys->i_buffer );
817 }
818 p_block->i_dts =
819 p_block->i_pts = i_pts;
820
821 CheckPcr( p_demux, tk, p_block->i_pts );
822 es_out_Send( p_demux->out, tk->p_es, p_block );
823 }
824
825 // Sipr packet re-ordering code and index table borrowed from
826 // the MPlayer Realmedia demuxer.
827 static const uint8_t sipr_swap_index_table[38][2] = {
828 { 0, 63 }, { 1, 22 }, { 2, 44 }, { 3, 90 },
829 { 5, 81 }, { 7, 31 }, { 8, 86 }, { 9, 58 },
830 { 10, 36 }, { 12, 68 }, { 13, 39 }, { 14, 73 },
831 { 15, 53 }, { 16, 69 }, { 17, 57 }, { 19, 88 },
832 { 20, 34 }, { 21, 71 }, { 24, 46 }, { 25, 94 },
833 { 26, 54 }, { 28, 75 }, { 29, 50 }, { 32, 70 },
834 { 33, 92 }, { 35, 74 }, { 38, 85 }, { 40, 56 },
835 { 42, 87 }, { 43, 65 }, { 45, 59 }, { 48, 79 },
836 { 49, 93 }, { 51, 89 }, { 55, 95 }, { 61, 76 },
837 { 67, 83 }, { 77, 80 }
838 };
839
840 static void SiprPacketReorder(uint8_t *buf, int sub_packet_h, int framesize)
841 {
842 int n, bs = sub_packet_h * framesize * 2 / 96; // nibbles per subpacket
843
844 for (n = 0; n < 38; n++) {
845 int j;
846 int i = bs * sipr_swap_index_table[n][0];
847 int o = bs * sipr_swap_index_table[n][1];
848
849 /* swap 4 bit-nibbles of block 'i' with 'o' */
850 for (j = 0; j < bs; j++, i++, o++) {
851 int x = (buf[i >> 1] >> (4 * (i & 1))) & 0xF,
852 y = (buf[o >> 1] >> (4 * (o & 1))) & 0xF;
853
854 buf[o >> 1] = (x << (4 * (o & 1))) |
855 (buf[o >> 1] & (0xF << (4 * !(o & 1))));
856 buf[i >> 1] = (y << (4 * (i & 1))) |
857 (buf[i >> 1] & (0xF << (4 * !(i & 1))));
858 }
859 }
860 }
861
862 static void DemuxAudioSipr( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_pts )
863 {
864 demux_sys_t *p_sys = p_demux->p_sys;
865 block_t *p_block = tk->p_sipr_packet;
866
867 if( p_sys->i_buffer < tk->i_frame_size
868 || tk->i_sipr_subpacket_count >= tk->i_subpacket_h )
869 return;
870
871 if( !p_block )
872 {
873 p_block = block_Alloc( tk->i_frame_size * tk->i_subpacket_h );
874 if( !p_block )
875 return;
876 tk->p_sipr_packet = p_block;
877 }
878 memcpy( p_block->p_buffer + tk->i_sipr_subpacket_count * tk->i_frame_size,
879 p_sys->buffer, tk->i_frame_size );
880 if (!tk->i_sipr_subpacket_count)
881 {
882 p_block->i_dts =
883 p_block->i_pts = i_pts;
884 }
885
886 if( ++tk->i_sipr_subpacket_count < tk->i_subpacket_h )
887 return;
888
889 SiprPacketReorder(p_block->p_buffer, tk->i_subpacket_h, tk->i_frame_size);
890 CheckPcr( p_demux, tk, p_block->i_pts );
891 es_out_Send( p_demux->out, tk->p_es, p_block );
892 tk->i_sipr_subpacket_count = 0;
893 tk->p_sipr_packet = NULL;
894 }
895
896 static void DemuxAudio( demux_t *p_demux, real_track_t *tk, vlc_tick_t i_pts, unsigned i_flags )
897 {
898 switch( tk->fmt.i_codec )
899 {
900 case VLC_CODEC_COOK:
901 case VLC_CODEC_ATRAC3:
902 case VLC_CODEC_RA_288:
903 DemuxAudioMethod1( p_demux, tk, i_pts, i_flags );
904 break;
905 case VLC_CODEC_MP4A:
906 DemuxAudioMethod2( p_demux, tk, i_pts );
907 break;
908 case VLC_CODEC_SIPR:
909 DemuxAudioSipr( p_demux, tk, i_pts );
910 break;
911 default:
912 DemuxAudioMethod3( p_demux, tk, i_pts );
913 break;
914 }
915 }
916
917 /*****************************************************************************
918 * Helpers: seek/control
919 *****************************************************************************/
920 static int ControlGoToIndex( demux_t *p_demux, real_index_t *p_index )
921 {
922 demux_sys_t *p_sys = p_demux->p_sys;
923
924 p_sys->b_seek = true;
925 p_sys->i_pcr = INT64_C(1000) * p_index->i_time_offset;
926 for( int i = 0; i < p_sys->i_track; i++ )
927 p_sys->track[i]->i_last_dts = 0;
928 return vlc_stream_Seek( p_demux->s, p_index->i_file_offset );
929 }
930 static int ControlSeekTime( demux_t *p_demux, vlc_tick_t i_time )
931 {
932 demux_sys_t *p_sys = p_demux->p_sys;
933 real_index_t *p_index = p_sys->p_index;
934
935 while( p_index->i_file_offset != 0 )
936 {
937 if( p_index->i_time_offset * INT64_C(1000) > i_time )
938 {
939 if( p_index != p_sys->p_index )
940 p_index--;
941 break;
942 }
943 p_index++;
944 }
945 if( p_index->i_file_offset == 0 )
946 return VLC_EGENERIC;
947 return ControlGoToIndex( p_demux, p_index );
948 }
949 static int ControlSeekByte( demux_t *p_demux, int64_t i_bytes )
950 {
951 demux_sys_t *p_sys = p_demux->p_sys;
952 real_index_t *p_index = p_sys->p_index;
953
954 while( p_index->i_file_offset != 0 )
955 {
956 if( p_index->i_file_offset > i_bytes )
957 {
958 if( p_index != p_sys->p_index )
959 p_index--;
960 break;
961 }
962 p_index++;
963 }
964 if( p_index->i_file_offset == 0 )
965 return VLC_EGENERIC;
966 return ControlGoToIndex( p_demux, p_index );
967 }
968
969 /*****************************************************************************
970 * Helpers: header reading
971 *****************************************************************************/
972
973 /**
974 * This function will read a Pascal string with size stored in 2 bytes from
975 * a stream_t.
976 *
977 * FIXME what is the right charset ?
978 */
979 static char *StreamReadString2( stream_t *s )
980 {
981 uint8_t p_tmp[2];
982
983 if( vlc_stream_Read( s, p_tmp, 2 ) < 2 )
984 return NULL;
985
986 const int i_length = GetWBE( p_tmp );
987 if( i_length <= 0 )
988 return NULL;
989
990 char *psz_string = malloc( i_length + 1 );
991 if( unlikely(psz_string == NULL) )
992 return NULL;
993
994 if( vlc_stream_Read( s, psz_string, i_length ) < i_length )
995 {
996 free( psz_string );
997 return NULL;
998 }
999
1000 psz_string[i_length] = '\0';
1001 EnsureUTF8( psz_string );
1002 return psz_string;
1003 }
1004
1005 /**
1006 * This function will read a pascal string with size stored in 1 byte from a
1007 * memory buffer.
1008 *
1009 * FIXME what is the right charset ?
1010 */
1011 static char *MemoryReadString1( const uint8_t **pp_data, int *pi_data )
1012 {
1013 const uint8_t *p_data = *pp_data;
1014 int i_data = *pi_data;
1015
1016 char *psz_string = NULL;
1017
1018 if( i_data < 1 )
1019 goto exit;
1020
1021 int i_length = *p_data++; i_data--;
1022 if( i_length > i_data )
1023 i_length = i_data;
1024
1025 if( i_length > 0 )
1026 {
1027 psz_string = strndup( (const char*)p_data, i_length );
1028 if( psz_string )
1029 EnsureUTF8( psz_string );
1030
1031 p_data += i_length;
1032 i_data -= i_length;
1033 }
1034
1035 exit:
1036 *pp_data = p_data;
1037 *pi_data = i_data;
1038 return psz_string;
1039 }
1040
1041 /**
1042 * This function parses(skip) the .RMF identification chunk.
1043 */
1044 static int HeaderRMF( demux_t *p_demux )
1045 {
1046 uint8_t p_buffer[8];
1047
1048 if( vlc_stream_Read( p_demux->s, p_buffer, 8 ) < 8 )
1049 return VLC_EGENERIC;
1050
1051 msg_Dbg( p_demux, " - file version=0x%x num headers=%d",
1052 GetDWBE( &p_buffer[0] ), GetDWBE( &p_buffer[4] ) );
1053 return VLC_SUCCESS;
1054 }
1055 /**
1056 * This function parses the PROP properties chunk.
1057 */
1058 static int HeaderPROP( demux_t *p_demux )
1059 {
1060 demux_sys_t *p_sys = p_demux->p_sys;
1061
1062 uint8_t p_buffer[40];
1063 int i_flags;
1064
1065 if( vlc_stream_Read( p_demux->s, p_buffer, 40 ) < 40 )
1066 return VLC_EGENERIC;
1067
1068 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
1069 GetDWBE(&p_buffer[0]), GetDWBE(&p_buffer[4]) );
1070 msg_Dbg( p_demux, " - max packet size=%d avg bitrate=%d",
1071 GetDWBE(&p_buffer[8]), GetDWBE(&p_buffer[12]) );
1072 msg_Dbg( p_demux, " - packets count=%d", GetDWBE(&p_buffer[16]) );
1073 msg_Dbg( p_demux, " - duration=%d ms", GetDWBE(&p_buffer[20]) );
1074 msg_Dbg( p_demux, " - preroll=%d ms", GetDWBE(&p_buffer[24]) );
1075 msg_Dbg( p_demux, " - index offset=%d", GetDWBE(&p_buffer[28]) );
1076 msg_Dbg( p_demux, " - data offset=%d", GetDWBE(&p_buffer[32]) );
1077 msg_Dbg( p_demux, " - num streams=%d", GetWBE(&p_buffer[36]) );
1078
1079 /* set the duration for export in control */
1080 p_sys->i_our_duration = GetDWBE(&p_buffer[20]);
1081
1082 p_sys->i_index_offset = GetDWBE(&p_buffer[28]);
1083
1084 i_flags = GetWBE(&p_buffer[38]);
1085 msg_Dbg( p_demux, " - flags=0x%x %s%s%s",
1086 i_flags,
1087 i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
1088 i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
1089 i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
1090
1091 return VLC_SUCCESS;
1092 }
1093 /**
1094 * This functions parses the CONT commentairs chunk.
1095 */
1096 static int HeaderCONT( demux_t *p_demux )
1097 {
1098 demux_sys_t *p_sys = p_demux->p_sys;
1099
1100 /* */
1101 p_sys->psz_title = StreamReadString2( p_demux->s );
1102 if( p_sys->psz_title )
1103 msg_Dbg( p_demux, " - title=`%s'", p_sys->psz_title );
1104
1105 /* */
1106 p_sys->psz_artist = StreamReadString2( p_demux->s );
1107 if( p_sys->psz_artist )
1108 msg_Dbg( p_demux, " - artist=`%s'", p_sys->psz_artist );
1109
1110 /* */
1111 p_sys->psz_copyright = StreamReadString2( p_demux->s );
1112 if( p_sys->psz_copyright )
1113 msg_Dbg( p_demux, " - copyright=`%s'", p_sys->psz_copyright );
1114
1115 /* */
1116 p_sys->psz_description = StreamReadString2( p_demux->s );
1117 if( p_sys->psz_description )
1118 msg_Dbg( p_demux, " - comment=`%s'", p_sys->psz_description );
1119
1120 return VLC_SUCCESS;
1121 }
1122 /**
1123 * This function parses the MDPR (Media properties) chunk.
1124 */
1125 static int HeaderMDPR( demux_t *p_demux )
1126 {
1127 uint8_t p_buffer[30];
1128
1129 if( vlc_stream_Read( p_demux->s, p_buffer, 30 ) < 30 )
1130 return VLC_EGENERIC;
1131
1132 const int i_num = GetWBE( &p_buffer[0] );
1133 msg_Dbg( p_demux, " - id=0x%x", i_num );
1134 msg_Dbg( p_demux, " - max bitrate=%d avg bitrate=%d",
1135 GetDWBE(&p_buffer[2]), GetDWBE(&p_buffer[6]) );
1136 msg_Dbg( p_demux, " - max packet size=%d avg packet size=%d",
1137 GetDWBE(&p_buffer[10]), GetDWBE(&p_buffer[14]) );
1138 msg_Dbg( p_demux, " - start time=%d", GetDWBE(&p_buffer[18]) );
1139 msg_Dbg( p_demux, " - preroll=%d", GetDWBE(&p_buffer[22]) );
1140 msg_Dbg( p_demux, " - duration=%d", GetDWBE(&p_buffer[26]) );
1141
1142 /* */
1143 const uint8_t *p_peek;
1144 int i_peek_org = vlc_stream_Peek( p_demux->s, &p_peek, 2 * 256 );
1145 int i_peek = i_peek_org;
1146 if( i_peek <= 0 )
1147 return VLC_EGENERIC;
1148
1149 char *psz_name = MemoryReadString1( &p_peek, &i_peek );
1150 if( psz_name )
1151 {
1152 msg_Dbg( p_demux, " - name=`%s'", psz_name );
1153 free( psz_name );
1154 }
1155 char *psz_mime = MemoryReadString1( &p_peek, &i_peek );
1156 if( psz_mime )
1157 {
1158 msg_Dbg( p_demux, " - mime=`%s'", psz_mime );
1159 free( psz_mime );
1160 }
1161 const int i_skip = i_peek_org - i_peek;
1162 if( i_skip > 0 && vlc_stream_Read( p_demux->s, NULL, i_skip ) < i_skip )
1163 return VLC_EGENERIC;
1164
1165 /* */
1166 if( vlc_stream_Read( p_demux->s, p_buffer, 4 ) < 4 )
1167 return VLC_EGENERIC;
1168
1169 const uint32_t i_size = GetDWBE( p_buffer );
1170 if( i_size > 0 )
1171 {
1172 CodecParse( p_demux, i_size, i_num );
1173 unsigned size = vlc_stream_Read( p_demux->s, NULL, i_size );
1174 if( size < i_size )
1175 return VLC_EGENERIC;
1176 }
1177 return VLC_SUCCESS;
1178 }
1179 /**
1180 * This function parses DATA chunk (it contains the actual movie data).
1181 */
1182 static int HeaderDATA( demux_t *p_demux, uint32_t i_size )
1183 {
1184 demux_sys_t *p_sys = p_demux->p_sys;
1185 uint8_t p_buffer[8];
1186
1187 if( vlc_stream_Read( p_demux->s, p_buffer, 8 ) < 8 )
1188 return VLC_EGENERIC;
1189
1190 p_sys->i_data_offset = vlc_stream_Tell( p_demux->s ) - 10;
1191 p_sys->i_data_size = i_size;
1192 p_sys->i_data_packets_count = GetDWBE( p_buffer );
1193 p_sys->i_data_packets = 0;
1194 p_sys->i_data_offset_next = GetDWBE( &p_buffer[4] );
1195
1196 msg_Dbg( p_demux, " - packets count=%d next=%u",
1197 p_sys->i_data_packets_count,
1198 (unsigned int)p_sys->i_data_offset_next );
1199 return VLC_SUCCESS;
1200 }
1201 /**
1202 * This function parses the INDX (movie index chunk).
1203 * It is optional but seeking without it is ... hard.
1204 */
1205 static void HeaderINDX( demux_t *p_demux )
1206 {
1207 demux_sys_t *p_sys = p_demux->p_sys;
1208 uint8_t buffer[20];
1209
1210 uint32_t i_index_count;
1211
1212 if( p_sys->i_index_offset == 0 )
1213 return;
1214
1215 if( vlc_stream_Seek( p_demux->s, p_sys->i_index_offset )
1216 || vlc_stream_Read( p_demux->s, buffer, 20 ) < 20 )
1217 return ;
1218
1219 const uint32_t i_id = VLC_FOURCC( buffer[0], buffer[1], buffer[2], buffer[3] );
1220 const uint32_t i_size = GetDWBE( &buffer[4] );
1221 int i_version = GetWBE( &buffer[8] );
1222
1223 msg_Dbg( p_demux, "Real index %4.4s size=%d version=%d",
1224 (char*)&i_id, i_size, i_version );
1225
1226 if( (i_size < 20) && (i_id != VLC_FOURCC('I','N','D','X')) )
1227 return;
1228
1229 i_index_count = GetDWBE( &buffer[10] );
1230
1231 msg_Dbg( p_demux, "Real Index : num : %d ", i_index_count );
1232
1233 if( i_index_count >= ( 0xffffffff / sizeof(*p_sys->p_index) ) )
1234 return;
1235
1236 if( GetDWBE( &buffer[16] ) > 0 )
1237 msg_Dbg( p_demux, "Real Index: Does next index exist? %d ",
1238 GetDWBE( &buffer[16] ) );
1239
1240 /* One extra entry is allocated (that MUST be set to 0) to identify the
1241 * end of the index.
1242 * TODO add a clean entry count (easier to build index on the fly) */
1243 p_sys->p_index = calloc( i_index_count + 1, sizeof(*p_sys->p_index) );
1244 if( !p_sys->p_index )
1245 return;
1246
1247 for( unsigned int i = 0; i < i_index_count; i++ )
1248 {
1249 uint8_t p_entry[14];
1250
1251 if( vlc_stream_Read( p_demux->s, p_entry, 14 ) < 14 )
1252 return ;
1253
1254 if( GetWBE( &p_entry[0] ) != 0 )
1255 {
1256 msg_Dbg( p_demux, "Real Index: invaild version of index entry %d ",
1257 GetWBE( &p_entry[0] ) );
1258 return;
1259 }
1260
1261 real_index_t *p_idx = &p_sys->p_index[i];
1262
1263 p_idx->i_time_offset = GetDWBE( &p_entry[2] );
1264 p_idx->i_file_offset = GetDWBE( &p_entry[6] );
1265 p_idx->i_frame_index = GetDWBE( &p_entry[10] );
1266
1267 #if 0
1268 msg_Dbg( p_demux,
1269 "Real Index: time %"PRIu32" file %"PRIu32" frame %"PRIu32,
1270 p_idx->i_time_offset,
1271 p_idx->i_file_offset,
1272 p_idx->i_frame_index );
1273 #endif
1274 }
1275 }
1276
1277
1278 /**
1279 * This function parses the complete RM headers and move the
1280 * stream pointer to the data to be read.
1281 */
1282 static int HeaderRead( demux_t *p_demux )
1283 {
1284 demux_sys_t *p_sys = p_demux->p_sys;
1285
1286 for( ;; )
1287 {
1288 const int64_t i_stream_position = vlc_stream_Tell( p_demux->s );
1289 uint8_t header[100]; /* FIXME */
1290
1291 /* Read the header */
1292 if( vlc_stream_Read( p_demux->s, header, 10 ) < 10 )
1293 return VLC_EGENERIC;
1294
1295 const uint32_t i_id = VLC_FOURCC( header[0], header[1],
1296 header[2], header[3] );
1297 const uint32_t i_size = GetDWBE( &header[4] );
1298 const int i_version = GetWBE( &header[8] );
1299
1300 msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
1301 (char*)&i_id, i_size, i_version );
1302
1303 /* */
1304 if( i_size < 10 && i_id != VLC_FOURCC('D','A','T','A') )
1305 {
1306 msg_Dbg( p_demux, "invalid size for object %4.4s", (char*)&i_id );
1307 return VLC_EGENERIC;
1308 }
1309
1310 int i_ret;
1311 switch( i_id )
1312 {
1313 case VLC_FOURCC('.','R','M','F'):
1314 i_ret = HeaderRMF( p_demux );
1315 break;
1316 case VLC_FOURCC('P','R','O','P'):
1317 i_ret = HeaderPROP( p_demux );
1318 break;
1319 case VLC_FOURCC('C','O','N','T'):
1320 i_ret = HeaderCONT( p_demux );
1321 break;
1322 case VLC_FOURCC('M','D','P','R'):
1323 i_ret = HeaderMDPR( p_demux );
1324 break;
1325 case VLC_FOURCC('D','A','T','A'):
1326 i_ret = HeaderDATA( p_demux, i_size );
1327 break;
1328 default:
1329 /* unknown header */
1330 msg_Dbg( p_demux, "unknown chunk" );
1331 i_ret = VLC_SUCCESS;
1332 break;
1333 }
1334 if( i_ret )
1335 return i_ret;
1336
1337 if( i_id == VLC_FOURCC('D','A','T','A') ) /* In this case, parsing is finished */
1338 break;
1339
1340 /* Skip unread data */
1341 const int64_t i_stream_current = vlc_stream_Tell( p_demux->s );
1342 const int64_t i_stream_skip = (i_stream_position + i_size) - i_stream_current;
1343
1344 if( i_stream_skip > 0 )
1345 {
1346 if( vlc_stream_Read( p_demux->s, NULL, i_stream_skip ) != i_stream_skip )
1347 return VLC_EGENERIC;
1348 }
1349 else if( i_stream_skip < 0 )
1350 {
1351 return VLC_EGENERIC;
1352 }
1353 }
1354
1355 /* read index if possible */
1356 if( p_sys->i_index_offset > 0 )
1357 {
1358 const int64_t i_position = vlc_stream_Tell( p_demux->s );
1359
1360 HeaderINDX( p_demux );
1361
1362 if( vlc_stream_Seek( p_demux->s, i_position ) )
1363 return VLC_EGENERIC;
1364 }
1365 return VLC_SUCCESS;
1366 }
1367
1368 static void CodecMetaRead( demux_t *p_demux, const uint8_t **pp_data, int *pi_data )
1369 {
1370 demux_sys_t *p_sys = p_demux->p_sys;
1371
1372 /* Title */
1373 p_sys->psz_title = MemoryReadString1( pp_data, pi_data );
1374 if( p_sys->psz_title )
1375 msg_Dbg( p_demux, " - title=`%s'", p_sys->psz_title );
1376
1377 /* Authors */
1378 p_sys->psz_artist = MemoryReadString1( pp_data, pi_data );
1379 if( p_sys->psz_artist )
1380 msg_Dbg( p_demux, " - artist=`%s'", p_sys->psz_artist );
1381
1382 /* Copyright */
1383 p_sys->psz_copyright = MemoryReadString1( pp_data, pi_data );
1384 if( p_sys->psz_copyright )
1385 msg_Dbg( p_demux, " - copyright=`%s'", p_sys->psz_copyright );
1386
1387 /* Comment */
1388 p_sys->psz_description = MemoryReadString1( pp_data, pi_data );
1389 if( p_sys->psz_description )
1390 msg_Dbg( p_demux, " - Comment=`%s'", p_sys->psz_description );
1391 }
1392
1393 static int CodecVideoParse( demux_t *p_demux, int i_tk_id, const uint8_t *p_data, int i_data )
1394 {
1395 demux_sys_t *p_sys = p_demux->p_sys;
1396
1397 if( i_data < 34 )
1398 return VLC_EGENERIC;
1399
1400 /* */
1401 es_format_t fmt;
1402 es_format_Init( &fmt, VIDEO_ES,
1403 VLC_FOURCC( p_data[8], p_data[9], p_data[10], p_data[11] ) );
1404 fmt.video.i_width = GetWBE( &p_data[12] );
1405 fmt.video.i_height= GetWBE( &p_data[14] );
1406 fmt.video.i_visible_width = fmt.video.i_width;
1407 fmt.video.i_visible_height = fmt.video.i_height;
1408 fmt.video.i_frame_rate = (GetWBE( &p_data[22] ) << 16) | GetWBE( &p_data[24] );
1409 fmt.video.i_frame_rate_base = 1 << 16;
1410
1411 fmt.i_extra = i_data - 26;
1412 fmt.p_extra = malloc( fmt.i_extra );
1413 if( !fmt.p_extra )
1414 return VLC_ENOMEM;
1415
1416 memcpy( fmt.p_extra, &p_data[26], fmt.i_extra );
1417
1418 //msg_Dbg( p_demux, " - video 0x%08x 0x%08x", dw0, dw1 );
1419
1420 /* */
1421 switch( GetDWBE( &p_data[30] ) )
1422 {
1423 case 0x10003000:
1424 case 0x10003001:
1425 fmt.i_codec = VLC_CODEC_RV13;
1426 break;
1427 case 0x20001000:
1428 case 0x20100001:
1429 case 0x20200002:
1430 case 0x20201002:
1431 fmt.i_codec = VLC_CODEC_RV20;
1432 break;
1433 case 0x30202002:
1434 fmt.i_codec = VLC_CODEC_RV30;
1435 break;
1436 case 0x40000000:
1437 fmt.i_codec = VLC_CODEC_RV40;
1438 break;
1439 }
1440 msg_Dbg( p_demux, " - video %4.4s %dx%d - %8.8x",
1441 (char*)&fmt.i_codec, fmt.video.i_width, fmt.video.i_height, GetDWBE( &p_data[30] ) );
1442
1443 real_track_t *tk = malloc( sizeof( *tk ) );
1444 if( !tk )
1445 {
1446 es_format_Clean( &fmt );
1447 return VLC_ENOMEM;
1448 }
1449 tk->i_out_subpacket = 0;
1450 tk->i_subpacket = 0;
1451 tk->i_subpackets = 0;
1452 tk->p_subpackets = NULL;
1453 tk->p_subpackets_timecode = NULL;
1454 tk->i_id = i_tk_id;
1455 tk->fmt = fmt;
1456 tk->i_frame_num = -1;
1457 tk->i_frame_size = 0;
1458 tk->p_frame = NULL;
1459 tk->i_last_dts = 0;
1460 tk->p_sipr_packet = NULL;
1461 tk->p_es = es_out_Add( p_demux->out, &fmt );
1462
1463 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
1464 return VLC_SUCCESS;
1465 }
1466 static int CodecAudioParse( demux_t *p_demux, int i_tk_id, const uint8_t *p_data, int i_data )
1467 {
1468 demux_sys_t *p_sys = p_demux->p_sys;
1469 es_format_t fmt;
1470
1471 if( i_data < 6 )
1472 return VLC_EGENERIC;
1473
1474 int i_flavor = 0;
1475 int i_coded_frame_size = 0;
1476 int i_subpacket_h = 0;
1477 int i_frame_size = 0;
1478 int i_subpacket_size = 0;
1479 char p_genr[4] = {0,0,0,0};
1480 int i_version = GetWBE( &p_data[4] );
1481 int i_extra_codec = 0;
1482
1483 msg_Dbg( p_demux, " - audio version=%d", i_version );
1484
1485 es_format_Init( &fmt, AUDIO_ES, 0 );
1486
1487 RVoid( &p_data, &i_data, 6 ); /* 4 + version */
1488 if( i_version == 3 ) /* RMF version 3 or .ra version 3 */
1489 {
1490 RVoid( &p_data, &i_data, 2 + 10 + 4 );
1491
1492 /* Meta Datas */
1493 CodecMetaRead( p_demux, &p_data, &i_data );
1494
1495 RVoid( &p_data, &i_data, 1 + 1 );
1496 if( i_data >= 4 )
1497 memcpy( &fmt.i_codec, p_data, 4 );
1498 RVoid( &p_data, &i_data, 4 );
1499
1500 fmt.audio.i_channels = 1; /* This is always the case in rm3 */
1501 fmt.audio.i_rate = 8000;
1502
1503 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
1504 (char*)&fmt.i_codec, fmt.audio.i_channels, fmt.audio.i_rate );
1505 }
1506 else /* RMF version 4/5 or .ra version 4 */
1507 {
1508 RVoid( &p_data, &i_data, 2 + 4 + 4 + 2 + 4 );
1509 i_flavor = R16( &p_data, &i_data );
1510 i_coded_frame_size = R32( &p_data, &i_data );
1511 RVoid( &p_data, &i_data, 4 + 4 + 4 );
1512 i_subpacket_h = R16( &p_data, &i_data );
1513 i_frame_size = R16( &p_data, &i_data );
1514 i_subpacket_size = R16( &p_data, &i_data );
1515 if( !i_frame_size || !i_coded_frame_size )
1516 {
1517 es_format_Clean( &fmt );
1518 return VLC_EGENERIC;
1519 }
1520
1521 RVoid( &p_data, &i_data, 2 + (i_version == 5 ? 6 : 0 ) );
1522
1523 fmt.audio.i_rate = R16( &p_data, &i_data );
1524 RVoid( &p_data, &i_data, 2 );
1525 fmt.audio.i_bitspersample = R16( &p_data, &i_data );
1526 fmt.audio.i_channels = R16( &p_data, &i_data );
1527 fmt.audio.i_blockalign = i_frame_size;
1528
1529 if( i_version == 5 )
1530 {
1531 if( i_data >= 8 )
1532 {
1533 memcpy( p_genr, &p_data[0], 4 );
1534 memcpy( &fmt.i_codec, &p_data[4], 4 );
1535 }
1536 RVoid( &p_data, &i_data, 8 );
1537 }
1538 else /* version 4 */
1539 {
1540 if( i_data > 0 )
1541 RVoid( &p_data, &i_data, 1 + *p_data );
1542 if( i_data >= 1 + 4 )
1543 memcpy( &fmt.i_codec, &p_data[1], 4 );
1544 if( i_data > 0 )
1545 RVoid( &p_data, &i_data, 1 + *p_data );
1546 }
1547
1548 msg_Dbg( p_demux, " - audio codec=%4.4s channels=%d rate=%dHz",
1549 (char*)&fmt.i_codec, fmt.audio.i_channels, fmt.audio.i_rate );
1550
1551 RVoid( &p_data, &i_data, 3 );
1552
1553 if( p_sys->b_real_audio )
1554 {
1555 CodecMetaRead( p_demux, &p_data, &i_data );
1556 }
1557 else
1558 {
1559 if( i_version == 5 )
1560 RVoid( &p_data, &i_data, 1 );
1561 i_extra_codec = R32( &p_data, &i_data );
1562 }
1563 }
1564
1565 switch( fmt.i_codec )
1566 {
1567 case VLC_FOURCC('l','p','c','J'):
1568 case VLC_FOURCC('1','4','_','4'):
1569 fmt.i_codec = VLC_CODEC_RA_144;
1570 fmt.audio.i_blockalign = 0x14 ;
1571 break;
1572
1573 case VLC_FOURCC('2','8','_','8'):
1574 if( i_coded_frame_size <= 0 )
1575 {
1576 es_format_Clean( &fmt );
1577 return VLC_EGENERIC;
1578 }
1579 fmt.i_codec = VLC_CODEC_RA_288;
1580 fmt.audio.i_blockalign = i_coded_frame_size;
1581 break;
1582
1583 case VLC_FOURCC( 'a','5','2',' ' ):
1584 case VLC_FOURCC( 'd','n','e','t' ):
1585 fmt.i_codec = VLC_CODEC_A52;
1586 break;
1587
1588 case VLC_FOURCC( 'r','a','a','c' ):
1589 case VLC_FOURCC( 'r','a','c','p' ):
1590 fmt.i_codec = VLC_CODEC_MP4A;
1591
1592 if( i_extra_codec > 0 )
1593 {
1594 i_extra_codec--;
1595 RVoid( &p_data, &i_data, 1 );
1596 }
1597 if( i_extra_codec > 0 )
1598 {
1599 fmt.p_extra = malloc( i_extra_codec );
1600 if( !fmt.p_extra || i_extra_codec > i_data )
1601 {
1602 free( fmt.p_extra );
1603 return VLC_ENOMEM;
1604 }
1605
1606 fmt.i_extra = i_extra_codec;
1607 memcpy( fmt.p_extra, p_data, fmt.i_extra );
1608 }
1609 break;
1610
1611 case VLC_CODEC_SIPR:
1612 fmt.i_codec = VLC_CODEC_SIPR;
1613 if( i_flavor > 3 )
1614 {
1615 msg_Dbg( p_demux, " - unsupported sipr flavorc=%i", i_flavor );
1616 es_format_Clean( &fmt );
1617 return VLC_EGENERIC;
1618 }
1619
1620 i_subpacket_size = i_subpacket_size_sipr[i_flavor];
1621 // The libavcodec sipr decoder requires stream bitrate
1622 // to be set during initialization so that the correct mode
1623 // can be selected.
1624 fmt.i_bitrate = fmt.audio.i_rate;
1625 msg_Dbg( p_demux, " - sipr flavor=%i", i_flavor );
1626 /* fall through */
1627
1628 case VLC_CODEC_COOK:
1629 case VLC_CODEC_ATRAC3:
1630 if( i_subpacket_size <= 0 || i_frame_size / i_subpacket_size <= 0 )
1631 {
1632 es_format_Clean( &fmt );
1633 return VLC_EGENERIC;
1634 }
1635 if( !memcmp( p_genr, "genr", 4 ) )
1636 fmt.audio.i_blockalign = i_subpacket_size;
1637 else
1638 fmt.audio.i_blockalign = i_coded_frame_size;
1639
1640 if( i_extra_codec > 0 )
1641 {
1642 fmt.p_extra = malloc( i_extra_codec );
1643 if( !fmt.p_extra || i_extra_codec > i_data )
1644 {
1645 free( fmt.p_extra );
1646 return VLC_ENOMEM;
1647 }
1648
1649 fmt.i_extra = i_extra_codec;
1650 memcpy( fmt.p_extra, p_data, fmt.i_extra );
1651 }
1652
1653 break;
1654
1655 case VLC_FOURCC('r','a','l','f'):
1656 default:
1657 msg_Dbg( p_demux, " - unknown audio codec=%4.4s",
1658 (char*)&fmt.i_codec );
1659 es_format_Clean( &fmt );
1660 return VLC_EGENERIC;
1661 }
1662 msg_Dbg( p_demux, " - extra data=%d", fmt.i_extra );
1663
1664 /* */
1665 real_track_t *tk = malloc( sizeof( *tk ) );
1666 if( !tk )
1667 {
1668 es_format_Clean( &fmt );
1669 return VLC_ENOMEM;
1670 }
1671 tk->i_id = i_tk_id;
1672 tk->fmt = fmt;
1673 tk->i_frame_size = 0;
1674 tk->p_frame = NULL;
1675
1676 tk->i_subpacket_h = i_subpacket_h;
1677 tk->i_subpacket_size = i_subpacket_size;
1678 tk->i_coded_frame_size = i_coded_frame_size;
1679 tk->i_frame_size = i_frame_size;
1680
1681 tk->i_out_subpacket = 0;
1682 tk->i_subpacket = 0;
1683 tk->i_subpackets = 0;
1684 tk->p_subpackets = NULL;
1685 tk->p_subpackets_timecode = NULL;
1686
1687 tk->p_sipr_packet = NULL;
1688 tk->i_sipr_subpacket_count = 0;
1689
1690 if( fmt.i_codec == VLC_CODEC_COOK ||
1691 fmt.i_codec == VLC_CODEC_ATRAC3 )
1692 {
1693 tk->i_subpackets =
1694 i_subpacket_h * i_frame_size / tk->i_subpacket_size;
1695 tk->p_subpackets =
1696 xcalloc( tk->i_subpackets, sizeof(block_t *) );
1697 tk->p_subpackets_timecode =
1698 xcalloc( tk->i_subpackets , sizeof( int64_t ) );
1699 }
1700 else if( fmt.i_codec == VLC_CODEC_RA_288 )
1701 {
1702 tk->i_subpackets =
1703 i_subpacket_h * i_frame_size / tk->i_coded_frame_size;
1704 tk->p_subpackets =
1705 xcalloc( tk->i_subpackets, sizeof(block_t *) );
1706 tk->p_subpackets_timecode =
1707 xcalloc( tk->i_subpackets , sizeof( int64_t ) );
1708 }
1709
1710 /* Check if the calloc went correctly */
1711 if( tk->i_subpacket > 0 && ( !tk->p_subpackets || !tk->p_subpackets_timecode ) )
1712 {
1713 free( tk->p_subpackets_timecode );
1714 free( tk->p_subpackets );
1715 free( tk );
1716 msg_Err( p_demux, "Can't alloc subpacket" );
1717 return VLC_EGENERIC;
1718 }
1719
1720 tk->i_last_dts = 0;
1721 tk->p_es = es_out_Add( p_demux->out, &fmt );
1722
1723 TAB_APPEND( p_sys->i_track, p_sys->track, tk );
1724
1725 return VLC_SUCCESS;
1726 }
1727
1728
1729 static int CodecParse( demux_t *p_demux, int i_len, int i_num )
1730 {
1731 const uint8_t *p_peek;
1732
1733 msg_Dbg( p_demux, " - specific data len=%d", i_len );
1734 if( vlc_stream_Peek( p_demux->s, &p_peek, i_len ) < i_len )
1735 return VLC_EGENERIC;
1736
1737 if( i_len >= 8 && !memcmp( &p_peek[4], "VIDO", 4 ) )
1738 {
1739 return CodecVideoParse( p_demux, i_num, p_peek, i_len );
1740 }
1741 else if( i_len >= 4 && !memcmp( &p_peek[0], ".ra\xfd", 4 ) )
1742 {
1743 return CodecAudioParse( p_demux, i_num, p_peek, i_len );
1744 }
1745 return VLC_SUCCESS;
1746 }
1747
1748 /*****************************************************************************
1749 * Helpers: memory buffer fct.
1750 *****************************************************************************/
1751 static void RVoid( const uint8_t **pp_data, int *pi_data, int i_size )
1752 {
1753 if( i_size > *pi_data )
1754 i_size = *pi_data;
1755
1756 *pp_data += i_size;
1757 *pi_data -= i_size;
1758 }
1759 #define RX(name, type, size, code ) \
1760 static type name( const uint8_t **pp_data, int *pi_data ) { \
1761 if( *pi_data < (size) ) \
1762 return 0; \
1763 type v = code; \
1764 RVoid( pp_data, pi_data, size ); \
1765 return v; \
1766 }
1767 RX(R8, uint8_t, 1, **pp_data )
1768 RX(R16, uint16_t, 2, GetWBE( *pp_data ) )
1769 RX(R32, uint32_t, 4, GetDWBE( *pp_data ) )
1770 static int RLength( const uint8_t **pp_data, int *pi_data )
1771 {
1772 const int v0 = R16( pp_data, pi_data ) & 0x7FFF;
1773 if( v0 >= 0x4000 )
1774 return v0 - 0x4000;
1775 return (v0 << 16) | R16( pp_data, pi_data );
1776 }
1777
1778
VLC Development tree mirror