Message ID | CAPf3GF=6J4_MmRTgdxTn0rdgcAkO-w1239SGHpF3H1Qus90DHQ@mail.gmail.com |
---|---|
State | New |
Headers | show |
Series | [FFmpeg-devel] avfilter/vf_libvmaf: fix string comparison bug | expand |
Context | Check | Description |
---|---|---|
andriy/configure_x86 | warning | Failed to apply patch |
Hi, On Mon, Dec 4, 2023 at 7:59 AM Nil Fons Miret via ffmpeg-devel <ffmpeg-devel@ffmpeg.org> wrote: > > The libvmaf filter was doing substring checks in place of string > equality comparisons. This led to a bug when the user specified the > pooling method "harmonic_mean", since "mean" was checked first and the > substring comparison returned true. This patch changes all substring > comparisons for string equality comparisons. This is both correct and > more efficient than the existing method. > > Signed-off-by: nilfm <nilf@netflix.com> > --- > libavfilter/vf_libvmaf.c | 30 +++++++++++++++--------------- > 1 file changed, 15 insertions(+), 15 deletions(-) > > diff --git a/libavfilter/vf_libvmaf.c b/libavfilter/vf_libvmaf.c > index 12810b7267..46ff8154ef 100644 > --- a/libavfilter/vf_libvmaf.c > +++ b/libavfilter/vf_libvmaf.c > @@ -251,7 +251,7 @@ static int parse_features(AVFilterContext *ctx) > const AVDictionaryEntry *e = NULL; > > while (e = av_dict_iterate(dict[i], e)) { > - if (av_stristr(e->key, "name")) { > + if (!strcmp(e->key, "name")) { > feature_name = e->value; > continue; > } > @@ -312,29 +312,29 @@ static int parse_models(AVFilterContext *ctx) > char *path = NULL; > > while (e = av_dict_iterate(dict[i], e)) { > - if (av_stristr(e->key, "disable_clip")) { > - model_cfg.flags |= av_stristr(e->value, "true") ? > + if (!strcmp(e->key, "disable_clip")) { > + model_cfg.flags |= !strcmp(e->value, "true") ? > VMAF_MODEL_FLAG_DISABLE_CLIP : 0; > continue; > } > > - if (av_stristr(e->key, "enable_transform")) { > - model_cfg.flags |= av_stristr(e->value, "true") ? > + if (!strcmp(e->key, "enable_transform")) { > + model_cfg.flags |= !strcmp(e->value, "true") ? > VMAF_MODEL_FLAG_ENABLE_TRANSFORM : 0; > continue; > } > > - if (av_stristr(e->key, "name")) { > + if (!strcmp(e->key, "name")) { > model_cfg.name = e->value; > continue; > } > > - if (av_stristr(e->key, "version")) { > + if (!strcmp(e->key, "version")) { > version = e->value; > continue; > } > > - if (av_stristr(e->key, "path")) { > + if (!strcmp(e->key, "path")) { > path = e->value; > continue; > } > @@ -529,13 +529,13 @@ static int activate(AVFilterContext *ctx) > static enum VmafOutputFormat log_fmt_map(const char *log_fmt) > { > if (log_fmt) { > - if (av_stristr(log_fmt, "xml")) > + if (!strcmp(log_fmt, "xml")) > return VMAF_OUTPUT_FORMAT_XML; > - if (av_stristr(log_fmt, "json")) > + if (!strcmp(log_fmt, "json")) > return VMAF_OUTPUT_FORMAT_JSON; > - if (av_stristr(log_fmt, "csv")) > + if (!strcmp(log_fmt, "csv")) > return VMAF_OUTPUT_FORMAT_CSV; > - if (av_stristr(log_fmt, "sub")) > + if (!strcmp(log_fmt, "sub")) > return VMAF_OUTPUT_FORMAT_SUB; > } > > @@ -545,11 +545,11 @@ static enum VmafOutputFormat log_fmt_map(const > char *log_fmt) > static enum VmafPoolingMethod pool_method_map(const char *pool_method) > { > if (pool_method) { > - if (av_stristr(pool_method, "min")) > + if (!strcmp(pool_method, "min")) > return VMAF_POOL_METHOD_MIN; > - if (av_stristr(pool_method, "mean")) > + if (!strcmp(pool_method, "mean")) > return VMAF_POOL_METHOD_MEAN; > - if (av_stristr(pool_method, "harmonic_mean")) > + if (!strcmp(pool_method, "harmonic_mean")) > return VMAF_POOL_METHOD_HARMONIC_MEAN; > } > > -- > 2.37.1 (Apple Git-137.1) > _______________________________________________ > ffmpeg-devel mailing list > ffmpeg-devel@ffmpeg.org > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > To unsubscribe, visit link above, or email > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe". Will push this tomorrow. Thanks, Kyle
On Wed, Dec 6, 2023 at 9:54 AM Kyle Swanson <k@ylo.ph> wrote:
> Will push this tomorrow.
Pushed, thanks.
diff --git a/libavfilter/vf_libvmaf.c b/libavfilter/vf_libvmaf.c index 12810b7267..46ff8154ef 100644 --- a/libavfilter/vf_libvmaf.c +++ b/libavfilter/vf_libvmaf.c @@ -251,7 +251,7 @@ static int parse_features(AVFilterContext *ctx) const AVDictionaryEntry *e = NULL; while (e = av_dict_iterate(dict[i], e)) { - if (av_stristr(e->key, "name")) { + if (!strcmp(e->key, "name")) { feature_name = e->value; continue; } @@ -312,29 +312,29 @@ static int parse_models(AVFilterContext *ctx) char *path = NULL; while (e = av_dict_iterate(dict[i], e)) { - if (av_stristr(e->key, "disable_clip")) { - model_cfg.flags |= av_stristr(e->value, "true") ? + if (!strcmp(e->key, "disable_clip")) { + model_cfg.flags |= !strcmp(e->value, "true") ? VMAF_MODEL_FLAG_DISABLE_CLIP : 0; continue; } - if (av_stristr(e->key, "enable_transform")) { - model_cfg.flags |= av_stristr(e->value, "true") ? + if (!strcmp(e->key, "enable_transform")) { + model_cfg.flags |= !strcmp(e->value, "true") ? VMAF_MODEL_FLAG_ENABLE_TRANSFORM : 0; continue; } - if (av_stristr(e->key, "name")) { + if (!strcmp(e->key, "name")) { model_cfg.name = e->value; continue; } - if (av_stristr(e->key, "version")) { + if (!strcmp(e->key, "version")) { version = e->value; continue; } - if (av_stristr(e->key, "path")) { + if (!strcmp(e->key, "path")) { path = e->value; continue; } @@ -529,13 +529,13 @@ static int activate(AVFilterContext *ctx) static enum VmafOutputFormat log_fmt_map(const char *log_fmt) { if (log_fmt) { - if (av_stristr(log_fmt, "xml")) + if (!strcmp(log_fmt, "xml")) return VMAF_OUTPUT_FORMAT_XML; - if (av_stristr(log_fmt, "json")) + if (!strcmp(log_fmt, "json")) return VMAF_OUTPUT_FORMAT_JSON; - if (av_stristr(log_fmt, "csv")) + if (!strcmp(log_fmt, "csv")) return VMAF_OUTPUT_FORMAT_CSV; - if (av_stristr(log_fmt, "sub")) + if (!strcmp(log_fmt, "sub")) return VMAF_OUTPUT_FORMAT_SUB; } @@ -545,11 +545,11 @@ static enum VmafOutputFormat log_fmt_map(const char *log_fmt) static enum VmafPoolingMethod pool_method_map(const char *pool_method) { if (pool_method) { - if (av_stristr(pool_method, "min")) + if (!strcmp(pool_method, "min")) return VMAF_POOL_METHOD_MIN; - if (av_stristr(pool_method, "mean")) + if (!strcmp(pool_method, "mean")) return VMAF_POOL_METHOD_MEAN; - if (av_stristr(pool_method, "harmonic_mean")) + if (!strcmp(pool_method, "harmonic_mean")) return VMAF_POOL_METHOD_HARMONIC_MEAN; }
The libvmaf filter was doing substring checks in place of string equality comparisons. This led to a bug when the user specified the pooling method "harmonic_mean", since "mean" was checked first and the substring comparison returned true. This patch changes all substring comparisons for string equality comparisons. This is both correct and more efficient than the existing method. Signed-off-by: nilfm <nilf@netflix.com> --- libavfilter/vf_libvmaf.c | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-)