repositories / deb_ffmpeg.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Imported Debian version 2.5.0~trusty1.1
[deb_ffmpeg.git] / ffmpeg / ffprobe.c
CommitLineData
2ba45a60
DM		 1/*
2 * Copyright (c) 2007-2010 Stefano Sabatini
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21/**
22 * @file
23 * simple media prober based on the FFmpeg libraries
24 */
25
26#include "config.h"
27#include "libavutil/ffversion.h"
28
29#include <string.h>
30
31#include "libavformat/avformat.h"
32#include "libavcodec/avcodec.h"
33#include "libavutil/avassert.h"
34#include "libavutil/avstring.h"
35#include "libavutil/bprint.h"
36#include "libavutil/hash.h"
37#include "libavutil/opt.h"
38#include "libavutil/pixdesc.h"
39#include "libavutil/dict.h"
40#include "libavutil/libm.h"
41#include "libavutil/parseutils.h"
42#include "libavutil/timecode.h"
43#include "libavutil/timestamp.h"
44#include "libavdevice/avdevice.h"
45#include "libswscale/swscale.h"
46#include "libswresample/swresample.h"
47#include "libpostproc/postprocess.h"
48#include "cmdutils.h"
49
50const char program_name[] = "ffprobe";
51const int program_birth_year = 2007;
52
53static int do_bitexact = 0;
54static int do_count_frames = 0;
55static int do_count_packets = 0;
56static int do_read_frames = 0;
57static int do_read_packets = 0;
58static int do_show_chapters = 0;
59static int do_show_error = 0;
60static int do_show_format = 0;
61static int do_show_frames = 0;
62static int do_show_packets = 0;
63static int do_show_programs = 0;
64static int do_show_streams = 0;
65static int do_show_stream_disposition = 0;
66static int do_show_data = 0;
67static int do_show_program_version = 0;
68static int do_show_library_versions = 0;
f6fa7814
DM		 69static int do_show_pixel_formats = 0;
70static int do_show_pixel_format_flags = 0;
71static int do_show_pixel_format_components = 0;
2ba45a60
DM		 72
73static int do_show_chapter_tags = 0;
74static int do_show_format_tags = 0;
75static int do_show_frame_tags = 0;
76static int do_show_program_tags = 0;
77static int do_show_stream_tags = 0;
78
79static int show_value_unit = 0;
80static int use_value_prefix = 0;
81static int use_byte_value_binary_prefix = 0;
82static int use_value_sexagesimal_format = 0;
83static int show_private_data = 1;
84
85static char *print_format;
86static char *stream_specifier;
87static char *show_data_hash;
88
89typedef struct {
90 int id; ///< identifier
91 int64_t start, end; ///< start, end in second/AV_TIME_BASE units
92 int has_start, has_end;
93 int start_is_offset, end_is_offset;
94 int duration_frames;
95} ReadInterval;
96
97static ReadInterval *read_intervals;
98static int read_intervals_nb = 0;
99
100/* section structure definition */
101
102#define SECTION_MAX_NB_CHILDREN 10
103
104struct section {
105 int id; ///< unique id identifying a section
106 const char *name;
107
108#define SECTION_FLAG_IS_WRAPPER 1 ///< the section only contains other sections, but has no data at its own level
109#define SECTION_FLAG_IS_ARRAY 2 ///< the section contains an array of elements of the same type
110#define SECTION_FLAG_HAS_VARIABLE_FIELDS 4 ///< the section may contain a variable number of fields with variable keys.
111 /// For these sections the element_name field is mandatory.
112 int flags;
113 int children_ids[SECTION_MAX_NB_CHILDREN+1]; ///< list of children section IDS, terminated by -1
114 const char *element_name; ///< name of the contained element, if provided
115 const char *unique_name; ///< unique section name, in case the name is ambiguous
116 AVDictionary *entries_to_show;
117 int show_all_entries;
118};
119
120typedef enum {
121 SECTION_ID_NONE = -1,
122 SECTION_ID_CHAPTER,
123 SECTION_ID_CHAPTER_TAGS,
124 SECTION_ID_CHAPTERS,
125 SECTION_ID_ERROR,
126 SECTION_ID_FORMAT,
127 SECTION_ID_FORMAT_TAGS,
128 SECTION_ID_FRAME,
129 SECTION_ID_FRAMES,
130 SECTION_ID_FRAME_TAGS,
131 SECTION_ID_FRAME_SIDE_DATA_LIST,
132 SECTION_ID_FRAME_SIDE_DATA,
133 SECTION_ID_LIBRARY_VERSION,
134 SECTION_ID_LIBRARY_VERSIONS,
135 SECTION_ID_PACKET,
136 SECTION_ID_PACKETS,
137 SECTION_ID_PACKETS_AND_FRAMES,
f6fa7814
DM		 138 SECTION_ID_PIXEL_FORMAT,
139 SECTION_ID_PIXEL_FORMAT_FLAGS,
140 SECTION_ID_PIXEL_FORMAT_COMPONENT,
141 SECTION_ID_PIXEL_FORMAT_COMPONENTS,
142 SECTION_ID_PIXEL_FORMATS,
2ba45a60
DM		 143 SECTION_ID_PROGRAM_STREAM_DISPOSITION,
144 SECTION_ID_PROGRAM_STREAM_TAGS,
145 SECTION_ID_PROGRAM,
146 SECTION_ID_PROGRAM_STREAMS,
147 SECTION_ID_PROGRAM_STREAM,
148 SECTION_ID_PROGRAM_TAGS,
149 SECTION_ID_PROGRAM_VERSION,
150 SECTION_ID_PROGRAMS,
151 SECTION_ID_ROOT,
152 SECTION_ID_STREAM,
153 SECTION_ID_STREAM_DISPOSITION,
154 SECTION_ID_STREAMS,
155 SECTION_ID_STREAM_TAGS,
156 SECTION_ID_SUBTITLE,
157} SectionID;
158
159static struct section sections[] = {
160 [SECTION_ID_CHAPTERS] = { SECTION_ID_CHAPTERS, "chapters", SECTION_FLAG_IS_ARRAY, { SECTION_ID_CHAPTER, -1 } },
161 [SECTION_ID_CHAPTER] = { SECTION_ID_CHAPTER, "chapter", 0, { SECTION_ID_CHAPTER_TAGS, -1 } },
162 [SECTION_ID_CHAPTER_TAGS] = { SECTION_ID_CHAPTER_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "chapter_tags" },
163 [SECTION_ID_ERROR] = { SECTION_ID_ERROR, "error", 0, { -1 } },
164 [SECTION_ID_FORMAT] = { SECTION_ID_FORMAT, "format", 0, { SECTION_ID_FORMAT_TAGS, -1 } },
165 [SECTION_ID_FORMAT_TAGS] = { SECTION_ID_FORMAT_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "format_tags" },
166 [SECTION_ID_FRAMES] = { SECTION_ID_FRAMES, "frames", SECTION_FLAG_IS_ARRAY, { SECTION_ID_FRAME, SECTION_ID_SUBTITLE, -1 } },
167 [SECTION_ID_FRAME] = { SECTION_ID_FRAME, "frame", 0, { SECTION_ID_FRAME_TAGS, SECTION_ID_FRAME_SIDE_DATA_LIST, -1 } },
168 [SECTION_ID_FRAME_TAGS] = { SECTION_ID_FRAME_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "frame_tags" },
169 [SECTION_ID_FRAME_SIDE_DATA_LIST] ={ SECTION_ID_FRAME_SIDE_DATA_LIST, "side_data_list", SECTION_FLAG_IS_ARRAY, { SECTION_ID_FRAME_SIDE_DATA, -1 } },
170 [SECTION_ID_FRAME_SIDE_DATA] = { SECTION_ID_FRAME_SIDE_DATA, "side_data", 0, { -1 } },
171 [SECTION_ID_LIBRARY_VERSIONS] = { SECTION_ID_LIBRARY_VERSIONS, "library_versions", SECTION_FLAG_IS_ARRAY, { SECTION_ID_LIBRARY_VERSION, -1 } },
172 [SECTION_ID_LIBRARY_VERSION] = { SECTION_ID_LIBRARY_VERSION, "library_version", 0, { -1 } },
173 [SECTION_ID_PACKETS] = { SECTION_ID_PACKETS, "packets", SECTION_FLAG_IS_ARRAY, { SECTION_ID_PACKET, -1} },
174 [SECTION_ID_PACKETS_AND_FRAMES] = { SECTION_ID_PACKETS_AND_FRAMES, "packets_and_frames", SECTION_FLAG_IS_ARRAY, { SECTION_ID_PACKET, -1} },
175 [SECTION_ID_PACKET] = { SECTION_ID_PACKET, "packet", 0, { -1 } },
f6fa7814
DM		 176 [SECTION_ID_PIXEL_FORMATS] = { SECTION_ID_PIXEL_FORMATS, "pixel_formats", SECTION_FLAG_IS_ARRAY, { SECTION_ID_PIXEL_FORMAT, -1 } },
177 [SECTION_ID_PIXEL_FORMAT] = { SECTION_ID_PIXEL_FORMAT, "pixel_format", 0, { SECTION_ID_PIXEL_FORMAT_FLAGS, SECTION_ID_PIXEL_FORMAT_COMPONENTS, -1 } },
178 [SECTION_ID_PIXEL_FORMAT_FLAGS] = { SECTION_ID_PIXEL_FORMAT_FLAGS, "flags", 0, { -1 }, .unique_name = "pixel_format_flags" },
179 [SECTION_ID_PIXEL_FORMAT_COMPONENTS] = { SECTION_ID_PIXEL_FORMAT_COMPONENTS, "components", SECTION_FLAG_IS_ARRAY, {SECTION_ID_PIXEL_FORMAT_COMPONENT, -1 }, .unique_name = "pixel_format_components" },
180 [SECTION_ID_PIXEL_FORMAT_COMPONENT] = { SECTION_ID_PIXEL_FORMAT_COMPONENT, "component", 0, { -1 } },
2ba45a60
DM		 181 [SECTION_ID_PROGRAM_STREAM_DISPOSITION] = { SECTION_ID_PROGRAM_STREAM_DISPOSITION, "disposition", 0, { -1 }, .unique_name = "program_stream_disposition" },
182 [SECTION_ID_PROGRAM_STREAM_TAGS] = { SECTION_ID_PROGRAM_STREAM_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "program_stream_tags" },
183 [SECTION_ID_PROGRAM] = { SECTION_ID_PROGRAM, "program", 0, { SECTION_ID_PROGRAM_TAGS, SECTION_ID_PROGRAM_STREAMS, -1 } },
184 [SECTION_ID_PROGRAM_STREAMS] = { SECTION_ID_PROGRAM_STREAMS, "streams", SECTION_FLAG_IS_ARRAY, { SECTION_ID_PROGRAM_STREAM, -1 }, .unique_name = "program_streams" },
185 [SECTION_ID_PROGRAM_STREAM] = { SECTION_ID_PROGRAM_STREAM, "stream", 0, { SECTION_ID_PROGRAM_STREAM_DISPOSITION, SECTION_ID_PROGRAM_STREAM_TAGS, -1 }, .unique_name = "program_stream" },
186 [SECTION_ID_PROGRAM_TAGS] = { SECTION_ID_PROGRAM_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "program_tags" },
187 [SECTION_ID_PROGRAM_VERSION] = { SECTION_ID_PROGRAM_VERSION, "program_version", 0, { -1 } },
188 [SECTION_ID_PROGRAMS] = { SECTION_ID_PROGRAMS, "programs", SECTION_FLAG_IS_ARRAY, { SECTION_ID_PROGRAM, -1 } },
189 [SECTION_ID_ROOT] = { SECTION_ID_ROOT, "root", SECTION_FLAG_IS_WRAPPER,
190 { SECTION_ID_CHAPTERS, SECTION_ID_FORMAT, SECTION_ID_FRAMES, SECTION_ID_PROGRAMS, SECTION_ID_STREAMS,
f6fa7814
DM		 191 SECTION_ID_PACKETS, SECTION_ID_ERROR, SECTION_ID_PROGRAM_VERSION, SECTION_ID_LIBRARY_VERSIONS,
192 SECTION_ID_PIXEL_FORMATS, -1} },
2ba45a60
DM		 193 [SECTION_ID_STREAMS] = { SECTION_ID_STREAMS, "streams", SECTION_FLAG_IS_ARRAY, { SECTION_ID_STREAM, -1 } },
194 [SECTION_ID_STREAM] = { SECTION_ID_STREAM, "stream", 0, { SECTION_ID_STREAM_DISPOSITION, SECTION_ID_STREAM_TAGS, -1 } },
195 [SECTION_ID_STREAM_DISPOSITION] = { SECTION_ID_STREAM_DISPOSITION, "disposition", 0, { -1 }, .unique_name = "stream_disposition" },
196 [SECTION_ID_STREAM_TAGS] = { SECTION_ID_STREAM_TAGS, "tags", SECTION_FLAG_HAS_VARIABLE_FIELDS, { -1 }, .element_name = "tag", .unique_name = "stream_tags" },
197 [SECTION_ID_SUBTITLE] = { SECTION_ID_SUBTITLE, "subtitle", 0, { -1 } },
198};
199
200static const OptionDef *options;
201
202/* FFprobe context */
203static const char *input_filename;
204static AVInputFormat *iformat = NULL;
205
206static struct AVHashContext *hash;
207
208static const char *const binary_unit_prefixes [] = { "", "Ki", "Mi", "Gi", "Ti", "Pi" };
209static const char *const decimal_unit_prefixes[] = { "", "K" , "M" , "G" , "T" , "P" };
210
211static const char unit_second_str[] = "s" ;
212static const char unit_hertz_str[] = "Hz" ;
213static const char unit_byte_str[] = "byte" ;
214static const char unit_bit_per_second_str[] = "bit/s";
215
216static int nb_streams;
217static uint64_t *nb_streams_packets;
218static uint64_t *nb_streams_frames;
219static int *selected_streams;
220
221static void ffprobe_cleanup(int ret)
222{
223 int i;
224 for (i = 0; i < FF_ARRAY_ELEMS(sections); i++)
225 av_dict_free(&(sections[i].entries_to_show));
226}
227
228struct unit_value {
229 union { double d; long long int i; } val;
230 const char *unit;
231};
232
233static char *value_string(char *buf, int buf_size, struct unit_value uv)
234{
235 double vald;
236 long long int vali;
237 int show_float = 0;
238
239 if (uv.unit == unit_second_str) {
240 vald = uv.val.d;
241 show_float = 1;
242 } else {
243 vald = vali = uv.val.i;
244 }
245
246 if (uv.unit == unit_second_str && use_value_sexagesimal_format) {
247 double secs;
248 int hours, mins;
249 secs = vald;
250 mins = (int)secs / 60;
251 secs = secs - mins * 60;
252 hours = mins / 60;
253 mins %= 60;
254 snprintf(buf, buf_size, "%d:%02d:%09.6f", hours, mins, secs);
255 } else {
256 const char *prefix_string = "";
257
258 if (use_value_prefix && vald > 1) {
259 long long int index;
260
261 if (uv.unit == unit_byte_str && use_byte_value_binary_prefix) {
262 index = (long long int) (log2(vald)) / 10;
263 index = av_clip(index, 0, FF_ARRAY_ELEMS(binary_unit_prefixes) - 1);
264 vald /= exp2(index * 10);
265 prefix_string = binary_unit_prefixes[index];
266 } else {
267 index = (long long int) (log10(vald)) / 3;
268 index = av_clip(index, 0, FF_ARRAY_ELEMS(decimal_unit_prefixes) - 1);
269 vald /= pow(10, index * 3);
270 prefix_string = decimal_unit_prefixes[index];
271 }
272 vali = vald;
273 }
274
275 if (show_float || (use_value_prefix && vald != (long long int)vald))
276 snprintf(buf, buf_size, "%f", vald);
277 else
278 snprintf(buf, buf_size, "%lld", vali);
279 av_strlcatf(buf, buf_size, "%s%s%s", *prefix_string || show_value_unit ? " " : "",
280 prefix_string, show_value_unit ? uv.unit : "");
281 }
282
283 return buf;
284}
285
286/* WRITERS API */
287
288typedef struct WriterContext WriterContext;
289
290#define WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS 1
291#define WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER 2
292
293typedef enum {
294 WRITER_STRING_VALIDATION_FAIL,
295 WRITER_STRING_VALIDATION_REPLACE,
296 WRITER_STRING_VALIDATION_IGNORE,
297 WRITER_STRING_VALIDATION_NB
298} StringValidation;
299
300typedef struct Writer {
301 const AVClass *priv_class; ///< private class of the writer, if any
302 int priv_size; ///< private size for the writer context
303 const char *name;
304
305 int (*init) (WriterContext *wctx);
306 void (*uninit)(WriterContext *wctx);
307
308 void (*print_section_header)(WriterContext *wctx);
309 void (*print_section_footer)(WriterContext *wctx);
310 void (*print_integer) (WriterContext *wctx, const char *, long long int);
311 void (*print_rational) (WriterContext *wctx, AVRational *q, char *sep);
312 void (*print_string) (WriterContext *wctx, const char *, const char *);
313 int flags; ///< a combination or WRITER_FLAG_*
314} Writer;
315
316#define SECTION_MAX_NB_LEVELS 10
317
318struct WriterContext {
319 const AVClass *class; ///< class of the writer
320 const Writer *writer; ///< the Writer of which this is an instance
321 char *name; ///< name of this writer instance
322 void *priv; ///< private data for use by the filter
323
324 const struct section *sections; ///< array containing all sections
325 int nb_sections; ///< number of sections
326
327 int level; ///< current level, starting from 0
328
329 /** number of the item printed in the given section, starting from 0 */
330 unsigned int nb_item[SECTION_MAX_NB_LEVELS];
331
332 /** section per each level */
333 const struct section *section[SECTION_MAX_NB_LEVELS];
334 AVBPrint section_pbuf[SECTION_MAX_NB_LEVELS]; ///< generic print buffer dedicated to each section,
335 /// used by various writers
336
337 unsigned int nb_section_packet; ///< number of the packet section in case we are in "packets_and_frames" section
338 unsigned int nb_section_frame; ///< number of the frame section in case we are in "packets_and_frames" section
339 unsigned int nb_section_packet_frame; ///< nb_section_packet or nb_section_frame according if is_packets_and_frames
340
341 StringValidation string_validation;
342 char *string_validation_replacement;
343 unsigned int string_validation_utf8_flags;
344};
345
346static const char *writer_get_name(void *p)
347{
348 WriterContext *wctx = p;
349 return wctx->writer->name;
350}
351
352#define OFFSET(x) offsetof(WriterContext, x)
353
354static const AVOption writer_options[] = {
355 { "string_validation", "set string validation mode",
356 OFFSET(string_validation), AV_OPT_TYPE_INT, {.i64=WRITER_STRING_VALIDATION_REPLACE}, 0, WRITER_STRING_VALIDATION_NB-1, .unit = "sv" },
357 { "sv", "set string validation mode",
358 OFFSET(string_validation), AV_OPT_TYPE_INT, {.i64=WRITER_STRING_VALIDATION_REPLACE}, 0, WRITER_STRING_VALIDATION_NB-1, .unit = "sv" },
359 { "ignore", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = WRITER_STRING_VALIDATION_IGNORE}, .unit = "sv" },
360 { "replace", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = WRITER_STRING_VALIDATION_REPLACE}, .unit = "sv" },
361 { "fail", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = WRITER_STRING_VALIDATION_FAIL}, .unit = "sv" },
362 { "string_validation_replacement", "set string validation replacement string", OFFSET(string_validation_replacement), AV_OPT_TYPE_STRING, {.str=""}},
363 { "svr", "set string validation replacement string", OFFSET(string_validation_replacement), AV_OPT_TYPE_STRING, {.str="\xEF\xBF\xBD"}},
364 { NULL }
365};
366
367static void *writer_child_next(void *obj, void *prev)
368{
369 WriterContext *ctx = obj;
370 if (!prev && ctx->writer && ctx->writer->priv_class && ctx->priv)
371 return ctx->priv;
372 return NULL;
373}
374
375static const AVClass writer_class = {
376 .class_name = "Writer",
377 .item_name = writer_get_name,
378 .option = writer_options,
379 .version = LIBAVUTIL_VERSION_INT,
380 .child_next = writer_child_next,
381};
382
383static void writer_close(WriterContext **wctx)
384{
385 int i;
386
387 if (!*wctx)
388 return;
389
390 if ((*wctx)->writer->uninit)
391 (*wctx)->writer->uninit(*wctx);
392 for (i = 0; i < SECTION_MAX_NB_LEVELS; i++)
393 av_bprint_finalize(&(*wctx)->section_pbuf[i], NULL);
394 if ((*wctx)->writer->priv_class)
395 av_opt_free((*wctx)->priv);
396 av_freep(&((*wctx)->priv));
397 av_opt_free(*wctx);
398 av_freep(wctx);
399}
400
401static void bprint_bytes(AVBPrint *bp, const uint8_t *ubuf, size_t ubuf_size)
402{
403 int i;
404 av_bprintf(bp, "0X");
405 for (i = 0; i < ubuf_size; i++)
406 av_bprintf(bp, "%02X", ubuf[i]);
407}
408
409
410static int writer_open(WriterContext **wctx, const Writer *writer, const char *args,
411 const struct section *sections, int nb_sections)
412{
413 int i, ret = 0;
414
415 if (!(*wctx = av_mallocz(sizeof(WriterContext)))) {
416 ret = AVERROR(ENOMEM);
417 goto fail;
418 }
419
420 if (!((*wctx)->priv = av_mallocz(writer->priv_size))) {
421 ret = AVERROR(ENOMEM);
422 goto fail;
423 }
424
425 (*wctx)->class = &writer_class;
426 (*wctx)->writer = writer;
427 (*wctx)->level = -1;
428 (*wctx)->sections = sections;
429 (*wctx)->nb_sections = nb_sections;
430
431 av_opt_set_defaults(*wctx);
432
433 if (writer->priv_class) {
434 void *priv_ctx = (*wctx)->priv;
435 *((const AVClass **)priv_ctx) = writer->priv_class;
436 av_opt_set_defaults(priv_ctx);
437 }
438
439 /* convert options to dictionary */
440 if (args) {
441 AVDictionary *opts = NULL;
442 AVDictionaryEntry *opt = NULL;
443
444 if ((ret = av_dict_parse_string(&opts, args, "=", ":", 0)) < 0) {
445 av_log(*wctx, AV_LOG_ERROR, "Failed to parse option string '%s' provided to writer context\n", args);
446 av_dict_free(&opts);
447 goto fail;
448 }
449
450 while ((opt = av_dict_get(opts, "", opt, AV_DICT_IGNORE_SUFFIX))) {
451 if ((ret = av_opt_set(*wctx, opt->key, opt->value, AV_OPT_SEARCH_CHILDREN)) < 0) {
452 av_log(*wctx, AV_LOG_ERROR, "Failed to set option '%s' with value '%s' provided to writer context\n",
453 opt->key, opt->value);
454 av_dict_free(&opts);
455 goto fail;
456 }
457 }
458
459 av_dict_free(&opts);
460 }
461
462 /* validate replace string */
463 {
464 const uint8_t *p = (*wctx)->string_validation_replacement;
465 const uint8_t *endp = p + strlen(p);
466 while (*p) {
467 const uint8_t *p0 = p;
468 int32_t code;
469 ret = av_utf8_decode(&code, &p, endp, (*wctx)->string_validation_utf8_flags);
470 if (ret < 0) {
471 AVBPrint bp;
472 av_bprint_init(&bp, 0, AV_BPRINT_SIZE_AUTOMATIC);
473 bprint_bytes(&bp, p0, p-p0),
474 av_log(wctx, AV_LOG_ERROR,
475 "Invalid UTF8 sequence %s found in string validation replace '%s'\n",
476 bp.str, (*wctx)->string_validation_replacement);
477 return ret;
478 }
479 }
480 }
481
482 for (i = 0; i < SECTION_MAX_NB_LEVELS; i++)
483 av_bprint_init(&(*wctx)->section_pbuf[i], 1, AV_BPRINT_SIZE_UNLIMITED);
484
485 if ((*wctx)->writer->init)
486 ret = (*wctx)->writer->init(*wctx);
487 if (ret < 0)
488 goto fail;
489
490 return 0;
491
492fail:
493 writer_close(wctx);
494 return ret;
495}
496
497static inline void writer_print_section_header(WriterContext *wctx,
498 int section_id)
499{
500 int parent_section_id;
501 wctx->level++;
502 av_assert0(wctx->level < SECTION_MAX_NB_LEVELS);
503 parent_section_id = wctx->level ?
504 (wctx->section[wctx->level-1])->id : SECTION_ID_NONE;
505
506 wctx->nb_item[wctx->level] = 0;
507 wctx->section[wctx->level] = &wctx->sections[section_id];
508
509 if (section_id == SECTION_ID_PACKETS_AND_FRAMES) {
510 wctx->nb_section_packet = wctx->nb_section_frame =
511 wctx->nb_section_packet_frame = 0;
512 } else if (parent_section_id == SECTION_ID_PACKETS_AND_FRAMES) {
513 wctx->nb_section_packet_frame = section_id == SECTION_ID_PACKET ?
514 wctx->nb_section_packet : wctx->nb_section_frame;
515 }
516
517 if (wctx->writer->print_section_header)
518 wctx->writer->print_section_header(wctx);
519}
520
521static inline void writer_print_section_footer(WriterContext *wctx)
522{
523 int section_id = wctx->section[wctx->level]->id;
524 int parent_section_id = wctx->level ?
525 wctx->section[wctx->level-1]->id : SECTION_ID_NONE;
526
527 if (parent_section_id != SECTION_ID_NONE)
528 wctx->nb_item[wctx->level-1]++;
529 if (parent_section_id == SECTION_ID_PACKETS_AND_FRAMES) {
530 if (section_id == SECTION_ID_PACKET) wctx->nb_section_packet++;
531 else wctx->nb_section_frame++;
532 }
533 if (wctx->writer->print_section_footer)
534 wctx->writer->print_section_footer(wctx);
535 wctx->level--;
536}
537
538static inline void writer_print_integer(WriterContext *wctx,
539 const char *key, long long int val)
540{
541 const struct section *section = wctx->section[wctx->level];
542
543 if (section->show_all_entries || av_dict_get(section->entries_to_show, key, NULL, 0)) {
544 wctx->writer->print_integer(wctx, key, val);
545 wctx->nb_item[wctx->level]++;
546 }
547}
548
549static inline int validate_string(WriterContext *wctx, char **dstp, const char *src)
550{
551 const uint8_t *p, *endp;
552 AVBPrint dstbuf;
553 int invalid_chars_nb = 0, ret = 0;
554
555 av_bprint_init(&dstbuf, 0, AV_BPRINT_SIZE_UNLIMITED);
556
557 endp = src + strlen(src);
558 for (p = (uint8_t *)src; *p;) {
559 uint32_t code;
560 int invalid = 0;
561 const uint8_t *p0 = p;
562
563 if (av_utf8_decode(&code, &p, endp, wctx->string_validation_utf8_flags) < 0) {
564 AVBPrint bp;
565 av_bprint_init(&bp, 0, AV_BPRINT_SIZE_AUTOMATIC);
566 bprint_bytes(&bp, p0, p-p0);
567 av_log(wctx, AV_LOG_DEBUG,
568 "Invalid UTF-8 sequence %s found in string '%s'\n", bp.str, src);
569 invalid = 1;
570 }
571
572 if (invalid) {
573 invalid_chars_nb++;
574
575 switch (wctx->string_validation) {
576 case WRITER_STRING_VALIDATION_FAIL:
577 av_log(wctx, AV_LOG_ERROR,
578 "Invalid UTF-8 sequence found in string '%s'\n", src);
579 ret = AVERROR_INVALIDDATA;
580 goto end;
581 break;
582
583 case WRITER_STRING_VALIDATION_REPLACE:
584 av_bprintf(&dstbuf, "%s", wctx->string_validation_replacement);
585 break;
586 }
587 }
588
589 if (!invalid || wctx->string_validation == WRITER_STRING_VALIDATION_IGNORE)
590 av_bprint_append_data(&dstbuf, p0, p-p0);
591 }
592
593 if (invalid_chars_nb && wctx->string_validation == WRITER_STRING_VALIDATION_REPLACE) {
594 av_log(wctx, AV_LOG_WARNING,
595 "%d invalid UTF-8 sequence(s) found in string '%s', replaced with '%s'\n",
596 invalid_chars_nb, src, wctx->string_validation_replacement);
597 }
598
599end:
600 av_bprint_finalize(&dstbuf, dstp);
601 return ret;
602}
603
604#define PRINT_STRING_OPT 1
605#define PRINT_STRING_VALIDATE 2
606
607static inline int writer_print_string(WriterContext *wctx,
608 const char *key, const char *val, int flags)
609{
610 const struct section *section = wctx->section[wctx->level];
611 int ret = 0;
612
613 if ((flags & PRINT_STRING_OPT)
614 && !(wctx->writer->flags & WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS))
615 return 0;
616
617 if (section->show_all_entries || av_dict_get(section->entries_to_show, key, NULL, 0)) {
618 if (flags & PRINT_STRING_VALIDATE) {
619 char *key1 = NULL, *val1 = NULL;
620 ret = validate_string(wctx, &key1, key);
621 if (ret < 0) goto end;
622 ret = validate_string(wctx, &val1, val);
623 if (ret < 0) goto end;
624 wctx->writer->print_string(wctx, key1, val1);
625 end:
626 if (ret < 0) {
627 av_log(wctx, AV_LOG_ERROR,
628 "Invalid key=value string combination %s=%s in section %s\n",
629 key, val, section->unique_name);
630 }
631 av_free(key1);
632 av_free(val1);
633 } else {
634 wctx->writer->print_string(wctx, key, val);
635 }
636
637 wctx->nb_item[wctx->level]++;
638 }
639
640 return ret;
641}
642
643static inline void writer_print_rational(WriterContext *wctx,
644 const char *key, AVRational q, char sep)
645{
646 AVBPrint buf;
647 av_bprint_init(&buf, 0, AV_BPRINT_SIZE_AUTOMATIC);
648 av_bprintf(&buf, "%d%c%d", q.num, sep, q.den);
649 writer_print_string(wctx, key, buf.str, 0);
650}
651
652static void writer_print_time(WriterContext *wctx, const char *key,
653 int64_t ts, const AVRational *time_base, int is_duration)
654{
655 char buf[128];
656
657 if ((!is_duration && ts == AV_NOPTS_VALUE) || (is_duration && ts == 0)) {
658 writer_print_string(wctx, key, "N/A", PRINT_STRING_OPT);
659 } else {
660 double d = ts * av_q2d(*time_base);
661 struct unit_value uv;
662 uv.val.d = d;
663 uv.unit = unit_second_str;
664 value_string(buf, sizeof(buf), uv);
665 writer_print_string(wctx, key, buf, 0);
666 }
667}
668
669static void writer_print_ts(WriterContext *wctx, const char *key, int64_t ts, int is_duration)
670{
671 if ((!is_duration && ts == AV_NOPTS_VALUE) || (is_duration && ts == 0)) {
672 writer_print_string(wctx, key, "N/A", PRINT_STRING_OPT);
673 } else {
674 writer_print_integer(wctx, key, ts);
675 }
676}
677
678static void writer_print_data(WriterContext *wctx, const char *name,
679 uint8_t *data, int size)
680{
681 AVBPrint bp;
682 int offset = 0, l, i;
683
684 av_bprint_init(&bp, 0, AV_BPRINT_SIZE_UNLIMITED);
685 av_bprintf(&bp, "\n");
686 while (size) {
687 av_bprintf(&bp, "%08x: ", offset);
688 l = FFMIN(size, 16);
689 for (i = 0; i < l; i++) {
690 av_bprintf(&bp, "%02x", data[i]);
691 if (i & 1)
692 av_bprintf(&bp, " ");
693 }
694 av_bprint_chars(&bp, ' ', 41 - 2 * i - i / 2);
695 for (i = 0; i < l; i++)
696 av_bprint_chars(&bp, data[i] - 32U < 95 ? data[i] : '.', 1);
697 av_bprintf(&bp, "\n");
698 offset += l;
699 data += l;
700 size -= l;
701 }
702 writer_print_string(wctx, name, bp.str, 0);
703 av_bprint_finalize(&bp, NULL);
704}
705
706static void writer_print_data_hash(WriterContext *wctx, const char *name,
707 uint8_t *data, int size)
708{
709 char *p, buf[AV_HASH_MAX_SIZE * 2 + 64] = { 0 };
710
711 if (!hash)
712 return;
713 av_hash_init(hash);
714 av_hash_update(hash, data, size);
715 snprintf(buf, sizeof(buf), "%s:", av_hash_get_name(hash));
716 p = buf + strlen(buf);
717 av_hash_final_hex(hash, p, buf + sizeof(buf) - p);
718 writer_print_string(wctx, name, buf, 0);
719}
720
721#define MAX_REGISTERED_WRITERS_NB 64
722
723static const Writer *registered_writers[MAX_REGISTERED_WRITERS_NB + 1];
724
725static int writer_register(const Writer *writer)
726{
727 static int next_registered_writer_idx = 0;
728
729 if (next_registered_writer_idx == MAX_REGISTERED_WRITERS_NB)
730 return AVERROR(ENOMEM);
731
732 registered_writers[next_registered_writer_idx++] = writer;
733 return 0;
734}
735
736static const Writer *writer_get_by_name(const char *name)
737{
738 int i;
739
740 for (i = 0; registered_writers[i]; i++)
741 if (!strcmp(registered_writers[i]->name, name))
742 return registered_writers[i];
743
744 return NULL;
745}
746
747
748/* WRITERS */
749
750#define DEFINE_WRITER_CLASS(name) \
751static const char *name##_get_name(void *ctx) \
752{ \
753 return #name ; \
754} \
755static const AVClass name##_class = { \
756 .class_name = #name, \
757 .item_name = name##_get_name, \
758 .option = name##_options \
759}
760
761/* Default output */
762
763typedef struct DefaultContext {
764 const AVClass *class;
765 int nokey;
766 int noprint_wrappers;
767 int nested_section[SECTION_MAX_NB_LEVELS];
768} DefaultContext;
769
770#undef OFFSET
771#define OFFSET(x) offsetof(DefaultContext, x)
772
773static const AVOption default_options[] = {
774 { "noprint_wrappers", "do not print headers and footers", OFFSET(noprint_wrappers), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
775 { "nw", "do not print headers and footers", OFFSET(noprint_wrappers), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
776 { "nokey", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
777 { "nk", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
778 {NULL},
779};
780
781DEFINE_WRITER_CLASS(default);
782
783/* lame uppercasing routine, assumes the string is lower case ASCII */
784static inline char *upcase_string(char *dst, size_t dst_size, const char *src)
785{
786 int i;
787 for (i = 0; src[i] && i < dst_size-1; i++)
788 dst[i] = av_toupper(src[i]);
789 dst[i] = 0;
790 return dst;
791}
792
793static void default_print_section_header(WriterContext *wctx)
794{
795 DefaultContext *def = wctx->priv;
796 char buf[32];
797 const struct section *section = wctx->section[wctx->level];
798 const struct section *parent_section = wctx->level ?
799 wctx->section[wctx->level-1] : NULL;
800
801 av_bprint_clear(&wctx->section_pbuf[wctx->level]);
802 if (parent_section &&
803 !(parent_section->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY))) {
804 def->nested_section[wctx->level] = 1;
805 av_bprintf(&wctx->section_pbuf[wctx->level], "%s%s:",
806 wctx->section_pbuf[wctx->level-1].str,
807 upcase_string(buf, sizeof(buf),
808 av_x_if_null(section->element_name, section->name)));
809 }
810
811 if (def->noprint_wrappers || def->nested_section[wctx->level])
812 return;
813
814 if (!(section->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY)))
815 printf("[%s]\n", upcase_string(buf, sizeof(buf), section->name));
816}
817
818static void default_print_section_footer(WriterContext *wctx)
819{
820 DefaultContext *def = wctx->priv;
821 const struct section *section = wctx->section[wctx->level];
822 char buf[32];
823
824 if (def->noprint_wrappers || def->nested_section[wctx->level])
825 return;
826
827 if (!(section->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY)))
828 printf("[/%s]\n", upcase_string(buf, sizeof(buf), section->name));
829}
830
831static void default_print_str(WriterContext *wctx, const char *key, const char *value)
832{
833 DefaultContext *def = wctx->priv;
834
835 if (!def->nokey)
836 printf("%s%s=", wctx->section_pbuf[wctx->level].str, key);
837 printf("%s\n", value);
838}
839
840static void default_print_int(WriterContext *wctx, const char *key, long long int value)
841{
842 DefaultContext *def = wctx->priv;
843
844 if (!def->nokey)
845 printf("%s%s=", wctx->section_pbuf[wctx->level].str, key);
846 printf("%lld\n", value);
847}
848
849static const Writer default_writer = {
850 .name = "default",
851 .priv_size = sizeof(DefaultContext),
852 .print_section_header = default_print_section_header,
853 .print_section_footer = default_print_section_footer,
854 .print_integer = default_print_int,
855 .print_string = default_print_str,
856 .flags = WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS,
857 .priv_class = &default_class,
858};
859
860/* Compact output */
861
862/**
863 * Apply C-language-like string escaping.
864 */
865static const char *c_escape_str(AVBPrint *dst, const char *src, const char sep, void *log_ctx)
866{
867 const char *p;
868
869 for (p = src; *p; p++) {
870 switch (*p) {
871 case '\b': av_bprintf(dst, "%s", "\\b"); break;
872 case '\f': av_bprintf(dst, "%s", "\\f"); break;
873 case '\n': av_bprintf(dst, "%s", "\\n"); break;
874 case '\r': av_bprintf(dst, "%s", "\\r"); break;
875 case '\\': av_bprintf(dst, "%s", "\\\\"); break;
876 default:
877 if (*p == sep)
878 av_bprint_chars(dst, '\\', 1);
879 av_bprint_chars(dst, *p, 1);
880 }
881 }
882 return dst->str;
883}
884
885/**
886 * Quote fields containing special characters, check RFC4180.
887 */
888static const char *csv_escape_str(AVBPrint *dst, const char *src, const char sep, void *log_ctx)
889{
890 char meta_chars[] = { sep, '"', '\n', '\r', '\0' };
891 int needs_quoting = !!src[strcspn(src, meta_chars)];
892
893 if (needs_quoting)
894 av_bprint_chars(dst, '"', 1);
895
896 for (; *src; src++) {
897 if (*src == '"')
898 av_bprint_chars(dst, '"', 1);
899 av_bprint_chars(dst, *src, 1);
900 }
901 if (needs_quoting)
902 av_bprint_chars(dst, '"', 1);
903 return dst->str;
904}
905
906static const char *none_escape_str(AVBPrint *dst, const char *src, const char sep, void *log_ctx)
907{
908 return src;
909}
910
911typedef struct CompactContext {
912 const AVClass *class;
913 char *item_sep_str;
914 char item_sep;
915 int nokey;
916 int print_section;
917 char *escape_mode_str;
918 const char * (*escape_str)(AVBPrint *dst, const char *src, const char sep, void *log_ctx);
919 int nested_section[SECTION_MAX_NB_LEVELS];
920 int has_nested_elems[SECTION_MAX_NB_LEVELS];
921 int terminate_line[SECTION_MAX_NB_LEVELS];
922} CompactContext;
923
924#undef OFFSET
925#define OFFSET(x) offsetof(CompactContext, x)
926
927static const AVOption compact_options[]= {
928 {"item_sep", "set item separator", OFFSET(item_sep_str), AV_OPT_TYPE_STRING, {.str="|"}, CHAR_MIN, CHAR_MAX },
929 {"s", "set item separator", OFFSET(item_sep_str), AV_OPT_TYPE_STRING, {.str="|"}, CHAR_MIN, CHAR_MAX },
930 {"nokey", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
931 {"nk", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
932 {"escape", "set escape mode", OFFSET(escape_mode_str), AV_OPT_TYPE_STRING, {.str="c"}, CHAR_MIN, CHAR_MAX },
933 {"e", "set escape mode", OFFSET(escape_mode_str), AV_OPT_TYPE_STRING, {.str="c"}, CHAR_MIN, CHAR_MAX },
934 {"print_section", "print section name", OFFSET(print_section), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
935 {"p", "print section name", OFFSET(print_section), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
936 {NULL},
937};
938
939DEFINE_WRITER_CLASS(compact);
940
941static av_cold int compact_init(WriterContext *wctx)
942{
943 CompactContext *compact = wctx->priv;
944
945 if (strlen(compact->item_sep_str) != 1) {
946 av_log(wctx, AV_LOG_ERROR, "Item separator '%s' specified, but must contain a single character\n",
947 compact->item_sep_str);
948 return AVERROR(EINVAL);
949 }
950 compact->item_sep = compact->item_sep_str[0];
951
952 if (!strcmp(compact->escape_mode_str, "none")) compact->escape_str = none_escape_str;
953 else if (!strcmp(compact->escape_mode_str, "c" )) compact->escape_str = c_escape_str;
954 else if (!strcmp(compact->escape_mode_str, "csv" )) compact->escape_str = csv_escape_str;
955 else {
956 av_log(wctx, AV_LOG_ERROR, "Unknown escape mode '%s'\n", compact->escape_mode_str);
957 return AVERROR(EINVAL);
958 }
959
960 return 0;
961}
962
963static void compact_print_section_header(WriterContext *wctx)
964{
965 CompactContext *compact = wctx->priv;
966 const struct section *section = wctx->section[wctx->level];
967 const struct section *parent_section = wctx->level ?
968 wctx->section[wctx->level-1] : NULL;
969 compact->terminate_line[wctx->level] = 1;
970 compact->has_nested_elems[wctx->level] = 0;
971
972 av_bprint_clear(&wctx->section_pbuf[wctx->level]);
973 if (!(section->flags & SECTION_FLAG_IS_ARRAY) && parent_section &&
974 !(parent_section->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY))) {
975 compact->nested_section[wctx->level] = 1;
976 compact->has_nested_elems[wctx->level-1] = 1;
977 av_bprintf(&wctx->section_pbuf[wctx->level], "%s%s:",
978 wctx->section_pbuf[wctx->level-1].str,
979 (char *)av_x_if_null(section->element_name, section->name));
980 wctx->nb_item[wctx->level] = wctx->nb_item[wctx->level-1];
981 } else {
982 if (parent_section && compact->has_nested_elems[wctx->level-1] &&
983 (section->flags & SECTION_FLAG_IS_ARRAY)) {
984 compact->terminate_line[wctx->level-1] = 0;
985 printf("\n");
986 }
987 if (compact->print_section &&
988 !(section->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY)))
989 printf("%s%c", section->name, compact->item_sep);
990 }
991}
992
993static void compact_print_section_footer(WriterContext *wctx)
994{
995 CompactContext *compact = wctx->priv;
996
997 if (!compact->nested_section[wctx->level] &&
998 compact->terminate_line[wctx->level] &&
999 !(wctx->section[wctx->level]->flags & (SECTION_FLAG_IS_WRAPPER|SECTION_FLAG_IS_ARRAY)))
1000 printf("\n");
1001}
1002
1003static void compact_print_str(WriterContext *wctx, const char *key, const char *value)
1004{
1005 CompactContext *compact = wctx->priv;
1006 AVBPrint buf;
1007
1008 if (wctx->nb_item[wctx->level]) printf("%c", compact->item_sep);
1009 if (!compact->nokey)
1010 printf("%s%s=", wctx->section_pbuf[wctx->level].str, key);
1011 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1012 printf("%s", compact->escape_str(&buf, value, compact->item_sep, wctx));
1013 av_bprint_finalize(&buf, NULL);
1014}
1015
1016static void compact_print_int(WriterContext *wctx, const char *key, long long int value)
1017{
1018 CompactContext *compact = wctx->priv;
1019
1020 if (wctx->nb_item[wctx->level]) printf("%c", compact->item_sep);
1021 if (!compact->nokey)
1022 printf("%s%s=", wctx->section_pbuf[wctx->level].str, key);
1023 printf("%lld", value);
1024}
1025
1026static const Writer compact_writer = {
1027 .name = "compact",
1028 .priv_size = sizeof(CompactContext),
1029 .init = compact_init,
1030 .print_section_header = compact_print_section_header,
1031 .print_section_footer = compact_print_section_footer,
1032 .print_integer = compact_print_int,
1033 .print_string = compact_print_str,
1034 .flags = WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS,
1035 .priv_class = &compact_class,
1036};
1037
1038/* CSV output */
1039
1040#undef OFFSET
1041#define OFFSET(x) offsetof(CompactContext, x)
1042
1043static const AVOption csv_options[] = {
1044 {"item_sep", "set item separator", OFFSET(item_sep_str), AV_OPT_TYPE_STRING, {.str=","}, CHAR_MIN, CHAR_MAX },
1045 {"s", "set item separator", OFFSET(item_sep_str), AV_OPT_TYPE_STRING, {.str=","}, CHAR_MIN, CHAR_MAX },
1046 {"nokey", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1047 {"nk", "force no key printing", OFFSET(nokey), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1048 {"escape", "set escape mode", OFFSET(escape_mode_str), AV_OPT_TYPE_STRING, {.str="csv"}, CHAR_MIN, CHAR_MAX },
1049 {"e", "set escape mode", OFFSET(escape_mode_str), AV_OPT_TYPE_STRING, {.str="csv"}, CHAR_MIN, CHAR_MAX },
1050 {"print_section", "print section name", OFFSET(print_section), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1051 {"p", "print section name", OFFSET(print_section), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1052 {NULL},
1053};
1054
1055DEFINE_WRITER_CLASS(csv);
1056
1057static const Writer csv_writer = {
1058 .name = "csv",
1059 .priv_size = sizeof(CompactContext),
1060 .init = compact_init,
1061 .print_section_header = compact_print_section_header,
1062 .print_section_footer = compact_print_section_footer,
1063 .print_integer = compact_print_int,
1064 .print_string = compact_print_str,
1065 .flags = WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS,
1066 .priv_class = &csv_class,
1067};
1068
1069/* Flat output */
1070
1071typedef struct FlatContext {
1072 const AVClass *class;
1073 const char *sep_str;
1074 char sep;
1075 int hierarchical;
1076} FlatContext;
1077
1078#undef OFFSET
1079#define OFFSET(x) offsetof(FlatContext, x)
1080
1081static const AVOption flat_options[]= {
1082 {"sep_char", "set separator", OFFSET(sep_str), AV_OPT_TYPE_STRING, {.str="."}, CHAR_MIN, CHAR_MAX },
1083 {"s", "set separator", OFFSET(sep_str), AV_OPT_TYPE_STRING, {.str="."}, CHAR_MIN, CHAR_MAX },
1084 {"hierarchical", "specify if the section specification should be hierarchical", OFFSET(hierarchical), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1085 {"h", "specify if the section specification should be hierarchical", OFFSET(hierarchical), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1086 {NULL},
1087};
1088
1089DEFINE_WRITER_CLASS(flat);
1090
1091static av_cold int flat_init(WriterContext *wctx)
1092{
1093 FlatContext *flat = wctx->priv;
1094
1095 if (strlen(flat->sep_str) != 1) {
1096 av_log(wctx, AV_LOG_ERROR, "Item separator '%s' specified, but must contain a single character\n",
1097 flat->sep_str);
1098 return AVERROR(EINVAL);
1099 }
1100 flat->sep = flat->sep_str[0];
1101
1102 return 0;
1103}
1104
1105static const char *flat_escape_key_str(AVBPrint *dst, const char *src, const char sep)
1106{
1107 const char *p;
1108
1109 for (p = src; *p; p++) {
1110 if (!((*p >= '0' && *p <= '9') ||
1111 (*p >= 'a' && *p <= 'z') ||
1112 (*p >= 'A' && *p <= 'Z')))
1113 av_bprint_chars(dst, '_', 1);
1114 else
1115 av_bprint_chars(dst, *p, 1);
1116 }
1117 return dst->str;
1118}
1119
1120static const char *flat_escape_value_str(AVBPrint *dst, const char *src)
1121{
1122 const char *p;
1123
1124 for (p = src; *p; p++) {
1125 switch (*p) {
1126 case '\n': av_bprintf(dst, "%s", "\\n"); break;
1127 case '\r': av_bprintf(dst, "%s", "\\r"); break;
1128 case '\\': av_bprintf(dst, "%s", "\\\\"); break;
1129 case '"': av_bprintf(dst, "%s", "\\\""); break;
1130 case '`': av_bprintf(dst, "%s", "\\`"); break;
1131 case '$': av_bprintf(dst, "%s", "\\$"); break;
1132 default: av_bprint_chars(dst, *p, 1); break;
1133 }
1134 }
1135 return dst->str;
1136}
1137
1138static void flat_print_section_header(WriterContext *wctx)
1139{
1140 FlatContext *flat = wctx->priv;
1141 AVBPrint *buf = &wctx->section_pbuf[wctx->level];
1142 const struct section *section = wctx->section[wctx->level];
1143 const struct section *parent_section = wctx->level ?
1144 wctx->section[wctx->level-1] : NULL;
1145
1146 /* build section header */
1147 av_bprint_clear(buf);
1148 if (!parent_section)
1149 return;
1150 av_bprintf(buf, "%s", wctx->section_pbuf[wctx->level-1].str);
1151
1152 if (flat->hierarchical ||
1153 !(section->flags & (SECTION_FLAG_IS_ARRAY|SECTION_FLAG_IS_WRAPPER))) {
1154 av_bprintf(buf, "%s%s", wctx->section[wctx->level]->name, flat->sep_str);
1155
1156 if (parent_section->flags & SECTION_FLAG_IS_ARRAY) {
1157 int n = parent_section->id == SECTION_ID_PACKETS_AND_FRAMES ?
1158 wctx->nb_section_packet_frame : wctx->nb_item[wctx->level-1];
1159 av_bprintf(buf, "%d%s", n, flat->sep_str);
1160 }
1161 }
1162}
1163
1164static void flat_print_int(WriterContext *wctx, const char *key, long long int value)
1165{
1166 printf("%s%s=%lld\n", wctx->section_pbuf[wctx->level].str, key, value);
1167}
1168
1169static void flat_print_str(WriterContext *wctx, const char *key, const char *value)
1170{
1171 FlatContext *flat = wctx->priv;
1172 AVBPrint buf;
1173
1174 printf("%s", wctx->section_pbuf[wctx->level].str);
1175 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1176 printf("%s=", flat_escape_key_str(&buf, key, flat->sep));
1177 av_bprint_clear(&buf);
1178 printf("\"%s\"\n", flat_escape_value_str(&buf, value));
1179 av_bprint_finalize(&buf, NULL);
1180}
1181
1182static const Writer flat_writer = {
1183 .name = "flat",
1184 .priv_size = sizeof(FlatContext),
1185 .init = flat_init,
1186 .print_section_header = flat_print_section_header,
1187 .print_integer = flat_print_int,
1188 .print_string = flat_print_str,
1189 .flags = WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS|WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER,
1190 .priv_class = &flat_class,
1191};
1192
1193/* INI format output */
1194
1195typedef struct {
1196 const AVClass *class;
1197 int hierarchical;
1198} INIContext;
1199
1200#undef OFFSET
1201#define OFFSET(x) offsetof(INIContext, x)
1202
1203static const AVOption ini_options[] = {
1204 {"hierarchical", "specify if the section specification should be hierarchical", OFFSET(hierarchical), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1205 {"h", "specify if the section specification should be hierarchical", OFFSET(hierarchical), AV_OPT_TYPE_INT, {.i64=1}, 0, 1 },
1206 {NULL},
1207};
1208
1209DEFINE_WRITER_CLASS(ini);
1210
1211static char *ini_escape_str(AVBPrint *dst, const char *src)
1212{
1213 int i = 0;
1214 char c = 0;
1215
1216 while (c = src[i++]) {
1217 switch (c) {
1218 case '\b': av_bprintf(dst, "%s", "\\b"); break;
1219 case '\f': av_bprintf(dst, "%s", "\\f"); break;
1220 case '\n': av_bprintf(dst, "%s", "\\n"); break;
1221 case '\r': av_bprintf(dst, "%s", "\\r"); break;
1222 case '\t': av_bprintf(dst, "%s", "\\t"); break;
1223 case '\\':
1224 case '#' :
1225 case '=' :
1226 case ':' : av_bprint_chars(dst, '\\', 1);
1227 default:
1228 if ((unsigned char)c < 32)
1229 av_bprintf(dst, "\\x00%02x", c & 0xff);
1230 else
1231 av_bprint_chars(dst, c, 1);
1232 break;
1233 }
1234 }
1235 return dst->str;
1236}
1237
1238static void ini_print_section_header(WriterContext *wctx)
1239{
1240 INIContext *ini = wctx->priv;
1241 AVBPrint *buf = &wctx->section_pbuf[wctx->level];
1242 const struct section *section = wctx->section[wctx->level];
1243 const struct section *parent_section = wctx->level ?
1244 wctx->section[wctx->level-1] : NULL;
1245
1246 av_bprint_clear(buf);
1247 if (!parent_section) {
1248 printf("# ffprobe output\n\n");
1249 return;
1250 }
1251
1252 if (wctx->nb_item[wctx->level-1])
1253 printf("\n");
1254
1255 av_bprintf(buf, "%s", wctx->section_pbuf[wctx->level-1].str);
1256 if (ini->hierarchical ||
1257 !(section->flags & (SECTION_FLAG_IS_ARRAY|SECTION_FLAG_IS_WRAPPER))) {
1258 av_bprintf(buf, "%s%s", buf->str[0] ? "." : "", wctx->section[wctx->level]->name);
1259
1260 if (parent_section->flags & SECTION_FLAG_IS_ARRAY) {
1261 int n = parent_section->id == SECTION_ID_PACKETS_AND_FRAMES ?
1262 wctx->nb_section_packet_frame : wctx->nb_item[wctx->level-1];
1263 av_bprintf(buf, ".%d", n);
1264 }
1265 }
1266
1267 if (!(section->flags & (SECTION_FLAG_IS_ARRAY|SECTION_FLAG_IS_WRAPPER)))
1268 printf("[%s]\n", buf->str);
1269}
1270
1271static void ini_print_str(WriterContext *wctx, const char *key, const char *value)
1272{
1273 AVBPrint buf;
1274
1275 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1276 printf("%s=", ini_escape_str(&buf, key));
1277 av_bprint_clear(&buf);
1278 printf("%s\n", ini_escape_str(&buf, value));
1279 av_bprint_finalize(&buf, NULL);
1280}
1281
1282static void ini_print_int(WriterContext *wctx, const char *key, long long int value)
1283{
1284 printf("%s=%lld\n", key, value);
1285}
1286
1287static const Writer ini_writer = {
1288 .name = "ini",
1289 .priv_size = sizeof(INIContext),
1290 .print_section_header = ini_print_section_header,
1291 .print_integer = ini_print_int,
1292 .print_string = ini_print_str,
1293 .flags = WRITER_FLAG_DISPLAY_OPTIONAL_FIELDS|WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER,
1294 .priv_class = &ini_class,
1295};
1296
1297/* JSON output */
1298
1299typedef struct {
1300 const AVClass *class;
1301 int indent_level;
1302 int compact;
1303 const char *item_sep, *item_start_end;
1304} JSONContext;
1305
1306#undef OFFSET
1307#define OFFSET(x) offsetof(JSONContext, x)
1308
1309static const AVOption json_options[]= {
1310 { "compact", "enable compact output", OFFSET(compact), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1311 { "c", "enable compact output", OFFSET(compact), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1312 { NULL }
1313};
1314
1315DEFINE_WRITER_CLASS(json);
1316
1317static av_cold int json_init(WriterContext *wctx)
1318{
1319 JSONContext *json = wctx->priv;
1320
1321 json->item_sep = json->compact ? ", " : ",\n";
1322 json->item_start_end = json->compact ? " " : "\n";
1323
1324 return 0;
1325}
1326
1327static const char *json_escape_str(AVBPrint *dst, const char *src, void *log_ctx)
1328{
1329 static const char json_escape[] = {'"', '\\', '\b', '\f', '\n', '\r', '\t', 0};
1330 static const char json_subst[] = {'"', '\\', 'b', 'f', 'n', 'r', 't', 0};
1331 const char *p;
1332
1333 for (p = src; *p; p++) {
1334 char *s = strchr(json_escape, *p);
1335 if (s) {
1336 av_bprint_chars(dst, '\\', 1);
1337 av_bprint_chars(dst, json_subst[s - json_escape], 1);
1338 } else if ((unsigned char)*p < 32) {
1339 av_bprintf(dst, "\\u00%02x", *p & 0xff);
1340 } else {
1341 av_bprint_chars(dst, *p, 1);
1342 }
1343 }
1344 return dst->str;
1345}
1346
1347#define JSON_INDENT() printf("%*c", json->indent_level * 4, ' ')
1348
1349static void json_print_section_header(WriterContext *wctx)
1350{
1351 JSONContext *json = wctx->priv;
1352 AVBPrint buf;
1353 const struct section *section = wctx->section[wctx->level];
1354 const struct section *parent_section = wctx->level ?
1355 wctx->section[wctx->level-1] : NULL;
1356
1357 if (wctx->level && wctx->nb_item[wctx->level-1])
1358 printf(",\n");
1359
1360 if (section->flags & SECTION_FLAG_IS_WRAPPER) {
1361 printf("{\n");
1362 json->indent_level++;
1363 } else {
1364 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1365 json_escape_str(&buf, section->name, wctx);
1366 JSON_INDENT();
1367
1368 json->indent_level++;
1369 if (section->flags & SECTION_FLAG_IS_ARRAY) {
1370 printf("\"%s\": [\n", buf.str);
1371 } else if (parent_section && !(parent_section->flags & SECTION_FLAG_IS_ARRAY)) {
1372 printf("\"%s\": {%s", buf.str, json->item_start_end);
1373 } else {
1374 printf("{%s", json->item_start_end);
1375
1376 /* this is required so the parser can distinguish between packets and frames */
1377 if (parent_section && parent_section->id == SECTION_ID_PACKETS_AND_FRAMES) {
1378 if (!json->compact)
1379 JSON_INDENT();
1380 printf("\"type\": \"%s\"%s", section->name, json->item_sep);
1381 }
1382 }
1383 av_bprint_finalize(&buf, NULL);
1384 }
1385}
1386
1387static void json_print_section_footer(WriterContext *wctx)
1388{
1389 JSONContext *json = wctx->priv;
1390 const struct section *section = wctx->section[wctx->level];
1391
1392 if (wctx->level == 0) {
1393 json->indent_level--;
1394 printf("\n}\n");
1395 } else if (section->flags & SECTION_FLAG_IS_ARRAY) {
1396 printf("\n");
1397 json->indent_level--;
1398 JSON_INDENT();
1399 printf("]");
1400 } else {
1401 printf("%s", json->item_start_end);
1402 json->indent_level--;
1403 if (!json->compact)
1404 JSON_INDENT();
1405 printf("}");
1406 }
1407}
1408
1409static inline void json_print_item_str(WriterContext *wctx,
1410 const char *key, const char *value)
1411{
1412 AVBPrint buf;
1413
1414 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1415 printf("\"%s\":", json_escape_str(&buf, key, wctx));
1416 av_bprint_clear(&buf);
1417 printf(" \"%s\"", json_escape_str(&buf, value, wctx));
1418 av_bprint_finalize(&buf, NULL);
1419}
1420
1421static void json_print_str(WriterContext *wctx, const char *key, const char *value)
1422{
1423 JSONContext *json = wctx->priv;
1424
1425 if (wctx->nb_item[wctx->level])
1426 printf("%s", json->item_sep);
1427 if (!json->compact)
1428 JSON_INDENT();
1429 json_print_item_str(wctx, key, value);
1430}
1431
1432static void json_print_int(WriterContext *wctx, const char *key, long long int value)
1433{
1434 JSONContext *json = wctx->priv;
1435 AVBPrint buf;
1436
1437 if (wctx->nb_item[wctx->level])
1438 printf("%s", json->item_sep);
1439 if (!json->compact)
1440 JSON_INDENT();
1441
1442 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1443 printf("\"%s\": %lld", json_escape_str(&buf, key, wctx), value);
1444 av_bprint_finalize(&buf, NULL);
1445}
1446
1447static const Writer json_writer = {
1448 .name = "json",
1449 .priv_size = sizeof(JSONContext),
1450 .init = json_init,
1451 .print_section_header = json_print_section_header,
1452 .print_section_footer = json_print_section_footer,
1453 .print_integer = json_print_int,
1454 .print_string = json_print_str,
1455 .flags = WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER,
1456 .priv_class = &json_class,
1457};
1458
1459/* XML output */
1460
1461typedef struct {
1462 const AVClass *class;
1463 int within_tag;
1464 int indent_level;
1465 int fully_qualified;
1466 int xsd_strict;
1467} XMLContext;
1468
1469#undef OFFSET
1470#define OFFSET(x) offsetof(XMLContext, x)
1471
1472static const AVOption xml_options[] = {
1473 {"fully_qualified", "specify if the output should be fully qualified", OFFSET(fully_qualified), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1474 {"q", "specify if the output should be fully qualified", OFFSET(fully_qualified), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1475 {"xsd_strict", "ensure that the output is XSD compliant", OFFSET(xsd_strict), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1476 {"x", "ensure that the output is XSD compliant", OFFSET(xsd_strict), AV_OPT_TYPE_INT, {.i64=0}, 0, 1 },
1477 {NULL},
1478};
1479
1480DEFINE_WRITER_CLASS(xml);
1481
1482static av_cold int xml_init(WriterContext *wctx)
1483{
1484 XMLContext *xml = wctx->priv;
1485
1486 if (xml->xsd_strict) {
1487 xml->fully_qualified = 1;
1488#define CHECK_COMPLIANCE(opt, opt_name) \
1489 if (opt) { \
1490 av_log(wctx, AV_LOG_ERROR, \
1491 "XSD-compliant output selected but option '%s' was selected, XML output may be non-compliant.\n" \
1492 "You need to disable such option with '-no%s'\n", opt_name, opt_name); \
1493 return AVERROR(EINVAL); \
1494 }
1495 CHECK_COMPLIANCE(show_private_data, "private");
1496 CHECK_COMPLIANCE(show_value_unit, "unit");
1497 CHECK_COMPLIANCE(use_value_prefix, "prefix");
1498
1499 if (do_show_frames && do_show_packets) {
1500 av_log(wctx, AV_LOG_ERROR,
1501 "Interleaved frames and packets are not allowed in XSD. "
1502 "Select only one between the -show_frames and the -show_packets options.\n");
1503 return AVERROR(EINVAL);
1504 }
1505 }
1506
1507 return 0;
1508}
1509
1510static const char *xml_escape_str(AVBPrint *dst, const char *src, void *log_ctx)
1511{
1512 const char *p;
1513
1514 for (p = src; *p; p++) {
1515 switch (*p) {
1516 case '&' : av_bprintf(dst, "%s", "&amp;"); break;
1517 case '<' : av_bprintf(dst, "%s", "&lt;"); break;
1518 case '>' : av_bprintf(dst, "%s", "&gt;"); break;
1519 case '"' : av_bprintf(dst, "%s", "&quot;"); break;
1520 case '\'': av_bprintf(dst, "%s", "&apos;"); break;
1521 default: av_bprint_chars(dst, *p, 1);
1522 }
1523 }
1524
1525 return dst->str;
1526}
1527
1528#define XML_INDENT() printf("%*c", xml->indent_level * 4, ' ')
1529
1530static void xml_print_section_header(WriterContext *wctx)
1531{
1532 XMLContext *xml = wctx->priv;
1533 const struct section *section = wctx->section[wctx->level];
1534 const struct section *parent_section = wctx->level ?
1535 wctx->section[wctx->level-1] : NULL;
1536
1537 if (wctx->level == 0) {
1538 const char *qual = " xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' "
1539 "xmlns:ffprobe='http://www.ffmpeg.org/schema/ffprobe' "
1540 "xsi:schemaLocation='http://www.ffmpeg.org/schema/ffprobe ffprobe.xsd'";
1541
1542 printf("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
1543 printf("<%sffprobe%s>\n",
1544 xml->fully_qualified ? "ffprobe:" : "",
1545 xml->fully_qualified ? qual : "");
1546 return;
1547 }
1548
1549 if (xml->within_tag) {
1550 xml->within_tag = 0;
1551 printf(">\n");
1552 }
1553 if (section->flags & SECTION_FLAG_HAS_VARIABLE_FIELDS) {
1554 xml->indent_level++;
1555 } else {
1556 if (parent_section && (parent_section->flags & SECTION_FLAG_IS_WRAPPER) &&
1557 wctx->level && wctx->nb_item[wctx->level-1])
1558 printf("\n");
1559 xml->indent_level++;
1560
1561 if (section->flags & SECTION_FLAG_IS_ARRAY) {
1562 XML_INDENT(); printf("<%s>\n", section->name);
1563 } else {
1564 XML_INDENT(); printf("<%s ", section->name);
1565 xml->within_tag = 1;
1566 }
1567 }
1568}
1569
1570static void xml_print_section_footer(WriterContext *wctx)
1571{
1572 XMLContext *xml = wctx->priv;
1573 const struct section *section = wctx->section[wctx->level];
1574
1575 if (wctx->level == 0) {
1576 printf("</%sffprobe>\n", xml->fully_qualified ? "ffprobe:" : "");
1577 } else if (xml->within_tag) {
1578 xml->within_tag = 0;
1579 printf("/>\n");
1580 xml->indent_level--;
1581 } else if (section->flags & SECTION_FLAG_HAS_VARIABLE_FIELDS) {
1582 xml->indent_level--;
1583 } else {
1584 XML_INDENT(); printf("</%s>\n", section->name);
1585 xml->indent_level--;
1586 }
1587}
1588
1589static void xml_print_str(WriterContext *wctx, const char *key, const char *value)
1590{
1591 AVBPrint buf;
1592 XMLContext *xml = wctx->priv;
1593 const struct section *section = wctx->section[wctx->level];
1594
1595 av_bprint_init(&buf, 1, AV_BPRINT_SIZE_UNLIMITED);
1596
1597 if (section->flags & SECTION_FLAG_HAS_VARIABLE_FIELDS) {
1598 XML_INDENT();
1599 printf("<%s key=\"%s\"",
1600 section->element_name, xml_escape_str(&buf, key, wctx));
1601 av_bprint_clear(&buf);
1602 printf(" value=\"%s\"/>\n", xml_escape_str(&buf, value, wctx));
1603 } else {
1604 if (wctx->nb_item[wctx->level])
1605 printf(" ");
1606 printf("%s=\"%s\"", key, xml_escape_str(&buf, value, wctx));
1607 }
1608
1609 av_bprint_finalize(&buf, NULL);
1610}
1611
1612static void xml_print_int(WriterContext *wctx, const char *key, long long int value)
1613{
1614 if (wctx->nb_item[wctx->level])
1615 printf(" ");
1616 printf("%s=\"%lld\"", key, value);
1617}
1618
1619static Writer xml_writer = {
1620 .name = "xml",
1621 .priv_size = sizeof(XMLContext),
1622 .init = xml_init,
1623 .print_section_header = xml_print_section_header,
1624 .print_section_footer = xml_print_section_footer,
1625 .print_integer = xml_print_int,
1626 .print_string = xml_print_str,
1627 .flags = WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER,
1628 .priv_class = &xml_class,
1629};
1630
1631static void writer_register_all(void)
1632{
1633 static int initialized;
1634
1635 if (initialized)
1636 return;
1637 initialized = 1;
1638
1639 writer_register(&default_writer);
1640 writer_register(&compact_writer);
1641 writer_register(&csv_writer);
1642 writer_register(&flat_writer);
1643 writer_register(&ini_writer);
1644 writer_register(&json_writer);
1645 writer_register(&xml_writer);
1646}
1647
1648#define print_fmt(k, f, ...) do { \
1649 av_bprint_clear(&pbuf); \
1650 av_bprintf(&pbuf, f, __VA_ARGS__); \
1651 writer_print_string(w, k, pbuf.str, 0); \
1652} while (0)
1653
1654#define print_int(k, v) writer_print_integer(w, k, v)
1655#define print_q(k, v, s) writer_print_rational(w, k, v, s)
1656#define print_str(k, v) writer_print_string(w, k, v, 0)
1657#define print_str_opt(k, v) writer_print_string(w, k, v, PRINT_STRING_OPT)
1658#define print_str_validate(k, v) writer_print_string(w, k, v, PRINT_STRING_VALIDATE)
1659#define print_time(k, v, tb) writer_print_time(w, k, v, tb, 0)
1660#define print_ts(k, v) writer_print_ts(w, k, v, 0)
1661#define print_duration_time(k, v, tb) writer_print_time(w, k, v, tb, 1)
1662#define print_duration_ts(k, v) writer_print_ts(w, k, v, 1)
1663#define print_val(k, v, u) do { \
1664 struct unit_value uv; \
1665 uv.val.i = v; \
1666 uv.unit = u; \
1667 writer_print_string(w, k, value_string(val_str, sizeof(val_str), uv), 0); \
1668} while (0)
1669
1670#define print_section_header(s) writer_print_section_header(w, s)
1671#define print_section_footer(s) writer_print_section_footer(w, s)
1672
1673#define REALLOCZ_ARRAY_STREAM(ptr, cur_n, new_n) \
1674{ \
1675 ret = av_reallocp_array(&(ptr), (new_n), sizeof(*(ptr))); \
1676 if (ret < 0) \
1677 goto end; \
1678 memset( (ptr) + (cur_n), 0, ((new_n) - (cur_n)) * sizeof(*(ptr)) ); \
1679}
1680
1681static inline int show_tags(WriterContext *w, AVDictionary *tags, int section_id)
1682{
1683 AVDictionaryEntry *tag = NULL;
1684 int ret = 0;
1685
1686 if (!tags)
1687 return 0;
1688 writer_print_section_header(w, section_id);
1689
1690 while ((tag = av_dict_get(tags, "", tag, AV_DICT_IGNORE_SUFFIX))) {
1691 if ((ret = print_str_validate(tag->key, tag->value)) < 0)
1692 break;
1693 }
1694 writer_print_section_footer(w);
1695
1696 return ret;
1697}
1698
1699static void show_packet(WriterContext *w, AVFormatContext *fmt_ctx, AVPacket *pkt, int packet_idx)
1700{
1701 char val_str[128];
1702 AVStream *st = fmt_ctx->streams[pkt->stream_index];
1703 AVBPrint pbuf;
1704 const char *s;
1705
1706 av_bprint_init(&pbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
1707
1708 writer_print_section_header(w, SECTION_ID_PACKET);
1709
1710 s = av_get_media_type_string(st->codec->codec_type);
1711 if (s) print_str ("codec_type", s);
1712 else print_str_opt("codec_type", "unknown");
1713 print_int("stream_index", pkt->stream_index);
1714 print_ts ("pts", pkt->pts);
1715 print_time("pts_time", pkt->pts, &st->time_base);
1716 print_ts ("dts", pkt->dts);
1717 print_time("dts_time", pkt->dts, &st->time_base);
1718 print_duration_ts("duration", pkt->duration);
1719 print_duration_time("duration_time", pkt->duration, &st->time_base);
1720 print_duration_ts("convergence_duration", pkt->convergence_duration);
1721 print_duration_time("convergence_duration_time", pkt->convergence_duration, &st->time_base);
1722 print_val("size", pkt->size, unit_byte_str);
1723 if (pkt->pos != -1) print_fmt ("pos", "%"PRId64, pkt->pos);
1724 else print_str_opt("pos", "N/A");
1725 print_fmt("flags", "%c", pkt->flags & AV_PKT_FLAG_KEY ? 'K' : '_');
1726 if (do_show_data)
1727 writer_print_data(w, "data", pkt->data, pkt->size);
1728 writer_print_data_hash(w, "data_hash", pkt->data, pkt->size);
1729 writer_print_section_footer(w);
1730
1731 av_bprint_finalize(&pbuf, NULL);
1732 fflush(stdout);
1733}
1734
1735static void show_subtitle(WriterContext *w, AVSubtitle *sub, AVStream *stream,
1736 AVFormatContext *fmt_ctx)
1737{
1738 AVBPrint pbuf;
1739
1740 av_bprint_init(&pbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
1741
1742 writer_print_section_header(w, SECTION_ID_SUBTITLE);
1743
1744 print_str ("media_type", "subtitle");
1745 print_ts ("pts", sub->pts);
1746 print_time("pts_time", sub->pts, &AV_TIME_BASE_Q);
1747 print_int ("format", sub->format);
1748 print_int ("start_display_time", sub->start_display_time);
1749 print_int ("end_display_time", sub->end_display_time);
1750 print_int ("num_rects", sub->num_rects);
1751
1752 writer_print_section_footer(w);
1753
1754 av_bprint_finalize(&pbuf, NULL);
1755 fflush(stdout);
1756}
1757
1758static void show_frame(WriterContext *w, AVFrame *frame, AVStream *stream,
1759 AVFormatContext *fmt_ctx)
1760{
1761 AVBPrint pbuf;
1762 const char *s;
1763 int i;
1764
1765 av_bprint_init(&pbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
1766
1767 writer_print_section_header(w, SECTION_ID_FRAME);
1768
1769 s = av_get_media_type_string(stream->codec->codec_type);
1770 if (s) print_str ("media_type", s);
1771 else print_str_opt("media_type", "unknown");
1772 print_int("key_frame", frame->key_frame);
1773 print_ts ("pkt_pts", frame->pkt_pts);
1774 print_time("pkt_pts_time", frame->pkt_pts, &stream->time_base);
1775 print_ts ("pkt_dts", frame->pkt_dts);
1776 print_time("pkt_dts_time", frame->pkt_dts, &stream->time_base);
1777 print_ts ("best_effort_timestamp", av_frame_get_best_effort_timestamp(frame));
1778 print_time("best_effort_timestamp_time", av_frame_get_best_effort_timestamp(frame), &stream->time_base);
1779 print_duration_ts ("pkt_duration", av_frame_get_pkt_duration(frame));
1780 print_duration_time("pkt_duration_time", av_frame_get_pkt_duration(frame), &stream->time_base);
1781 if (av_frame_get_pkt_pos (frame) != -1) print_fmt ("pkt_pos", "%"PRId64, av_frame_get_pkt_pos(frame));
1782 else print_str_opt("pkt_pos", "N/A");
1783 if (av_frame_get_pkt_size(frame) != -1) print_fmt ("pkt_size", "%d", av_frame_get_pkt_size(frame));
1784 else print_str_opt("pkt_size", "N/A");
1785
1786 switch (stream->codec->codec_type) {
1787 AVRational sar;
1788
1789 case AVMEDIA_TYPE_VIDEO:
1790 print_int("width", frame->width);
1791 print_int("height", frame->height);
1792 s = av_get_pix_fmt_name(frame->format);
1793 if (s) print_str ("pix_fmt", s);
1794 else print_str_opt("pix_fmt", "unknown");
1795 sar = av_guess_sample_aspect_ratio(fmt_ctx, stream, frame);
1796 if (sar.num) {
1797 print_q("sample_aspect_ratio", sar, ':');
1798 } else {
1799 print_str_opt("sample_aspect_ratio", "N/A");
1800 }
1801 print_fmt("pict_type", "%c", av_get_picture_type_char(frame->pict_type));
1802 print_int("coded_picture_number", frame->coded_picture_number);
1803 print_int("display_picture_number", frame->display_picture_number);
1804 print_int("interlaced_frame", frame->interlaced_frame);
1805 print_int("top_field_first", frame->top_field_first);
1806 print_int("repeat_pict", frame->repeat_pict);
1807 break;
1808
1809 case AVMEDIA_TYPE_AUDIO:
1810 s = av_get_sample_fmt_name(frame->format);
1811 if (s) print_str ("sample_fmt", s);
1812 else print_str_opt("sample_fmt", "unknown");
1813 print_int("nb_samples", frame->nb_samples);
1814 print_int("channels", av_frame_get_channels(frame));
1815 if (av_frame_get_channel_layout(frame)) {
1816 av_bprint_clear(&pbuf);
1817 av_bprint_channel_layout(&pbuf, av_frame_get_channels(frame),
1818 av_frame_get_channel_layout(frame));
1819 print_str ("channel_layout", pbuf.str);
1820 } else
1821 print_str_opt("channel_layout", "unknown");
1822 break;
1823 }
1824 if (do_show_frame_tags)
1825 show_tags(w, av_frame_get_metadata(frame), SECTION_ID_FRAME_TAGS);
1826 if (frame->nb_side_data) {
1827 writer_print_section_header(w, SECTION_ID_FRAME_SIDE_DATA_LIST);
1828 for (i = 0; i < frame->nb_side_data; i++) {
1829 AVFrameSideData *sd = frame->side_data[i];
1830 const char *name;
1831
1832 writer_print_section_header(w, SECTION_ID_FRAME_SIDE_DATA);
1833 name = av_frame_side_data_name(sd->type);
1834 print_str("side_data_type", name ? name : "unknown");
1835 print_int("side_data_size", sd->size);
1836 writer_print_section_footer(w);
1837 }
1838 writer_print_section_footer(w);
1839 }
1840
1841 writer_print_section_footer(w);
1842
1843 av_bprint_finalize(&pbuf, NULL);
1844 fflush(stdout);
1845}
1846
1847static av_always_inline int process_frame(WriterContext *w,
1848 AVFormatContext *fmt_ctx,
1849 AVFrame *frame, AVPacket *pkt)
1850{
1851 AVCodecContext *dec_ctx = fmt_ctx->streams[pkt->stream_index]->codec;
1852 AVSubtitle sub;
1853 int ret = 0, got_frame = 0;
1854
1855 if (dec_ctx->codec) {
1856 switch (dec_ctx->codec_type) {
1857 case AVMEDIA_TYPE_VIDEO:
1858 ret = avcodec_decode_video2(dec_ctx, frame, &got_frame, pkt);
1859 break;
1860
1861 case AVMEDIA_TYPE_AUDIO:
1862 ret = avcodec_decode_audio4(dec_ctx, frame, &got_frame, pkt);
1863 break;
1864
1865 case AVMEDIA_TYPE_SUBTITLE:
1866 ret = avcodec_decode_subtitle2(dec_ctx, &sub, &got_frame, pkt);
1867 break;
1868 }
1869 }
1870
1871 if (ret < 0)
1872 return ret;
1873 ret = FFMIN(ret, pkt->size); /* guard against bogus return values */
1874 pkt->data += ret;
1875 pkt->size -= ret;
1876 if (got_frame) {
1877 int is_sub = (dec_ctx->codec_type == AVMEDIA_TYPE_SUBTITLE);
1878 nb_streams_frames[pkt->stream_index]++;
1879 if (do_show_frames)
1880 if (is_sub)
1881 show_subtitle(w, &sub, fmt_ctx->streams[pkt->stream_index], fmt_ctx);
1882 else
1883 show_frame(w, frame, fmt_ctx->streams[pkt->stream_index], fmt_ctx);
1884 if (is_sub)
1885 avsubtitle_free(&sub);
1886 }
1887 return got_frame;
1888}
1889
1890static void log_read_interval(const ReadInterval *interval, void *log_ctx, int log_level)
1891{
1892 av_log(log_ctx, log_level, "id:%d", interval->id);
1893
1894 if (interval->has_start) {
1895 av_log(log_ctx, log_level, " start:%s%s", interval->start_is_offset ? "+" : "",
1896 av_ts2timestr(interval->start, &AV_TIME_BASE_Q));
1897 } else {
1898 av_log(log_ctx, log_level, " start:N/A");
1899 }
1900
1901 if (interval->has_end) {
1902 av_log(log_ctx, log_level, " end:%s", interval->end_is_offset ? "+" : "");
1903 if (interval->duration_frames)
1904 av_log(log_ctx, log_level, "#%"PRId64, interval->end);
1905 else
1906 av_log(log_ctx, log_level, "%s", av_ts2timestr(interval->end, &AV_TIME_BASE_Q));
1907 } else {
1908 av_log(log_ctx, log_level, " end:N/A");
1909 }
1910
1911 av_log(log_ctx, log_level, "\n");
1912}
1913
1914static int read_interval_packets(WriterContext *w, AVFormatContext *fmt_ctx,
1915 const ReadInterval *interval, int64_t *cur_ts)
1916{
1917 AVPacket pkt, pkt1;
1918 AVFrame *frame = NULL;
1919 int ret = 0, i = 0, frame_count = 0;
1920 int64_t start = -INT64_MAX, end = interval->end;
1921 int has_start = 0, has_end = interval->has_end && !interval->end_is_offset;
1922
1923 av_init_packet(&pkt);
1924
1925 av_log(NULL, AV_LOG_VERBOSE, "Processing read interval ");
1926 log_read_interval(interval, NULL, AV_LOG_VERBOSE);
1927
1928 if (interval->has_start) {
1929 int64_t target;
1930 if (interval->start_is_offset) {
1931 if (*cur_ts == AV_NOPTS_VALUE) {
1932 av_log(NULL, AV_LOG_ERROR,
1933 "Could not seek to relative position since current "
1934 "timestamp is not defined\n");
1935 ret = AVERROR(EINVAL);
1936 goto end;
1937 }
1938 target = *cur_ts + interval->start;
1939 } else {
1940 target = interval->start;
1941 }
1942
1943 av_log(NULL, AV_LOG_VERBOSE, "Seeking to read interval start point %s\n",
1944 av_ts2timestr(target, &AV_TIME_BASE_Q));
1945 if ((ret = avformat_seek_file(fmt_ctx, -1, -INT64_MAX, target, INT64_MAX, 0)) < 0) {
1946 av_log(NULL, AV_LOG_ERROR, "Could not seek to position %"PRId64": %s\n",
1947 interval->start, av_err2str(ret));
1948 goto end;
1949 }
1950 }
1951
1952 frame = av_frame_alloc();
1953 if (!frame) {
1954 ret = AVERROR(ENOMEM);
1955 goto end;
1956 }
1957 while (!av_read_frame(fmt_ctx, &pkt)) {
1958 if (fmt_ctx->nb_streams > nb_streams) {
1959 REALLOCZ_ARRAY_STREAM(nb_streams_frames, nb_streams, fmt_ctx->nb_streams);
1960 REALLOCZ_ARRAY_STREAM(nb_streams_packets, nb_streams, fmt_ctx->nb_streams);
1961 REALLOCZ_ARRAY_STREAM(selected_streams, nb_streams, fmt_ctx->nb_streams);
1962 nb_streams = fmt_ctx->nb_streams;
1963 }
1964 if (selected_streams[pkt.stream_index]) {
1965 AVRational tb = fmt_ctx->streams[pkt.stream_index]->time_base;
1966
1967 if (pkt.pts != AV_NOPTS_VALUE)
1968 *cur_ts = av_rescale_q(pkt.pts, tb, AV_TIME_BASE_Q);
1969
1970 if (!has_start && *cur_ts != AV_NOPTS_VALUE) {
1971 start = *cur_ts;
1972 has_start = 1;
1973 }
1974
1975 if (has_start && !has_end && interval->end_is_offset) {
1976 end = start + interval->end;
1977 has_end = 1;
1978 }
1979
1980 if (interval->end_is_offset && interval->duration_frames) {
1981 if (frame_count >= interval->end)
1982 break;
1983 } else if (has_end && *cur_ts != AV_NOPTS_VALUE && *cur_ts >= end) {
1984 break;
1985 }
1986
1987 frame_count++;
1988 if (do_read_packets) {
1989 if (do_show_packets)
1990 show_packet(w, fmt_ctx, &pkt, i++);
1991 nb_streams_packets[pkt.stream_index]++;
1992 }
1993 if (do_read_frames) {
1994 pkt1 = pkt;
1995 while (pkt1.size && process_frame(w, fmt_ctx, frame, &pkt1) > 0);
1996 }
1997 }
1998 av_free_packet(&pkt);
1999 }
2000 av_init_packet(&pkt);
2001 pkt.data = NULL;
2002 pkt.size = 0;
2003 //Flush remaining frames that are cached in the decoder
2004 for (i = 0; i < fmt_ctx->nb_streams; i++) {
2005 pkt.stream_index = i;
2006 if (do_read_frames)
2007 while (process_frame(w, fmt_ctx, frame, &pkt) > 0);
2008 }
2009
2010end:
2011 av_frame_free(&frame);
2012 if (ret < 0) {
2013 av_log(NULL, AV_LOG_ERROR, "Could not read packets in interval ");
2014 log_read_interval(interval, NULL, AV_LOG_ERROR);
2015 }
2016 return ret;
2017}
2018
2019static int read_packets(WriterContext *w, AVFormatContext *fmt_ctx)
2020{
2021 int i, ret = 0;
2022 int64_t cur_ts = fmt_ctx->start_time;
2023
2024 if (read_intervals_nb == 0) {
2025 ReadInterval interval = (ReadInterval) { .has_start = 0, .has_end = 0 };
2026 ret = read_interval_packets(w, fmt_ctx, &interval, &cur_ts);
2027 } else {
2028 for (i = 0; i < read_intervals_nb; i++) {
2029 ret = read_interval_packets(w, fmt_ctx, &read_intervals[i], &cur_ts);
2030 if (ret < 0)
2031 break;
2032 }
2033 }
2034
2035 return ret;
2036}
2037
2038static int show_stream(WriterContext *w, AVFormatContext *fmt_ctx, int stream_idx, int in_program)
2039{
2040 AVStream *stream = fmt_ctx->streams[stream_idx];
2041 AVCodecContext *dec_ctx;
2042 const AVCodec *dec;
2043 char val_str[128];
2044 const char *s;
2045 AVRational sar, dar;
2046 AVBPrint pbuf;
2047 const AVCodecDescriptor *cd;
2048 int ret = 0;
2049
2050 av_bprint_init(&pbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
2051
2052 writer_print_section_header(w, in_program ? SECTION_ID_PROGRAM_STREAM : SECTION_ID_STREAM);
2053
2054 print_int("index", stream->index);
2055
2056 if ((dec_ctx = stream->codec)) {
2057 const char *profile = NULL;
2058 dec = dec_ctx->codec;
2059 if (dec) {
2060 print_str("codec_name", dec->name);
2061 if (!do_bitexact) {
2062 if (dec->long_name) print_str ("codec_long_name", dec->long_name);
2063 else print_str_opt("codec_long_name", "unknown");
2064 }
2065 } else if ((cd = avcodec_descriptor_get(stream->codec->codec_id))) {
2066 print_str_opt("codec_name", cd->name);
2067 if (!do_bitexact) {
2068 print_str_opt("codec_long_name",
2069 cd->long_name ? cd->long_name : "unknown");
2070 }
2071 } else {
2072 print_str_opt("codec_name", "unknown");
2073 if (!do_bitexact) {
2074 print_str_opt("codec_long_name", "unknown");
2075 }
2076 }
2077
2078 if (dec && (profile = av_get_profile_name(dec, dec_ctx->profile)))
2079 print_str("profile", profile);
2080 else
2081 print_str_opt("profile", "unknown");
2082
2083 s = av_get_media_type_string(dec_ctx->codec_type);
2084 if (s) print_str ("codec_type", s);
2085 else print_str_opt("codec_type", "unknown");
2086 print_q("codec_time_base", dec_ctx->time_base, '/');
2087
2088 /* print AVI/FourCC tag */
2089 av_get_codec_tag_string(val_str, sizeof(val_str), dec_ctx->codec_tag);
2090 print_str("codec_tag_string", val_str);
2091 print_fmt("codec_tag", "0x%04x", dec_ctx->codec_tag);
2092
2093 switch (dec_ctx->codec_type) {
2094 case AVMEDIA_TYPE_VIDEO:
2095 print_int("width", dec_ctx->width);
2096 print_int("height", dec_ctx->height);
2097 print_int("has_b_frames", dec_ctx->has_b_frames);
2098 sar = av_guess_sample_aspect_ratio(fmt_ctx, stream, NULL);
2099 if (sar.den) {
2100 print_q("sample_aspect_ratio", sar, ':');
2101 av_reduce(&dar.num, &dar.den,
2102 dec_ctx->width * sar.num,
2103 dec_ctx->height * sar.den,
2104 1024*1024);
2105 print_q("display_aspect_ratio", dar, ':');
2106 } else {
2107 print_str_opt("sample_aspect_ratio", "N/A");
2108 print_str_opt("display_aspect_ratio", "N/A");
2109 }
2110 s = av_get_pix_fmt_name(dec_ctx->pix_fmt);
2111 if (s) print_str ("pix_fmt", s);
2112 else print_str_opt("pix_fmt", "unknown");
2113 print_int("level", dec_ctx->level);
2114 if (dec_ctx->color_range != AVCOL_RANGE_UNSPECIFIED)
f6fa7814 2115 print_str ("color_range", av_color_range_name(dec_ctx->color_range));
2ba45a60
DM		 2116 else
2117 print_str_opt("color_range", "N/A");
2118 s = av_get_colorspace_name(dec_ctx->colorspace);
2119 if (s) print_str ("color_space", s);
2120 else print_str_opt("color_space", "unknown");
f6fa7814
DM		 2121
2122 if (dec_ctx->color_trc != AVCOL_TRC_UNSPECIFIED)
2123 print_str("color_transfer", av_color_transfer_name(dec_ctx->color_trc));
2124 else
2125 print_str_opt("color_transfer", av_color_transfer_name(dec_ctx->color_trc));
2126
2127 if (dec_ctx->color_primaries != AVCOL_PRI_UNSPECIFIED)
2128 print_str("color_primaries", av_color_primaries_name(dec_ctx->color_primaries));
2129 else
2130 print_str_opt("color_primaries", av_color_primaries_name(dec_ctx->color_primaries));
2131
2132 if (dec_ctx->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED)
2133 print_str("chroma_location", av_chroma_location_name(dec_ctx->chroma_sample_location));
2134 else
2135 print_str_opt("chroma_location", av_chroma_location_name(dec_ctx->chroma_sample_location));
2136
2ba45a60
DM		 2137 if (dec_ctx->timecode_frame_start >= 0) {
2138 char tcbuf[AV_TIMECODE_STR_SIZE];
2139 av_timecode_make_mpeg_tc_string(tcbuf, dec_ctx->timecode_frame_start);
2140 print_str("timecode", tcbuf);
2141 } else {
2142 print_str_opt("timecode", "N/A");
2143 }
2144 break;
2145
2146 case AVMEDIA_TYPE_AUDIO:
2147 s = av_get_sample_fmt_name(dec_ctx->sample_fmt);
2148 if (s) print_str ("sample_fmt", s);
2149 else print_str_opt("sample_fmt", "unknown");
2150 print_val("sample_rate", dec_ctx->sample_rate, unit_hertz_str);
2151 print_int("channels", dec_ctx->channels);
2152
2153 if (dec_ctx->channel_layout) {
2154 av_bprint_clear(&pbuf);
2155 av_bprint_channel_layout(&pbuf, dec_ctx->channels, dec_ctx->channel_layout);
2156 print_str ("channel_layout", pbuf.str);
2157 } else {
2158 print_str_opt("channel_layout", "unknown");
2159 }
2160
2161 print_int("bits_per_sample", av_get_bits_per_sample(dec_ctx->codec_id));
2162 break;
2163
2164 case AVMEDIA_TYPE_SUBTITLE:
2165 if (dec_ctx->width)
2166 print_int("width", dec_ctx->width);
2167 else
2168 print_str_opt("width", "N/A");
2169 if (dec_ctx->height)
2170 print_int("height", dec_ctx->height);
2171 else
2172 print_str_opt("height", "N/A");
2173 break;
2174 }
2175 } else {
2176 print_str_opt("codec_type", "unknown");
2177 }
2178 if (dec_ctx->codec && dec_ctx->codec->priv_class && show_private_data) {
2179 const AVOption *opt = NULL;
2180 while (opt = av_opt_next(dec_ctx->priv_data,opt)) {
2181 uint8_t *str;
2182 if (opt->flags) continue;
2183 if (av_opt_get(dec_ctx->priv_data, opt->name, 0, &str) >= 0) {
2184 print_str(opt->name, str);
2185 av_free(str);
2186 }
2187 }
2188 }
2189
2190 if (fmt_ctx->iformat->flags & AVFMT_SHOW_IDS) print_fmt ("id", "0x%x", stream->id);
2191 else print_str_opt("id", "N/A");
2192 print_q("r_frame_rate", stream->r_frame_rate, '/');
2193 print_q("avg_frame_rate", stream->avg_frame_rate, '/');
2194 print_q("time_base", stream->time_base, '/');
2195 print_ts ("start_pts", stream->start_time);
2196 print_time("start_time", stream->start_time, &stream->time_base);
2197 print_ts ("duration_ts", stream->duration);
2198 print_time("duration", stream->duration, &stream->time_base);
2199 if (dec_ctx->bit_rate > 0) print_val ("bit_rate", dec_ctx->bit_rate, unit_bit_per_second_str);
2200 else print_str_opt("bit_rate", "N/A");
2201 if (dec_ctx->rc_max_rate > 0) print_val ("max_bit_rate", dec_ctx->rc_max_rate, unit_bit_per_second_str);
2202 else print_str_opt("max_bit_rate", "N/A");
2203 if (dec_ctx->bits_per_raw_sample > 0) print_fmt("bits_per_raw_sample", "%d", dec_ctx->bits_per_raw_sample);
2204 else print_str_opt("bits_per_raw_sample", "N/A");
2205 if (stream->nb_frames) print_fmt ("nb_frames", "%"PRId64, stream->nb_frames);
2206 else print_str_opt("nb_frames", "N/A");
2207 if (nb_streams_frames[stream_idx]) print_fmt ("nb_read_frames", "%"PRIu64, nb_streams_frames[stream_idx]);
2208 else print_str_opt("nb_read_frames", "N/A");
2209 if (nb_streams_packets[stream_idx]) print_fmt ("nb_read_packets", "%"PRIu64, nb_streams_packets[stream_idx]);
2210 else print_str_opt("nb_read_packets", "N/A");
2211 if (do_show_data)
2212 writer_print_data(w, "extradata", dec_ctx->extradata,
2213 dec_ctx->extradata_size);
2214 writer_print_data_hash(w, "extradata_hash", dec_ctx->extradata,
2215 dec_ctx->extradata_size);
2216
2217 /* Print disposition information */
2218#define PRINT_DISPOSITION(flagname, name) do { \
2219 print_int(name, !!(stream->disposition & AV_DISPOSITION_##flagname)); \
2220 } while (0)
2221
2222 if (do_show_stream_disposition) {
2223 writer_print_section_header(w, in_program ? SECTION_ID_PROGRAM_STREAM_DISPOSITION : SECTION_ID_STREAM_DISPOSITION);
2224 PRINT_DISPOSITION(DEFAULT, "default");
2225 PRINT_DISPOSITION(DUB, "dub");
2226 PRINT_DISPOSITION(ORIGINAL, "original");
2227 PRINT_DISPOSITION(COMMENT, "comment");
2228 PRINT_DISPOSITION(LYRICS, "lyrics");
2229 PRINT_DISPOSITION(KARAOKE, "karaoke");
2230 PRINT_DISPOSITION(FORCED, "forced");
2231 PRINT_DISPOSITION(HEARING_IMPAIRED, "hearing_impaired");
2232 PRINT_DISPOSITION(VISUAL_IMPAIRED, "visual_impaired");
2233 PRINT_DISPOSITION(CLEAN_EFFECTS, "clean_effects");
2234 PRINT_DISPOSITION(ATTACHED_PIC, "attached_pic");
2235 writer_print_section_footer(w);
2236 }
2237
2238 if (do_show_stream_tags)
2239 ret = show_tags(w, stream->metadata, in_program ? SECTION_ID_PROGRAM_STREAM_TAGS : SECTION_ID_STREAM_TAGS);
2240
2241 writer_print_section_footer(w);
2242 av_bprint_finalize(&pbuf, NULL);
2243 fflush(stdout);
2244
2245 return ret;
2246}
2247
2248static int show_streams(WriterContext *w, AVFormatContext *fmt_ctx)
2249{
2250 int i, ret = 0;
2251
2252 writer_print_section_header(w, SECTION_ID_STREAMS);
2253 for (i = 0; i < fmt_ctx->nb_streams; i++)
2254 if (selected_streams[i]) {
2255 ret = show_stream(w, fmt_ctx, i, 0);
2256 if (ret < 0)
2257 break;
2258 }
2259 writer_print_section_footer(w);
2260
2261 return ret;
2262}
2263
2264static int show_program(WriterContext *w, AVFormatContext *fmt_ctx, AVProgram *program)
2265{
2266 int i, ret = 0;
2267
2268 writer_print_section_header(w, SECTION_ID_PROGRAM);
2269 print_int("program_id", program->id);
2270 print_int("program_num", program->program_num);
2271 print_int("nb_streams", program->nb_stream_indexes);
2272 print_int("pmt_pid", program->pmt_pid);
2273 print_int("pcr_pid", program->pcr_pid);
2274 print_ts("start_pts", program->start_time);
2275 print_time("start_time", program->start_time, &AV_TIME_BASE_Q);
2276 print_ts("end_pts", program->end_time);
2277 print_time("end_time", program->end_time, &AV_TIME_BASE_Q);
2278 if (do_show_program_tags)
2279 ret = show_tags(w, program->metadata, SECTION_ID_PROGRAM_TAGS);
2280 if (ret < 0)
2281 goto end;
2282
2283 writer_print_section_header(w, SECTION_ID_PROGRAM_STREAMS);
2284 for (i = 0; i < program->nb_stream_indexes; i++) {
2285 if (selected_streams[program->stream_index[i]]) {
2286 ret = show_stream(w, fmt_ctx, program->stream_index[i], 1);
2287 if (ret < 0)
2288 break;
2289 }
2290 }
2291 writer_print_section_footer(w);
2292
2293end:
2294 writer_print_section_footer(w);
2295 return ret;
2296}
2297
2298static int show_programs(WriterContext *w, AVFormatContext *fmt_ctx)
2299{
2300 int i, ret = 0;
2301
2302 writer_print_section_header(w, SECTION_ID_PROGRAMS);
2303 for (i = 0; i < fmt_ctx->nb_programs; i++) {
2304 AVProgram *program = fmt_ctx->programs[i];
2305 if (!program)
2306 continue;
2307 ret = show_program(w, fmt_ctx, program);
2308 if (ret < 0)
2309 break;
2310 }
2311 writer_print_section_footer(w);
2312 return ret;
2313}
2314
2315static int show_chapters(WriterContext *w, AVFormatContext *fmt_ctx)
2316{
2317 int i, ret = 0;
2318
2319 writer_print_section_header(w, SECTION_ID_CHAPTERS);
2320 for (i = 0; i < fmt_ctx->nb_chapters; i++) {
2321 AVChapter *chapter = fmt_ctx->chapters[i];
2322
2323 writer_print_section_header(w, SECTION_ID_CHAPTER);
2324 print_int("id", chapter->id);
2325 print_q ("time_base", chapter->time_base, '/');
2326 print_int("start", chapter->start);
2327 print_time("start_time", chapter->start, &chapter->time_base);
2328 print_int("end", chapter->end);
2329 print_time("end_time", chapter->end, &chapter->time_base);
2330 if (do_show_chapter_tags)
2331 ret = show_tags(w, chapter->metadata, SECTION_ID_CHAPTER_TAGS);
2332 writer_print_section_footer(w);
2333 }
2334 writer_print_section_footer(w);
2335
2336 return ret;
2337}
2338
2339static int show_format(WriterContext *w, AVFormatContext *fmt_ctx)
2340{
2341 char val_str[128];
2342 int64_t size = fmt_ctx->pb ? avio_size(fmt_ctx->pb) : -1;
2343 int ret = 0;
2344
2345 writer_print_section_header(w, SECTION_ID_FORMAT);
2346 print_str_validate("filename", fmt_ctx->filename);
2347 print_int("nb_streams", fmt_ctx->nb_streams);
2348 print_int("nb_programs", fmt_ctx->nb_programs);
2349 print_str("format_name", fmt_ctx->iformat->name);
2350 if (!do_bitexact) {
2351 if (fmt_ctx->iformat->long_name) print_str ("format_long_name", fmt_ctx->iformat->long_name);
2352 else print_str_opt("format_long_name", "unknown");
2353 }
2354 print_time("start_time", fmt_ctx->start_time, &AV_TIME_BASE_Q);
2355 print_time("duration", fmt_ctx->duration, &AV_TIME_BASE_Q);
2356 if (size >= 0) print_val ("size", size, unit_byte_str);
2357 else print_str_opt("size", "N/A");
2358 if (fmt_ctx->bit_rate > 0) print_val ("bit_rate", fmt_ctx->bit_rate, unit_bit_per_second_str);
2359 else print_str_opt("bit_rate", "N/A");
2360 print_int("probe_score", av_format_get_probe_score(fmt_ctx));
2361 if (do_show_format_tags)
2362 ret = show_tags(w, fmt_ctx->metadata, SECTION_ID_FORMAT_TAGS);
2363
2364 writer_print_section_footer(w);
2365 fflush(stdout);
2366 return ret;
2367}
2368
2369static void show_error(WriterContext *w, int err)
2370{
2371 char errbuf[128];
2372 const char *errbuf_ptr = errbuf;
2373
2374 if (av_strerror(err, errbuf, sizeof(errbuf)) < 0)
2375 errbuf_ptr = strerror(AVUNERROR(err));
2376
2377 writer_print_section_header(w, SECTION_ID_ERROR);
2378 print_int("code", err);
2379 print_str("string", errbuf_ptr);
2380 writer_print_section_footer(w);
2381}
2382
2383static int open_input_file(AVFormatContext **fmt_ctx_ptr, const char *filename)
2384{
2385 int err, i, orig_nb_streams;
2386 AVFormatContext *fmt_ctx = NULL;
2387 AVDictionaryEntry *t;
2388 AVDictionary **opts;
f6fa7814 2389 int scan_all_pmts_set = 0;
2ba45a60 2390
f6fa7814
DM		 2391 if (!av_dict_get(format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE)) {
2392 av_dict_set(&format_opts, "scan_all_pmts", "1", AV_DICT_DONT_OVERWRITE);
2393 scan_all_pmts_set = 1;
2394 }
2ba45a60
DM		 2395 if ((err = avformat_open_input(&fmt_ctx, filename,
2396 iformat, &format_opts)) < 0) {
2397 print_error(filename, err);
2398 return err;
2399 }
f6fa7814
DM		 2400 if (scan_all_pmts_set)
2401 av_dict_set(&format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE);
2ba45a60
DM		 2402 if ((t = av_dict_get(format_opts, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
2403 av_log(NULL, AV_LOG_ERROR, "Option %s not found.\n", t->key);
2404 return AVERROR_OPTION_NOT_FOUND;
2405 }
2406
2407 /* fill the streams in the format context */
2408 opts = setup_find_stream_info_opts(fmt_ctx, codec_opts);
2409 orig_nb_streams = fmt_ctx->nb_streams;
2410
2411 if ((err = avformat_find_stream_info(fmt_ctx, opts)) < 0) {
2412 print_error(filename, err);
2413 return err;
2414 }
2415 for (i = 0; i < orig_nb_streams; i++)
2416 av_dict_free(&opts[i]);
2417 av_freep(&opts);
2418
2419 av_dump_format(fmt_ctx, 0, filename, 0);
2420
2421 /* bind a decoder to each input stream */
2422 for (i = 0; i < fmt_ctx->nb_streams; i++) {
2423 AVStream *stream = fmt_ctx->streams[i];
2424 AVCodec *codec;
2425
2426 if (stream->codec->codec_id == AV_CODEC_ID_PROBE) {
2427 av_log(NULL, AV_LOG_WARNING,
2428 "Failed to probe codec for input stream %d\n",
2429 stream->index);
2430 } else if (!(codec = avcodec_find_decoder(stream->codec->codec_id))) {
2431 av_log(NULL, AV_LOG_WARNING,
2432 "Unsupported codec with id %d for input stream %d\n",
2433 stream->codec->codec_id, stream->index);
2434 } else {
2435 AVDictionary *opts = filter_codec_opts(codec_opts, stream->codec->codec_id,
2436 fmt_ctx, stream, codec);
2437 if (avcodec_open2(stream->codec, codec, &opts) < 0) {
2438 av_log(NULL, AV_LOG_WARNING, "Could not open codec for input stream %d\n",
2439 stream->index);
2440 }
2441 if ((t = av_dict_get(opts, "", NULL, AV_DICT_IGNORE_SUFFIX))) {
2442 av_log(NULL, AV_LOG_ERROR, "Option %s for input stream %d not found\n",
2443 t->key, stream->index);
2444 return AVERROR_OPTION_NOT_FOUND;
2445 }
2446 }
2447 }
2448
2449 *fmt_ctx_ptr = fmt_ctx;
2450 return 0;
2451}
2452
2453static void close_input_file(AVFormatContext **ctx_ptr)
2454{
2455 int i;
2456 AVFormatContext *fmt_ctx = *ctx_ptr;
2457
2458 /* close decoder for each stream */
2459 for (i = 0; i < fmt_ctx->nb_streams; i++)
2460 if (fmt_ctx->streams[i]->codec->codec_id != AV_CODEC_ID_NONE)
2461 avcodec_close(fmt_ctx->streams[i]->codec);
2462
2463 avformat_close_input(ctx_ptr);
2464}
2465
2466static int probe_file(WriterContext *wctx, const char *filename)
2467{
2468 AVFormatContext *fmt_ctx;
2469 int ret, i;
2470 int section_id;
2471
2472 do_read_frames = do_show_frames || do_count_frames;
2473 do_read_packets = do_show_packets || do_count_packets;
2474
2475 ret = open_input_file(&fmt_ctx, filename);
2476 if (ret < 0)
2477 return ret;
2478
2479#define CHECK_END if (ret < 0) goto end
2480
2481 nb_streams = fmt_ctx->nb_streams;
2482 REALLOCZ_ARRAY_STREAM(nb_streams_frames,0,fmt_ctx->nb_streams);
2483 REALLOCZ_ARRAY_STREAM(nb_streams_packets,0,fmt_ctx->nb_streams);
2484 REALLOCZ_ARRAY_STREAM(selected_streams,0,fmt_ctx->nb_streams);
2485
2486 for (i = 0; i < fmt_ctx->nb_streams; i++) {
2487 if (stream_specifier) {
2488 ret = avformat_match_stream_specifier(fmt_ctx,
2489 fmt_ctx->streams[i],
2490 stream_specifier);
2491 CHECK_END;
2492 else
2493 selected_streams[i] = ret;
2494 ret = 0;
2495 } else {
2496 selected_streams[i] = 1;
2497 }
2498 }
2499
2500 if (do_read_frames || do_read_packets) {
2501 if (do_show_frames && do_show_packets &&
2502 wctx->writer->flags & WRITER_FLAG_PUT_PACKETS_AND_FRAMES_IN_SAME_CHAPTER)
2503 section_id = SECTION_ID_PACKETS_AND_FRAMES;
2504 else if (do_show_packets && !do_show_frames)
2505 section_id = SECTION_ID_PACKETS;
2506 else // (!do_show_packets && do_show_frames)
2507 section_id = SECTION_ID_FRAMES;
2508 if (do_show_frames || do_show_packets)
2509 writer_print_section_header(wctx, section_id);
2510 ret = read_packets(wctx, fmt_ctx);
2511 if (do_show_frames || do_show_packets)
2512 writer_print_section_footer(wctx);
2513 CHECK_END;
2514 }
2515
2516 if (do_show_programs) {
2517 ret = show_programs(wctx, fmt_ctx);
2518 CHECK_END;
2519 }
2520
2521 if (do_show_streams) {
2522 ret = show_streams(wctx, fmt_ctx);
2523 CHECK_END;
2524 }
2525 if (do_show_chapters) {
2526 ret = show_chapters(wctx, fmt_ctx);
2527 CHECK_END;
2528 }
2529 if (do_show_format) {
2530 ret = show_format(wctx, fmt_ctx);
2531 CHECK_END;
2532 }
2533
2534end:
2535 close_input_file(&fmt_ctx);
2536 av_freep(&nb_streams_frames);
2537 av_freep(&nb_streams_packets);
2538 av_freep(&selected_streams);
2539
2540 return ret;
2541}
2542
2543static void show_usage(void)
2544{
2545 av_log(NULL, AV_LOG_INFO, "Simple multimedia streams analyzer\n");
2546 av_log(NULL, AV_LOG_INFO, "usage: %s [OPTIONS] [INPUT_FILE]\n", program_name);
2547 av_log(NULL, AV_LOG_INFO, "\n");
2548}
2549
2550static void ffprobe_show_program_version(WriterContext *w)
2551{
2552 AVBPrint pbuf;
2553 av_bprint_init(&pbuf, 1, AV_BPRINT_SIZE_UNLIMITED);
2554
2555 writer_print_section_header(w, SECTION_ID_PROGRAM_VERSION);
2556 print_str("version", FFMPEG_VERSION);
2557 print_fmt("copyright", "Copyright (c) %d-%d the FFmpeg developers",
2558 program_birth_year, CONFIG_THIS_YEAR);
2559 print_str("build_date", __DATE__);
2560 print_str("build_time", __TIME__);
2561 print_str("compiler_ident", CC_IDENT);
2562 print_str("configuration", FFMPEG_CONFIGURATION);
2563 writer_print_section_footer(w);
2564
2565 av_bprint_finalize(&pbuf, NULL);
2566}
2567
2568#define SHOW_LIB_VERSION(libname, LIBNAME) \
2569 do { \
2570 if (CONFIG_##LIBNAME) { \
2571 unsigned int version = libname##_version(); \
2572 writer_print_section_header(w, SECTION_ID_LIBRARY_VERSION); \
2573 print_str("name", "lib" #libname); \
2574 print_int("major", LIB##LIBNAME##_VERSION_MAJOR); \
2575 print_int("minor", LIB##LIBNAME##_VERSION_MINOR); \
2576 print_int("micro", LIB##LIBNAME##_VERSION_MICRO); \
2577 print_int("version", version); \
2578 print_str("ident", LIB##LIBNAME##_IDENT); \
2579 writer_print_section_footer(w); \
2580 } \
2581 } while (0)
2582
2583static void ffprobe_show_library_versions(WriterContext *w)
2584{
2585 writer_print_section_header(w, SECTION_ID_LIBRARY_VERSIONS);
2586 SHOW_LIB_VERSION(avutil, AVUTIL);
2587 SHOW_LIB_VERSION(avcodec, AVCODEC);
2588 SHOW_LIB_VERSION(avformat, AVFORMAT);
2589 SHOW_LIB_VERSION(avdevice, AVDEVICE);
2590 SHOW_LIB_VERSION(avfilter, AVFILTER);
2591 SHOW_LIB_VERSION(swscale, SWSCALE);
2592 SHOW_LIB_VERSION(swresample, SWRESAMPLE);
2593 SHOW_LIB_VERSION(postproc, POSTPROC);
2594 writer_print_section_footer(w);
2595}
2596
f6fa7814
DM		 2597#define PRINT_PIX_FMT_FLAG(flagname, name) \
2598 do { \
2599 print_int(name, !!(pixdesc->flags & AV_PIX_FMT_FLAG_##flagname)); \
2600 } while (0)
2601
2602static void ffprobe_show_pixel_formats(WriterContext *w)
2603{
2604 const AVPixFmtDescriptor *pixdesc = NULL;
2605 int i, n;
2606
2607 writer_print_section_header(w, SECTION_ID_PIXEL_FORMATS);
2608 while (pixdesc = av_pix_fmt_desc_next(pixdesc)) {
2609 writer_print_section_header(w, SECTION_ID_PIXEL_FORMAT);
2610 print_str("name", pixdesc->name);
2611 print_int("nb_components", pixdesc->nb_components);
2612 if ((pixdesc->nb_components >= 3) && !(pixdesc->flags & AV_PIX_FMT_FLAG_RGB)) {
2613 print_int ("log2_chroma_w", pixdesc->log2_chroma_w);
2614 print_int ("log2_chroma_h", pixdesc->log2_chroma_h);
2615 } else {
2616 print_str_opt("log2_chroma_w", "N/A");
2617 print_str_opt("log2_chroma_h", "N/A");
2618 }
2619 n = av_get_bits_per_pixel(pixdesc);
2620 if (n) print_int ("bits_per_pixel", n);
2621 else print_str_opt("bits_per_pixel", "N/A");
2622 if (do_show_pixel_format_flags) {
2623 writer_print_section_header(w, SECTION_ID_PIXEL_FORMAT_FLAGS);
2624 PRINT_PIX_FMT_FLAG(BE, "big_endian");
2625 PRINT_PIX_FMT_FLAG(PAL, "palette");
2626 PRINT_PIX_FMT_FLAG(BITSTREAM, "bitstream");
2627 PRINT_PIX_FMT_FLAG(HWACCEL, "hwaccel");
2628 PRINT_PIX_FMT_FLAG(PLANAR, "planar");
2629 PRINT_PIX_FMT_FLAG(RGB, "rgb");
2630 PRINT_PIX_FMT_FLAG(PSEUDOPAL, "pseudopal");
2631 PRINT_PIX_FMT_FLAG(ALPHA, "alpha");
2632 writer_print_section_footer(w);
2633 }
2634 if (do_show_pixel_format_components && (pixdesc->nb_components > 0)) {
2635 writer_print_section_header(w, SECTION_ID_PIXEL_FORMAT_COMPONENTS);
2636 for (i = 0; i < pixdesc->nb_components; i++) {
2637 writer_print_section_header(w, SECTION_ID_PIXEL_FORMAT_COMPONENT);
2638 print_int("index", i + 1);
2639 print_int("bit_depth", pixdesc->comp[i].depth_minus1 + 1);
2640 writer_print_section_footer(w);
2641 }
2642 writer_print_section_footer(w);
2643 }
2644 writer_print_section_footer(w);
2645 }
2646 writer_print_section_footer(w);
2647}
2648
2ba45a60
DM		 2649static int opt_format(void *optctx, const char *opt, const char *arg)
2650{
2651 iformat = av_find_input_format(arg);
2652 if (!iformat) {
2653 av_log(NULL, AV_LOG_ERROR, "Unknown input format: %s\n", arg);
2654 return AVERROR(EINVAL);
2655 }
2656 return 0;
2657}
2658
2659static inline void mark_section_show_entries(SectionID section_id,
2660 int show_all_entries, AVDictionary *entries)
2661{
2662 struct section *section = &sections[section_id];
2663
2664 section->show_all_entries = show_all_entries;
2665 if (show_all_entries) {
2666 SectionID *id;
2667 for (id = section->children_ids; *id != -1; id++)
2668 mark_section_show_entries(*id, show_all_entries, entries);
2669 } else {
2670 av_dict_copy(&section->entries_to_show, entries, 0);
2671 }
2672}
2673
2674static int match_section(const char *section_name,
2675 int show_all_entries, AVDictionary *entries)
2676{
2677 int i, ret = 0;
2678
2679 for (i = 0; i < FF_ARRAY_ELEMS(sections); i++) {
2680 const struct section *section = &sections[i];
2681 if (!strcmp(section_name, section->name) ||
2682 (section->unique_name && !strcmp(section_name, section->unique_name))) {
2683 av_log(NULL, AV_LOG_DEBUG,
2684 "'%s' matches section with unique name '%s'\n", section_name,
2685 (char *)av_x_if_null(section->unique_name, section->name));
2686 ret++;
2687 mark_section_show_entries(section->id, show_all_entries, entries);
2688 }
2689 }
2690 return ret;
2691}
2692
2693static int opt_show_entries(void *optctx, const char *opt, const char *arg)
2694{
2695 const char *p = arg;
2696 int ret = 0;
2697
2698 while (*p) {
2699 AVDictionary *entries = NULL;
2700 char *section_name = av_get_token(&p, "=:");
2701 int show_all_entries = 0;
2702
2703 if (!section_name) {
2704 av_log(NULL, AV_LOG_ERROR,
2705 "Missing section name for option '%s'\n", opt);
2706 return AVERROR(EINVAL);
2707 }
2708
2709 if (*p == '=') {
2710 p++;
2711 while (*p && *p != ':') {
2712 char *entry = av_get_token(&p, ",:");
2713 if (!entry)
2714 break;
2715 av_log(NULL, AV_LOG_VERBOSE,
2716 "Adding '%s' to the entries to show in section '%s'\n",
2717 entry, section_name);
2718 av_dict_set(&entries, entry, "", AV_DICT_DONT_STRDUP_KEY);
2719 if (*p == ',')
2720 p++;
2721 }
2722 } else {
2723 show_all_entries = 1;
2724 }
2725
2726 ret = match_section(section_name, show_all_entries, entries);
2727 if (ret == 0) {
2728 av_log(NULL, AV_LOG_ERROR, "No match for section '%s'\n", section_name);
2729 ret = AVERROR(EINVAL);
2730 }
2731 av_dict_free(&entries);
2732 av_free(section_name);
2733
2734 if (ret <= 0)
2735 break;
2736 if (*p)
2737 p++;
2738 }
2739
2740 return ret;
2741}
2742
2743static int opt_show_format_entry(void *optctx, const char *opt, const char *arg)
2744{
2745 char *buf = av_asprintf("format=%s", arg);
2746 int ret;
2747
2748 av_log(NULL, AV_LOG_WARNING,
2749 "Option '%s' is deprecated, use '-show_entries format=%s' instead\n",
2750 opt, arg);
2751 ret = opt_show_entries(optctx, opt, buf);
2752 av_free(buf);
2753 return ret;
2754}
2755
2756static void opt_input_file(void *optctx, const char *arg)
2757{
2758 if (input_filename) {
2759 av_log(NULL, AV_LOG_ERROR,
2760 "Argument '%s' provided as input filename, but '%s' was already specified.\n",
2761 arg, input_filename);
2762 exit_program(1);
2763 }
2764 if (!strcmp(arg, "-"))
2765 arg = "pipe:";
2766 input_filename = arg;
2767}
2768
2769static int opt_input_file_i(void *optctx, const char *opt, const char *arg)
2770{
2771 opt_input_file(optctx, arg);
2772 return 0;
2773}
2774
2775void show_help_default(const char *opt, const char *arg)
2776{
2777 av_log_set_callback(log_callback_help);
2778 show_usage();
2779 show_help_options(options, "Main options:", 0, 0, 0);
2780 printf("\n");
2781
2782 show_help_children(avformat_get_class(), AV_OPT_FLAG_DECODING_PARAM);
2783}
2784
2785/**
2786 * Parse interval specification, according to the format:
2787 * INTERVAL ::= [START|+START_OFFSET][%[END|+END_OFFSET]]
2788 * INTERVALS ::= INTERVAL[,INTERVALS]
2789*/
2790static int parse_read_interval(const char *interval_spec,
2791 ReadInterval *interval)
2792{
2793 int ret = 0;
2794 char *next, *p, *spec = av_strdup(interval_spec);
2795 if (!spec)
2796 return AVERROR(ENOMEM);
2797
2798 if (!*spec) {
2799 av_log(NULL, AV_LOG_ERROR, "Invalid empty interval specification\n");
2800 ret = AVERROR(EINVAL);
2801 goto end;
2802 }
2803
2804 p = spec;
2805 next = strchr(spec, '%');
2806 if (next)
2807 *next++ = 0;
2808
2809 /* parse first part */
2810 if (*p) {
2811 interval->has_start = 1;
2812
2813 if (*p == '+') {
2814 interval->start_is_offset = 1;
2815 p++;
2816 } else {
2817 interval->start_is_offset = 0;
2818 }
2819
2820 ret = av_parse_time(&interval->start, p, 1);
2821 if (ret < 0) {
2822 av_log(NULL, AV_LOG_ERROR, "Invalid interval start specification '%s'\n", p);
2823 goto end;
2824 }
2825 } else {
2826 interval->has_start = 0;
2827 }
2828
2829 /* parse second part */
2830 p = next;
2831 if (p && *p) {
2832 int64_t us;
2833 interval->has_end = 1;
2834
2835 if (*p == '+') {
2836 interval->end_is_offset = 1;
2837 p++;
2838 } else {
2839 interval->end_is_offset = 0;
2840 }
2841
2842 if (interval->end_is_offset && *p == '#') {
2843 long long int lli;
2844 char *tail;
2845 interval->duration_frames = 1;
2846 p++;
2847 lli = strtoll(p, &tail, 10);
2848 if (*tail || lli < 0) {
2849 av_log(NULL, AV_LOG_ERROR,
2850 "Invalid or negative value '%s' for duration number of frames\n", p);
2851 goto end;
2852 }
2853 interval->end = lli;
2854 } else {
2855 ret = av_parse_time(&us, p, 1);
2856 if (ret < 0) {
2857 av_log(NULL, AV_LOG_ERROR, "Invalid interval end/duration specification '%s'\n", p);
2858 goto end;
2859 }
2860 interval->end = us;
2861 }
2862 } else {
2863 interval->has_end = 0;
2864 }
2865
2866end:
2867 av_free(spec);
2868 return ret;
2869}
2870
2871static int parse_read_intervals(const char *intervals_spec)
2872{
2873 int ret, n, i;
2874 char *p, *spec = av_strdup(intervals_spec);
2875 if (!spec)
2876 return AVERROR(ENOMEM);
2877
2878 /* preparse specification, get number of intervals */
2879 for (n = 0, p = spec; *p; p++)
2880 if (*p == ',')
2881 n++;
2882 n++;
2883
2884 read_intervals = av_malloc_array(n, sizeof(*read_intervals));
2885 if (!read_intervals) {
2886 ret = AVERROR(ENOMEM);
2887 goto end;
2888 }
2889 read_intervals_nb = n;
2890
2891 /* parse intervals */
2892 p = spec;
2893 for (i = 0; p; i++) {
2894 char *next;
2895
2896 av_assert0(i < read_intervals_nb);
2897 next = strchr(p, ',');
2898 if (next)
2899 *next++ = 0;
2900
2901 read_intervals[i].id = i;
2902 ret = parse_read_interval(p, &read_intervals[i]);
2903 if (ret < 0) {
2904 av_log(NULL, AV_LOG_ERROR, "Error parsing read interval #%d '%s'\n",
2905 i, p);
2906 goto end;
2907 }
2908 av_log(NULL, AV_LOG_VERBOSE, "Parsed log interval ");
2909 log_read_interval(&read_intervals[i], NULL, AV_LOG_VERBOSE);
2910 p = next;
2911 }
2912 av_assert0(i == read_intervals_nb);
2913
2914end:
2915 av_free(spec);
2916 return ret;
2917}
2918
2919static int opt_read_intervals(void *optctx, const char *opt, const char *arg)
2920{
2921 return parse_read_intervals(arg);
2922}
2923
2924static int opt_pretty(void *optctx, const char *opt, const char *arg)
2925{
2926 show_value_unit = 1;
2927 use_value_prefix = 1;
2928 use_byte_value_binary_prefix = 1;
2929 use_value_sexagesimal_format = 1;
2930 return 0;
2931}
2932
2933static void print_section(SectionID id, int level)
2934{
2935 const SectionID *pid;
2936 const struct section *section = &sections[id];
2937 printf("%c%c%c",
2938 section->flags & SECTION_FLAG_IS_WRAPPER ? 'W' : '.',
2939 section->flags & SECTION_FLAG_IS_ARRAY ? 'A' : '.',
2940 section->flags & SECTION_FLAG_HAS_VARIABLE_FIELDS ? 'V' : '.');
2941 printf("%*c %s", level * 4, ' ', section->name);
2942 if (section->unique_name)
2943 printf("/%s", section->unique_name);
2944 printf("\n");
2945
2946 for (pid = section->children_ids; *pid != -1; pid++)
2947 print_section(*pid, level+1);
2948}
2949
2950static int opt_sections(void *optctx, const char *opt, const char *arg)
2951{
2952 printf("Sections:\n"
2953 "W.. = Section is a wrapper (contains other sections, no local entries)\n"
2954 ".A. = Section contains an array of elements of the same type\n"
2955 "..V = Section may contain a variable number of fields with variable keys\n"
2956 "FLAGS NAME/UNIQUE_NAME\n"
2957 "---\n");
2958 print_section(SECTION_ID_ROOT, 0);
2959 return 0;
2960}
2961
2962static int opt_show_versions(const char *opt, const char *arg)
2963{
2964 mark_section_show_entries(SECTION_ID_PROGRAM_VERSION, 1, NULL);
2965 mark_section_show_entries(SECTION_ID_LIBRARY_VERSION, 1, NULL);
2966 return 0;
2967}
2968
2969#define DEFINE_OPT_SHOW_SECTION(section, target_section_id) \
2970 static int opt_show_##section(const char *opt, const char *arg) \
2971 { \
2972 mark_section_show_entries(SECTION_ID_##target_section_id, 1, NULL); \
2973 return 0; \
2974 }
2975
2976DEFINE_OPT_SHOW_SECTION(chapters, CHAPTERS);
2977DEFINE_OPT_SHOW_SECTION(error, ERROR);
2978DEFINE_OPT_SHOW_SECTION(format, FORMAT);
2979DEFINE_OPT_SHOW_SECTION(frames, FRAMES);
2980DEFINE_OPT_SHOW_SECTION(library_versions, LIBRARY_VERSIONS);
2981DEFINE_OPT_SHOW_SECTION(packets, PACKETS);
f6fa7814 2982DEFINE_OPT_SHOW_SECTION(pixel_formats, PIXEL_FORMATS);
2ba45a60
DM		 2983DEFINE_OPT_SHOW_SECTION(program_version, PROGRAM_VERSION);
2984DEFINE_OPT_SHOW_SECTION(streams, STREAMS);
2985DEFINE_OPT_SHOW_SECTION(programs, PROGRAMS);
2986
2987static const OptionDef real_options[] = {
2988#include "cmdutils_common_opts.h"
2989 { "f", HAS_ARG, {.func_arg = opt_format}, "force format", "format" },
2990 { "unit", OPT_BOOL, {&show_value_unit}, "show unit of the displayed values" },
2991 { "prefix", OPT_BOOL, {&use_value_prefix}, "use SI prefixes for the displayed values" },
2992 { "byte_binary_prefix", OPT_BOOL, {&use_byte_value_binary_prefix},
2993 "use binary prefixes for byte units" },
2994 { "sexagesimal", OPT_BOOL, {&use_value_sexagesimal_format},
2995 "use sexagesimal format HOURS:MM:SS.MICROSECONDS for time units" },
2996 { "pretty", 0, {.func_arg = opt_pretty},
2997 "prettify the format of displayed values, make it more human readable" },
2998 { "print_format", OPT_STRING | HAS_ARG, {(void*)&print_format},
2999 "set the output printing format (available formats are: default, compact, csv, flat, ini, json, xml)", "format" },
3000 { "of", OPT_STRING | HAS_ARG, {(void*)&print_format}, "alias for -print_format", "format" },
3001 { "select_streams", OPT_STRING | HAS_ARG, {(void*)&stream_specifier}, "select the specified streams", "stream_specifier" },
3002 { "sections", OPT_EXIT, {.func_arg = opt_sections}, "print sections structure and section information, and exit" },
3003 { "show_data", OPT_BOOL, {(void*)&do_show_data}, "show packets data" },
3004 { "show_data_hash", OPT_STRING | HAS_ARG, {(void*)&show_data_hash}, "show packets data hash" },
3005 { "show_error", 0, {(void*)&opt_show_error}, "show probing error" },
3006 { "show_format", 0, {(void*)&opt_show_format}, "show format/container info" },
3007 { "show_frames", 0, {(void*)&opt_show_frames}, "show frames info" },
3008 { "show_format_entry", HAS_ARG, {.func_arg = opt_show_format_entry},
3009 "show a particular entry from the format/container info", "entry" },
3010 { "show_entries", HAS_ARG, {.func_arg = opt_show_entries},
3011 "show a set of specified entries", "entry_list" },
3012 { "show_packets", 0, {(void*)&opt_show_packets}, "show packets info" },
3013 { "show_programs", 0, {(void*)&opt_show_programs}, "show programs info" },
3014 { "show_streams", 0, {(void*)&opt_show_streams}, "show streams info" },
3015 { "show_chapters", 0, {(void*)&opt_show_chapters}, "show chapters info" },
3016 { "count_frames", OPT_BOOL, {(void*)&do_count_frames}, "count the number of frames per stream" },
3017 { "count_packets", OPT_BOOL, {(void*)&do_count_packets}, "count the number of packets per stream" },
3018 { "show_program_version", 0, {(void*)&opt_show_program_version}, "show ffprobe version" },
3019 { "show_library_versions", 0, {(void*)&opt_show_library_versions}, "show library versions" },
3020 { "show_versions", 0, {(void*)&opt_show_versions}, "show program and library versions" },
f6fa7814 3021 { "show_pixel_formats", 0, {(void*)&opt_show_pixel_formats}, "show pixel format descriptions" },
2ba45a60
DM		 3022 { "show_private_data", OPT_BOOL, {(void*)&show_private_data}, "show private data" },
3023 { "private", OPT_BOOL, {(void*)&show_private_data}, "same as show_private_data" },
3024 { "bitexact", OPT_BOOL, {&do_bitexact}, "force bitexact output" },
3025 { "read_intervals", HAS_ARG, {.func_arg = opt_read_intervals}, "set read intervals", "read_intervals" },
3026 { "default", HAS_ARG | OPT_AUDIO | OPT_VIDEO | OPT_EXPERT, {.func_arg = opt_default}, "generic catch all option", "" },
3027 { "i", HAS_ARG, {.func_arg = opt_input_file_i}, "read specified file", "input_file"},
3028 { NULL, },
3029};
3030
3031static inline int check_section_show_entries(int section_id)
3032{
3033 int *id;
3034 struct section *section = &sections[section_id];
3035 if (sections[section_id].show_all_entries || sections[section_id].entries_to_show)
3036 return 1;
3037 for (id = section->children_ids; *id != -1; id++)
3038 if (check_section_show_entries(*id))
3039 return 1;
3040 return 0;
3041}
3042
3043#define SET_DO_SHOW(id, varname) do { \
3044 if (check_section_show_entries(SECTION_ID_##id)) \
3045 do_show_##varname = 1; \
3046 } while (0)
3047
3048int main(int argc, char **argv)
3049{
3050 const Writer *w;
3051 WriterContext *wctx;
3052 char *buf;
3053 char *w_name = NULL, *w_args = NULL;
3054 int ret, i;
3055
3056 av_log_set_flags(AV_LOG_SKIP_REPEATED);
3057 register_exit(ffprobe_cleanup);
3058
3059 options = real_options;
3060 parse_loglevel(argc, argv, options);
3061 av_register_all();
3062 avformat_network_init();
3063 init_opts();
3064#if CONFIG_AVDEVICE
3065 avdevice_register_all();
3066#endif
3067
3068 show_banner(argc, argv, options);
3069 parse_options(NULL, argc, argv, options, opt_input_file);
3070
3071 /* mark things to show, based on -show_entries */
3072 SET_DO_SHOW(CHAPTERS, chapters);
3073 SET_DO_SHOW(ERROR, error);
3074 SET_DO_SHOW(FORMAT, format);
3075 SET_DO_SHOW(FRAMES, frames);
3076 SET_DO_SHOW(LIBRARY_VERSIONS, library_versions);
3077 SET_DO_SHOW(PACKETS, packets);
f6fa7814
DM		 3078 SET_DO_SHOW(PIXEL_FORMATS, pixel_formats);
3079 SET_DO_SHOW(PIXEL_FORMAT_FLAGS, pixel_format_flags);
3080 SET_DO_SHOW(PIXEL_FORMAT_COMPONENTS, pixel_format_components);
2ba45a60
DM		 3081 SET_DO_SHOW(PROGRAM_VERSION, program_version);
3082 SET_DO_SHOW(PROGRAMS, programs);
3083 SET_DO_SHOW(STREAMS, streams);
3084 SET_DO_SHOW(STREAM_DISPOSITION, stream_disposition);
3085 SET_DO_SHOW(PROGRAM_STREAM_DISPOSITION, stream_disposition);
3086
3087 SET_DO_SHOW(CHAPTER_TAGS, chapter_tags);
3088 SET_DO_SHOW(FORMAT_TAGS, format_tags);
3089 SET_DO_SHOW(FRAME_TAGS, frame_tags);
3090 SET_DO_SHOW(PROGRAM_TAGS, program_tags);
3091 SET_DO_SHOW(STREAM_TAGS, stream_tags);
3092
3093 if (do_bitexact && (do_show_program_version || do_show_library_versions)) {
3094 av_log(NULL, AV_LOG_ERROR,
3095 "-bitexact and -show_program_version or -show_library_versions "
3096 "options are incompatible\n");
3097 ret = AVERROR(EINVAL);
3098 goto end;
3099 }
3100
3101 writer_register_all();
3102
3103 if (!print_format)
3104 print_format = av_strdup("default");
3105 if (!print_format) {
3106 ret = AVERROR(ENOMEM);
3107 goto end;
3108 }
3109 w_name = av_strtok(print_format, "=", &buf);
3110 w_args = buf;
3111
3112 if (show_data_hash) {
3113 if ((ret = av_hash_alloc(&hash, show_data_hash)) < 0) {
3114 if (ret == AVERROR(EINVAL)) {
3115 const char *n;
3116 av_log(NULL, AV_LOG_ERROR,
3117 "Unknown hash algorithm '%s'\nKnown algorithms:",
3118 show_data_hash);
3119 for (i = 0; (n = av_hash_names(i)); i++)
3120 av_log(NULL, AV_LOG_ERROR, " %s", n);
3121 av_log(NULL, AV_LOG_ERROR, "\n");
3122 }
3123 goto end;
3124 }
3125 }
3126
3127 w = writer_get_by_name(w_name);
3128 if (!w) {
3129 av_log(NULL, AV_LOG_ERROR, "Unknown output format with name '%s'\n", w_name);
3130 ret = AVERROR(EINVAL);
3131 goto end;
3132 }
3133
3134 if ((ret = writer_open(&wctx, w, w_args,
3135 sections, FF_ARRAY_ELEMS(sections))) >= 0) {
3136 if (w == &xml_writer)
3137 wctx->string_validation_utf8_flags |= AV_UTF8_FLAG_EXCLUDE_XML_INVALID_CONTROL_CODES;
3138
3139 writer_print_section_header(wctx, SECTION_ID_ROOT);
3140
3141 if (do_show_program_version)
3142 ffprobe_show_program_version(wctx);
3143 if (do_show_library_versions)
3144 ffprobe_show_library_versions(wctx);
f6fa7814
DM		 3145 if (do_show_pixel_formats)
3146 ffprobe_show_pixel_formats(wctx);
2ba45a60
DM		 3147
3148 if (!input_filename &&
3149 ((do_show_format || do_show_programs || do_show_streams || do_show_chapters || do_show_packets || do_show_error) ||
f6fa7814 3150 (!do_show_program_version && !do_show_library_versions && !do_show_pixel_formats))) {
2ba45a60
DM		 3151 show_usage();
3152 av_log(NULL, AV_LOG_ERROR, "You have to specify one input file.\n");
3153 av_log(NULL, AV_LOG_ERROR, "Use -h to get full help or, even better, run 'man %s'.\n", program_name);
3154 ret = AVERROR(EINVAL);
3155 } else if (input_filename) {
3156 ret = probe_file(wctx, input_filename);
3157 if (ret < 0 && do_show_error)
3158 show_error(wctx, ret);
3159 }
3160
3161 writer_print_section_footer(wctx);
3162 writer_close(&wctx);
3163 }
3164
3165end:
3166 av_freep(&print_format);
3167 av_freep(&read_intervals);
3168 av_hash_freep(&hash);
3169
3170 uninit_opts();
3171 for (i = 0; i < FF_ARRAY_ELEMS(sections); i++)
3172 av_dict_free(&(sections[i].entries_to_show));
3173
3174 avformat_network_deinit();
3175
3176 return ret < 0;
3177}
ffmpeg Piment Noir Debian package repository