diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2015-03-30 04:37:42 +0200 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2015-04-22 12:27:24 +0200 |
commit | 44d447f31cf97388b644e221cc60aa89083b7918 (patch) | |
tree | a2051bb64eb6916cbfdf3bac93d0a36baac9f58a | |
parent | 99a69249837079417ca8bec6dd0515ca996a748e (diff) | |
download | ffmpeg-44d447f31cf97388b644e221cc60aa89083b7918.tar.gz |
avutil/pca: Check for av_malloc* failures
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
(cherry picked from commit dadc43eee4d9036aa532665a04720238cc15e922)
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
-rw-r--r-- | libavutil/pca.c | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/libavutil/pca.c b/libavutil/pca.c index 311b6bc9cb..a745136e1d 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(sizeof(*pca->z) * n); 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; } |