From fb7b222b91825e6cd692d354b213c192d4e974a9 Mon Sep 17 00:00:00 2001 From: Paul B Mahol Date: Thu, 17 Oct 2019 11:02:02 +0200 Subject: [PATCH] avfilter/af_arnndn: make sure that vad_output nb_neurons is always 1 --- libavfilter/af_arnndn.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/libavfilter/af_arnndn.c b/libavfilter/af_arnndn.c index 781d0dc934..7688b31b0c 100644 --- a/libavfilter/af_arnndn.c +++ b/libavfilter/af_arnndn.c @@ -300,6 +300,11 @@ static RNNModel *rnnoise_model_from_file(FILE *f) INPUT_DENSE(denoise_output); INPUT_DENSE(vad_output); + if (vad_output->nb_neurons != 1) { + rnnoise_model_free(ret); + return NULL; + } + return ret; }