diff --git a/cmdutils.c b/cmdutils.c index e072d75970..adf1b5fea3 100644 --- a/cmdutils.c +++ b/cmdutils.c @@ -510,6 +510,20 @@ int opt_report(const char *opt) return 0; } +int opt_max_alloc(const char *opt, const char *arg) +{ + char *tail; + size_t max; + + max = strtol(arg, &tail, 10); + if (*tail) { + av_log(NULL, AV_LOG_FATAL, "Invalid max_alloc \"%s\".\n", arg); + exit_program(1); + } + av_max_alloc(max); + return 0; +} + int opt_codec_debug(const char *opt, const char *arg) { av_log_set_level(AV_LOG_DEBUG); diff --git a/cmdutils.h b/cmdutils.h index a682c96d79..b9ea1cfc69 100644 --- a/cmdutils.h +++ b/cmdutils.h @@ -78,6 +78,8 @@ int opt_loglevel(const char *opt, const char *arg); int opt_report(const char *opt); +int opt_max_alloc(const char *opt, const char *arg); + int opt_codec_debug(const char *opt, const char *arg); /** diff --git a/cmdutils_common_opts.h b/cmdutils_common_opts.h index 058f2b1e2f..0aa6d99db6 100644 --- a/cmdutils_common_opts.h +++ b/cmdutils_common_opts.h @@ -15,3 +15,4 @@ { "v", HAS_ARG, {(void*)opt_loglevel}, "set libav* logging level", "loglevel" }, { "debug", HAS_ARG, {(void*)opt_codec_debug}, "set debug flags", "flags" }, { "report", 0, {(void*)opt_report}, "generate a report" }, + { "max_alloc", HAS_ARG, {(void*)opt_max_alloc}, "set maximum size of a single allocated block", "bytes" },