avutil/pca: Check for av_malloc* failures
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
		
							parent
							
								
									4a10811617
								
							
						
					
					
						commit
						dadc43eee4
					
				@ -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;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user