Merge commit '0b72a47f5592ecc035edd5bac57558ad9604612d'

[ffmpeg] / libavutil / pca.c

diff --git a/libavutil/pca.c b/libavutil/pca.c
index f7ae350b48c73732886d6dcbe0c9ddd4808d3cdc..1d88ff300afb34563ab974423c4304660c348419 100644 (file)
--- 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;
 }