Unified · Split
| @@ -364,6 +364,8 @@ av_cold int swri_rematrix_init(SwrContext *s){ · 363 unmodified lines | ||
| 364 | 364 | if (s->midbuf.fmt == AV_SAMPLE_FMT_S16P){ |
| 365 | 365 | s->native_matrix = av_calloc(nb_in * nb_out, sizeof(int)); |
| 366 | 366 | s->native_one = av_mallocz(sizeof(int)); |
| 367 | + if (!s->native_matrix || !s->native_one) | |
| 368 | + return AVERROR(ENOMEM); | |
| 367 | 369 | for (i = 0; i < nb_out; i++) |
| 368 | 370 | for (j = 0; j < nb_in; j++) |
| 369 | 371 | ((int*)s->native_matrix)[i * nb_in + j] = lrintf(s->matrix[i][j] * 32768); |
| @@ -374,6 +376,8 @@ av_cold int swri_rematrix_init(SwrContext *s){ · 4 unmodified lines | ||
| 374 | 376 | }else if(s->midbuf.fmt == AV_SAMPLE_FMT_FLTP){ |
| 375 | 377 | s->native_matrix = av_calloc(nb_in * nb_out, sizeof(float)); |
| 376 | 378 | s->native_one = av_mallocz(sizeof(float)); |
| 379 | + if (!s->native_matrix || !s->native_one) | |
| 380 | + return AVERROR(ENOMEM); | |
| 377 | 381 | for (i = 0; i < nb_out; i++) |
| 378 | 382 | for (j = 0; j < nb_in; j++) |
| 379 | 383 | ((float*)s->native_matrix)[i * nb_in + j] = s->matrix[i][j]; |
| @@ -384,6 +388,8 @@ av_cold int swri_rematrix_init(SwrContext *s){ · 4 unmodified lines | ||
| 384 | 388 | }else if(s->midbuf.fmt == AV_SAMPLE_FMT_DBLP){ |
| 385 | 389 | s->native_matrix = av_calloc(nb_in * nb_out, sizeof(double)); |
| 386 | 390 | s->native_one = av_mallocz(sizeof(double)); |
| 391 | + if (!s->native_matrix || !s->native_one) | |
| 392 | + return AVERROR(ENOMEM); | |
| 387 | 393 | for (i = 0; i < nb_out; i++) |
| 388 | 394 | for (j = 0; j < nb_in; j++) |
| 389 | 395 | ((double*)s->native_matrix)[i * nb_in + j] = s->matrix[i][j]; |
| @@ -395,6 +401,8 @@ av_cold int swri_rematrix_init(SwrContext *s){ · 5 unmodified lines | ||
| 395 | 401 | // Only for dithering currently |
| 396 | 402 | // s->native_matrix = av_calloc(nb_in * nb_out, sizeof(double)); |
| 397 | 403 | s->native_one = av_mallocz(sizeof(int)); |
| 404 | + if (!s->native_one) | |
| 405 | + return AVERROR(ENOMEM); | |
| 398 | 406 | // for (i = 0; i < nb_out; i++) |
| 399 | 407 | // for (j = 0; j < nb_in; j++) |
| 400 | 408 | // ((double*)s->native_matrix)[i * nb_in + j] = s->matrix[i][j]; |