FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/xwddec.c
Date: 2024-04-26 14:42:52
Exec Total Coverage
Lines: 97 143 67.8%
Functions: 1 1 100.0%
Branches: 80 160 50.0%
Line Branch Exec Source
1 /*
2 * XWD image format
3 *
4 * Copyright (c) 2012 Paul B Mahol
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 #include <inttypes.h>
24
25 #include "libavutil/imgutils.h"
26 #include "avcodec.h"
27 #include "bytestream.h"
28 #include "codec_internal.h"
29 #include "decode.h"
30 #include "xwd.h"
31
32 138 static int xwd_decode_frame(AVCodecContext *avctx, AVFrame *p,
33 int *got_frame, AVPacket *avpkt)
34 {
35 uint32_t version, header_size, vclass, ncolors;
36 uint32_t xoffset, be, bpp, lsize, rsize;
37 uint32_t pixformat, pixdepth, bunit, bitorder, bpad;
38 uint32_t rgb[3];
39 uint8_t *ptr;
40 int width, height;
41 GetByteContext gb;
42 int ret;
43
44
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (avpkt->size < XWD_HEADER_SIZE)
45 ✗ return AVERROR_INVALIDDATA;
46
47 138 bytestream2_init(&gb, avpkt->data, avpkt->size);
48 138 header_size = bytestream2_get_be32u(&gb);
49
50 138 version = bytestream2_get_be32u(&gb);
51
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (version != XWD_VERSION) {
52 ✗ av_log(avctx, AV_LOG_ERROR, "unsupported version\n");
53 ✗ return AVERROR_INVALIDDATA;
54 }
55
56
2/4
✓ Branch 0 taken 138 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 138 times.
 138 if (avpkt->size < header_size || header_size < XWD_HEADER_SIZE) {
57 ✗ av_log(avctx, AV_LOG_ERROR, "invalid header size\n");
58 ✗ return AVERROR_INVALIDDATA;
59 }
60
61 138 pixformat = bytestream2_get_be32u(&gb);
62 138 pixdepth = bytestream2_get_be32u(&gb);
63 138 width = bytestream2_get_be32u(&gb);
64 138 height = bytestream2_get_be32u(&gb);
65 138 xoffset = bytestream2_get_be32u(&gb);
66 138 be = bytestream2_get_be32u(&gb);
67 138 bunit = bytestream2_get_be32u(&gb);
68 138 bitorder = bytestream2_get_be32u(&gb);
69 138 bpad = bytestream2_get_be32u(&gb);
70 138 bpp = bytestream2_get_be32u(&gb);
71 138 lsize = bytestream2_get_be32u(&gb);
72 138 vclass = bytestream2_get_be32u(&gb);
73 138 rgb[0] = bytestream2_get_be32u(&gb);
74 138 rgb[1] = bytestream2_get_be32u(&gb);
75 138 rgb[2] = bytestream2_get_be32u(&gb);
76 138 bytestream2_skipu(&gb, 8);
77 138 ncolors = bytestream2_get_be32u(&gb);
78 138 bytestream2_skipu(&gb, header_size - (XWD_HEADER_SIZE - 20));
79
80
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 138 times.
 138 if ((ret = ff_set_dimensions(avctx, width, height)) < 0)
81 ✗ return ret;
82
83 138 av_log(avctx, AV_LOG_DEBUG,
84 "pixformat %"PRIu32", pixdepth %"PRIu32", bunit %"PRIu32", bitorder %"PRIu32", bpad %"PRIu32"\n",
85 pixformat, pixdepth, bunit, bitorder, bpad);
86 138 av_log(avctx, AV_LOG_DEBUG,
87 "vclass %"PRIu32", ncolors %"PRIu32", bpp %"PRIu32", be %"PRIu32", lsize %"PRIu32", xoffset %"PRIu32"\n",
88 vclass, ncolors, bpp, be, lsize, xoffset);
89 138 av_log(avctx, AV_LOG_DEBUG,
90 "red %0"PRIx32", green %0"PRIx32", blue %0"PRIx32"\n",
91 rgb[0], rgb[1], rgb[2]);
92
93
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (pixformat > XWD_Z_PIXMAP) {
94 ✗ av_log(avctx, AV_LOG_ERROR, "invalid pixmap format\n");
95 ✗ return AVERROR_INVALIDDATA;
96 }
97
98
2/4
✓ Branch 0 taken 138 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 138 times.
 138 if (pixdepth == 0 || pixdepth > 32) {
99 ✗ av_log(avctx, AV_LOG_ERROR, "invalid pixmap depth\n");
100 ✗ return AVERROR_INVALIDDATA;
101 }
102
103
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (xoffset) {
104 ✗ avpriv_request_sample(avctx, "xoffset %"PRIu32"", xoffset);
105 ✗ return AVERROR_PATCHWELCOME;
106 }
107
108
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (be > 1) {
109 ✗ av_log(avctx, AV_LOG_ERROR, "invalid byte order\n");
110 ✗ return AVERROR_INVALIDDATA;
111 }
112
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (bitorder > 1) {
114 ✗ av_log(avctx, AV_LOG_ERROR, "invalid bitmap bit order\n");
115 ✗ return AVERROR_INVALIDDATA;
116 }
117
118
3/6
✓ Branch 0 taken 138 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 138 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 138 times.
 138 if (bunit != 8 && bunit != 16 && bunit != 32) {
119 ✗ av_log(avctx, AV_LOG_ERROR, "invalid bitmap unit\n");
120 ✗ return AVERROR_INVALIDDATA;
121 }
122
123
5/6
✓ Branch 0 taken 82 times.
✓ Branch 1 taken 56 times.
✓ Branch 2 taken 54 times.
✓ Branch 3 taken 28 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 54 times.
 138 if (bpad != 8 && bpad != 16 && bpad != 32) {
124 ✗ av_log(avctx, AV_LOG_ERROR, "invalid bitmap scan-line pad\n");
125 ✗ return AVERROR_INVALIDDATA;
126 }
127
128
2/4
✓ Branch 0 taken 138 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 138 times.
 138 if (bpp == 0 || bpp > 32) {
129 ✗ av_log(avctx, AV_LOG_ERROR, "invalid bits per pixel\n");
130 ✗ return AVERROR_INVALIDDATA;
131 }
132
133
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (ncolors > 256) {
134 ✗ av_log(avctx, AV_LOG_ERROR, "invalid number of entries in colormap\n");
135 ✗ return AVERROR_INVALIDDATA;
136 }
137
138
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 138 times.
 138 if ((ret = av_image_check_size(avctx->width, avctx->height, 0, NULL)) < 0)
139 ✗ return ret;
140
141 138 rsize = FFALIGN(avctx->width * bpp, bpad) / 8;
142
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (lsize < rsize) {
143 ✗ av_log(avctx, AV_LOG_ERROR, "invalid bytes per scan-line\n");
144 ✗ return AVERROR_INVALIDDATA;
145 }
146
147
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 138 times.
 138 if (bytestream2_get_bytes_left(&gb) < ncolors * XWD_CMAP_SIZE + (uint64_t)avctx->height * lsize) {
148 ✗ av_log(avctx, AV_LOG_ERROR, "input buffer too small\n");
149 ✗ return AVERROR_INVALIDDATA;
150 }
151
152
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (pixformat != XWD_Z_PIXMAP) {
153 ✗ avpriv_report_missing_feature(avctx, "Pixmap format %"PRIu32, pixformat);
154 ✗ return AVERROR_PATCHWELCOME;
155 }
156
157 138 avctx->pix_fmt = AV_PIX_FMT_NONE;
158
3/4
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 82 times.
✗ Branch 3 not taken.
 138 switch (vclass) {
159 28 case XWD_STATIC_GRAY:
160 case XWD_GRAY_SCALE:
161
3/4
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 14 times.
 28 if (bpp != 1 && bpp != 8)
162 ✗ return AVERROR_INVALIDDATA;
163
3/4
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
 28 if (bpp == 1 && pixdepth == 1) {
164 14 avctx->pix_fmt = AV_PIX_FMT_MONOWHITE;
165
2/4
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
 14 } else if (bpp == 8 && pixdepth == 8) {
166 14 avctx->pix_fmt = AV_PIX_FMT_GRAY8;
167 }
168 28 break;
169 28 case XWD_STATIC_COLOR:
170 case XWD_PSEUDO_COLOR:
171
1/2
✓ Branch 0 taken 28 times.
✗ Branch 1 not taken.
 28 if (bpp == 8)
172 28 avctx->pix_fmt = AV_PIX_FMT_PAL8;
173 28 break;
174 82 case XWD_TRUE_COLOR:
175 case XWD_DIRECT_COLOR:
176
5/6
✓ Branch 0 taken 54 times.
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 14 times.
✓ Branch 3 taken 40 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 14 times.
 82 if (bpp != 16 && bpp != 24 && bpp != 32)
177 ✗ return AVERROR_INVALIDDATA;
178
4/4
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 54 times.
✓ Branch 2 taken 14 times.
✓ Branch 3 taken 14 times.
 82 if (bpp == 16 && pixdepth == 15) {
179
3/6
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 14 times.
✗ Branch 5 not taken.
 14 if (rgb[0] == 0x7C00 && rgb[1] == 0x3E0 && rgb[2] == 0x1F)
180
1/2
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
 14 avctx->pix_fmt = be ? AV_PIX_FMT_RGB555BE : AV_PIX_FMT_RGB555LE;
181 ✗ else if (rgb[0] == 0x1F && rgb[1] == 0x3E0 && rgb[2] == 0x7C00)
182 ✗ avctx->pix_fmt = be ? AV_PIX_FMT_BGR555BE : AV_PIX_FMT_BGR555LE;
183
3/4
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 54 times.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
 68 } else if (bpp == 16 && pixdepth == 16) {
184
3/6
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 14 times.
✗ Branch 5 not taken.
 14 if (rgb[0] == 0xF800 && rgb[1] == 0x7E0 && rgb[2] == 0x1F)
185
1/2
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
 14 avctx->pix_fmt = be ? AV_PIX_FMT_RGB565BE : AV_PIX_FMT_RGB565LE;
186 ✗ else if (rgb[0] == 0x1F && rgb[1] == 0x7E0 && rgb[2] == 0xF800)
187 ✗ avctx->pix_fmt = be ? AV_PIX_FMT_BGR565BE : AV_PIX_FMT_BGR565LE;
188
2/2
✓ Branch 0 taken 40 times.
✓ Branch 1 taken 14 times.
 54 } else if (bpp == 24) {
189
3/6
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 40 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 40 times.
✗ Branch 5 not taken.
 40 if (rgb[0] == 0xFF0000 && rgb[1] == 0xFF00 && rgb[2] == 0xFF)
190
1/2
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
 40 avctx->pix_fmt = be ? AV_PIX_FMT_RGB24 : AV_PIX_FMT_BGR24;
191 ✗ else if (rgb[0] == 0xFF && rgb[1] == 0xFF00 && rgb[2] == 0xFF0000)
192 ✗ avctx->pix_fmt = be ? AV_PIX_FMT_BGR24 : AV_PIX_FMT_RGB24;
193
1/2
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
 14 } else if (bpp == 32) {
194
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 14 if (rgb[0] == 0xFF0000 && rgb[1] == 0xFF00 && rgb[2] == 0xFF)
195 ✗ avctx->pix_fmt = be ? AV_PIX_FMT_ARGB : AV_PIX_FMT_BGRA;
196
3/6
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 14 times.
✗ Branch 5 not taken.
 14 else if (rgb[0] == 0xFF && rgb[1] == 0xFF00 && rgb[2] == 0xFF0000)
197
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
 14 avctx->pix_fmt = be ? AV_PIX_FMT_ABGR : AV_PIX_FMT_RGBA;
198 }
199 82 bytestream2_skipu(&gb, ncolors * XWD_CMAP_SIZE);
200 82 break;
201 ✗ default:
202 ✗ av_log(avctx, AV_LOG_ERROR, "invalid visual class\n");
203 ✗ return AVERROR_INVALIDDATA;
204 }
205
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138 times.
 138 if (avctx->pix_fmt == AV_PIX_FMT_NONE) {
207 ✗ avpriv_request_sample(avctx,
208 "Unknown file: bpp %"PRIu32", pixdepth %"PRIu32", vclass %"PRIu32"",
209 bpp, pixdepth, vclass);
210 ✗ return AVERROR_PATCHWELCOME;
211 }
212
213
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 129 times.
 138 if (avctx->skip_frame >= AVDISCARD_ALL)
214 9 return avpkt->size;
215
216
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 129 times.
 129 if ((ret = ff_get_buffer(avctx, p, 0)) < 0)
217 ✗ return ret;
218
219 129 p->flags |= AV_FRAME_FLAG_KEY;
220 129 p->pict_type = AV_PICTURE_TYPE_I;
221
222
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 103 times.
 129 if (avctx->pix_fmt == AV_PIX_FMT_PAL8) {
223 26 uint32_t *dst = (uint32_t *)p->data[1];
224 uint8_t red, green, blue;
225
226
2/2
✓ Branch 0 taken 6656 times.
✓ Branch 1 taken 26 times.
 6682 for (int i = 0; i < ncolors; i++) {
227 6656 bytestream2_skipu(&gb, 4); // skip colormap entry number
228 6656 red = bytestream2_get_byteu(&gb);
229 6656 bytestream2_skipu(&gb, 1);
230 6656 green = bytestream2_get_byteu(&gb);
231 6656 bytestream2_skipu(&gb, 1);
232 6656 blue = bytestream2_get_byteu(&gb);
233 6656 bytestream2_skipu(&gb, 3); // skip bitmask flag and padding
234
235 6656 dst[i] = 0xFFU << 24 | red << 16 | green << 8 | blue;
236 }
237 }
238
239 129 ptr = p->data[0];
240
2/2
✓ Branch 0 taken 37152 times.
✓ Branch 1 taken 129 times.
 37281 for (int i = 0; i < avctx->height; i++) {
241 37152 bytestream2_get_bufferu(&gb, ptr, rsize);
242 37152 bytestream2_skipu(&gb, lsize - rsize);
243 37152 ptr += p->linesize[0];
244 }
245
246 129 *got_frame = 1;
247
248 129 return avpkt->size;
249 }
250
251 const FFCodec ff_xwd_decoder = {
252 .p.name = "xwd",
253 CODEC_LONG_NAME("XWD (X Window Dump) image"),
254 .p.type = AVMEDIA_TYPE_VIDEO,
255 .p.id = AV_CODEC_ID_XWD,
256 .p.capabilities = AV_CODEC_CAP_DR1,
257 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
258 FF_CODEC_DECODE_CB(xwd_decode_frame),
259 };
260