From dadc43eee4d9036aa532665a04720238cc15e922 Mon Sep 17 00:00:00 2001
From: Michael Niedermayer <michaelni@gmx.at>
Date: Mon, 30 Mar 2015 04:37:42 +0200
Subject: [PATCH] avutil/pca: Check for av_malloc* failures
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
---
libavutil/pca.c | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/libavutil/pca.c b/libavutil/pca.c
index f7ae350b48c..1d88ff300af 100644
--- a/libavutil/pca.c
+++ b/libavutil/pca.c
@@ -41,12 +41,20 @@ PCA *ff_pca_init(int n){
return NULL;
pca= av_mallocz(sizeof(*pca));
+ if (!pca)
+ return NULL;
+
pca->n= n;
pca->z = av_malloc_array(n, sizeof(*pca->z));
pca->count=0;
pca->covariance= av_calloc(n*n, sizeof(double));
pca->mean= av_calloc(n, sizeof(double));
+ if (!pca->z || !pca->covariance || !pca->mean) {
+ ff_pca_free(pca);
+ return NULL;
+ }
+
return pca;
}
--
GitLab