FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
ffprobe.c
Go to the documentation of this file.
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 
50 const char program_name[] = "ffprobe";
51 const int program_birth_year = 2007;
52 
53 static int do_bitexact = 0;
54 static int do_count_frames = 0;
55 static int do_count_packets = 0;
56 static int do_read_frames = 0;
57 static int do_read_packets = 0;
58 static int do_show_chapters = 0;
59 static int do_show_error = 0;
60 static int do_show_format = 0;
61 static int do_show_frames = 0;
62 static int do_show_packets = 0;
63 static int do_show_programs = 0;
64 static int do_show_streams = 0;
65 static int do_show_stream_disposition = 0;
66 static int do_show_data = 0;
67 static int do_show_program_version = 0;
68 static int do_show_library_versions = 0;
69 static int do_show_pixel_formats = 0;
70 static int do_show_pixel_format_flags = 0;
71 static int do_show_pixel_format_components = 0;
72 
73 static int do_show_chapter_tags = 0;
74 static int do_show_format_tags = 0;
75 static int do_show_frame_tags = 0;
76 static int do_show_program_tags = 0;
77 static int do_show_stream_tags = 0;
78 
79 static int show_value_unit = 0;
80 static int use_value_prefix = 0;
81 static int use_byte_value_binary_prefix = 0;
82 static int use_value_sexagesimal_format = 0;
83 static int show_private_data = 1;
84 
85 static char *print_format;
86 static char *stream_specifier;
87 static char *show_data_hash;
88 
89 typedef 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 
97 static ReadInterval *read_intervals;
98 static int read_intervals_nb = 0;
99 
100 /* section structure definition */
101 
102 #define SECTION_MAX_NB_CHILDREN 10
103 
104 struct 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 
120 typedef 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,
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,
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 
159 static 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 } },
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 } },
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,
191  SECTION_ID_PACKETS, SECTION_ID_ERROR, SECTION_ID_PROGRAM_VERSION, SECTION_ID_LIBRARY_VERSIONS,
192  SECTION_ID_PIXEL_FORMATS, -1} },
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 
200 static const OptionDef *options;
201 
202 /* FFprobe context */
203 static const char *input_filename;
204 static AVInputFormat *iformat = NULL;
205 
206 static struct AVHashContext *hash;
207 
208 static const char *const binary_unit_prefixes [] = { "", "Ki", "Mi", "Gi", "Ti", "Pi" };
209 static const char *const decimal_unit_prefixes[] = { "", "K" , "M" , "G" , "T" , "P" };
210 
211 static const char unit_second_str[] = "s" ;
212 static const char unit_hertz_str[] = "Hz" ;
213 static const char unit_byte_str[] = "byte" ;
214 static const char unit_bit_per_second_str[] = "bit/s";
215 
216 static int nb_streams;
217 static uint64_t *nb_streams_packets;
218 static uint64_t *nb_streams_frames;
219 static int *selected_streams;
220 
221 static 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 
228 struct unit_value {
229  union { double d; long long int i; } val;
230  const char *unit;
231 };
232 
233 static 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 
288 typedef 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 
293 typedef enum {
294  WRITER_STRING_VALIDATION_FAIL,
295  WRITER_STRING_VALIDATION_REPLACE,
296  WRITER_STRING_VALIDATION_IGNORE,
297  WRITER_STRING_VALIDATION_NB
298 } StringValidation;
299 
300 typedef 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 
318 struct 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 
346 static 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 
354 static 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 
367 static 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 
375 static 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 
383 static 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 
401 static 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 
410 static 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 
492 fail:
493  writer_close(wctx);
494  return ret;
495 }
496 
497 static 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 
521 static 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 
538 static 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 
549 static 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 
599 end:
600  av_bprint_finalize(&dstbuf, dstp);
601  return ret;
602 }
603 
604 #define PRINT_STRING_OPT 1
605 #define PRINT_STRING_VALIDATE 2
606 
607 static 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 
643 static 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 
652 static 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 
669 static 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 
678 static 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 
706 static 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 
723 static const Writer *registered_writers[MAX_REGISTERED_WRITERS_NB + 1];
724 
725 static 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 
736 static 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) \
751 static const char *name##_get_name(void *ctx) \
752 { \
753  return #name ; \
754 } \
755 static const AVClass name##_class = { \
756  .class_name = #name, \
757  .item_name = name##_get_name, \
758  .option = name##_options \
759 }
760 
761 /* Default output */
762 
763 typedef 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 
773 static 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 
781 DEFINE_WRITER_CLASS(default);
782 
783 /* lame uppercasing routine, assumes the string is lower case ASCII */
784 static 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 
793 static 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 
818 static 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 
831 static 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 
840 static 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 
849 static 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  */
865 static 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  */
888 static 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 
906 static const char *none_escape_str(AVBPrint *dst, const char *src, const char sep, void *log_ctx)
907 {
908  return src;
909 }
910 
911 typedef 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 
927 static 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 
939 DEFINE_WRITER_CLASS(compact);
940 
941 static 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 
963 static 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 
993 static 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 
1003 static 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 
1016 static 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 
1026 static 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 
1043 static 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 
1055 DEFINE_WRITER_CLASS(csv);
1056 
1057 static 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 
1071 typedef 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 
1081 static 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 
1089 DEFINE_WRITER_CLASS(flat);
1090 
1091 static 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 
1105 static 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 
1120 static 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 
1138 static 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 
1164 static 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 
1169 static 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 
1182 static 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 
1195 typedef struct {
1196  const AVClass *class;
1197  int hierarchical;
1198 } INIContext;
1199 
1200 #undef OFFSET
1201 #define OFFSET(x) offsetof(INIContext, x)
1202 
1203 static 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 
1209 DEFINE_WRITER_CLASS(ini);
1210 
1211 static 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 
1238 static 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 
1271 static 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 
1282 static void ini_print_int(WriterContext *wctx, const char *key, long long int value)
1283 {
1284  printf("%s=%lld\n", key, value);
1285 }
1286 
1287 static 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 
1299 typedef 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 
1309 static 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 
1315 DEFINE_WRITER_CLASS(json);
1316 
1317 static 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 
1327 static 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 
1349 static 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 
1387 static 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 
1409 static 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 
1421 static 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 
1432 static 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 
1447 static 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 
1461 typedef 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 
1472 static 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 
1480 DEFINE_WRITER_CLASS(xml);
1481 
1482 static 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 
1510 static 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 
1530 static 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 
1570 static 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 
1589 static 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 
1612 static 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 
1619 static 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 
1631 static 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 
1681 static 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 
1699 static 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 
1735 static 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 
1758 static 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 
1847 static 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 
1890 static 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 
1914 static 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 
2010 end:
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 
2019 static 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 
2038 static 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)
2115  print_str ("color_range", av_color_range_name(dec_ctx->color_range));
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");
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 
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 
2248 static 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 
2264 static 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 
2293 end:
2294  writer_print_section_footer(w);
2295  return ret;
2296 }
2297 
2298 static 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 
2315 static 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 
2339 static 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 
2369 static 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 
2383 static 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;
2389  int scan_all_pmts_set = 0;
2390 
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  }
2395  if ((err = avformat_open_input(&fmt_ctx, filename,
2396  iformat, &format_opts)) < 0) {
2397  print_error(filename, err);
2398  return err;
2399  }
2400  if (scan_all_pmts_set)
2401  av_dict_set(&format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE);
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 
2453 static 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 
2466 static 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 
2534 end:
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 
2543 static 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 
2550 static 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 
2583 static 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 
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 
2602 static 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 
2649 static 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 
2659 static 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 
2674 static 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 
2693 static 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 
2743 static 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 
2756 static 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 
2769 static 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 
2775 void 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 */
2790 static 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 
2866 end:
2867  av_free(spec);
2868  return ret;
2869 }
2870 
2871 static 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 
2914 end:
2915  av_free(spec);
2916  return ret;
2917 }
2918 
2919 static int opt_read_intervals(void *optctx, const char *opt, const char *arg)
2920 {
2921  return parse_read_intervals(arg);
2922 }
2923 
2924 static 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 
2933 static 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 
2950 static 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 
2962 static 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 
2976 DEFINE_OPT_SHOW_SECTION(chapters, CHAPTERS);
2977 DEFINE_OPT_SHOW_SECTION(error, ERROR);
2978 DEFINE_OPT_SHOW_SECTION(format, FORMAT);
2979 DEFINE_OPT_SHOW_SECTION(frames, FRAMES);
2980 DEFINE_OPT_SHOW_SECTION(library_versions, LIBRARY_VERSIONS);
2981 DEFINE_OPT_SHOW_SECTION(packets, PACKETS);
2982 DEFINE_OPT_SHOW_SECTION(pixel_formats, PIXEL_FORMATS);
2983 DEFINE_OPT_SHOW_SECTION(program_version, PROGRAM_VERSION);
2984 DEFINE_OPT_SHOW_SECTION(streams, STREAMS);
2985 DEFINE_OPT_SHOW_SECTION(programs, PROGRAMS);
2986 
2987 static 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" },
3021  { "show_pixel_formats", 0, {(void*)&opt_show_pixel_formats}, "show pixel format descriptions" },
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 
3031 static 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 
3048 int 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);
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);
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);
3145  if (do_show_pixel_formats)
3146  ffprobe_show_pixel_formats(wctx);
3147 
3148  if (!input_filename &&
3149  ((do_show_format || do_show_programs || do_show_streams || do_show_chapters || do_show_packets || do_show_error) ||
3150  (!do_show_program_version && !do_show_library_versions && !do_show_pixel_formats))) {
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 
3165 end:
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 }
Generated on Fri Dec 5 2014 04:41:48 for FFmpeg by   doxygen 1.8.2