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