GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavcodec/msrle.c Lines: 33 64 51.6 %
Date: 2021-04-18 21:26:34 Branches: 11 32 34.4 %

Line Branch Exec Source
1
/*
2
 * Microsoft RLE video decoder
3
 * Copyright (C) 2003 The FFmpeg project
4
 *
5
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg is free software; you can redistribute it and/or
8
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21
22
/**
23
 * @file
24
 * MS RLE video decoder by Mike Melanson (melanson@pcisys.net)
25
 * For more information about the MS RLE format, visit:
26
 *   http://www.pcisys.net/~melanson/codecs/
27
 *
28
 * The MS RLE decoder outputs PAL8 colorspace data.
29
 */
30
31
#include <stdio.h>
32
#include <stdlib.h>
33
#include <string.h>
34
35
#include "avcodec.h"
36
#include "decode.h"
37
#include "internal.h"
38
#include "msrledec.h"
39
#include "libavutil/imgutils.h"
40
41
typedef struct MsrleContext {
42
    AVCodecContext *avctx;
43
    AVFrame *frame;
44
45
    GetByteContext gb;
46
47
    uint32_t pal[256];
48
} MsrleContext;
49
50
2
static av_cold int msrle_decode_init(AVCodecContext *avctx)
51
{
52
2
    MsrleContext *s = avctx->priv_data;
53
    int i;
54
55
2
    s->avctx = avctx;
56
57

2
    switch (avctx->bits_per_coded_sample) {
58
    case 1:
59
        avctx->pix_fmt = AV_PIX_FMT_MONOWHITE;
60
        break;
61
2
    case 4:
62
    case 8:
63
2
        avctx->pix_fmt = AV_PIX_FMT_PAL8;
64
2
        break;
65
    case 24:
66
        avctx->pix_fmt = AV_PIX_FMT_BGR24;
67
        break;
68
    default:
69
        av_log(avctx, AV_LOG_ERROR, "unsupported bits per sample\n");
70
        return AVERROR_INVALIDDATA;
71
    }
72
73
2
    s->frame = av_frame_alloc();
74
2
    if (!s->frame)
75
        return AVERROR(ENOMEM);
76
77
2
    if (avctx->extradata_size >= 4)
78
36
        for (i = 0; i < FFMIN(avctx->extradata_size, AVPALETTE_SIZE)/4; i++)
79
34
            s->pal[i] = 0xFFU<<24 | AV_RL32(avctx->extradata+4*i);
80
81
2
    return 0;
82
}
83
84
29
static int msrle_decode_frame(AVCodecContext *avctx,
85
                              void *data, int *got_frame,
86
                              AVPacket *avpkt)
87
{
88
29
    const uint8_t *buf = avpkt->data;
89
29
    int buf_size = avpkt->size;
90
29
    MsrleContext *s = avctx->priv_data;
91
29
    int istride = FFALIGN(avctx->width*avctx->bits_per_coded_sample, 32) / 8;
92
    int ret;
93
94
29
    if (buf_size < 2) //Minimally a end of picture code should be there
95
        return AVERROR_INVALIDDATA;
96
97
29
    if ((ret = ff_reget_buffer(avctx, s->frame, 0)) < 0)
98
        return ret;
99
100

29
    if (avctx->bits_per_coded_sample > 1 && avctx->bits_per_coded_sample <= 8) {
101
29
        s->frame->palette_has_changed = ff_copy_palette(s->pal, avpkt, avctx);
102
103
        /* make the palette available */
104
29
        memcpy(s->frame->data[1], s->pal, AVPALETTE_SIZE);
105
    }
106
107
    /* FIXME how to correctly detect RLE ??? */
108
29
    if (avctx->height * istride == avpkt->size) { /* assume uncompressed */
109
        int linesize = av_image_get_linesize(avctx->pix_fmt, avctx->width, 0);
110
        uint8_t *ptr = s->frame->data[0];
111
        uint8_t *buf = avpkt->data + (avctx->height-1)*istride;
112
        int i, j;
113
114
        if (linesize < 0)
115
            return linesize;
116
117
        for (i = 0; i < avctx->height; i++) {
118
            if (avctx->bits_per_coded_sample == 4) {
119
                for (j = 0; j < avctx->width - 1; j += 2) {
120
                    ptr[j+0] = buf[j>>1] >> 4;
121
                    ptr[j+1] = buf[j>>1] & 0xF;
122
                }
123
                if (avctx->width & 1)
124
                    ptr[j+0] = buf[j>>1] >> 4;
125
            } else {
126
                memcpy(ptr, buf, linesize);
127
            }
128
            buf -= istride;
129
            ptr += s->frame->linesize[0];
130
        }
131
    } else {
132
29
        bytestream2_init(&s->gb, buf, buf_size);
133
29
        ff_msrle_decode(avctx, s->frame, avctx->bits_per_coded_sample, &s->gb);
134
    }
135
136
29
    if ((ret = av_frame_ref(data, s->frame)) < 0)
137
        return ret;
138
139
29
    *got_frame      = 1;
140
141
    /* report that the buffer was completely consumed */
142
29
    return buf_size;
143
}
144
145
static void msrle_decode_flush(AVCodecContext *avctx)
146
{
147
    MsrleContext *s = avctx->priv_data;
148
149
    av_frame_unref(s->frame);
150
}
151
152
2
static av_cold int msrle_decode_end(AVCodecContext *avctx)
153
{
154
2
    MsrleContext *s = avctx->priv_data;
155
156
    /* release the last frame */
157
2
    av_frame_free(&s->frame);
158
159
2
    return 0;
160
}
161
162
AVCodec ff_msrle_decoder = {
163
    .name           = "msrle",
164
    .long_name      = NULL_IF_CONFIG_SMALL("Microsoft RLE"),
165
    .type           = AVMEDIA_TYPE_VIDEO,
166
    .id             = AV_CODEC_ID_MSRLE,
167
    .priv_data_size = sizeof(MsrleContext),
168
    .init           = msrle_decode_init,
169
    .close          = msrle_decode_end,
170
    .decode         = msrle_decode_frame,
171
    .flush          = msrle_decode_flush,
172
    .capabilities   = AV_CODEC_CAP_DR1,
173
};