1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-02-04 06:08:26 +02:00

avcodec/wmavoice: Check operations that can fail

There might be segfaults on failure.

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2021-04-03 17:45:17 +02:00
parent a5b2f06b0c
commit e93875b756

View File

@ -367,7 +367,7 @@ static av_cold void wmavoice_flush(AVCodecContext *ctx)
static av_cold int wmavoice_decode_init(AVCodecContext *ctx) static av_cold int wmavoice_decode_init(AVCodecContext *ctx)
{ {
static AVOnce init_static_once = AV_ONCE_INIT; static AVOnce init_static_once = AV_ONCE_INIT;
int n, flags, pitch_range, lsp16_flag; int n, flags, pitch_range, lsp16_flag, ret;
WMAVoiceContext *s = ctx->priv_data; WMAVoiceContext *s = ctx->priv_data;
ff_thread_once(&init_static_once, wmavoice_init_static_data); ff_thread_once(&init_static_once, wmavoice_init_static_data);
@ -395,10 +395,11 @@ static av_cold int wmavoice_decode_init(AVCodecContext *ctx)
s->spillover_bitsize = 3 + av_ceil_log2(ctx->block_align); s->spillover_bitsize = 3 + av_ceil_log2(ctx->block_align);
s->do_apf = flags & 0x1; s->do_apf = flags & 0x1;
if (s->do_apf) { if (s->do_apf) {
ff_rdft_init(&s->rdft, 7, DFT_R2C); if ((ret = ff_rdft_init(&s->rdft, 7, DFT_R2C)) < 0 ||
ff_rdft_init(&s->irdft, 7, IDFT_C2R); (ret = ff_rdft_init(&s->irdft, 7, IDFT_C2R)) < 0 ||
ff_dct_init(&s->dct, 6, DCT_I); (ret = ff_dct_init (&s->dct, 6, DCT_I)) < 0 ||
ff_dct_init(&s->dst, 6, DST_I); (ret = ff_dct_init (&s->dst, 6, DST_I)) < 0)
return ret;
ff_sine_window_init(s->cos, 256); ff_sine_window_init(s->cos, 256);
memcpy(&s->sin[255], s->cos, 256 * sizeof(s->cos[0])); memcpy(&s->sin[255], s->cos, 256 * sizeof(s->cos[0]));