Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * MPEG-2 transport stream (aka DVB) demuxer | ||
3 | * Copyright (c) 2002-2003 Fabrice Bellard | ||
4 | * | ||
5 | * This file is part of FFmpeg. | ||
6 | * | ||
7 | * FFmpeg is free software; you can redistribute it and/or | ||
8 | * modify it under the terms of the GNU Lesser General Public | ||
9 | * License as published by the Free Software Foundation; either | ||
10 | * version 2.1 of the License, or (at your option) any later version. | ||
11 | * | ||
12 | * FFmpeg is distributed in the hope that it will be useful, | ||
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
15 | * Lesser General Public License for more details. | ||
16 | * | ||
17 | * You should have received a copy of the GNU Lesser General Public | ||
18 | * License along with FFmpeg; if not, write to the Free Software | ||
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
20 | */ | ||
21 | |||
22 | #include "config_components.h" | ||
23 | |||
24 | #include "libavutil/buffer.h" | ||
25 | #include "libavutil/crc.h" | ||
26 | #include "libavutil/internal.h" | ||
27 | #include "libavutil/intreadwrite.h" | ||
28 | #include "libavutil/log.h" | ||
29 | #include "libavutil/dict.h" | ||
30 | #include "libavutil/mem.h" | ||
31 | #include "libavutil/opt.h" | ||
32 | #include "libavutil/avassert.h" | ||
33 | #include "libavutil/dovi_meta.h" | ||
34 | #include "libavcodec/bytestream.h" | ||
35 | #include "libavcodec/defs.h" | ||
36 | #include "libavcodec/get_bits.h" | ||
37 | #include "libavcodec/opus/opus.h" | ||
38 | #include "avformat.h" | ||
39 | #include "mpegts.h" | ||
40 | #include "internal.h" | ||
41 | #include "avio_internal.h" | ||
42 | #include "demux.h" | ||
43 | #include "mpeg.h" | ||
44 | #include "isom.h" | ||
45 | #if CONFIG_ICONV | ||
46 | #include <iconv.h> | ||
47 | #endif | ||
48 | |||
49 | /* maximum size in which we look for synchronization if | ||
50 | * synchronization is lost */ | ||
51 | #define MAX_RESYNC_SIZE 65536 | ||
52 | |||
53 | #define MAX_MP4_DESCR_COUNT 16 | ||
54 | |||
55 | #define MOD_UNLIKELY(modulus, dividend, divisor, prev_dividend) \ | ||
56 | do { \ | ||
57 | if ((prev_dividend) == 0 || (dividend) - (prev_dividend) != (divisor)) \ | ||
58 | (modulus) = (dividend) % (divisor); \ | ||
59 | (prev_dividend) = (dividend); \ | ||
60 | } while (0) | ||
61 | |||
62 | #define PROBE_PACKET_MAX_BUF 8192 | ||
63 | #define PROBE_PACKET_MARGIN 5 | ||
64 | |||
65 | enum MpegTSFilterType { | ||
66 | MPEGTS_PES, | ||
67 | MPEGTS_SECTION, | ||
68 | MPEGTS_PCR, | ||
69 | }; | ||
70 | |||
71 | typedef struct MpegTSFilter MpegTSFilter; | ||
72 | |||
73 | typedef int PESCallback (MpegTSFilter *f, const uint8_t *buf, int len, | ||
74 | int is_start, int64_t pos); | ||
75 | |||
76 | typedef struct MpegTSPESFilter { | ||
77 | PESCallback *pes_cb; | ||
78 | void *opaque; | ||
79 | } MpegTSPESFilter; | ||
80 | |||
81 | typedef void SectionCallback (MpegTSFilter *f, const uint8_t *buf, int len); | ||
82 | |||
83 | typedef void SetServiceCallback (void *opaque, int ret); | ||
84 | |||
85 | typedef struct MpegTSSectionFilter { | ||
86 | int section_index; | ||
87 | int section_h_size; | ||
88 | int last_ver; | ||
89 | unsigned crc; | ||
90 | unsigned last_crc; | ||
91 | uint8_t *section_buf; | ||
92 | unsigned int check_crc : 1; | ||
93 | unsigned int end_of_section_reached : 1; | ||
94 | SectionCallback *section_cb; | ||
95 | void *opaque; | ||
96 | } MpegTSSectionFilter; | ||
97 | |||
98 | struct MpegTSFilter { | ||
99 | int pid; | ||
100 | int es_id; | ||
101 | int last_cc; /* last cc code (-1 if first packet) */ | ||
102 | int64_t last_pcr; | ||
103 | int discard; | ||
104 | enum MpegTSFilterType type; | ||
105 | union { | ||
106 | MpegTSPESFilter pes_filter; | ||
107 | MpegTSSectionFilter section_filter; | ||
108 | } u; | ||
109 | }; | ||
110 | |||
111 | struct Stream { | ||
112 | int idx; | ||
113 | int stream_identifier; | ||
114 | }; | ||
115 | |||
116 | #define MAX_STREAMS_PER_PROGRAM 128 | ||
117 | #define MAX_PIDS_PER_PROGRAM (MAX_STREAMS_PER_PROGRAM + 2) | ||
118 | struct Program { | ||
119 | unsigned int id; // program id/service id | ||
120 | unsigned int nb_pids; | ||
121 | unsigned int pids[MAX_PIDS_PER_PROGRAM]; | ||
122 | unsigned int nb_streams; | ||
123 | struct Stream streams[MAX_STREAMS_PER_PROGRAM]; | ||
124 | |||
125 | /** have we found pmt for this program */ | ||
126 | int pmt_found; | ||
127 | }; | ||
128 | |||
129 | struct MpegTSContext { | ||
130 | const AVClass *class; | ||
131 | /* user data */ | ||
132 | AVFormatContext *stream; | ||
133 | /** raw packet size, including FEC if present */ | ||
134 | int raw_packet_size; | ||
135 | |||
136 | int64_t pos47_full; | ||
137 | |||
138 | /** if true, all pids are analyzed to find streams */ | ||
139 | int auto_guess; | ||
140 | |||
141 | /** compute exact PCR for each transport stream packet */ | ||
142 | int mpeg2ts_compute_pcr; | ||
143 | |||
144 | /** fix dvb teletext pts */ | ||
145 | int fix_teletext_pts; | ||
146 | |||
147 | int64_t cur_pcr; /**< used to estimate the exact PCR */ | ||
148 | int64_t pcr_incr; /**< used to estimate the exact PCR */ | ||
149 | |||
150 | /* data needed to handle file based ts */ | ||
151 | /** stop parsing loop */ | ||
152 | int stop_parse; | ||
153 | /** packet containing Audio/Video data */ | ||
154 | AVPacket *pkt; | ||
155 | /** to detect seek */ | ||
156 | int64_t last_pos; | ||
157 | |||
158 | int skip_changes; | ||
159 | int skip_clear; | ||
160 | int skip_unknown_pmt; | ||
161 | |||
162 | int scan_all_pmts; | ||
163 | |||
164 | int resync_size; | ||
165 | int merge_pmt_versions; | ||
166 | int max_packet_size; | ||
167 | |||
168 | int id; | ||
169 | |||
170 | /******************************************/ | ||
171 | /* private mpegts data */ | ||
172 | /* scan context */ | ||
173 | /** structure to keep track of Program->pids mapping */ | ||
174 | unsigned int nb_prg; | ||
175 | struct Program *prg; | ||
176 | |||
177 | int8_t crc_validity[NB_PID_MAX]; | ||
178 | /** filters for various streams specified by PMT + for the PAT and PMT */ | ||
179 | MpegTSFilter *pids[NB_PID_MAX]; | ||
180 | int current_pid; | ||
181 | |||
182 | AVStream *epg_stream; | ||
183 | AVBufferPool* pools[32]; | ||
184 | }; | ||
185 | |||
186 | #define MPEGTS_OPTIONS \ | ||
187 | { "resync_size", "set size limit for looking up a new synchronization", \ | ||
188 | offsetof(MpegTSContext, resync_size), AV_OPT_TYPE_INT, \ | ||
189 | { .i64 = MAX_RESYNC_SIZE}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, \ | ||
190 | { "ts_id", "transport stream id", \ | ||
191 | offsetof(MpegTSContext, id), AV_OPT_TYPE_INT, \ | ||
192 | { .i64 = 0 }, 0, INT_MAX, AV_OPT_FLAG_EXPORT | AV_OPT_FLAG_READONLY }, \ | ||
193 | { "ts_packetsize", "output option carrying the raw packet size", \ | ||
194 | offsetof(MpegTSContext, raw_packet_size), AV_OPT_TYPE_INT, \ | ||
195 | { .i64 = 0 }, 0, INT_MAX, AV_OPT_FLAG_EXPORT | AV_OPT_FLAG_READONLY } | ||
196 | |||
197 | static const AVOption options[] = { | ||
198 | MPEGTS_OPTIONS, | ||
199 | {"fix_teletext_pts", "try to fix pts values of dvb teletext streams", offsetof(MpegTSContext, fix_teletext_pts), AV_OPT_TYPE_BOOL, | ||
200 | {.i64 = 1}, 0, 1, AV_OPT_FLAG_DECODING_PARAM }, | ||
201 | {"scan_all_pmts", "scan and combine all PMTs", offsetof(MpegTSContext, scan_all_pmts), AV_OPT_TYPE_BOOL, | ||
202 | {.i64 = -1}, -1, 1, AV_OPT_FLAG_DECODING_PARAM }, | ||
203 | {"skip_unknown_pmt", "skip PMTs for programs not advertised in the PAT", offsetof(MpegTSContext, skip_unknown_pmt), AV_OPT_TYPE_BOOL, | ||
204 | {.i64 = 0}, 0, 1, AV_OPT_FLAG_DECODING_PARAM }, | ||
205 | {"merge_pmt_versions", "re-use streams when PMT's version/pids change", offsetof(MpegTSContext, merge_pmt_versions), AV_OPT_TYPE_BOOL, | ||
206 | {.i64 = 0}, 0, 1, AV_OPT_FLAG_DECODING_PARAM }, | ||
207 | {"skip_changes", "skip changing / adding streams / programs", offsetof(MpegTSContext, skip_changes), AV_OPT_TYPE_BOOL, | ||
208 | {.i64 = 0}, 0, 1, 0 }, | ||
209 | {"skip_clear", "skip clearing programs", offsetof(MpegTSContext, skip_clear), AV_OPT_TYPE_BOOL, | ||
210 | {.i64 = 0}, 0, 1, 0 }, | ||
211 | {"max_packet_size", "maximum size of emitted packet", offsetof(MpegTSContext, max_packet_size), AV_OPT_TYPE_INT, | ||
212 | {.i64 = 204800}, 1, INT_MAX/2, AV_OPT_FLAG_DECODING_PARAM }, | ||
213 | { NULL }, | ||
214 | }; | ||
215 | |||
216 | static const AVClass mpegts_class = { | ||
217 | .class_name = "mpegts demuxer", | ||
218 | .item_name = av_default_item_name, | ||
219 | .option = options, | ||
220 | .version = LIBAVUTIL_VERSION_INT, | ||
221 | }; | ||
222 | |||
223 | static const AVOption raw_options[] = { | ||
224 | MPEGTS_OPTIONS, | ||
225 | { "compute_pcr", "compute exact PCR for each transport stream packet", | ||
226 | offsetof(MpegTSContext, mpeg2ts_compute_pcr), AV_OPT_TYPE_BOOL, | ||
227 | { .i64 = 0 }, 0, 1, AV_OPT_FLAG_DECODING_PARAM }, | ||
228 | { NULL }, | ||
229 | }; | ||
230 | |||
231 | static const AVClass mpegtsraw_class = { | ||
232 | .class_name = "mpegtsraw demuxer", | ||
233 | .item_name = av_default_item_name, | ||
234 | .option = raw_options, | ||
235 | .version = LIBAVUTIL_VERSION_INT, | ||
236 | }; | ||
237 | |||
238 | /* TS stream handling */ | ||
239 | |||
240 | enum MpegTSState { | ||
241 | MPEGTS_HEADER = 0, | ||
242 | MPEGTS_PESHEADER, | ||
243 | MPEGTS_PESHEADER_FILL, | ||
244 | MPEGTS_PAYLOAD, | ||
245 | MPEGTS_SKIP, | ||
246 | }; | ||
247 | |||
248 | /* enough for PES header + length */ | ||
249 | #define PES_START_SIZE 6 | ||
250 | #define PES_HEADER_SIZE 9 | ||
251 | #define MAX_PES_HEADER_SIZE (9 + 255) | ||
252 | |||
253 | typedef struct PESContext { | ||
254 | int pid; | ||
255 | int pcr_pid; /**< if -1 then all packets containing PCR are considered */ | ||
256 | int stream_type; | ||
257 | MpegTSContext *ts; | ||
258 | AVFormatContext *stream; | ||
259 | AVStream *st; | ||
260 | AVStream *sub_st; /**< stream for the embedded AC3 stream in HDMV TrueHD */ | ||
261 | enum MpegTSState state; | ||
262 | /* used to get the format */ | ||
263 | int data_index; | ||
264 | int flags; /**< copied to the AVPacket flags */ | ||
265 | int PES_packet_length; | ||
266 | int pes_header_size; | ||
267 | int extended_stream_id; | ||
268 | uint8_t stream_id; | ||
269 | int64_t pts, dts; | ||
270 | int64_t ts_packet_pos; /**< position of first TS packet of this PES packet */ | ||
271 | uint8_t header[MAX_PES_HEADER_SIZE]; | ||
272 | AVBufferRef *buffer; | ||
273 | SLConfigDescr sl; | ||
274 | int merged_st; | ||
275 | } PESContext; | ||
276 | |||
277 | extern const FFInputFormat ff_mpegts_demuxer; | ||
278 | |||
279 | 484 | static struct Program * get_program(MpegTSContext *ts, unsigned int programid) | |
280 | { | ||
281 | int i; | ||
282 |
2/2✓ Branch 0 taken 429 times.
✓ Branch 1 taken 55 times.
|
484 | for (i = 0; i < ts->nb_prg; i++) { |
283 |
1/2✓ Branch 0 taken 429 times.
✗ Branch 1 not taken.
|
429 | if (ts->prg[i].id == programid) { |
284 | 429 | return &ts->prg[i]; | |
285 | } | ||
286 | } | ||
287 | 55 | return NULL; | |
288 | } | ||
289 | |||
290 | 172 | static void clear_avprogram(MpegTSContext *ts, unsigned int programid) | |
291 | { | ||
292 | 172 | AVProgram *prg = NULL; | |
293 | int i; | ||
294 | |||
295 |
1/2✓ Branch 0 taken 172 times.
✗ Branch 1 not taken.
|
172 | for (i = 0; i < ts->stream->nb_programs; i++) |
296 |
1/2✓ Branch 0 taken 172 times.
✗ Branch 1 not taken.
|
172 | if (ts->stream->programs[i]->id == programid) { |
297 | 172 | prg = ts->stream->programs[i]; | |
298 | 172 | break; | |
299 | } | ||
300 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 172 times.
|
172 | if (!prg) |
301 | ✗ | return; | |
302 | 172 | prg->nb_stream_indexes = 0; | |
303 | } | ||
304 | |||
305 | 326 | static void clear_program(struct Program *p) | |
306 | { | ||
307 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 326 times.
|
326 | if (!p) |
308 | ✗ | return; | |
309 | 326 | p->nb_pids = 0; | |
310 | 326 | p->nb_streams = 0; | |
311 | 326 | p->pmt_found = 0; | |
312 | } | ||
313 | |||
314 | 55 | static void clear_programs(MpegTSContext *ts) | |
315 | { | ||
316 | 55 | av_freep(&ts->prg); | |
317 | 55 | ts->nb_prg = 0; | |
318 | 55 | } | |
319 | |||
320 | 213 | static struct Program * add_program(MpegTSContext *ts, unsigned int programid) | |
321 | { | ||
322 | 213 | struct Program *p = get_program(ts, programid); | |
323 |
2/2✓ Branch 0 taken 158 times.
✓ Branch 1 taken 55 times.
|
213 | if (p) |
324 | 158 | return p; | |
325 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 55 times.
|
55 | if (av_reallocp_array(&ts->prg, ts->nb_prg + 1, sizeof(*ts->prg)) < 0) { |
326 | ✗ | ts->nb_prg = 0; | |
327 | ✗ | return NULL; | |
328 | } | ||
329 | 55 | p = &ts->prg[ts->nb_prg]; | |
330 | 55 | p->id = programid; | |
331 | 55 | clear_program(p); | |
332 | 55 | ts->nb_prg++; | |
333 | 55 | return p; | |
334 | } | ||
335 | |||
336 | 1314 | static void add_pid_to_program(struct Program *p, unsigned int pid) | |
337 | { | ||
338 | int i; | ||
339 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1314 times.
|
1314 | if (!p) |
340 | ✗ | return; | |
341 | |||
342 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1314 times.
|
1314 | if (p->nb_pids >= MAX_PIDS_PER_PROGRAM) |
343 | ✗ | return; | |
344 | |||
345 |
2/2✓ Branch 0 taken 1759 times.
✓ Branch 1 taken 879 times.
|
2638 | for (i = 0; i < p->nb_pids; i++) |
346 |
2/2✓ Branch 0 taken 435 times.
✓ Branch 1 taken 1324 times.
|
1759 | if (p->pids[i] == pid) |
347 | 435 | return; | |
348 | |||
349 | 879 | p->pids[p->nb_pids++] = pid; | |
350 | } | ||
351 | |||
352 | 271 | static void update_av_program_info(AVFormatContext *s, unsigned int programid, | |
353 | unsigned int pid, int version) | ||
354 | { | ||
355 | int i; | ||
356 |
1/2✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
|
271 | for (i = 0; i < s->nb_programs; i++) { |
357 | 271 | AVProgram *program = s->programs[i]; | |
358 |
1/2✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
|
271 | if (program->id == programid) { |
359 | 271 | int old_pcr_pid = program->pcr_pid, | |
360 | 271 | old_version = program->pmt_version; | |
361 | 271 | program->pcr_pid = pid; | |
362 | 271 | program->pmt_version = version; | |
363 | |||
364 |
4/4✓ Branch 0 taken 216 times.
✓ Branch 1 taken 55 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 198 times.
|
271 | if (old_version != -1 && old_version != version) { |
365 | 18 | av_log(s, AV_LOG_VERBOSE, | |
366 | "detected PMT change (program=%d, version=%d/%d, pcr_pid=0x%x/0x%x)\n", | ||
367 | programid, old_version, version, old_pcr_pid, pid); | ||
368 | } | ||
369 | 271 | break; | |
370 | } | ||
371 | } | ||
372 | 271 | } | |
373 | |||
374 | /** | ||
375 | * @brief discard_pid() decides if the pid is to be discarded according | ||
376 | * to caller's programs selection | ||
377 | * @param ts : - TS context | ||
378 | * @param pid : - pid | ||
379 | * @return 1 if the pid is only comprised in programs that have .discard=AVDISCARD_ALL | ||
380 | * 0 otherwise | ||
381 | */ | ||
382 | 36133 | static int discard_pid(MpegTSContext *ts, unsigned int pid) | |
383 | { | ||
384 | int i, j, k; | ||
385 | 36133 | int used = 0, discarded = 0; | |
386 | struct Program *p; | ||
387 | |||
388 |
2/2✓ Branch 0 taken 3009 times.
✓ Branch 1 taken 33124 times.
|
36133 | if (pid == PAT_PID) |
389 | 3009 | return 0; | |
390 | |||
391 | /* If none of the programs have .discard=AVDISCARD_ALL then there's | ||
392 | * no way we have to discard this packet */ | ||
393 |
2/2✓ Branch 0 taken 33097 times.
✓ Branch 1 taken 33124 times.
|
66221 | for (k = 0; k < ts->stream->nb_programs; k++) |
394 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 33097 times.
|
33097 | if (ts->stream->programs[k]->discard == AVDISCARD_ALL) |
395 | ✗ | break; | |
396 |
1/2✓ Branch 0 taken 33124 times.
✗ Branch 1 not taken.
|
33124 | if (k == ts->stream->nb_programs) |
397 | 33124 | return 0; | |
398 | |||
399 | ✗ | for (i = 0; i < ts->nb_prg; i++) { | |
400 | ✗ | p = &ts->prg[i]; | |
401 | ✗ | for (j = 0; j < p->nb_pids; j++) { | |
402 | ✗ | if (p->pids[j] != pid) | |
403 | ✗ | continue; | |
404 | // is program with id p->id set to be discarded? | ||
405 | ✗ | for (k = 0; k < ts->stream->nb_programs; k++) { | |
406 | ✗ | if (ts->stream->programs[k]->id == p->id) { | |
407 | ✗ | if (ts->stream->programs[k]->discard == AVDISCARD_ALL) | |
408 | ✗ | discarded++; | |
409 | else | ||
410 | ✗ | used++; | |
411 | } | ||
412 | } | ||
413 | } | ||
414 | } | ||
415 | |||
416 | ✗ | return !used && discarded; | |
417 | } | ||
418 | |||
419 | /** | ||
420 | * Assemble PES packets out of TS packets, and then call the "section_cb" | ||
421 | * function when they are complete. | ||
422 | */ | ||
423 | 6482 | static void write_section_data(MpegTSContext *ts, MpegTSFilter *tss1, | |
424 | const uint8_t *buf, int buf_size, int is_start) | ||
425 | { | ||
426 | 6482 | MpegTSSectionFilter *tss = &tss1->u.section_filter; | |
427 | 6482 | uint8_t *cur_section_buf = NULL; | |
428 | int len, offset; | ||
429 | |||
430 |
2/2✓ Branch 0 taken 6447 times.
✓ Branch 1 taken 35 times.
|
6482 | if (is_start) { |
431 | 6447 | memcpy(tss->section_buf, buf, buf_size); | |
432 | 6447 | tss->section_index = buf_size; | |
433 | 6447 | tss->section_h_size = -1; | |
434 | 6447 | tss->end_of_section_reached = 0; | |
435 | } else { | ||
436 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 35 times.
|
35 | if (tss->end_of_section_reached) |
437 | ✗ | return; | |
438 | 35 | len = MAX_SECTION_SIZE - tss->section_index; | |
439 |
1/2✓ Branch 0 taken 35 times.
✗ Branch 1 not taken.
|
35 | if (buf_size < len) |
440 | 35 | len = buf_size; | |
441 | 35 | memcpy(tss->section_buf + tss->section_index, buf, len); | |
442 | 35 | tss->section_index += len; | |
443 | } | ||
444 | |||
445 | 6482 | offset = 0; | |
446 | 6482 | cur_section_buf = tss->section_buf; | |
447 |
3/4✓ Branch 0 taken 12929 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 6690 times.
✓ Branch 3 taken 6239 times.
|
12929 | while (cur_section_buf - tss->section_buf < MAX_SECTION_SIZE && cur_section_buf[0] != 0xff) { |
448 | /* compute section length if possible */ | ||
449 |
3/4✓ Branch 0 taken 6690 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 6482 times.
✓ Branch 3 taken 208 times.
|
6690 | if (tss->section_h_size == -1 && tss->section_index - offset >= 3) { |
450 | 6482 | len = (AV_RB16(cur_section_buf + 1) & 0xfff) + 3; | |
451 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6482 times.
|
6482 | if (len > MAX_SECTION_SIZE) |
452 | ✗ | return; | |
453 | 6482 | tss->section_h_size = len; | |
454 | } | ||
455 | |||
456 |
2/2✓ Branch 0 taken 6482 times.
✓ Branch 1 taken 208 times.
|
6690 | if (tss->section_h_size != -1 && |
457 |
2/2✓ Branch 0 taken 6447 times.
✓ Branch 1 taken 35 times.
|
6482 | tss->section_index >= offset + tss->section_h_size) { |
458 | 6447 | int crc_valid = 1; | |
459 | 6447 | tss->end_of_section_reached = 1; | |
460 | |||
461 |
1/2✓ Branch 0 taken 6447 times.
✗ Branch 1 not taken.
|
6447 | if (tss->check_crc) { |
462 | 6447 | crc_valid = !av_crc(av_crc_get_table(AV_CRC_32_IEEE), -1, cur_section_buf, tss->section_h_size); | |
463 |
1/2✓ Branch 0 taken 6447 times.
✗ Branch 1 not taken.
|
6447 | if (tss->section_h_size >= 4) |
464 | 6447 | tss->crc = AV_RB32(cur_section_buf + tss->section_h_size - 4); | |
465 | |||
466 |
1/2✓ Branch 0 taken 6447 times.
✗ Branch 1 not taken.
|
6447 | if (crc_valid) { |
467 | 6447 | ts->crc_validity[ tss1->pid ] = 100; | |
468 | ✗ | }else if (ts->crc_validity[ tss1->pid ] > -10) { | |
469 | ✗ | ts->crc_validity[ tss1->pid ]--; | |
470 | }else | ||
471 | ✗ | crc_valid = 2; | |
472 | } | ||
473 |
1/2✓ Branch 0 taken 6447 times.
✗ Branch 1 not taken.
|
6447 | if (crc_valid) { |
474 | 6447 | tss->section_cb(tss1, cur_section_buf, tss->section_h_size); | |
475 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6447 times.
|
6447 | if (crc_valid != 1) |
476 | ✗ | tss->last_ver = -1; | |
477 | } | ||
478 | |||
479 | 6447 | cur_section_buf += tss->section_h_size; | |
480 | 6447 | offset += tss->section_h_size; | |
481 | 6447 | tss->section_h_size = -1; | |
482 | } else { | ||
483 | 243 | tss->section_h_size = -1; | |
484 | 243 | tss->end_of_section_reached = 0; | |
485 | 243 | break; | |
486 | } | ||
487 | } | ||
488 | } | ||
489 | |||
490 | 353 | static MpegTSFilter *mpegts_open_filter(MpegTSContext *ts, unsigned int pid, | |
491 | enum MpegTSFilterType type) | ||
492 | { | ||
493 | MpegTSFilter *filter; | ||
494 | |||
495 | 353 | av_log(ts->stream, AV_LOG_TRACE, "Filter: pid=0x%x type=%d\n", pid, type); | |
496 | |||
497 |
2/4✓ Branch 0 taken 353 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 353 times.
|
353 | if (pid >= NB_PID_MAX || ts->pids[pid]) |
498 | ✗ | return NULL; | |
499 | 353 | filter = av_mallocz(sizeof(MpegTSFilter)); | |
500 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 353 times.
|
353 | if (!filter) |
501 | ✗ | return NULL; | |
502 | 353 | ts->pids[pid] = filter; | |
503 | |||
504 | 353 | filter->type = type; | |
505 | 353 | filter->pid = pid; | |
506 | 353 | filter->es_id = -1; | |
507 | 353 | filter->last_cc = -1; | |
508 | 353 | filter->last_pcr= -1; | |
509 | |||
510 | 353 | return filter; | |
511 | } | ||
512 | |||
513 | 227 | static MpegTSFilter *mpegts_open_section_filter(MpegTSContext *ts, | |
514 | unsigned int pid, | ||
515 | SectionCallback *section_cb, | ||
516 | void *opaque, | ||
517 | int check_crc) | ||
518 | { | ||
519 | MpegTSFilter *filter; | ||
520 | MpegTSSectionFilter *sec; | ||
521 | 227 | uint8_t *section_buf = av_mallocz(MAX_SECTION_SIZE); | |
522 | |||
523 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 227 times.
|
227 | if (!section_buf) |
524 | ✗ | return NULL; | |
525 | |||
526 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 227 times.
|
227 | if (!(filter = mpegts_open_filter(ts, pid, MPEGTS_SECTION))) { |
527 | ✗ | av_free(section_buf); | |
528 | ✗ | return NULL; | |
529 | } | ||
530 | 227 | sec = &filter->u.section_filter; | |
531 | 227 | sec->section_cb = section_cb; | |
532 | 227 | sec->opaque = opaque; | |
533 | 227 | sec->section_buf = section_buf; | |
534 | 227 | sec->check_crc = check_crc; | |
535 | 227 | sec->last_ver = -1; | |
536 | |||
537 | 227 | return filter; | |
538 | } | ||
539 | |||
540 | 126 | static MpegTSFilter *mpegts_open_pes_filter(MpegTSContext *ts, unsigned int pid, | |
541 | PESCallback *pes_cb, | ||
542 | void *opaque) | ||
543 | { | ||
544 | MpegTSFilter *filter; | ||
545 | MpegTSPESFilter *pes; | ||
546 | |||
547 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 126 times.
|
126 | if (!(filter = mpegts_open_filter(ts, pid, MPEGTS_PES))) |
548 | ✗ | return NULL; | |
549 | |||
550 | 126 | pes = &filter->u.pes_filter; | |
551 | 126 | pes->pes_cb = pes_cb; | |
552 | 126 | pes->opaque = opaque; | |
553 | 126 | return filter; | |
554 | } | ||
555 | |||
556 | ✗ | static MpegTSFilter *mpegts_open_pcr_filter(MpegTSContext *ts, unsigned int pid) | |
557 | { | ||
558 | ✗ | return mpegts_open_filter(ts, pid, MPEGTS_PCR); | |
559 | } | ||
560 | |||
561 | 353 | static void mpegts_close_filter(MpegTSContext *ts, MpegTSFilter *filter) | |
562 | { | ||
563 | int pid; | ||
564 | |||
565 | 353 | pid = filter->pid; | |
566 |
2/2✓ Branch 0 taken 227 times.
✓ Branch 1 taken 126 times.
|
353 | if (filter->type == MPEGTS_SECTION) |
567 | 227 | av_freep(&filter->u.section_filter.section_buf); | |
568 |
1/2✓ Branch 0 taken 126 times.
✗ Branch 1 not taken.
|
126 | else if (filter->type == MPEGTS_PES) { |
569 | 126 | PESContext *pes = filter->u.pes_filter.opaque; | |
570 | 126 | av_buffer_unref(&pes->buffer); | |
571 | /* referenced private data will be freed later in | ||
572 | * avformat_close_input (pes->st->priv_data == pes) */ | ||
573 |
4/4✓ Branch 0 taken 109 times.
✓ Branch 1 taken 17 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 105 times.
|
126 | if (!pes->st || pes->merged_st) { |
574 | 21 | av_freep(&filter->u.pes_filter.opaque); | |
575 | } | ||
576 | } | ||
577 | |||
578 | 353 | av_free(filter); | |
579 | 353 | ts->pids[pid] = NULL; | |
580 | 353 | } | |
581 | |||
582 | 70839 | static int analyze(const uint8_t *buf, int size, int packet_size, | |
583 | int probe) | ||
584 | { | ||
585 | int stat[TS_MAX_PACKET_SIZE]; | ||
586 | 70839 | int stat_all = 0; | |
587 | int i; | ||
588 | 70839 | int best_score = 0; | |
589 | |||
590 | 70839 | memset(stat, 0, packet_size * sizeof(*stat)); | |
591 | |||
592 |
2/2✓ Branch 0 taken 1077570587 times.
✓ Branch 1 taken 70839 times.
|
1077641426 | for (i = 0; i < size - 3; i++) { |
593 |
2/2✓ Branch 0 taken 1906571 times.
✓ Branch 1 taken 1075664016 times.
|
1077570587 | if (buf[i] == 0x47) { |
594 | 1906571 | int pid = AV_RB16(buf+1) & 0x1FFF; | |
595 | 1906571 | int asc = buf[i + 3] & 0x30; | |
596 |
6/6✓ Branch 0 taken 1884614 times.
✓ Branch 1 taken 21957 times.
✓ Branch 2 taken 1882772 times.
✓ Branch 3 taken 1842 times.
✓ Branch 4 taken 1542453 times.
✓ Branch 5 taken 340319 times.
|
1906571 | if (!probe || pid == 0x1FFF || asc) { |
597 | 1566252 | int x = i % packet_size; | |
598 | 1566252 | stat[x]++; | |
599 | 1566252 | stat_all++; | |
600 |
2/2✓ Branch 0 taken 118765 times.
✓ Branch 1 taken 1447487 times.
|
1566252 | if (stat[x] > best_score) { |
601 | 118765 | best_score = stat[x]; | |
602 | } | ||
603 | } | ||
604 | } | ||
605 | } | ||
606 | |||
607 | 70839 | return best_score - FFMAX(stat_all - 10*best_score, 0)/10; | |
608 | } | ||
609 | |||
610 | /* autodetect fec presence */ | ||
611 | 114 | static int get_packet_size(AVFormatContext* s) | |
612 | { | ||
613 | int score, fec_score, dvhs_score; | ||
614 | int margin; | ||
615 | int ret; | ||
616 | |||
617 | /*init buffer to store stream for probing */ | ||
618 | 114 | uint8_t buf[PROBE_PACKET_MAX_BUF] = {0}; | |
619 | 114 | int buf_size = 0; | |
620 | 114 | int max_iterations = 16; | |
621 | |||
622 |
2/4✓ Branch 0 taken 114 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 114 times.
✗ Branch 3 not taken.
|
114 | while (buf_size < PROBE_PACKET_MAX_BUF && max_iterations--) { |
623 | 114 | ret = avio_read_partial(s->pb, buf + buf_size, PROBE_PACKET_MAX_BUF - buf_size); | |
624 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 114 times.
|
114 | if (ret < 0) |
625 | ✗ | return AVERROR_INVALIDDATA; | |
626 | 114 | buf_size += ret; | |
627 | |||
628 | 114 | score = analyze(buf, buf_size, TS_PACKET_SIZE, 0); | |
629 | 114 | dvhs_score = analyze(buf, buf_size, TS_DVHS_PACKET_SIZE, 0); | |
630 | 114 | fec_score = analyze(buf, buf_size, TS_FEC_PACKET_SIZE, 0); | |
631 | 114 | av_log(s, AV_LOG_TRACE, "Probe: %d, score: %d, dvhs_score: %d, fec_score: %d \n", | |
632 | buf_size, score, dvhs_score, fec_score); | ||
633 | |||
634 | 114 | margin = mid_pred(score, fec_score, dvhs_score); | |
635 | |||
636 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 113 times.
|
114 | if (buf_size < PROBE_PACKET_MAX_BUF) |
637 | 1 | margin += PROBE_PACKET_MARGIN; /*if buffer not filled */ | |
638 | |||
639 |
1/2✓ Branch 0 taken 114 times.
✗ Branch 1 not taken.
|
114 | if (score > margin) |
640 | 114 | return TS_PACKET_SIZE; | |
641 | ✗ | else if (dvhs_score > margin) | |
642 | ✗ | return TS_DVHS_PACKET_SIZE; | |
643 | ✗ | else if (fec_score > margin) | |
644 | ✗ | return TS_FEC_PACKET_SIZE; | |
645 | } | ||
646 | ✗ | return AVERROR_INVALIDDATA; | |
647 | } | ||
648 | |||
649 | typedef struct SectionHeader { | ||
650 | uint8_t tid; | ||
651 | uint16_t id; | ||
652 | uint8_t version; | ||
653 | uint8_t current_next; | ||
654 | uint8_t sec_num; | ||
655 | uint8_t last_sec_num; | ||
656 | } SectionHeader; | ||
657 | |||
658 | 5767 | static int skip_identical(const SectionHeader *h, MpegTSSectionFilter *tssf) | |
659 | { | ||
660 |
3/4✓ Branch 0 taken 5183 times.
✓ Branch 1 taken 584 times.
✓ Branch 2 taken 5183 times.
✗ Branch 3 not taken.
|
5767 | if (h->version == tssf->last_ver && tssf->last_crc == tssf->crc) |
661 | 5183 | return 1; | |
662 | |||
663 | 584 | tssf->last_ver = h->version; | |
664 | 584 | tssf->last_crc = tssf->crc; | |
665 | |||
666 | 584 | return 0; | |
667 | } | ||
668 | |||
669 | 31558 | static inline int get8(const uint8_t **pp, const uint8_t *p_end) | |
670 | { | ||
671 | const uint8_t *p; | ||
672 | int c; | ||
673 | |||
674 | 31558 | p = *pp; | |
675 |
2/2✓ Branch 0 taken 1461 times.
✓ Branch 1 taken 30097 times.
|
31558 | if (p >= p_end) |
676 | 1461 | return AVERROR_INVALIDDATA; | |
677 | 30097 | c = *p++; | |
678 | 30097 | *pp = p; | |
679 | 30097 | return c; | |
680 | } | ||
681 | |||
682 | 9707 | static inline int get16(const uint8_t **pp, const uint8_t *p_end) | |
683 | { | ||
684 | const uint8_t *p; | ||
685 | int c; | ||
686 | |||
687 | 9707 | p = *pp; | |
688 |
2/2✓ Branch 0 taken 313 times.
✓ Branch 1 taken 9394 times.
|
9707 | if (1 >= p_end - p) |
689 | 313 | return AVERROR_INVALIDDATA; | |
690 | 9394 | c = AV_RB16(p); | |
691 | 9394 | p += 2; | |
692 | 9394 | *pp = p; | |
693 | 9394 | return c; | |
694 | } | ||
695 | |||
696 | /* read and allocate a DVB string preceded by its length */ | ||
697 | 200 | static char *getstr8(const uint8_t **pp, const uint8_t *p_end) | |
698 | { | ||
699 | int len; | ||
700 | const uint8_t *p; | ||
701 | char *str; | ||
702 | |||
703 | 200 | p = *pp; | |
704 | 200 | len = get8(&p, p_end); | |
705 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
|
200 | if (len < 0) |
706 | ✗ | return NULL; | |
707 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
|
200 | if (len > p_end - p) |
708 | ✗ | return NULL; | |
709 | #if CONFIG_ICONV | ||
710 |
1/2✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
|
200 | if (len) { |
711 | 200 | const char *encodings[] = { | |
712 | "ISO6937", "ISO-8859-5", "ISO-8859-6", "ISO-8859-7", | ||
713 | "ISO-8859-8", "ISO-8859-9", "ISO-8859-10", "ISO-8859-11", | ||
714 | "", "ISO-8859-13", "ISO-8859-14", "ISO-8859-15", "", "", "", "", | ||
715 | "", "UCS-2BE", "KSC_5601", "GB2312", "UCS-2BE", "UTF-8", "", "", | ||
716 | "", "", "", "", "", "", "", "" | ||
717 | }; | ||
718 | iconv_t cd; | ||
719 | char *in, *out; | ||
720 | 200 | size_t inlen = len, outlen = inlen * 6 + 1; | |
721 |
2/12✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 200 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
✗ Branch 10 not taken.
✗ Branch 11 not taken.
|
200 | if (len >= 3 && p[0] == 0x10 && !p[1] && p[2] && p[2] <= 0xf && p[2] != 0xc) { |
722 | char iso8859[12]; | ||
723 | ✗ | snprintf(iso8859, sizeof(iso8859), "ISO-8859-%d", p[2]); | |
724 | ✗ | inlen -= 3; | |
725 | ✗ | in = (char *)p + 3; | |
726 | ✗ | cd = iconv_open("UTF-8", iso8859); | |
727 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 192 times.
|
200 | } else if (p[0] < 0x20) { |
728 | 8 | inlen -= 1; | |
729 | 8 | in = (char *)p + 1; | |
730 | 8 | cd = iconv_open("UTF-8", encodings[*p]); | |
731 | } else { | ||
732 | 192 | in = (char *)p; | |
733 | 192 | cd = iconv_open("UTF-8", encodings[0]); | |
734 | } | ||
735 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
|
200 | if (cd == (iconv_t)-1) |
736 | ✗ | goto no_iconv; | |
737 | 200 | str = out = av_malloc(outlen); | |
738 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
|
200 | if (!str) { |
739 | ✗ | iconv_close(cd); | |
740 | 200 | return NULL; | |
741 | } | ||
742 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 200 times.
|
200 | if (iconv(cd, &in, &inlen, &out, &outlen) == -1) { |
743 | ✗ | iconv_close(cd); | |
744 | ✗ | av_freep(&str); | |
745 | ✗ | goto no_iconv; | |
746 | } | ||
747 | 200 | iconv_close(cd); | |
748 | 200 | *out = 0; | |
749 | 200 | *pp = p + len; | |
750 | 200 | return str; | |
751 | } | ||
752 | ✗ | no_iconv: | |
753 | #endif | ||
754 | ✗ | str = av_malloc(len + 1); | |
755 | ✗ | if (!str) | |
756 | ✗ | return NULL; | |
757 | ✗ | memcpy(str, p, len); | |
758 | ✗ | str[len] = '\0'; | |
759 | ✗ | p += len; | |
760 | ✗ | *pp = p; | |
761 | ✗ | return str; | |
762 | } | ||
763 | |||
764 | 6445 | static int parse_section_header(SectionHeader *h, | |
765 | const uint8_t **pp, const uint8_t *p_end) | ||
766 | { | ||
767 | int val; | ||
768 | |||
769 | 6445 | val = get8(pp, p_end); | |
770 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6445 times.
|
6445 | if (val < 0) |
771 | ✗ | return val; | |
772 | 6445 | h->tid = val; | |
773 | 6445 | *pp += 2; | |
774 | 6445 | val = get16(pp, p_end); | |
775 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6445 times.
|
6445 | if (val < 0) |
776 | ✗ | return val; | |
777 | 6445 | h->id = val; | |
778 | 6445 | val = get8(pp, p_end); | |
779 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6445 times.
|
6445 | if (val < 0) |
780 | ✗ | return val; | |
781 | 6445 | h->version = (val >> 1) & 0x1f; | |
782 | 6445 | h->current_next = val & 0x01; | |
783 | 6445 | val = get8(pp, p_end); | |
784 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6445 times.
|
6445 | if (val < 0) |
785 | ✗ | return val; | |
786 | 6445 | h->sec_num = val; | |
787 | 6445 | val = get8(pp, p_end); | |
788 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6445 times.
|
6445 | if (val < 0) |
789 | ✗ | return val; | |
790 | 6445 | h->last_sec_num = val; | |
791 | 6445 | return 0; | |
792 | } | ||
793 | |||
794 | typedef struct StreamType { | ||
795 | uint32_t stream_type; | ||
796 | enum AVMediaType codec_type; | ||
797 | enum AVCodecID codec_id; | ||
798 | } StreamType; | ||
799 | |||
800 | static const StreamType ISO_types[] = { | ||
801 | { 0x01, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_MPEG2VIDEO }, | ||
802 | { 0x02, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_MPEG2VIDEO }, | ||
803 | { 0x03, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_MP3 }, | ||
804 | { 0x04, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_MP3 }, | ||
805 | { 0x0f, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AAC }, | ||
806 | { 0x10, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_MPEG4 }, | ||
807 | /* Makito encoder sets stream type 0x11 for AAC, | ||
808 | * so auto-detect LOAS/LATM instead of hardcoding it. */ | ||
809 | #if !CONFIG_LOAS_DEMUXER | ||
810 | { 0x11, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AAC_LATM }, /* LATM syntax */ | ||
811 | #endif | ||
812 | { 0x1b, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_H264 }, | ||
813 | { 0x1c, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AAC }, | ||
814 | { 0x20, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_H264 }, | ||
815 | { 0x21, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_JPEG2000 }, | ||
816 | { 0x24, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_HEVC }, | ||
817 | { 0x33, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_VVC }, | ||
818 | { 0x42, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_CAVS }, | ||
819 | { 0xd1, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_DIRAC }, | ||
820 | { 0xd2, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_AVS2 }, | ||
821 | { 0xd4, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_AVS3 }, | ||
822 | { 0xea, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_VC1 }, | ||
823 | { 0 }, | ||
824 | }; | ||
825 | |||
826 | static const StreamType HDMV_types[] = { | ||
827 | { 0x80, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_PCM_BLURAY }, | ||
828 | { 0x81, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC3 }, | ||
829 | { 0x82, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
830 | { 0x83, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_TRUEHD }, | ||
831 | { 0x84, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_EAC3 }, | ||
832 | { 0x85, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, /* DTS HD */ | ||
833 | { 0x86, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, /* DTS HD MASTER*/ | ||
834 | { 0xa1, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_EAC3 }, /* E-AC3 Secondary Audio */ | ||
835 | { 0xa2, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, /* DTS Express Secondary Audio */ | ||
836 | { 0x90, AVMEDIA_TYPE_SUBTITLE, AV_CODEC_ID_HDMV_PGS_SUBTITLE }, | ||
837 | { 0x92, AVMEDIA_TYPE_SUBTITLE, AV_CODEC_ID_HDMV_TEXT_SUBTITLE }, | ||
838 | { 0 }, | ||
839 | }; | ||
840 | |||
841 | /* SCTE types */ | ||
842 | static const StreamType SCTE_types[] = { | ||
843 | { 0x86, AVMEDIA_TYPE_DATA, AV_CODEC_ID_SCTE_35 }, | ||
844 | { 0 }, | ||
845 | }; | ||
846 | |||
847 | /* ATSC ? */ | ||
848 | static const StreamType MISC_types[] = { | ||
849 | { 0x81, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC3 }, | ||
850 | { 0x8a, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
851 | { 0 }, | ||
852 | }; | ||
853 | |||
854 | /* HLS Sample Encryption Types */ | ||
855 | static const StreamType HLS_SAMPLE_ENC_types[] = { | ||
856 | { 0xdb, AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_H264}, | ||
857 | { 0xcf, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AAC }, | ||
858 | { 0xc1, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC3 }, | ||
859 | { 0xc2, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_EAC3}, | ||
860 | { 0 }, | ||
861 | }; | ||
862 | |||
863 | static const StreamType REGD_types[] = { | ||
864 | { MKTAG('d', 'r', 'a', 'c'), AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_DIRAC }, | ||
865 | { MKTAG('A', 'C', '-', '3'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC3 }, | ||
866 | { MKTAG('A', 'C', '-', '4'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC4 }, | ||
867 | { MKTAG('B', 'S', 'S', 'D'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_S302M }, | ||
868 | { MKTAG('D', 'T', 'S', '1'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
869 | { MKTAG('D', 'T', 'S', '2'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
870 | { MKTAG('D', 'T', 'S', '3'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
871 | { MKTAG('E', 'A', 'C', '3'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_EAC3 }, | ||
872 | { MKTAG('H', 'E', 'V', 'C'), AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_HEVC }, | ||
873 | { MKTAG('V', 'V', 'C', ' '), AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_VVC }, | ||
874 | { MKTAG('K', 'L', 'V', 'A'), AVMEDIA_TYPE_DATA, AV_CODEC_ID_SMPTE_KLV }, | ||
875 | { MKTAG('V', 'A', 'N', 'C'), AVMEDIA_TYPE_DATA, AV_CODEC_ID_SMPTE_2038 }, | ||
876 | { MKTAG('I', 'D', '3', ' '), AVMEDIA_TYPE_DATA, AV_CODEC_ID_TIMED_ID3 }, | ||
877 | { MKTAG('V', 'C', '-', '1'), AVMEDIA_TYPE_VIDEO, AV_CODEC_ID_VC1 }, | ||
878 | { MKTAG('O', 'p', 'u', 's'), AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_OPUS }, | ||
879 | { 0 }, | ||
880 | }; | ||
881 | |||
882 | static const StreamType METADATA_types[] = { | ||
883 | { MKTAG('K','L','V','A'), AVMEDIA_TYPE_DATA, AV_CODEC_ID_SMPTE_KLV }, | ||
884 | { MKTAG('I','D','3',' '), AVMEDIA_TYPE_DATA, AV_CODEC_ID_TIMED_ID3 }, | ||
885 | { 0 }, | ||
886 | }; | ||
887 | |||
888 | /* descriptor present */ | ||
889 | static const StreamType DESC_types[] = { | ||
890 | { 0x6a, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_AC3 }, /* AC-3 descriptor */ | ||
891 | { 0x7a, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_EAC3 }, /* E-AC-3 descriptor */ | ||
892 | { 0x7b, AVMEDIA_TYPE_AUDIO, AV_CODEC_ID_DTS }, | ||
893 | { 0x56, AVMEDIA_TYPE_SUBTITLE, AV_CODEC_ID_DVB_TELETEXT }, | ||
894 | { 0x59, AVMEDIA_TYPE_SUBTITLE, AV_CODEC_ID_DVB_SUBTITLE }, /* subtitling descriptor */ | ||
895 | { 0 }, | ||
896 | }; | ||
897 | |||
898 | 165 | static void mpegts_find_stream_type(AVStream *st, | |
899 | uint32_t stream_type, | ||
900 | const StreamType *types) | ||
901 | { | ||
902 | 165 | FFStream *const sti = ffstream(st); | |
903 |
2/2✓ Branch 0 taken 894 times.
✓ Branch 1 taken 53 times.
|
947 | for (; types->stream_type; types++) |
904 |
2/2✓ Branch 0 taken 112 times.
✓ Branch 1 taken 782 times.
|
894 | if (stream_type == types->stream_type) { |
905 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 102 times.
|
112 | if (st->codecpar->codec_type != types->codec_type || |
906 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 1 times.
|
10 | st->codecpar->codec_id != types->codec_id) { |
907 | 111 | st->codecpar->codec_type = types->codec_type; | |
908 | 111 | st->codecpar->codec_id = types->codec_id; | |
909 | 111 | sti->need_context_update = 1; | |
910 | } | ||
911 | 112 | sti->request_probe = 0; | |
912 | 112 | return; | |
913 | } | ||
914 | } | ||
915 | |||
916 | 105 | static int mpegts_set_stream_info(AVStream *st, PESContext *pes, | |
917 | uint32_t stream_type, uint32_t prog_reg_desc) | ||
918 | { | ||
919 | 105 | FFStream *const sti = ffstream(st); | |
920 | 105 | int old_codec_type = st->codecpar->codec_type; | |
921 | 105 | int old_codec_id = st->codecpar->codec_id; | |
922 | 105 | int old_codec_tag = st->codecpar->codec_tag; | |
923 | |||
924 | 105 | avpriv_set_pts_info(st, 33, 1, 90000); | |
925 | 105 | st->priv_data = pes; | |
926 | 105 | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
927 | 105 | st->codecpar->codec_id = AV_CODEC_ID_NONE; | |
928 | 105 | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
929 | 105 | pes->st = st; | |
930 | 105 | pes->stream_type = stream_type; | |
931 | |||
932 | 105 | av_log(pes->stream, AV_LOG_DEBUG, | |
933 | "stream=%d stream_type=%x pid=%x prog_reg_desc=%.4s\n", | ||
934 | st->index, pes->stream_type, pes->pid, (char *)&prog_reg_desc); | ||
935 | |||
936 | 105 | st->codecpar->codec_tag = pes->stream_type; | |
937 | |||
938 | 105 | mpegts_find_stream_type(st, pes->stream_type, ISO_types); | |
939 |
4/4✓ Branch 0 taken 101 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 98 times.
|
105 | if (pes->stream_type == 4 || pes->stream_type == 0x0f) |
940 | 7 | sti->request_probe = 50; | |
941 |
1/2✓ Branch 0 taken 105 times.
✗ Branch 1 not taken.
|
105 | if ((prog_reg_desc == AV_RL32("HDMV") || |
942 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 105 times.
|
105 | prog_reg_desc == AV_RL32("HDPR")) && |
943 | ✗ | st->codecpar->codec_id == AV_CODEC_ID_NONE) { | |
944 | ✗ | mpegts_find_stream_type(st, pes->stream_type, HDMV_types); | |
945 | ✗ | if (pes->stream_type == 0x83) { | |
946 | // HDMV TrueHD streams also contain an AC3 coded version of the | ||
947 | // audio track - add a second stream for this | ||
948 | AVStream *sub_st; | ||
949 | // priv_data cannot be shared between streams | ||
950 | ✗ | PESContext *sub_pes = av_memdup(pes, sizeof(*sub_pes)); | |
951 | ✗ | if (!sub_pes) | |
952 | ✗ | return AVERROR(ENOMEM); | |
953 | |||
954 | ✗ | sub_st = avformat_new_stream(pes->stream, NULL); | |
955 | ✗ | if (!sub_st) { | |
956 | ✗ | av_free(sub_pes); | |
957 | ✗ | return AVERROR(ENOMEM); | |
958 | } | ||
959 | |||
960 | ✗ | sub_st->id = pes->pid; | |
961 | ✗ | avpriv_set_pts_info(sub_st, 33, 1, 90000); | |
962 | ✗ | sub_st->priv_data = sub_pes; | |
963 | ✗ | sub_st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; | |
964 | ✗ | sub_st->codecpar->codec_id = AV_CODEC_ID_AC3; | |
965 | ✗ | ffstream(sub_st)->need_parsing = AVSTREAM_PARSE_FULL; | |
966 | ✗ | sub_pes->sub_st = pes->sub_st = sub_st; | |
967 | } | ||
968 | } | ||
969 |
2/2✓ Branch 0 taken 29 times.
✓ Branch 1 taken 76 times.
|
105 | if (st->codecpar->codec_id == AV_CODEC_ID_NONE) |
970 | 29 | mpegts_find_stream_type(st, pes->stream_type, MISC_types); | |
971 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 97 times.
|
105 | if (st->codecpar->codec_id == AV_CODEC_ID_NONE) |
972 | 8 | mpegts_find_stream_type(st, pes->stream_type, HLS_SAMPLE_ENC_types); | |
973 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 97 times.
|
105 | if (st->codecpar->codec_id == AV_CODEC_ID_NONE) { |
974 | 8 | st->codecpar->codec_id = old_codec_id; | |
975 | 8 | st->codecpar->codec_type = old_codec_type; | |
976 | } | ||
977 |
2/2✓ Branch 0 taken 97 times.
✓ Branch 1 taken 8 times.
|
105 | if ((st->codecpar->codec_id == AV_CODEC_ID_NONE || |
978 |
3/4✓ Branch 0 taken 7 times.
✓ Branch 1 taken 90 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 7 times.
|
97 | (sti->request_probe > 0 && sti->request_probe < AVPROBE_SCORE_STREAM_RETRY / 5)) && |
979 |
3/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 5 times.
✓ Branch 3 taken 3 times.
|
8 | sti->probe_packets > 0 && |
980 | stream_type == STREAM_TYPE_PRIVATE_DATA) { | ||
981 | 5 | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
982 | 5 | st->codecpar->codec_id = AV_CODEC_ID_BIN_DATA; | |
983 | 5 | sti->request_probe = AVPROBE_SCORE_STREAM_RETRY / 5; | |
984 | } | ||
985 | |||
986 | /* queue a context update if properties changed */ | ||
987 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 102 times.
|
105 | if (old_codec_type != st->codecpar->codec_type || |
988 |
1/2✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
|
3 | old_codec_id != st->codecpar->codec_id || |
989 |
1/2✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
|
3 | old_codec_tag != st->codecpar->codec_tag) |
990 | 105 | sti->need_context_update = 1; | |
991 | |||
992 | 105 | return 0; | |
993 | } | ||
994 | |||
995 | 55796 | static void reset_pes_packet_state(PESContext *pes) | |
996 | { | ||
997 | 55796 | pes->pts = AV_NOPTS_VALUE; | |
998 | 55796 | pes->dts = AV_NOPTS_VALUE; | |
999 | 55796 | pes->data_index = 0; | |
1000 | 55796 | pes->flags = 0; | |
1001 | 55796 | av_buffer_unref(&pes->buffer); | |
1002 | 55796 | } | |
1003 | |||
1004 | 12 | static void new_data_packet(const uint8_t *buffer, int len, AVPacket *pkt) | |
1005 | { | ||
1006 | 12 | av_packet_unref(pkt); | |
1007 | 12 | pkt->data = (uint8_t *)buffer; | |
1008 | 12 | pkt->size = len; | |
1009 | 12 | } | |
1010 | |||
1011 | 28973 | static int new_pes_packet(PESContext *pes, AVPacket *pkt) | |
1012 | { | ||
1013 | uint8_t *sd; | ||
1014 | |||
1015 | 28973 | av_packet_unref(pkt); | |
1016 | |||
1017 | 28973 | pkt->buf = pes->buffer; | |
1018 | 28973 | pkt->data = pes->buffer->data; | |
1019 | 28973 | pkt->size = pes->data_index; | |
1020 | |||
1021 |
2/2✓ Branch 0 taken 26050 times.
✓ Branch 1 taken 2923 times.
|
28973 | if (pes->PES_packet_length && |
1022 |
2/2✓ Branch 0 taken 92 times.
✓ Branch 1 taken 25958 times.
|
26050 | pes->pes_header_size + pes->data_index != pes->PES_packet_length + |
1023 | PES_START_SIZE) { | ||
1024 | 92 | av_log(pes->stream, AV_LOG_WARNING, "PES packet size mismatch\n"); | |
1025 | 92 | pes->flags |= AV_PKT_FLAG_CORRUPT; | |
1026 | } | ||
1027 | 28973 | memset(pkt->data + pkt->size, 0, AV_INPUT_BUFFER_PADDING_SIZE); | |
1028 | |||
1029 | // Separate out the AC3 substream from an HDMV combined TrueHD/AC3 PID | ||
1030 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 28973 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
28973 | if (pes->sub_st && pes->stream_type == 0x83 && pes->extended_stream_id == 0x76) |
1031 | ✗ | pkt->stream_index = pes->sub_st->index; | |
1032 | else | ||
1033 | 28973 | pkt->stream_index = pes->st->index; | |
1034 | 28973 | pkt->pts = pes->pts; | |
1035 | 28973 | pkt->dts = pes->dts; | |
1036 | /* store position of first TS packet of this PES packet */ | ||
1037 | 28973 | pkt->pos = pes->ts_packet_pos; | |
1038 | 28973 | pkt->flags = pes->flags; | |
1039 | |||
1040 | 28973 | pes->buffer = NULL; | |
1041 | 28973 | reset_pes_packet_state(pes); | |
1042 | |||
1043 | 28973 | sd = av_packet_new_side_data(pkt, AV_PKT_DATA_MPEGTS_STREAM_ID, 1); | |
1044 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 28973 times.
|
28973 | if (!sd) |
1045 | ✗ | return AVERROR(ENOMEM); | |
1046 | 28973 | *sd = pes->stream_id; | |
1047 | |||
1048 | 28973 | return 0; | |
1049 | } | ||
1050 | |||
1051 | ✗ | static uint64_t get_ts64(GetBitContext *gb, int bits) | |
1052 | { | ||
1053 | ✗ | if (get_bits_left(gb) < bits) | |
1054 | ✗ | return AV_NOPTS_VALUE; | |
1055 | ✗ | return get_bits64(gb, bits); | |
1056 | } | ||
1057 | |||
1058 | ✗ | static int read_sl_header(PESContext *pes, SLConfigDescr *sl, | |
1059 | const uint8_t *buf, int buf_size) | ||
1060 | { | ||
1061 | GetBitContext gb; | ||
1062 | ✗ | int au_start_flag = 0, au_end_flag = 0, ocr_flag = 0, idle_flag = 0; | |
1063 | ✗ | int padding_flag = 0, padding_bits = 0, inst_bitrate_flag = 0; | |
1064 | ✗ | int dts_flag = -1, cts_flag = -1; | |
1065 | ✗ | int64_t dts = AV_NOPTS_VALUE, cts = AV_NOPTS_VALUE; | |
1066 | uint8_t buf_padded[128 + AV_INPUT_BUFFER_PADDING_SIZE]; | ||
1067 | ✗ | int buf_padded_size = FFMIN(buf_size, sizeof(buf_padded) - AV_INPUT_BUFFER_PADDING_SIZE); | |
1068 | |||
1069 | ✗ | memcpy(buf_padded, buf, buf_padded_size); | |
1070 | |||
1071 | ✗ | init_get_bits(&gb, buf_padded, buf_padded_size * 8); | |
1072 | |||
1073 | ✗ | if (sl->use_au_start) | |
1074 | ✗ | au_start_flag = get_bits1(&gb); | |
1075 | ✗ | if (sl->use_au_end) | |
1076 | ✗ | au_end_flag = get_bits1(&gb); | |
1077 | ✗ | if (!sl->use_au_start && !sl->use_au_end) | |
1078 | ✗ | au_start_flag = au_end_flag = 1; | |
1079 | ✗ | if (sl->ocr_len > 0) | |
1080 | ✗ | ocr_flag = get_bits1(&gb); | |
1081 | ✗ | if (sl->use_idle) | |
1082 | ✗ | idle_flag = get_bits1(&gb); | |
1083 | ✗ | if (sl->use_padding) | |
1084 | ✗ | padding_flag = get_bits1(&gb); | |
1085 | ✗ | if (padding_flag) | |
1086 | ✗ | padding_bits = get_bits(&gb, 3); | |
1087 | |||
1088 | ✗ | if (!idle_flag && (!padding_flag || padding_bits != 0)) { | |
1089 | ✗ | if (sl->packet_seq_num_len) | |
1090 | ✗ | skip_bits_long(&gb, sl->packet_seq_num_len); | |
1091 | ✗ | if (sl->degr_prior_len) | |
1092 | ✗ | if (get_bits1(&gb)) | |
1093 | ✗ | skip_bits(&gb, sl->degr_prior_len); | |
1094 | ✗ | if (ocr_flag) | |
1095 | ✗ | skip_bits_long(&gb, sl->ocr_len); | |
1096 | ✗ | if (au_start_flag) { | |
1097 | ✗ | if (sl->use_rand_acc_pt) | |
1098 | ✗ | get_bits1(&gb); | |
1099 | ✗ | if (sl->au_seq_num_len > 0) | |
1100 | ✗ | skip_bits_long(&gb, sl->au_seq_num_len); | |
1101 | ✗ | if (sl->use_timestamps) { | |
1102 | ✗ | dts_flag = get_bits1(&gb); | |
1103 | ✗ | cts_flag = get_bits1(&gb); | |
1104 | } | ||
1105 | } | ||
1106 | ✗ | if (sl->inst_bitrate_len) | |
1107 | ✗ | inst_bitrate_flag = get_bits1(&gb); | |
1108 | ✗ | if (dts_flag == 1) | |
1109 | ✗ | dts = get_ts64(&gb, sl->timestamp_len); | |
1110 | ✗ | if (cts_flag == 1) | |
1111 | ✗ | cts = get_ts64(&gb, sl->timestamp_len); | |
1112 | ✗ | if (sl->au_len > 0) | |
1113 | ✗ | skip_bits_long(&gb, sl->au_len); | |
1114 | ✗ | if (inst_bitrate_flag) | |
1115 | ✗ | skip_bits_long(&gb, sl->inst_bitrate_len); | |
1116 | } | ||
1117 | |||
1118 | ✗ | if (dts != AV_NOPTS_VALUE) | |
1119 | ✗ | pes->dts = dts; | |
1120 | ✗ | if (cts != AV_NOPTS_VALUE) | |
1121 | ✗ | pes->pts = cts; | |
1122 | |||
1123 | ✗ | if (sl->timestamp_len && sl->timestamp_res) | |
1124 | ✗ | avpriv_set_pts_info(pes->st, sl->timestamp_len, 1, sl->timestamp_res); | |
1125 | |||
1126 | ✗ | return (get_bits_count(&gb) + 7) >> 3; | |
1127 | } | ||
1128 | |||
1129 | 29074 | static AVBufferRef *buffer_pool_get(MpegTSContext *ts, int size) | |
1130 | { | ||
1131 | 29074 | int index = av_log2(size + AV_INPUT_BUFFER_PADDING_SIZE); | |
1132 |
2/2✓ Branch 0 taken 158 times.
✓ Branch 1 taken 28916 times.
|
29074 | if (!ts->pools[index]) { |
1133 |
2/2✓ Branch 0 taken 133 times.
✓ Branch 1 taken 25 times.
|
158 | int pool_size = FFMIN(ts->max_packet_size + AV_INPUT_BUFFER_PADDING_SIZE, 2 << index); |
1134 | 158 | ts->pools[index] = av_buffer_pool_init(pool_size, NULL); | |
1135 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 158 times.
|
158 | if (!ts->pools[index]) |
1136 | ✗ | return NULL; | |
1137 | } | ||
1138 | 29074 | return av_buffer_pool_get(ts->pools[index]); | |
1139 | } | ||
1140 | |||
1141 | /* return non zero if a packet could be constructed */ | ||
1142 | 374489 | static int mpegts_push_data(MpegTSFilter *filter, | |
1143 | const uint8_t *buf, int buf_size, int is_start, | ||
1144 | int64_t pos) | ||
1145 | { | ||
1146 | 374489 | PESContext *pes = filter->u.pes_filter.opaque; | |
1147 | 374489 | MpegTSContext *ts = pes->ts; | |
1148 | const uint8_t *p; | ||
1149 | int ret, len; | ||
1150 | |||
1151 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 374489 times.
|
374489 | if (!ts->pkt) |
1152 | ✗ | return 0; | |
1153 | |||
1154 |
2/2✓ Branch 0 taken 29674 times.
✓ Branch 1 taken 344815 times.
|
374489 | if (is_start) { |
1155 |
3/4✓ Branch 0 taken 2851 times.
✓ Branch 1 taken 26823 times.
✓ Branch 2 taken 2851 times.
✗ Branch 3 not taken.
|
29674 | if (pes->state == MPEGTS_PAYLOAD && pes->data_index > 0) { |
1156 | 2851 | ret = new_pes_packet(pes, ts->pkt); | |
1157 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2851 times.
|
2851 | if (ret < 0) |
1158 | ✗ | return ret; | |
1159 | 2851 | ts->stop_parse = 1; | |
1160 | } else { | ||
1161 | 26823 | reset_pes_packet_state(pes); | |
1162 | } | ||
1163 | 29674 | pes->state = MPEGTS_HEADER; | |
1164 | 29674 | pes->ts_packet_pos = pos; | |
1165 | } | ||
1166 | 374489 | p = buf; | |
1167 |
2/2✓ Branch 0 taken 462309 times.
✓ Branch 1 taken 374489 times.
|
1211287 | while (buf_size > 0) { |
1168 |
5/6✓ Branch 0 taken 29674 times.
✓ Branch 1 taken 29074 times.
✓ Branch 2 taken 29074 times.
✓ Branch 3 taken 355945 times.
✓ Branch 4 taken 18542 times.
✗ Branch 5 not taken.
|
462309 | switch (pes->state) { |
1169 | 29674 | case MPEGTS_HEADER: | |
1170 | 29674 | len = PES_START_SIZE - pes->data_index; | |
1171 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 29672 times.
|
29674 | if (len > buf_size) |
1172 | 2 | len = buf_size; | |
1173 | 29674 | memcpy(pes->header + pes->data_index, p, len); | |
1174 | 29674 | pes->data_index += len; | |
1175 | 29674 | p += len; | |
1176 | 29674 | buf_size -= len; | |
1177 |
2/2✓ Branch 0 taken 29672 times.
✓ Branch 1 taken 2 times.
|
29674 | if (pes->data_index == PES_START_SIZE) { |
1178 | /* we got all the PES or section header. We can now | ||
1179 | * decide */ | ||
1180 |
3/4✓ Branch 0 taken 29665 times.
✓ Branch 1 taken 7 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 29665 times.
|
29672 | if (pes->header[0] == 0x00 && pes->header[1] == 0x00 && |
1181 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29665 times.
|
29665 | pes->header[2] == 0x01) { |
1182 | /* it must be an MPEG-2 PES stream */ | ||
1183 | 29665 | pes->stream_id = pes->header[3]; | |
1184 | 29665 | av_log(pes->stream, AV_LOG_TRACE, "pid=%x stream_id=%#x\n", pes->pid, pes->stream_id); | |
1185 | |||
1186 |
3/4✓ Branch 0 taken 29665 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 591 times.
✓ Branch 3 taken 29074 times.
|
29665 | if ((pes->st && pes->st->discard == AVDISCARD_ALL && |
1187 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 591 times.
|
591 | (!pes->sub_st || |
1188 | ✗ | pes->sub_st->discard == AVDISCARD_ALL)) || | |
1189 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | pes->stream_id == STREAM_ID_PADDING_STREAM) |
1190 | 591 | goto skip; | |
1191 | |||
1192 | /* stream not present in PMT */ | ||
1193 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (!pes->st) { |
1194 | ✗ | if (ts->skip_changes) | |
1195 | ✗ | goto skip; | |
1196 | ✗ | if (ts->merge_pmt_versions) | |
1197 | ✗ | goto skip; /* wait for PMT to merge new stream */ | |
1198 | |||
1199 | ✗ | pes->st = avformat_new_stream(ts->stream, NULL); | |
1200 | ✗ | if (!pes->st) | |
1201 | ✗ | return AVERROR(ENOMEM); | |
1202 | ✗ | pes->st->id = pes->pid; | |
1203 | ✗ | mpegts_set_stream_info(pes->st, pes, 0, 0); | |
1204 | } | ||
1205 | |||
1206 | 29074 | pes->PES_packet_length = AV_RB16(pes->header + 4); | |
1207 | /* NOTE: zero length means the PES size is unbounded */ | ||
1208 | |||
1209 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | if (pes->stream_id != STREAM_ID_PROGRAM_STREAM_MAP && |
1210 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | pes->stream_id != STREAM_ID_PRIVATE_STREAM_2 && |
1211 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | pes->stream_id != STREAM_ID_ECM_STREAM && |
1212 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | pes->stream_id != STREAM_ID_EMM_STREAM && |
1213 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | pes->stream_id != STREAM_ID_PROGRAM_STREAM_DIRECTORY && |
1214 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | pes->stream_id != STREAM_ID_DSMCC_STREAM && |
1215 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
58148 | pes->stream_id != STREAM_ID_TYPE_E_STREAM) { |
1216 | 29074 | FFStream *const pes_sti = ffstream(pes->st); | |
1217 | 29074 | pes->state = MPEGTS_PESHEADER; | |
1218 |
3/4✓ Branch 0 taken 2 times.
✓ Branch 1 taken 29072 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
29074 | if (pes->st->codecpar->codec_id == AV_CODEC_ID_NONE && !pes_sti->request_probe) { |
1219 | 2 | av_log(pes->stream, AV_LOG_TRACE, | |
1220 | "pid=%x stream_type=%x probing\n", | ||
1221 | pes->pid, | ||
1222 | pes->stream_type); | ||
1223 | 2 | pes_sti->request_probe = 1; | |
1224 | } | ||
1225 | } else { | ||
1226 | ✗ | pes->pes_header_size = 6; | |
1227 | ✗ | pes->state = MPEGTS_PAYLOAD; | |
1228 | ✗ | pes->data_index = 0; | |
1229 | } | ||
1230 | } else { | ||
1231 | /* otherwise, it should be a table */ | ||
1232 | /* skip packet */ | ||
1233 | 7 | skip: | |
1234 | 598 | pes->state = MPEGTS_SKIP; | |
1235 | 598 | continue; | |
1236 | } | ||
1237 | } | ||
1238 | 29076 | break; | |
1239 | /**********************************************/ | ||
1240 | /* PES packing parsing */ | ||
1241 | 29074 | case MPEGTS_PESHEADER: | |
1242 | 29074 | len = PES_HEADER_SIZE - pes->data_index; | |
1243 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (len < 0) |
1244 | ✗ | return AVERROR_INVALIDDATA; | |
1245 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (len > buf_size) |
1246 | ✗ | len = buf_size; | |
1247 | 29074 | memcpy(pes->header + pes->data_index, p, len); | |
1248 | 29074 | pes->data_index += len; | |
1249 | 29074 | p += len; | |
1250 | 29074 | buf_size -= len; | |
1251 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | if (pes->data_index == PES_HEADER_SIZE) { |
1252 | 29074 | pes->pes_header_size = pes->header[8] + 9; | |
1253 | 29074 | pes->state = MPEGTS_PESHEADER_FILL; | |
1254 | } | ||
1255 | 29074 | break; | |
1256 | 29074 | case MPEGTS_PESHEADER_FILL: | |
1257 | 29074 | len = pes->pes_header_size - pes->data_index; | |
1258 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (len < 0) |
1259 | ✗ | return AVERROR_INVALIDDATA; | |
1260 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (len > buf_size) |
1261 | ✗ | len = buf_size; | |
1262 | 29074 | memcpy(pes->header + pes->data_index, p, len); | |
1263 | 29074 | pes->data_index += len; | |
1264 | 29074 | p += len; | |
1265 | 29074 | buf_size -= len; | |
1266 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | if (pes->data_index == pes->pes_header_size) { |
1267 | const uint8_t *r; | ||
1268 | unsigned int flags, pes_ext, skip; | ||
1269 | |||
1270 | 29074 | flags = pes->header[7]; | |
1271 | 29074 | r = pes->header + 9; | |
1272 | 29074 | pes->pts = AV_NOPTS_VALUE; | |
1273 | 29074 | pes->dts = AV_NOPTS_VALUE; | |
1274 |
2/2✓ Branch 0 taken 10075 times.
✓ Branch 1 taken 18999 times.
|
29074 | if ((flags & 0xc0) == 0x80) { |
1275 | 10075 | pes->dts = pes->pts = ff_parse_pes_pts(r); | |
1276 | 10075 | r += 5; | |
1277 |
2/2✓ Branch 0 taken 1854 times.
✓ Branch 1 taken 17145 times.
|
18999 | } else if ((flags & 0xc0) == 0xc0) { |
1278 | 1854 | pes->pts = ff_parse_pes_pts(r); | |
1279 | 1854 | r += 5; | |
1280 | 1854 | pes->dts = ff_parse_pes_pts(r); | |
1281 | 1854 | r += 5; | |
1282 | } | ||
1283 | 29074 | pes->extended_stream_id = -1; | |
1284 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (flags & 0x01) { /* PES extension */ |
1285 | ✗ | pes_ext = *r++; | |
1286 | /* Skip PES private data, program packet sequence counter and P-STD buffer */ | ||
1287 | ✗ | skip = (pes_ext >> 4) & 0xb; | |
1288 | ✗ | skip += skip & 0x9; | |
1289 | ✗ | r += skip; | |
1290 | ✗ | if ((pes_ext & 0x41) == 0x01 && | |
1291 | ✗ | (r + 2) <= (pes->header + pes->pes_header_size)) { | |
1292 | /* PES extension 2 */ | ||
1293 | ✗ | if ((r[0] & 0x7f) > 0 && (r[1] & 0x80) == 0) | |
1294 | ✗ | pes->extended_stream_id = r[1]; | |
1295 | } | ||
1296 | } | ||
1297 | |||
1298 | /* we got the full header. We parse it and get the payload */ | ||
1299 | 29074 | pes->state = MPEGTS_PAYLOAD; | |
1300 | 29074 | pes->data_index = 0; | |
1301 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
29074 | if (pes->stream_type == 0x12 && buf_size > 0) { |
1302 | ✗ | int sl_header_bytes = read_sl_header(pes, &pes->sl, p, | |
1303 | buf_size); | ||
1304 | ✗ | pes->pes_header_size += sl_header_bytes; | |
1305 | ✗ | p += sl_header_bytes; | |
1306 | ✗ | buf_size -= sl_header_bytes; | |
1307 | } | ||
1308 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 29068 times.
|
29074 | if (pes->stream_type == STREAM_TYPE_METADATA && |
1309 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | pes->stream_id == STREAM_ID_METADATA_STREAM && |
1310 | ✗ | pes->st->codecpar->codec_id == AV_CODEC_ID_SMPTE_KLV && | |
1311 | buf_size >= 5) { | ||
1312 | /* skip metadata access unit header - see MISB ST 1402 */ | ||
1313 | ✗ | pes->pes_header_size += 5; | |
1314 | ✗ | p += 5; | |
1315 | ✗ | buf_size -= 5; | |
1316 | } | ||
1317 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | if ( pes->ts->fix_teletext_pts |
1318 |
1/2✓ Branch 0 taken 29074 times.
✗ Branch 1 not taken.
|
29074 | && ( pes->st->codecpar->codec_id == AV_CODEC_ID_DVB_TELETEXT |
1319 |
2/2✓ Branch 0 taken 253 times.
✓ Branch 1 taken 28821 times.
|
29074 | || pes->st->codecpar->codec_id == AV_CODEC_ID_DVB_SUBTITLE) |
1320 | ) { | ||
1321 | 253 | AVProgram *p = NULL; | |
1322 | 253 | int pcr_found = 0; | |
1323 |
2/2✓ Branch 1 taken 253 times.
✓ Branch 2 taken 253 times.
|
506 | while ((p = av_find_program_from_stream(pes->stream, p, pes->st->index))) { |
1324 |
2/4✓ Branch 0 taken 253 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 253 times.
✗ Branch 3 not taken.
|
253 | if (p->pcr_pid != -1 && p->discard != AVDISCARD_ALL) { |
1325 | 253 | MpegTSFilter *f = pes->ts->pids[p->pcr_pid]; | |
1326 |
1/2✓ Branch 0 taken 253 times.
✗ Branch 1 not taken.
|
253 | if (f) { |
1327 | 253 | AVStream *st = NULL; | |
1328 |
1/2✓ Branch 0 taken 253 times.
✗ Branch 1 not taken.
|
253 | if (f->type == MPEGTS_PES) { |
1329 | 253 | PESContext *pcrpes = f->u.pes_filter.opaque; | |
1330 |
1/2✓ Branch 0 taken 253 times.
✗ Branch 1 not taken.
|
253 | if (pcrpes) |
1331 | 253 | st = pcrpes->st; | |
1332 | ✗ | } else if (f->type == MPEGTS_PCR) { | |
1333 | int i; | ||
1334 | ✗ | for (i = 0; i < p->nb_stream_indexes; i++) { | |
1335 | ✗ | AVStream *pst = pes->stream->streams[p->stream_index[i]]; | |
1336 | ✗ | if (pst->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) | |
1337 | ✗ | st = pst; | |
1338 | } | ||
1339 | } | ||
1340 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 253 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
253 | if (f->last_pcr != -1 && !f->discard) { |
1341 | // teletext packets do not always have correct timestamps, | ||
1342 | // the standard says they should be handled after 40.6 ms at most, | ||
1343 | // and the pcr error to this packet should be no more than 100 ms. | ||
1344 | // TODO: we should interpolate the PCR, not just use the last one | ||
1345 | ✗ | int64_t pcr = f->last_pcr / 300; | |
1346 | ✗ | pcr_found = 1; | |
1347 | ✗ | if (st) { | |
1348 | ✗ | const FFStream *const sti = ffstream(st); | |
1349 | ✗ | FFStream *const pes_sti = ffstream(pes->st); | |
1350 | |||
1351 | ✗ | pes_sti->pts_wrap_reference = sti->pts_wrap_reference; | |
1352 | ✗ | pes_sti->pts_wrap_behavior = sti->pts_wrap_behavior; | |
1353 | } | ||
1354 | ✗ | if (pes->dts == AV_NOPTS_VALUE || pes->dts < pcr) { | |
1355 | ✗ | pes->pts = pes->dts = pcr; | |
1356 | ✗ | } else if (pes->st->codecpar->codec_id == AV_CODEC_ID_DVB_TELETEXT && | |
1357 | ✗ | pes->dts > pcr + 3654 + 9000) { | |
1358 | ✗ | pes->pts = pes->dts = pcr + 3654 + 9000; | |
1359 | ✗ | } else if (pes->st->codecpar->codec_id == AV_CODEC_ID_DVB_SUBTITLE && | |
1360 | ✗ | pes->dts > pcr + 10*90000) { //10sec | |
1361 | ✗ | pes->pts = pes->dts = pcr + 3654 + 9000; | |
1362 | } | ||
1363 | ✗ | break; | |
1364 | } | ||
1365 | } | ||
1366 | } | ||
1367 | } | ||
1368 | |||
1369 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 253 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
253 | if (pes->st->codecpar->codec_id == AV_CODEC_ID_DVB_TELETEXT && |
1370 | !pcr_found) { | ||
1371 | ✗ | av_log(pes->stream, AV_LOG_VERBOSE, | |
1372 | "Forcing DTS/PTS to be unset for a " | ||
1373 | "non-trustworthy PES packet for PID %d as " | ||
1374 | "PCR hasn't been received yet.\n", | ||
1375 | pes->pid); | ||
1376 | ✗ | pes->dts = pes->pts = AV_NOPTS_VALUE; | |
1377 | } | ||
1378 | } | ||
1379 | } | ||
1380 | 29074 | break; | |
1381 | 355945 | case MPEGTS_PAYLOAD: | |
1382 | do { | ||
1383 | 355945 | int max_packet_size = ts->max_packet_size; | |
1384 |
3/4✓ Branch 0 taken 195689 times.
✓ Branch 1 taken 160256 times.
✓ Branch 2 taken 195689 times.
✗ Branch 3 not taken.
|
355945 | if (pes->PES_packet_length && pes->PES_packet_length + PES_START_SIZE > pes->pes_header_size) |
1385 | 195689 | max_packet_size = pes->PES_packet_length + PES_START_SIZE - pes->pes_header_size; | |
1386 | |||
1387 |
2/2✓ Branch 0 taken 326871 times.
✓ Branch 1 taken 29074 times.
|
355945 | if (pes->data_index > 0 && |
1388 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 326871 times.
|
326871 | pes->data_index + buf_size > max_packet_size) { |
1389 | ✗ | ret = new_pes_packet(pes, ts->pkt); | |
1390 | ✗ | if (ret < 0) | |
1391 | ✗ | return ret; | |
1392 | ✗ | pes->PES_packet_length = 0; | |
1393 | ✗ | max_packet_size = ts->max_packet_size; | |
1394 | ✗ | ts->stop_parse = 1; | |
1395 |
3/4✓ Branch 0 taken 29074 times.
✓ Branch 1 taken 326871 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 29074 times.
|
355945 | } else if (pes->data_index == 0 && |
1396 | buf_size > max_packet_size) { | ||
1397 | // pes packet size is < ts size packet and pes data is padded with 0xff | ||
1398 | // not sure if this is legal in ts but see issue #2392 | ||
1399 | ✗ | buf_size = max_packet_size; | |
1400 | } | ||
1401 | |||
1402 |
2/2✓ Branch 0 taken 29074 times.
✓ Branch 1 taken 326871 times.
|
355945 | if (!pes->buffer) { |
1403 | 29074 | pes->buffer = buffer_pool_get(ts, max_packet_size); | |
1404 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29074 times.
|
29074 | if (!pes->buffer) |
1405 | ✗ | return AVERROR(ENOMEM); | |
1406 | } | ||
1407 | |||
1408 | 355945 | memcpy(pes->buffer->data + pes->data_index, p, buf_size); | |
1409 | 355945 | pes->data_index += buf_size; | |
1410 | /* emit complete packets with known packet size | ||
1411 | * decreases demuxer delay for infrequent packets like subtitles from | ||
1412 | * a couple of seconds to milliseconds for properly muxed files. */ | ||
1413 |
4/4✓ Branch 0 taken 353094 times.
✓ Branch 1 taken 2851 times.
✓ Branch 2 taken 195689 times.
✓ Branch 3 taken 157405 times.
|
355945 | if (!ts->stop_parse && pes->PES_packet_length && |
1414 |
2/2✓ Branch 0 taken 25958 times.
✓ Branch 1 taken 169731 times.
|
195689 | pes->pes_header_size + pes->data_index == pes->PES_packet_length + PES_START_SIZE) { |
1415 | 25958 | ts->stop_parse = 1; | |
1416 | 25958 | ret = new_pes_packet(pes, ts->pkt); | |
1417 | 25958 | pes->state = MPEGTS_SKIP; | |
1418 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 25958 times.
|
25958 | if (ret < 0) |
1419 | ✗ | return ret; | |
1420 | } | ||
1421 | } while (0); | ||
1422 | 355945 | buf_size = 0; | |
1423 | 355945 | break; | |
1424 | 18542 | case MPEGTS_SKIP: | |
1425 | 18542 | buf_size = 0; | |
1426 | 18542 | break; | |
1427 | } | ||
1428 | } | ||
1429 | |||
1430 | 374489 | return 0; | |
1431 | } | ||
1432 | |||
1433 | 126 | static PESContext *add_pes_stream(MpegTSContext *ts, int pid, int pcr_pid) | |
1434 | { | ||
1435 | MpegTSFilter *tss; | ||
1436 | PESContext *pes; | ||
1437 | |||
1438 | /* if no pid found, then add a pid context */ | ||
1439 | 126 | pes = av_mallocz(sizeof(PESContext)); | |
1440 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 126 times.
|
126 | if (!pes) |
1441 | ✗ | return 0; | |
1442 | 126 | pes->ts = ts; | |
1443 | 126 | pes->stream = ts->stream; | |
1444 | 126 | pes->pid = pid; | |
1445 | 126 | pes->pcr_pid = pcr_pid; | |
1446 | 126 | pes->state = MPEGTS_SKIP; | |
1447 | 126 | pes->pts = AV_NOPTS_VALUE; | |
1448 | 126 | pes->dts = AV_NOPTS_VALUE; | |
1449 | 126 | tss = mpegts_open_pes_filter(ts, pid, mpegts_push_data, pes); | |
1450 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 126 times.
|
126 | if (!tss) { |
1451 | ✗ | av_free(pes); | |
1452 | ✗ | return 0; | |
1453 | } | ||
1454 | 126 | return pes; | |
1455 | } | ||
1456 | |||
1457 | #define MAX_LEVEL 4 | ||
1458 | typedef struct MP4DescrParseContext { | ||
1459 | AVFormatContext *s; | ||
1460 | FFIOContext pb; | ||
1461 | Mp4Descr *descr; | ||
1462 | Mp4Descr *active_descr; | ||
1463 | int descr_count; | ||
1464 | int max_descr_count; | ||
1465 | int level; | ||
1466 | int predefined_SLConfigDescriptor_seen; | ||
1467 | } MP4DescrParseContext; | ||
1468 | |||
1469 | ✗ | static int init_MP4DescrParseContext(MP4DescrParseContext *d, AVFormatContext *s, | |
1470 | const uint8_t *buf, unsigned size, | ||
1471 | Mp4Descr *descr, int max_descr_count) | ||
1472 | { | ||
1473 | ✗ | if (size > (1 << 30)) | |
1474 | ✗ | return AVERROR_INVALIDDATA; | |
1475 | |||
1476 | ✗ | ffio_init_read_context(&d->pb, buf, size); | |
1477 | |||
1478 | ✗ | d->s = s; | |
1479 | ✗ | d->level = 0; | |
1480 | ✗ | d->descr_count = 0; | |
1481 | ✗ | d->descr = descr; | |
1482 | ✗ | d->active_descr = NULL; | |
1483 | ✗ | d->max_descr_count = max_descr_count; | |
1484 | |||
1485 | ✗ | return 0; | |
1486 | } | ||
1487 | |||
1488 | ✗ | static void update_offsets(AVIOContext *pb, int64_t *off, int *len) | |
1489 | { | ||
1490 | ✗ | int64_t new_off = avio_tell(pb); | |
1491 | ✗ | (*len) -= new_off - *off; | |
1492 | ✗ | *off = new_off; | |
1493 | ✗ | } | |
1494 | |||
1495 | static int parse_mp4_descr(MP4DescrParseContext *d, int64_t off, int len, | ||
1496 | int target_tag); | ||
1497 | |||
1498 | ✗ | static int parse_mp4_descr_arr(MP4DescrParseContext *d, int64_t off, int len) | |
1499 | { | ||
1500 | ✗ | while (len > 0) { | |
1501 | ✗ | int ret = parse_mp4_descr(d, off, len, 0); | |
1502 | ✗ | if (ret < 0) | |
1503 | ✗ | return ret; | |
1504 | ✗ | update_offsets(&d->pb.pub, &off, &len); | |
1505 | } | ||
1506 | ✗ | return 0; | |
1507 | } | ||
1508 | |||
1509 | ✗ | static int parse_MP4IODescrTag(MP4DescrParseContext *d, int64_t off, int len) | |
1510 | { | ||
1511 | ✗ | AVIOContext *const pb = &d->pb.pub; | |
1512 | ✗ | avio_rb16(pb); // ID | |
1513 | ✗ | avio_r8(pb); | |
1514 | ✗ | avio_r8(pb); | |
1515 | ✗ | avio_r8(pb); | |
1516 | ✗ | avio_r8(pb); | |
1517 | ✗ | avio_r8(pb); | |
1518 | ✗ | update_offsets(pb, &off, &len); | |
1519 | ✗ | return parse_mp4_descr_arr(d, off, len); | |
1520 | } | ||
1521 | |||
1522 | ✗ | static int parse_MP4ODescrTag(MP4DescrParseContext *d, int64_t off, int len) | |
1523 | { | ||
1524 | int id_flags; | ||
1525 | ✗ | if (len < 2) | |
1526 | ✗ | return 0; | |
1527 | ✗ | id_flags = avio_rb16(&d->pb.pub); | |
1528 | ✗ | if (!(id_flags & 0x0020)) { // URL_Flag | |
1529 | ✗ | update_offsets(&d->pb.pub, &off, &len); | |
1530 | ✗ | return parse_mp4_descr_arr(d, off, len); // ES_Descriptor[] | |
1531 | } else { | ||
1532 | ✗ | return 0; | |
1533 | } | ||
1534 | } | ||
1535 | |||
1536 | ✗ | static int parse_MP4ESDescrTag(MP4DescrParseContext *d, int64_t off, int len) | |
1537 | { | ||
1538 | ✗ | AVIOContext *const pb = &d->pb.pub; | |
1539 | ✗ | int es_id = 0; | |
1540 | ✗ | int ret = 0; | |
1541 | |||
1542 | ✗ | if (d->descr_count >= d->max_descr_count) | |
1543 | ✗ | return AVERROR_INVALIDDATA; | |
1544 | ✗ | ff_mp4_parse_es_descr(pb, &es_id); | |
1545 | ✗ | d->active_descr = d->descr + (d->descr_count++); | |
1546 | |||
1547 | ✗ | d->active_descr->es_id = es_id; | |
1548 | ✗ | update_offsets(pb, &off, &len); | |
1549 | ✗ | if ((ret = parse_mp4_descr(d, off, len, MP4DecConfigDescrTag)) < 0) | |
1550 | ✗ | return ret; | |
1551 | ✗ | update_offsets(pb, &off, &len); | |
1552 | ✗ | if (len > 0) | |
1553 | ✗ | ret = parse_mp4_descr(d, off, len, MP4SLDescrTag); | |
1554 | ✗ | d->active_descr = NULL; | |
1555 | ✗ | return ret; | |
1556 | } | ||
1557 | |||
1558 | ✗ | static int parse_MP4DecConfigDescrTag(MP4DescrParseContext *d, int64_t off, | |
1559 | int len) | ||
1560 | { | ||
1561 | ✗ | Mp4Descr *descr = d->active_descr; | |
1562 | ✗ | if (!descr) | |
1563 | ✗ | return AVERROR_INVALIDDATA; | |
1564 | ✗ | d->active_descr->dec_config_descr = av_malloc(len); | |
1565 | ✗ | if (!descr->dec_config_descr) | |
1566 | ✗ | return AVERROR(ENOMEM); | |
1567 | ✗ | descr->dec_config_descr_len = len; | |
1568 | ✗ | avio_read(&d->pb.pub, descr->dec_config_descr, len); | |
1569 | ✗ | return 0; | |
1570 | } | ||
1571 | |||
1572 | ✗ | static int parse_MP4SLDescrTag(MP4DescrParseContext *d, int64_t off, int len) | |
1573 | { | ||
1574 | ✗ | Mp4Descr *descr = d->active_descr; | |
1575 | ✗ | AVIOContext *const pb = &d->pb.pub; | |
1576 | int predefined; | ||
1577 | ✗ | if (!descr) | |
1578 | ✗ | return AVERROR_INVALIDDATA; | |
1579 | |||
1580 | #define R8_CHECK_CLIP_MAX(dst, maxv) do { \ | ||
1581 | descr->sl.dst = avio_r8(pb); \ | ||
1582 | if (descr->sl.dst > maxv) { \ | ||
1583 | descr->sl.dst = maxv; \ | ||
1584 | return AVERROR_INVALIDDATA; \ | ||
1585 | } \ | ||
1586 | } while (0) | ||
1587 | |||
1588 | ✗ | predefined = avio_r8(pb); | |
1589 | ✗ | if (!predefined) { | |
1590 | int lengths; | ||
1591 | ✗ | int flags = avio_r8(pb); | |
1592 | ✗ | descr->sl.use_au_start = !!(flags & 0x80); | |
1593 | ✗ | descr->sl.use_au_end = !!(flags & 0x40); | |
1594 | ✗ | descr->sl.use_rand_acc_pt = !!(flags & 0x20); | |
1595 | ✗ | descr->sl.use_padding = !!(flags & 0x08); | |
1596 | ✗ | descr->sl.use_timestamps = !!(flags & 0x04); | |
1597 | ✗ | descr->sl.use_idle = !!(flags & 0x02); | |
1598 | ✗ | descr->sl.timestamp_res = avio_rb32(pb); | |
1599 | ✗ | avio_rb32(pb); | |
1600 | ✗ | R8_CHECK_CLIP_MAX(timestamp_len, 63); | |
1601 | ✗ | R8_CHECK_CLIP_MAX(ocr_len, 63); | |
1602 | ✗ | R8_CHECK_CLIP_MAX(au_len, 31); | |
1603 | ✗ | descr->sl.inst_bitrate_len = avio_r8(pb); | |
1604 | ✗ | lengths = avio_rb16(pb); | |
1605 | ✗ | descr->sl.degr_prior_len = lengths >> 12; | |
1606 | ✗ | descr->sl.au_seq_num_len = (lengths >> 7) & 0x1f; | |
1607 | ✗ | descr->sl.packet_seq_num_len = (lengths >> 2) & 0x1f; | |
1608 | ✗ | } else if (!d->predefined_SLConfigDescriptor_seen){ | |
1609 | ✗ | avpriv_report_missing_feature(d->s, "Predefined SLConfigDescriptor"); | |
1610 | ✗ | d->predefined_SLConfigDescriptor_seen = 1; | |
1611 | } | ||
1612 | ✗ | return 0; | |
1613 | } | ||
1614 | |||
1615 | ✗ | static int parse_mp4_descr(MP4DescrParseContext *d, int64_t off, int len, | |
1616 | int target_tag) | ||
1617 | { | ||
1618 | int tag; | ||
1619 | ✗ | AVIOContext *const pb = &d->pb.pub; | |
1620 | ✗ | int len1 = ff_mp4_read_descr(d->s, pb, &tag); | |
1621 | ✗ | int ret = 0; | |
1622 | |||
1623 | ✗ | update_offsets(pb, &off, &len); | |
1624 | ✗ | if (len < 0 || len1 > len || len1 <= 0) { | |
1625 | ✗ | av_log(d->s, AV_LOG_ERROR, | |
1626 | "Tag %x length violation new length %d bytes remaining %d\n", | ||
1627 | tag, len1, len); | ||
1628 | ✗ | return AVERROR_INVALIDDATA; | |
1629 | } | ||
1630 | |||
1631 | ✗ | if (d->level++ >= MAX_LEVEL) { | |
1632 | ✗ | av_log(d->s, AV_LOG_ERROR, "Maximum MP4 descriptor level exceeded\n"); | |
1633 | ✗ | ret = AVERROR_INVALIDDATA; | |
1634 | ✗ | goto done; | |
1635 | } | ||
1636 | |||
1637 | ✗ | if (target_tag && tag != target_tag) { | |
1638 | ✗ | av_log(d->s, AV_LOG_ERROR, "Found tag %x expected %x\n", tag, | |
1639 | target_tag); | ||
1640 | ✗ | ret = AVERROR_INVALIDDATA; | |
1641 | ✗ | goto done; | |
1642 | } | ||
1643 | |||
1644 | ✗ | switch (tag) { | |
1645 | ✗ | case MP4IODescrTag: | |
1646 | ✗ | ret = parse_MP4IODescrTag(d, off, len1); | |
1647 | ✗ | break; | |
1648 | ✗ | case MP4ODescrTag: | |
1649 | ✗ | ret = parse_MP4ODescrTag(d, off, len1); | |
1650 | ✗ | break; | |
1651 | ✗ | case MP4ESDescrTag: | |
1652 | ✗ | ret = parse_MP4ESDescrTag(d, off, len1); | |
1653 | ✗ | break; | |
1654 | ✗ | case MP4DecConfigDescrTag: | |
1655 | ✗ | ret = parse_MP4DecConfigDescrTag(d, off, len1); | |
1656 | ✗ | break; | |
1657 | ✗ | case MP4SLDescrTag: | |
1658 | ✗ | ret = parse_MP4SLDescrTag(d, off, len1); | |
1659 | ✗ | break; | |
1660 | } | ||
1661 | |||
1662 | |||
1663 | ✗ | done: | |
1664 | ✗ | d->level--; | |
1665 | ✗ | avio_seek(pb, off + len1, SEEK_SET); | |
1666 | ✗ | return ret; | |
1667 | } | ||
1668 | |||
1669 | ✗ | static int mp4_read_iods(AVFormatContext *s, const uint8_t *buf, unsigned size, | |
1670 | Mp4Descr *descr, int *descr_count, int max_descr_count) | ||
1671 | { | ||
1672 | MP4DescrParseContext d; | ||
1673 | int ret; | ||
1674 | |||
1675 | ✗ | d.predefined_SLConfigDescriptor_seen = 0; | |
1676 | |||
1677 | ✗ | ret = init_MP4DescrParseContext(&d, s, buf, size, descr, max_descr_count); | |
1678 | ✗ | if (ret < 0) | |
1679 | ✗ | return ret; | |
1680 | |||
1681 | ✗ | ret = parse_mp4_descr(&d, avio_tell(&d.pb.pub), size, MP4IODescrTag); | |
1682 | |||
1683 | ✗ | *descr_count = d.descr_count; | |
1684 | ✗ | return ret; | |
1685 | } | ||
1686 | |||
1687 | ✗ | static int mp4_read_od(AVFormatContext *s, const uint8_t *buf, unsigned size, | |
1688 | Mp4Descr *descr, int *descr_count, int max_descr_count) | ||
1689 | { | ||
1690 | MP4DescrParseContext d; | ||
1691 | int ret; | ||
1692 | |||
1693 | ✗ | ret = init_MP4DescrParseContext(&d, s, buf, size, descr, max_descr_count); | |
1694 | ✗ | if (ret < 0) | |
1695 | ✗ | return ret; | |
1696 | |||
1697 | ✗ | ret = parse_mp4_descr_arr(&d, avio_tell(&d.pb.pub), size); | |
1698 | |||
1699 | ✗ | *descr_count = d.descr_count; | |
1700 | ✗ | return ret; | |
1701 | } | ||
1702 | |||
1703 | ✗ | static void m4sl_cb(MpegTSFilter *filter, const uint8_t *section, | |
1704 | int section_len) | ||
1705 | { | ||
1706 | ✗ | MpegTSContext *ts = filter->u.section_filter.opaque; | |
1707 | ✗ | MpegTSSectionFilter *tssf = &filter->u.section_filter; | |
1708 | SectionHeader h; | ||
1709 | const uint8_t *p, *p_end; | ||
1710 | ✗ | int mp4_descr_count = 0; | |
1711 | ✗ | Mp4Descr mp4_descr[MAX_MP4_DESCR_COUNT] = { { 0 } }; | |
1712 | int i, pid; | ||
1713 | ✗ | AVFormatContext *s = ts->stream; | |
1714 | |||
1715 | ✗ | p_end = section + section_len - 4; | |
1716 | ✗ | p = section; | |
1717 | ✗ | if (parse_section_header(&h, &p, p_end) < 0) | |
1718 | ✗ | return; | |
1719 | ✗ | if (h.tid != M4OD_TID) | |
1720 | ✗ | return; | |
1721 | ✗ | if (skip_identical(&h, tssf)) | |
1722 | ✗ | return; | |
1723 | |||
1724 | ✗ | mp4_read_od(s, p, (unsigned) (p_end - p), mp4_descr, &mp4_descr_count, | |
1725 | MAX_MP4_DESCR_COUNT); | ||
1726 | |||
1727 | ✗ | for (pid = 0; pid < NB_PID_MAX; pid++) { | |
1728 | ✗ | if (!ts->pids[pid]) | |
1729 | ✗ | continue; | |
1730 | ✗ | for (i = 0; i < mp4_descr_count; i++) { | |
1731 | PESContext *pes; | ||
1732 | AVStream *st; | ||
1733 | FFStream *sti; | ||
1734 | FFIOContext pb; | ||
1735 | ✗ | if (ts->pids[pid]->es_id != mp4_descr[i].es_id) | |
1736 | ✗ | continue; | |
1737 | ✗ | if (ts->pids[pid]->type != MPEGTS_PES) { | |
1738 | ✗ | av_log(s, AV_LOG_ERROR, "pid %x is not PES\n", pid); | |
1739 | ✗ | continue; | |
1740 | } | ||
1741 | ✗ | pes = ts->pids[pid]->u.pes_filter.opaque; | |
1742 | ✗ | st = pes->st; | |
1743 | ✗ | if (!st) | |
1744 | ✗ | continue; | |
1745 | ✗ | sti = ffstream(st); | |
1746 | |||
1747 | ✗ | pes->sl = mp4_descr[i].sl; | |
1748 | |||
1749 | ✗ | ffio_init_read_context(&pb, mp4_descr[i].dec_config_descr, | |
1750 | mp4_descr[i].dec_config_descr_len); | ||
1751 | ✗ | ff_mp4_read_dec_config_descr(s, st, &pb.pub); | |
1752 | ✗ | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && | |
1753 | ✗ | st->codecpar->extradata_size > 0) | |
1754 | ✗ | sti->need_parsing = 0; | |
1755 | ✗ | if (st->codecpar->codec_id == AV_CODEC_ID_H264 && | |
1756 | ✗ | st->codecpar->extradata_size > 0) | |
1757 | ✗ | sti->need_parsing = 0; | |
1758 | |||
1759 | ✗ | st->codecpar->codec_type = avcodec_get_type(st->codecpar->codec_id); | |
1760 | ✗ | sti->need_context_update = 1; | |
1761 | } | ||
1762 | } | ||
1763 | ✗ | for (i = 0; i < mp4_descr_count; i++) | |
1764 | ✗ | av_free(mp4_descr[i].dec_config_descr); | |
1765 | } | ||
1766 | |||
1767 | ✗ | static void scte_data_cb(MpegTSFilter *filter, const uint8_t *section, | |
1768 | int section_len) | ||
1769 | { | ||
1770 | ✗ | AVProgram *prg = NULL; | |
1771 | ✗ | MpegTSContext *ts = filter->u.section_filter.opaque; | |
1772 | |||
1773 | ✗ | int idx = ff_find_stream_index(ts->stream, filter->pid); | |
1774 | ✗ | if (idx < 0) | |
1775 | ✗ | return; | |
1776 | |||
1777 | /** | ||
1778 | * In case we receive an SCTE-35 packet before mpegts context is fully | ||
1779 | * initialized. | ||
1780 | */ | ||
1781 | ✗ | if (!ts->pkt) | |
1782 | ✗ | return; | |
1783 | |||
1784 | ✗ | new_data_packet(section, section_len, ts->pkt); | |
1785 | ✗ | ts->pkt->stream_index = idx; | |
1786 | ✗ | prg = av_find_program_from_stream(ts->stream, NULL, idx); | |
1787 | ✗ | if (prg && prg->pcr_pid != -1 && prg->discard != AVDISCARD_ALL) { | |
1788 | ✗ | MpegTSFilter *f = ts->pids[prg->pcr_pid]; | |
1789 | ✗ | if (f && f->last_pcr != -1) | |
1790 | ✗ | ts->pkt->pts = ts->pkt->dts = f->last_pcr/300; | |
1791 | } | ||
1792 | ✗ | ts->stop_parse = 1; | |
1793 | |||
1794 | } | ||
1795 | |||
1796 | static const uint8_t opus_coupled_stream_cnt[9] = { | ||
1797 | 1, 0, 1, 1, 2, 2, 2, 3, 3 | ||
1798 | }; | ||
1799 | |||
1800 | static const uint8_t opus_stream_cnt[9] = { | ||
1801 | 1, 1, 1, 2, 2, 3, 4, 4, 5, | ||
1802 | }; | ||
1803 | |||
1804 | static const uint8_t opus_channel_map[8][8] = { | ||
1805 | { 0 }, | ||
1806 | { 0,1 }, | ||
1807 | { 0,2,1 }, | ||
1808 | { 0,1,2,3 }, | ||
1809 | { 0,4,1,2,3 }, | ||
1810 | { 0,4,1,2,3,5 }, | ||
1811 | { 0,4,1,2,3,5,6 }, | ||
1812 | { 0,6,1,2,3,4,5,7 }, | ||
1813 | }; | ||
1814 | |||
1815 | 933 | int ff_parse_mpeg2_descriptor(AVFormatContext *fc, AVStream *st, int stream_type, | |
1816 | const uint8_t **pp, const uint8_t *desc_list_end, | ||
1817 | Mp4Descr *mp4_descr, int mp4_descr_count, int pid, | ||
1818 | MpegTSContext *ts) | ||
1819 | { | ||
1820 | 933 | FFStream *const sti = ffstream(st); | |
1821 | const uint8_t *desc_end; | ||
1822 | int desc_len, desc_tag, desc_es_id, ext_desc_tag, channels, channel_config_code; | ||
1823 | char language[252]; | ||
1824 | int i; | ||
1825 | |||
1826 | 933 | desc_tag = get8(pp, desc_list_end); | |
1827 |
2/2✓ Branch 0 taken 559 times.
✓ Branch 1 taken 374 times.
|
933 | if (desc_tag < 0) |
1828 | 559 | return AVERROR_INVALIDDATA; | |
1829 | 374 | desc_len = get8(pp, desc_list_end); | |
1830 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 374 times.
|
374 | if (desc_len < 0) |
1831 | ✗ | return AVERROR_INVALIDDATA; | |
1832 | 374 | desc_end = *pp + desc_len; | |
1833 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 374 times.
|
374 | if (desc_end > desc_list_end) |
1834 | ✗ | return AVERROR_INVALIDDATA; | |
1835 | |||
1836 | 374 | av_log(fc, AV_LOG_TRACE, "tag: 0x%02x len=%d\n", desc_tag, desc_len); | |
1837 | |||
1838 |
6/6✓ Branch 0 taken 369 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 25 times.
✓ Branch 3 taken 344 times.
✓ Branch 4 taken 12 times.
✓ Branch 5 taken 18 times.
|
374 | if ((st->codecpar->codec_id == AV_CODEC_ID_NONE || sti->request_probe > 0) && |
1839 | stream_type == STREAM_TYPE_PRIVATE_DATA) | ||
1840 | 12 | mpegts_find_stream_type(st, desc_tag, DESC_types); | |
1841 | |||
1842 |
8/15✗ Branch 0 not taken.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 9 times.
✓ Branch 5 taken 252 times.
✓ Branch 6 taken 8 times.
✓ Branch 7 taken 31 times.
✓ Branch 8 taken 4 times.
✓ Branch 9 taken 4 times.
✓ Branch 10 taken 6 times.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✓ Branch 14 taken 60 times.
|
374 | switch (desc_tag) { |
1843 | ✗ | case VIDEO_STREAM_DESCRIPTOR: | |
1844 | ✗ | if (get8(pp, desc_end) & 0x1) { | |
1845 | ✗ | st->disposition |= AV_DISPOSITION_STILL_IMAGE; | |
1846 | } | ||
1847 | ✗ | break; | |
1848 | ✗ | case SL_DESCRIPTOR: | |
1849 | ✗ | desc_es_id = get16(pp, desc_end); | |
1850 | ✗ | if (desc_es_id < 0) | |
1851 | ✗ | break; | |
1852 | ✗ | if (ts && ts->pids[pid]) | |
1853 | ✗ | ts->pids[pid]->es_id = desc_es_id; | |
1854 | ✗ | for (i = 0; i < mp4_descr_count; i++) | |
1855 | ✗ | if (mp4_descr[i].dec_config_descr_len && | |
1856 | ✗ | mp4_descr[i].es_id == desc_es_id) { | |
1857 | FFIOContext pb; | ||
1858 | ✗ | ffio_init_read_context(&pb, mp4_descr[i].dec_config_descr, | |
1859 | ✗ | mp4_descr[i].dec_config_descr_len); | |
1860 | ✗ | ff_mp4_read_dec_config_descr(fc, st, &pb.pub); | |
1861 | ✗ | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && | |
1862 | ✗ | st->codecpar->extradata_size > 0) { | |
1863 | ✗ | sti->need_parsing = 0; | |
1864 | ✗ | sti->need_context_update = 1; | |
1865 | } | ||
1866 | ✗ | if (st->codecpar->codec_id == AV_CODEC_ID_MPEG4SYSTEMS) | |
1867 | ✗ | mpegts_open_section_filter(ts, pid, m4sl_cb, ts, 1); | |
1868 | } | ||
1869 | ✗ | break; | |
1870 | ✗ | case FMC_DESCRIPTOR: | |
1871 | ✗ | if (get16(pp, desc_end) < 0) | |
1872 | ✗ | break; | |
1873 | ✗ | if (mp4_descr_count > 0 && | |
1874 | ✗ | (st->codecpar->codec_id == AV_CODEC_ID_AAC_LATM || | |
1875 | ✗ | (sti->request_probe == 0 && st->codecpar->codec_id == AV_CODEC_ID_NONE) || | |
1876 | ✗ | sti->request_probe > 0) && | |
1877 | ✗ | mp4_descr->dec_config_descr_len && mp4_descr->es_id == pid) { | |
1878 | FFIOContext pb; | ||
1879 | ✗ | ffio_init_read_context(&pb, mp4_descr->dec_config_descr, | |
1880 | mp4_descr->dec_config_descr_len); | ||
1881 | ✗ | ff_mp4_read_dec_config_descr(fc, st, &pb.pub); | |
1882 | ✗ | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && | |
1883 | ✗ | st->codecpar->extradata_size > 0) { | |
1884 | ✗ | sti->request_probe = sti->need_parsing = 0; | |
1885 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; | |
1886 | ✗ | sti->need_context_update = 1; | |
1887 | } | ||
1888 | } | ||
1889 | ✗ | break; | |
1890 | ✗ | case 0x56: /* DVB teletext descriptor */ | |
1891 | { | ||
1892 | ✗ | uint8_t *extradata = NULL; | |
1893 | ✗ | int language_count = desc_len / 5, ret; | |
1894 | |||
1895 | ✗ | if (desc_len > 0 && desc_len % 5 != 0) | |
1896 | ✗ | return AVERROR_INVALIDDATA; | |
1897 | |||
1898 | ✗ | if (language_count > 0) { | |
1899 | /* 4 bytes per language code (3 bytes) with comma or NUL byte should fit language buffer */ | ||
1900 | ✗ | av_assert0(language_count <= sizeof(language) / 4); | |
1901 | |||
1902 | ✗ | if (st->codecpar->extradata == NULL) { | |
1903 | ✗ | ret = ff_alloc_extradata(st->codecpar, language_count * 2); | |
1904 | ✗ | if (ret < 0) | |
1905 | ✗ | return ret; | |
1906 | } | ||
1907 | |||
1908 | ✗ | if (st->codecpar->extradata_size < language_count * 2) | |
1909 | ✗ | return AVERROR_INVALIDDATA; | |
1910 | |||
1911 | ✗ | extradata = st->codecpar->extradata; | |
1912 | |||
1913 | ✗ | for (i = 0; i < language_count; i++) { | |
1914 | ✗ | language[i * 4 + 0] = get8(pp, desc_end); | |
1915 | ✗ | language[i * 4 + 1] = get8(pp, desc_end); | |
1916 | ✗ | language[i * 4 + 2] = get8(pp, desc_end); | |
1917 | ✗ | language[i * 4 + 3] = ','; | |
1918 | |||
1919 | ✗ | memcpy(extradata, *pp, 2); | |
1920 | ✗ | extradata += 2; | |
1921 | |||
1922 | ✗ | *pp += 2; | |
1923 | } | ||
1924 | |||
1925 | ✗ | language[i * 4 - 1] = 0; | |
1926 | ✗ | av_dict_set(&st->metadata, "language", language, 0); | |
1927 | ✗ | sti->need_context_update = 1; | |
1928 | } | ||
1929 | } | ||
1930 | ✗ | break; | |
1931 | 9 | case 0x59: /* subtitling descriptor */ | |
1932 | { | ||
1933 | /* 8 bytes per DVB subtitle substream data: | ||
1934 | * ISO_639_language_code (3 bytes), | ||
1935 | * subtitling_type (1 byte), | ||
1936 | * composition_page_id (2 bytes), | ||
1937 | * ancillary_page_id (2 bytes) */ | ||
1938 | 9 | int language_count = desc_len / 8, ret; | |
1939 | |||
1940 |
2/4✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
|
9 | if (desc_len > 0 && desc_len % 8 != 0) |
1941 | ✗ | return AVERROR_INVALIDDATA; | |
1942 | |||
1943 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (language_count > 1) { |
1944 | ✗ | avpriv_request_sample(fc, "DVB subtitles with multiple languages"); | |
1945 | } | ||
1946 | |||
1947 |
1/2✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
|
9 | if (language_count > 0) { |
1948 | uint8_t *extradata; | ||
1949 | |||
1950 | /* 4 bytes per language code (3 bytes) with comma or NUL byte should fit language buffer */ | ||
1951 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | av_assert0(language_count <= sizeof(language) / 4); |
1952 | |||
1953 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 6 times.
|
9 | if (st->codecpar->extradata == NULL) { |
1954 | 3 | ret = ff_alloc_extradata(st->codecpar, language_count * 5); | |
1955 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
1956 | ✗ | return ret; | |
1957 | } | ||
1958 | |||
1959 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (st->codecpar->extradata_size < language_count * 5) |
1960 | ✗ | return AVERROR_INVALIDDATA; | |
1961 | |||
1962 | 9 | extradata = st->codecpar->extradata; | |
1963 | |||
1964 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 9 times.
|
18 | for (i = 0; i < language_count; i++) { |
1965 | 9 | language[i * 4 + 0] = get8(pp, desc_end); | |
1966 | 9 | language[i * 4 + 1] = get8(pp, desc_end); | |
1967 | 9 | language[i * 4 + 2] = get8(pp, desc_end); | |
1968 | 9 | language[i * 4 + 3] = ','; | |
1969 | |||
1970 | /* hearing impaired subtitles detection using subtitling_type */ | ||
1971 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | switch (*pp[0]) { |
1972 | ✗ | case 0x20: /* DVB subtitles (for the hard of hearing) with no monitor aspect ratio criticality */ | |
1973 | case 0x21: /* DVB subtitles (for the hard of hearing) for display on 4:3 aspect ratio monitor */ | ||
1974 | case 0x22: /* DVB subtitles (for the hard of hearing) for display on 16:9 aspect ratio monitor */ | ||
1975 | case 0x23: /* DVB subtitles (for the hard of hearing) for display on 2.21:1 aspect ratio monitor */ | ||
1976 | case 0x24: /* DVB subtitles (for the hard of hearing) for display on a high definition monitor */ | ||
1977 | case 0x25: /* DVB subtitles (for the hard of hearing) with plano-stereoscopic disparity for display on a high definition monitor */ | ||
1978 | ✗ | st->disposition |= AV_DISPOSITION_HEARING_IMPAIRED; | |
1979 | ✗ | break; | |
1980 | } | ||
1981 | |||
1982 | 9 | extradata[4] = get8(pp, desc_end); /* subtitling_type */ | |
1983 | 9 | memcpy(extradata, *pp, 4); /* composition_page_id and ancillary_page_id */ | |
1984 | 9 | extradata += 5; | |
1985 | |||
1986 | 9 | *pp += 4; | |
1987 | } | ||
1988 | |||
1989 | 9 | language[i * 4 - 1] = 0; | |
1990 | 9 | av_dict_set(&st->metadata, "language", language, 0); | |
1991 | 9 | sti->need_context_update = 1; | |
1992 | } | ||
1993 | } | ||
1994 | 9 | break; | |
1995 | 252 | case ISO_639_LANGUAGE_DESCRIPTOR: | |
1996 |
2/2✓ Branch 0 taken 246 times.
✓ Branch 1 taken 252 times.
|
498 | for (i = 0; i + 4 <= desc_len; i += 4) { |
1997 | 246 | language[i + 0] = get8(pp, desc_end); | |
1998 | 246 | language[i + 1] = get8(pp, desc_end); | |
1999 | 246 | language[i + 2] = get8(pp, desc_end); | |
2000 | 246 | language[i + 3] = ','; | |
2001 |
2/4✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 60 times.
✓ Branch 4 taken 186 times.
|
246 | switch (get8(pp, desc_end)) { |
2002 | ✗ | case 0x01: | |
2003 | ✗ | st->disposition |= AV_DISPOSITION_CLEAN_EFFECTS; | |
2004 | ✗ | break; | |
2005 | ✗ | case 0x02: | |
2006 | ✗ | st->disposition |= AV_DISPOSITION_HEARING_IMPAIRED; | |
2007 | ✗ | break; | |
2008 | 60 | case 0x03: | |
2009 | 60 | st->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED; | |
2010 | 60 | st->disposition |= AV_DISPOSITION_DESCRIPTIONS; | |
2011 | 60 | break; | |
2012 | } | ||
2013 | } | ||
2014 |
4/4✓ Branch 0 taken 246 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 242 times.
✓ Branch 3 taken 4 times.
|
252 | if (i && language[0]) { |
2015 | 242 | language[i - 1] = 0; | |
2016 | /* don't overwrite language, as it may already have been set by | ||
2017 | * another, more specific descriptor (e.g. supplementary audio) */ | ||
2018 | 242 | av_dict_set(&st->metadata, "language", language, AV_DICT_DONT_OVERWRITE); | |
2019 | } | ||
2020 | 252 | break; | |
2021 | 8 | case REGISTRATION_DESCRIPTOR: | |
2022 | 8 | st->codecpar->codec_tag = bytestream_get_le32(pp); | |
2023 | 8 | av_log(fc, AV_LOG_TRACE, "reg_desc=%.4s\n", (char *)&st->codecpar->codec_tag); | |
2024 |
3/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 5 times.
|
8 | if (st->codecpar->codec_id == AV_CODEC_ID_NONE || sti->request_probe > 0) { |
2025 | 3 | mpegts_find_stream_type(st, st->codecpar->codec_tag, REGD_types); | |
2026 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | if (st->codecpar->codec_tag == MKTAG('B', 'S', 'S', 'D')) |
2027 | 2 | sti->request_probe = 50; | |
2028 | } | ||
2029 | 8 | break; | |
2030 | 31 | case 0x52: /* stream identifier descriptor */ | |
2031 | 31 | sti->stream_identifier = 1 + get8(pp, desc_end); | |
2032 | 31 | break; | |
2033 | 4 | case METADATA_DESCRIPTOR: | |
2034 |
1/2✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
|
4 | if (get16(pp, desc_end) == 0xFFFF) |
2035 | 4 | *pp += 4; | |
2036 |
1/2✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
|
4 | if (get8(pp, desc_end) == 0xFF) { |
2037 | 4 | st->codecpar->codec_tag = bytestream_get_le32(pp); | |
2038 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
|
4 | if (st->codecpar->codec_id == AV_CODEC_ID_NONE) |
2039 | 1 | mpegts_find_stream_type(st, st->codecpar->codec_tag, METADATA_types); | |
2040 | } | ||
2041 | 4 | break; | |
2042 | 4 | case 0x7f: /* DVB extension descriptor */ | |
2043 | 4 | ext_desc_tag = get8(pp, desc_end); | |
2044 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ext_desc_tag < 0) |
2045 | ✗ | return AVERROR_INVALIDDATA; | |
2046 |
3/4✓ Branch 0 taken 3 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 3 times.
✗ Branch 3 not taken.
|
4 | if (st->codecpar->codec_id == AV_CODEC_ID_OPUS && |
2047 | ext_desc_tag == 0x80) { /* User defined (provisional Opus) */ | ||
2048 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
|
3 | if (!st->codecpar->extradata) { |
2049 | 1 | st->codecpar->extradata = av_mallocz(sizeof(opus_default_extradata) + | |
2050 | AV_INPUT_BUFFER_PADDING_SIZE); | ||
2051 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!st->codecpar->extradata) |
2052 | ✗ | return AVERROR(ENOMEM); | |
2053 | |||
2054 | 1 | st->codecpar->extradata_size = sizeof(opus_default_extradata); | |
2055 | 1 | memcpy(st->codecpar->extradata, opus_default_extradata, sizeof(opus_default_extradata)); | |
2056 | |||
2057 | 1 | channel_config_code = get8(pp, desc_end); | |
2058 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (channel_config_code < 0) |
2059 | ✗ | return AVERROR_INVALIDDATA; | |
2060 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (channel_config_code <= 0x8) { |
2061 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | st->codecpar->extradata[9] = channels = channel_config_code ? channel_config_code : 2; |
2062 | 1 | AV_WL32(&st->codecpar->extradata[12], 48000); | |
2063 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | st->codecpar->extradata[18] = channel_config_code ? (channels > 2) : /* Dual Mono */ 255; |
2064 | 1 | st->codecpar->extradata[19] = opus_stream_cnt[channel_config_code]; | |
2065 | 1 | st->codecpar->extradata[20] = opus_coupled_stream_cnt[channel_config_code]; | |
2066 | 1 | memcpy(&st->codecpar->extradata[21], opus_channel_map[channels - 1], channels); | |
2067 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | st->codecpar->extradata_size = st->codecpar->extradata[18] ? 21 + channels : 19; |
2068 | } else { | ||
2069 | ✗ | avpriv_request_sample(fc, "Opus in MPEG-TS - channel_config_code > 0x8"); | |
2070 | } | ||
2071 | 1 | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
2072 | 1 | sti->need_context_update = 1; | |
2073 | } | ||
2074 | } | ||
2075 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ext_desc_tag == 0x06) { /* supplementary audio descriptor */ |
2076 | int flags; | ||
2077 | |||
2078 | ✗ | if (desc_len < 1) | |
2079 | ✗ | return AVERROR_INVALIDDATA; | |
2080 | ✗ | flags = get8(pp, desc_end); | |
2081 | |||
2082 | ✗ | if ((flags & 0x80) == 0) /* mix_type */ | |
2083 | ✗ | st->disposition |= AV_DISPOSITION_DEPENDENT; | |
2084 | |||
2085 | ✗ | switch ((flags >> 2) & 0x1F) { /* editorial_classification */ | |
2086 | ✗ | case 0x01: | |
2087 | ✗ | st->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED; | |
2088 | ✗ | st->disposition |= AV_DISPOSITION_DESCRIPTIONS; | |
2089 | ✗ | break; | |
2090 | ✗ | case 0x02: | |
2091 | ✗ | st->disposition |= AV_DISPOSITION_HEARING_IMPAIRED; | |
2092 | ✗ | break; | |
2093 | ✗ | case 0x03: | |
2094 | ✗ | st->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED; | |
2095 | ✗ | break; | |
2096 | } | ||
2097 | |||
2098 | ✗ | if (flags & 0x01) { /* language_code_present */ | |
2099 | ✗ | if (desc_len < 4) | |
2100 | ✗ | return AVERROR_INVALIDDATA; | |
2101 | ✗ | language[0] = get8(pp, desc_end); | |
2102 | ✗ | language[1] = get8(pp, desc_end); | |
2103 | ✗ | language[2] = get8(pp, desc_end); | |
2104 | ✗ | language[3] = 0; | |
2105 | |||
2106 | /* This language always has to override a possible | ||
2107 | * ISO 639 language descriptor language */ | ||
2108 | ✗ | if (language[0]) | |
2109 | ✗ | av_dict_set(&st->metadata, "language", language, 0); | |
2110 | } | ||
2111 | } | ||
2112 | 4 | break; | |
2113 | 6 | case 0x6a: /* ac-3_descriptor */ | |
2114 | { | ||
2115 | 6 | int component_type_flag = get8(pp, desc_end) & (1 << 7); | |
2116 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (component_type_flag) { |
2117 | ✗ | int component_type = get8(pp, desc_end); | |
2118 | ✗ | int service_type_mask = 0x38; // 0b00111000 | |
2119 | ✗ | int service_type = ((component_type & service_type_mask) >> 3); | |
2120 | ✗ | if (service_type == 0x02 /* 0b010 */) { | |
2121 | ✗ | st->disposition |= AV_DISPOSITION_DESCRIPTIONS; | |
2122 | ✗ | av_log(ts ? ts->stream : fc, AV_LOG_DEBUG, "New track disposition for id %u: %u\n", st->id, st->disposition); | |
2123 | } | ||
2124 | } | ||
2125 | } | ||
2126 | 6 | break; | |
2127 | ✗ | case 0x7a: /* enhanced_ac-3_descriptor */ | |
2128 | { | ||
2129 | ✗ | int component_type_flag = get8(pp, desc_end) & (1 << 7); | |
2130 | ✗ | if (component_type_flag) { | |
2131 | ✗ | int component_type = get8(pp, desc_end); | |
2132 | ✗ | int service_type_mask = 0x38; // 0b00111000 | |
2133 | ✗ | int service_type = ((component_type & service_type_mask) >> 3); | |
2134 | ✗ | if (service_type == 0x02 /* 0b010 */) { | |
2135 | ✗ | st->disposition |= AV_DISPOSITION_DESCRIPTIONS; | |
2136 | ✗ | av_log(ts ? ts->stream : fc, AV_LOG_DEBUG, "New track disposition for id %u: %u\n", st->id, st->disposition); | |
2137 | } | ||
2138 | } | ||
2139 | } | ||
2140 | ✗ | break; | |
2141 | ✗ | case 0xfd: /* ARIB data coding type descriptor */ | |
2142 | // STD-B24, fascicle 3, chapter 4 defines private_stream_1 | ||
2143 | // for captions | ||
2144 | ✗ | if (stream_type == STREAM_TYPE_PRIVATE_DATA) { | |
2145 | // This structure is defined in STD-B10, part 1, listing 5.4 and | ||
2146 | // part 2, 6.2.20). | ||
2147 | // Listing of data_component_ids is in STD-B10, part 2, Annex J. | ||
2148 | // Component tag limits are documented in TR-B14, fascicle 2, | ||
2149 | // Vol. 3, Section 2, 4.2.8.1 | ||
2150 | ✗ | int actual_component_tag = sti->stream_identifier - 1; | |
2151 | ✗ | int picked_profile = AV_PROFILE_UNKNOWN; | |
2152 | ✗ | int data_component_id = get16(pp, desc_end); | |
2153 | ✗ | if (data_component_id < 0) | |
2154 | ✗ | return AVERROR_INVALIDDATA; | |
2155 | |||
2156 | ✗ | switch (data_component_id) { | |
2157 | ✗ | case 0x0008: | |
2158 | // [0x30..0x37] are component tags utilized for | ||
2159 | // non-mobile captioning service ("profile A"). | ||
2160 | ✗ | if (actual_component_tag >= 0x30 && | |
2161 | actual_component_tag <= 0x37) { | ||
2162 | ✗ | picked_profile = AV_PROFILE_ARIB_PROFILE_A; | |
2163 | } | ||
2164 | ✗ | break; | |
2165 | ✗ | case 0x0012: | |
2166 | // component tag 0x87 signifies a mobile/partial reception | ||
2167 | // (1seg) captioning service ("profile C"). | ||
2168 | ✗ | if (actual_component_tag == 0x87) { | |
2169 | ✗ | picked_profile = AV_PROFILE_ARIB_PROFILE_C; | |
2170 | } | ||
2171 | ✗ | break; | |
2172 | ✗ | default: | |
2173 | ✗ | break; | |
2174 | } | ||
2175 | |||
2176 | ✗ | if (picked_profile == AV_PROFILE_UNKNOWN) | |
2177 | ✗ | break; | |
2178 | |||
2179 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; | |
2180 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_ARIB_CAPTION; | |
2181 | ✗ | if (st->codecpar->profile != picked_profile) { | |
2182 | ✗ | st->codecpar->profile = picked_profile; | |
2183 | ✗ | sti->need_context_update = 1; | |
2184 | } | ||
2185 | ✗ | sti->request_probe = 0; | |
2186 | ✗ | sti->need_parsing = 0; | |
2187 | } | ||
2188 | ✗ | break; | |
2189 | ✗ | case 0xb0: /* DOVI video stream descriptor */ | |
2190 | { | ||
2191 | uint32_t buf; | ||
2192 | AVDOVIDecoderConfigurationRecord *dovi; | ||
2193 | size_t dovi_size; | ||
2194 | ✗ | int dependency_pid = -1; // Unset | |
2195 | |||
2196 | ✗ | if (desc_end - *pp < 4) // (8 + 8 + 7 + 6 + 1 + 1 + 1) / 8 | |
2197 | ✗ | return AVERROR_INVALIDDATA; | |
2198 | |||
2199 | ✗ | dovi = av_dovi_alloc(&dovi_size); | |
2200 | ✗ | if (!dovi) | |
2201 | ✗ | return AVERROR(ENOMEM); | |
2202 | |||
2203 | ✗ | dovi->dv_version_major = get8(pp, desc_end); | |
2204 | ✗ | dovi->dv_version_minor = get8(pp, desc_end); | |
2205 | ✗ | buf = get16(pp, desc_end); | |
2206 | ✗ | dovi->dv_profile = (buf >> 9) & 0x7f; // 7 bits | |
2207 | ✗ | dovi->dv_level = (buf >> 3) & 0x3f; // 6 bits | |
2208 | ✗ | dovi->rpu_present_flag = (buf >> 2) & 0x01; // 1 bit | |
2209 | ✗ | dovi->el_present_flag = (buf >> 1) & 0x01; // 1 bit | |
2210 | ✗ | dovi->bl_present_flag = buf & 0x01; // 1 bit | |
2211 | ✗ | if (!dovi->bl_present_flag && desc_end - *pp >= 2) { | |
2212 | ✗ | buf = get16(pp, desc_end); | |
2213 | ✗ | dependency_pid = buf >> 3; // 13 bits | |
2214 | } | ||
2215 | ✗ | if (desc_end - *pp >= 1) { // 8 bits | |
2216 | ✗ | buf = get8(pp, desc_end); | |
2217 | ✗ | dovi->dv_bl_signal_compatibility_id = (buf >> 4) & 0x0f; // 4 bits | |
2218 | ✗ | dovi->dv_md_compression = (buf >> 2) & 0x03; // 2 bits | |
2219 | } else { | ||
2220 | // 0 stands for None | ||
2221 | // Dolby Vision V1.2.93 profiles and levels | ||
2222 | ✗ | dovi->dv_bl_signal_compatibility_id = 0; | |
2223 | ✗ | dovi->dv_md_compression = AV_DOVI_COMPRESSION_NONE; | |
2224 | } | ||
2225 | |||
2226 | ✗ | if (!av_packet_side_data_add(&st->codecpar->coded_side_data, | |
2227 | ✗ | &st->codecpar->nb_coded_side_data, | |
2228 | AV_PKT_DATA_DOVI_CONF, | ||
2229 | (uint8_t *)dovi, dovi_size, 0)) { | ||
2230 | ✗ | av_free(dovi); | |
2231 | ✗ | return AVERROR(ENOMEM); | |
2232 | } | ||
2233 | |||
2234 | ✗ | av_log(fc, AV_LOG_TRACE, "DOVI, version: %d.%d, profile: %d, level: %d, " | |
2235 | "rpu flag: %d, el flag: %d, bl flag: %d, dependency_pid: %d, " | ||
2236 | "compatibility id: %d, compression: %d\n", | ||
2237 | ✗ | dovi->dv_version_major, dovi->dv_version_minor, | |
2238 | ✗ | dovi->dv_profile, dovi->dv_level, | |
2239 | ✗ | dovi->rpu_present_flag, | |
2240 | ✗ | dovi->el_present_flag, | |
2241 | ✗ | dovi->bl_present_flag, | |
2242 | dependency_pid, | ||
2243 | ✗ | dovi->dv_bl_signal_compatibility_id, | |
2244 | ✗ | dovi->dv_md_compression); | |
2245 | } | ||
2246 | ✗ | break; | |
2247 | 60 | default: | |
2248 | 60 | break; | |
2249 | } | ||
2250 | 374 | *pp = desc_end; | |
2251 | 374 | return 0; | |
2252 | } | ||
2253 | |||
2254 | 12 | static AVStream *find_matching_stream(MpegTSContext *ts, int pid, unsigned int programid, | |
2255 | int stream_identifier, int pmt_stream_idx, struct Program *p) | ||
2256 | { | ||
2257 | 12 | AVFormatContext *s = ts->stream; | |
2258 | 12 | AVStream *found = NULL; | |
2259 | |||
2260 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if (stream_identifier) { /* match based on "stream identifier descriptor" if present */ |
2261 | ✗ | for (int i = 0; i < p->nb_streams; i++) { | |
2262 | ✗ | if (p->streams[i].stream_identifier == stream_identifier) | |
2263 | ✗ | if (!found || pmt_stream_idx == i) /* fallback to idx based guess if multiple streams have the same identifier */ | |
2264 | ✗ | found = s->streams[p->streams[i].idx]; | |
2265 | } | ||
2266 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 5 times.
|
12 | } else if (pmt_stream_idx < p->nb_streams) { /* match based on position within the PMT */ |
2267 | 7 | found = s->streams[p->streams[pmt_stream_idx].idx]; | |
2268 | } | ||
2269 | |||
2270 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 5 times.
|
12 | if (found) { |
2271 | 7 | av_log(ts->stream, AV_LOG_VERBOSE, | |
2272 | "re-using existing %s stream %d (pid=0x%x) for new pid=0x%x\n", | ||
2273 | 7 | av_get_media_type_string(found->codecpar->codec_type), | |
2274 | found->index, found->id, pid); | ||
2275 | } | ||
2276 | |||
2277 | 12 | return found; | |
2278 | } | ||
2279 | |||
2280 | 559 | static int parse_stream_identifier_desc(const uint8_t *p, const uint8_t *p_end) | |
2281 | { | ||
2282 | 559 | const uint8_t **pp = &p; | |
2283 | const uint8_t *desc_list_end; | ||
2284 | const uint8_t *desc_end; | ||
2285 | int desc_list_len; | ||
2286 | int desc_len, desc_tag; | ||
2287 | |||
2288 | 559 | desc_list_len = get16(pp, p_end); | |
2289 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 559 times.
|
559 | if (desc_list_len < 0) |
2290 | ✗ | return -1; | |
2291 | 559 | desc_list_len &= 0xfff; | |
2292 | 559 | desc_list_end = p + desc_list_len; | |
2293 |
1/2✓ Branch 0 taken 559 times.
✗ Branch 1 not taken.
|
559 | if (desc_list_end > p_end) |
2294 | ✗ | return -1; | |
2295 | |||
2296 | while (1) { | ||
2297 | 893 | desc_tag = get8(pp, desc_list_end); | |
2298 |
2/2✓ Branch 0 taken 531 times.
✓ Branch 1 taken 362 times.
|
893 | if (desc_tag < 0) |
2299 | 531 | return -1; | |
2300 | 362 | desc_len = get8(pp, desc_list_end); | |
2301 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 362 times.
|
362 | if (desc_len < 0) |
2302 | ✗ | return -1; | |
2303 | 362 | desc_end = *pp + desc_len; | |
2304 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 362 times.
|
362 | if (desc_end > desc_list_end) |
2305 | ✗ | return -1; | |
2306 | |||
2307 |
2/2✓ Branch 0 taken 28 times.
✓ Branch 1 taken 334 times.
|
362 | if (desc_tag == 0x52) { |
2308 | 28 | return get8(pp, desc_end); | |
2309 | } | ||
2310 | 334 | *pp = desc_end; | |
2311 | } | ||
2312 | |||
2313 | return -1; | ||
2314 | } | ||
2315 | |||
2316 | 139 | static int is_pes_stream(int stream_type, uint32_t prog_reg_desc) | |
2317 | { | ||
2318 |
3/4✓ Branch 0 taken 139 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 30 times.
✓ Branch 3 taken 109 times.
|
169 | return !(stream_type == 0x13 || |
2319 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30 times.
|
30 | (stream_type == 0x86 && prog_reg_desc == AV_RL32("CUEI")) ); |
2320 | } | ||
2321 | |||
2322 | 2982 | static void pmt_cb(MpegTSFilter *filter, const uint8_t *section, int section_len) | |
2323 | { | ||
2324 | 2982 | MpegTSContext *ts = filter->u.section_filter.opaque; | |
2325 | 2982 | MpegTSSectionFilter *tssf = &filter->u.section_filter; | |
2326 | struct Program old_program; | ||
2327 | 2982 | SectionHeader h1, *h = &h1; | |
2328 | PESContext *pes; | ||
2329 | AVStream *st; | ||
2330 | const uint8_t *p, *p_end, *desc_list_end; | ||
2331 | int program_info_length, pcr_pid, pid, stream_type; | ||
2332 | int desc_list_len; | ||
2333 | 2982 | uint32_t prog_reg_desc = 0; /* registration descriptor */ | |
2334 | 2982 | int stream_identifier = -1; | |
2335 | struct Program *prg; | ||
2336 | |||
2337 | 2982 | int mp4_descr_count = 0; | |
2338 | 2982 | Mp4Descr mp4_descr[MAX_MP4_DESCR_COUNT] = { { 0 } }; | |
2339 | int i; | ||
2340 | |||
2341 | 2982 | av_log(ts->stream, AV_LOG_TRACE, "PMT: len %i\n", section_len); | |
2342 | hex_dump_debug(ts->stream, section, section_len); | ||
2343 | |||
2344 | 2982 | p_end = section + section_len - 4; | |
2345 | 2982 | p = section; | |
2346 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 2982 times.
|
2982 | if (parse_section_header(h, &p, p_end) < 0) |
2347 | 2711 | return; | |
2348 |
2/2✓ Branch 0 taken 75 times.
✓ Branch 1 taken 2907 times.
|
2982 | if (h->tid != PMT_TID) |
2349 | 75 | return; | |
2350 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2907 times.
|
2907 | if (!h->current_next) |
2351 | ✗ | return; | |
2352 |
2/2✓ Branch 1 taken 2636 times.
✓ Branch 2 taken 271 times.
|
2907 | if (skip_identical(h, tssf)) |
2353 | 2636 | return; | |
2354 | |||
2355 | 271 | av_log(ts->stream, AV_LOG_TRACE, "sid=0x%x sec_num=%d/%d version=%d tid=%d\n", | |
2356 | 271 | h->id, h->sec_num, h->last_sec_num, h->version, h->tid); | |
2357 | |||
2358 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
271 | if (!ts->scan_all_pmts && ts->skip_changes) |
2359 | ✗ | return; | |
2360 | |||
2361 | 271 | prg = get_program(ts, h->id); | |
2362 |
1/2✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
|
271 | if (prg) |
2363 | 271 | old_program = *prg; | |
2364 | else | ||
2365 | ✗ | clear_program(&old_program); | |
2366 | |||
2367 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
271 | if (ts->skip_unknown_pmt && !prg) |
2368 | ✗ | return; | |
2369 |
3/6✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 271 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 271 times.
|
271 | if (prg && prg->nb_pids && prg->pids[0] != ts->current_pid) |
2370 | ✗ | return; | |
2371 |
2/2✓ Branch 0 taken 172 times.
✓ Branch 1 taken 99 times.
|
271 | if (!ts->skip_clear) |
2372 | 172 | clear_avprogram(ts, h->id); | |
2373 | 271 | clear_program(prg); | |
2374 | 271 | add_pid_to_program(prg, ts->current_pid); | |
2375 | |||
2376 | 271 | pcr_pid = get16(&p, p_end); | |
2377 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
|
271 | if (pcr_pid < 0) |
2378 | ✗ | return; | |
2379 | 271 | pcr_pid &= 0x1fff; | |
2380 | 271 | add_pid_to_program(prg, pcr_pid); | |
2381 | 271 | update_av_program_info(ts->stream, h->id, pcr_pid, h->version); | |
2382 | |||
2383 | 271 | av_log(ts->stream, AV_LOG_TRACE, "pcr_pid=0x%x\n", pcr_pid); | |
2384 | |||
2385 | 271 | program_info_length = get16(&p, p_end); | |
2386 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
|
271 | if (program_info_length < 0) |
2387 | ✗ | return; | |
2388 | 271 | program_info_length &= 0xfff; | |
2389 |
2/2✓ Branch 0 taken 246 times.
✓ Branch 1 taken 271 times.
|
517 | while (program_info_length >= 2) { |
2390 | uint8_t tag, len; | ||
2391 | 246 | tag = get8(&p, p_end); | |
2392 | 246 | len = get8(&p, p_end); | |
2393 | |||
2394 | 246 | av_log(ts->stream, AV_LOG_TRACE, "program tag: 0x%02x len=%d\n", tag, len); | |
2395 | |||
2396 | 246 | program_info_length -= 2; | |
2397 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 246 times.
|
246 | if (len > program_info_length) |
2398 | // something else is broken, exit the program_descriptors_loop | ||
2399 | ✗ | break; | |
2400 | 246 | program_info_length -= len; | |
2401 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 246 times.
|
246 | if (tag == IOD_DESCRIPTOR) { |
2402 | ✗ | get8(&p, p_end); // scope | |
2403 | ✗ | get8(&p, p_end); // label | |
2404 | ✗ | len -= 2; | |
2405 | ✗ | mp4_read_iods(ts->stream, p, len, mp4_descr + mp4_descr_count, | |
2406 | &mp4_descr_count, MAX_MP4_DESCR_COUNT); | ||
2407 |
3/4✓ Branch 0 taken 120 times.
✓ Branch 1 taken 126 times.
✓ Branch 2 taken 120 times.
✗ Branch 3 not taken.
|
246 | } else if (tag == REGISTRATION_DESCRIPTOR && len >= 4) { |
2408 | 120 | prog_reg_desc = bytestream_get_le32(&p); | |
2409 | 120 | len -= 4; | |
2410 | } | ||
2411 | 246 | p += len; | |
2412 | } | ||
2413 | 271 | p += program_info_length; | |
2414 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
|
271 | if (p >= p_end) |
2415 | ✗ | goto out; | |
2416 | |||
2417 | // stop parsing after pmt, we found header | ||
2418 |
2/2✓ Branch 0 taken 55 times.
✓ Branch 1 taken 216 times.
|
271 | if (!ts->pkt) |
2419 | 55 | ts->stop_parse = 2; | |
2420 | |||
2421 |
1/2✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
|
271 | if (prg) |
2422 | 271 | prg->pmt_found = 1; | |
2423 | |||
2424 |
1/2✓ Branch 0 taken 830 times.
✗ Branch 1 not taken.
|
830 | for (i = 0; i < MAX_STREAMS_PER_PROGRAM; i++) { |
2425 | 830 | st = 0; | |
2426 | 830 | pes = NULL; | |
2427 | 830 | stream_type = get8(&p, p_end); | |
2428 |
2/2✓ Branch 0 taken 271 times.
✓ Branch 1 taken 559 times.
|
830 | if (stream_type < 0) |
2429 | 271 | break; | |
2430 | 559 | pid = get16(&p, p_end); | |
2431 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 559 times.
|
559 | if (pid < 0) |
2432 | ✗ | goto out; | |
2433 | 559 | pid &= 0x1fff; | |
2434 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 559 times.
|
559 | if (pid == ts->current_pid) |
2435 | ✗ | goto out; | |
2436 | |||
2437 | 559 | stream_identifier = parse_stream_identifier_desc(p, p_end) + 1; | |
2438 | |||
2439 | /* now create stream */ | ||
2440 |
4/4✓ Branch 0 taken 440 times.
✓ Branch 1 taken 119 times.
✓ Branch 2 taken 420 times.
✓ Branch 3 taken 20 times.
|
559 | if (ts->pids[pid] && ts->pids[pid]->type == MPEGTS_PES) { |
2441 | 420 | pes = ts->pids[pid]->u.pes_filter.opaque; | |
2442 |
3/4✓ Branch 0 taken 16 times.
✓ Branch 1 taken 404 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 16 times.
|
420 | if (ts->merge_pmt_versions && !pes->st) { |
2443 | ✗ | st = find_matching_stream(ts, pid, h->id, stream_identifier, i, &old_program); | |
2444 | ✗ | if (st) { | |
2445 | ✗ | pes->st = st; | |
2446 | ✗ | pes->stream_type = stream_type; | |
2447 | ✗ | pes->merged_st = 1; | |
2448 | } | ||
2449 | } | ||
2450 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 420 times.
|
420 | if (!pes->st) { |
2451 | ✗ | pes->st = avformat_new_stream(pes->stream, NULL); | |
2452 | ✗ | if (!pes->st) | |
2453 | ✗ | goto out; | |
2454 | ✗ | pes->st->id = pes->pid; | |
2455 | } | ||
2456 | 420 | st = pes->st; | |
2457 |
2/2✓ Branch 1 taken 109 times.
✓ Branch 2 taken 30 times.
|
139 | } else if (is_pes_stream(stream_type, prog_reg_desc)) { |
2458 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 109 times.
|
109 | if (ts->pids[pid]) |
2459 | ✗ | mpegts_close_filter(ts, ts->pids[pid]); // wrongly added sdt filter probably | |
2460 | 109 | pes = add_pes_stream(ts, pid, pcr_pid); | |
2461 |
4/6✓ Branch 0 taken 8 times.
✓ Branch 1 taken 101 times.
✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 8 times.
✗ Branch 5 not taken.
|
109 | if (ts->merge_pmt_versions && pes && !pes->st) { |
2462 | 8 | st = find_matching_stream(ts, pid, h->id, stream_identifier, i, &old_program); | |
2463 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
|
8 | if (st) { |
2464 | 4 | pes->st = st; | |
2465 | 4 | pes->stream_type = stream_type; | |
2466 | 4 | pes->merged_st = 1; | |
2467 | } | ||
2468 | } | ||
2469 |
3/4✓ Branch 0 taken 109 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 105 times.
✓ Branch 3 taken 4 times.
|
109 | if (pes && !pes->st) { |
2470 | 105 | st = avformat_new_stream(pes->stream, NULL); | |
2471 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 105 times.
|
105 | if (!st) |
2472 | ✗ | goto out; | |
2473 | 105 | st->id = pes->pid; | |
2474 | } | ||
2475 | } else { | ||
2476 | 30 | int idx = ff_find_stream_index(ts->stream, pid); | |
2477 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 10 times.
|
30 | if (idx >= 0) { |
2478 | 20 | st = ts->stream->streams[idx]; | |
2479 | } | ||
2480 |
4/4✓ Branch 0 taken 6 times.
✓ Branch 1 taken 24 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 2 times.
|
30 | if (ts->merge_pmt_versions && !st) { |
2481 | 4 | st = find_matching_stream(ts, pid, h->id, stream_identifier, i, &old_program); | |
2482 | } | ||
2483 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 23 times.
|
30 | if (!st) { |
2484 | 7 | st = avformat_new_stream(ts->stream, NULL); | |
2485 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
|
7 | if (!st) |
2486 | ✗ | goto out; | |
2487 | 7 | st->id = pid; | |
2488 | 7 | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
2489 |
2/4✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
✗ Branch 3 not taken.
|
7 | if (stream_type == 0x86 && prog_reg_desc == AV_RL32("CUEI")) { |
2490 | 7 | mpegts_find_stream_type(st, stream_type, SCTE_types); | |
2491 | 7 | mpegts_open_section_filter(ts, pid, scte_data_cb, ts, 1); | |
2492 | } | ||
2493 | } | ||
2494 | } | ||
2495 | |||
2496 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 559 times.
|
559 | if (!st) |
2497 | ✗ | goto out; | |
2498 | |||
2499 |
4/4✓ Branch 0 taken 529 times.
✓ Branch 1 taken 30 times.
✓ Branch 2 taken 105 times.
✓ Branch 3 taken 424 times.
|
559 | if (pes && !pes->stream_type) |
2500 | 105 | mpegts_set_stream_info(st, pes, stream_type, prog_reg_desc); | |
2501 | |||
2502 | 559 | add_pid_to_program(prg, pid); | |
2503 |
1/2✓ Branch 0 taken 559 times.
✗ Branch 1 not taken.
|
559 | if (prg) { |
2504 | 559 | prg->streams[i].idx = st->index; | |
2505 | 559 | prg->streams[i].stream_identifier = stream_identifier; | |
2506 | 559 | prg->nb_streams++; | |
2507 | } | ||
2508 | |||
2509 | 559 | av_program_add_stream_index(ts->stream, h->id, st->index); | |
2510 | |||
2511 | 559 | desc_list_len = get16(&p, p_end); | |
2512 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 559 times.
|
559 | if (desc_list_len < 0) |
2513 | ✗ | goto out; | |
2514 | 559 | desc_list_len &= 0xfff; | |
2515 | 559 | desc_list_end = p + desc_list_len; | |
2516 |
1/2✓ Branch 0 taken 559 times.
✗ Branch 1 not taken.
|
559 | if (desc_list_end > p_end) |
2517 | ✗ | goto out; | |
2518 | for (;;) { | ||
2519 |
2/2✓ Branch 1 taken 559 times.
✓ Branch 2 taken 370 times.
|
929 | if (ff_parse_mpeg2_descriptor(ts->stream, st, stream_type, &p, |
2520 | desc_list_end, mp4_descr, | ||
2521 | mp4_descr_count, pid, ts) < 0) | ||
2522 | 559 | break; | |
2523 | |||
2524 |
2/6✓ Branch 0 taken 370 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 370 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
370 | if (pes && prog_reg_desc == AV_RL32("HDMV") && |
2525 | ✗ | stream_type == 0x83 && pes->sub_st) { | |
2526 | ✗ | av_program_add_stream_index(ts->stream, h->id, | |
2527 | ✗ | pes->sub_st->index); | |
2528 | ✗ | pes->sub_st->codecpar->codec_tag = st->codecpar->codec_tag; | |
2529 | } | ||
2530 | } | ||
2531 | 559 | p = desc_list_end; | |
2532 | } | ||
2533 | |||
2534 |
1/2✓ Branch 0 taken 271 times.
✗ Branch 1 not taken.
|
271 | if (!ts->pids[pcr_pid]) |
2535 | ✗ | mpegts_open_pcr_filter(ts, pcr_pid); | |
2536 | |||
2537 | 271 | out: | |
2538 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 271 times.
|
271 | for (i = 0; i < mp4_descr_count; i++) |
2539 | ✗ | av_free(mp4_descr[i].dec_config_descr); | |
2540 | } | ||
2541 | |||
2542 | 3009 | static void pat_cb(MpegTSFilter *filter, const uint8_t *section, int section_len) | |
2543 | { | ||
2544 | 3009 | MpegTSContext *ts = filter->u.section_filter.opaque; | |
2545 | 3009 | MpegTSSectionFilter *tssf = &filter->u.section_filter; | |
2546 | 3009 | SectionHeader h1, *h = &h1; | |
2547 | const uint8_t *p, *p_end; | ||
2548 | int sid, pmt_pid; | ||
2549 | 3009 | int nb_prg = 0; | |
2550 | AVProgram *program; | ||
2551 | |||
2552 | 3009 | av_log(ts->stream, AV_LOG_TRACE, "PAT:\n"); | |
2553 | hex_dump_debug(ts->stream, section, section_len); | ||
2554 | |||
2555 | 3009 | p_end = section + section_len - 4; | |
2556 | 3009 | p = section; | |
2557 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 3009 times.
|
3009 | if (parse_section_header(h, &p, p_end) < 0) |
2558 | 2796 | return; | |
2559 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3009 times.
|
3009 | if (h->tid != PAT_TID) |
2560 | ✗ | return; | |
2561 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3009 times.
|
3009 | if (!h->current_next) |
2562 | ✗ | return; | |
2563 |
2/2✓ Branch 0 taken 536 times.
✓ Branch 1 taken 2473 times.
|
3009 | if (ts->skip_changes) |
2564 | 536 | return; | |
2565 | |||
2566 |
2/2✓ Branch 1 taken 2260 times.
✓ Branch 2 taken 213 times.
|
2473 | if (skip_identical(h, tssf)) |
2567 | 2260 | return; | |
2568 | 213 | ts->id = h->id; | |
2569 | |||
2570 | for (;;) { | ||
2571 | 426 | sid = get16(&p, p_end); | |
2572 |
2/2✓ Branch 0 taken 213 times.
✓ Branch 1 taken 213 times.
|
426 | if (sid < 0) |
2573 | 213 | break; | |
2574 | 213 | pmt_pid = get16(&p, p_end); | |
2575 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 213 times.
|
213 | if (pmt_pid < 0) |
2576 | ✗ | break; | |
2577 | 213 | pmt_pid &= 0x1fff; | |
2578 | |||
2579 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 213 times.
|
213 | if (pmt_pid == ts->current_pid) |
2580 | ✗ | break; | |
2581 | |||
2582 | 213 | av_log(ts->stream, AV_LOG_TRACE, "sid=0x%x pid=0x%x\n", sid, pmt_pid); | |
2583 | |||
2584 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (sid == 0x0000) { |
2585 | /* NIT info */ | ||
2586 | } else { | ||
2587 | 213 | MpegTSFilter *fil = ts->pids[pmt_pid]; | |
2588 | struct Program *prg; | ||
2589 | 213 | program = av_new_program(ts->stream, sid); | |
2590 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (program) { |
2591 | 213 | program->program_num = sid; | |
2592 | 213 | program->pmt_pid = pmt_pid; | |
2593 | } | ||
2594 |
2/2✓ Branch 0 taken 158 times.
✓ Branch 1 taken 55 times.
|
213 | if (fil) |
2595 |
1/2✓ Branch 0 taken 158 times.
✗ Branch 1 not taken.
|
158 | if ( fil->type != MPEGTS_SECTION |
2596 |
1/2✓ Branch 0 taken 158 times.
✗ Branch 1 not taken.
|
158 | || fil->pid != pmt_pid |
2597 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 158 times.
|
158 | || fil->u.section_filter.section_cb != pmt_cb) |
2598 | ✗ | mpegts_close_filter(ts, ts->pids[pmt_pid]); | |
2599 | |||
2600 |
2/2✓ Branch 0 taken 55 times.
✓ Branch 1 taken 158 times.
|
213 | if (!ts->pids[pmt_pid]) |
2601 | 55 | mpegts_open_section_filter(ts, pmt_pid, pmt_cb, ts, 1); | |
2602 | 213 | prg = add_program(ts, sid); | |
2603 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (prg) { |
2604 | 213 | unsigned prg_idx = prg - ts->prg; | |
2605 |
3/4✓ Branch 0 taken 158 times.
✓ Branch 1 taken 55 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 158 times.
|
213 | if (prg->nb_pids && prg->pids[0] != pmt_pid) |
2606 | ✗ | clear_program(prg); | |
2607 | 213 | add_pid_to_program(prg, pmt_pid); | |
2608 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 213 times.
|
213 | if (prg_idx > nb_prg) |
2609 | ✗ | FFSWAP(struct Program, ts->prg[nb_prg], ts->prg[prg_idx]); | |
2610 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (prg_idx >= nb_prg) |
2611 | 213 | nb_prg++; | |
2612 | } else | ||
2613 | ✗ | nb_prg = 0; | |
2614 | } | ||
2615 | } | ||
2616 | 213 | ts->nb_prg = nb_prg; | |
2617 | |||
2618 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (sid < 0) { |
2619 | int i,j; | ||
2620 |
2/2✓ Branch 0 taken 213 times.
✓ Branch 1 taken 213 times.
|
426 | for (j=0; j<ts->stream->nb_programs; j++) { |
2621 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | for (i = 0; i < ts->nb_prg; i++) |
2622 |
1/2✓ Branch 0 taken 213 times.
✗ Branch 1 not taken.
|
213 | if (ts->prg[i].id == ts->stream->programs[j]->id) |
2623 | 213 | break; | |
2624 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 213 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
213 | if (i==ts->nb_prg && !ts->skip_clear) |
2625 | ✗ | clear_avprogram(ts, ts->stream->programs[j]->id); | |
2626 | } | ||
2627 | } | ||
2628 | } | ||
2629 | |||
2630 | 14 | static void eit_cb(MpegTSFilter *filter, const uint8_t *section, int section_len) | |
2631 | { | ||
2632 | 14 | MpegTSContext *ts = filter->u.section_filter.opaque; | |
2633 | const uint8_t *p, *p_end; | ||
2634 | 14 | SectionHeader h1, *h = &h1; | |
2635 | |||
2636 | /* | ||
2637 | * Sometimes we receive EPG packets but SDT table do not have | ||
2638 | * eit_pres_following or eit_sched turned on, so we open EPG | ||
2639 | * stream directly here. | ||
2640 | */ | ||
2641 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 11 times.
|
14 | if (!ts->epg_stream) { |
2642 | 3 | ts->epg_stream = avformat_new_stream(ts->stream, NULL); | |
2643 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (!ts->epg_stream) |
2644 | 2 | return; | |
2645 | 3 | ts->epg_stream->id = EIT_PID; | |
2646 | 3 | ts->epg_stream->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
2647 | 3 | ts->epg_stream->codecpar->codec_id = AV_CODEC_ID_EPG; | |
2648 | } | ||
2649 | |||
2650 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 12 times.
|
14 | if (ts->epg_stream->discard == AVDISCARD_ALL) |
2651 | 2 | return; | |
2652 | |||
2653 | 12 | p_end = section + section_len - 4; | |
2654 | 12 | p = section; | |
2655 | |||
2656 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
|
12 | if (parse_section_header(h, &p, p_end) < 0) |
2657 | ✗ | return; | |
2658 |
2/4✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 12 times.
|
12 | if (h->tid < EIT_TID || h->tid > OEITS_END_TID) |
2659 | ✗ | return; | |
2660 | |||
2661 | 12 | av_log(ts->stream, AV_LOG_TRACE, "EIT: tid received = %.02x\n", h->tid); | |
2662 | |||
2663 | /** | ||
2664 | * Service_id 0xFFFF is reserved, it indicates that the current EIT table | ||
2665 | * is scrambled. | ||
2666 | */ | ||
2667 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if (h->id == 0xFFFF) { |
2668 | ✗ | av_log(ts->stream, AV_LOG_TRACE, "Scrambled EIT table received.\n"); | |
2669 | ✗ | return; | |
2670 | } | ||
2671 | |||
2672 | /** | ||
2673 | * In case we receive an EPG packet before mpegts context is fully | ||
2674 | * initialized. | ||
2675 | */ | ||
2676 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if (!ts->pkt) |
2677 | ✗ | return; | |
2678 | |||
2679 | 12 | new_data_packet(section, section_len, ts->pkt); | |
2680 | 12 | ts->pkt->stream_index = ts->epg_stream->index; | |
2681 | 12 | ts->stop_parse = 1; | |
2682 | } | ||
2683 | |||
2684 | 442 | static void sdt_cb(MpegTSFilter *filter, const uint8_t *section, int section_len) | |
2685 | { | ||
2686 | 442 | MpegTSContext *ts = filter->u.section_filter.opaque; | |
2687 | 442 | MpegTSSectionFilter *tssf = &filter->u.section_filter; | |
2688 | 442 | SectionHeader h1, *h = &h1; | |
2689 | const uint8_t *p, *p_end, *desc_list_end, *desc_end; | ||
2690 | int onid, val, sid, desc_list_len, desc_tag, desc_len, service_type; | ||
2691 | char *name, *provider_name; | ||
2692 | |||
2693 | 442 | av_log(ts->stream, AV_LOG_TRACE, "SDT:\n"); | |
2694 | hex_dump_debug(ts->stream, section, section_len); | ||
2695 | |||
2696 | 442 | p_end = section + section_len - 4; | |
2697 | 442 | p = section; | |
2698 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 442 times.
|
442 | if (parse_section_header(h, &p, p_end) < 0) |
2699 | 342 | return; | |
2700 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 442 times.
|
442 | if (h->tid != SDT_TID) |
2701 | ✗ | return; | |
2702 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 442 times.
|
442 | if (!h->current_next) |
2703 | ✗ | return; | |
2704 |
2/2✓ Branch 0 taken 55 times.
✓ Branch 1 taken 387 times.
|
442 | if (ts->skip_changes) |
2705 | 55 | return; | |
2706 |
2/2✓ Branch 1 taken 287 times.
✓ Branch 2 taken 100 times.
|
387 | if (skip_identical(h, tssf)) |
2707 | 287 | return; | |
2708 | |||
2709 | 100 | onid = get16(&p, p_end); | |
2710 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
|
100 | if (onid < 0) |
2711 | ✗ | return; | |
2712 | 100 | val = get8(&p, p_end); | |
2713 |
1/2✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
|
100 | if (val < 0) |
2714 | ✗ | return; | |
2715 | for (;;) { | ||
2716 | 200 | sid = get16(&p, p_end); | |
2717 |
2/2✓ Branch 0 taken 100 times.
✓ Branch 1 taken 100 times.
|
200 | if (sid < 0) |
2718 | 100 | break; | |
2719 | 100 | val = get8(&p, p_end); | |
2720 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
|
100 | if (val < 0) |
2721 | ✗ | break; | |
2722 | 100 | desc_list_len = get16(&p, p_end); | |
2723 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
|
100 | if (desc_list_len < 0) |
2724 | ✗ | break; | |
2725 | 100 | desc_list_len &= 0xfff; | |
2726 | 100 | desc_list_end = p + desc_list_len; | |
2727 |
1/2✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
|
100 | if (desc_list_end > p_end) |
2728 | ✗ | break; | |
2729 | for (;;) { | ||
2730 | 200 | desc_tag = get8(&p, desc_list_end); | |
2731 |
2/2✓ Branch 0 taken 100 times.
✓ Branch 1 taken 100 times.
|
200 | if (desc_tag < 0) |
2732 | 100 | break; | |
2733 | 100 | desc_len = get8(&p, desc_list_end); | |
2734 | 100 | desc_end = p + desc_len; | |
2735 |
2/4✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 100 times.
✗ Branch 3 not taken.
|
100 | if (desc_len < 0 || desc_end > desc_list_end) |
2736 | break; | ||
2737 | |||
2738 | 100 | av_log(ts->stream, AV_LOG_TRACE, "tag: 0x%02x len=%d\n", | |
2739 | desc_tag, desc_len); | ||
2740 | |||
2741 |
1/2✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
|
100 | switch (desc_tag) { |
2742 | 100 | case 0x48: | |
2743 | 100 | service_type = get8(&p, desc_end); | |
2744 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
|
100 | if (service_type < 0) |
2745 | ✗ | break; | |
2746 | 100 | provider_name = getstr8(&p, desc_end); | |
2747 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 100 times.
|
100 | if (!provider_name) |
2748 | ✗ | break; | |
2749 | 100 | name = getstr8(&p, desc_end); | |
2750 |
1/2✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
|
100 | if (name) { |
2751 | 100 | AVProgram *program = av_new_program(ts->stream, sid); | |
2752 |
1/2✓ Branch 0 taken 100 times.
✗ Branch 1 not taken.
|
100 | if (program) { |
2753 | 100 | av_dict_set(&program->metadata, "service_name", name, 0); | |
2754 | 100 | av_dict_set(&program->metadata, "service_provider", | |
2755 | provider_name, 0); | ||
2756 | } | ||
2757 | } | ||
2758 | 100 | av_free(name); | |
2759 | 100 | av_free(provider_name); | |
2760 | 100 | break; | |
2761 | ✗ | default: | |
2762 | ✗ | break; | |
2763 | } | ||
2764 | 100 | p = desc_end; | |
2765 | } | ||
2766 | 100 | p = desc_list_end; | |
2767 | } | ||
2768 | } | ||
2769 | |||
2770 | static int parse_pcr(int64_t *ppcr_high, int *ppcr_low, | ||
2771 | const uint8_t *packet); | ||
2772 | |||
2773 | /* handle one TS packet */ | ||
2774 | 398491 | static int handle_packet(MpegTSContext *ts, const uint8_t *packet, int64_t pos) | |
2775 | { | ||
2776 | MpegTSFilter *tss; | ||
2777 | int len, pid, cc, expected_cc, cc_ok, afc, is_start, is_discontinuity, | ||
2778 | has_adaptation, has_payload; | ||
2779 | const uint8_t *p, *p_end; | ||
2780 | |||
2781 | 398491 | pid = AV_RB16(packet + 1) & 0x1fff; | |
2782 | 398491 | is_start = packet[1] & 0x40; | |
2783 | 398491 | tss = ts->pids[pid]; | |
2784 |
6/6✓ Branch 0 taken 390480 times.
✓ Branch 1 taken 8011 times.
✓ Branch 2 taken 6809 times.
✓ Branch 3 taken 383671 times.
✓ Branch 4 taken 17 times.
✓ Branch 5 taken 6792 times.
|
398491 | if (ts->auto_guess && !tss && is_start) { |
2785 | 17 | add_pes_stream(ts, pid, -1); | |
2786 | 17 | tss = ts->pids[pid]; | |
2787 | } | ||
2788 |
2/2✓ Branch 0 taken 14662 times.
✓ Branch 1 taken 383829 times.
|
398491 | if (!tss) |
2789 | 14662 | return 0; | |
2790 |
2/2✓ Branch 0 taken 36133 times.
✓ Branch 1 taken 347696 times.
|
383829 | if (is_start) |
2791 | 36133 | tss->discard = discard_pid(ts, pid); | |
2792 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 383829 times.
|
383829 | if (tss->discard) |
2793 | ✗ | return 0; | |
2794 | 383829 | ts->current_pid = pid; | |
2795 | |||
2796 | 383829 | afc = (packet[3] >> 4) & 3; | |
2797 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 383829 times.
|
383829 | if (afc == 0) /* reserved value */ |
2798 | ✗ | return 0; | |
2799 | 383829 | has_adaptation = afc & 2; | |
2800 | 383829 | has_payload = afc & 1; | |
2801 | 443389 | is_discontinuity = has_adaptation && | |
2802 |
4/4✓ Branch 0 taken 59560 times.
✓ Branch 1 taken 324269 times.
✓ Branch 2 taken 59522 times.
✓ Branch 3 taken 38 times.
|
443351 | packet[4] != 0 && /* with length > 0 */ |
2803 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59522 times.
|
59522 | (packet[5] & 0x80); /* and discontinuity indicated */ |
2804 | |||
2805 | /* continuity check (currently not used) */ | ||
2806 | 383829 | cc = (packet[3] & 0xf); | |
2807 |
2/2✓ Branch 0 taken 380983 times.
✓ Branch 1 taken 2846 times.
|
383829 | expected_cc = has_payload ? (tss->last_cc + 1) & 0x0f : tss->last_cc; |
2808 |
1/2✓ Branch 0 taken 383829 times.
✗ Branch 1 not taken.
|
383829 | cc_ok = pid == 0x1FFF || // null packet PID |
2809 | 383829 | is_discontinuity || | |
2810 |
5/6✓ Branch 0 taken 383829 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 382681 times.
✓ Branch 3 taken 1148 times.
✓ Branch 4 taken 382037 times.
✓ Branch 5 taken 644 times.
|
767658 | tss->last_cc < 0 || |
2811 | expected_cc == cc; | ||
2812 | |||
2813 | 383829 | tss->last_cc = cc; | |
2814 |
2/2✓ Branch 0 taken 644 times.
✓ Branch 1 taken 383185 times.
|
383829 | if (!cc_ok) { |
2815 | 644 | av_log(ts->stream, AV_LOG_DEBUG, | |
2816 | "Continuity check failed for pid %d expected %d got %d\n", | ||
2817 | pid, expected_cc, cc); | ||
2818 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 637 times.
|
644 | if (tss->type == MPEGTS_PES) { |
2819 | 7 | PESContext *pc = tss->u.pes_filter.opaque; | |
2820 | 7 | pc->flags |= AV_PKT_FLAG_CORRUPT; | |
2821 | } | ||
2822 | } | ||
2823 | |||
2824 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 383817 times.
|
383829 | if (packet[1] & 0x80) { |
2825 | 12 | av_log(ts->stream, AV_LOG_DEBUG, "Packet had TEI flag set; marking as corrupt\n"); | |
2826 |
1/2✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
|
12 | if (tss->type == MPEGTS_PES) { |
2827 | 12 | PESContext *pc = tss->u.pes_filter.opaque; | |
2828 | 12 | pc->flags |= AV_PKT_FLAG_CORRUPT; | |
2829 | } | ||
2830 | } | ||
2831 | |||
2832 | 383829 | p = packet + 4; | |
2833 |
2/2✓ Branch 0 taken 59560 times.
✓ Branch 1 taken 324269 times.
|
383829 | if (has_adaptation) { |
2834 | int64_t pcr_h; | ||
2835 | int pcr_l; | ||
2836 |
2/2✓ Branch 1 taken 27593 times.
✓ Branch 2 taken 31967 times.
|
59560 | if (parse_pcr(&pcr_h, &pcr_l, packet) == 0) |
2837 | 27593 | tss->last_pcr = pcr_h * 300 + pcr_l; | |
2838 | /* skip adaptation field */ | ||
2839 | 59560 | p += p[0] + 1; | |
2840 | } | ||
2841 | /* if past the end of packet, ignore */ | ||
2842 | 383829 | p_end = packet + TS_PACKET_SIZE; | |
2843 |
3/4✓ Branch 0 taken 380971 times.
✓ Branch 1 taken 2858 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 380971 times.
|
383829 | if (p >= p_end || !has_payload) |
2844 | 2858 | return 0; | |
2845 | |||
2846 |
1/2✓ Branch 0 taken 380971 times.
✗ Branch 1 not taken.
|
380971 | if (pos >= 0) { |
2847 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 380971 times.
|
380971 | av_assert0(pos >= TS_PACKET_SIZE); |
2848 | 380971 | ts->pos47_full = pos - TS_PACKET_SIZE; | |
2849 | } | ||
2850 | |||
2851 |
2/2✓ Branch 0 taken 6482 times.
✓ Branch 1 taken 374489 times.
|
380971 | if (tss->type == MPEGTS_SECTION) { |
2852 |
2/2✓ Branch 0 taken 6447 times.
✓ Branch 1 taken 35 times.
|
6482 | if (is_start) { |
2853 | /* pointer field present */ | ||
2854 | 6447 | len = *p++; | |
2855 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6447 times.
|
6447 | if (len > p_end - p) |
2856 | ✗ | return 0; | |
2857 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 6447 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
6447 | if (len && cc_ok) { |
2858 | /* write remaining section bytes */ | ||
2859 | ✗ | write_section_data(ts, tss, | |
2860 | p, len, 0); | ||
2861 | /* check whether filter has been closed */ | ||
2862 | ✗ | if (!ts->pids[pid]) | |
2863 | ✗ | return 0; | |
2864 | } | ||
2865 | 6447 | p += len; | |
2866 |
1/2✓ Branch 0 taken 6447 times.
✗ Branch 1 not taken.
|
6447 | if (p < p_end) { |
2867 | 6447 | write_section_data(ts, tss, | |
2868 | 6447 | p, p_end - p, 1); | |
2869 | } | ||
2870 | } else { | ||
2871 |
1/2✓ Branch 0 taken 35 times.
✗ Branch 1 not taken.
|
35 | if (cc_ok) { |
2872 | 35 | write_section_data(ts, tss, | |
2873 | 35 | p, p_end - p, 0); | |
2874 | } | ||
2875 | } | ||
2876 | |||
2877 | // stop find_stream_info from waiting for more streams | ||
2878 | // when all programs have received a PMT | ||
2879 |
4/4✓ Branch 0 taken 4439 times.
✓ Branch 1 taken 2043 times.
✓ Branch 2 taken 17 times.
✓ Branch 3 taken 4422 times.
|
6482 | if (ts->stream->ctx_flags & AVFMTCTX_NOHEADER && ts->scan_all_pmts <= 0) { |
2880 | int i; | ||
2881 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 17 times.
|
34 | for (i = 0; i < ts->nb_prg; i++) { |
2882 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
|
17 | if (!ts->prg[i].pmt_found) |
2883 | ✗ | break; | |
2884 | } | ||
2885 |
2/4✓ Branch 0 taken 17 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 17 times.
✗ Branch 3 not taken.
|
17 | if (i == ts->nb_prg && ts->nb_prg > 0) { |
2886 | 17 | av_log(ts->stream, AV_LOG_DEBUG, "All programs have pmt, headers found\n"); | |
2887 | 17 | ts->stream->ctx_flags &= ~AVFMTCTX_NOHEADER; | |
2888 | } | ||
2889 | } | ||
2890 | |||
2891 | } else { | ||
2892 | int ret; | ||
2893 | // Note: The position here points actually behind the current packet. | ||
2894 |
1/2✓ Branch 0 taken 374489 times.
✗ Branch 1 not taken.
|
374489 | if (tss->type == MPEGTS_PES) { |
2895 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 374489 times.
|
374489 | if ((ret = tss->u.pes_filter.pes_cb(tss, p, p_end - p, is_start, |
2896 | 374489 | pos - ts->raw_packet_size)) < 0) | |
2897 | ✗ | return ret; | |
2898 | } | ||
2899 | } | ||
2900 | |||
2901 | 380971 | return 0; | |
2902 | } | ||
2903 | |||
2904 | 59 | static int mpegts_resync(AVFormatContext *s, int seekback, const uint8_t *current_packet) | |
2905 | { | ||
2906 | 59 | MpegTSContext *ts = s->priv_data; | |
2907 | 59 | AVIOContext *pb = s->pb; | |
2908 | int c, i; | ||
2909 | 59 | uint64_t pos = avio_tell(pb); | |
2910 | 59 | int64_t back = FFMIN(seekback, pos); | |
2911 | |||
2912 | //Special case for files like 01c56b0dc1.ts | ||
2913 |
3/6✓ Branch 0 taken 6 times.
✓ Branch 1 taken 53 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
59 | if (current_packet[0] == 0x80 && current_packet[12] == 0x47 && pos >= TS_PACKET_SIZE) { |
2914 | ✗ | avio_seek(pb, 12 - TS_PACKET_SIZE, SEEK_CUR); | |
2915 | ✗ | return 0; | |
2916 | } | ||
2917 | |||
2918 | 59 | avio_seek(pb, -back, SEEK_CUR); | |
2919 | |||
2920 |
1/2✓ Branch 0 taken 4257 times.
✗ Branch 1 not taken.
|
4257 | for (i = 0; i < ts->resync_size; i++) { |
2921 | 4257 | c = avio_r8(pb); | |
2922 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4257 times.
|
4257 | if (avio_feof(pb)) |
2923 | ✗ | return AVERROR_EOF; | |
2924 |
2/2✓ Branch 0 taken 59 times.
✓ Branch 1 taken 4198 times.
|
4257 | if (c == 0x47) { |
2925 | int new_packet_size, ret; | ||
2926 | 59 | avio_seek(pb, -1, SEEK_CUR); | |
2927 | 59 | pos = avio_tell(pb); | |
2928 | 59 | ret = ffio_ensure_seekback(pb, PROBE_PACKET_MAX_BUF); | |
2929 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (ret < 0) |
2930 | ✗ | return ret; | |
2931 | 59 | new_packet_size = get_packet_size(s); | |
2932 |
2/4✓ Branch 0 taken 59 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 59 times.
|
59 | if (new_packet_size > 0 && new_packet_size != ts->raw_packet_size) { |
2933 | ✗ | av_log(ts->stream, AV_LOG_WARNING, "changing packet size to %d\n", new_packet_size); | |
2934 | ✗ | ts->raw_packet_size = new_packet_size; | |
2935 | } | ||
2936 | 59 | avio_seek(pb, pos, SEEK_SET); | |
2937 | 59 | return 0; | |
2938 | } | ||
2939 | } | ||
2940 | ✗ | av_log(s, AV_LOG_ERROR, | |
2941 | "max resync size reached, could not find sync byte\n"); | ||
2942 | /* no sync found */ | ||
2943 | ✗ | return AVERROR_INVALIDDATA; | |
2944 | } | ||
2945 | |||
2946 | /* return AVERROR_something if error or EOF. Return 0 if OK. */ | ||
2947 | 398864 | static int read_packet(AVFormatContext *s, uint8_t *buf, int raw_packet_size, | |
2948 | const uint8_t **data) | ||
2949 | { | ||
2950 | 398864 | AVIOContext *pb = s->pb; | |
2951 | int len; | ||
2952 | |||
2953 | for (;;) { | ||
2954 | 398923 | len = ffio_read_indirect(pb, buf, TS_PACKET_SIZE, data); | |
2955 |
2/2✓ Branch 0 taken 373 times.
✓ Branch 1 taken 398550 times.
|
398923 | if (len != TS_PACKET_SIZE) |
2956 |
2/2✓ Branch 0 taken 316 times.
✓ Branch 1 taken 57 times.
|
373 | return len < 0 ? len : AVERROR_EOF; |
2957 | /* check packet sync byte */ | ||
2958 |
2/2✓ Branch 0 taken 59 times.
✓ Branch 1 taken 398491 times.
|
398550 | if ((*data)[0] != 0x47) { |
2959 | /* find a new packet start */ | ||
2960 | |||
2961 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 59 times.
|
59 | if (mpegts_resync(s, raw_packet_size, *data) < 0) |
2962 | ✗ | return AVERROR(EAGAIN); | |
2963 | else | ||
2964 | 59 | continue; | |
2965 | } else { | ||
2966 | 398491 | break; | |
2967 | } | ||
2968 | } | ||
2969 | 398491 | return 0; | |
2970 | } | ||
2971 | |||
2972 | 398491 | static void finished_reading_packet(AVFormatContext *s, int raw_packet_size) | |
2973 | { | ||
2974 | 398491 | AVIOContext *pb = s->pb; | |
2975 | 398491 | int skip = raw_packet_size - TS_PACKET_SIZE; | |
2976 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 398491 times.
|
398491 | if (skip > 0) |
2977 | ✗ | avio_skip(pb, skip); | |
2978 | 398491 | } | |
2979 | |||
2980 | 29249 | static int handle_packets(MpegTSContext *ts, int64_t nb_packets) | |
2981 | { | ||
2982 | 29249 | AVFormatContext *s = ts->stream; | |
2983 | uint8_t packet[TS_PACKET_SIZE + AV_INPUT_BUFFER_PADDING_SIZE]; | ||
2984 | const uint8_t *data; | ||
2985 | int64_t packet_num; | ||
2986 | 29249 | int ret = 0; | |
2987 | |||
2988 |
2/2✓ Branch 1 taken 260 times.
✓ Branch 2 taken 28989 times.
|
29249 | if (avio_tell(s->pb) != ts->last_pos) { |
2989 | int i; | ||
2990 | 260 | av_log(ts->stream, AV_LOG_TRACE, "Skipping after seek\n"); | |
2991 | /* seek detected, flush pes buffer */ | ||
2992 |
2/2✓ Branch 0 taken 2129920 times.
✓ Branch 1 taken 260 times.
|
2130180 | for (i = 0; i < NB_PID_MAX; i++) { |
2993 |
2/2✓ Branch 0 taken 1623 times.
✓ Branch 1 taken 2128297 times.
|
2129920 | if (ts->pids[i]) { |
2994 |
2/2✓ Branch 0 taken 559 times.
✓ Branch 1 taken 1064 times.
|
1623 | if (ts->pids[i]->type == MPEGTS_PES) { |
2995 | 559 | PESContext *pes = ts->pids[i]->u.pes_filter.opaque; | |
2996 | 559 | av_buffer_unref(&pes->buffer); | |
2997 | 559 | pes->data_index = 0; | |
2998 | 559 | pes->state = MPEGTS_SKIP; /* skip until pes header */ | |
2999 |
1/2✓ Branch 0 taken 1064 times.
✗ Branch 1 not taken.
|
1064 | } else if (ts->pids[i]->type == MPEGTS_SECTION) { |
3000 | 1064 | ts->pids[i]->u.section_filter.last_ver = -1; | |
3001 | } | ||
3002 | 1623 | ts->pids[i]->last_cc = -1; | |
3003 | 1623 | ts->pids[i]->last_pcr = -1; | |
3004 | } | ||
3005 | } | ||
3006 | } | ||
3007 | |||
3008 | 29249 | ts->stop_parse = 0; | |
3009 | 29249 | packet_num = 0; | |
3010 | 29249 | memset(packet + TS_PACKET_SIZE, 0, AV_INPUT_BUFFER_PADDING_SIZE); | |
3011 | for (;;) { | ||
3012 | 427740 | packet_num++; | |
3013 |
3/4✓ Branch 0 taken 8066 times.
✓ Branch 1 taken 419674 times.
✓ Branch 2 taken 8066 times.
✗ Branch 3 not taken.
|
427740 | if (nb_packets != 0 && packet_num >= nb_packets || |
3014 |
2/2✓ Branch 0 taken 55 times.
✓ Branch 1 taken 427685 times.
|
427740 | ts->stop_parse > 1) { |
3015 | 55 | ret = AVERROR(EAGAIN); | |
3016 | 55 | break; | |
3017 | } | ||
3018 |
2/2✓ Branch 0 taken 28821 times.
✓ Branch 1 taken 398864 times.
|
427685 | if (ts->stop_parse > 0) |
3019 | 28821 | break; | |
3020 | |||
3021 | 398864 | ret = read_packet(s, packet, ts->raw_packet_size, &data); | |
3022 |
2/2✓ Branch 0 taken 373 times.
✓ Branch 1 taken 398491 times.
|
398864 | if (ret != 0) |
3023 | 373 | break; | |
3024 | 398491 | ret = handle_packet(ts, data, avio_tell(s->pb)); | |
3025 | 398491 | finished_reading_packet(s, ts->raw_packet_size); | |
3026 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 398491 times.
|
398491 | if (ret != 0) |
3027 | ✗ | break; | |
3028 | } | ||
3029 | 29249 | ts->last_pos = avio_tell(s->pb); | |
3030 | 29249 | return ret; | |
3031 | } | ||
3032 | |||
3033 | 7186 | static int mpegts_probe(const AVProbeData *p) | |
3034 | { | ||
3035 | 7186 | const int size = p->buf_size; | |
3036 | 7186 | int maxscore = 0; | |
3037 | 7186 | int sumscore = 0; | |
3038 | int i; | ||
3039 | 7186 | int check_count = size / TS_FEC_PACKET_SIZE; | |
3040 | #define CHECK_COUNT 10 | ||
3041 | #define CHECK_BLOCK 100 | ||
3042 | |||
3043 |
2/2✓ Branch 0 taken 13 times.
✓ Branch 1 taken 7173 times.
|
7186 | if (!check_count) |
3044 | 13 | return 0; | |
3045 | |||
3046 |
2/2✓ Branch 0 taken 23499 times.
✓ Branch 1 taken 7173 times.
|
30672 | for (i = 0; i<check_count; i+=CHECK_BLOCK) { |
3047 | 23499 | int left = FFMIN(check_count - i, CHECK_BLOCK); | |
3048 | 23499 | int score = analyze(p->buf + TS_PACKET_SIZE *i, TS_PACKET_SIZE *left, TS_PACKET_SIZE , 1); | |
3049 | 23499 | int dvhs_score = analyze(p->buf + TS_DVHS_PACKET_SIZE*i, TS_DVHS_PACKET_SIZE*left, TS_DVHS_PACKET_SIZE, 1); | |
3050 | 23499 | int fec_score = analyze(p->buf + TS_FEC_PACKET_SIZE *i, TS_FEC_PACKET_SIZE *left, TS_FEC_PACKET_SIZE , 1); | |
3051 | 23499 | score = FFMAX3(score, dvhs_score, fec_score); | |
3052 | 23499 | sumscore += score; | |
3053 | 23499 | maxscore = FFMAX(maxscore, score); | |
3054 | } | ||
3055 | |||
3056 | 7173 | sumscore = sumscore * CHECK_COUNT / check_count; | |
3057 | 7173 | maxscore = maxscore * CHECK_COUNT / CHECK_BLOCK; | |
3058 | |||
3059 | ff_dlog(0, "TS score: %d %d\n", sumscore, maxscore); | ||
3060 | |||
3061 |
3/4✓ Branch 0 taken 3201 times.
✓ Branch 1 taken 3972 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3201 times.
|
7173 | if (check_count > CHECK_COUNT && sumscore > 6) { |
3062 | ✗ | return AVPROBE_SCORE_MAX + sumscore - CHECK_COUNT; | |
3063 |
4/4✓ Branch 0 taken 6819 times.
✓ Branch 1 taken 354 times.
✓ Branch 2 taken 47 times.
✓ Branch 3 taken 6772 times.
|
7173 | } else if (check_count >= CHECK_COUNT && sumscore > 6) { |
3064 | 47 | return AVPROBE_SCORE_MAX/2 + sumscore - CHECK_COUNT; | |
3065 |
3/4✓ Branch 0 taken 6772 times.
✓ Branch 1 taken 354 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6772 times.
|
7126 | } else if (check_count >= CHECK_COUNT && maxscore > 6) { |
3066 | ✗ | return AVPROBE_SCORE_MAX/2 + sumscore - CHECK_COUNT; | |
3067 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 7121 times.
|
7126 | } else if (sumscore > 6) { |
3068 | 5 | return 2; | |
3069 | } else { | ||
3070 | 7121 | return 0; | |
3071 | } | ||
3072 | } | ||
3073 | |||
3074 | /* return the 90kHz PCR and the extension for the 27MHz PCR. return | ||
3075 | * (-1) if not available */ | ||
3076 | 59560 | static int parse_pcr(int64_t *ppcr_high, int *ppcr_low, const uint8_t *packet) | |
3077 | { | ||
3078 | int afc, len, flags; | ||
3079 | const uint8_t *p; | ||
3080 | unsigned int v; | ||
3081 | |||
3082 | 59560 | afc = (packet[3] >> 4) & 3; | |
3083 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59560 times.
|
59560 | if (afc <= 1) |
3084 | ✗ | return AVERROR_INVALIDDATA; | |
3085 | 59560 | p = packet + 4; | |
3086 | 59560 | len = p[0]; | |
3087 | 59560 | p++; | |
3088 |
2/2✓ Branch 0 taken 38 times.
✓ Branch 1 taken 59522 times.
|
59560 | if (len == 0) |
3089 | 38 | return AVERROR_INVALIDDATA; | |
3090 | 59522 | flags = *p++; | |
3091 | 59522 | len--; | |
3092 |
2/2✓ Branch 0 taken 31929 times.
✓ Branch 1 taken 27593 times.
|
59522 | if (!(flags & 0x10)) |
3093 | 31929 | return AVERROR_INVALIDDATA; | |
3094 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27593 times.
|
27593 | if (len < 6) |
3095 | ✗ | return AVERROR_INVALIDDATA; | |
3096 | 27593 | v = AV_RB32(p); | |
3097 | 27593 | *ppcr_high = ((int64_t) v << 1) | (p[4] >> 7); | |
3098 | 27593 | *ppcr_low = ((p[4] & 1) << 8) | p[5]; | |
3099 | 27593 | return 0; | |
3100 | } | ||
3101 | |||
3102 | 110 | static void seek_back(AVFormatContext *s, AVIOContext *pb, int64_t pos) { | |
3103 | |||
3104 | /* NOTE: We attempt to seek on non-seekable files as well, as the | ||
3105 | * probe buffer usually is big enough. Only warn if the seek failed | ||
3106 | * on files where the seek should work. */ | ||
3107 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 110 times.
|
110 | if (avio_seek(pb, pos, SEEK_SET) < 0) |
3108 | ✗ | av_log(s, (pb->seekable & AVIO_SEEKABLE_NORMAL) ? AV_LOG_ERROR : AV_LOG_INFO, "Unable to seek back to the start\n"); | |
3109 | 110 | } | |
3110 | |||
3111 | 55 | static int mpegts_read_header(AVFormatContext *s) | |
3112 | { | ||
3113 | 55 | MpegTSContext *ts = s->priv_data; | |
3114 | 55 | AVIOContext *pb = s->pb; | |
3115 | 55 | int64_t pos, probesize = s->probesize; | |
3116 | 55 | int64_t seekback = FFMAX(s->probesize, (int64_t)ts->resync_size + PROBE_PACKET_MAX_BUF); | |
3117 | |||
3118 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 55 times.
|
55 | if (ffio_ensure_seekback(pb, seekback) < 0) |
3119 | ✗ | av_log(s, AV_LOG_WARNING, "Failed to allocate buffers for seekback\n"); | |
3120 | |||
3121 | 55 | pos = avio_tell(pb); | |
3122 | 55 | ts->raw_packet_size = get_packet_size(s); | |
3123 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 55 times.
|
55 | if (ts->raw_packet_size <= 0) { |
3124 | ✗ | av_log(s, AV_LOG_WARNING, "Could not detect TS packet size, defaulting to non-FEC/DVHS\n"); | |
3125 | ✗ | ts->raw_packet_size = TS_PACKET_SIZE; | |
3126 | } | ||
3127 | 55 | ts->stream = s; | |
3128 | 55 | ts->auto_guess = 0; | |
3129 | |||
3130 |
1/2✓ Branch 0 taken 55 times.
✗ Branch 1 not taken.
|
55 | if (s->iformat == &ff_mpegts_demuxer.p) { |
3131 | /* normal demux */ | ||
3132 | |||
3133 | /* first do a scan to get all the services */ | ||
3134 | 55 | seek_back(s, pb, pos); | |
3135 | |||
3136 | 55 | mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1); | |
3137 | 55 | mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1); | |
3138 | 55 | mpegts_open_section_filter(ts, EIT_PID, eit_cb, ts, 1); | |
3139 | |||
3140 | 55 | handle_packets(ts, probesize / ts->raw_packet_size); | |
3141 | /* if could not find service, enable auto_guess */ | ||
3142 | |||
3143 | 55 | ts->auto_guess = 1; | |
3144 | |||
3145 | 55 | av_log(ts->stream, AV_LOG_TRACE, "tuning done\n"); | |
3146 | |||
3147 | 55 | s->ctx_flags |= AVFMTCTX_NOHEADER; | |
3148 | } else { | ||
3149 | AVStream *st; | ||
3150 | int pcr_pid, pid, nb_packets, nb_pcrs, ret, pcr_l; | ||
3151 | int64_t pcrs[2], pcr_h; | ||
3152 | uint8_t packet[TS_PACKET_SIZE]; | ||
3153 | const uint8_t *data; | ||
3154 | |||
3155 | /* only read packets */ | ||
3156 | |||
3157 | ✗ | st = avformat_new_stream(s, NULL); | |
3158 | ✗ | if (!st) | |
3159 | ✗ | return AVERROR(ENOMEM); | |
3160 | ✗ | avpriv_set_pts_info(st, 60, 1, 27000000); | |
3161 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
3162 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_MPEG2TS; | |
3163 | |||
3164 | /* we iterate until we find two PCRs to estimate the bitrate */ | ||
3165 | ✗ | pcr_pid = -1; | |
3166 | ✗ | nb_pcrs = 0; | |
3167 | ✗ | nb_packets = 0; | |
3168 | for (;;) { | ||
3169 | ✗ | ret = read_packet(s, packet, ts->raw_packet_size, &data); | |
3170 | ✗ | if (ret < 0) | |
3171 | ✗ | return ret; | |
3172 | ✗ | pid = AV_RB16(data + 1) & 0x1fff; | |
3173 | ✗ | if ((pcr_pid == -1 || pcr_pid == pid) && | |
3174 | ✗ | parse_pcr(&pcr_h, &pcr_l, data) == 0) { | |
3175 | ✗ | finished_reading_packet(s, ts->raw_packet_size); | |
3176 | ✗ | pcr_pid = pid; | |
3177 | ✗ | pcrs[nb_pcrs] = pcr_h * 300 + pcr_l; | |
3178 | ✗ | nb_pcrs++; | |
3179 | ✗ | if (nb_pcrs >= 2) { | |
3180 | ✗ | if (pcrs[1] - pcrs[0] > 0) { | |
3181 | /* the difference needs to be positive to make sense for bitrate computation */ | ||
3182 | ✗ | break; | |
3183 | } else { | ||
3184 | ✗ | av_log(ts->stream, AV_LOG_WARNING, "invalid pcr pair %"PRId64" >= %"PRId64"\n", pcrs[0], pcrs[1]); | |
3185 | ✗ | pcrs[0] = pcrs[1]; | |
3186 | ✗ | nb_pcrs--; | |
3187 | } | ||
3188 | } | ||
3189 | } else { | ||
3190 | ✗ | finished_reading_packet(s, ts->raw_packet_size); | |
3191 | } | ||
3192 | ✗ | nb_packets++; | |
3193 | } | ||
3194 | |||
3195 | /* NOTE1: the bitrate is computed without the FEC */ | ||
3196 | /* NOTE2: it is only the bitrate of the start of the stream */ | ||
3197 | ✗ | ts->pcr_incr = pcrs[1] - pcrs[0]; | |
3198 | ✗ | ts->cur_pcr = pcrs[0] - ts->pcr_incr * (nb_packets - 1); | |
3199 | ✗ | s->bit_rate = TS_PACKET_SIZE * 8 * 27000000LL / ts->pcr_incr; | |
3200 | ✗ | st->codecpar->bit_rate = s->bit_rate; | |
3201 | ✗ | st->start_time = ts->cur_pcr; | |
3202 | ✗ | av_log(ts->stream, AV_LOG_TRACE, "start=%0.3f pcr=%0.3f incr=%"PRId64"\n", | |
3203 | ✗ | st->start_time / 1000000.0, pcrs[0] / 27e6, ts->pcr_incr); | |
3204 | } | ||
3205 | |||
3206 | 55 | seek_back(s, pb, pos); | |
3207 | 55 | return 0; | |
3208 | } | ||
3209 | |||
3210 | #define MAX_PACKET_READAHEAD ((128 * 1024) / 188) | ||
3211 | |||
3212 | ✗ | static int mpegts_raw_read_packet(AVFormatContext *s, AVPacket *pkt) | |
3213 | { | ||
3214 | ✗ | MpegTSContext *ts = s->priv_data; | |
3215 | int ret, i; | ||
3216 | int64_t pcr_h, next_pcr_h, pos; | ||
3217 | int pcr_l, next_pcr_l; | ||
3218 | uint8_t pcr_buf[12]; | ||
3219 | const uint8_t *data; | ||
3220 | |||
3221 | ✗ | if ((ret = av_new_packet(pkt, TS_PACKET_SIZE)) < 0) | |
3222 | ✗ | return ret; | |
3223 | ✗ | ret = read_packet(s, pkt->data, ts->raw_packet_size, &data); | |
3224 | ✗ | pkt->pos = avio_tell(s->pb); | |
3225 | ✗ | if (ret < 0) { | |
3226 | ✗ | return ret; | |
3227 | } | ||
3228 | ✗ | if (data != pkt->data) | |
3229 | ✗ | memcpy(pkt->data, data, TS_PACKET_SIZE); | |
3230 | ✗ | finished_reading_packet(s, ts->raw_packet_size); | |
3231 | ✗ | if (ts->mpeg2ts_compute_pcr) { | |
3232 | /* compute exact PCR for each packet */ | ||
3233 | ✗ | if (parse_pcr(&pcr_h, &pcr_l, pkt->data) == 0) { | |
3234 | /* we read the next PCR (XXX: optimize it by using a bigger buffer */ | ||
3235 | ✗ | pos = avio_tell(s->pb); | |
3236 | ✗ | for (i = 0; i < MAX_PACKET_READAHEAD; i++) { | |
3237 | ✗ | avio_seek(s->pb, pos + i * ts->raw_packet_size, SEEK_SET); | |
3238 | ✗ | avio_read(s->pb, pcr_buf, 12); | |
3239 | ✗ | if (parse_pcr(&next_pcr_h, &next_pcr_l, pcr_buf) == 0) { | |
3240 | /* XXX: not precise enough */ | ||
3241 | ✗ | ts->pcr_incr = | |
3242 | ✗ | ((next_pcr_h - pcr_h) * 300 + (next_pcr_l - pcr_l)) / | |
3243 | ✗ | (i + 1); | |
3244 | ✗ | break; | |
3245 | } | ||
3246 | } | ||
3247 | ✗ | avio_seek(s->pb, pos, SEEK_SET); | |
3248 | /* no next PCR found: we use previous increment */ | ||
3249 | ✗ | ts->cur_pcr = pcr_h * 300 + pcr_l; | |
3250 | } | ||
3251 | ✗ | pkt->pts = ts->cur_pcr; | |
3252 | ✗ | pkt->duration = ts->pcr_incr; | |
3253 | ✗ | ts->cur_pcr += ts->pcr_incr; | |
3254 | } | ||
3255 | ✗ | pkt->stream_index = 0; | |
3256 | ✗ | return 0; | |
3257 | } | ||
3258 | |||
3259 | 29194 | static int mpegts_read_packet(AVFormatContext *s, AVPacket *pkt) | |
3260 | { | ||
3261 | 29194 | MpegTSContext *ts = s->priv_data; | |
3262 | int ret, i; | ||
3263 | |||
3264 | 29194 | pkt->size = -1; | |
3265 | 29194 | ts->pkt = pkt; | |
3266 | 29194 | ret = handle_packets(ts, 0); | |
3267 |
2/2✓ Branch 0 taken 373 times.
✓ Branch 1 taken 28821 times.
|
29194 | if (ret < 0) { |
3268 | 373 | av_packet_unref(ts->pkt); | |
3269 | /* flush pes data left */ | ||
3270 |
2/2✓ Branch 0 taken 2214625 times.
✓ Branch 1 taken 209 times.
|
2214834 | for (i = 0; i < NB_PID_MAX; i++) |
3271 |
4/4✓ Branch 0 taken 2349 times.
✓ Branch 1 taken 2212276 times.
✓ Branch 2 taken 862 times.
✓ Branch 3 taken 1487 times.
|
2214625 | if (ts->pids[i] && ts->pids[i]->type == MPEGTS_PES) { |
3272 | 862 | PESContext *pes = ts->pids[i]->u.pes_filter.opaque; | |
3273 |
3/4✓ Branch 0 taken 164 times.
✓ Branch 1 taken 698 times.
✓ Branch 2 taken 164 times.
✗ Branch 3 not taken.
|
862 | if (pes->state == MPEGTS_PAYLOAD && pes->data_index > 0) { |
3274 | 164 | ret = new_pes_packet(pes, pkt); | |
3275 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 164 times.
|
164 | if (ret < 0) |
3276 | ✗ | return ret; | |
3277 | 164 | pes->state = MPEGTS_SKIP; | |
3278 | 164 | ret = 0; | |
3279 | 164 | break; | |
3280 | } | ||
3281 | } | ||
3282 | } | ||
3283 | |||
3284 |
3/4✓ Branch 0 taken 28985 times.
✓ Branch 1 taken 209 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 28985 times.
|
29194 | if (!ret && pkt->size < 0) |
3285 | ✗ | ret = AVERROR_INVALIDDATA; | |
3286 | 29194 | return ret; | |
3287 | } | ||
3288 | |||
3289 | 55 | static void mpegts_free(MpegTSContext *ts) | |
3290 | { | ||
3291 | int i; | ||
3292 | |||
3293 | 55 | clear_programs(ts); | |
3294 | |||
3295 |
2/2✓ Branch 0 taken 1760 times.
✓ Branch 1 taken 55 times.
|
1815 | for (i = 0; i < FF_ARRAY_ELEMS(ts->pools); i++) |
3296 | 1760 | av_buffer_pool_uninit(&ts->pools[i]); | |
3297 | |||
3298 |
2/2✓ Branch 0 taken 450560 times.
✓ Branch 1 taken 55 times.
|
450615 | for (i = 0; i < NB_PID_MAX; i++) |
3299 |
2/2✓ Branch 0 taken 353 times.
✓ Branch 1 taken 450207 times.
|
450560 | if (ts->pids[i]) |
3300 | 353 | mpegts_close_filter(ts, ts->pids[i]); | |
3301 | 55 | } | |
3302 | |||
3303 | 55 | static int mpegts_read_close(AVFormatContext *s) | |
3304 | { | ||
3305 | 55 | MpegTSContext *ts = s->priv_data; | |
3306 | 55 | mpegts_free(ts); | |
3307 | 55 | return 0; | |
3308 | } | ||
3309 | |||
3310 | ✗ | static av_unused int64_t mpegts_get_pcr(AVFormatContext *s, int stream_index, | |
3311 | int64_t *ppos, int64_t pos_limit) | ||
3312 | { | ||
3313 | ✗ | MpegTSContext *ts = s->priv_data; | |
3314 | int64_t pos, timestamp; | ||
3315 | uint8_t buf[TS_PACKET_SIZE]; | ||
3316 | ✗ | int pcr_l, pcr_pid = | |
3317 | ✗ | ((PESContext *)s->streams[stream_index]->priv_data)->pcr_pid; | |
3318 | ✗ | int pos47 = ts->pos47_full % ts->raw_packet_size; | |
3319 | ✗ | pos = | |
3320 | ✗ | ((*ppos + ts->raw_packet_size - 1 - pos47) / ts->raw_packet_size) * | |
3321 | ✗ | ts->raw_packet_size + pos47; | |
3322 | ✗ | while(pos < pos_limit) { | |
3323 | ✗ | if (avio_seek(s->pb, pos, SEEK_SET) < 0) | |
3324 | ✗ | return AV_NOPTS_VALUE; | |
3325 | ✗ | if (avio_read(s->pb, buf, TS_PACKET_SIZE) != TS_PACKET_SIZE) | |
3326 | ✗ | return AV_NOPTS_VALUE; | |
3327 | ✗ | if (buf[0] != 0x47) { | |
3328 | ✗ | if (mpegts_resync(s, TS_PACKET_SIZE, buf) < 0) | |
3329 | ✗ | return AV_NOPTS_VALUE; | |
3330 | ✗ | pos = avio_tell(s->pb); | |
3331 | ✗ | continue; | |
3332 | } | ||
3333 | ✗ | if ((pcr_pid < 0 || (AV_RB16(buf + 1) & 0x1fff) == pcr_pid) && | |
3334 | ✗ | parse_pcr(×tamp, &pcr_l, buf) == 0) { | |
3335 | ✗ | *ppos = pos; | |
3336 | ✗ | return timestamp; | |
3337 | } | ||
3338 | ✗ | pos += ts->raw_packet_size; | |
3339 | } | ||
3340 | |||
3341 | ✗ | return AV_NOPTS_VALUE; | |
3342 | } | ||
3343 | |||
3344 | 90 | static int64_t mpegts_get_dts(AVFormatContext *s, int stream_index, | |
3345 | int64_t *ppos, int64_t pos_limit) | ||
3346 | { | ||
3347 | 90 | MpegTSContext *ts = s->priv_data; | |
3348 | AVPacket *pkt; | ||
3349 | int64_t pos; | ||
3350 | 90 | int pos47 = ts->pos47_full % ts->raw_packet_size; | |
3351 | 90 | pos = ((*ppos + ts->raw_packet_size - 1 - pos47) / ts->raw_packet_size) * ts->raw_packet_size + pos47; | |
3352 | 90 | ff_read_frame_flush(s); | |
3353 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 90 times.
|
90 | if (avio_seek(s->pb, pos, SEEK_SET) < 0) |
3354 | ✗ | return AV_NOPTS_VALUE; | |
3355 | 90 | pkt = av_packet_alloc(); | |
3356 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | if (!pkt) |
3357 | ✗ | return AV_NOPTS_VALUE; | |
3358 |
2/2✓ Branch 0 taken 484 times.
✓ Branch 1 taken 12 times.
|
496 | while(pos < pos_limit) { |
3359 | 484 | int ret = av_read_frame(s, pkt); | |
3360 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 466 times.
|
484 | if (ret < 0) { |
3361 | 18 | av_packet_free(&pkt); | |
3362 | 18 | return AV_NOPTS_VALUE; | |
3363 | } | ||
3364 |
3/4✓ Branch 0 taken 466 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 163 times.
✓ Branch 3 taken 303 times.
|
466 | if (pkt->dts != AV_NOPTS_VALUE && pkt->pos >= 0) { |
3365 | 163 | ff_reduce_index(s, pkt->stream_index); | |
3366 | 163 | av_add_index_entry(s->streams[pkt->stream_index], pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME /* FIXME keyframe? */); | |
3367 |
3/4✓ Branch 0 taken 60 times.
✓ Branch 1 taken 103 times.
✓ Branch 2 taken 60 times.
✗ Branch 3 not taken.
|
163 | if (pkt->stream_index == stream_index && pkt->pos >= *ppos) { |
3368 | 60 | int64_t dts = pkt->dts; | |
3369 | 60 | *ppos = pkt->pos; | |
3370 | 60 | av_packet_free(&pkt); | |
3371 | 60 | return dts; | |
3372 | } | ||
3373 | } | ||
3374 | 406 | pos = pkt->pos; | |
3375 | 406 | av_packet_unref(pkt); | |
3376 | } | ||
3377 | |||
3378 | 12 | av_packet_free(&pkt); | |
3379 | 12 | return AV_NOPTS_VALUE; | |
3380 | } | ||
3381 | |||
3382 | /**************************************************************/ | ||
3383 | /* parsing functions - called from other demuxers such as RTP */ | ||
3384 | |||
3385 | ✗ | MpegTSContext *avpriv_mpegts_parse_open(AVFormatContext *s) | |
3386 | { | ||
3387 | MpegTSContext *ts; | ||
3388 | |||
3389 | ✗ | ts = av_mallocz(sizeof(MpegTSContext)); | |
3390 | ✗ | if (!ts) | |
3391 | ✗ | return NULL; | |
3392 | /* no stream case, currently used by RTP */ | ||
3393 | ✗ | ts->raw_packet_size = TS_PACKET_SIZE; | |
3394 | ✗ | ts->max_packet_size = 2048000; | |
3395 | ✗ | ts->stream = s; | |
3396 | ✗ | ts->auto_guess = 1; | |
3397 | |||
3398 | ✗ | mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1); | |
3399 | ✗ | mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1); | |
3400 | ✗ | mpegts_open_section_filter(ts, EIT_PID, eit_cb, ts, 1); | |
3401 | |||
3402 | ✗ | return ts; | |
3403 | } | ||
3404 | |||
3405 | /* return the consumed length if a packet was output, or -1 if no | ||
3406 | * packet is output */ | ||
3407 | ✗ | int avpriv_mpegts_parse_packet(MpegTSContext *ts, AVPacket *pkt, | |
3408 | const uint8_t *buf, int len) | ||
3409 | { | ||
3410 | int len1; | ||
3411 | |||
3412 | ✗ | len1 = len; | |
3413 | ✗ | ts->pkt = pkt; | |
3414 | for (;;) { | ||
3415 | ✗ | ts->stop_parse = 0; | |
3416 | ✗ | if (len < TS_PACKET_SIZE) | |
3417 | ✗ | return AVERROR_INVALIDDATA; | |
3418 | ✗ | if (buf[0] != 0x47) { | |
3419 | ✗ | buf++; | |
3420 | ✗ | len--; | |
3421 | } else { | ||
3422 | ✗ | handle_packet(ts, buf, len1 - len + TS_PACKET_SIZE); | |
3423 | ✗ | buf += TS_PACKET_SIZE; | |
3424 | ✗ | len -= TS_PACKET_SIZE; | |
3425 | ✗ | if (ts->stop_parse == 1) | |
3426 | ✗ | break; | |
3427 | } | ||
3428 | } | ||
3429 | ✗ | return len1 - len; | |
3430 | } | ||
3431 | |||
3432 | ✗ | void avpriv_mpegts_parse_close(MpegTSContext *ts) | |
3433 | { | ||
3434 | ✗ | mpegts_free(ts); | |
3435 | ✗ | av_free(ts); | |
3436 | ✗ | } | |
3437 | |||
3438 | const FFInputFormat ff_mpegts_demuxer = { | ||
3439 | .p.name = "mpegts", | ||
3440 | .p.long_name = NULL_IF_CONFIG_SMALL("MPEG-TS (MPEG-2 Transport Stream)"), | ||
3441 | .p.flags = AVFMT_SHOW_IDS | AVFMT_TS_DISCONT, | ||
3442 | .p.priv_class = &mpegts_class, | ||
3443 | .priv_data_size = sizeof(MpegTSContext), | ||
3444 | .read_probe = mpegts_probe, | ||
3445 | .read_header = mpegts_read_header, | ||
3446 | .read_packet = mpegts_read_packet, | ||
3447 | .read_close = mpegts_read_close, | ||
3448 | .read_timestamp = mpegts_get_dts, | ||
3449 | .flags_internal = FF_INFMT_FLAG_PREFER_CODEC_FRAMERATE, | ||
3450 | }; | ||
3451 | |||
3452 | const FFInputFormat ff_mpegtsraw_demuxer = { | ||
3453 | .p.name = "mpegtsraw", | ||
3454 | .p.long_name = NULL_IF_CONFIG_SMALL("raw MPEG-TS (MPEG-2 Transport Stream)"), | ||
3455 | .p.flags = AVFMT_SHOW_IDS | AVFMT_TS_DISCONT, | ||
3456 | .p.priv_class = &mpegtsraw_class, | ||
3457 | .priv_data_size = sizeof(MpegTSContext), | ||
3458 | .read_header = mpegts_read_header, | ||
3459 | .read_packet = mpegts_raw_read_packet, | ||
3460 | .read_close = mpegts_read_close, | ||
3461 | .read_timestamp = mpegts_get_dts, | ||
3462 | .flags_internal = FF_INFMT_FLAG_PREFER_CODEC_FRAMERATE, | ||
3463 | }; | ||
3464 |