rtpdec_hevc: Use av_realloc instead of av_malloc+memcpy

Signed-off-by: Martin Storsjö <martin@martin.st>
This commit is contained in:
Martin Storsjö 2014-10-15 16:21:27 +03:00
parent 752e71e74f
commit ced7238cd0
1 changed files with 5 additions and 9 deletions

View File

@ -124,24 +124,20 @@ static av_cold int hevc_sdp_parse_fmtp_config(AVFormatContext *s,
decoded_packet_size = av_base64_decode(decoded_packet, base64packet, decoded_packet_size = av_base64_decode(decoded_packet, base64packet,
sizeof(decoded_packet)); sizeof(decoded_packet));
if (decoded_packet_size > 0) { if (decoded_packet_size > 0) {
uint8_t *dest = av_malloc(decoded_packet_size + uint8_t *tmp = av_realloc(*data_ptr, decoded_packet_size +
sizeof(start_sequence) + *size_ptr); sizeof(start_sequence) + *size_ptr);
if (!dest) { if (!tmp) {
av_log(s, AV_LOG_ERROR, av_log(s, AV_LOG_ERROR,
"Unable to allocate memory for extradata!\n"); "Unable to allocate memory for extradata!\n");
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
} }
if (*size_ptr) { *data_ptr = tmp;
memcpy(dest, *data_ptr, *size_ptr);
av_free(*data_ptr);
}
memcpy(dest + *size_ptr, start_sequence, memcpy(*data_ptr + *size_ptr, start_sequence,
sizeof(start_sequence)); sizeof(start_sequence));
memcpy(dest + *size_ptr + sizeof(start_sequence), memcpy(*data_ptr + *size_ptr + sizeof(start_sequence),
decoded_packet, decoded_packet_size); decoded_packet, decoded_packet_size);
*data_ptr = dest;
*size_ptr += sizeof(start_sequence) + decoded_packet_size; *size_ptr += sizeof(start_sequence) + decoded_packet_size;
} }
} }