RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

avformat/amvenc: use av_packet_alloc() to allocate packets 

Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer 2021-01-28 23:27:10 -03:00
parent 01f4d33b41
commit 82faeb56ce
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
libavformat/amvenc.c
View File

@ -62,8 +62,8 @@ typedef struct AMVContext
int32_t aframe_size; /* Expected audio frame size. */
int32_t ablock_align; /* Expected audio block align. */
AVPacket apad; /* Dummy audio packet for padding. */
AVPacket vpad; /* Most recent video frame, for padding. */
AVPacket *apad; /* Dummy audio packet for padding. */
AVPacket *vpad; /* Most recent video frame, for padding. */
/*
* Cumulative PTS values for each stream, used for the final
@ -183,16 +183,25 @@ static av_cold int amv_init(AVFormatContext *s)
}
/* Allocate and fill dummy packet so we can pad the audio. */
if ((ret = av_new_packet(&amv->apad, amv->ablock_align)) < 0)
amv->apad = av_packet_alloc();
if (!amv->apad)
return AVERROR(ENOMEM);
if ((ret = av_new_packet(amv->apad, amv->ablock_align)) < 0) {
av_packet_free(&amv->apad);
return ret;
}
amv->apad.stream_index = AMV_STREAM_AUDIO;
memset(amv->apad.data, 0, amv->ablock_align);
AV_WL32(amv->apad.data + 4, amv->aframe_size);
amv->apad->stream_index = AMV_STREAM_AUDIO;
memset(amv->apad->data, 0, amv->ablock_align);
AV_WL32(amv->apad->data + 4, amv->aframe_size);
av_init_packet(&amv->vpad);
amv->vpad.stream_index = AMV_STREAM_VIDEO;
amv->vpad.duration = 1;
amv->vpad = av_packet_alloc();
if (!amv->vpad) {
av_packet_free(&amv->apad);
return AVERROR(ENOMEM);
}
amv->vpad->stream_index = AMV_STREAM_VIDEO;
amv->vpad->duration = 1;
return 0;
}
@ -200,8 +209,8 @@ static void amv_deinit(AVFormatContext *s)
{
AMVContext *amv = s->priv_data;
av_packet_unref(&amv->apad);
av_packet_unref(&amv->vpad);
av_packet_free(&amv->apad);
av_packet_free(&amv->vpad);
}
static void amv_write_vlist(AVFormatContext *s, AVCodecParameters *par)
@ -325,9 +334,9 @@ static int amv_pad(AVFormatContext *s, AVPacket *pkt)
stream_index = (stream_index + 1) % s->nb_streams;
if (stream_index == AMV_STREAM_VIDEO)
return amv_write_packet_internal(s, &amv->vpad);
return amv_write_packet_internal(s, amv->vpad);
else if (stream_index == AMV_STREAM_AUDIO)
return amv_write_packet_internal(s, &amv->apad);
return amv_write_packet_internal(s, amv->apad);
else
av_assert0(0);
@ -348,8 +357,8 @@ static int amv_write_packet(AVFormatContext *s, AVPacket *pkt)
if (pkt->stream_index == AMV_STREAM_VIDEO) {
/* Save the last packet for padding. */
av_packet_unref(&amv->vpad);
if ((ret = av_packet_ref(&amv->vpad, pkt)) < 0)
av_packet_unref(amv->vpad);
if ((ret = av_packet_ref(amv->vpad, pkt)) < 0)
return ret;
}
@ -366,7 +375,7 @@ static int amv_write_trailer(AVFormatContext *s)
/* Pad-out one last audio frame if needed. */
if (amv->last_stream == AMV_STREAM_VIDEO) {
if ((ret = amv_write_packet_internal(s, &amv->apad)) < 0)
if ((ret = amv_write_packet_internal(s, amv->apad)) < 0)
return ret;
}