X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=sidebyside;f=ffprobe.c;h=ee3c899ce82279ed4b7a3411186f6bb1c4aa2f41;hb=15b219fae9da1691dfb264f51637805e1ca63d1a;hp=52f07e72ff3ecf7cfd54cd693757994ef63dd57d;hpb=0b9a69f244e399565d67100a6862886201a594a4;p=ffmpeg diff --git a/ffprobe.c b/ffprobe.c index 52f07e72ff3..ee3c899ce82 100644 --- a/ffprobe.c +++ b/ffprobe.c @@ -1,5 +1,4 @@ /* - * ffprobe : Simple Media Prober based on the FFmpeg libraries * Copyright (c) 2007-2010 Stefano Sabatini * * This file is part of FFmpeg. @@ -19,6 +18,11 @@ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +/** + * @file + * simple media prober based on the FFmpeg libraries + */ + #include "config.h" #include "libavformat/avformat.h" @@ -33,6 +37,7 @@ const char program_name[] = "ffprobe"; const int program_birth_year = 2007; +static int do_show_error = 0; static int do_show_format = 0; static int do_show_packets = 0; static int do_show_streams = 0; @@ -41,6 +46,7 @@ static int show_value_unit = 0; static int use_value_prefix = 0; static int use_byte_value_binary_prefix = 0; static int use_value_sexagesimal_format = 0; +static int show_private_data = 1; static char *print_format; @@ -862,6 +868,248 @@ static const Writer json_writer = { .show_tags = json_show_tags, }; +/* XML output */ + +typedef struct { + const AVClass *class; + int within_tag; + int multiple_entries; ///< tells if the given chapter requires multiple entries + int indent_level; + int fully_qualified; + int xsd_strict; + char *buf; + size_t buf_size; +} XMLContext; + +#undef OFFSET +#define OFFSET(x) offsetof(XMLContext, x) + +static const AVOption xml_options[] = { + {"fully_qualified", "specify if the output should be fully qualified", OFFSET(fully_qualified), AV_OPT_TYPE_INT, {.dbl=0}, 0, 1 }, + {"q", "specify if the output should be fully qualified", OFFSET(fully_qualified), AV_OPT_TYPE_INT, {.dbl=0}, 0, 1 }, + {"xsd_strict", "ensure that the output is XSD compliant", OFFSET(xsd_strict), AV_OPT_TYPE_INT, {.dbl=0}, 0, 1 }, + {"x", "ensure that the output is XSD compliant", OFFSET(xsd_strict), AV_OPT_TYPE_INT, {.dbl=0}, 0, 1 }, + {NULL}, +}; + +static const char *xml_get_name(void *ctx) +{ + return "xml"; +} + +static const AVClass xml_class = { + "XMLContext", + xml_get_name, + xml_options +}; + +static av_cold int xml_init(WriterContext *wctx, const char *args, void *opaque) +{ + XMLContext *xml = wctx->priv; + int err; + + xml->class = &xml_class; + av_opt_set_defaults(xml); + + if (args && + (err = (av_set_options_string(xml, args, "=", ":"))) < 0) { + av_log(wctx, AV_LOG_ERROR, "Error parsing options string: '%s'\n", args); + return err; + } + + if (xml->xsd_strict) { + xml->fully_qualified = 1; +#define CHECK_COMPLIANCE(opt, opt_name) \ + if (opt) { \ + av_log(wctx, AV_LOG_ERROR, \ + "XSD-compliant output selected but option '%s' was selected, XML output may be non-compliant.\n" \ + "You need to disable such option with '-no%s'\n", opt_name, opt_name); \ + } + CHECK_COMPLIANCE(show_private_data, "private"); + CHECK_COMPLIANCE(show_value_unit, "unit"); + CHECK_COMPLIANCE(use_value_prefix, "prefix"); + } + + xml->buf_size = ESCAPE_INIT_BUF_SIZE; + if (!(xml->buf = av_malloc(xml->buf_size))) + return AVERROR(ENOMEM); + return 0; +} + +static av_cold void xml_uninit(WriterContext *wctx) +{ + XMLContext *xml = wctx->priv; + av_freep(&xml->buf); +} + +static const char *xml_escape_str(char **dst, size_t *dst_size, const char *src, + void *log_ctx) +{ + const char *p; + char *q; + size_t size = 1; + + /* precompute size */ + for (p = src; *p; p++, size++) { + ESCAPE_CHECK_SIZE(src, size, SIZE_MAX-10); + switch (*p) { + case '&' : size += strlen("&"); break; + case '<' : size += strlen("<"); break; + case '>' : size += strlen(">"); break; + case '\"': size += strlen("""); break; + case '\'': size += strlen("'"); break; + default: size++; + } + } + ESCAPE_REALLOC_BUF(dst_size, dst, src, size); + +#define COPY_STR(str) { \ + const char *s = str; \ + while (*s) \ + *q++ = *s++; \ + } + + p = src; + q = *dst; + while (*p) { + switch (*p) { + case '&' : COPY_STR("&"); break; + case '<' : COPY_STR("<"); break; + case '>' : COPY_STR(">"); break; + case '\"': COPY_STR("""); break; + case '\'': COPY_STR("'"); break; + default: *q++ = *p; + } + p++; + } + *q = 0; + + return *dst; +} + +static void xml_print_header(WriterContext *wctx) +{ + XMLContext *xml = wctx->priv; + const char *qual = " xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' " + "xmlns:ffprobe='http://www.ffmpeg.org/schema/ffprobe' " + "xsi:schemaLocation='http://www.ffmpeg.org/schema/ffprobe ffprobe.xsd'"; + + printf("\n"); + printf("<%sffprobe%s>\n", + xml->fully_qualified ? "ffprobe:" : "", + xml->fully_qualified ? qual : ""); + + xml->indent_level++; +} + +static void xml_print_footer(WriterContext *wctx) +{ + XMLContext *xml = wctx->priv; + + xml->indent_level--; + printf("\n", xml->fully_qualified ? "ffprobe:" : ""); +} + +#define XML_INDENT() { int i; for (i = 0; i < xml->indent_level; i++) printf(INDENT); } + +static void xml_print_chapter_header(WriterContext *wctx, const char *chapter) +{ + XMLContext *xml = wctx->priv; + + if (wctx->nb_chapter) + printf("\n"); + xml->multiple_entries = !strcmp(chapter, "packets") || !strcmp(chapter, "streams"); + + if (xml->multiple_entries) { + XML_INDENT(); printf("<%s>\n", chapter); + xml->indent_level++; + } +} + +static void xml_print_chapter_footer(WriterContext *wctx, const char *chapter) +{ + XMLContext *xml = wctx->priv; + + if (xml->multiple_entries) { + xml->indent_level--; + XML_INDENT(); printf("\n", chapter); + } +} + +static void xml_print_section_header(WriterContext *wctx, const char *section) +{ + XMLContext *xml = wctx->priv; + + XML_INDENT(); printf("<%s ", section); + xml->within_tag = 1; +} + +static void xml_print_section_footer(WriterContext *wctx, const char *section) +{ + XMLContext *xml = wctx->priv; + + if (xml->within_tag) + printf("/>\n"); + else { + XML_INDENT(); printf("\n", section); + } +} + +static void xml_print_str(WriterContext *wctx, const char *key, const char *value) +{ + XMLContext *xml = wctx->priv; + + if (wctx->nb_item) + printf(" "); + printf("%s=\"%s\"", key, xml_escape_str(&xml->buf, &xml->buf_size, value, wctx)); +} + +static void xml_print_int(WriterContext *wctx, const char *key, long long int value) +{ + if (wctx->nb_item) + printf(" "); + printf("%s=\"%lld\"", key, value); +} + +static void xml_show_tags(WriterContext *wctx, AVDictionary *dict) +{ + XMLContext *xml = wctx->priv; + AVDictionaryEntry *tag = NULL; + int is_first = 1; + + xml->indent_level++; + while ((tag = av_dict_get(dict, "", tag, AV_DICT_IGNORE_SUFFIX))) { + if (is_first) { + /* close section tag */ + printf(">\n"); + xml->within_tag = 0; + is_first = 0; + } + XML_INDENT(); + printf("buf, &xml->buf_size, tag->key, wctx)); + printf(" value=\"%s\"/>\n", + xml_escape_str(&xml->buf, &xml->buf_size, tag->value, wctx)); + } + xml->indent_level--; +} + +static Writer xml_writer = { + .name = "xml", + .priv_size = sizeof(XMLContext), + .init = xml_init, + .uninit = xml_uninit, + .print_header = xml_print_header, + .print_footer = xml_print_footer, + .print_chapter_header = xml_print_chapter_header, + .print_chapter_footer = xml_print_chapter_footer, + .print_section_header = xml_print_section_header, + .print_section_footer = xml_print_section_footer, + .print_integer = xml_print_int, + .print_string = xml_print_str, + .show_tags = xml_show_tags, +}; + static void writer_register_all(void) { static int initialized; @@ -874,6 +1122,7 @@ static void writer_register_all(void) writer_register(&compact_writer); writer_register(&csv_writer); writer_register(&json_writer); + writer_register(&xml_writer); } #define print_fmt(k, f, ...) do { \ @@ -892,7 +1141,7 @@ static void writer_register_all(void) #define print_time(k, v, tb) writer_print_time(w, k, v, tb) #define print_ts(k, v) writer_print_ts(w, k, v) #define print_val(k, v, u) writer_print_string(w, k, \ - value_string(val_str, sizeof(val_str), (struct unit_value){.val.i = v, .unit=u}), 1) + value_string(val_str, sizeof(val_str), (struct unit_value){.val.i = v, .unit=u}), 0) #define print_section_header(s) writer_print_section_header(w, s) #define print_section_footer(s) writer_print_section_footer(w, s) #define show_tags(metadata) writer_show_tags(w, metadata) @@ -993,6 +1242,17 @@ static void show_stream(WriterContext *w, AVFormatContext *fmt_ctx, int stream_i if (s) print_str ("pix_fmt", s); else print_str_opt("pix_fmt", "unknown"); print_int("level", dec_ctx->level); + if (dec_ctx->timecode_frame_start >= 0) { + uint32_t tc = dec_ctx->timecode_frame_start; + print_fmt("timecode", "%02d:%02d:%02d%c%02d", + tc>>19 & 0x1f, // hours + tc>>13 & 0x3f, // minutes + tc>>6 & 0x3f, // seconds + tc & 1<<24 ? ';' : ':', // drop + tc & 0x3f); // frames + } else { + print_str_opt("timecode", "N/A"); + } break; case AVMEDIA_TYPE_AUDIO: @@ -1007,7 +1267,7 @@ static void show_stream(WriterContext *w, AVFormatContext *fmt_ctx, int stream_i } else { print_str_opt("codec_type", "unknown"); } - if (dec_ctx->codec && dec_ctx->codec->priv_class) { + if (dec_ctx->codec && dec_ctx->codec->priv_class && show_private_data) { const AVOption *opt = NULL; while (opt = av_opt_next(dec_ctx->priv_data,opt)) { uint8_t *str; @@ -1046,7 +1306,6 @@ static void show_format(WriterContext *w, AVFormatContext *fmt_ctx) { char val_str[128]; int64_t size = avio_size(fmt_ctx->pb); - struct print_buf pbuf = {.s = NULL}; print_section_header("format"); print_str("filename", fmt_ctx->filename); @@ -1061,10 +1320,25 @@ static void show_format(WriterContext *w, AVFormatContext *fmt_ctx) else print_str_opt("bit_rate", "N/A"); show_tags(fmt_ctx->metadata); print_section_footer("format"); - av_free(pbuf.s); fflush(stdout); } +static void show_error(WriterContext *w, int err) +{ + char errbuf[128]; + const char *errbuf_ptr = errbuf; + + if (av_strerror(err, errbuf, sizeof(errbuf)) < 0) + errbuf_ptr = strerror(AVUNERROR(err)); + + writer_print_chapter_header(w, "error"); + print_section_header("error"); + print_int("code", err); + print_str("string", errbuf_ptr); + print_section_footer("error"); + writer_print_chapter_footer(w, "error"); +} + static int open_input_file(AVFormatContext **fmt_ctx_ptr, const char *filename) { int err, i; @@ -1095,11 +1369,11 @@ static int open_input_file(AVFormatContext **fmt_ctx_ptr, const char *filename) AVCodec *codec; if (!(codec = avcodec_find_decoder(stream->codec->codec_id))) { - fprintf(stderr, "Unsupported codec with id %d for input stream %d\n", - stream->codec->codec_id, stream->index); + av_log(NULL, AV_LOG_ERROR, "Unsupported codec with id %d for input stream %d\n", + stream->codec->codec_id, stream->index); } else if (avcodec_open2(stream->codec, codec, NULL) < 0) { - fprintf(stderr, "Error while opening codec for input stream %d\n", - stream->index); + av_log(NULL, AV_LOG_ERROR, "Error while opening codec for input stream %d\n", + stream->index); } } @@ -1140,16 +1414,18 @@ static int probe_file(const char *filename) if ((ret = writer_open(&wctx, w, w_args, NULL)) < 0) goto end; - if ((ret = open_input_file(&fmt_ctx, filename))) - goto end; writer_print_header(wctx); - PRINT_CHAPTER(packets); - PRINT_CHAPTER(streams); - PRINT_CHAPTER(format); + ret = open_input_file(&fmt_ctx, filename); + if (ret >= 0) { + PRINT_CHAPTER(packets); + PRINT_CHAPTER(streams); + PRINT_CHAPTER(format); + avformat_close_input(&fmt_ctx); + } else if (do_show_error) { + show_error(wctx, ret); + } writer_print_footer(wctx); - - av_close_input_file(fmt_ctx); writer_close(&wctx); end: @@ -1169,7 +1445,7 @@ static int opt_format(const char *opt, const char *arg) { iformat = av_find_input_format(arg); if (!iformat) { - fprintf(stderr, "Unknown input format: %s\n", arg); + av_log(NULL, AV_LOG_ERROR, "Unknown input format: %s\n", arg); return AVERROR(EINVAL); } return 0; @@ -1178,8 +1454,8 @@ static int opt_format(const char *opt, const char *arg) static void opt_input_file(void *optctx, const char *arg) { if (input_filename) { - fprintf(stderr, "Argument '%s' provided as input filename, but '%s' was already specified.\n", - arg, input_filename); + av_log(NULL, AV_LOG_ERROR, "Argument '%s' provided as input filename, but '%s' was already specified.\n", + arg, input_filename); exit(1); } if (!strcmp(arg, "-")) @@ -1220,10 +1496,13 @@ static const OptionDef options[] = { { "pretty", 0, {(void*)&opt_pretty}, "prettify the format of displayed values, make it more human readable" }, { "print_format", OPT_STRING | HAS_ARG, {(void*)&print_format}, - "set the output printing format (available formats are: default, compact, csv, json)", "format" }, + "set the output printing format (available formats are: default, compact, csv, json, xml)", "format" }, + { "show_error", OPT_BOOL, {(void*)&do_show_error} , "show probing error" }, { "show_format", OPT_BOOL, {(void*)&do_show_format} , "show format/container info" }, { "show_packets", OPT_BOOL, {(void*)&do_show_packets}, "show packets info" }, { "show_streams", OPT_BOOL, {(void*)&do_show_streams}, "show streams info" }, + { "show_private_data", OPT_BOOL, {(void*)&show_private_data}, "show private data" }, + { "private", OPT_BOOL, {(void*)&show_private_data}, "same as show_private_data" }, { "default", HAS_ARG | OPT_AUDIO | OPT_VIDEO | OPT_EXPERT, {(void*)opt_default}, "generic catch all option", "" }, { "i", HAS_ARG, {(void *)opt_input_file}, "read specified file", "input_file"}, { NULL, }, @@ -1241,13 +1520,13 @@ int main(int argc, char **argv) avdevice_register_all(); #endif - show_banner(); + show_banner(argc, argv, options); parse_options(NULL, argc, argv, options, opt_input_file); if (!input_filename) { show_usage(); - fprintf(stderr, "You have to specify one input file.\n"); - fprintf(stderr, "Use -h to get full help or, even better, run 'man %s'.\n", program_name); + av_log(NULL, AV_LOG_ERROR, "You have to specify one input file.\n"); + av_log(NULL, AV_LOG_ERROR, "Use -h to get full help or, even better, run 'man %s'.\n", program_name); exit(1); }