cmdutils.c

   1 /*
   2  * Various utilities for command line tools
   3  * Copyright (c) 2000-2003 Fabrice Bellard
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 #include <string.h>
  23 #include <stdint.h>
  24 #include <stdlib.h>
  25 #include <errno.h>
  26 #include <math.h>
  27
  28 /* Include only the enabled headers since some compilers (namely, Sun
  29    Studio) will not omit unused inline functions and create undefined
  30    references to libraries that are not being built. */
  31
  32 #include "config.h"
  33 #include "compat/va_copy.h"
  34 #include "libavformat/avformat.h"
  35 #include "libavfilter/avfilter.h"
  36 #include "libavdevice/avdevice.h"
  37 #include "libavresample/avresample.h"
  38 #include "libswscale/swscale.h"
  39 #include "libswresample/swresample.h"
  40 #include "libpostproc/postprocess.h"
  41 #include "libavutil/avassert.h"
  42 #include "libavutil/avstring.h"
  43 #include "libavutil/bprint.h"
  44 #include "libavutil/mathematics.h"
  45 #include "libavutil/imgutils.h"
  46 #include "libavutil/parseutils.h"
  47 #include "libavutil/pixdesc.h"
  48 #include "libavutil/eval.h"
  49 #include "libavutil/dict.h"
  50 #include "libavutil/opt.h"
  51 #include "libavutil/cpu.h"
  52 #include "libavutil/ffversion.h"
  53 #include "cmdutils.h"
  54 #if CONFIG_NETWORK
  55 #include "libavformat/network.h"
  56 #endif
  57 #if HAVE_SYS_RESOURCE_H
  58 #include <sys/time.h>
  59 #include <sys/resource.h>
  60 #endif
  61
  62 static int init_report(const char *env);
  63
  64 struct SwsContext *sws_opts;
  65 AVDictionary *swr_opts;
  66 AVDictionary *format_opts, *codec_opts, *resample_opts;
  67
  68 static FILE *report_file;
  69 static int report_file_level = AV_LOG_DEBUG;
  70 int hide_banner = 0;
  71
  72 void init_opts(void)
  73 {
  74
  75     if(CONFIG_SWSCALE)
  76         sws_opts = sws_getContext(16, 16, 0, 16, 16, 0, SWS_BICUBIC,
  77                               NULL, NULL, NULL);
  78 }
  79
  80 void uninit_opts(void)
  81 {
  82 #if CONFIG_SWSCALE
  83     sws_freeContext(sws_opts);
  84     sws_opts = NULL;
  85 #endif
  86
  87     av_dict_free(&swr_opts);
  88     av_dict_free(&format_opts);
  89     av_dict_free(&codec_opts);
  90     av_dict_free(&resample_opts);
  91 }
  92
  93 void log_callback_help(void *ptr, int level, const char *fmt, va_list vl)
  94 {
  95     vfprintf(stdout, fmt, vl);
  96 }
  97
  98 static void log_callback_report(void *ptr, int level, const char *fmt, va_list vl)
  99 {
 100     va_list vl2;
 101     char line[1024];
 102     static int print_prefix = 1;
 103
 104     va_copy(vl2, vl);
 105     av_log_default_callback(ptr, level, fmt, vl);
 106     av_log_format_line(ptr, level, fmt, vl2, line, sizeof(line), &print_prefix);
 107     va_end(vl2);
 108     if (report_file_level >= level) {
 109         fputs(line, report_file);
 110         fflush(report_file);
 111     }
 112 }
 113
 114 static void (*program_exit)(int ret);
 115
 116 void register_exit(void (*cb)(int ret))
 117 {
 118     program_exit = cb;
 119 }
 120
 121 void exit_program(int ret)
 122 {
 123     if (program_exit)
 124         program_exit(ret);
 125
 126     exit(ret);
 127 }
 128
 129 double parse_number_or_die(const char *context, const char *numstr, int type,
 130                            double min, double max)
 131 {
 132     char *tail;
 133     const char *error;
 134     double d = av_strtod(numstr, &tail);
 135     if (*tail)
 136         error = "Expected number for %s but found: %s\n";
 137     else if (d < min || d > max)
 138         error = "The value for %s was %s which is not within %f - %f\n";
 139     else if (type == OPT_INT64 && (int64_t)d != d)
 140         error = "Expected int64 for %s but found %s\n";
 141     else if (type == OPT_INT && (int)d != d)
 142         error = "Expected int for %s but found %s\n";
 143     else
 144         return d;
 145     av_log(NULL, AV_LOG_FATAL, error, context, numstr, min, max);
 146     exit_program(1);
 147     return 0;
 148 }
 149
 150 int64_t parse_time_or_die(const char *context, const char *timestr,
 151                           int is_duration)
 152 {
 153     int64_t us;
 154     if (av_parse_time(&us, timestr, is_duration) < 0) {
 155         av_log(NULL, AV_LOG_FATAL, "Invalid %s specification for %s: %s\n",
 156                is_duration ? "duration" : "date", context, timestr);
 157         exit_program(1);
 158     }
 159     return us;
 160 }
 161
 162 void show_help_options(const OptionDef *options, const char *msg, int req_flags,
 163                        int rej_flags, int alt_flags)
 164 {
 165     const OptionDef *po;
 166     int first;
 167
 168     first = 1;
 169     for (po = options; po->name; po++) {
 170         char buf[64];
 171
 172         if (((po->flags & req_flags) != req_flags) ||
 173             (alt_flags && !(po->flags & alt_flags)) ||
 174             (po->flags & rej_flags))
 175             continue;
 176
 177         if (first) {
 178             printf("%s\n", msg);
 179             first = 0;
 180         }
 181         av_strlcpy(buf, po->name, sizeof(buf));
 182         if (po->argname) {
 183             av_strlcat(buf, " ", sizeof(buf));
 184             av_strlcat(buf, po->argname, sizeof(buf));
 185         }
 186         printf("-%-17s  %s\n", buf, po->help);
 187     }
 188     printf("\n");
 189 }
 190
 191 void show_help_children(const AVClass *class, int flags)
 192 {
 193     const AVClass *child = NULL;
 194     if (class->option) {
 195         av_opt_show2(&class, NULL, flags, 0);
 196         printf("\n");
 197     }
 198
 199     while (child = av_opt_child_class_next(class, child))
 200         show_help_children(child, flags);
 201 }
 202
 203 static const OptionDef *find_option(const OptionDef *po, const char *name)
 204 {
 205     const char *p = strchr(name, ':');
 206     int len = p ? p - name : strlen(name);
 207
 208     while (po->name) {
 209         if (!strncmp(name, po->name, len) && strlen(po->name) == len)
 210             break;
 211         po++;
 212     }
 213     return po;
 214 }
 215
 216 /* _WIN32 means using the windows libc - cygwin doesn't define that
 217  * by default. HAVE_COMMANDLINETOARGVW is true on cygwin, while
 218  * it doesn't provide the actual command line via GetCommandLineW(). */
 219 #if HAVE_COMMANDLINETOARGVW && defined(_WIN32)
 220 #include <windows.h>
 221 #include <shellapi.h>
 222 /* Will be leaked on exit */
 223 static char** win32_argv_utf8 = NULL;
 224 static int win32_argc = 0;
 225
 226 /**
 227  * Prepare command line arguments for executable.
 228  * For Windows - perform wide-char to UTF-8 conversion.
 229  * Input arguments should be main() function arguments.
 230  * @param argc_ptr Arguments number (including executable)
 231  * @param argv_ptr Arguments list.
 232  */
 233 static void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
 234 {
 235     char *argstr_flat;
 236     wchar_t **argv_w;
 237     int i, buffsize = 0, offset = 0;
 238
 239     if (win32_argv_utf8) {
 240         *argc_ptr = win32_argc;
 241         *argv_ptr = win32_argv_utf8;
 242         return;
 243     }
 244
 245     win32_argc = 0;
 246     argv_w = CommandLineToArgvW(GetCommandLineW(), &win32_argc);
 247     if (win32_argc <= 0 || !argv_w)
 248         return;
 249
 250     /* determine the UTF-8 buffer size (including NULL-termination symbols) */
 251     for (i = 0; i < win32_argc; i++)
 252         buffsize += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
 253                                         NULL, 0, NULL, NULL);
 254
 255     win32_argv_utf8 = av_mallocz(sizeof(char *) * (win32_argc + 1) + buffsize);
 256     argstr_flat     = (char *)win32_argv_utf8 + sizeof(char *) * (win32_argc + 1);
 257     if (!win32_argv_utf8) {
 258         LocalFree(argv_w);
 259         return;
 260     }
 261
 262     for (i = 0; i < win32_argc; i++) {
 263         win32_argv_utf8[i] = &argstr_flat[offset];
 264         offset += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
 265                                       &argstr_flat[offset],
 266                                       buffsize - offset, NULL, NULL);
 267     }
 268     win32_argv_utf8[i] = NULL;
 269     LocalFree(argv_w);
 270
 271     *argc_ptr = win32_argc;
 272     *argv_ptr = win32_argv_utf8;
 273 }
 274 #else
 275 static inline void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
 276 {
 277     /* nothing to do */
 278 }
 279 #endif /* HAVE_COMMANDLINETOARGVW */
 280
 281 static int write_option(void *optctx, const OptionDef *po, const char *opt,
 282                         const char *arg)
 283 {
 284     /* new-style options contain an offset into optctx, old-style address of
 285      * a global var*/
 286     void *dst = po->flags & (OPT_OFFSET | OPT_SPEC) ?
 287                 (uint8_t *)optctx + po->u.off : po->u.dst_ptr;
 288     int *dstcount;
 289
 290     if (po->flags & OPT_SPEC) {
 291         SpecifierOpt **so = dst;
 292         char *p = strchr(opt, ':');
 293         char *str;
 294
 295         dstcount = (int *)(so + 1);
 296         *so = grow_array(*so, sizeof(**so), dstcount, *dstcount + 1);
 297         str = av_strdup(p ? p + 1 : "");
 298         if (!str)
 299             return AVERROR(ENOMEM);
 300         (*so)[*dstcount - 1].specifier = str;
 301         dst = &(*so)[*dstcount - 1].u;
 302     }
 303
 304     if (po->flags & OPT_STRING) {
 305         char *str;
 306         str = av_strdup(arg);
 307         av_freep(dst);
 308         if (!str)
 309             return AVERROR(ENOMEM);
 310         *(char **)dst = str;
 311     } else if (po->flags & OPT_BOOL || po->flags & OPT_INT) {
 312         *(int *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);
 313     } else if (po->flags & OPT_INT64) {
 314         *(int64_t *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT64_MIN, INT64_MAX);
 315     } else if (po->flags & OPT_TIME) {
 316         *(int64_t *)dst = parse_time_or_die(opt, arg, 1);
 317     } else if (po->flags & OPT_FLOAT) {
 318         *(float *)dst = parse_number_or_die(opt, arg, OPT_FLOAT, -INFINITY, INFINITY);
 319     } else if (po->flags & OPT_DOUBLE) {
 320         *(double *)dst = parse_number_or_die(opt, arg, OPT_DOUBLE, -INFINITY, INFINITY);
 321     } else if (po->u.func_arg) {
 322         int ret = po->u.func_arg(optctx, opt, arg);
 323         if (ret < 0) {
 324             av_log(NULL, AV_LOG_ERROR,
 325                    "Failed to set value '%s' for option '%s': %s\n",
 326                    arg, opt, av_err2str(ret));
 327             return ret;
 328         }
 329     }
 330     if (po->flags & OPT_EXIT)
 331         exit_program(0);
 332
 333     return 0;
 334 }
 335
 336 int parse_option(void *optctx, const char *opt, const char *arg,
 337                  const OptionDef *options)
 338 {
 339     const OptionDef *po;
 340     int ret;
 341
 342     po = find_option(options, opt);
 343     if (!po->name && opt[0] == 'n' && opt[1] == 'o') {
 344         /* handle 'no' bool option */
 345         po = find_option(options, opt + 2);
 346         if ((po->name && (po->flags & OPT_BOOL)))
 347             arg = "0";
 348     } else if (po->flags & OPT_BOOL)
 349         arg = "1";
 350
 351     if (!po->name)
 352         po = find_option(options, "default");
 353     if (!po->name) {
 354         av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'\n", opt);
 355         return AVERROR(EINVAL);
 356     }
 357     if (po->flags & HAS_ARG && !arg) {
 358         av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'\n", opt);
 359         return AVERROR(EINVAL);
 360     }
 361
 362     ret = write_option(optctx, po, opt, arg);
 363     if (ret < 0)
 364         return ret;
 365
 366     return !!(po->flags & HAS_ARG);
 367 }
 368
 369 void parse_options(void *optctx, int argc, char **argv, const OptionDef *options,
 370                    void (*parse_arg_function)(void *, const char*))
 371 {
 372     const char *opt;
 373     int optindex, handleoptions = 1, ret;
 374
 375     /* perform system-dependent conversions for arguments list */
 376     prepare_app_arguments(&argc, &argv);
 377
 378     /* parse options */
 379     optindex = 1;
 380     while (optindex < argc) {
 381         opt = argv[optindex++];
 382
 383         if (handleoptions && opt[0] == '-' && opt[1] != '\0') {
 384             if (opt[1] == '-' && opt[2] == '\0') {
 385                 handleoptions = 0;
 386                 continue;
 387             }
 388             opt++;
 389
 390             if ((ret = parse_option(optctx, opt, argv[optindex], options)) < 0)
 391                 exit_program(1);
 392             optindex += ret;
 393         } else {
 394             if (parse_arg_function)
 395                 parse_arg_function(optctx, opt);
 396         }
 397     }
 398 }
 399
 400 int parse_optgroup(void *optctx, OptionGroup *g)
 401 {
 402     int i, ret;
 403
 404     av_log(NULL, AV_LOG_DEBUG, "Parsing a group of options: %s %s.\n",
 405            g->group_def->name, g->arg);
 406
 407     for (i = 0; i < g->nb_opts; i++) {
 408         Option *o = &g->opts[i];
 409
 410         if (g->group_def->flags &&
 411             !(g->group_def->flags & o->opt->flags)) {
 412             av_log(NULL, AV_LOG_ERROR, "Option %s (%s) cannot be applied to "
 413                    "%s %s -- you are trying to apply an input option to an "
 414                    "output file or vice versa. Move this option before the "
 415                    "file it belongs to.\n", o->key, o->opt->help,
 416                    g->group_def->name, g->arg);
 417             return AVERROR(EINVAL);
 418         }
 419
 420         av_log(NULL, AV_LOG_DEBUG, "Applying option %s (%s) with argument %s.\n",
 421                o->key, o->opt->help, o->val);
 422
 423         ret = write_option(optctx, o->opt, o->key, o->val);
 424         if (ret < 0)
 425             return ret;
 426     }
 427
 428     av_log(NULL, AV_LOG_DEBUG, "Successfully parsed a group of options.\n");
 429
 430     return 0;
 431 }
 432
 433 int locate_option(int argc, char **argv, const OptionDef *options,
 434                   const char *optname)
 435 {
 436     const OptionDef *po;
 437     int i;
 438
 439     for (i = 1; i < argc; i++) {
 440         const char *cur_opt = argv[i];
 441
 442         if (*cur_opt++ != '-')
 443             continue;
 444
 445         po = find_option(options, cur_opt);
 446         if (!po->name && cur_opt[0] == 'n' && cur_opt[1] == 'o')
 447             po = find_option(options, cur_opt + 2);
 448
 449         if ((!po->name && !strcmp(cur_opt, optname)) ||
 450              (po->name && !strcmp(optname, po->name)))
 451             return i;
 452
 453         if (!po->name || po->flags & HAS_ARG)
 454             i++;
 455     }
 456     return 0;
 457 }
 458
 459 static void dump_argument(const char *a)
 460 {
 461     const unsigned char *p;
 462
 463     for (p = a; *p; p++)
 464         if (!((*p >= '+' && *p <= ':') || (*p >= '@' && *p <= 'Z') ||
 465               *p == '_' || (*p >= 'a' && *p <= 'z')))
 466             break;
 467     if (!*p) {
 468         fputs(a, report_file);
 469         return;
 470     }
 471     fputc('"', report_file);
 472     for (p = a; *p; p++) {
 473         if (*p == '\\' || *p == '"' || *p == '$' || *p == '`')
 474             fprintf(report_file, "\\%c", *p);
 475         else if (*p < ' ' || *p > '~')
 476             fprintf(report_file, "\\x%02x", *p);
 477         else
 478             fputc(*p, report_file);
 479     }
 480     fputc('"', report_file);
 481 }
 482
 483 void parse_loglevel(int argc, char **argv, const OptionDef *options)
 484 {
 485     int idx = locate_option(argc, argv, options, "loglevel");
 486     const char *env;
 487     if (!idx)
 488         idx = locate_option(argc, argv, options, "v");
 489     if (idx && argv[idx + 1])
 490         opt_loglevel(NULL, "loglevel", argv[idx + 1]);
 491     idx = locate_option(argc, argv, options, "report");
 492     if ((env = getenv("FFREPORT")) || idx) {
 493         init_report(env);
 494         if (report_file) {
 495             int i;
 496             fprintf(report_file, "Command line:\n");
 497             for (i = 0; i < argc; i++) {
 498                 dump_argument(argv[i]);
 499                 fputc(i < argc - 1 ? ' ' : '\n', report_file);
 500             }
 501             fflush(report_file);
 502         }
 503     }
 504     idx = locate_option(argc, argv, options, "hide_banner");
 505     if (idx)
 506         hide_banner = 1;
 507 }
 508
 509 static const AVOption *opt_find(void *obj, const char *name, const char *unit,
 510                             int opt_flags, int search_flags)
 511 {
 512     const AVOption *o = av_opt_find(obj, name, unit, opt_flags, search_flags);
 513     if(o && !o->flags)
 514         return NULL;
 515     return o;
 516 }
 517
 518 #define FLAGS (o->type == AV_OPT_TYPE_FLAGS) ? AV_DICT_APPEND : 0
 519 int opt_default(void *optctx, const char *opt, const char *arg)
 520 {
 521     const AVOption *o;
 522     int consumed = 0;
 523     char opt_stripped[128];
 524     const char *p;
 525     const AVClass *cc = avcodec_get_class(), *fc = avformat_get_class();
 526 #if CONFIG_AVRESAMPLE
 527     const AVClass *rc = avresample_get_class();
 528 #endif
 529     const AVClass *sc, *swr_class;
 530
 531     if (!strcmp(opt, "debug") || !strcmp(opt, "fdebug"))
 532         av_log_set_level(AV_LOG_DEBUG);
 533
 534     if (!(p = strchr(opt, ':')))
 535         p = opt + strlen(opt);
 536     av_strlcpy(opt_stripped, opt, FFMIN(sizeof(opt_stripped), p - opt + 1));
 537
 538     if ((o = opt_find(&cc, opt_stripped, NULL, 0,
 539                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) ||
 540         ((opt[0] == 'v' || opt[0] == 'a' || opt[0] == 's') &&
 541          (o = opt_find(&cc, opt + 1, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ)))) {
 542         av_dict_set(&codec_opts, opt, arg, FLAGS);
 543         consumed = 1;
 544     }
 545     if ((o = opt_find(&fc, opt, NULL, 0,
 546                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
 547         av_dict_set(&format_opts, opt, arg, FLAGS);
 548         if (consumed)
 549             av_log(NULL, AV_LOG_VERBOSE, "Routing option %s to both codec and muxer layer\n", opt);
 550         consumed = 1;
 551     }
 552 #if CONFIG_SWSCALE
 553     sc = sws_get_class();
 554     if (!consumed && opt_find(&sc, opt, NULL, 0,
 555                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) {
 556         // XXX we only support sws_flags, not arbitrary sws options
 557         int ret = av_opt_set(sws_opts, opt, arg, 0);
 558         if (ret < 0) {
 559             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
 560             return ret;
 561         }
 562         consumed = 1;
 563     }
 564 #else
 565     if (!consumed && !strcmp(opt, "sws_flags")) {
 566         av_log(NULL, AV_LOG_WARNING, "Ignoring %s %s, due to disabled swscale\n", opt, arg);
 567         consumed = 1;
 568     }
 569 #endif
 570 #if CONFIG_SWRESAMPLE
 571     swr_class = swr_get_class();
 572     if (!consumed && (o=opt_find(&swr_class, opt, NULL, 0,
 573                                     AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
 574         struct SwrContext *swr = swr_alloc();
 575         int ret = av_opt_set(swr, opt, arg, 0);
 576         swr_free(&swr);
 577         if (ret < 0) {
 578             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
 579             return ret;
 580         }
 581         av_dict_set(&swr_opts, opt, arg, FLAGS);
 582         consumed = 1;
 583     }
 584 #endif
 585 #if CONFIG_AVRESAMPLE
 586     if ((o=opt_find(&rc, opt, NULL, 0,
 587                        AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
 588         av_dict_set(&resample_opts, opt, arg, FLAGS);
 589         consumed = 1;
 590     }
 591 #endif
 592
 593     if (consumed)
 594         return 0;
 595     return AVERROR_OPTION_NOT_FOUND;
 596 }
 597
 598 /*
 599  * Check whether given option is a group separator.
 600  *
 601  * @return index of the group definition that matched or -1 if none
 602  */
 603 static int match_group_separator(const OptionGroupDef *groups, int nb_groups,
 604                                  const char *opt)
 605 {
 606     int i;
 607
 608     for (i = 0; i < nb_groups; i++) {
 609         const OptionGroupDef *p = &groups[i];
 610         if (p->sep && !strcmp(p->sep, opt))
 611             return i;
 612     }
 613
 614     return -1;
 615 }
 616
 617 /*
 618  * Finish parsing an option group.
 619  *
 620  * @param group_idx which group definition should this group belong to
 621  * @param arg argument of the group delimiting option
 622  */
 623 static void finish_group(OptionParseContext *octx, int group_idx,
 624                          const char *arg)
 625 {
 626     OptionGroupList *l = &octx->groups[group_idx];
 627     OptionGroup *g;
 628
 629     GROW_ARRAY(l->groups, l->nb_groups);
 630     g = &l->groups[l->nb_groups - 1];
 631
 632     *g             = octx->cur_group;
 633     g->arg         = arg;
 634     g->group_def   = l->group_def;
 635 #if CONFIG_SWSCALE
 636     g->sws_opts    = sws_opts;
 637 #endif
 638     g->swr_opts    = swr_opts;
 639     g->codec_opts  = codec_opts;
 640     g->format_opts = format_opts;
 641     g->resample_opts = resample_opts;
 642
 643     codec_opts  = NULL;
 644     format_opts = NULL;
 645     resample_opts = NULL;
 646 #if CONFIG_SWSCALE
 647     sws_opts    = NULL;
 648 #endif
 649     swr_opts    = NULL;
 650     init_opts();
 651
 652     memset(&octx->cur_group, 0, sizeof(octx->cur_group));
 653 }
 654
 655 /*
 656  * Add an option instance to currently parsed group.
 657  */
 658 static void add_opt(OptionParseContext *octx, const OptionDef *opt,
 659                     const char *key, const char *val)
 660 {
 661     int global = !(opt->flags & (OPT_PERFILE | OPT_SPEC | OPT_OFFSET));
 662     OptionGroup *g = global ? &octx->global_opts : &octx->cur_group;
 663
 664     GROW_ARRAY(g->opts, g->nb_opts);
 665     g->opts[g->nb_opts - 1].opt = opt;
 666     g->opts[g->nb_opts - 1].key = key;
 667     g->opts[g->nb_opts - 1].val = val;
 668 }
 669
 670 static void init_parse_context(OptionParseContext *octx,
 671                                const OptionGroupDef *groups, int nb_groups)
 672 {
 673     static const OptionGroupDef global_group = { "global" };
 674     int i;
 675
 676     memset(octx, 0, sizeof(*octx));
 677
 678     octx->nb_groups = nb_groups;
 679     octx->groups    = av_mallocz_array(octx->nb_groups, sizeof(*octx->groups));
 680     if (!octx->groups)
 681         exit_program(1);
 682
 683     for (i = 0; i < octx->nb_groups; i++)
 684         octx->groups[i].group_def = &groups[i];
 685
 686     octx->global_opts.group_def = &global_group;
 687     octx->global_opts.arg       = "";
 688
 689     init_opts();
 690 }
 691
 692 void uninit_parse_context(OptionParseContext *octx)
 693 {
 694     int i, j;
 695
 696     for (i = 0; i < octx->nb_groups; i++) {
 697         OptionGroupList *l = &octx->groups[i];
 698
 699         for (j = 0; j < l->nb_groups; j++) {
 700             av_freep(&l->groups[j].opts);
 701             av_dict_free(&l->groups[j].codec_opts);
 702             av_dict_free(&l->groups[j].format_opts);
 703             av_dict_free(&l->groups[j].resample_opts);
 704 #if CONFIG_SWSCALE
 705             sws_freeContext(l->groups[j].sws_opts);
 706 #endif
 707             av_dict_free(&l->groups[j].swr_opts);
 708         }
 709         av_freep(&l->groups);
 710     }
 711     av_freep(&octx->groups);
 712
 713     av_freep(&octx->cur_group.opts);
 714     av_freep(&octx->global_opts.opts);
 715
 716     uninit_opts();
 717 }
 718
 719 int split_commandline(OptionParseContext *octx, int argc, char *argv[],
 720                       const OptionDef *options,
 721                       const OptionGroupDef *groups, int nb_groups)
 722 {
 723     int optindex = 1;
 724     int dashdash = -2;
 725
 726     /* perform system-dependent conversions for arguments list */
 727     prepare_app_arguments(&argc, &argv);
 728
 729     init_parse_context(octx, groups, nb_groups);
 730     av_log(NULL, AV_LOG_DEBUG, "Splitting the commandline.\n");
 731
 732     while (optindex < argc) {
 733         const char *opt = argv[optindex++], *arg;
 734         const OptionDef *po;
 735         int ret;
 736
 737         av_log(NULL, AV_LOG_DEBUG, "Reading option '%s' ...", opt);
 738
 739         if (opt[0] == '-' && opt[1] == '-' && !opt[2]) {
 740             dashdash = optindex;
 741             continue;
 742         }
 743         /* unnamed group separators, e.g. output filename */
 744         if (opt[0] != '-' || !opt[1] || dashdash+1 == optindex) {
 745             finish_group(octx, 0, opt);
 746             av_log(NULL, AV_LOG_DEBUG, " matched as %s.\n", groups[0].name);
 747             continue;
 748         }
 749         opt++;
 750
 751 #define GET_ARG(arg)                                                           \
 752 do {                                                                           \
 753     arg = argv[optindex++];                                                    \
 754     if (!arg) {                                                                \
 755         av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'.\n", opt);\
 756         return AVERROR(EINVAL);                                                \
 757     }                                                                          \
 758 } while (0)
 759
 760         /* named group separators, e.g. -i */
 761         if ((ret = match_group_separator(groups, nb_groups, opt)) >= 0) {
 762             GET_ARG(arg);
 763             finish_group(octx, ret, arg);
 764             av_log(NULL, AV_LOG_DEBUG, " matched as %s with argument '%s'.\n",
 765                    groups[ret].name, arg);
 766             continue;
 767         }
 768
 769         /* normal options */
 770         po = find_option(options, opt);
 771         if (po->name) {
 772             if (po->flags & OPT_EXIT) {
 773                 /* optional argument, e.g. -h */
 774                 arg = argv[optindex++];
 775             } else if (po->flags & HAS_ARG) {
 776                 GET_ARG(arg);
 777             } else {
 778                 arg = "1";
 779             }
 780
 781             add_opt(octx, po, opt, arg);
 782             av_log(NULL, AV_LOG_DEBUG, " matched as option '%s' (%s) with "
 783                    "argument '%s'.\n", po->name, po->help, arg);
 784             continue;
 785         }
 786
 787         /* AVOptions */
 788         if (argv[optindex]) {
 789             ret = opt_default(NULL, opt, argv[optindex]);
 790             if (ret >= 0) {
 791                 av_log(NULL, AV_LOG_DEBUG, " matched as AVOption '%s' with "
 792                        "argument '%s'.\n", opt, argv[optindex]);
 793                 optindex++;
 794                 continue;
 795             } else if (ret != AVERROR_OPTION_NOT_FOUND) {
 796                 av_log(NULL, AV_LOG_ERROR, "Error parsing option '%s' "
 797                        "with argument '%s'.\n", opt, argv[optindex]);
 798                 return ret;
 799             }
 800         }
 801
 802         /* boolean -nofoo options */
 803         if (opt[0] == 'n' && opt[1] == 'o' &&
 804             (po = find_option(options, opt + 2)) &&
 805             po->name && po->flags & OPT_BOOL) {
 806             add_opt(octx, po, opt, "0");
 807             av_log(NULL, AV_LOG_DEBUG, " matched as option '%s' (%s) with "
 808                    "argument 0.\n", po->name, po->help);
 809             continue;
 810         }
 811
 812         av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'.\n", opt);
 813         return AVERROR_OPTION_NOT_FOUND;
 814     }
 815
 816     if (octx->cur_group.nb_opts || codec_opts || format_opts || resample_opts)
 817         av_log(NULL, AV_LOG_WARNING, "Trailing options were found on the "
 818                "commandline.\n");
 819
 820     av_log(NULL, AV_LOG_DEBUG, "Finished splitting the commandline.\n");
 821
 822     return 0;
 823 }
 824
 825 int opt_cpuflags(void *optctx, const char *opt, const char *arg)
 826 {
 827     int ret;
 828     unsigned flags = av_get_cpu_flags();
 829
 830     if ((ret = av_parse_cpu_caps(&flags, arg)) < 0)
 831         return ret;
 832
 833     av_force_cpu_flags(flags);
 834     return 0;
 835 }
 836
 837 int opt_loglevel(void *optctx, const char *opt, const char *arg)
 838 {
 839     const struct { const char *name; int level; } log_levels[] = {
 840         { "quiet"  , AV_LOG_QUIET   },
 841         { "panic"  , AV_LOG_PANIC   },
 842         { "fatal"  , AV_LOG_FATAL   },
 843         { "error"  , AV_LOG_ERROR   },
 844         { "warning", AV_LOG_WARNING },
 845         { "info"   , AV_LOG_INFO    },
 846         { "verbose", AV_LOG_VERBOSE },
 847         { "debug"  , AV_LOG_DEBUG   },
 848     };
 849     char *tail;
 850     int level;
 851     int flags;
 852     int i;
 853
 854     flags = av_log_get_flags();
 855     tail = strstr(arg, "repeat");
 856     if (tail)
 857         flags &= ~AV_LOG_SKIP_REPEATED;
 858     else
 859         flags |= AV_LOG_SKIP_REPEATED;
 860
 861     av_log_set_flags(flags);
 862     if (tail == arg)
 863         arg += 6 + (arg[6]=='+');
 864     if(tail && !*arg)
 865         return 0;
 866
 867     for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++) {
 868         if (!strcmp(log_levels[i].name, arg)) {
 869             av_log_set_level(log_levels[i].level);
 870             return 0;
 871         }
 872     }
 873
 874     level = strtol(arg, &tail, 10);
 875     if (*tail) {
 876         av_log(NULL, AV_LOG_FATAL, "Invalid loglevel \"%s\". "
 877                "Possible levels are numbers or:\n", arg);
 878         for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++)
 879             av_log(NULL, AV_LOG_FATAL, "\"%s\"\n", log_levels[i].name);
 880         exit_program(1);
 881     }
 882     av_log_set_level(level);
 883     return 0;
 884 }
 885
 886 static void expand_filename_template(AVBPrint *bp, const char *template,
 887                                      struct tm *tm)
 888 {
 889     int c;
 890
 891     while ((c = *(template++))) {
 892         if (c == '%') {
 893             if (!(c = *(template++)))
 894                 break;
 895             switch (c) {
 896             case 'p':
 897                 av_bprintf(bp, "%s", program_name);
 898                 break;
 899             case 't':
 900                 av_bprintf(bp, "%04d%02d%02d-%02d%02d%02d",
 901                            tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
 902                            tm->tm_hour, tm->tm_min, tm->tm_sec);
 903                 break;
 904             case '%':
 905                 av_bprint_chars(bp, c, 1);
 906                 break;
 907             }
 908         } else {
 909             av_bprint_chars(bp, c, 1);
 910         }
 911     }
 912 }
 913
 914 static int init_report(const char *env)
 915 {
 916     char *filename_template = NULL;
 917     char *key, *val;
 918     int ret, count = 0;
 919     time_t now;
 920     struct tm *tm;
 921     AVBPrint filename;
 922
 923     if (report_file) /* already opened */
 924         return 0;
 925     time(&now);
 926     tm = localtime(&now);
 927
 928     while (env && *env) {
 929         if ((ret = av_opt_get_key_value(&env, "=", ":", 0, &key, &val)) < 0) {
 930             if (count)
 931                 av_log(NULL, AV_LOG_ERROR,
 932                        "Failed to parse FFREPORT environment variable: %s\n",
 933                        av_err2str(ret));
 934             break;
 935         }
 936         if (*env)
 937             env++;
 938         count++;
 939         if (!strcmp(key, "file")) {
 940             av_free(filename_template);
 941             filename_template = val;
 942             val = NULL;
 943         } else if (!strcmp(key, "level")) {
 944             char *tail;
 945             report_file_level = strtol(val, &tail, 10);
 946             if (*tail) {
 947                 av_log(NULL, AV_LOG_FATAL, "Invalid report file level\n");
 948                 exit_program(1);
 949             }
 950         } else {
 951             av_log(NULL, AV_LOG_ERROR, "Unknown key '%s' in FFREPORT\n", key);
 952         }
 953         av_free(val);
 954         av_free(key);
 955     }
 956
 957     av_bprint_init(&filename, 0, 1);
 958     expand_filename_template(&filename,
 959                              av_x_if_null(filename_template, "%p-%t.log"), tm);
 960     av_free(filename_template);
 961     if (!av_bprint_is_complete(&filename)) {
 962         av_log(NULL, AV_LOG_ERROR, "Out of memory building report file name\n");
 963         return AVERROR(ENOMEM);
 964     }
 965
 966     report_file = fopen(filename.str, "w");
 967     if (!report_file) {
 968         int ret = AVERROR(errno);
 969         av_log(NULL, AV_LOG_ERROR, "Failed to open report \"%s\": %s\n",
 970                filename.str, strerror(errno));
 971         return ret;
 972     }
 973     av_log_set_callback(log_callback_report);
 974     av_log(NULL, AV_LOG_INFO,
 975            "%s started on %04d-%02d-%02d at %02d:%02d:%02d\n"
 976            "Report written to \"%s\"\n",
 977            program_name,
 978            tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
 979            tm->tm_hour, tm->tm_min, tm->tm_sec,
 980            filename.str);
 981     av_bprint_finalize(&filename, NULL);
 982     return 0;
 983 }
 984
 985 int opt_report(const char *opt)
 986 {
 987     return init_report(NULL);
 988 }
 989
 990 int opt_max_alloc(void *optctx, const char *opt, const char *arg)
 991 {
 992     char *tail;
 993     size_t max;
 994
 995     max = strtol(arg, &tail, 10);
 996     if (*tail) {
 997         av_log(NULL, AV_LOG_FATAL, "Invalid max_alloc \"%s\".\n", arg);
 998         exit_program(1);
 999     }
1000     av_max_alloc(max);
1001     return 0;
1002 }
1003
1004 int opt_timelimit(void *optctx, const char *opt, const char *arg)
1005 {
1006 #if HAVE_SETRLIMIT
1007     int lim = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
1008     struct rlimit rl = { lim, lim + 1 };
1009     if (setrlimit(RLIMIT_CPU, &rl))
1010         perror("setrlimit");
1011 #else
1012     av_log(NULL, AV_LOG_WARNING, "-%s not implemented on this OS\n", opt);
1013 #endif
1014     return 0;
1015 }
1016
1017 void print_error(const char *filename, int err)
1018 {
1019     char errbuf[128];
1020     const char *errbuf_ptr = errbuf;
1021
1022     if (av_strerror(err, errbuf, sizeof(errbuf)) < 0)
1023         errbuf_ptr = strerror(AVUNERROR(err));
1024     av_log(NULL, AV_LOG_ERROR, "%s: %s\n", filename, errbuf_ptr);
1025 }
1026
1027 static int warned_cfg = 0;
1028
1029 #define INDENT        1
1030 #define SHOW_VERSION  2
1031 #define SHOW_CONFIG   4
1032 #define SHOW_COPYRIGHT 8
1033
1034 #define PRINT_LIB_INFO(libname, LIBNAME, flags, level)                  \
1035     if (CONFIG_##LIBNAME) {                                             \
1036         const char *indent = flags & INDENT? "  " : "";                 \
1037         if (flags & SHOW_VERSION) {                                     \
1038             unsigned int version = libname##_version();                 \
1039             av_log(NULL, level,                                         \
1040                    "%slib%-11s %2d.%3d.%3d / %2d.%3d.%3d\n",            \
1041                    indent, #libname,                                    \
1042                    LIB##LIBNAME##_VERSION_MAJOR,                        \
1043                    LIB##LIBNAME##_VERSION_MINOR,                        \
1044                    LIB##LIBNAME##_VERSION_MICRO,                        \
1045                    version >> 16, version >> 8 & 0xff, version & 0xff); \
1046         }                                                               \
1047         if (flags & SHOW_CONFIG) {                                      \
1048             const char *cfg = libname##_configuration();                \
1049             if (strcmp(FFMPEG_CONFIGURATION, cfg)) {                    \
1050                 if (!warned_cfg) {                                      \
1051                     av_log(NULL, level,                                 \
1052                             "%sWARNING: library configuration mismatch\n", \
1053                             indent);                                    \
1054                     warned_cfg = 1;                                     \
1055                 }                                                       \
1056                 av_log(NULL, level, "%s%-11s configuration: %s\n",      \
1057                         indent, #libname, cfg);                         \
1058             }                                                           \
1059         }                                                               \
1060     }                                                                   \
1061
1062 static void print_all_libs_info(int flags, int level)
1063 {
1064     PRINT_LIB_INFO(avutil,   AVUTIL,   flags, level);
1065     PRINT_LIB_INFO(avcodec,  AVCODEC,  flags, level);
1066     PRINT_LIB_INFO(avformat, AVFORMAT, flags, level);
1067     PRINT_LIB_INFO(avdevice, AVDEVICE, flags, level);
1068     PRINT_LIB_INFO(avfilter, AVFILTER, flags, level);
1069     PRINT_LIB_INFO(avresample, AVRESAMPLE, flags, level);
1070     PRINT_LIB_INFO(swscale,  SWSCALE,  flags, level);
1071     PRINT_LIB_INFO(swresample,SWRESAMPLE,  flags, level);
1072     PRINT_LIB_INFO(postproc, POSTPROC, flags, level);
1073 }
1074
1075 static void print_program_info(int flags, int level)
1076 {
1077     const char *indent = flags & INDENT? "  " : "";
1078
1079     av_log(NULL, level, "%s version " FFMPEG_VERSION, program_name);
1080     if (flags & SHOW_COPYRIGHT)
1081         av_log(NULL, level, " Copyright (c) %d-%d the FFmpeg developers",
1082                program_birth_year, CONFIG_THIS_YEAR);
1083     av_log(NULL, level, "\n");
1084     av_log(NULL, level, "%sbuilt on %s %s with %s\n",
1085            indent, __DATE__, __TIME__, CC_IDENT);
1086
1087     av_log(NULL, level, "%sconfiguration: " FFMPEG_CONFIGURATION "\n", indent);
1088 }
1089
1090 static void print_buildconf(int flags, int level)
1091 {
1092     const char *indent = flags & INDENT ? "  " : "";
1093     char str[] = { FFMPEG_CONFIGURATION };
1094     char *conflist, *remove_tilde, *splitconf;
1095
1096     // Change all the ' --' strings to '~--' so that
1097     // they can be identified as tokens.
1098     while ((conflist = strstr(str, " --")) != NULL) {
1099         strncpy(conflist, "~--", 3);
1100     }
1101
1102     // Compensate for the weirdness this would cause
1103     // when passing 'pkg-config --static'.
1104     while ((remove_tilde = strstr(str, "pkg-config~")) != NULL) {
1105         strncpy(remove_tilde, "pkg-config ", 11);
1106     }
1107
1108     splitconf = strtok(str, "~");
1109     av_log(NULL, level, "\n%sconfiguration:\n", indent);
1110     while (splitconf != NULL) {
1111         av_log(NULL, level, "%s%s%s\n", indent, indent, splitconf);
1112         splitconf = strtok(NULL, "~");
1113     }
1114 }
1115
1116 void show_banner(int argc, char **argv, const OptionDef *options)
1117 {
1118     int idx = locate_option(argc, argv, options, "version");
1119     if (hide_banner || idx)
1120         return;
1121
1122     print_program_info (INDENT|SHOW_COPYRIGHT, AV_LOG_INFO);
1123     print_all_libs_info(INDENT|SHOW_CONFIG,  AV_LOG_INFO);
1124     print_all_libs_info(INDENT|SHOW_VERSION, AV_LOG_INFO);
1125 }
1126
1127 int show_version(void *optctx, const char *opt, const char *arg)
1128 {
1129     av_log_set_callback(log_callback_help);
1130     print_program_info (SHOW_COPYRIGHT, AV_LOG_INFO);
1131     print_all_libs_info(SHOW_VERSION, AV_LOG_INFO);
1132
1133     return 0;
1134 }
1135
1136 int show_buildconf(void *optctx, const char *opt, const char *arg)
1137 {
1138     av_log_set_callback(log_callback_help);
1139     print_buildconf      (INDENT|0, AV_LOG_INFO);
1140
1141     return 0;
1142 }
1143
1144 int show_license(void *optctx, const char *opt, const char *arg)
1145 {
1146 #if CONFIG_NONFREE
1147     printf(
1148     "This version of %s has nonfree parts compiled in.\n"
1149     "Therefore it is not legally redistributable.\n",
1150     program_name );
1151 #elif CONFIG_GPLV3
1152     printf(
1153     "%s is free software; you can redistribute it and/or modify\n"
1154     "it under the terms of the GNU General Public License as published by\n"
1155     "the Free Software Foundation; either version 3 of the License, or\n"
1156     "(at your option) any later version.\n"
1157     "\n"
1158     "%s is distributed in the hope that it will be useful,\n"
1159     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1160     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1161     "GNU General Public License for more details.\n"
1162     "\n"
1163     "You should have received a copy of the GNU General Public License\n"
1164     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
1165     program_name, program_name, program_name );
1166 #elif CONFIG_GPL
1167     printf(
1168     "%s is free software; you can redistribute it and/or modify\n"
1169     "it under the terms of the GNU General Public License as published by\n"
1170     "the Free Software Foundation; either version 2 of the License, or\n"
1171     "(at your option) any later version.\n"
1172     "\n"
1173     "%s is distributed in the hope that it will be useful,\n"
1174     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1175     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1176     "GNU General Public License for more details.\n"
1177     "\n"
1178     "You should have received a copy of the GNU General Public License\n"
1179     "along with %s; if not, write to the Free Software\n"
1180     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
1181     program_name, program_name, program_name );
1182 #elif CONFIG_LGPLV3
1183     printf(
1184     "%s is free software; you can redistribute it and/or modify\n"
1185     "it under the terms of the GNU Lesser General Public License as published by\n"
1186     "the Free Software Foundation; either version 3 of the License, or\n"
1187     "(at your option) any later version.\n"
1188     "\n"
1189     "%s is distributed in the hope that it will be useful,\n"
1190     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1191     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
1192     "GNU Lesser General Public License for more details.\n"
1193     "\n"
1194     "You should have received a copy of the GNU Lesser General Public License\n"
1195     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
1196     program_name, program_name, program_name );
1197 #else
1198     printf(
1199     "%s is free software; you can redistribute it and/or\n"
1200     "modify it under the terms of the GNU Lesser General Public\n"
1201     "License as published by the Free Software Foundation; either\n"
1202     "version 2.1 of the License, or (at your option) any later version.\n"
1203     "\n"
1204     "%s is distributed in the hope that it will be useful,\n"
1205     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
1206     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
1207     "Lesser General Public License for more details.\n"
1208     "\n"
1209     "You should have received a copy of the GNU Lesser General Public\n"
1210     "License along with %s; if not, write to the Free Software\n"
1211     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
1212     program_name, program_name, program_name );
1213 #endif
1214
1215     return 0;
1216 }
1217
1218 static int is_device(const AVClass *avclass)
1219 {
1220     if (!avclass)
1221         return 0;
1222     return AV_IS_INPUT_DEVICE(avclass->category) || AV_IS_OUTPUT_DEVICE(avclass->category);
1223 }
1224
1225 static int show_formats_devices(void *optctx, const char *opt, const char *arg, int device_only)
1226 {
1227     AVInputFormat *ifmt  = NULL;
1228     AVOutputFormat *ofmt = NULL;
1229     const char *last_name;
1230     int is_dev;
1231
1232     printf("%s\n"
1233            " D. = Demuxing supported\n"
1234            " .E = Muxing supported\n"
1235            " --\n", device_only ? "Devices:" : "File formats:");
1236     last_name = "000";
1237     for (;;) {
1238         int decode = 0;
1239         int encode = 0;
1240         const char *name      = NULL;
1241         const char *long_name = NULL;
1242
1243         while ((ofmt = av_oformat_next(ofmt))) {
1244             is_dev = is_device(ofmt->priv_class);
1245             if (!is_dev && device_only)
1246                 continue;
1247             if ((!name || strcmp(ofmt->name, name) < 0) &&
1248                 strcmp(ofmt->name, last_name) > 0) {
1249                 name      = ofmt->name;
1250                 long_name = ofmt->long_name;
1251                 encode    = 1;
1252             }
1253         }
1254         while ((ifmt = av_iformat_next(ifmt))) {
1255             is_dev = is_device(ifmt->priv_class);
1256             if (!is_dev && device_only)
1257                 continue;
1258             if ((!name || strcmp(ifmt->name, name) < 0) &&
1259                 strcmp(ifmt->name, last_name) > 0) {
1260                 name      = ifmt->name;
1261                 long_name = ifmt->long_name;
1262                 encode    = 0;
1263             }
1264             if (name && strcmp(ifmt->name, name) == 0)
1265                 decode = 1;
1266         }
1267         if (!name)
1268             break;
1269         last_name = name;
1270
1271         printf(" %s%s %-15s %s\n",
1272                decode ? "D" : " ",
1273                encode ? "E" : " ",
1274                name,
1275             long_name ? long_name:" ");
1276     }
1277     return 0;
1278 }
1279
1280 int show_formats(void *optctx, const char *opt, const char *arg)
1281 {
1282     return show_formats_devices(optctx, opt, arg, 0);
1283 }
1284
1285 int show_devices(void *optctx, const char *opt, const char *arg)
1286 {
1287     return show_formats_devices(optctx, opt, arg, 1);
1288 }
1289
1290 #define PRINT_CODEC_SUPPORTED(codec, field, type, list_name, term, get_name) \
1291     if (codec->field) {                                                      \
1292         const type *p = codec->field;                                        \
1293                                                                              \
1294         printf("    Supported " list_name ":");                              \
1295         while (*p != term) {                                                 \
1296             get_name(*p);                                                    \
1297             printf(" %s", name);                                             \
1298             p++;                                                             \
1299         }                                                                    \
1300         printf("\n");                                                        \
1301     }                                                                        \
1302
1303 static void print_codec(const AVCodec *c)
1304 {
1305     int encoder = av_codec_is_encoder(c);
1306
1307     printf("%s %s [%s]:\n", encoder ? "Encoder" : "Decoder", c->name,
1308            c->long_name ? c->long_name : "");
1309
1310     if (c->type == AVMEDIA_TYPE_VIDEO ||
1311         c->type == AVMEDIA_TYPE_AUDIO) {
1312         printf("    Threading capabilities: ");
1313         switch (c->capabilities & (CODEC_CAP_FRAME_THREADS |
1314                                    CODEC_CAP_SLICE_THREADS)) {
1315         case CODEC_CAP_FRAME_THREADS |
1316              CODEC_CAP_SLICE_THREADS: printf("frame and slice"); break;
1317         case CODEC_CAP_FRAME_THREADS: printf("frame");           break;
1318         case CODEC_CAP_SLICE_THREADS: printf("slice");           break;
1319         default:                      printf("no");              break;
1320         }
1321         printf("\n");
1322     }
1323
1324     if (c->supported_framerates) {
1325         const AVRational *fps = c->supported_framerates;
1326
1327         printf("    Supported framerates:");
1328         while (fps->num) {
1329             printf(" %d/%d", fps->num, fps->den);
1330             fps++;
1331         }
1332         printf("\n");
1333     }
1334     PRINT_CODEC_SUPPORTED(c, pix_fmts, enum AVPixelFormat, "pixel formats",
1335                           AV_PIX_FMT_NONE, GET_PIX_FMT_NAME);
1336     PRINT_CODEC_SUPPORTED(c, supported_samplerates, int, "sample rates", 0,
1337                           GET_SAMPLE_RATE_NAME);
1338     PRINT_CODEC_SUPPORTED(c, sample_fmts, enum AVSampleFormat, "sample formats",
1339                           AV_SAMPLE_FMT_NONE, GET_SAMPLE_FMT_NAME);
1340     PRINT_CODEC_SUPPORTED(c, channel_layouts, uint64_t, "channel layouts",
1341                           0, GET_CH_LAYOUT_DESC);
1342
1343     if (c->priv_class) {
1344         show_help_children(c->priv_class,
1345                            AV_OPT_FLAG_ENCODING_PARAM |
1346                            AV_OPT_FLAG_DECODING_PARAM);
1347     }
1348 }
1349
1350 static char get_media_type_char(enum AVMediaType type)
1351 {
1352     switch (type) {
1353         case AVMEDIA_TYPE_VIDEO:    return 'V';
1354         case AVMEDIA_TYPE_AUDIO:    return 'A';
1355         case AVMEDIA_TYPE_DATA:     return 'D';
1356         case AVMEDIA_TYPE_SUBTITLE: return 'S';
1357         case AVMEDIA_TYPE_ATTACHMENT:return 'T';
1358         default:                    return '?';
1359     }
1360 }
1361
1362 static const AVCodec *next_codec_for_id(enum AVCodecID id, const AVCodec *prev,
1363                                         int encoder)
1364 {
1365     while ((prev = av_codec_next(prev))) {
1366         if (prev->id == id &&
1367             (encoder ? av_codec_is_encoder(prev) : av_codec_is_decoder(prev)))
1368             return prev;
1369     }
1370     return NULL;
1371 }
1372
1373 static int compare_codec_desc(const void *a, const void *b)
1374 {
1375     const AVCodecDescriptor * const *da = a;
1376     const AVCodecDescriptor * const *db = b;
1377
1378     return (*da)->type != (*db)->type ? (*da)->type - (*db)->type :
1379            strcmp((*da)->name, (*db)->name);
1380 }
1381
1382 static unsigned get_codecs_sorted(const AVCodecDescriptor ***rcodecs)
1383 {
1384     const AVCodecDescriptor *desc = NULL;
1385     const AVCodecDescriptor **codecs;
1386     unsigned nb_codecs = 0, i = 0;
1387
1388     while ((desc = avcodec_descriptor_next(desc)))
1389         nb_codecs++;
1390     if (!(codecs = av_calloc(nb_codecs, sizeof(*codecs)))) {
1391         av_log(NULL, AV_LOG_ERROR, "Out of memory\n");
1392         exit_program(1);
1393     }
1394     desc = NULL;
1395     while ((desc = avcodec_descriptor_next(desc)))
1396         codecs[i++] = desc;
1397     av_assert0(i == nb_codecs);
1398     qsort(codecs, nb_codecs, sizeof(*codecs), compare_codec_desc);
1399     *rcodecs = codecs;
1400     return nb_codecs;
1401 }
1402
1403 static void print_codecs_for_id(enum AVCodecID id, int encoder)
1404 {
1405     const AVCodec *codec = NULL;
1406
1407     printf(" (%s: ", encoder ? "encoders" : "decoders");
1408
1409     while ((codec = next_codec_for_id(id, codec, encoder)))
1410         printf("%s ", codec->name);
1411
1412     printf(")");
1413 }
1414
1415 int show_codecs(void *optctx, const char *opt, const char *arg)
1416 {
1417     const AVCodecDescriptor **codecs;
1418     unsigned i, nb_codecs = get_codecs_sorted(&codecs);
1419
1420     printf("Codecs:\n"
1421            " D..... = Decoding supported\n"
1422            " .E.... = Encoding supported\n"
1423            " ..V... = Video codec\n"
1424            " ..A... = Audio codec\n"
1425            " ..S... = Subtitle codec\n"
1426            " ...I.. = Intra frame-only codec\n"
1427            " ....L. = Lossy compression\n"
1428            " .....S = Lossless compression\n"
1429            " -------\n");
1430     for (i = 0; i < nb_codecs; i++) {
1431         const AVCodecDescriptor *desc = codecs[i];
1432         const AVCodec *codec = NULL;
1433
1434         if (strstr(desc->name, "_deprecated"))
1435             continue;
1436
1437         printf(" ");
1438         printf(avcodec_find_decoder(desc->id) ? "D" : ".");
1439         printf(avcodec_find_encoder(desc->id) ? "E" : ".");
1440
1441         printf("%c", get_media_type_char(desc->type));
1442         printf((desc->props & AV_CODEC_PROP_INTRA_ONLY) ? "I" : ".");
1443         printf((desc->props & AV_CODEC_PROP_LOSSY)      ? "L" : ".");
1444         printf((desc->props & AV_CODEC_PROP_LOSSLESS)   ? "S" : ".");
1445
1446         printf(" %-20s %s", desc->name, desc->long_name ? desc->long_name : "");
1447
1448         /* print decoders/encoders when there's more than one or their
1449          * names are different from codec name */
1450         while ((codec = next_codec_for_id(desc->id, codec, 0))) {
1451             if (strcmp(codec->name, desc->name)) {
1452                 print_codecs_for_id(desc->id, 0);
1453                 break;
1454             }
1455         }
1456         codec = NULL;
1457         while ((codec = next_codec_for_id(desc->id, codec, 1))) {
1458             if (strcmp(codec->name, desc->name)) {
1459                 print_codecs_for_id(desc->id, 1);
1460                 break;
1461             }
1462         }
1463
1464         printf("\n");
1465     }
1466     av_free(codecs);
1467     return 0;
1468 }
1469
1470 static void print_codecs(int encoder)
1471 {
1472     const AVCodecDescriptor **codecs;
1473     unsigned i, nb_codecs = get_codecs_sorted(&codecs);
1474
1475     printf("%s:\n"
1476            " V..... = Video\n"
1477            " A..... = Audio\n"
1478            " S..... = Subtitle\n"
1479            " .F.... = Frame-level multithreading\n"
1480            " ..S... = Slice-level multithreading\n"
1481            " ...X.. = Codec is experimental\n"
1482            " ....B. = Supports draw_horiz_band\n"
1483            " .....D = Supports direct rendering method 1\n"
1484            " ------\n",
1485            encoder ? "Encoders" : "Decoders");
1486     for (i = 0; i < nb_codecs; i++) {
1487         const AVCodecDescriptor *desc = codecs[i];
1488         const AVCodec *codec = NULL;
1489
1490         while ((codec = next_codec_for_id(desc->id, codec, encoder))) {
1491             printf(" %c", get_media_type_char(desc->type));
1492             printf((codec->capabilities & CODEC_CAP_FRAME_THREADS) ? "F" : ".");
1493             printf((codec->capabilities & CODEC_CAP_SLICE_THREADS) ? "S" : ".");
1494             printf((codec->capabilities & CODEC_CAP_EXPERIMENTAL)  ? "X" : ".");
1495             printf((codec->capabilities & CODEC_CAP_DRAW_HORIZ_BAND)?"B" : ".");
1496             printf((codec->capabilities & CODEC_CAP_DR1)           ? "D" : ".");
1497
1498             printf(" %-20s %s", codec->name, codec->long_name ? codec->long_name : "");
1499             if (strcmp(codec->name, desc->name))
1500                 printf(" (codec %s)", desc->name);
1501
1502             printf("\n");
1503         }
1504     }
1505     av_free(codecs);
1506 }
1507
1508 int show_decoders(void *optctx, const char *opt, const char *arg)
1509 {
1510     print_codecs(0);
1511     return 0;
1512 }
1513
1514 int show_encoders(void *optctx, const char *opt, const char *arg)
1515 {
1516     print_codecs(1);
1517     return 0;
1518 }
1519
1520 int show_bsfs(void *optctx, const char *opt, const char *arg)
1521 {
1522     AVBitStreamFilter *bsf = NULL;
1523
1524     printf("Bitstream filters:\n");
1525     while ((bsf = av_bitstream_filter_next(bsf)))
1526         printf("%s\n", bsf->name);
1527     printf("\n");
1528     return 0;
1529 }
1530
1531 int show_protocols(void *optctx, const char *opt, const char *arg)
1532 {
1533     void *opaque = NULL;
1534     const char *name;
1535
1536     printf("Supported file protocols:\n"
1537            "Input:\n");
1538     while ((name = avio_enum_protocols(&opaque, 0)))
1539         printf("%s\n", name);
1540     printf("Output:\n");
1541     while ((name = avio_enum_protocols(&opaque, 1)))
1542         printf("%s\n", name);
1543     return 0;
1544 }
1545
1546 int show_filters(void *optctx, const char *opt, const char *arg)
1547 {
1548 #if CONFIG_AVFILTER
1549     const AVFilter *filter = NULL;
1550     char descr[64], *descr_cur;
1551     int i, j;
1552     const AVFilterPad *pad;
1553
1554     printf("Filters:\n"
1555            "  T.. = Timeline support\n"
1556            "  .S. = Slice threading\n"
1557            "  ..C = Commmand support\n"
1558            "  A = Audio input/output\n"
1559            "  V = Video input/output\n"
1560            "  N = Dynamic number and/or type of input/output\n"
1561            "  | = Source or sink filter\n");
1562     while ((filter = avfilter_next(filter))) {
1563         descr_cur = descr;
1564         for (i = 0; i < 2; i++) {
1565             if (i) {
1566                 *(descr_cur++) = '-';
1567                 *(descr_cur++) = '>';
1568             }
1569             pad = i ? filter->outputs : filter->inputs;
1570             for (j = 0; pad && pad[j].name; j++) {
1571                 if (descr_cur >= descr + sizeof(descr) - 4)
1572                     break;
1573                 *(descr_cur++) = get_media_type_char(pad[j].type);
1574             }
1575             if (!j)
1576                 *(descr_cur++) = ((!i && (filter->flags & AVFILTER_FLAG_DYNAMIC_INPUTS)) ||
1577                                   ( i && (filter->flags & AVFILTER_FLAG_DYNAMIC_OUTPUTS))) ? 'N' : '|';
1578         }
1579         *descr_cur = 0;
1580         printf(" %c%c%c %-16s %-10s %s\n",
1581                filter->flags & AVFILTER_FLAG_SUPPORT_TIMELINE ? 'T' : '.',
1582                filter->flags & AVFILTER_FLAG_SLICE_THREADS    ? 'S' : '.',
1583                filter->process_command                        ? 'C' : '.',
1584                filter->name, descr, filter->description);
1585     }
1586 #else
1587     printf("No filters available: libavfilter disabled\n");
1588 #endif
1589     return 0;
1590 }
1591
1592 int show_colors(void *optctx, const char *opt, const char *arg)
1593 {
1594     const char *name;
1595     const uint8_t *rgb;
1596     int i;
1597
1598     printf("%-32s #RRGGBB\n", "name");
1599
1600     for (i = 0; name = av_get_known_color_name(i, &rgb); i++)
1601         printf("%-32s #%02x%02x%02x\n", name, rgb[0], rgb[1], rgb[2]);
1602
1603     return 0;
1604 }
1605
1606 int show_pix_fmts(void *optctx, const char *opt, const char *arg)
1607 {
1608     const AVPixFmtDescriptor *pix_desc = NULL;
1609
1610     printf("Pixel formats:\n"
1611            "I.... = Supported Input  format for conversion\n"
1612            ".O... = Supported Output format for conversion\n"
1613            "..H.. = Hardware accelerated format\n"
1614            "...P. = Paletted format\n"
1615            "....B = Bitstream format\n"
1616            "FLAGS NAME            NB_COMPONENTS BITS_PER_PIXEL\n"
1617            "-----\n");
1618
1619 #if !CONFIG_SWSCALE
1620 #   define sws_isSupportedInput(x)  0
1621 #   define sws_isSupportedOutput(x) 0
1622 #endif
1623
1624     while ((pix_desc = av_pix_fmt_desc_next(pix_desc))) {
1625         enum AVPixelFormat pix_fmt = av_pix_fmt_desc_get_id(pix_desc);
1626         printf("%c%c%c%c%c %-16s       %d            %2d\n",
1627                sws_isSupportedInput (pix_fmt)              ? 'I' : '.',
1628                sws_isSupportedOutput(pix_fmt)              ? 'O' : '.',
1629                pix_desc->flags & AV_PIX_FMT_FLAG_HWACCEL   ? 'H' : '.',
1630                pix_desc->flags & AV_PIX_FMT_FLAG_PAL       ? 'P' : '.',
1631                pix_desc->flags & AV_PIX_FMT_FLAG_BITSTREAM ? 'B' : '.',
1632                pix_desc->name,
1633                pix_desc->nb_components,
1634                av_get_bits_per_pixel(pix_desc));
1635     }
1636     return 0;
1637 }
1638
1639 int show_layouts(void *optctx, const char *opt, const char *arg)
1640 {
1641     int i = 0;
1642     uint64_t layout, j;
1643     const char *name, *descr;
1644
1645     printf("Individual channels:\n"
1646            "NAME           DESCRIPTION\n");
1647     for (i = 0; i < 63; i++) {
1648         name = av_get_channel_name((uint64_t)1 << i);
1649         if (!name)
1650             continue;
1651         descr = av_get_channel_description((uint64_t)1 << i);
1652         printf("%-14s %s\n", name, descr);
1653     }
1654     printf("\nStandard channel layouts:\n"
1655            "NAME           DECOMPOSITION\n");
1656     for (i = 0; !av_get_standard_channel_layout(i, &layout, &name); i++) {
1657         if (name) {
1658             printf("%-14s ", name);
1659             for (j = 1; j; j <<= 1)
1660                 if ((layout & j))
1661                     printf("%s%s", (layout & (j - 1)) ? "+" : "", av_get_channel_name(j));
1662             printf("\n");
1663         }
1664     }
1665     return 0;
1666 }
1667
1668 int show_sample_fmts(void *optctx, const char *opt, const char *arg)
1669 {
1670     int i;
1671     char fmt_str[128];
1672     for (i = -1; i < AV_SAMPLE_FMT_NB; i++)
1673         printf("%s\n", av_get_sample_fmt_string(fmt_str, sizeof(fmt_str), i));
1674     return 0;
1675 }
1676
1677 static void show_help_codec(const char *name, int encoder)
1678 {
1679     const AVCodecDescriptor *desc;
1680     const AVCodec *codec;
1681
1682     if (!name) {
1683         av_log(NULL, AV_LOG_ERROR, "No codec name specified.\n");
1684         return;
1685     }
1686
1687     codec = encoder ? avcodec_find_encoder_by_name(name) :
1688                       avcodec_find_decoder_by_name(name);
1689
1690     if (codec)
1691         print_codec(codec);
1692     else if ((desc = avcodec_descriptor_get_by_name(name))) {
1693         int printed = 0;
1694
1695         while ((codec = next_codec_for_id(desc->id, codec, encoder))) {
1696             printed = 1;
1697             print_codec(codec);
1698         }
1699
1700         if (!printed) {
1701             av_log(NULL, AV_LOG_ERROR, "Codec '%s' is known to FFmpeg, "
1702                    "but no %s for it are available. FFmpeg might need to be "
1703                    "recompiled with additional external libraries.\n",
1704                    name, encoder ? "encoders" : "decoders");
1705         }
1706     } else {
1707         av_log(NULL, AV_LOG_ERROR, "Codec '%s' is not recognized by FFmpeg.\n",
1708                name);
1709     }
1710 }
1711
1712 static void show_help_demuxer(const char *name)
1713 {
1714     const AVInputFormat *fmt = av_find_input_format(name);
1715
1716     if (!fmt) {
1717         av_log(NULL, AV_LOG_ERROR, "Unknown format '%s'.\n", name);
1718         return;
1719     }
1720
1721     printf("Demuxer %s [%s]:\n", fmt->name, fmt->long_name);
1722
1723     if (fmt->extensions)
1724         printf("    Common extensions: %s.\n", fmt->extensions);
1725
1726     if (fmt->priv_class)
1727         show_help_children(fmt->priv_class, AV_OPT_FLAG_DECODING_PARAM);
1728 }
1729
1730 static void show_help_muxer(const char *name)
1731 {
1732     const AVCodecDescriptor *desc;
1733     const AVOutputFormat *fmt = av_guess_format(name, NULL, NULL);
1734
1735     if (!fmt) {
1736         av_log(NULL, AV_LOG_ERROR, "Unknown format '%s'.\n", name);
1737         return;
1738     }
1739
1740     printf("Muxer %s [%s]:\n", fmt->name, fmt->long_name);
1741
1742     if (fmt->extensions)
1743         printf("    Common extensions: %s.\n", fmt->extensions);
1744     if (fmt->mime_type)
1745         printf("    Mime type: %s.\n", fmt->mime_type);
1746     if (fmt->video_codec != AV_CODEC_ID_NONE &&
1747         (desc = avcodec_descriptor_get(fmt->video_codec))) {
1748         printf("    Default video codec: %s.\n", desc->name);
1749     }
1750     if (fmt->audio_codec != AV_CODEC_ID_NONE &&
1751         (desc = avcodec_descriptor_get(fmt->audio_codec))) {
1752         printf("    Default audio codec: %s.\n", desc->name);
1753     }
1754     if (fmt->subtitle_codec != AV_CODEC_ID_NONE &&
1755         (desc = avcodec_descriptor_get(fmt->subtitle_codec))) {
1756         printf("    Default subtitle codec: %s.\n", desc->name);
1757     }
1758
1759     if (fmt->priv_class)
1760         show_help_children(fmt->priv_class, AV_OPT_FLAG_ENCODING_PARAM);
1761 }
1762
1763 #if CONFIG_AVFILTER
1764 static void show_help_filter(const char *name)
1765 {
1766 #if CONFIG_AVFILTER
1767     const AVFilter *f = avfilter_get_by_name(name);
1768     int i, count;
1769
1770     if (!name) {
1771         av_log(NULL, AV_LOG_ERROR, "No filter name specified.\n");
1772         return;
1773     } else if (!f) {
1774         av_log(NULL, AV_LOG_ERROR, "Unknown filter '%s'.\n", name);
1775         return;
1776     }
1777
1778     printf("Filter %s\n", f->name);
1779     if (f->description)
1780         printf("  %s\n", f->description);
1781
1782     if (f->flags & AVFILTER_FLAG_SLICE_THREADS)
1783         printf("    slice threading supported\n");
1784
1785     printf("    Inputs:\n");
1786     count = avfilter_pad_count(f->inputs);
1787     for (i = 0; i < count; i++) {
1788         printf("       #%d: %s (%s)\n", i, avfilter_pad_get_name(f->inputs, i),
1789                media_type_string(avfilter_pad_get_type(f->inputs, i)));
1790     }
1791     if (f->flags & AVFILTER_FLAG_DYNAMIC_INPUTS)
1792         printf("        dynamic (depending on the options)\n");
1793     else if (!count)
1794         printf("        none (source filter)\n");
1795
1796     printf("    Outputs:\n");
1797     count = avfilter_pad_count(f->outputs);
1798     for (i = 0; i < count; i++) {
1799         printf("       #%d: %s (%s)\n", i, avfilter_pad_get_name(f->outputs, i),
1800                media_type_string(avfilter_pad_get_type(f->outputs, i)));
1801     }
1802     if (f->flags & AVFILTER_FLAG_DYNAMIC_OUTPUTS)
1803         printf("        dynamic (depending on the options)\n");
1804     else if (!count)
1805         printf("        none (sink filter)\n");
1806
1807     if (f->priv_class)
1808         show_help_children(f->priv_class, AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM |
1809                                           AV_OPT_FLAG_AUDIO_PARAM);
1810     if (f->flags & AVFILTER_FLAG_SUPPORT_TIMELINE)
1811         printf("This filter has support for timeline through the 'enable' option.\n");
1812 #else
1813     av_log(NULL, AV_LOG_ERROR, "Build without libavfilter; "
1814            "can not to satisfy request\n");
1815 #endif
1816 }
1817 #endif
1818
1819 int show_help(void *optctx, const char *opt, const char *arg)
1820 {
1821     char *topic, *par;
1822     av_log_set_callback(log_callback_help);
1823
1824     topic = av_strdup(arg ? arg : "");
1825     if (!topic)
1826         return AVERROR(ENOMEM);
1827     par = strchr(topic, '=');
1828     if (par)
1829         *par++ = 0;
1830
1831     if (!*topic) {
1832         show_help_default(topic, par);
1833     } else if (!strcmp(topic, "decoder")) {
1834         show_help_codec(par, 0);
1835     } else if (!strcmp(topic, "encoder")) {
1836         show_help_codec(par, 1);
1837     } else if (!strcmp(topic, "demuxer")) {
1838         show_help_demuxer(par);
1839     } else if (!strcmp(topic, "muxer")) {
1840         show_help_muxer(par);
1841 #if CONFIG_AVFILTER
1842     } else if (!strcmp(topic, "filter")) {
1843         show_help_filter(par);
1844 #endif
1845     } else {
1846         show_help_default(topic, par);
1847     }
1848
1849     av_freep(&topic);
1850     return 0;
1851 }
1852
1853 int read_yesno(void)
1854 {
1855     int c = getchar();
1856     int yesno = (av_toupper(c) == 'Y');
1857
1858     while (c != '\n' && c != EOF)
1859         c = getchar();
1860
1861     return yesno;
1862 }
1863
1864 int cmdutils_read_file(const char *filename, char **bufptr, size_t *size)
1865 {
1866     int64_t ret;
1867     FILE *f = av_fopen_utf8(filename, "rb");
1868
1869     if (!f) {
1870         ret = AVERROR(errno);
1871         av_log(NULL, AV_LOG_ERROR, "Cannot read file '%s': %s\n", filename,
1872                strerror(errno));
1873         return ret;
1874     }
1875
1876     ret = fseek(f, 0, SEEK_END);
1877     if (ret == -1) {
1878         ret = AVERROR(errno);
1879         goto out;
1880     }
1881
1882     ret = ftell(f);
1883     if (ret < 0) {
1884         ret = AVERROR(errno);
1885         goto out;
1886     }
1887     *size = ret;
1888
1889     ret = fseek(f, 0, SEEK_SET);
1890     if (ret == -1) {
1891         ret = AVERROR(errno);
1892         goto out;
1893     }
1894
1895     *bufptr = av_malloc(*size + 1);
1896     if (!*bufptr) {
1897         av_log(NULL, AV_LOG_ERROR, "Could not allocate file buffer\n");
1898         ret = AVERROR(ENOMEM);
1899         goto out;
1900     }
1901     ret = fread(*bufptr, 1, *size, f);
1902     if (ret < *size) {
1903         av_free(*bufptr);
1904         if (ferror(f)) {
1905             ret = AVERROR(errno);
1906             av_log(NULL, AV_LOG_ERROR, "Error while reading file '%s': %s\n",
1907                    filename, strerror(errno));
1908         } else
1909             ret = AVERROR_EOF;
1910     } else {
1911         ret = 0;
1912         (*bufptr)[(*size)++] = '\0';
1913     }
1914
1915 out:
1916     av_log(NULL, AV_LOG_ERROR, "IO error: %s\n", av_err2str(ret));
1917     fclose(f);
1918     return ret;
1919 }
1920
1921 FILE *get_preset_file(char *filename, size_t filename_size,
1922                       const char *preset_name, int is_path,
1923                       const char *codec_name)
1924 {
1925     FILE *f = NULL;
1926     int i;
1927     const char *base[3] = { getenv("FFMPEG_DATADIR"),
1928                             getenv("HOME"),
1929                             FFMPEG_DATADIR, };
1930
1931     if (is_path) {
1932         av_strlcpy(filename, preset_name, filename_size);
1933         f = fopen(filename, "r");
1934     } else {
1935 #ifdef _WIN32
1936         char datadir[MAX_PATH], *ls;
1937         base[2] = NULL;
1938
1939         if (GetModuleFileNameA(GetModuleHandleA(NULL), datadir, sizeof(datadir) - 1))
1940         {
1941             for (ls = datadir; ls < datadir + strlen(datadir); ls++)
1942                 if (*ls == '\\') *ls = '/';
1943
1944             if (ls = strrchr(datadir, '/'))
1945             {
1946                 *ls = 0;
1947                 strncat(datadir, "/ffpresets",  sizeof(datadir) - 1 - strlen(datadir));
1948                 base[2] = datadir;
1949             }
1950         }
1951 #endif
1952         for (i = 0; i < 3 && !f; i++) {
1953             if (!base[i])
1954                 continue;
1955             snprintf(filename, filename_size, "%s%s/%s.ffpreset", base[i],
1956                      i != 1 ? "" : "/.ffmpeg", preset_name);
1957             f = fopen(filename, "r");
1958             if (!f && codec_name) {
1959                 snprintf(filename, filename_size,
1960                          "%s%s/%s-%s.ffpreset",
1961                          base[i], i != 1 ? "" : "/.ffmpeg", codec_name,
1962                          preset_name);
1963                 f = fopen(filename, "r");
1964             }
1965         }
1966     }
1967
1968     return f;
1969 }
1970
1971 int check_stream_specifier(AVFormatContext *s, AVStream *st, const char *spec)
1972 {
1973     int ret = avformat_match_stream_specifier(s, st, spec);
1974     if (ret < 0)
1975         av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
1976     return ret;
1977 }
1978
1979 AVDictionary *filter_codec_opts(AVDictionary *opts, enum AVCodecID codec_id,
1980                                 AVFormatContext *s, AVStream *st, AVCodec *codec)
1981 {
1982     AVDictionary    *ret = NULL;
1983     AVDictionaryEntry *t = NULL;
1984     int            flags = s->oformat ? AV_OPT_FLAG_ENCODING_PARAM
1985                                       : AV_OPT_FLAG_DECODING_PARAM;
1986     char          prefix = 0;
1987     const AVClass    *cc = avcodec_get_class();
1988
1989     if (!codec)
1990         codec            = s->oformat ? avcodec_find_encoder(codec_id)
1991                                       : avcodec_find_decoder(codec_id);
1992
1993     switch (st->codec->codec_type) {
1994     case AVMEDIA_TYPE_VIDEO:
1995         prefix  = 'v';
1996         flags  |= AV_OPT_FLAG_VIDEO_PARAM;
1997         break;
1998     case AVMEDIA_TYPE_AUDIO:
1999         prefix  = 'a';
2000         flags  |= AV_OPT_FLAG_AUDIO_PARAM;
2001         break;
2002     case AVMEDIA_TYPE_SUBTITLE:
2003         prefix  = 's';
2004         flags  |= AV_OPT_FLAG_SUBTITLE_PARAM;
2005         break;
2006     }
2007
2008     while (t = av_dict_get(opts, "", t, AV_DICT_IGNORE_SUFFIX)) {
2009         char *p = strchr(t->key, ':');
2010
2011         /* check stream specification in opt name */
2012         if (p)
2013             switch (check_stream_specifier(s, st, p + 1)) {
2014             case  1: *p = 0; break;
2015             case  0:         continue;
2016             default:         exit_program(1);
2017             }
2018
2019         if (av_opt_find(&cc, t->key, NULL, flags, AV_OPT_SEARCH_FAKE_OBJ) ||
2020             !codec ||
2021             (codec->priv_class &&
2022              av_opt_find(&codec->priv_class, t->key, NULL, flags,
2023                          AV_OPT_SEARCH_FAKE_OBJ)))
2024             av_dict_set(&ret, t->key, t->value, 0);
2025         else if (t->key[0] == prefix &&
2026                  av_opt_find(&cc, t->key + 1, NULL, flags,
2027                              AV_OPT_SEARCH_FAKE_OBJ))
2028             av_dict_set(&ret, t->key + 1, t->value, 0);
2029
2030         if (p)
2031             *p = ':';
2032     }
2033     return ret;
2034 }
2035
2036 AVDictionary **setup_find_stream_info_opts(AVFormatContext *s,
2037                                            AVDictionary *codec_opts)
2038 {
2039     int i;
2040     AVDictionary **opts;
2041
2042     if (!s->nb_streams)
2043         return NULL;
2044     opts = av_mallocz_array(s->nb_streams, sizeof(*opts));
2045     if (!opts) {
2046         av_log(NULL, AV_LOG_ERROR,
2047                "Could not alloc memory for stream options.\n");
2048         return NULL;
2049     }
2050     for (i = 0; i < s->nb_streams; i++)
2051         opts[i] = filter_codec_opts(codec_opts, s->streams[i]->codec->codec_id,
2052                                     s, s->streams[i], NULL);
2053     return opts;
2054 }
2055
2056 void *grow_array(void *array, int elem_size, int *size, int new_size)
2057 {
2058     if (new_size >= INT_MAX / elem_size) {
2059         av_log(NULL, AV_LOG_ERROR, "Array too big.\n");
2060         exit_program(1);
2061     }
2062     if (*size < new_size) {
2063         uint8_t *tmp = av_realloc_array(array, new_size, elem_size);
2064         if (!tmp) {
2065             av_log(NULL, AV_LOG_ERROR, "Could not alloc buffer.\n");
2066             exit_program(1);
2067         }
2068         memset(tmp + *size*elem_size, 0, (new_size-*size) * elem_size);
2069         *size = new_size;
2070         return tmp;
2071     }
2072     return array;
2073 }
2074
2075 #if CONFIG_AVDEVICE
2076 static int print_device_sources(AVInputFormat *fmt, AVDictionary *opts)
2077 {
2078     int ret, i;
2079     AVFormatContext *dev = NULL;
2080     AVDeviceInfoList *device_list = NULL;
2081     AVDictionary *tmp_opts = NULL;
2082
2083     if (!fmt || !fmt->priv_class  || !AV_IS_INPUT_DEVICE(fmt->priv_class->category))
2084         return AVERROR(EINVAL);
2085
2086     printf("Audo-detected sources for %s:\n", fmt->name);
2087     if (!fmt->get_device_list) {
2088         ret = AVERROR(ENOSYS);
2089         printf("Cannot list sources. Not implemented.\n");
2090         goto fail;
2091     }
2092
2093     /* TODO: avformat_open_input calls read_header callback which is not necessary.
2094              Function like avformat_alloc_output_context2 for input could be helpful here. */
2095     av_dict_copy(&tmp_opts, opts, 0);
2096     if ((ret = avformat_open_input(&dev, NULL, fmt, &tmp_opts)) < 0) {
2097         printf("Cannot open device: %s.\n", fmt->name);
2098         goto fail;
2099     }
2100
2101     if ((ret = avdevice_list_devices(dev, &device_list)) < 0) {
2102         printf("Cannot list sources.\n");
2103         goto fail;
2104     }
2105
2106     for (i = 0; i < device_list->nb_devices; i++) {
2107         printf("%s %s [%s]\n", device_list->default_device == i ? "*" : " ",
2108                device_list->devices[i]->device_name, device_list->devices[i]->device_description);
2109     }
2110
2111   fail:
2112     av_dict_free(&tmp_opts);
2113     avdevice_free_list_devices(&device_list);
2114     avformat_close_input(&dev);
2115     return ret;
2116 }
2117
2118 static int print_device_sinks(AVOutputFormat *fmt, AVDictionary *opts)
2119 {
2120     int ret, i;
2121     AVFormatContext *dev = NULL;
2122     AVDeviceInfoList *device_list = NULL;
2123     AVDictionary *tmp_opts = NULL;
2124
2125     if (!fmt || !fmt->priv_class  || !AV_IS_OUTPUT_DEVICE(fmt->priv_class->category))
2126         return AVERROR(EINVAL);
2127
2128     printf("Audo-detected sinks for %s:\n", fmt->name);
2129     if (!fmt->get_device_list) {
2130         ret = AVERROR(ENOSYS);
2131         printf("Cannot list sinks. Not implemented.\n");
2132         goto fail;
2133     }
2134
2135     if ((ret = avformat_alloc_output_context2(&dev, fmt, NULL, NULL)) < 0) {
2136         printf("Cannot open device: %s.\n", fmt->name);
2137         goto fail;
2138     }
2139     av_dict_copy(&tmp_opts, opts, 0);
2140     av_opt_set_dict2(dev, &tmp_opts, AV_OPT_SEARCH_CHILDREN);
2141
2142     if ((ret = avdevice_list_devices(dev, &device_list)) < 0) {
2143         printf("Cannot list sinks.\n");
2144         goto fail;
2145     }
2146
2147     for (i = 0; i < device_list->nb_devices; i++) {
2148         printf("%s %s [%s]\n", device_list->default_device == i ? "*" : " ",
2149                device_list->devices[i]->device_name, device_list->devices[i]->device_description);
2150     }
2151
2152   fail:
2153     av_dict_free(&tmp_opts);
2154     avdevice_free_list_devices(&device_list);
2155     avformat_free_context(dev);
2156     return ret;
2157 }
2158
2159 static int show_sinks_sources_parse_arg(const char *arg, char **dev, AVDictionary **opts)
2160 {
2161     int ret;
2162     if (arg) {
2163         char *opts_str = NULL;
2164         av_assert0(dev && opts);
2165         *dev = av_strdup(arg);
2166         if (!*dev)
2167             return AVERROR(ENOMEM);
2168         if ((opts_str = strchr(*dev, ','))) {
2169             *(opts_str++) = '\0';
2170             if (opts_str[0] && ((ret = av_dict_parse_string(opts, opts_str, "=", ":", 0)) < 0)) {
2171                 av_freep(dev);
2172                 return ret;
2173             }
2174         }
2175     } else
2176         printf("\nDevice name is not provided.\n"
2177                 "You can pass devicename[,opt1=val1[,opt2=val2...]] as an argument.\n\n");
2178     return 0;
2179 }
2180
2181 int show_sources(void *optctx, const char *opt, const char *arg)
2182 {
2183     AVInputFormat *fmt = NULL;
2184     char *dev = NULL;
2185     AVDictionary *opts = NULL;
2186     int ret = 0;
2187     int error_level = av_log_get_level();
2188
2189     av_log_set_level(AV_LOG_ERROR);
2190
2191     if ((ret = show_sinks_sources_parse_arg(arg, &dev, &opts)) < 0)
2192         goto fail;
2193
2194     do {
2195         fmt = av_input_audio_device_next(fmt);
2196         if (fmt) {
2197             if (!strcmp(fmt->name, "lavfi"))
2198                 continue; //it's pointless to probe lavfi
2199             if (dev && strcmp(fmt->name, dev))
2200                 continue;
2201             print_device_sources(fmt, opts);
2202         }
2203     } while (fmt);
2204     do {
2205         fmt = av_input_video_device_next(fmt);
2206         if (fmt) {
2207             if (dev && strcmp(fmt->name, dev))
2208                 continue;
2209             print_device_sources(fmt, opts);
2210         }
2211     } while (fmt);
2212   fail:
2213     av_dict_free(&opts);
2214     av_free(dev);
2215     av_log_set_level(error_level);
2216     return ret;
2217 }
2218
2219 int show_sinks(void *optctx, const char *opt, const char *arg)
2220 {
2221     AVOutputFormat *fmt = NULL;
2222     char *dev = NULL;
2223     AVDictionary *opts = NULL;
2224     int ret = 0;
2225     int error_level = av_log_get_level();
2226
2227     av_log_set_level(AV_LOG_ERROR);
2228
2229     if ((ret = show_sinks_sources_parse_arg(arg, &dev, &opts)) < 0)
2230         goto fail;
2231
2232     do {
2233         fmt = av_output_audio_device_next(fmt);
2234         if (fmt) {
2235             if (dev && strcmp(fmt->name, dev))
2236                 continue;
2237             print_device_sinks(fmt, opts);
2238         }
2239     } while (fmt);
2240     do {
2241         fmt = av_output_video_device_next(fmt);
2242         if (fmt) {
2243             if (dev && strcmp(fmt->name, dev))
2244                 continue;
2245             print_device_sinks(fmt, opts);
2246         }
2247     } while (fmt);
2248   fail:
2249     av_dict_free(&opts);
2250     av_free(dev);
2251     av_log_set_level(error_level);
2252     return ret;
2253 }
2254 #endif