ty: fix variable aliasing bug
[vlc.git] / modules / demux / ty.c
blobdce11581bb51ae990c4ada769c785eebb59b501b
1 /*****************************************************************************
2 * ty.c - TiVo ty stream video demuxer for VLC
3 *****************************************************************************
4 * Copyright (C) 2005 VLC authors and VideoLAN
5 * Copyright (C) 2005 by Neal Symms (tivo@freakinzoo.com) - February 2005
6 * based on code by Christopher Wingert for tivo-mplayer
7 * tivo(at)wingert.org, February 2003
9 * $Id$
11 * This program is free software; you can redistribute it and/or modify it
12 * under the terms of the GNU Lesser General Public License as published by
13 * the Free Software Foundation; either version 2.1 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU Lesser General Public License for more details.
21 * You should have received a copy of the GNU Lesser General Public License
22 * along with this program; if not, write to the Free Software Foundation,
23 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25 * CODE CHANGES:
26 * v1.0.0 - 24-Feb-2005 - Initial release - Series 1 support ONLY!
27 * v1.0.1 - 25-Feb-2005 - Added fix for bad GOP headers - Neal
28 * v1.0.2 - 26-Feb-2005 - No longer require "seekable" input stream - Neal
29 * v2.0.0 - 21-Mar-2005 - Series 2 support! No AC-3 on S2 DTivo yet.
30 * v2.1.0 - 22-Mar-2005 - Support for AC-3 on S2 DTivo (long ac3 packets)
31 * v3.0.0 - 14-Jul-2005 - Support for skipping fwd/back via VLC hotkeys
32 *****************************************************************************/
34 /*****************************************************************************
35 * Preamble
36 *****************************************************************************/
38 #ifdef HAVE_CONFIG_H
39 # include "config.h"
40 #endif
42 #include <vlc_common.h>
43 #include <vlc_plugin.h>
44 #include <vlc_demux.h>
45 #include <vlc_codec.h>
46 #include <vlc_meta.h>
47 #include <vlc_input.h>
48 #include "../codec/cc.h"
50 #include <assert.h>
52 /*****************************************************************************
53 * Module descriptor
54 *****************************************************************************/
55 static int Open ( vlc_object_t * );
56 static void Close( vlc_object_t * );
58 vlc_module_begin ()
59 set_shortname( N_("TY") )
60 set_description(N_("TY Stream audio/video demux"))
61 set_category( CAT_INPUT )
62 set_subcategory( SUBCAT_INPUT_DEMUX )
63 set_capability("demux", 6)
64 /* FIXME: there seems to be a segfault when using PVR access
65 * and TY demux has a bigger priority than PS
66 * Something must be wrong.
68 set_callbacks( Open, Close )
69 add_shortcut("ty", "tivo")
70 vlc_module_end ()
72 /*****************************************************************************
73 * Local prototypes
74 *****************************************************************************/
75 static int Demux ( demux_t * );
76 static int Control( demux_t *, int, va_list );
78 #define SERIES1_PES_LENGTH (11) /* length of audio PES hdr on S1 */
79 #define SERIES2_PES_LENGTH (16) /* length of audio PES hdr on S2 */
80 #define AC3_PES_LENGTH (14) /* length of audio PES hdr for AC3 */
81 #define VIDEO_PES_LENGTH (16) /* length of video PES header */
82 #define DTIVO_PTS_OFFSET (6) /* offs into PES for MPEG PTS on DTivo */
83 #define SA_PTS_OFFSET (9) /* offset into PES for MPEG PTS on SA */
84 #define AC3_PTS_OFFSET (9) /* offset into PES for AC3 PTS on DTivo */
85 #define VIDEO_PTS_OFFSET (9) /* offset into PES for video PTS on all */
86 #define AC3_PKT_LENGTH (1536) /* size of TiVo AC3 pkts (w/o PES hdr) */
87 static const uint8_t ty_VideoPacket[] = { 0x00, 0x00, 0x01, 0xe0 };
88 static const uint8_t ty_MPEGAudioPacket[] = { 0x00, 0x00, 0x01, 0xc0 };
89 static const uint8_t ty_AC3AudioPacket[] = { 0x00, 0x00, 0x01, 0xbd };
91 #define CHUNK_PEEK_COUNT (3) /* number of chunks to probe */
93 /* packet types for reference:
94 2/c0: audio data continued
95 3/c0: audio packet header (PES header)
96 4/c0: audio data (S/A only?)
97 9/c0: audio packet header, AC-3 audio
98 2/e0: video data continued
99 6/e0: video packet header (PES header)
100 7/e0: video sequence header start
101 8/e0: video I-frame header start
102 a/e0: video P-frame header start
103 b/e0: video B-frame header start
104 c/e0: video GOP header start
105 e/01: closed-caption data
106 e/02: Extended data services data
107 e/03: ipreview data ("thumbs up to record" signal)
108 e/05: UK Teletext
111 #define TIVO_PES_FILEID ( 0xf5467abd )
112 #define TIVO_PART_LENGTH ( 0x20000000 ) /* 536,870,912 bytes */
113 #define CHUNK_SIZE ( 128 * 1024 )
115 typedef struct
117 long l_rec_size;
118 uint8_t ex[2];
119 uint8_t rec_type;
120 uint8_t subrec_type;
121 bool b_ext;
122 uint64_t l_ty_pts; /* TY PTS in the record header */
123 } ty_rec_hdr_t;
125 typedef struct
127 uint64_t l_timestamp;
128 uint8_t chunk_bitmask[8];
129 } ty_seq_table_t;
131 typedef enum
133 TIVO_TYPE_UNKNOWN,
134 TIVO_TYPE_SA,
135 TIVO_TYPE_DTIVO
136 } tivo_type_t;
138 typedef enum
140 TIVO_SERIES_UNKNOWN,
141 TIVO_SERIES1,
142 TIVO_SERIES2
143 } tivo_series_t;
145 typedef enum
147 TIVO_AUDIO_UNKNOWN,
148 TIVO_AUDIO_AC3,
149 TIVO_AUDIO_MPEG
150 } tivo_audio_t;
152 #define XDS_MAX_DATA_SIZE (32)
153 typedef enum
155 XDS_CLASS_CURRENT = 0,
156 XDS_CLASS_FUTURE = 1,
157 XDS_CLASS_CHANNEL = 2,
158 XDS_CLASS_MISCELLANEOUS = 3,
159 XDS_CLASS_PUBLIC_SERVICE = 4,
160 XDS_CLASS_RESERVED = 5,
161 XDS_CLASS_UNDEFINED = 6,
162 XDS_CLASS_OTHER = 7,
164 XDS_MAX_CLASS_COUNT
165 } xds_class_t;
166 typedef struct
168 bool b_started;
169 int i_data;
170 uint8_t p_data[XDS_MAX_DATA_SIZE];
171 int i_sum;
172 } xds_packet_t;
173 typedef enum
175 XDS_META_PROGRAM_RATING_NONE,
176 XDS_META_PROGRAM_RATING_MPAA,
177 XDS_META_PROGRAM_RATING_TPG,
178 /* TODO add CA/CE rating */
179 } xds_meta_program_rating_t;
180 typedef struct
182 char *psz_name;
183 xds_meta_program_rating_t rating;
184 char *psz_rating;
185 /* Add the other fields once I have the samples */
186 } xds_meta_program_t;
187 typedef struct
189 char *psz_channel_name;
190 char *psz_channel_call_letter;
191 char *psz_channel_number;
193 xds_meta_program_t current;
194 xds_meta_program_t future;
195 } xds_meta_t;
196 typedef struct
198 /* Are we in XDS mode */
199 bool b_xds;
201 /* Current class type */
202 xds_class_t i_class;
203 int i_type;
204 bool b_future;
206 /* */
207 xds_packet_t pkt[XDS_MAX_CLASS_COUNT][128]; /* XXX it is way too much, but simpler */
209 /* */
210 bool b_meta_changed;
211 xds_meta_t meta;
213 } xds_t;
215 struct demux_sys_t
217 es_out_id_t *p_video; /* ptr to video codec */
218 es_out_id_t *p_audio; /* holds either ac3 or mpeg codec ptr */
220 cc_data_t cc;
221 es_out_id_t *p_cc[4];
223 xds_t xds;
225 int i_cur_chunk;
226 int i_stuff_cnt;
227 size_t i_stream_size; /* size of input stream (if known) */
228 //uint64_t l_program_len; /* length of this stream in msec */
229 bool b_seekable; /* is this stream seekable? */
230 bool b_have_master; /* are master chunks present? */
231 tivo_type_t tivo_type; /* tivo type (SA / DTiVo) */
232 tivo_series_t tivo_series; /* Series1 or Series2 */
233 tivo_audio_t audio_type; /* AC3 or MPEG */
234 int i_Pes_Length; /* Length of Audio PES header */
235 int i_Pts_Offset; /* offset into audio PES of PTS */
236 uint8_t pes_buffer[20]; /* holds incomplete pes headers */
237 int i_pes_buf_cnt; /* how many bytes in our buffer */
238 size_t l_ac3_pkt_size; /* len of ac3 pkt we've seen so far */
239 uint64_t l_last_ty_pts; /* last TY timestamp we've seen */
240 //mtime_t l_last_ty_pts_sync; /* audio PTS at time of last TY PTS */
241 uint64_t l_first_ty_pts; /* first TY PTS in this master chunk */
242 uint64_t l_final_ty_pts; /* final TY PTS in this master chunk */
243 unsigned i_seq_table_size; /* number of entries in SEQ table */
244 unsigned i_bits_per_seq_entry; /* # of bits in SEQ table bitmask */
246 mtime_t firstAudioPTS;
247 mtime_t lastAudioPTS;
248 mtime_t lastVideoPTS;
250 ty_rec_hdr_t *rec_hdrs; /* record headers array */
251 int i_cur_rec; /* current record in this chunk */
252 int i_num_recs; /* number of recs in this chunk */
253 int i_seq_rec; /* record number where seq start is */
254 ty_seq_table_t *seq_table; /* table of SEQ entries from mstr chk */
255 bool eof;
256 bool b_first_chunk;
259 static int get_chunk_header(demux_t *);
260 static mtime_t get_pts( const uint8_t *buf );
261 static int find_es_header( const uint8_t *header,
262 const uint8_t *buffer, int i_search_len );
263 static int ty_stream_seek_pct(demux_t *p_demux, double seek_pct);
264 static int ty_stream_seek_time(demux_t *, uint64_t);
266 static ty_rec_hdr_t *parse_chunk_headers( const uint8_t *p_buf,
267 int i_num_recs, int *pi_payload_size);
268 static int probe_stream(demux_t *p_demux);
269 static void analyze_chunk(demux_t *p_demux, const uint8_t *p_chunk);
270 static void parse_master(demux_t *p_demux);
272 static int DemuxRecVideo( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in );
273 static int DemuxRecAudio( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in );
274 static int DemuxRecCc( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in );
276 static void DemuxDecodeXds( demux_t *p_demux, uint8_t d1, uint8_t d2 );
278 static void XdsInit( xds_t * );
279 static void XdsExit( xds_t * );
281 #define TY_ES_GROUP (1)
284 * Open: check file and initialize demux structures
286 * here's what we do:
287 * 1. peek at the first 12 bytes of the stream for the
288 * magic TiVo PART header & stream type & chunk size
289 * 2. if it's not there, error with VLC_EGENERIC
290 * 3. set up video (mpgv) codec
291 * 4. return VLC_SUCCESS
293 static int Open(vlc_object_t *p_this)
295 demux_t *p_demux = (demux_t *)p_this;
296 demux_sys_t *p_sys;
297 es_format_t fmt;
298 const uint8_t *p_peek;
299 int i;
301 /* peek at the first 12 bytes. */
302 /* for TY streams, they're always the same */
303 if( vlc_stream_Peek( p_demux->s, &p_peek, 12 ) < 12 )
304 return VLC_EGENERIC;
306 if ( U32_AT(p_peek) != TIVO_PES_FILEID ||
307 U32_AT(&p_peek[4]) != 0x02 ||
308 U32_AT(&p_peek[8]) != CHUNK_SIZE )
310 if( !p_demux->obj.force &&
311 !demux_IsPathExtension( p_demux, ".ty" ) &&
312 !demux_IsPathExtension( p_demux, ".ty+" ) )
313 return VLC_EGENERIC;
314 msg_Warn( p_demux, "this does not look like a TY file, "
315 "continuing anyway..." );
318 /* at this point, we assume we have a valid TY stream */
319 msg_Dbg( p_demux, "valid TY stream detected" );
321 p_sys = malloc(sizeof(demux_sys_t));
322 if( unlikely(p_sys == NULL) )
323 return VLC_ENOMEM;
325 /* Set exported functions */
326 p_demux->pf_demux = Demux;
327 p_demux->pf_control = Control;
329 /* create our structure that will hold all data */
330 p_demux->p_sys = p_sys;
331 memset(p_sys, 0, sizeof(demux_sys_t));
333 /* set up our struct (most were zero'd out with the memset above) */
334 p_sys->b_first_chunk = true;
335 p_sys->b_have_master = (U32_AT(p_peek) == TIVO_PES_FILEID);
336 p_sys->firstAudioPTS = -1;
337 p_sys->lastAudioPTS = VLC_TS_INVALID;
338 p_sys->lastVideoPTS = VLC_TS_INVALID;
339 p_sys->i_stream_size = stream_Size(p_demux->s);
340 p_sys->tivo_type = TIVO_TYPE_UNKNOWN;
341 p_sys->audio_type = TIVO_AUDIO_UNKNOWN;
342 p_sys->tivo_series = TIVO_SERIES_UNKNOWN;
343 p_sys->i_Pes_Length = 0;
344 p_sys->i_Pts_Offset = 0;
345 p_sys->l_ac3_pkt_size = 0;
347 /* see if this stream is seekable */
348 vlc_stream_Control( p_demux->s, STREAM_CAN_SEEK, &p_sys->b_seekable );
350 if (probe_stream(p_demux) != VLC_SUCCESS) {
351 //TyClose(p_demux);
352 return VLC_EGENERIC;
355 if (!p_sys->b_have_master)
356 msg_Warn(p_demux, "No master chunk found; seeking will be limited.");
358 /* register the proper audio codec */
359 if (p_sys->audio_type == TIVO_AUDIO_MPEG) {
360 es_format_Init( &fmt, AUDIO_ES, VLC_CODEC_MPGA );
361 } else {
362 es_format_Init( &fmt, AUDIO_ES, VLC_CODEC_A52 );
364 fmt.i_group = TY_ES_GROUP;
365 p_sys->p_audio = es_out_Add( p_demux->out, &fmt );
367 /* register the video stream */
368 es_format_Init( &fmt, VIDEO_ES, VLC_CODEC_MPGV );
369 fmt.i_group = TY_ES_GROUP;
370 p_sys->p_video = es_out_Add( p_demux->out, &fmt );
372 /* */
373 for( i = 0; i < 4; i++ )
374 p_sys->p_cc[i] = NULL;
375 cc_Init( &p_sys->cc );
377 XdsInit( &p_sys->xds );
379 return VLC_SUCCESS;
382 /* =========================================================================== */
383 /* Demux: Read & Demux one record from the chunk
385 * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
387 * NOTE: I think we can return the number of packets sent instead of just 1.
388 * that means we can demux an entire chunk and shoot it back (may be more efficient)
389 * -- should try that some day :) --
391 static int Demux( demux_t *p_demux )
393 demux_sys_t *p_sys = p_demux->p_sys;
394 ty_rec_hdr_t *p_rec;
395 block_t *p_block_in = NULL;
397 /*msg_Dbg(p_demux, "ty demux processing" );*/
399 /* did we hit EOF earlier? */
400 if( p_sys->eof )
401 return 0;
404 * what we do (1 record now.. maybe more later):
405 * - use vlc_stream_Read() to read the chunk header & record headers
406 * - discard entire chunk if it is a PART header chunk
407 * - parse all the headers into record header array
408 * - keep a pointer of which record we're on
409 * - use vlc_stream_Block() to fetch each record
410 * - parse out PTS from PES headers
411 * - set PTS for data packets
412 * - pass the data on to the proper codec via es_out_Send()
414 * if this is the first time or
415 * if we're at the end of this chunk, start a new one
417 /* parse the next chunk's record headers */
418 if( p_sys->b_first_chunk || p_sys->i_cur_rec >= p_sys->i_num_recs )
420 if( get_chunk_header(p_demux) == 0 || p_sys->i_num_recs == 0 )
421 return 0;
424 /*======================================================================
425 * parse & send one record of the chunk
426 *====================================================================== */
427 p_rec = &p_sys->rec_hdrs[p_sys->i_cur_rec];
429 if( !p_rec->b_ext )
431 const long l_rec_size = p_rec->l_rec_size;
432 /*msg_Dbg(p_demux, "Record Type 0x%x/%02x %ld bytes",
433 subrec_type, p_rec->rec_type, l_rec_size );*/
435 /* some normal records are 0 length, so check for that... */
436 if( l_rec_size <= 0 )
438 /* no data in payload; we're done */
439 p_sys->i_cur_rec++;
440 return 1;
443 /* read in this record's payload */
444 if( !( p_block_in = vlc_stream_Block( p_demux->s, l_rec_size ) ) )
445 return 0;
447 /* set these as 'unknown' for now */
448 p_block_in->i_pts =
449 p_block_in->i_dts = VLC_TS_INVALID;
451 /*else
453 -- don't read any data from the stream, data was in the record header --
454 msg_Dbg(p_demux,
455 "Record Type 0x%02x/%02x, ext data = %02x, %02x", subrec_type,
456 p_rec->rec_type, p_rec->ex1, p_rec->ex2);
459 switch( p_rec->rec_type )
461 case 0xe0: /* video */
462 DemuxRecVideo( p_demux, p_rec, p_block_in );
463 break;
465 case 0xc0: /* audio */
466 DemuxRecAudio( p_demux, p_rec, p_block_in );
467 break;
469 case 0x01:
470 case 0x02:
471 /* closed captions/XDS */
472 DemuxRecCc( p_demux, p_rec, p_block_in );
473 break;
475 default:
476 msg_Dbg(p_demux, "Invalid record type 0x%02x", p_rec->rec_type );
478 case 0x03: /* tivo data services */
479 case 0x05: /* unknown, but seen regularly */
480 if( p_block_in )
481 block_Release( p_block_in );
484 /* */
485 p_sys->i_cur_rec++;
486 return 1;
489 /* Control */
490 static int Control(demux_t *p_demux, int i_query, va_list args)
492 demux_sys_t *p_sys = p_demux->p_sys;
493 double f, *pf;
494 int64_t i64, *p_i64;
496 /*msg_Info(p_demux, "control cmd %d", i_query);*/
497 switch( i_query )
499 case DEMUX_CAN_SEEK:
500 *va_arg( args, bool * ) = p_sys->b_seekable;
501 return VLC_SUCCESS;
503 case DEMUX_GET_POSITION:
504 /* arg is 0.0 - 1.0 percent of overall file position */
505 if( ( i64 = p_sys->i_stream_size ) > 0 )
507 pf = va_arg( args, double* );
508 *pf = ((double)1.0) * vlc_stream_Tell( p_demux->s ) / (double) i64;
509 return VLC_SUCCESS;
511 return VLC_EGENERIC;
513 case DEMUX_SET_POSITION:
514 /* arg is 0.0 - 1.0 percent of overall file position */
515 f = (double) va_arg( args, double );
516 /* msg_Dbg(p_demux, "Control - set position to %2.3f", f); */
517 if ((i64 = p_sys->i_stream_size) > 0)
518 return ty_stream_seek_pct(p_demux, f);
519 return VLC_EGENERIC;
520 case DEMUX_GET_TIME:
521 /* return TiVo timestamp */
522 p_i64 = va_arg(args, int64_t *);
523 //*p_i64 = p_sys->lastAudioPTS - p_sys->firstAudioPTS;
524 //*p_i64 = (p_sys->l_last_ty_pts / 1000) + (p_sys->lastAudioPTS -
525 // p_sys->l_last_ty_pts_sync);
526 *p_i64 = (p_sys->l_last_ty_pts / 1000);
527 return VLC_SUCCESS;
528 case DEMUX_GET_LENGTH: /* length of program in microseconds, 0 if unk */
529 /* size / bitrate */
530 p_i64 = va_arg(args, int64_t *);
531 *p_i64 = 0;
532 return VLC_SUCCESS;
533 case DEMUX_SET_TIME: /* arg is time in microsecs */
534 i64 = va_arg( args, int64_t );
535 return ty_stream_seek_time(p_demux, i64 * 1000);
536 case DEMUX_GET_FPS:
537 default:
538 return VLC_EGENERIC;
542 /* Close */
543 static void Close( vlc_object_t *p_this )
545 demux_t *p_demux = (demux_t*)p_this;
546 demux_sys_t *p_sys = p_demux->p_sys;
548 XdsExit( &p_sys->xds );
549 cc_Exit( &p_sys->cc );
550 free( p_sys->rec_hdrs );
551 free( p_sys->seq_table );
552 free(p_sys);
556 /* =========================================================================== */
557 /* Compute Presentation Time Stamp (PTS)
558 * Assume buf points to beginning of PTS */
559 static mtime_t get_pts( const uint8_t *buf )
561 mtime_t i_pts;
563 i_pts = ((mtime_t)(buf[0]&0x0e ) << 29)|
564 (mtime_t)(buf[1] << 22)|
565 ((mtime_t)(buf[2]&0xfe) << 14)|
566 (mtime_t)(buf[3] << 7)|
567 (mtime_t)(buf[4] >> 1);
568 i_pts *= 100 / 9; /* convert PTS (90Khz clock) to microseconds */
569 return i_pts;
573 /* =========================================================================== */
574 static int find_es_header( const uint8_t *header,
575 const uint8_t *buffer, int i_search_len )
577 int count;
579 for( count = 0; count < i_search_len; count++ )
581 if( !memcmp( &buffer[count], header, 4 ) )
582 return count;
584 return -1;
588 /* =========================================================================== */
589 /* check if we have a full PES header, if not, then save what we have.
590 * this is called when audio-start packets are encountered.
591 * Returns:
592 * 1 partial PES hdr found, some audio data found (buffer adjusted),
593 * -1 partial PES hdr found, no audio data found
594 * 0 otherwise (complete PES found, pts extracted, pts set, buffer adjusted) */
595 /* TODO: HD support -- nothing known about those streams */
596 static int check_sync_pes( demux_t *p_demux, block_t *p_block,
597 int32_t offset, int32_t rec_len )
599 demux_sys_t *p_sys = p_demux->p_sys;
601 if ( offset < 0 || offset + p_sys->i_Pes_Length > rec_len )
603 /* entire PES header not present */
604 msg_Dbg( p_demux, "PES header at %d not complete in record. storing.",
605 offset );
606 /* save the partial pes header */
607 if( offset < 0 )
609 /* no header found, fake some 00's (this works, believe me) */
610 memset( p_sys->pes_buffer, 0, 4 );
611 p_sys->i_pes_buf_cnt = 4;
612 if( rec_len > 4 )
613 msg_Err( p_demux, "PES header not found in record of %d bytes!",
614 rec_len );
615 return -1;
617 /* copy the partial pes header we found */
618 memcpy( p_sys->pes_buffer, p_block->p_buffer + offset,
619 rec_len - offset );
620 p_sys->i_pes_buf_cnt = rec_len - offset;
622 if( offset > 0 )
624 /* PES Header was found, but not complete, so trim the end of this record */
625 p_block->i_buffer -= rec_len - offset;
626 return 1;
628 return -1; /* partial PES, no audio data */
630 /* full PES header present, extract PTS */
631 p_sys->lastAudioPTS = VLC_TS_0 + get_pts( &p_block->p_buffer[ offset +
632 p_sys->i_Pts_Offset ] );
633 if (p_sys->firstAudioPTS < 0)
634 p_sys->firstAudioPTS = p_sys->lastAudioPTS;
635 p_block->i_pts = p_sys->lastAudioPTS;
636 /*msg_Dbg(p_demux, "Audio PTS %"PRId64, p_sys->lastAudioPTS );*/
637 /* adjust audio record to remove PES header */
638 memmove(p_block->p_buffer + offset, p_block->p_buffer + offset +
639 p_sys->i_Pes_Length, rec_len - p_sys->i_Pes_Length);
640 p_block->i_buffer -= p_sys->i_Pes_Length;
641 #if 0
642 msg_Dbg(p_demux, "pes hdr removed; buffer len=%d and has "
643 "%02x %02x %02x %02x %02x %02x %02x %02x "
644 "%02x %02x %02x %02x %02x %02x %02x %02x", p_block->i_buffer,
645 p_block->p_buffer[0], p_block->p_buffer[1],
646 p_block->p_buffer[2], p_block->p_buffer[3],
647 p_block->p_buffer[4], p_block->p_buffer[5],
648 p_block->p_buffer[6], p_block->p_buffer[7],
649 p_block->p_buffer[8], p_block->p_buffer[9],
650 p_block->p_buffer[10], p_block->p_buffer[11],
651 p_block->p_buffer[12], p_block->p_buffer[13],
652 p_block->p_buffer[14], p_block->p_buffer[15]);
653 #endif
654 return 0;
657 static int DemuxRecVideo( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
659 demux_sys_t *p_sys = p_demux->p_sys;
660 const int subrec_type = rec_hdr->subrec_type;
661 const long l_rec_size = rec_hdr->l_rec_size; // p_block_in->i_buffer might be better
662 int esOffset1;
663 int i;
665 assert( rec_hdr->rec_type == 0xe0 );
666 if( !p_block_in )
667 return -1;
669 #if 0
670 msg_Dbg(p_demux, "packet buffer has "
671 "%02x %02x %02x %02x %02x %02x %02x %02x "
672 "%02x %02x %02x %02x %02x %02x %02x %02x",
673 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
674 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
675 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
676 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
677 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
678 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
679 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
680 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
681 #endif
682 //if( subrec_type == 0x06 || subrec_type == 0x07 )
683 if( subrec_type != 0x02 && subrec_type != 0x0c &&
684 subrec_type != 0x08 && l_rec_size > 4 )
686 /* get the PTS from this packet if it has one.
687 * on S1, only 0x06 has PES. On S2, however, most all do.
688 * Do NOT Pass the PES Header to the MPEG2 codec */
689 esOffset1 = find_es_header( ty_VideoPacket, p_block_in->p_buffer, 5 );
690 if( esOffset1 != -1 )
692 //msg_Dbg(p_demux, "Video PES hdr in pkt type 0x%02x at offset %d",
693 //subrec_type, esOffset1);
694 p_sys->lastVideoPTS = VLC_TS_0 + get_pts(
695 &p_block_in->p_buffer[ esOffset1 + VIDEO_PTS_OFFSET ] );
696 /*msg_Dbg(p_demux, "Video rec %d PTS %"PRId64, p_sys->i_cur_rec,
697 p_sys->lastVideoPTS );*/
698 if (subrec_type != 0x06) {
699 /* if we found a PES, and it's not type 6, then we're S2 */
700 /* The packet will have video data (& other headers) so we
701 * chop out the PES header and send the rest */
702 if (l_rec_size >= VIDEO_PES_LENGTH) {
703 p_block_in->p_buffer += VIDEO_PES_LENGTH + esOffset1;
704 p_block_in->i_buffer -= VIDEO_PES_LENGTH + esOffset1;
705 } else {
706 msg_Dbg(p_demux, "video rec type 0x%02x has short PES"
707 " (%ld bytes)", subrec_type, l_rec_size);
708 /* nuke this block; it's too short, but has PES marker */
709 p_block_in->i_buffer = 0;
712 }/* else
713 msg_Dbg(p_demux, "No Video PES hdr in pkt type 0x%02x",
714 subrec_type); */
717 if(subrec_type == 0x06 )
719 /* type 6 (S1 DTivo) has no data, so we're done */
720 block_Release(p_block_in);
721 return 0;
724 /* if it's not a continue blk, then set PTS */
725 if( subrec_type != 0x02 )
727 /*msg_Dbg(p_demux, "Video rec %d type 0x%02X", p_sys->i_cur_rec,
728 subrec_type);*/
729 /* if it's a GOP header, make sure it's legal
730 * (if we have enough data) */
731 /* Some ty files don't have this bit set
732 * and it causes problems */
733 if (subrec_type == 0x0c && l_rec_size >= 6)
734 p_block_in->p_buffer[5] |= 0x08;
735 /* store the TY PTS if there is one */
736 if (subrec_type == 0x07) {
737 p_sys->l_last_ty_pts = rec_hdr->l_ty_pts;
738 /* should we use audio or video PTS? */
739 //p_sys->l_last_ty_pts_sync = p_sys->lastAudioPTS;
740 } else {
741 /* yes I know this is a cheap hack. It's the timestamp
742 used for display and skipping fwd/back, so it
743 doesn't have to be accurate to the millisecond.
744 I adjust it here by roughly one 1/30 sec. Yes it
745 will be slightly off for UK streams, but it's OK.
747 p_sys->l_last_ty_pts += 35000000;
748 //p_sys->l_last_ty_pts += 33366667;
750 /* set PTS for this block before we send */
751 if (p_sys->lastVideoPTS > VLC_TS_INVALID)
753 p_block_in->i_pts = p_sys->lastVideoPTS;
754 /* PTS gets used ONCE.
755 * Any subsequent frames we get BEFORE next PES
756 * header will have their PTS computed in the codec */
757 p_sys->lastVideoPTS = VLC_TS_INVALID;
761 /* Register the CC decoders when needed */
762 for( i = 0; i < 4; i++ )
764 static const vlc_fourcc_t fcc[4] = {
765 VLC_CODEC_EIA608_1,
766 VLC_CODEC_EIA608_2,
767 VLC_CODEC_EIA608_3,
768 VLC_CODEC_EIA608_4,
770 static const char *ppsz_description[4] = {
771 N_("Closed captions 1"),
772 N_("Closed captions 2"),
773 N_("Closed captions 3"),
774 N_("Closed captions 4"),
777 es_format_t fmt;
779 if( !p_sys->cc.pb_present[i] || p_sys->p_cc[i] )
780 continue;
782 es_format_Init( &fmt, SPU_ES, fcc[i] );
783 fmt.psz_description = strdup( vlc_gettext(ppsz_description[i]) );
784 fmt.i_group = TY_ES_GROUP;
785 p_sys->p_cc[i] = es_out_Add( p_demux->out, &fmt );
786 es_format_Clean( &fmt );
789 /* Send the CC data */
790 if( p_block_in->i_pts > VLC_TS_INVALID && p_sys->cc.i_data > 0 )
792 for( i = 0; i < 4; i++ )
794 if( p_sys->p_cc[i] )
796 block_t *p_cc = block_Alloc( p_sys->cc.i_data );
797 p_cc->i_flags |= BLOCK_FLAG_TYPE_I;
798 p_cc->i_pts = p_block_in->i_pts;
799 memcpy( p_cc->p_buffer, p_sys->cc.p_data, p_sys->cc.i_data );
801 es_out_Send( p_demux->out, p_sys->p_cc[i], p_cc );
804 cc_Flush( &p_sys->cc );
807 //msg_Dbg(p_demux, "sending rec %d as video type 0x%02x",
808 //p_sys->i_cur_rec, subrec_type);
809 es_out_Send(p_demux->out, p_sys->p_video, p_block_in);
810 return 0;
812 static int DemuxRecAudio( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
814 demux_sys_t *p_sys = p_demux->p_sys;
815 const int subrec_type = rec_hdr->subrec_type;
816 const long l_rec_size = rec_hdr->l_rec_size;
817 int esOffset1;
819 assert( rec_hdr->rec_type == 0xc0 );
820 if( !p_block_in )
821 return -1;
822 #if 0
823 int i;
824 fprintf( stderr, "Audio Packet Header " );
825 for( i = 0 ; i < 24 ; i++ )
826 fprintf( stderr, "%2.2x ", p_block_in->p_buffer[i] );
827 fprintf( stderr, "\n" );
828 #endif
830 if( subrec_type == 2 )
832 /* SA or DTiVo Audio Data, no PES (continued block)
833 * ================================================
836 /* continue PES if previous was incomplete */
837 if (p_sys->i_pes_buf_cnt > 0)
839 const int i_need = p_sys->i_Pes_Length - p_sys->i_pes_buf_cnt;
841 msg_Dbg(p_demux, "continuing PES header");
842 /* do we have enough data to complete? */
843 if (i_need >= l_rec_size)
845 /* don't have complete PES hdr; save what we have and return */
846 memcpy(&p_sys->pes_buffer[p_sys->i_pes_buf_cnt],
847 p_block_in->p_buffer, l_rec_size);
848 p_sys->i_pes_buf_cnt += l_rec_size;
849 /* */
850 block_Release(p_block_in);
851 return 0;
854 /* we have enough; reconstruct this p_frame with the new hdr */
855 memcpy(&p_sys->pes_buffer[p_sys->i_pes_buf_cnt],
856 p_block_in->p_buffer, i_need);
857 /* advance the block past the PES header (don't want to send it) */
858 p_block_in->p_buffer += i_need;
859 p_block_in->i_buffer -= i_need;
860 /* get the PTS out of this PES header (MPEG or AC3) */
861 if (p_sys->audio_type == TIVO_AUDIO_MPEG)
862 esOffset1 = find_es_header(ty_MPEGAudioPacket,
863 p_sys->pes_buffer, 5);
864 else
865 esOffset1 = find_es_header(ty_AC3AudioPacket,
866 p_sys->pes_buffer, 5);
867 if (esOffset1 < 0)
869 /* god help us; something's really wrong */
870 msg_Err(p_demux, "can't find audio PES header in packet");
872 else
874 p_sys->lastAudioPTS = VLC_TS_0 + get_pts(
875 &p_sys->pes_buffer[ esOffset1 + p_sys->i_Pts_Offset ] );
876 p_block_in->i_pts = p_sys->lastAudioPTS;
878 p_sys->i_pes_buf_cnt = 0;
880 /* S2 DTivo has AC3 packets with 2 padding bytes at end. This is
881 * not allowed in the AC3 spec and will cause problems. So here
882 * we try to trim things. */
883 /* Also, S1 DTivo has alternating short / long AC3 packets. That
884 * is, one packet is short (incomplete) and the next packet has
885 * the first one's missing data, plus all of its own. Strange. */
886 if (p_sys->audio_type == TIVO_AUDIO_AC3 &&
887 p_sys->tivo_series == TIVO_SERIES2) {
888 if (p_sys->l_ac3_pkt_size + p_block_in->i_buffer >
889 AC3_PKT_LENGTH) {
890 p_block_in->i_buffer -= 2;
891 p_sys->l_ac3_pkt_size = 0;
892 } else {
893 p_sys->l_ac3_pkt_size += p_block_in->i_buffer;
897 else if( subrec_type == 0x03 )
899 /* MPEG Audio with PES Header, either SA or DTiVo */
900 /* ================================================ */
901 esOffset1 = find_es_header( ty_MPEGAudioPacket,
902 p_block_in->p_buffer, 5 );
904 /*msg_Dbg(p_demux, "buffer has %#02x %#02x %#02x %#02x",
905 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
906 p_block_in->p_buffer[2], p_block_in->p_buffer[3]);
907 msg_Dbg(p_demux, "audio ES hdr at offset %d", esOffset1);*/
909 /* SA PES Header, No Audio Data */
910 /* ================================================ */
911 if ( ( esOffset1 == 0 ) && ( l_rec_size == 16 ) )
913 p_sys->lastAudioPTS = VLC_TS_0 + get_pts( &p_block_in->p_buffer[
914 SA_PTS_OFFSET ] );
915 if (p_sys->firstAudioPTS < 0)
916 p_sys->firstAudioPTS = p_sys->lastAudioPTS;
918 block_Release(p_block_in);
919 return 0;
920 /*msg_Dbg(p_demux, "SA Audio PTS %"PRId64, p_sys->lastAudioPTS );*/
922 /* DTiVo Audio with PES Header */
923 /* ================================================ */
925 /* Check for complete PES */
926 if (check_sync_pes(p_demux, p_block_in, esOffset1,
927 l_rec_size) == -1)
929 /* partial PES header found, nothing else.
930 * we're done. */
931 block_Release(p_block_in);
932 return 0;
934 #if 0
935 msg_Dbg(p_demux, "packet buffer has "
936 "%02x %02x %02x %02x %02x %02x %02x %02x "
937 "%02x %02x %02x %02x %02x %02x %02x %02x",
938 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
939 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
940 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
941 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
942 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
943 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
944 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
945 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
946 #endif
948 else if( subrec_type == 0x04 )
950 /* SA Audio with no PES Header */
951 /* ================================================ */
952 /*msg_Dbg(p_demux,
953 "Adding SA Audio Packet Size %ld", l_rec_size ); */
955 if (p_sys->lastAudioPTS > VLC_TS_INVALID )
956 p_block_in->i_pts = p_sys->lastAudioPTS;
958 else if( subrec_type == 0x09 )
960 /* DTiVo AC3 Audio Data with PES Header */
961 /* ================================================ */
962 esOffset1 = find_es_header( ty_AC3AudioPacket,
963 p_block_in->p_buffer, 5 );
965 #if 0
966 msg_Dbg(p_demux, "buffer has "
967 "%02x %02x %02x %02x %02x %02x %02x %02x "
968 "%02x %02x %02x %02x %02x %02x %02x %02x",
969 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
970 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
971 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
972 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
973 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
974 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
975 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
976 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
977 msg_Dbg(p_demux, "audio ES AC3 hdr at offset %d", esOffset1);
978 #endif
980 /* Check for complete PES */
981 if (check_sync_pes(p_demux, p_block_in, esOffset1,
982 l_rec_size) == -1)
984 /* partial PES header found, nothing else. we're done. */
985 block_Release(p_block_in);
986 return 0;
988 /* S2 DTivo has invalid long AC3 packets */
989 if (p_sys->tivo_series == TIVO_SERIES2) {
990 if (p_block_in->i_buffer > AC3_PKT_LENGTH) {
991 p_block_in->i_buffer -= 2;
992 p_sys->l_ac3_pkt_size = 0;
993 } else {
994 p_sys->l_ac3_pkt_size = p_block_in->i_buffer;
998 else
1000 /* Unsupported/Unknown */
1001 block_Release(p_block_in);
1002 return 0;
1005 /* set PCR before we send (if PTS found) */
1006 if( p_block_in->i_pts > VLC_TS_INVALID )
1007 es_out_Control( p_demux->out, ES_OUT_SET_PCR,
1008 p_block_in->i_pts );
1009 /* Send data */
1010 es_out_Send( p_demux->out, p_sys->p_audio, p_block_in );
1011 return 0;
1014 static int DemuxRecCc( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
1016 demux_sys_t *p_sys = p_demux->p_sys;
1017 int i_field;
1019 if( p_block_in )
1020 block_Release(p_block_in);
1022 if( rec_hdr->rec_type == 0x01 )
1023 i_field = 0;
1024 else if( rec_hdr->rec_type == 0x02 )
1025 i_field = 1;
1026 else
1027 return 0;
1029 /* XDS data (extract programs infos) transmitted on field 2 only */
1030 if( i_field == 1 )
1031 DemuxDecodeXds( p_demux, rec_hdr->ex[0], rec_hdr->ex[1] );
1033 if( p_sys->cc.i_data + 3 > CC_MAX_DATA_SIZE )
1034 return 0;
1036 cc_AppendData( &p_sys->cc, CC_PKT_BYTE0(i_field), rec_hdr->ex );
1037 return 0;
1040 /* seek to a position within the stream, if possible */
1041 static int ty_stream_seek_pct(demux_t *p_demux, double seek_pct)
1043 demux_sys_t *p_sys = p_demux->p_sys;
1044 int64_t seek_pos = p_sys->i_stream_size * seek_pct;
1045 uint64_t l_skip_amt;
1046 unsigned i_cur_part;
1048 /* if we're not seekable, there's nothing to do */
1049 if (!p_sys->b_seekable)
1050 return VLC_EGENERIC;
1052 /* figure out which part & chunk we want & go there */
1053 i_cur_part = seek_pos / TIVO_PART_LENGTH;
1054 p_sys->i_cur_chunk = seek_pos / CHUNK_SIZE;
1056 /* try to read the part header (master chunk) if it's there */
1057 if ( vlc_stream_Seek( p_demux->s, i_cur_part * TIVO_PART_LENGTH ))
1059 /* can't seek stream */
1060 return VLC_EGENERIC;
1062 parse_master(p_demux);
1064 /* now for the actual chunk */
1065 if ( vlc_stream_Seek( p_demux->s, p_sys->i_cur_chunk * CHUNK_SIZE))
1067 /* can't seek stream */
1068 return VLC_EGENERIC;
1070 /* load the chunk */
1071 p_sys->i_stuff_cnt = 0;
1072 get_chunk_header(p_demux);
1074 /* seek within the chunk to get roughly to where we want */
1075 p_sys->i_cur_rec = (int)
1076 ((double) ((seek_pos % CHUNK_SIZE) / (double) (CHUNK_SIZE)) * p_sys->i_num_recs);
1077 msg_Dbg(p_demux, "Seeked to file pos %"PRId64, seek_pos);
1078 msg_Dbg(p_demux, " (chunk %d, record %d)",
1079 p_sys->i_cur_chunk - 1, p_sys->i_cur_rec);
1081 /* seek to the start of this record's data.
1082 * to do that, we have to skip past all prior records */
1083 l_skip_amt = 0;
1084 for ( int i=0; i<p_sys->i_cur_rec; i++)
1085 l_skip_amt += p_sys->rec_hdrs[i].l_rec_size;
1086 vlc_stream_Seek(p_demux->s, ((p_sys->i_cur_chunk-1) * CHUNK_SIZE) +
1087 (p_sys->i_num_recs * 16) + l_skip_amt + 4);
1089 /* to hell with syncing any audio or video, just start reading records... :) */
1090 /*p_sys->lastAudioPTS = p_sys->lastVideoPTS = VLC_TS_INVALID;*/
1091 return VLC_SUCCESS;
1094 /* XDS decoder */
1095 //#define TY_XDS_DEBUG
1096 static void XdsInit( xds_t *h )
1098 h->b_xds = false;
1099 h->i_class = XDS_MAX_CLASS_COUNT;
1100 h->i_type = 0;
1101 h->b_future = false;
1102 for( int i = 0; i < XDS_MAX_CLASS_COUNT; i++ )
1104 for( int j = 0; j < 128; j++ )
1105 h->pkt[i][j].b_started = false;
1107 h->b_meta_changed = false;
1108 memset( &h->meta, 0, sizeof(h->meta) );
1110 static void XdsExit( xds_t *h )
1112 /* */
1113 free( h->meta.psz_channel_name );
1114 free( h->meta.psz_channel_call_letter );
1115 free( h->meta.psz_channel_number );
1117 /* */
1118 free( h->meta.current.psz_name );
1119 free( h->meta.current.psz_rating );
1120 /* */
1121 free( h->meta.future.psz_name );
1122 free( h->meta.future.psz_rating );
1124 static void XdsStringUtf8( char dst[2*32+1], const uint8_t *p_src, int i_src )
1126 int i_dst = 0;
1127 for( int i = 0; i < i_src; i++ )
1129 switch( p_src[i] )
1131 #define E2( c, u1, u2 ) case c: dst[i_dst++] = u1; dst[i_dst++] = u2; break
1132 E2( 0x2a, 0xc3,0xa1); // lowercase a, acute accent
1133 E2( 0x5c, 0xc3,0xa9); // lowercase e, acute accent
1134 E2( 0x5e, 0xc3,0xad); // lowercase i, acute accent
1135 E2( 0x5f, 0xc3,0xb3); // lowercase o, acute accent
1136 E2( 0x60, 0xc3,0xba); // lowercase u, acute accent
1137 E2( 0x7b, 0xc3,0xa7); // lowercase c with cedilla
1138 E2( 0x7c, 0xc3,0xb7); // division symbol
1139 E2( 0x7d, 0xc3,0x91); // uppercase N tilde
1140 E2( 0x7e, 0xc3,0xb1); // lowercase n tilde
1141 #undef E2
1142 default:
1143 dst[i_dst++] = p_src[i];
1144 break;
1147 dst[i_dst++] = '\0';
1149 static bool XdsChangeString( xds_t *h, char **ppsz_dst, const char *psz_new )
1151 if( *ppsz_dst && psz_new && !strcmp( *ppsz_dst, psz_new ) )
1152 return false;
1153 if( *ppsz_dst == NULL && psz_new == NULL )
1154 return false;
1156 free( *ppsz_dst );
1157 if( psz_new )
1158 *ppsz_dst = strdup( psz_new );
1159 else
1160 *ppsz_dst = NULL;
1162 h->b_meta_changed = true;
1163 return true;
1166 static void XdsDecodeCurrentFuture( xds_t *h, xds_packet_t *pk )
1168 xds_meta_program_t *p_prg = h->b_future ? &h->meta.future : &h->meta.current;
1169 char name[2*32+1];
1170 int i_rating;
1172 switch( h->i_type )
1174 case 0x03:
1175 XdsStringUtf8( name, pk->p_data, pk->i_data );
1176 if( XdsChangeString( h, &p_prg->psz_name, name ) )
1178 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Program Name) %d'\n", pk->i_data );
1179 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> program name %s\n", name );
1181 break;
1182 case 0x05:
1183 i_rating = (pk->p_data[0] & 0x18);
1184 if( i_rating == 0x08 )
1186 /* TPG */
1187 static const char *pppsz_ratings[8][2] = {
1188 { "None", "No rating (no content advisory)" },
1189 { "TV-Y", "All Children (no content advisory)" },
1190 { "TV-Y7", "Directed to Older Children (V = Fantasy Violence)" },
1191 { "TV-G", "General Audience (no content advisory)" },
1192 { "TV-PG", "Parental Guidance Suggested" },
1193 { "TV-14", "Parents Strongly Cautioned" },
1194 { "TV-MA", "Mature Audience Only" },
1195 { "None", "No rating (no content advisory)" }
1197 p_prg->rating = XDS_META_PROGRAM_RATING_TPG;
1198 if( XdsChangeString( h, &p_prg->psz_rating, pppsz_ratings[pk->p_data[1]&0x07][0] ) )
1200 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1201 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> TPG Rating %s (%s)\n",
1202 // pppsz_ratings[pk->p_data[1]&0x07][0], pppsz_ratings[pk->p_data[1]&0x07][1] );
1205 else if( i_rating == 0x00 || i_rating == 0x10 )
1207 /* MPAA */
1208 static const char *pppsz_ratings[8][2] = {
1209 { "N/A", "N/A" },
1210 { "G", "General Audiences" },
1211 { "PG", "Parental Guidance Suggested" },
1212 { "PG-13", "Parents Strongly Cautioned" },
1213 { "R", "Restricted" },
1214 { "NC-17", "No one 17 and under admitted" },
1215 { "X", "No one under 17 admitted" },
1216 { "NR", "Not Rated" },
1218 p_prg->rating = XDS_META_PROGRAM_RATING_MPAA;
1219 if( XdsChangeString( h, &p_prg->psz_rating, pppsz_ratings[pk->p_data[0]&0x07][0] ) )
1221 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1222 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> TPG Rating %s (%s)\n",
1223 // pppsz_ratings[pk->p_data[0]&0x07][0], pppsz_ratings[pk->p_data[0]&0x07][1] );
1226 else
1228 /* Non US Rating TODO */
1229 assert( i_rating == 0x18 ); // only left value possible */
1230 p_prg->rating = XDS_META_PROGRAM_RATING_NONE;
1231 if( XdsChangeString( h, &p_prg->psz_rating, NULL ) )
1233 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1234 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> 0x%2.2x 0x%2.2x\n", pk->p_data[0], pk->p_data[1] );
1237 break;
1239 default:
1240 #ifdef TY_XDS_DEBUG
1241 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Unknown 0x%x)'\n", h->i_type );
1242 #endif
1243 break;
1247 static void XdsDecodeChannel( xds_t *h, xds_packet_t *pk )
1249 char name[2*32+1];
1250 char chan[2*32+1];
1252 switch( h->i_type )
1254 case 0x01:
1255 if( pk->i_data < 2 )
1256 return;
1257 XdsStringUtf8( name, pk->p_data, pk->i_data );
1258 if( XdsChangeString( h, &h->meta.psz_channel_name, name ) )
1260 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Name) %d'\n", pk->i_data );
1261 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> %s\n", name );
1263 break;
1265 case 0x02:
1266 if( pk->i_data < 4 )
1267 return;
1269 XdsStringUtf8( name, pk->p_data, 4 );
1270 if( XdsChangeString( h, &h->meta.psz_channel_call_letter, name ) )
1272 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1273 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> call letter %s\n", name );
1275 if( pk->i_data >= 6 )
1277 XdsStringUtf8( chan, &pk->p_data[4], 2 );
1278 if( XdsChangeString( h, &h->meta.psz_channel_number, chan ) )
1280 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1281 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> channel number %s\n", chan );
1284 else
1286 if( XdsChangeString( h, &h->meta.psz_channel_number, NULL ) )
1288 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1289 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> no channel number letter anymore\n" );
1292 break;
1293 case 0x03:
1294 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Channel Tape Delay)'\n" );
1295 break;
1296 case 0x04:
1297 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Transmission Signal Identifier)'\n" );
1298 break;
1299 default:
1300 #ifdef TY_XDS_DEBUG
1301 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Unknown 0x%x)'\n", h->i_type );
1302 #endif
1303 break;
1307 static void XdsDecode( xds_t *h, xds_packet_t *pk )
1309 switch( h->i_class )
1311 case XDS_CLASS_CURRENT:
1312 case XDS_CLASS_FUTURE:
1313 XdsDecodeCurrentFuture( h, pk );
1314 break;
1315 case XDS_CLASS_CHANNEL:
1316 XdsDecodeChannel( h, pk );
1317 break;
1318 case XDS_CLASS_MISCELLANEOUS:
1319 #ifdef TY_XDS_DEBUG
1320 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Miscellaneous'\n" );
1321 #endif
1322 break;
1323 case XDS_CLASS_PUBLIC_SERVICE:
1324 #ifdef TY_XDS_DEBUG
1325 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Public Service'\n" );
1326 #endif
1327 break;
1328 default:
1329 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: unknown class\n" );
1330 break;
1334 static void XdsParse( xds_t *h, uint8_t d1, uint8_t d2 )
1336 /* TODO check parity */
1337 d1 &= 0x7f;
1338 d2 &= 0x7f;
1340 /* */
1341 if( d1 >= 0x01 && d1 <= 0x0e )
1343 const xds_class_t i_class = ( d1 - 1 ) >> 1;
1344 const int i_type = d2;
1345 const bool b_start = d1 & 0x01;
1346 xds_packet_t *pk = &h->pkt[i_class][i_type];
1348 if( !b_start && !pk->b_started )
1350 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS Continuying a non started packet, ignoring\n" );
1351 h->b_xds = false;
1352 return;
1355 h->b_xds = true;
1356 h->i_class = i_class;
1357 h->i_type = i_type;
1358 h->b_future = !b_start;
1359 pk->b_started = true;
1360 if( b_start )
1362 pk->i_data = 0;
1363 pk->i_sum = d1 + d2;
1366 else if( d1 == 0x0f && h->b_xds )
1368 xds_packet_t *pk = &h->pkt[h->i_class][h->i_type];
1370 /* TODO checksum and decode */
1371 pk->i_sum += d1 + d2;
1372 if( pk->i_sum & 0x7f )
1374 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS invalid checksum, ignoring ---------------------------------\n" );
1375 pk->b_started = false;
1376 return;
1378 if( pk->i_data <= 0 )
1380 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS empty packet, ignoring ---------------------------------\n" );
1381 pk->b_started = false;
1382 return;
1385 //if( pk->p_data[pk->i_data-1] == 0x40 ) /* Padding byte */
1386 // pk->i_data--;
1387 XdsDecode( h, pk );
1389 /* Reset it */
1390 pk->b_started = false;
1392 else if( d1 >= 0x20 && h->b_xds )
1394 xds_packet_t *pk = &h->pkt[h->i_class][h->i_type];
1396 if( pk->i_data+2 > XDS_MAX_DATA_SIZE )
1398 /* Broken -> reinit */
1399 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS broken, reset\n" );
1400 h->b_xds = false;
1401 pk->b_started = false;
1402 return;
1404 /* TODO check parity bit */
1405 pk->p_data[pk->i_data++] = d1 & 0x7f;
1406 pk->p_data[pk->i_data++] = d2 & 0x7f;
1407 pk->i_sum += d1+d2;
1409 else
1411 h->b_xds = false;
1415 static void DemuxDecodeXds( demux_t *p_demux, uint8_t d1, uint8_t d2 )
1417 demux_sys_t *p_sys = p_demux->p_sys;
1419 XdsParse( &p_demux->p_sys->xds, d1, d2 );
1420 if( p_demux->p_sys->xds.b_meta_changed )
1422 xds_meta_t *m = &p_sys->xds.meta;
1423 vlc_meta_t *p_meta;
1425 /* Channel meta data */
1426 p_meta = vlc_meta_New();
1427 if( m->psz_channel_name )
1428 vlc_meta_SetPublisher( p_meta, m->psz_channel_name );
1429 if( m->psz_channel_call_letter )
1430 vlc_meta_SetTitle( p_meta, m->psz_channel_call_letter );
1431 if( m->psz_channel_number )
1432 vlc_meta_AddExtra( p_meta, "Channel number", m->psz_channel_number );
1433 es_out_Control( p_demux->out, ES_OUT_SET_GROUP_META, TY_ES_GROUP, p_meta );
1434 vlc_meta_Delete( p_meta );
1436 /* Event meta data (current/future) */
1437 if( m->current.psz_name )
1439 vlc_epg_t *p_epg = vlc_epg_New( TY_ES_GROUP, TY_ES_GROUP );
1440 if ( p_epg )
1442 vlc_epg_event_t *p_evt = vlc_epg_event_New( 0, 0, 0 );
1443 if ( p_evt )
1445 if( m->current.psz_name )
1446 p_evt->psz_name = strdup( m->current.psz_name );
1447 if( !vlc_epg_AddEvent( p_epg, p_evt ) )
1448 vlc_epg_event_Delete( p_evt );
1450 //if( m->current.psz_rating )
1451 // TODO but VLC cannot yet handle rating per epg event
1452 vlc_epg_SetCurrent( p_epg, 0 );
1454 if( m->future.psz_name )
1457 if( p_epg->i_event > 0 )
1458 es_out_Control( p_demux->out, ES_OUT_SET_GROUP_EPG,
1459 TY_ES_GROUP, p_epg );
1460 vlc_epg_Delete( p_epg );
1464 p_demux->p_sys->xds.b_meta_changed = false;
1467 /* seek to an exact time position within the stream, if possible.
1468 * l_seek_time is in nanoseconds, the TIVO time standard.
1470 static int ty_stream_seek_time(demux_t *p_demux, uint64_t l_seek_time)
1472 demux_sys_t *p_sys = p_demux->p_sys;
1473 unsigned i_seq_entry = 0;
1474 unsigned i;
1475 int i_skip_cnt;
1476 int64_t l_cur_pos = vlc_stream_Tell(p_demux->s);
1477 unsigned i_cur_part = l_cur_pos / TIVO_PART_LENGTH;
1478 uint64_t l_seek_secs = l_seek_time / 1000000000;
1479 uint64_t l_fwd_stamp = 1;
1481 /* if we're not seekable, there's nothing to do */
1482 if (!p_sys->b_seekable || !p_sys->b_have_master)
1483 return VLC_EGENERIC;
1485 msg_Dbg(p_demux, "Skipping to time %02"PRIu64":%02"PRIu64":%02"PRIu64,
1486 l_seek_secs / 3600, (l_seek_secs / 60) % 60, l_seek_secs % 60);
1488 /* seek to the proper segment if necessary */
1489 /* first see if we need to go back */
1490 while (l_seek_time < p_sys->l_first_ty_pts) {
1491 msg_Dbg(p_demux, "skipping to prior segment.");
1492 /* load previous part */
1493 if (i_cur_part == 0) {
1494 vlc_stream_Seek(p_demux->s, l_cur_pos);
1495 msg_Err(p_demux, "Attempt to seek past BOF");
1496 return VLC_EGENERIC;
1498 vlc_stream_Seek(p_demux->s, (i_cur_part - 1) * TIVO_PART_LENGTH);
1499 i_cur_part--;
1500 parse_master(p_demux);
1502 /* maybe we need to go forward */
1503 while (l_seek_time > p_sys->l_final_ty_pts) {
1504 msg_Dbg(p_demux, "skipping to next segment.");
1505 /* load next part */
1506 if ((i_cur_part + 1) * TIVO_PART_LENGTH > p_sys->i_stream_size) {
1507 /* error; restore previous file position */
1508 vlc_stream_Seek(p_demux->s, l_cur_pos);
1509 msg_Err(p_demux, "seek error");
1510 return VLC_EGENERIC;
1512 vlc_stream_Seek(p_demux->s, (i_cur_part + 1) * TIVO_PART_LENGTH);
1513 i_cur_part++;
1514 parse_master(p_demux);
1517 /* our target is somewhere within this part;
1518 find the proper chunk using seq_table */
1519 for (i=1; i<p_sys->i_seq_table_size; i++) {
1520 if (p_sys->seq_table[i].l_timestamp > l_seek_time) {
1521 /* i-1 is the section we want; remember the next timestamp in case
1522 we have to use it (this section may not have a proper SEQ hdr
1523 for the time we're seeking) */
1524 msg_Dbg(p_demux, "stopping at seq entry %d.", i);
1525 l_fwd_stamp = p_sys->seq_table[i].l_timestamp;
1526 i_seq_entry = i-1;
1527 break;
1531 /* if we went through the entire last loop and didn't find our target,
1532 then we skip to the next part. What has happened is that the actual
1533 time we're seeking is within this part, but there isn't a SEQ hdr
1534 for it here. So we skip to the next part */
1535 if (i == p_sys->i_seq_table_size) {
1536 if ((i_cur_part + 1) * TIVO_PART_LENGTH > p_sys->i_stream_size) {
1537 /* error; restore previous file position */
1538 vlc_stream_Seek(p_demux->s, l_cur_pos);
1539 msg_Err(p_demux, "seek error");
1540 return VLC_EGENERIC;
1542 vlc_stream_Seek(p_demux->s, (i_cur_part + 1) * TIVO_PART_LENGTH);
1543 i_cur_part++;
1544 parse_master(p_demux);
1545 i_seq_entry = 0;
1548 /* determine which chunk has our seek_time */
1549 for (i=0; i<p_sys->i_bits_per_seq_entry; i++) {
1550 uint64_t l_chunk_nr = i_seq_entry * p_sys->i_bits_per_seq_entry + i;
1551 uint64_t l_chunk_offset = (l_chunk_nr + 1) * CHUNK_SIZE;
1552 msg_Dbg(p_demux, "testing part %d chunk %"PRIu64" mask 0x%02X bit %d",
1553 i_cur_part, l_chunk_nr,
1554 p_sys->seq_table[i_seq_entry].chunk_bitmask[i/8], i%8);
1555 if (p_sys->seq_table[i_seq_entry].chunk_bitmask[i/8] & (1 << (i%8))) {
1556 /* check this chunk's SEQ header timestamp */
1557 msg_Dbg(p_demux, "has SEQ. seeking to chunk at 0x%"PRIu64,
1558 (i_cur_part * TIVO_PART_LENGTH) + l_chunk_offset);
1559 vlc_stream_Seek(p_demux->s, (i_cur_part * TIVO_PART_LENGTH) +
1560 l_chunk_offset);
1561 // TODO: we don't have to parse the full header set;
1562 // just test the seq_rec entry for its timestamp
1563 p_sys->i_stuff_cnt = 0;
1564 get_chunk_header(p_demux);
1565 // check ty PTS for the SEQ entry in this chunk
1566 if (p_sys->i_seq_rec < 0 || p_sys->i_seq_rec > p_sys->i_num_recs) {
1567 msg_Err(p_demux, "no SEQ hdr in chunk; table had one.");
1568 /* Seek to beginning of original chunk & reload it */
1569 vlc_stream_Seek(p_demux->s, (l_cur_pos / CHUNK_SIZE) * CHUNK_SIZE);
1570 p_sys->i_stuff_cnt = 0;
1571 get_chunk_header(p_demux);
1572 return VLC_EGENERIC;
1574 l_seek_secs = p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts /
1575 1000000000;
1576 msg_Dbg(p_demux, "found SEQ hdr for timestamp %02"PRIu64":%02"PRIu64":%02"PRIu64,
1577 l_seek_secs / 3600,
1578 (l_seek_secs / 60) % 60, l_seek_secs % 60);
1579 if (p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts >= l_seek_time) {
1580 // keep this one? go back?
1581 /* for now, we take this one. it's the first SEQ hdr AFTER
1582 the time we were searching for. */
1583 msg_Dbg(p_demux, "seek target found.");
1584 break;
1586 msg_Dbg(p_demux, "timestamp too early. still scanning.");
1589 /* if we made it through this entire loop without finding our target,
1590 then we skip to the next section. What has happened is that the actual
1591 time we're seeking is within this section, but there isn't a SEQ hdr
1592 for it here. So we skip to the next closest one (l_fwd_stamp) */
1593 if (i == p_sys->i_bits_per_seq_entry)
1594 return ty_stream_seek_time(p_demux, l_fwd_stamp);
1596 /* current stream ptr is at beginning of data for this chunk,
1597 so we need to skip past any stream data prior to the seq_rec
1598 in this chunk */
1599 i_skip_cnt = 0;
1600 for (int j=0; j<p_sys->i_seq_rec; j++)
1601 i_skip_cnt += p_sys->rec_hdrs[j].l_rec_size;
1602 vlc_stream_Read(p_demux->s, NULL, i_skip_cnt);
1603 p_sys->i_cur_rec = p_sys->i_seq_rec;
1604 //p_sys->l_last_ty_pts = p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts;
1605 //p_sys->l_last_ty_pts_sync = p_sys->lastAudioPTS;
1607 return VLC_SUCCESS;
1611 /* parse a master chunk, filling the SEQ table and other variables.
1612 * We assume the stream is currently pointing to it.
1614 static void parse_master(demux_t *p_demux)
1616 demux_sys_t *p_sys = p_demux->p_sys;
1617 uint8_t mst_buf[32];
1618 int64_t i_save_pos = vlc_stream_Tell(p_demux->s);
1619 int64_t i_pts_secs;
1621 /* Note that the entries in the SEQ table in the stream may have
1622 different sizes depending on the bits per entry. We store them
1623 all in the same size structure, so we have to parse them out one
1624 by one. If we had a dynamic structure, we could simply read the
1625 entire table directly from the stream into memory in place. */
1627 /* clear the SEQ table */
1628 free(p_sys->seq_table);
1630 /* parse header info */
1631 vlc_stream_Read(p_demux->s, mst_buf, 32);
1633 uint32_t i_map_size = U32_AT(&mst_buf[20]); /* size of bitmask, in bytes */
1634 uint32_t i = U32_AT(&mst_buf[28]); /* size of SEQ table, in bytes */
1636 p_sys->i_bits_per_seq_entry = i_map_size * 8;
1637 p_sys->i_seq_table_size = i / (8 + i_map_size);
1639 if(p_sys->i_seq_table_size == 0)
1641 p_sys->seq_table = NULL;
1642 return;
1645 /* parse all the entries */
1646 p_sys->seq_table = calloc(p_sys->i_seq_table_size, sizeof(ty_seq_table_t));
1647 if (p_sys->seq_table == NULL)
1649 p_sys->i_seq_table_size = 0;
1650 return;
1652 for (unsigned i=0; i<p_sys->i_seq_table_size; i++) {
1653 vlc_stream_Read(p_demux->s, mst_buf, 8);
1654 p_sys->seq_table[i].l_timestamp = U64_AT(&mst_buf[0]);
1655 if (i_map_size > 8) {
1656 msg_Err(p_demux, "Unsupported SEQ bitmap size in master chunk");
1657 vlc_stream_Read(p_demux->s, NULL, i_map_size);
1658 } else {
1659 vlc_stream_Read(p_demux->s, mst_buf + 8, i_map_size);
1660 memcpy(p_sys->seq_table[i].chunk_bitmask, &mst_buf[8], i_map_size);
1664 /* set up a few of our variables */
1665 p_sys->l_first_ty_pts = p_sys->seq_table[0].l_timestamp;
1666 p_sys->l_final_ty_pts =
1667 p_sys->seq_table[p_sys->i_seq_table_size - 1].l_timestamp;
1668 p_sys->b_have_master = true;
1670 i_pts_secs = p_sys->l_first_ty_pts / 1000000000;
1671 msg_Dbg( p_demux,
1672 "first TY pts in master is %02"PRId64":%02"PRId64":%02"PRId64,
1673 i_pts_secs / 3600, (i_pts_secs / 60) % 60, i_pts_secs % 60 );
1674 i_pts_secs = p_sys->l_final_ty_pts / 1000000000;
1675 msg_Dbg( p_demux,
1676 "final TY pts in master is %02"PRId64":%02"PRId64":%02"PRId64,
1677 i_pts_secs / 3600, (i_pts_secs / 60) % 60, i_pts_secs % 60 );
1679 /* seek past this chunk */
1680 vlc_stream_Seek(p_demux->s, i_save_pos + CHUNK_SIZE);
1684 /* ======================================================================== */
1685 /* "Peek" at some chunks. Skip over the Part header if we find it.
1686 * We parse the peeked data and determine audio type,
1687 * SA vs. DTivo, & Tivo Series.
1688 * Set global vars i_Pes_Length, i_Pts_Offset,
1689 * p_sys->tivo_series, p_sys->tivo_type, p_sys->audio_type */
1690 static int probe_stream(demux_t *p_demux)
1692 demux_sys_t *p_sys = p_demux->p_sys;
1693 const uint8_t *p_buf;
1694 int i;
1695 bool b_probe_error = false;
1697 /* we need CHUNK_PEEK_COUNT chunks of data, first one might be a Part header, so ... */
1698 if (vlc_stream_Peek( p_demux->s, &p_buf, CHUNK_PEEK_COUNT * CHUNK_SIZE ) <
1699 CHUNK_PEEK_COUNT * CHUNK_SIZE) {
1700 msg_Err(p_demux, "Can't peek %d chunks", CHUNK_PEEK_COUNT);
1701 /* TODO: if seekable, then loop reading chunks into a temp buffer */
1702 return VLC_EGENERIC;
1705 /* the real work: analyze this chunk */
1706 for (i = 0; i < CHUNK_PEEK_COUNT; i++) {
1707 analyze_chunk(p_demux, p_buf);
1708 if (p_sys->tivo_series != TIVO_SERIES_UNKNOWN &&
1709 p_sys->audio_type != TIVO_AUDIO_UNKNOWN &&
1710 p_sys->tivo_type != TIVO_TYPE_UNKNOWN)
1711 break;
1712 p_buf += CHUNK_SIZE;
1715 /* the final tally */
1716 if (p_sys->tivo_series == TIVO_SERIES_UNKNOWN) {
1717 msg_Err(p_demux, "Can't determine Tivo Series.");
1718 b_probe_error = true;
1720 if (p_sys->audio_type == TIVO_AUDIO_UNKNOWN) {
1721 msg_Err(p_demux, "Can't determine Tivo Audio Type.");
1722 b_probe_error = true;
1724 if (p_sys->tivo_type == TIVO_TYPE_UNKNOWN) {
1725 msg_Err(p_demux, "Can't determine Tivo Type (SA/DTivo).");
1726 b_probe_error = true;
1728 return b_probe_error?VLC_EGENERIC:VLC_SUCCESS;
1732 /* ======================================================================== */
1733 /* gather statistics for this chunk & set our tivo-type vars accordingly */
1734 static void analyze_chunk(demux_t *p_demux, const uint8_t *p_chunk)
1736 demux_sys_t *p_sys = p_demux->p_sys;
1737 int i_num_recs, i;
1738 ty_rec_hdr_t *p_hdrs;
1739 int i_num_6e0, i_num_be0, i_num_9c0, i_num_3c0;
1740 int i_payload_size;
1742 /* skip if it's a Part header */
1743 if( U32_AT( &p_chunk[ 0 ] ) == TIVO_PES_FILEID )
1744 return;
1746 /* number of records in chunk (we ignore high order byte;
1747 * rarely are there > 256 chunks & we don't need that many anyway) */
1748 i_num_recs = p_chunk[0];
1749 if (i_num_recs < 5) {
1750 /* try again with the next chunk. Sometimes there are dead ones */
1751 return;
1754 p_chunk += 4; /* skip past rec count & SEQ bytes */
1755 //msg_Dbg(p_demux, "probe: chunk has %d recs", i_num_recs);
1756 p_hdrs = parse_chunk_headers(p_chunk, i_num_recs, &i_payload_size);
1757 /* scan headers.
1758 * 1. check video packets. Presence of 0x6e0 means S1.
1759 * No 6e0 but have be0 means S2.
1760 * 2. probe for audio 0x9c0 vs 0x3c0 (AC3 vs Mpeg)
1761 * If AC-3, then we have DTivo.
1762 * If MPEG, search for PTS offset. This will determine SA vs. DTivo.
1764 i_num_6e0 = i_num_be0 = i_num_9c0 = i_num_3c0 = 0;
1765 for (i=0; i<i_num_recs; i++) {
1766 //msg_Dbg(p_demux, "probe: rec is %d/%d = 0x%04x", p_hdrs[i].subrec_type,
1767 //p_hdrs[i].rec_type,
1768 //p_hdrs[i].subrec_type << 8 | p_hdrs[i].rec_type);
1769 switch (p_hdrs[i].subrec_type << 8 | p_hdrs[i].rec_type) {
1770 case 0x6e0:
1771 i_num_6e0++;
1772 break;
1773 case 0xbe0:
1774 i_num_be0++;
1775 break;
1776 case 0x3c0:
1777 i_num_3c0++;
1778 break;
1779 case 0x9c0:
1780 i_num_9c0++;
1781 break;
1784 msg_Dbg(p_demux, "probe: chunk has %d 0x6e0 recs, %d 0xbe0 recs.",
1785 i_num_6e0, i_num_be0);
1787 /* set up our variables */
1788 if (i_num_6e0 > 0) {
1789 msg_Dbg(p_demux, "detected Series 1 Tivo");
1790 p_sys->tivo_series = TIVO_SERIES1;
1791 p_sys->i_Pes_Length = SERIES1_PES_LENGTH;
1792 } else if (i_num_be0 > 0) {
1793 msg_Dbg(p_demux, "detected Series 2 Tivo");
1794 p_sys->tivo_series = TIVO_SERIES2;
1795 p_sys->i_Pes_Length = SERIES2_PES_LENGTH;
1797 if (i_num_9c0 > 0) {
1798 msg_Dbg(p_demux, "detected AC-3 Audio (DTivo)" );
1799 p_sys->audio_type = TIVO_AUDIO_AC3;
1800 p_sys->tivo_type = TIVO_TYPE_DTIVO;
1801 p_sys->i_Pts_Offset = AC3_PTS_OFFSET;
1802 p_sys->i_Pes_Length = AC3_PES_LENGTH;
1803 } else if (i_num_3c0 > 0) {
1804 p_sys->audio_type = TIVO_AUDIO_MPEG;
1805 msg_Dbg(p_demux, "detected MPEG Audio" );
1808 /* if tivo_type still unknown, we can check PTS location
1809 * in MPEG packets to determine tivo_type */
1810 if (p_sys->tivo_type == TIVO_TYPE_UNKNOWN) {
1811 uint32_t i_data_offset = (16 * i_num_recs);
1812 for (i=0; i<i_num_recs; i++) {
1813 if ((p_hdrs[i].subrec_type << 0x08 | p_hdrs[i].rec_type) == 0x3c0 &&
1814 p_hdrs[i].l_rec_size > 15) {
1815 /* first make sure we're aligned */
1816 int i_pes_offset = find_es_header(ty_MPEGAudioPacket,
1817 &p_chunk[i_data_offset], 5);
1818 if (i_pes_offset >= 0) {
1819 /* pes found. on SA, PES has hdr data at offset 6, not PTS. */
1820 //msg_Dbg(p_demux, "probe: mpeg es header found in rec %d at offset %d",
1821 //i, i_pes_offset);
1822 if ((p_chunk[i_data_offset + 6 + i_pes_offset] & 0x80) == 0x80) {
1823 /* S1SA or S2(any) Mpeg Audio (PES hdr, not a PTS start) */
1824 if (p_sys->tivo_series == TIVO_SERIES1)
1825 msg_Dbg(p_demux, "detected Stand-Alone Tivo" );
1826 p_sys->tivo_type = TIVO_TYPE_SA;
1827 p_sys->i_Pts_Offset = SA_PTS_OFFSET;
1828 } else {
1829 if (p_sys->tivo_series == TIVO_SERIES1)
1830 msg_Dbg(p_demux, "detected DirecTV Tivo" );
1831 p_sys->tivo_type = TIVO_TYPE_DTIVO;
1832 p_sys->i_Pts_Offset = DTIVO_PTS_OFFSET;
1834 break;
1837 i_data_offset += p_hdrs[i].l_rec_size;
1840 free(p_hdrs);
1844 /* =========================================================================== */
1845 static int get_chunk_header(demux_t *p_demux)
1847 int i_readSize, i_num_recs;
1848 uint8_t *p_hdr_buf;
1849 const uint8_t *p_peek;
1850 demux_sys_t *p_sys = p_demux->p_sys;
1851 int i_payload_size; /* sum of all records' sizes */
1853 msg_Dbg(p_demux, "parsing ty chunk #%d", p_sys->i_cur_chunk );
1855 /* if we have left-over filler space from the last chunk, get that */
1856 if (p_sys->i_stuff_cnt > 0) {
1857 vlc_stream_Read( p_demux->s, NULL, p_sys->i_stuff_cnt);
1858 p_sys->i_stuff_cnt = 0;
1861 /* read the TY packet header */
1862 i_readSize = vlc_stream_Peek( p_demux->s, &p_peek, 4 );
1863 p_sys->i_cur_chunk++;
1865 if ( (i_readSize < 4) || ( U32_AT(&p_peek[ 0 ] ) == 0 ))
1867 /* EOF */
1868 p_sys->eof = 1;
1869 return 0;
1872 /* check if it's a PART Header */
1873 if( U32_AT( &p_peek[ 0 ] ) == TIVO_PES_FILEID )
1875 /* parse master chunk */
1876 parse_master(p_demux);
1877 return get_chunk_header(p_demux);
1880 /* number of records in chunk (8- or 16-bit number) */
1881 if (p_peek[3] & 0x80)
1883 /* 16 bit rec cnt */
1884 p_sys->i_num_recs = i_num_recs = (p_peek[1] << 8) + p_peek[0];
1885 p_sys->i_seq_rec = (p_peek[3] << 8) + p_peek[2];
1886 if (p_sys->i_seq_rec != 0xffff)
1888 p_sys->i_seq_rec &= ~0x8000;
1891 else
1893 /* 8 bit reclen - tivo 1.3 format */
1894 p_sys->i_num_recs = i_num_recs = p_peek[0];
1895 p_sys->i_seq_rec = p_peek[1];
1897 p_sys->i_cur_rec = 0;
1898 p_sys->b_first_chunk = false;
1900 /*msg_Dbg( p_demux, "chunk has %d records", i_num_recs );*/
1902 free(p_sys->rec_hdrs);
1903 p_sys->rec_hdrs = NULL;
1905 /* skip past the 4 bytes we "peeked" earlier */
1906 vlc_stream_Read( p_demux->s, NULL, 4 );
1908 /* read the record headers into a temp buffer */
1909 p_hdr_buf = xmalloc(i_num_recs * 16);
1910 if (vlc_stream_Read(p_demux->s, p_hdr_buf, i_num_recs * 16) < i_num_recs * 16) {
1911 free( p_hdr_buf );
1912 p_sys->eof = true;
1913 return 0;
1915 /* parse them */
1916 p_sys->rec_hdrs = parse_chunk_headers(p_hdr_buf, i_num_recs,
1917 &i_payload_size);
1918 free(p_hdr_buf);
1920 p_sys->i_stuff_cnt = CHUNK_SIZE - 4 -
1921 (p_sys->i_num_recs * 16) - i_payload_size;
1922 if (p_sys->i_stuff_cnt > 0)
1923 msg_Dbg( p_demux, "chunk has %d stuff bytes at end",
1924 p_sys->i_stuff_cnt );
1925 return 1;
1929 static ty_rec_hdr_t *parse_chunk_headers( const uint8_t *p_buf,
1930 int i_num_recs, int *pi_payload_size)
1932 int i;
1933 ty_rec_hdr_t *p_hdrs, *p_rec_hdr;
1935 *pi_payload_size = 0;
1936 p_hdrs = xmalloc(i_num_recs * sizeof(ty_rec_hdr_t));
1938 for (i = 0; i < i_num_recs; i++)
1940 const uint8_t *record_header = p_buf + (i * 16);
1941 p_rec_hdr = &p_hdrs[i]; /* for brevity */
1942 p_rec_hdr->rec_type = record_header[3];
1943 p_rec_hdr->subrec_type = record_header[2] & 0x0f;
1944 if ((record_header[ 0 ] & 0x80) == 0x80)
1946 uint8_t b1, b2;
1947 /* marker bit 2 set, so read extended data */
1948 b1 = ( ( ( record_header[ 0 ] & 0x0f ) << 4 ) |
1949 ( ( record_header[ 1 ] & 0xf0 ) >> 4 ) );
1950 b2 = ( ( ( record_header[ 1 ] & 0x0f ) << 4 ) |
1951 ( ( record_header[ 2 ] & 0xf0 ) >> 4 ) );
1953 p_rec_hdr->ex[0] = b1;
1954 p_rec_hdr->ex[1] = b2;
1955 p_rec_hdr->l_rec_size = 0;
1956 p_rec_hdr->l_ty_pts = 0;
1957 p_rec_hdr->b_ext = true;
1959 else
1961 p_rec_hdr->l_rec_size = ( record_header[ 0 ] << 8 |
1962 record_header[ 1 ] ) << 4 | ( record_header[ 2 ] >> 4 );
1963 *pi_payload_size += p_rec_hdr->l_rec_size;
1964 p_rec_hdr->b_ext = false;
1965 p_rec_hdr->l_ty_pts = U64_AT( &record_header[ 8 ] );
1967 //fprintf( stderr, "parse_chunk_headers[%d] t=0x%x s=%d\n", i, p_rec_hdr->rec_type, p_rec_hdr->subrec_type );
1968 } /* end of record-header loop */
1969 return p_hdrs;