* dwbuiten/master: zerocodec: Fix memleak in decode_frame zerocodec: Cosmetics
Unified · Split
| @@ -23,7 +23,7 @@ · 22 unmodified lines | ||
| 23 | 23 | typedef struct { |
| 24 | 24 | AVFrame previous_frame; |
| 25 | 25 | z_stream zstream; |
| 26 | − int size; | |
| 26 | + int size; | |
| 27 | 27 | } ZeroCodecContext; |
| 28 | 28 | |
| 29 | 29 | static int zerocodec_decode_frame(AVCodecContext *avctx, void *data, |
| @@ -33,7 +33,8 @@ static int zerocodec_decode_frame(AVCodecContext *avctx, void *data, · 3 unmodified lines | ||
| 33 | 33 | AVFrame *pic = avctx->coded_frame; |
| 34 | 34 | AVFrame *prev_pic = &zc->previous_frame; |
| 35 | 35 | z_stream *zstream = &zc->zstream; |
| 36 | − uint8_t *prev = prev_pic->data[0], *dst; | |
| 36 | + uint8_t *prev = prev_pic->data[0]; | |
| 37 | + uint8_t *dst; | |
| 37 | 38 | int i, j, zret; |
| 38 | 39 | |
| 39 | 40 | pic->reference = 3; |
| @@ -43,29 +44,28 @@ static int zerocodec_decode_frame(AVCodecContext *avctx, void *data, · 3 unmodified lines | ||
| 43 | 44 | pic->pict_type = AV_PICTURE_TYPE_I; |
| 44 | 45 | } else { |
| 45 | 46 | if (!prev) { |
| 46 | − av_log(avctx, AV_LOG_ERROR, "Missing reference frame!\n"); | |
| 47 | + av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n"); | |
| 47 | 48 | return AVERROR_INVALIDDATA; |
| 48 | 49 | } |
| 49 | 50 | pic->key_frame = 0; |
| 50 | 51 | pic->pict_type = AV_PICTURE_TYPE_P; |
| 51 | 52 | } |
| 52 | 53 | |
| 54 | + zret = inflateReset(zstream); | |
| 55 | + if (zret != Z_OK) { | |
| 56 | + av_log(avctx, AV_LOG_ERROR, "Could not reset inflate: %d.\n", zret); | |
| 57 | + return AVERROR_INVALIDDATA; | |
| 58 | + } | |
| 59 | + | |
| 53 | 60 | if (avctx->get_buffer(avctx, pic) < 0) { |
| 54 | 61 | av_log(avctx, AV_LOG_ERROR, "Could not allocate buffer.\n"); |
| 55 | 62 | return AVERROR(ENOMEM); |
| 56 | 63 | } |
| 57 | 64 | |
| 58 | − zret = inflateReset(zstream); | |
| 65 | + zstream->next_in = avpkt->data; | |
| 66 | + zstream->avail_in = avpkt->size; | |
| 59 | 67 | |
| 60 | − if (zret != Z_OK) { | |
| 61 | − av_log(avctx, AV_LOG_ERROR, "Could not reset inflate: %d\n", zret); | |
| 62 | − return AVERROR(EINVAL); | |
| 63 | − } | |
| 64 | − | |
| 65 | − zstream->next_in = avpkt->data; | |
| 66 | − zstream->avail_in = avpkt->size; | |
| 67 | − | |
| 68 | − dst = pic->data[0]; | |
| 68 | + dst = pic->data[0]; | |
| 69 | 69 | |
| 70 | 70 | /** |
| 71 | 71 | * ZeroCodec has very simple interframe compression. If a value |
| @@ -75,11 +75,13 @@ static int zerocodec_decode_frame(AVCodecContext *avctx, void *data, · 3 unmodified lines | ||
| 75 | 75 | for (i = 0; i < avctx->height; i++) { |
| 76 | 76 | zstream->next_out = dst; |
| 77 | 77 | zstream->avail_out = avctx->width << 1; |
| 78 | + | |
| 78 | 79 | zret = inflate(zstream, Z_SYNC_FLUSH); |
| 79 | 80 | if (zret != Z_OK && zret != Z_STREAM_END) { |
| 81 | + avctx->release_buffer(avctx, pic); | |
| 80 | 82 | av_log(avctx, AV_LOG_ERROR, |
| 81 | − "Inflate failed with return code: %d\n", zret); | |
| 82 | − return AVERROR(EINVAL); | |
| 83 | + "Inflate failed with return code: %d.\n", zret); | |
| 84 | + return AVERROR_INVALIDDATA; | |
| 83 | 85 | } |
| 84 | 86 | |
| 85 | 87 | if (!(avpkt->flags & AV_PKT_FLAG_KEY)) |
| @@ -137,14 +139,12 @@ static av_cold int zerocodec_decode_init(AVCodecContext *avctx) · 51 unmodified lines | ||
| 137 | 139 | zstream->opaque = Z_NULL; |
| 138 | 140 | |
| 139 | 141 | zret = inflateInit(zstream); |
| 140 | − | |
| 141 | 142 | if (zret != Z_OK) { |
| 142 | − av_log(avctx, AV_LOG_ERROR, "Could not initialize inflate: %d\n", zret); | |
| 143 | + av_log(avctx, AV_LOG_ERROR, "Could not initialize inflate: %d.\n", zret); | |
| 143 | 144 | return AVERROR(ENOMEM); |
| 144 | 145 | } |
| 145 | 146 | |
| 146 | 147 | avctx->coded_frame = avcodec_alloc_frame(); |
| 147 | − | |
| 148 | 148 | if (!avctx->coded_frame) { |
| 149 | 149 | av_log(avctx, AV_LOG_ERROR, "Could not allocate frame buffer.\n"); |
| 150 | 150 | zerocodec_decode_close(avctx); |