1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00

avcodec/flashsv: Avoid copying packet

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2022-03-04 23:11:01 +01:00
parent c3fea6d83b
commit 88cccd1c73

View File

@ -44,7 +44,7 @@
#include "internal.h" #include "internal.h"
typedef struct BlockInfo { typedef struct BlockInfo {
uint8_t *pos; const uint8_t *pos;
int size; int size;
} BlockInfo; } BlockInfo;
@ -59,7 +59,8 @@ typedef struct FlashSVContext {
int ver; int ver;
const uint32_t *pal; const uint32_t *pal;
int is_keyframe; int is_keyframe;
uint8_t *keyframedata; const uint8_t *keyframedata;
AVBufferRef *keyframedata_buf;
uint8_t *keyframe; uint8_t *keyframe;
BlockInfo *blocks; BlockInfo *blocks;
uint8_t *deflate_block; uint8_t *deflate_block;
@ -138,7 +139,7 @@ static av_cold int flashsv_decode_init(AVCodecContext *avctx)
return 0; return 0;
} }
static int flashsv2_prime(FlashSVContext *s, uint8_t *src, int size) static int flashsv2_prime(FlashSVContext *s, const uint8_t *src, int size)
{ {
z_stream zs; z_stream zs;
int zret; // Zlib return code int zret; // Zlib return code
@ -355,10 +356,10 @@ static int flashsv_decode_frame(AVCodecContext *avctx, void *data,
/* we care for keyframes only in Screen Video v2 */ /* we care for keyframes only in Screen Video v2 */
s->is_keyframe = (avpkt->flags & AV_PKT_FLAG_KEY) && (s->ver == 2); s->is_keyframe = (avpkt->flags & AV_PKT_FLAG_KEY) && (s->ver == 2);
if (s->is_keyframe) { if (s->is_keyframe) {
int err; int err = av_buffer_replace(&s->keyframedata_buf, avpkt->buf);
if ((err = av_reallocp(&s->keyframedata, avpkt->size)) < 0) if (err < 0)
return err; return err;
memcpy(s->keyframedata, avpkt->data, avpkt->size); s->keyframedata = avpkt->data;
} }
if(s->ver == 2 && !s->blocks) if(s->ver == 2 && !s->blocks)
s->blocks = av_mallocz((v_blocks + !!v_part) * (h_blocks + !!h_part) * s->blocks = av_mallocz((v_blocks + !!v_part) * (h_blocks + !!h_part) *
@ -566,7 +567,7 @@ static av_cold int flashsv2_decode_end(AVCodecContext *avctx)
{ {
FlashSVContext *s = avctx->priv_data; FlashSVContext *s = avctx->priv_data;
av_freep(&s->keyframedata); av_buffer_unref(&s->keyframedata_buf);
av_freep(&s->blocks); av_freep(&s->blocks);
av_freep(&s->keyframe); av_freep(&s->keyframe);
av_freep(&s->deflate_block); av_freep(&s->deflate_block);