skins2: fix nested panels wrongly positioned if not the first child
[vlc.git] / modules / demux / ty.c
blobf2f30c888dffd39e61f449b3fc98d021eaf4e94f
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( 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->b_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 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 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 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 = 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 if( p_rec->rec_type == 0xe0 )
461 /* Video */
462 DemuxRecVideo( p_demux, p_rec, p_block_in );
464 else if ( p_rec->rec_type == 0xc0 )
466 /* Audio */
467 DemuxRecAudio( p_demux, p_rec, p_block_in );
469 else if( p_rec->rec_type == 0x01 || p_rec->rec_type == 0x02 )
471 /* Closed Captions/XDS */
472 DemuxRecCc( p_demux, p_rec, p_block_in );
474 else if ( p_rec->rec_type == 0x03 )
476 /* Tivo data services (e.g. "thumbs-up to record!") useless for us */
477 if( p_block_in )
478 block_Release(p_block_in);
480 else if ( p_rec->rec_type == 0x05 )
482 /* Unknown, but seen regularly */
483 if( p_block_in )
484 block_Release(p_block_in);
486 else
488 msg_Dbg(p_demux, "Invalid record type 0x%02x", p_rec->rec_type );
489 if( p_block_in )
490 block_Release(p_block_in);
493 /* */
494 p_sys->i_cur_rec++;
495 return 1;
498 /* Control */
499 static int Control(demux_t *p_demux, int i_query, va_list args)
501 demux_sys_t *p_sys = p_demux->p_sys;
502 double f, *pf;
503 int64_t i64, *p_i64;
505 /*msg_Info(p_demux, "control cmd %d", i_query);*/
506 switch( i_query )
508 case DEMUX_GET_POSITION:
509 /* arg is 0.0 - 1.0 percent of overall file position */
510 if( ( i64 = p_sys->i_stream_size ) > 0 )
512 pf = (double*) va_arg( args, double* );
513 *pf = ((double)1.0) * stream_Tell( p_demux->s ) / (double) i64;
514 return VLC_SUCCESS;
516 return VLC_EGENERIC;
518 case DEMUX_SET_POSITION:
519 /* arg is 0.0 - 1.0 percent of overall file position */
520 f = (double) va_arg( args, double );
521 /* msg_Dbg(p_demux, "Control - set position to %2.3f", f); */
522 if ((i64 = p_sys->i_stream_size) > 0)
523 return ty_stream_seek_pct(p_demux, f);
524 return VLC_EGENERIC;
525 case DEMUX_GET_TIME:
526 /* return TiVo timestamp */
527 p_i64 = (int64_t *) va_arg(args, int64_t *);
528 //*p_i64 = p_sys->lastAudioPTS - p_sys->firstAudioPTS;
529 //*p_i64 = (p_sys->l_last_ty_pts / 1000) + (p_sys->lastAudioPTS -
530 // p_sys->l_last_ty_pts_sync);
531 *p_i64 = (p_sys->l_last_ty_pts / 1000);
532 return VLC_SUCCESS;
533 case DEMUX_GET_LENGTH: /* length of program in microseconds, 0 if unk */
534 /* size / bitrate */
535 p_i64 = (int64_t *) va_arg(args, int64_t *);
536 *p_i64 = 0;
537 return VLC_SUCCESS;
538 case DEMUX_SET_TIME: /* arg is time in microsecs */
539 i64 = (int64_t) va_arg( args, int64_t );
540 return ty_stream_seek_time(p_demux, i64 * 1000);
541 case DEMUX_GET_FPS:
542 default:
543 return VLC_EGENERIC;
547 /* Close */
548 static void Close( vlc_object_t *p_this )
550 demux_t *p_demux = (demux_t*)p_this;
551 demux_sys_t *p_sys = p_demux->p_sys;
553 XdsExit( &p_sys->xds );
554 cc_Exit( &p_sys->cc );
555 free( p_sys->rec_hdrs );
556 free( p_sys->seq_table );
557 free(p_sys);
561 /* =========================================================================== */
562 /* Compute Presentation Time Stamp (PTS)
563 * Assume buf points to beginning of PTS */
564 static mtime_t get_pts( const uint8_t *buf )
566 mtime_t i_pts;
568 i_pts = ((mtime_t)(buf[0]&0x0e ) << 29)|
569 (mtime_t)(buf[1] << 22)|
570 ((mtime_t)(buf[2]&0xfe) << 14)|
571 (mtime_t)(buf[3] << 7)|
572 (mtime_t)(buf[4] >> 1);
573 i_pts *= 100 / 9; /* convert PTS (90Khz clock) to microseconds */
574 return i_pts;
578 /* =========================================================================== */
579 static int find_es_header( const uint8_t *header,
580 const uint8_t *buffer, int i_search_len )
582 int count;
584 for( count = 0; count < i_search_len; count++ )
586 if( !memcmp( &buffer[count], header, 4 ) )
587 return count;
589 return -1;
593 /* =========================================================================== */
594 /* check if we have a full PES header, if not, then save what we have.
595 * this is called when audio-start packets are encountered.
596 * Returns:
597 * 1 partial PES hdr found, some audio data found (buffer adjusted),
598 * -1 partial PES hdr found, no audio data found
599 * 0 otherwise (complete PES found, pts extracted, pts set, buffer adjusted) */
600 /* TODO: HD support -- nothing known about those streams */
601 static int check_sync_pes( demux_t *p_demux, block_t *p_block,
602 int32_t offset, int32_t rec_len )
604 demux_sys_t *p_sys = p_demux->p_sys;
606 if ( offset < 0 || offset + p_sys->i_Pes_Length > rec_len )
608 /* entire PES header not present */
609 msg_Dbg( p_demux, "PES header at %d not complete in record. storing.",
610 offset );
611 /* save the partial pes header */
612 if( offset < 0 )
614 /* no header found, fake some 00's (this works, believe me) */
615 memset( p_sys->pes_buffer, 0, 4 );
616 p_sys->i_pes_buf_cnt = 4;
617 if( rec_len > 4 )
618 msg_Err( p_demux, "PES header not found in record of %d bytes!",
619 rec_len );
620 return -1;
622 /* copy the partial pes header we found */
623 memcpy( p_sys->pes_buffer, p_block->p_buffer + offset,
624 rec_len - offset );
625 p_sys->i_pes_buf_cnt = rec_len - offset;
627 if( offset > 0 )
629 /* PES Header was found, but not complete, so trim the end of this record */
630 p_block->i_buffer -= rec_len - offset;
631 return 1;
633 return -1; /* partial PES, no audio data */
635 /* full PES header present, extract PTS */
636 p_sys->lastAudioPTS = VLC_TS_0 + get_pts( &p_block->p_buffer[ offset +
637 p_sys->i_Pts_Offset ] );
638 if (p_sys->firstAudioPTS < 0)
639 p_sys->firstAudioPTS = p_sys->lastAudioPTS;
640 p_block->i_pts = p_sys->lastAudioPTS;
641 /*msg_Dbg(p_demux, "Audio PTS %"PRId64, p_sys->lastAudioPTS );*/
642 /* adjust audio record to remove PES header */
643 memmove(p_block->p_buffer + offset, p_block->p_buffer + offset +
644 p_sys->i_Pes_Length, rec_len - p_sys->i_Pes_Length);
645 p_block->i_buffer -= p_sys->i_Pes_Length;
646 #if 0
647 msg_Dbg(p_demux, "pes hdr removed; buffer len=%d and has "
648 "%02x %02x %02x %02x %02x %02x %02x %02x "
649 "%02x %02x %02x %02x %02x %02x %02x %02x", p_block->i_buffer,
650 p_block->p_buffer[0], p_block->p_buffer[1],
651 p_block->p_buffer[2], p_block->p_buffer[3],
652 p_block->p_buffer[4], p_block->p_buffer[5],
653 p_block->p_buffer[6], p_block->p_buffer[7],
654 p_block->p_buffer[8], p_block->p_buffer[9],
655 p_block->p_buffer[10], p_block->p_buffer[11],
656 p_block->p_buffer[12], p_block->p_buffer[13],
657 p_block->p_buffer[14], p_block->p_buffer[15]);
658 #endif
659 return 0;
662 static int DemuxRecVideo( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
664 demux_sys_t *p_sys = p_demux->p_sys;
665 const int subrec_type = rec_hdr->subrec_type;
666 const long l_rec_size = rec_hdr->l_rec_size; // p_block_in->i_buffer might be better
667 int esOffset1;
668 int i;
670 assert( rec_hdr->rec_type == 0xe0 );
671 if( !p_block_in )
672 return -1;
674 #if 0
675 msg_Dbg(p_demux, "packet buffer has "
676 "%02x %02x %02x %02x %02x %02x %02x %02x "
677 "%02x %02x %02x %02x %02x %02x %02x %02x",
678 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
679 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
680 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
681 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
682 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
683 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
684 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
685 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
686 #endif
687 //if( subrec_type == 0x06 || subrec_type == 0x07 )
688 if( subrec_type != 0x02 && subrec_type != 0x0c &&
689 subrec_type != 0x08 && l_rec_size > 4 )
691 /* get the PTS from this packet if it has one.
692 * on S1, only 0x06 has PES. On S2, however, most all do.
693 * Do NOT Pass the PES Header to the MPEG2 codec */
694 esOffset1 = find_es_header( ty_VideoPacket, p_block_in->p_buffer, 5 );
695 if( esOffset1 != -1 )
697 //msg_Dbg(p_demux, "Video PES hdr in pkt type 0x%02x at offset %d",
698 //subrec_type, esOffset1);
699 p_sys->lastVideoPTS = VLC_TS_0 + get_pts(
700 &p_block_in->p_buffer[ esOffset1 + VIDEO_PTS_OFFSET ] );
701 /*msg_Dbg(p_demux, "Video rec %d PTS %"PRId64, p_sys->i_cur_rec,
702 p_sys->lastVideoPTS );*/
703 if (subrec_type != 0x06) {
704 /* if we found a PES, and it's not type 6, then we're S2 */
705 /* The packet will have video data (& other headers) so we
706 * chop out the PES header and send the rest */
707 if (l_rec_size >= VIDEO_PES_LENGTH) {
708 p_block_in->p_buffer += VIDEO_PES_LENGTH + esOffset1;
709 p_block_in->i_buffer -= VIDEO_PES_LENGTH + esOffset1;
710 } else {
711 msg_Dbg(p_demux, "video rec type 0x%02x has short PES"
712 " (%ld bytes)", subrec_type, l_rec_size);
713 /* nuke this block; it's too short, but has PES marker */
714 p_block_in->i_buffer = 0;
717 }/* else
718 msg_Dbg(p_demux, "No Video PES hdr in pkt type 0x%02x",
719 subrec_type); */
722 if(subrec_type == 0x06 )
724 /* type 6 (S1 DTivo) has no data, so we're done */
725 block_Release(p_block_in);
726 return 0;
729 /* if it's not a continue blk, then set PTS */
730 if( subrec_type != 0x02 )
732 /*msg_Dbg(p_demux, "Video rec %d type 0x%02X", p_sys->i_cur_rec,
733 subrec_type);*/
734 /* if it's a GOP header, make sure it's legal
735 * (if we have enough data) */
736 /* Some ty files don't have this bit set
737 * and it causes problems */
738 if (subrec_type == 0x0c && l_rec_size >= 6)
739 p_block_in->p_buffer[5] |= 0x08;
740 /* store the TY PTS if there is one */
741 if (subrec_type == 0x07) {
742 p_sys->l_last_ty_pts = rec_hdr->l_ty_pts;
743 /* should we use audio or video PTS? */
744 //p_sys->l_last_ty_pts_sync = p_sys->lastAudioPTS;
745 } else {
746 /* yes I know this is a cheap hack. It's the timestamp
747 used for display and skipping fwd/back, so it
748 doesn't have to be accurate to the millisecond.
749 I adjust it here by roughly one 1/30 sec. Yes it
750 will be slightly off for UK streams, but it's OK.
752 p_sys->l_last_ty_pts += 35000000;
753 //p_sys->l_last_ty_pts += 33366667;
755 /* set PTS for this block before we send */
756 if (p_sys->lastVideoPTS > VLC_TS_INVALID)
758 p_block_in->i_pts = p_sys->lastVideoPTS;
759 /* PTS gets used ONCE.
760 * Any subsequent frames we get BEFORE next PES
761 * header will have their PTS computed in the codec */
762 p_sys->lastVideoPTS = VLC_TS_INVALID;
766 /* Register the CC decoders when needed */
767 for( i = 0; i < 4; i++ )
769 static const vlc_fourcc_t fcc[4] = {
770 VLC_FOURCC('c', 'c', '1', ' '),
771 VLC_FOURCC('c', 'c', '2', ' '),
772 VLC_FOURCC('c', 'c', '3', ' '),
773 VLC_FOURCC('c', 'c', '4', ' ')
775 static const char *ppsz_description[4] = {
776 N_("Closed captions 1"),
777 N_("Closed captions 2"),
778 N_("Closed captions 3"),
779 N_("Closed captions 4"),
782 es_format_t fmt;
784 if( !p_sys->cc.pb_present[i] || p_sys->p_cc[i] )
785 continue;
787 es_format_Init( &fmt, SPU_ES, fcc[i] );
788 fmt.psz_description = strdup( vlc_gettext(ppsz_description[i]) );
789 fmt.i_group = TY_ES_GROUP;
790 p_sys->p_cc[i] = es_out_Add( p_demux->out, &fmt );
791 es_format_Clean( &fmt );
794 /* Send the CC data */
795 if( p_block_in->i_pts > VLC_TS_INVALID && p_sys->cc.i_data > 0 )
797 for( i = 0; i < 4; i++ )
799 if( p_sys->p_cc[i] )
801 block_t *p_cc = block_Alloc( p_sys->cc.i_data );
802 p_cc->i_flags |= BLOCK_FLAG_TYPE_I;
803 p_cc->i_pts = p_block_in->i_pts;
804 memcpy( p_cc->p_buffer, p_sys->cc.p_data, p_sys->cc.i_data );
806 es_out_Send( p_demux->out, p_sys->p_cc[i], p_cc );
809 cc_Flush( &p_sys->cc );
812 //msg_Dbg(p_demux, "sending rec %d as video type 0x%02x",
813 //p_sys->i_cur_rec, subrec_type);
814 es_out_Send(p_demux->out, p_sys->p_video, p_block_in);
815 return 0;
817 static int DemuxRecAudio( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
819 demux_sys_t *p_sys = p_demux->p_sys;
820 const int subrec_type = rec_hdr->subrec_type;
821 const long l_rec_size = rec_hdr->l_rec_size;
822 int esOffset1;
824 assert( rec_hdr->rec_type == 0xc0 );
825 if( !p_block_in )
826 return -1;
827 #if 0
828 int i;
829 fprintf( stderr, "Audio Packet Header " );
830 for( i = 0 ; i < 24 ; i++ )
831 fprintf( stderr, "%2.2x ", p_block_in->p_buffer[i] );
832 fprintf( stderr, "\n" );
833 #endif
835 if( subrec_type == 2 )
837 /* SA or DTiVo Audio Data, no PES (continued block)
838 * ================================================
841 /* continue PES if previous was incomplete */
842 if (p_sys->i_pes_buf_cnt > 0)
844 const int i_need = p_sys->i_Pes_Length - p_sys->i_pes_buf_cnt;
846 msg_Dbg(p_demux, "continuing PES header");
847 /* do we have enough data to complete? */
848 if (i_need >= l_rec_size)
850 /* don't have complete PES hdr; save what we have and return */
851 memcpy(&p_sys->pes_buffer[p_sys->i_pes_buf_cnt],
852 p_block_in->p_buffer, l_rec_size);
853 p_sys->i_pes_buf_cnt += l_rec_size;
854 /* */
855 block_Release(p_block_in);
856 return 0;
859 /* we have enough; reconstruct this p_frame with the new hdr */
860 memcpy(&p_sys->pes_buffer[p_sys->i_pes_buf_cnt],
861 p_block_in->p_buffer, i_need);
862 /* advance the block past the PES header (don't want to send it) */
863 p_block_in->p_buffer += i_need;
864 p_block_in->i_buffer -= i_need;
865 /* get the PTS out of this PES header (MPEG or AC3) */
866 if (p_sys->audio_type == TIVO_AUDIO_MPEG)
867 esOffset1 = find_es_header(ty_MPEGAudioPacket,
868 p_sys->pes_buffer, 5);
869 else
870 esOffset1 = find_es_header(ty_AC3AudioPacket,
871 p_sys->pes_buffer, 5);
872 if (esOffset1 < 0)
874 /* god help us; something's really wrong */
875 msg_Err(p_demux, "can't find audio PES header in packet");
877 else
879 p_sys->lastAudioPTS = VLC_TS_0 + get_pts(
880 &p_sys->pes_buffer[ esOffset1 + p_sys->i_Pts_Offset ] );
881 p_block_in->i_pts = p_sys->lastAudioPTS;
883 p_sys->i_pes_buf_cnt = 0;
885 /* S2 DTivo has AC3 packets with 2 padding bytes at end. This is
886 * not allowed in the AC3 spec and will cause problems. So here
887 * we try to trim things. */
888 /* Also, S1 DTivo has alternating short / long AC3 packets. That
889 * is, one packet is short (incomplete) and the next packet has
890 * the first one's missing data, plus all of its own. Strange. */
891 if (p_sys->audio_type == TIVO_AUDIO_AC3 &&
892 p_sys->tivo_series == TIVO_SERIES2) {
893 if (p_sys->l_ac3_pkt_size + p_block_in->i_buffer >
894 AC3_PKT_LENGTH) {
895 p_block_in->i_buffer -= 2;
896 p_sys->l_ac3_pkt_size = 0;
897 } else {
898 p_sys->l_ac3_pkt_size += p_block_in->i_buffer;
902 else if( subrec_type == 0x03 )
904 /* MPEG Audio with PES Header, either SA or DTiVo */
905 /* ================================================ */
906 esOffset1 = find_es_header( ty_MPEGAudioPacket,
907 p_block_in->p_buffer, 5 );
909 /*msg_Dbg(p_demux, "buffer has %#02x %#02x %#02x %#02x",
910 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
911 p_block_in->p_buffer[2], p_block_in->p_buffer[3]);
912 msg_Dbg(p_demux, "audio ES hdr at offset %d", esOffset1);*/
914 /* SA PES Header, No Audio Data */
915 /* ================================================ */
916 if ( ( esOffset1 == 0 ) && ( l_rec_size == 16 ) )
918 p_sys->lastAudioPTS = VLC_TS_0 + get_pts( &p_block_in->p_buffer[
919 SA_PTS_OFFSET ] );
920 if (p_sys->firstAudioPTS < 0)
921 p_sys->firstAudioPTS = p_sys->lastAudioPTS;
923 block_Release(p_block_in);
924 return 0;
925 /*msg_Dbg(p_demux, "SA Audio PTS %"PRId64, p_sys->lastAudioPTS );*/
927 /* DTiVo Audio with PES Header */
928 /* ================================================ */
930 /* Check for complete PES */
931 if (check_sync_pes(p_demux, p_block_in, esOffset1,
932 l_rec_size) == -1)
934 /* partial PES header found, nothing else.
935 * we're done. */
936 block_Release(p_block_in);
937 return 0;
939 #if 0
940 msg_Dbg(p_demux, "packet buffer has "
941 "%02x %02x %02x %02x %02x %02x %02x %02x "
942 "%02x %02x %02x %02x %02x %02x %02x %02x",
943 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
944 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
945 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
946 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
947 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
948 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
949 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
950 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
951 #endif
953 else if( subrec_type == 0x04 )
955 /* SA Audio with no PES Header */
956 /* ================================================ */
957 /*msg_Dbg(p_demux,
958 "Adding SA Audio Packet Size %ld", l_rec_size ); */
960 if (p_sys->lastAudioPTS > VLC_TS_INVALID )
961 p_block_in->i_pts = p_sys->lastAudioPTS;
963 else if( subrec_type == 0x09 )
965 /* DTiVo AC3 Audio Data with PES Header */
966 /* ================================================ */
967 esOffset1 = find_es_header( ty_AC3AudioPacket,
968 p_block_in->p_buffer, 5 );
970 #if 0
971 msg_Dbg(p_demux, "buffer has "
972 "%02x %02x %02x %02x %02x %02x %02x %02x "
973 "%02x %02x %02x %02x %02x %02x %02x %02x",
974 p_block_in->p_buffer[0], p_block_in->p_buffer[1],
975 p_block_in->p_buffer[2], p_block_in->p_buffer[3],
976 p_block_in->p_buffer[4], p_block_in->p_buffer[5],
977 p_block_in->p_buffer[6], p_block_in->p_buffer[7],
978 p_block_in->p_buffer[8], p_block_in->p_buffer[9],
979 p_block_in->p_buffer[10], p_block_in->p_buffer[11],
980 p_block_in->p_buffer[12], p_block_in->p_buffer[13],
981 p_block_in->p_buffer[14], p_block_in->p_buffer[15]);
982 msg_Dbg(p_demux, "audio ES AC3 hdr at offset %d", esOffset1);
983 #endif
985 /* Check for complete PES */
986 if (check_sync_pes(p_demux, p_block_in, esOffset1,
987 l_rec_size) == -1)
989 /* partial PES header found, nothing else. we're done. */
990 block_Release(p_block_in);
991 return 0;
993 /* S2 DTivo has invalid long AC3 packets */
994 if (p_sys->tivo_series == TIVO_SERIES2) {
995 if (p_block_in->i_buffer > AC3_PKT_LENGTH) {
996 p_block_in->i_buffer -= 2;
997 p_sys->l_ac3_pkt_size = 0;
998 } else {
999 p_sys->l_ac3_pkt_size = p_block_in->i_buffer;
1003 else
1005 /* Unsupported/Unknown */
1006 block_Release(p_block_in);
1007 return 0;
1010 /* set PCR before we send (if PTS found) */
1011 if( p_block_in->i_pts > VLC_TS_INVALID )
1012 es_out_Control( p_demux->out, ES_OUT_SET_PCR,
1013 p_block_in->i_pts );
1014 /* Send data */
1015 es_out_Send( p_demux->out, p_sys->p_audio, p_block_in );
1016 return 0;
1019 static int DemuxRecCc( demux_t *p_demux, ty_rec_hdr_t *rec_hdr, block_t *p_block_in )
1021 demux_sys_t *p_sys = p_demux->p_sys;
1022 int i_field;
1024 if( p_block_in )
1025 block_Release(p_block_in);
1027 if( rec_hdr->rec_type == 0x01 )
1028 i_field = 0;
1029 else if( rec_hdr->rec_type == 0x02 )
1030 i_field = 1;
1031 else
1032 return 0;
1034 /* XDS data (extract programs infos) transmitted on field 2 only */
1035 if( i_field == 1 )
1036 DemuxDecodeXds( p_demux, rec_hdr->ex[0], rec_hdr->ex[1] );
1038 if( p_sys->cc.i_data + 3 > CC_MAX_DATA_SIZE )
1039 return 0;
1041 cc_AppendData( &p_sys->cc, i_field, rec_hdr->ex );
1042 return 0;
1045 /* seek to a position within the stream, if possible */
1046 static int ty_stream_seek_pct(demux_t *p_demux, double seek_pct)
1048 demux_sys_t *p_sys = p_demux->p_sys;
1049 int64_t seek_pos = p_sys->i_stream_size * seek_pct;
1050 uint64_t l_skip_amt;
1051 unsigned i_cur_part;
1053 /* if we're not seekable, there's nothing to do */
1054 if (!p_sys->b_seekable)
1055 return VLC_EGENERIC;
1057 /* figure out which part & chunk we want & go there */
1058 i_cur_part = seek_pos / TIVO_PART_LENGTH;
1059 p_sys->i_cur_chunk = seek_pos / CHUNK_SIZE;
1061 /* try to read the part header (master chunk) if it's there */
1062 if ( stream_Seek( p_demux->s, i_cur_part * TIVO_PART_LENGTH ))
1064 /* can't seek stream */
1065 return VLC_EGENERIC;
1067 parse_master(p_demux);
1069 /* now for the actual chunk */
1070 if ( stream_Seek( p_demux->s, p_sys->i_cur_chunk * CHUNK_SIZE))
1072 /* can't seek stream */
1073 return VLC_EGENERIC;
1075 /* load the chunk */
1076 p_sys->i_stuff_cnt = 0;
1077 get_chunk_header(p_demux);
1079 /* seek within the chunk to get roughly to where we want */
1080 p_sys->i_cur_rec = (int)
1081 ((double) ((seek_pos % CHUNK_SIZE) / (double) (CHUNK_SIZE)) * p_sys->i_num_recs);
1082 msg_Dbg(p_demux, "Seeked to file pos %"PRId64, seek_pos);
1083 msg_Dbg(p_demux, " (chunk %d, record %d)",
1084 p_sys->i_cur_chunk - 1, p_sys->i_cur_rec);
1086 /* seek to the start of this record's data.
1087 * to do that, we have to skip past all prior records */
1088 l_skip_amt = 0;
1089 for ( int i=0; i<p_sys->i_cur_rec; i++)
1090 l_skip_amt += p_sys->rec_hdrs[i].l_rec_size;
1091 stream_Seek(p_demux->s, ((p_sys->i_cur_chunk-1) * CHUNK_SIZE) +
1092 (p_sys->i_num_recs * 16) + l_skip_amt + 4);
1094 /* to hell with syncing any audio or video, just start reading records... :) */
1095 /*p_sys->lastAudioPTS = p_sys->lastVideoPTS = VLC_TS_INVALID;*/
1096 return VLC_SUCCESS;
1099 /* XDS decoder */
1100 //#define TY_XDS_DEBUG
1101 static void XdsInit( xds_t *h )
1103 h->b_xds = false;
1104 h->i_class = XDS_MAX_CLASS_COUNT;
1105 h->i_type = 0;
1106 h->b_future = false;
1107 for( int i = 0; i < XDS_MAX_CLASS_COUNT; i++ )
1109 for( int j = 0; j < 128; j++ )
1110 h->pkt[i][j].b_started = false;
1112 h->b_meta_changed = false;
1113 memset( &h->meta, 0, sizeof(h->meta) );
1115 static void XdsExit( xds_t *h )
1117 /* */
1118 free( h->meta.psz_channel_name );
1119 free( h->meta.psz_channel_call_letter );
1120 free( h->meta.psz_channel_number );
1122 /* */
1123 free( h->meta.current.psz_name );
1124 free( h->meta.current.psz_rating );
1125 /* */
1126 free( h->meta.future.psz_name );
1127 free( h->meta.future.psz_rating );
1129 static void XdsStringUtf8( char dst[2*32+1], const uint8_t *p_src, int i_src )
1131 int i_dst = 0;
1132 for( int i = 0; i < i_src; i++ )
1134 switch( p_src[i] )
1136 #define E2( c, u1, u2 ) case c: dst[i_dst++] = u1; dst[i_dst++] = u2; break
1137 E2( 0x2a, 0xc3,0xa1); // lowercase a, acute accent
1138 E2( 0x5c, 0xc3,0xa9); // lowercase e, acute accent
1139 E2( 0x5e, 0xc3,0xad); // lowercase i, acute accent
1140 E2( 0x5f, 0xc3,0xb3); // lowercase o, acute accent
1141 E2( 0x60, 0xc3,0xba); // lowercase u, acute accent
1142 E2( 0x7b, 0xc3,0xa7); // lowercase c with cedilla
1143 E2( 0x7c, 0xc3,0xb7); // division symbol
1144 E2( 0x7d, 0xc3,0x91); // uppercase N tilde
1145 E2( 0x7e, 0xc3,0xb1); // lowercase n tilde
1146 #undef E2
1147 default:
1148 dst[i_dst++] = p_src[i];
1149 break;
1152 dst[i_dst++] = '\0';
1154 static bool XdsChangeString( xds_t *h, char **ppsz_dst, const char *psz_new )
1156 if( *ppsz_dst && psz_new && !strcmp( *ppsz_dst, psz_new ) )
1157 return false;
1158 if( *ppsz_dst == NULL && psz_new == NULL )
1159 return false;
1161 free( *ppsz_dst );
1162 if( psz_new )
1163 *ppsz_dst = strdup( psz_new );
1164 else
1165 *ppsz_dst = NULL;
1167 h->b_meta_changed = true;
1168 return true;
1171 static void XdsDecodeCurrentFuture( xds_t *h, xds_packet_t *pk )
1173 xds_meta_program_t *p_prg = h->b_future ? &h->meta.future : &h->meta.current;
1174 char name[2*32+1];
1175 int i_rating;
1177 switch( h->i_type )
1179 case 0x03:
1180 XdsStringUtf8( name, pk->p_data, pk->i_data );
1181 if( XdsChangeString( h, &p_prg->psz_name, name ) )
1183 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Program Name) %d'\n", pk->i_data );
1184 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> program name %s\n", name );
1186 break;
1187 case 0x05:
1188 i_rating = (pk->p_data[0] & 0x18);
1189 if( i_rating == 0x08 )
1191 /* TPG */
1192 static const char *pppsz_ratings[8][2] = {
1193 { "None", "No rating (no content advisory)" },
1194 { "TV-Y", "All Children (no content advisory)" },
1195 { "TV-Y7", "Directed to Older Children (V = Fantasy Violence)" },
1196 { "TV-G", "General Audience (no content advisory)" },
1197 { "TV-PG", "Parental Guidance Suggested" },
1198 { "TV-14", "Parents Strongly Cautioned" },
1199 { "TV-MA", "Mature Audience Only" },
1200 { "None", "No rating (no content advisory)" }
1202 p_prg->rating = XDS_META_PROGRAM_RATING_TPG;
1203 if( XdsChangeString( h, &p_prg->psz_rating, pppsz_ratings[pk->p_data[1]&0x07][0] ) )
1205 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1206 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> TPG Rating %s (%s)\n",
1207 // pppsz_ratings[pk->p_data[1]&0x07][0], pppsz_ratings[pk->p_data[1]&0x07][1] );
1210 else if( i_rating == 0x00 || i_rating == 0x10 )
1212 /* MPAA */
1213 static const char *pppsz_ratings[8][2] = {
1214 { "N/A", "N/A" },
1215 { "G", "General Audiences" },
1216 { "PG", "Parental Guidance Suggested" },
1217 { "PG-13", "Parents Strongly Cautioned" },
1218 { "R", "Restricted" },
1219 { "NC-17", "No one 17 and under admitted" },
1220 { "X", "No one under 17 admitted" },
1221 { "NR", "Not Rated" },
1223 p_prg->rating = XDS_META_PROGRAM_RATING_MPAA;
1224 if( XdsChangeString( h, &p_prg->psz_rating, pppsz_ratings[pk->p_data[0]&0x07][0] ) )
1226 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1227 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> TPG Rating %s (%s)\n",
1228 // pppsz_ratings[pk->p_data[0]&0x07][0], pppsz_ratings[pk->p_data[0]&0x07][1] );
1231 else
1233 /* Non US Rating TODO */
1234 assert( i_rating == 0x18 ); // only left value possible */
1235 p_prg->rating = XDS_META_PROGRAM_RATING_NONE;
1236 if( XdsChangeString( h, &p_prg->psz_rating, NULL ) )
1238 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Rating) %d'\n", pk->i_data );
1239 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> 0x%2.2x 0x%2.2x\n", pk->p_data[0], pk->p_data[1] );
1242 break;
1244 default:
1245 #ifdef TY_XDS_DEBUG
1246 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Current/Future (Unknown 0x%x)'\n", h->i_type );
1247 #endif
1248 break;
1252 static void XdsDecodeChannel( xds_t *h, xds_packet_t *pk )
1254 char name[2*32+1];
1255 char chan[2*32+1];
1257 switch( h->i_type )
1259 case 0x01:
1260 if( pk->i_data < 2 )
1261 return;
1262 XdsStringUtf8( name, pk->p_data, pk->i_data );
1263 if( XdsChangeString( h, &h->meta.psz_channel_name, name ) )
1265 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Name) %d'\n", pk->i_data );
1266 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> %s\n", name );
1268 break;
1270 case 0x02:
1271 if( pk->i_data < 4 )
1272 return;
1274 XdsStringUtf8( name, pk->p_data, 4 );
1275 if( XdsChangeString( h, &h->meta.psz_channel_call_letter, name ) )
1277 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1278 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> call letter %s\n", name );
1280 if( pk->i_data >= 6 )
1282 XdsStringUtf8( chan, &pk->p_data[4], 2 );
1283 if( XdsChangeString( h, &h->meta.psz_channel_number, chan ) )
1285 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1286 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> channel number %s\n", chan );
1289 else
1291 if( XdsChangeString( h, &h->meta.psz_channel_number, NULL ) )
1293 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Network Call Letter)' %d\n", pk->i_data );
1294 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: ====> no channel number letter anymore\n" );
1297 break;
1298 case 0x03:
1299 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Channel Tape Delay)'\n" );
1300 break;
1301 case 0x04:
1302 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Transmission Signal Identifier)'\n" );
1303 break;
1304 default:
1305 #ifdef TY_XDS_DEBUG
1306 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Channel (Unknown 0x%x)'\n", h->i_type );
1307 #endif
1308 break;
1312 static void XdsDecode( xds_t *h, xds_packet_t *pk )
1314 switch( h->i_class )
1316 case XDS_CLASS_CURRENT:
1317 case XDS_CLASS_FUTURE:
1318 XdsDecodeCurrentFuture( h, pk );
1319 break;
1320 case XDS_CLASS_CHANNEL:
1321 XdsDecodeChannel( h, pk );
1322 break;
1323 case XDS_CLASS_MISCELLANEOUS:
1324 #ifdef TY_XDS_DEBUG
1325 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Miscellaneous'\n" );
1326 #endif
1327 break;
1328 case XDS_CLASS_PUBLIC_SERVICE:
1329 #ifdef TY_XDS_DEBUG
1330 fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: class 'Public Service'\n" );
1331 #endif
1332 break;
1333 default:
1334 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS XdsDecode: unknown class\n" );
1335 break;
1339 static void XdsParse( xds_t *h, uint8_t d1, uint8_t d2 )
1341 /* TODO check parity */
1342 d1 &= 0x7f;
1343 d2 &= 0x7f;
1345 /* */
1346 if( d1 >= 0x01 && d1 <= 0x0e )
1348 const xds_class_t i_class = ( d1 - 1 ) >> 1;
1349 const int i_type = d2;
1350 const bool b_start = d1 & 0x01;
1351 xds_packet_t *pk = &h->pkt[i_class][i_type];
1353 if( !b_start && !pk->b_started )
1355 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS Continuying a non started packet, ignoring\n" );
1356 h->b_xds = false;
1357 return;
1360 h->b_xds = true;
1361 h->i_class = i_class;
1362 h->i_type = i_type;
1363 h->b_future = !b_start;
1364 pk->b_started = true;
1365 if( b_start )
1367 pk->i_data = 0;
1368 pk->i_sum = d1 + d2;
1371 else if( d1 == 0x0f && h->b_xds )
1373 xds_packet_t *pk = &h->pkt[h->i_class][h->i_type];
1375 /* TODO checksum and decode */
1376 pk->i_sum += d1 + d2;
1377 if( pk->i_sum & 0x7f )
1379 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS invalid checksum, ignoring ---------------------------------\n" );
1380 pk->b_started = false;
1381 return;
1383 if( pk->i_data <= 0 )
1385 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS empty packet, ignoring ---------------------------------\n" );
1386 pk->b_started = false;
1387 return;
1390 //if( pk->p_data[pk->i_data-1] == 0x40 ) /* Padding byte */
1391 // pk->i_data--;
1392 XdsDecode( h, pk );
1394 /* Reset it */
1395 pk->b_started = false;
1397 else if( d1 >= 0x20 && h->b_xds )
1399 xds_packet_t *pk = &h->pkt[h->i_class][h->i_type];
1401 if( pk->i_data+2 > XDS_MAX_DATA_SIZE )
1403 /* Broken -> reinit */
1404 //fprintf( stderr, "xxxxxxxxxxxxxxxXDS broken, reset\n" );
1405 h->b_xds = false;
1406 pk->b_started = false;
1407 return;
1409 /* TODO check parity bit */
1410 pk->p_data[pk->i_data++] = d1 & 0x7f;
1411 pk->p_data[pk->i_data++] = d2 & 0x7f;
1412 pk->i_sum += d1+d2;
1414 else
1416 h->b_xds = false;
1420 static void DemuxDecodeXds( demux_t *p_demux, uint8_t d1, uint8_t d2 )
1422 demux_sys_t *p_sys = p_demux->p_sys;
1424 XdsParse( &p_demux->p_sys->xds, d1, d2 );
1425 if( p_demux->p_sys->xds.b_meta_changed )
1427 xds_meta_t *m = &p_sys->xds.meta;
1428 vlc_meta_t *p_meta;
1429 vlc_epg_t *p_epg;
1431 /* Channel meta data */
1432 p_meta = vlc_meta_New();
1433 if( m->psz_channel_name )
1434 vlc_meta_SetPublisher( p_meta, m->psz_channel_name );
1435 if( m->psz_channel_call_letter )
1436 vlc_meta_SetTitle( p_meta, m->psz_channel_call_letter );
1437 if( m->psz_channel_number )
1438 vlc_meta_AddExtra( p_meta, "Channel number", m->psz_channel_number );
1439 es_out_Control( p_demux->out, ES_OUT_SET_GROUP_META, TY_ES_GROUP, p_meta );
1440 vlc_meta_Delete( p_meta );
1442 /* Event meta data (current/future) */
1443 p_epg = vlc_epg_New( NULL );
1444 if( m->current.psz_name )
1446 vlc_epg_AddEvent( p_epg, 0, 0, m->current.psz_name, NULL, NULL, 0 );
1447 //if( m->current.psz_rating )
1448 // TODO but VLC cannot yet handle rating per epg event
1449 vlc_epg_SetCurrent( p_epg, 0 );
1451 if( m->future.psz_name )
1454 if( p_epg->i_event > 0 )
1455 es_out_Control( p_demux->out, ES_OUT_SET_GROUP_EPG, TY_ES_GROUP, p_epg );
1456 vlc_epg_Delete( p_epg );
1458 p_demux->p_sys->xds.b_meta_changed = false;
1461 /* seek to an exact time position within the stream, if possible.
1462 * l_seek_time is in nanoseconds, the TIVO time standard.
1464 static int ty_stream_seek_time(demux_t *p_demux, uint64_t l_seek_time)
1466 demux_sys_t *p_sys = p_demux->p_sys;
1467 unsigned i_seq_entry = 0;
1468 unsigned i;
1469 int i_skip_cnt;
1470 int64_t l_cur_pos = stream_Tell(p_demux->s);
1471 unsigned i_cur_part = l_cur_pos / TIVO_PART_LENGTH;
1472 uint64_t l_seek_secs = l_seek_time / 1000000000;
1473 uint64_t l_fwd_stamp = 1;
1475 /* if we're not seekable, there's nothing to do */
1476 if (!p_sys->b_seekable || !p_sys->b_have_master)
1477 return VLC_EGENERIC;
1479 msg_Dbg(p_demux, "Skipping to time %02"PRIu64":%02"PRIu64":%02"PRIu64,
1480 l_seek_secs / 3600, (l_seek_secs / 60) % 60, l_seek_secs % 60);
1482 /* seek to the proper segment if necessary */
1483 /* first see if we need to go back */
1484 while (l_seek_time < p_sys->l_first_ty_pts) {
1485 msg_Dbg(p_demux, "skipping to prior segment.");
1486 /* load previous part */
1487 if (i_cur_part == 0) {
1488 stream_Seek(p_demux->s, l_cur_pos);
1489 msg_Err(p_demux, "Attempt to seek past BOF");
1490 return VLC_EGENERIC;
1492 stream_Seek(p_demux->s, (i_cur_part - 1) * TIVO_PART_LENGTH);
1493 i_cur_part--;
1494 parse_master(p_demux);
1496 /* maybe we need to go forward */
1497 while (l_seek_time > p_sys->l_final_ty_pts) {
1498 msg_Dbg(p_demux, "skipping to next segment.");
1499 /* load next part */
1500 if ((i_cur_part + 1) * TIVO_PART_LENGTH > p_sys->i_stream_size) {
1501 /* error; restore previous file position */
1502 stream_Seek(p_demux->s, l_cur_pos);
1503 msg_Err(p_demux, "seek error");
1504 return VLC_EGENERIC;
1506 stream_Seek(p_demux->s, (i_cur_part + 1) * TIVO_PART_LENGTH);
1507 i_cur_part++;
1508 parse_master(p_demux);
1511 /* our target is somewhere within this part;
1512 find the proper chunk using seq_table */
1513 for (i=1; i<p_sys->i_seq_table_size; i++) {
1514 if (p_sys->seq_table[i].l_timestamp > l_seek_time) {
1515 /* i-1 is the section we want; remember the next timestamp in case
1516 we have to use it (this section may not have a proper SEQ hdr
1517 for the time we're seeking) */
1518 msg_Dbg(p_demux, "stopping at seq entry %d.", i);
1519 l_fwd_stamp = p_sys->seq_table[i].l_timestamp;
1520 i_seq_entry = i-1;
1521 break;
1525 /* if we went through the entire last loop and didn't find our target,
1526 then we skip to the next part. What has happened is that the actual
1527 time we're seeking is within this part, but there isn't a SEQ hdr
1528 for it here. So we skip to the next part */
1529 if (i == p_sys->i_seq_table_size) {
1530 if ((i_cur_part + 1) * TIVO_PART_LENGTH > p_sys->i_stream_size) {
1531 /* error; restore previous file position */
1532 stream_Seek(p_demux->s, l_cur_pos);
1533 msg_Err(p_demux, "seek error");
1534 return VLC_EGENERIC;
1536 stream_Seek(p_demux->s, (i_cur_part + 1) * TIVO_PART_LENGTH);
1537 i_cur_part++;
1538 parse_master(p_demux);
1539 i_seq_entry = 0;
1542 /* determine which chunk has our seek_time */
1543 for (unsigned i=0; i<p_sys->i_bits_per_seq_entry; i++) {
1544 uint64_t l_chunk_nr = i_seq_entry * p_sys->i_bits_per_seq_entry + i;
1545 uint64_t l_chunk_offset = (l_chunk_nr + 1) * CHUNK_SIZE;
1546 msg_Dbg(p_demux, "testing part %d chunk %"PRIu64" mask 0x%02X bit %d",
1547 i_cur_part, l_chunk_nr,
1548 p_sys->seq_table[i_seq_entry].chunk_bitmask[i/8], i%8);
1549 if (p_sys->seq_table[i_seq_entry].chunk_bitmask[i/8] & (1 << (i%8))) {
1550 /* check this chunk's SEQ header timestamp */
1551 msg_Dbg(p_demux, "has SEQ. seeking to chunk at 0x%"PRIu64,
1552 (i_cur_part * TIVO_PART_LENGTH) + l_chunk_offset);
1553 stream_Seek(p_demux->s, (i_cur_part * TIVO_PART_LENGTH) +
1554 l_chunk_offset);
1555 // TODO: we don't have to parse the full header set;
1556 // just test the seq_rec entry for its timestamp
1557 p_sys->i_stuff_cnt = 0;
1558 get_chunk_header(p_demux);
1559 // check ty PTS for the SEQ entry in this chunk
1560 if (p_sys->i_seq_rec < 0 || p_sys->i_seq_rec > p_sys->i_num_recs) {
1561 msg_Err(p_demux, "no SEQ hdr in chunk; table had one.");
1562 /* Seek to beginning of original chunk & reload it */
1563 stream_Seek(p_demux->s, (l_cur_pos / CHUNK_SIZE) * CHUNK_SIZE);
1564 p_sys->i_stuff_cnt = 0;
1565 get_chunk_header(p_demux);
1566 return VLC_EGENERIC;
1568 l_seek_secs = p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts /
1569 1000000000;
1570 msg_Dbg(p_demux, "found SEQ hdr for timestamp %02"PRIu64":%02"PRIu64":%02"PRIu64,
1571 l_seek_secs / 3600,
1572 (l_seek_secs / 60) % 60, l_seek_secs % 60);
1573 if (p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts >= l_seek_time) {
1574 // keep this one? go back?
1575 /* for now, we take this one. it's the first SEQ hdr AFTER
1576 the time we were searching for. */
1577 msg_Dbg(p_demux, "seek target found.");
1578 break;
1580 msg_Dbg(p_demux, "timestamp too early. still scanning.");
1583 /* if we made it through this entire loop without finding our target,
1584 then we skip to the next section. What has happened is that the actual
1585 time we're seeking is within this section, but there isn't a SEQ hdr
1586 for it here. So we skip to the next closest one (l_fwd_stamp) */
1587 if (i == p_sys->i_bits_per_seq_entry)
1588 return ty_stream_seek_time(p_demux, l_fwd_stamp);
1590 /* current stream ptr is at beginning of data for this chunk,
1591 so we need to skip past any stream data prior to the seq_rec
1592 in this chunk */
1593 i_skip_cnt = 0;
1594 for (int j=0; j<p_sys->i_seq_rec; j++)
1595 i_skip_cnt += p_sys->rec_hdrs[j].l_rec_size;
1596 stream_Read(p_demux->s, NULL, i_skip_cnt);
1597 p_sys->i_cur_rec = p_sys->i_seq_rec;
1598 //p_sys->l_last_ty_pts = p_sys->rec_hdrs[p_sys->i_seq_rec].l_ty_pts;
1599 //p_sys->l_last_ty_pts_sync = p_sys->lastAudioPTS;
1601 return VLC_SUCCESS;
1605 /* parse a master chunk, filling the SEQ table and other variables.
1606 * We assume the stream is currently pointing to it.
1608 static void parse_master(demux_t *p_demux)
1610 demux_sys_t *p_sys = p_demux->p_sys;
1611 uint8_t mst_buf[32];
1612 uint32_t i, i_map_size;
1613 int64_t i_save_pos = stream_Tell(p_demux->s);
1614 int64_t i_pts_secs;
1616 /* Note that the entries in the SEQ table in the stream may have
1617 different sizes depending on the bits per entry. We store them
1618 all in the same size structure, so we have to parse them out one
1619 by one. If we had a dynamic structure, we could simply read the
1620 entire table directly from the stream into memory in place. */
1622 /* clear the SEQ table */
1623 free(p_sys->seq_table);
1625 /* parse header info */
1626 stream_Read(p_demux->s, mst_buf, 32);
1627 i_map_size = U32_AT(&mst_buf[20]); /* size of bitmask, in bytes */
1628 p_sys->i_bits_per_seq_entry = i_map_size * 8;
1629 i = U32_AT(&mst_buf[28]); /* size of SEQ table, in bytes */
1630 p_sys->i_seq_table_size = i / (8 + i_map_size);
1632 /* parse all the entries */
1633 p_sys->seq_table = calloc(p_sys->i_seq_table_size, sizeof(ty_seq_table_t));
1634 if (p_sys->seq_table == NULL)
1636 p_sys->i_seq_table_size = 0;
1637 return;
1639 for (unsigned i=0; i<p_sys->i_seq_table_size; i++) {
1640 stream_Read(p_demux->s, mst_buf, 8);
1641 p_sys->seq_table[i].l_timestamp = U64_AT(&mst_buf[0]);
1642 if (i_map_size > 8) {
1643 msg_Err(p_demux, "Unsupported SEQ bitmap size in master chunk");
1644 stream_Read(p_demux->s, NULL, i_map_size);
1645 } else {
1646 stream_Read(p_demux->s, mst_buf + 8, i_map_size);
1647 memcpy(p_sys->seq_table[i].chunk_bitmask, &mst_buf[8], i_map_size);
1651 /* set up a few of our variables */
1652 p_sys->l_first_ty_pts = p_sys->seq_table[0].l_timestamp;
1653 p_sys->l_final_ty_pts =
1654 p_sys->seq_table[p_sys->i_seq_table_size - 1].l_timestamp;
1655 p_sys->b_have_master = true;
1657 i_pts_secs = p_sys->l_first_ty_pts / 1000000000;
1658 msg_Dbg( p_demux,
1659 "first TY pts in master is %02"PRId64":%02"PRId64":%02"PRId64,
1660 i_pts_secs / 3600, (i_pts_secs / 60) % 60, i_pts_secs % 60 );
1661 i_pts_secs = p_sys->l_final_ty_pts / 1000000000;
1662 msg_Dbg( p_demux,
1663 "final TY pts in master is %02"PRId64":%02"PRId64":%02"PRId64,
1664 i_pts_secs / 3600, (i_pts_secs / 60) % 60, i_pts_secs % 60 );
1666 /* seek past this chunk */
1667 stream_Seek(p_demux->s, i_save_pos + CHUNK_SIZE);
1671 /* ======================================================================== */
1672 /* "Peek" at some chunks. Skip over the Part header if we find it.
1673 * We parse the peeked data and determine audio type,
1674 * SA vs. DTivo, & Tivo Series.
1675 * Set global vars i_Pes_Length, i_Pts_Offset,
1676 * p_sys->tivo_series, p_sys->tivo_type, p_sys->audio_type */
1677 static int probe_stream(demux_t *p_demux)
1679 demux_sys_t *p_sys = p_demux->p_sys;
1680 const uint8_t *p_buf;
1681 int i;
1682 bool b_probe_error = false;
1684 /* we need CHUNK_PEEK_COUNT chunks of data, first one might be a Part header, so ... */
1685 if (stream_Peek( p_demux->s, &p_buf, CHUNK_PEEK_COUNT * CHUNK_SIZE ) <
1686 CHUNK_PEEK_COUNT * CHUNK_SIZE) {
1687 msg_Err(p_demux, "Can't peek %d chunks", CHUNK_PEEK_COUNT);
1688 /* TODO: if seekable, then loop reading chunks into a temp buffer */
1689 return VLC_EGENERIC;
1692 /* the real work: analyze this chunk */
1693 for (i = 0; i < CHUNK_PEEK_COUNT; i++) {
1694 analyze_chunk(p_demux, p_buf);
1695 if (p_sys->tivo_series != TIVO_SERIES_UNKNOWN &&
1696 p_sys->audio_type != TIVO_AUDIO_UNKNOWN &&
1697 p_sys->tivo_type != TIVO_TYPE_UNKNOWN)
1698 break;
1699 p_buf += CHUNK_SIZE;
1702 /* the final tally */
1703 if (p_sys->tivo_series == TIVO_SERIES_UNKNOWN) {
1704 msg_Err(p_demux, "Can't determine Tivo Series.");
1705 b_probe_error = true;
1707 if (p_sys->audio_type == TIVO_AUDIO_UNKNOWN) {
1708 msg_Err(p_demux, "Can't determine Tivo Audio Type.");
1709 b_probe_error = true;
1711 if (p_sys->tivo_type == TIVO_TYPE_UNKNOWN) {
1712 msg_Err(p_demux, "Can't determine Tivo Type (SA/DTivo).");
1713 b_probe_error = true;
1715 return b_probe_error?VLC_EGENERIC:VLC_SUCCESS;
1719 /* ======================================================================== */
1720 /* gather statistics for this chunk & set our tivo-type vars accordingly */
1721 static void analyze_chunk(demux_t *p_demux, const uint8_t *p_chunk)
1723 demux_sys_t *p_sys = p_demux->p_sys;
1724 int i_num_recs, i;
1725 ty_rec_hdr_t *p_hdrs;
1726 int i_num_6e0, i_num_be0, i_num_9c0, i_num_3c0;
1727 int i_payload_size;
1729 /* skip if it's a Part header */
1730 if( U32_AT( &p_chunk[ 0 ] ) == TIVO_PES_FILEID )
1731 return;
1733 /* number of records in chunk (we ignore high order byte;
1734 * rarely are there > 256 chunks & we don't need that many anyway) */
1735 i_num_recs = p_chunk[0];
1736 if (i_num_recs < 5) {
1737 /* try again with the next chunk. Sometimes there are dead ones */
1738 return;
1741 p_chunk += 4; /* skip past rec count & SEQ bytes */
1742 //msg_Dbg(p_demux, "probe: chunk has %d recs", i_num_recs);
1743 p_hdrs = parse_chunk_headers(p_chunk, i_num_recs, &i_payload_size);
1744 /* scan headers.
1745 * 1. check video packets. Presence of 0x6e0 means S1.
1746 * No 6e0 but have be0 means S2.
1747 * 2. probe for audio 0x9c0 vs 0x3c0 (AC3 vs Mpeg)
1748 * If AC-3, then we have DTivo.
1749 * If MPEG, search for PTS offset. This will determine SA vs. DTivo.
1751 i_num_6e0 = i_num_be0 = i_num_9c0 = i_num_3c0 = 0;
1752 for (i=0; i<i_num_recs; i++) {
1753 //msg_Dbg(p_demux, "probe: rec is %d/%d = 0x%04x", p_hdrs[i].subrec_type,
1754 //p_hdrs[i].rec_type,
1755 //p_hdrs[i].subrec_type << 8 | p_hdrs[i].rec_type);
1756 switch (p_hdrs[i].subrec_type << 8 | p_hdrs[i].rec_type) {
1757 case 0x6e0:
1758 i_num_6e0++;
1759 break;
1760 case 0xbe0:
1761 i_num_be0++;
1762 break;
1763 case 0x3c0:
1764 i_num_3c0++;
1765 break;
1766 case 0x9c0:
1767 i_num_9c0++;
1768 break;
1771 msg_Dbg(p_demux, "probe: chunk has %d 0x6e0 recs, %d 0xbe0 recs.",
1772 i_num_6e0, i_num_be0);
1774 /* set up our variables */
1775 if (i_num_6e0 > 0) {
1776 msg_Dbg(p_demux, "detected Series 1 Tivo");
1777 p_sys->tivo_series = TIVO_SERIES1;
1778 p_sys->i_Pes_Length = SERIES1_PES_LENGTH;
1779 } else if (i_num_be0 > 0) {
1780 msg_Dbg(p_demux, "detected Series 2 Tivo");
1781 p_sys->tivo_series = TIVO_SERIES2;
1782 p_sys->i_Pes_Length = SERIES2_PES_LENGTH;
1784 if (i_num_9c0 > 0) {
1785 msg_Dbg(p_demux, "detected AC-3 Audio (DTivo)" );
1786 p_sys->audio_type = TIVO_AUDIO_AC3;
1787 p_sys->tivo_type = TIVO_TYPE_DTIVO;
1788 p_sys->i_Pts_Offset = AC3_PTS_OFFSET;
1789 p_sys->i_Pes_Length = AC3_PES_LENGTH;
1790 } else if (i_num_3c0 > 0) {
1791 p_sys->audio_type = TIVO_AUDIO_MPEG;
1792 msg_Dbg(p_demux, "detected MPEG Audio" );
1795 /* if tivo_type still unknown, we can check PTS location
1796 * in MPEG packets to determine tivo_type */
1797 if (p_sys->tivo_type == TIVO_TYPE_UNKNOWN) {
1798 uint32_t i_data_offset = (16 * i_num_recs);
1799 for (i=0; i<i_num_recs; i++) {
1800 if ((p_hdrs[i].subrec_type << 0x08 | p_hdrs[i].rec_type) == 0x3c0 &&
1801 p_hdrs[i].l_rec_size > 15) {
1802 /* first make sure we're aligned */
1803 int i_pes_offset = find_es_header(ty_MPEGAudioPacket,
1804 &p_chunk[i_data_offset], 5);
1805 if (i_pes_offset >= 0) {
1806 /* pes found. on SA, PES has hdr data at offset 6, not PTS. */
1807 //msg_Dbg(p_demux, "probe: mpeg es header found in rec %d at offset %d",
1808 //i, i_pes_offset);
1809 if ((p_chunk[i_data_offset + 6 + i_pes_offset] & 0x80) == 0x80) {
1810 /* S1SA or S2(any) Mpeg Audio (PES hdr, not a PTS start) */
1811 if (p_sys->tivo_series == TIVO_SERIES1)
1812 msg_Dbg(p_demux, "detected Stand-Alone Tivo" );
1813 p_sys->tivo_type = TIVO_TYPE_SA;
1814 p_sys->i_Pts_Offset = SA_PTS_OFFSET;
1815 } else {
1816 if (p_sys->tivo_series == TIVO_SERIES1)
1817 msg_Dbg(p_demux, "detected DirecTV Tivo" );
1818 p_sys->tivo_type = TIVO_TYPE_DTIVO;
1819 p_sys->i_Pts_Offset = DTIVO_PTS_OFFSET;
1821 break;
1824 i_data_offset += p_hdrs[i].l_rec_size;
1827 free(p_hdrs);
1831 /* =========================================================================== */
1832 static int get_chunk_header(demux_t *p_demux)
1834 int i_readSize, i_num_recs;
1835 uint8_t *p_hdr_buf;
1836 const uint8_t *p_peek;
1837 demux_sys_t *p_sys = p_demux->p_sys;
1838 int i_payload_size; /* sum of all records' sizes */
1840 msg_Dbg(p_demux, "parsing ty chunk #%d", p_sys->i_cur_chunk );
1842 /* if we have left-over filler space from the last chunk, get that */
1843 if (p_sys->i_stuff_cnt > 0) {
1844 stream_Read( p_demux->s, NULL, p_sys->i_stuff_cnt);
1845 p_sys->i_stuff_cnt = 0;
1848 /* read the TY packet header */
1849 i_readSize = stream_Peek( p_demux->s, &p_peek, 4 );
1850 p_sys->i_cur_chunk++;
1852 if ( (i_readSize < 4) || ( U32_AT(&p_peek[ 0 ] ) == 0 ))
1854 /* EOF */
1855 p_sys->eof = 1;
1856 return 0;
1859 /* check if it's a PART Header */
1860 if( U32_AT( &p_peek[ 0 ] ) == TIVO_PES_FILEID )
1862 /* parse master chunk */
1863 parse_master(p_demux);
1864 return get_chunk_header(p_demux);
1867 /* number of records in chunk (8- or 16-bit number) */
1868 if (p_peek[3] & 0x80)
1870 /* 16 bit rec cnt */
1871 p_sys->i_num_recs = i_num_recs = (p_peek[1] << 8) + p_peek[0];
1872 p_sys->i_seq_rec = (p_peek[3] << 8) + p_peek[2];
1873 if (p_sys->i_seq_rec != 0xffff)
1875 p_sys->i_seq_rec &= ~0x8000;
1878 else
1880 /* 8 bit reclen - tivo 1.3 format */
1881 p_sys->i_num_recs = i_num_recs = p_peek[0];
1882 p_sys->i_seq_rec = p_peek[1];
1884 p_sys->i_cur_rec = 0;
1885 p_sys->b_first_chunk = false;
1887 /*msg_Dbg( p_demux, "chunk has %d records", i_num_recs );*/
1889 free(p_sys->rec_hdrs);
1890 p_sys->rec_hdrs = NULL;
1892 /* skip past the 4 bytes we "peeked" earlier */
1893 stream_Read( p_demux->s, NULL, 4 );
1895 /* read the record headers into a temp buffer */
1896 p_hdr_buf = xmalloc(i_num_recs * 16);
1897 if (stream_Read(p_demux->s, p_hdr_buf, i_num_recs * 16) < i_num_recs * 16) {
1898 free( p_hdr_buf );
1899 p_sys->eof = true;
1900 return 0;
1902 /* parse them */
1903 p_sys->rec_hdrs = parse_chunk_headers(p_hdr_buf, i_num_recs,
1904 &i_payload_size);
1905 free(p_hdr_buf);
1907 p_sys->i_stuff_cnt = CHUNK_SIZE - 4 -
1908 (p_sys->i_num_recs * 16) - i_payload_size;
1909 if (p_sys->i_stuff_cnt > 0)
1910 msg_Dbg( p_demux, "chunk has %d stuff bytes at end",
1911 p_sys->i_stuff_cnt );
1912 return 1;
1916 static ty_rec_hdr_t *parse_chunk_headers( const uint8_t *p_buf,
1917 int i_num_recs, int *pi_payload_size)
1919 int i;
1920 ty_rec_hdr_t *p_hdrs, *p_rec_hdr;
1922 *pi_payload_size = 0;
1923 p_hdrs = xmalloc(i_num_recs * sizeof(ty_rec_hdr_t));
1925 for (i = 0; i < i_num_recs; i++)
1927 const uint8_t *record_header = p_buf + (i * 16);
1928 p_rec_hdr = &p_hdrs[i]; /* for brevity */
1929 p_rec_hdr->rec_type = record_header[3];
1930 p_rec_hdr->subrec_type = record_header[2] & 0x0f;
1931 if ((record_header[ 0 ] & 0x80) == 0x80)
1933 uint8_t b1, b2;
1934 /* marker bit 2 set, so read extended data */
1935 b1 = ( ( ( record_header[ 0 ] & 0x0f ) << 4 ) |
1936 ( ( record_header[ 1 ] & 0xf0 ) >> 4 ) );
1937 b2 = ( ( ( record_header[ 1 ] & 0x0f ) << 4 ) |
1938 ( ( record_header[ 2 ] & 0xf0 ) >> 4 ) );
1940 p_rec_hdr->ex[0] = b1;
1941 p_rec_hdr->ex[1] = b2;
1942 p_rec_hdr->l_rec_size = 0;
1943 p_rec_hdr->l_ty_pts = 0;
1944 p_rec_hdr->b_ext = true;
1946 else
1948 p_rec_hdr->l_rec_size = ( record_header[ 0 ] << 8 |
1949 record_header[ 1 ] ) << 4 | ( record_header[ 2 ] >> 4 );
1950 *pi_payload_size += p_rec_hdr->l_rec_size;
1951 p_rec_hdr->b_ext = false;
1952 p_rec_hdr->l_ty_pts = U64_AT( &record_header[ 8 ] );
1954 //fprintf( stderr, "parse_chunk_headers[%d] t=0x%x s=%d\n", i, p_rec_hdr->rec_type, p_rec_hdr->subrec_type );
1955 } /* end of record-header loop */
1956 return p_hdrs;