Commit 62f47225 authored by James Almer's avatar James Almer

avcodec/av1_parse: Use av_fast_realloc() for OBU array

Based on commits 22bec0d3 and
cebb4469.
Signed-off-by: 's avatarJames Almer <jamrial@gmail.com>
parent 1e0b6bc0
...@@ -66,7 +66,11 @@ int ff_av1_packet_split(AV1Packet *pkt, const uint8_t *buf, int length, void *lo ...@@ -66,7 +66,11 @@ int ff_av1_packet_split(AV1Packet *pkt, const uint8_t *buf, int length, void *lo
if (pkt->obus_allocated < pkt->nb_obus + 1) { if (pkt->obus_allocated < pkt->nb_obus + 1) {
int new_size = pkt->obus_allocated + 1; int new_size = pkt->obus_allocated + 1;
AV1OBU *tmp = av_realloc_array(pkt->obus, new_size, sizeof(*tmp)); AV1OBU *tmp;
if (new_size >= INT_MAX / sizeof(*tmp))
return AVERROR(ENOMEM);
tmp = av_fast_realloc(pkt->obus, &pkt->obus_allocated_size, new_size * sizeof(*tmp));
if (!tmp) if (!tmp)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
...@@ -102,5 +106,5 @@ int ff_av1_packet_split(AV1Packet *pkt, const uint8_t *buf, int length, void *lo ...@@ -102,5 +106,5 @@ int ff_av1_packet_split(AV1Packet *pkt, const uint8_t *buf, int length, void *lo
void ff_av1_packet_uninit(AV1Packet *pkt) void ff_av1_packet_uninit(AV1Packet *pkt)
{ {
av_freep(&pkt->obus); av_freep(&pkt->obus);
pkt->obus_allocated = 0; pkt->obus_allocated = pkt->obus_allocated_size = 0;
} }
...@@ -56,6 +56,7 @@ typedef struct AV1Packet { ...@@ -56,6 +56,7 @@ typedef struct AV1Packet {
AV1OBU *obus; AV1OBU *obus;
int nb_obus; int nb_obus;
int obus_allocated; int obus_allocated;
unsigned obus_allocated_size;
} AV1Packet; } AV1Packet;
/** /**
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment