FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavformat/jpegxl_anim_dec.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 82 100 82.0%
Functions: 4 4 100.0%
Branches: 37 62 59.7%
Line Branch Exec Source
1 /*
2 * Animated JPEG XL Demuxer
3 * Copyright (c) 2023 Leo Izen (thebombzen)
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 * Animated JPEG XL Demuxer
25 * @see ISO/IEC 18181-1 and 18181-2
26 */
27
28 #include <stdint.h>
29 #include <string.h>
30
31 #include "libavcodec/jpegxl.h"
32 #include "libavcodec/jpegxl_parse.h"
33 #include "libavutil/intreadwrite.h"
34 #include "libavutil/opt.h"
35
36 #include "avformat.h"
37 #include "demux.h"
38 #include "internal.h"
39
40 typedef struct JXLAnimDemuxContext {
41 AVBufferRef *initial;
42 } JXLAnimDemuxContext;
43
44 7186 static int jpegxl_anim_probe(const AVProbeData *p)
45 {
46 uint8_t buffer[4096 + AV_INPUT_BUFFER_PADDING_SIZE];
47 7186 int copied = 0, ret;
48 7186 FFJXLMetadata meta = { 0 };
49
50 /* this is a raw codestream */
51
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 7183 times.
 7186 if (AV_RL16(p->buf) == FF_JPEGXL_CODESTREAM_SIGNATURE_LE) {
52 3 ret = ff_jpegxl_parse_codestream_header(p->buf, p->buf_size, &meta, 5);
53
3/4
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 2 times.
 3 if (ret >= 0 && meta.animation_offset > 0)
54 1 return AVPROBE_SCORE_MAX;
55
56 2 return 0;
57 }
58
59 /* not a JPEG XL file at all */
60
2/2
✓ Branch 0 taken 7180 times.
✓ Branch 1 taken 3 times.
 7183 if (AV_RL64(p->buf) != FF_JPEGXL_CONTAINER_SIGNATURE_LE)
61 7180 return 0;
62
63
1/2
✓ Branch 1 taken 3 times.
✗ Branch 2 not taken.
 3 if (ff_jpegxl_collect_codestream_header(p->buf, p->buf_size, buffer,
64 sizeof(buffer) - AV_INPUT_BUFFER_PADDING_SIZE, &copied) <= 0
65
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 || copied <= 0)
66 ✗ return 0;
67
68 3 ret = ff_jpegxl_parse_codestream_header(buffer, copied, &meta, 10);
69
3/4
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 2 times.
 3 if (ret >= 0 && meta.animation_offset > 0)
70 1 return AVPROBE_SCORE_MAX;
71
72 2 return 0;
73 }
74
75 2 static int jpegxl_anim_read_header(AVFormatContext *s)
76 {
77 2 JXLAnimDemuxContext *ctx = s->priv_data;
78 2 AVIOContext *pb = s->pb;
79 AVStream *st;
80 uint8_t head[256 + AV_INPUT_BUFFER_PADDING_SIZE];
81 2 const int sizeofhead = sizeof(head) - AV_INPUT_BUFFER_PADDING_SIZE;
82 2 int headsize = 0, ret;
83 2 FFJXLMetadata meta = { 0 };
84
85 2 uint64_t sig16 = avio_rl16(pb);
86
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
 2 if (sig16 == FF_JPEGXL_CODESTREAM_SIGNATURE_LE) {
87 1 AV_WL16(head, sig16);
88 1 headsize = avio_read(s->pb, head + 2, sizeofhead - 2);
89
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (headsize < 0)
90 ✗ return headsize;
91 1 headsize += 2;
92 1 ctx->initial = av_buffer_alloc(headsize);
93
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (!ctx->initial)
94 ✗ return AVERROR(ENOMEM);
95 1 memcpy(ctx->initial->data, head, headsize);
96 } else {
97 1 uint64_t sig64 = avio_rl64(pb);
98 1 sig64 = (sig64 << 16) | sig16;
99
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (sig64 != FF_JPEGXL_CONTAINER_SIGNATURE_LE)
100 ✗ return AVERROR_INVALIDDATA;
101 1 avio_skip(pb, 2); // first box always 12 bytes
102 ✗ while (1) {
103 1 int copied = 0;
104 uint8_t buf[4096];
105 1 int read = avio_read(pb, buf, sizeof(buf));
106
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (read < 0)
107 ✗ return read;
108
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
 1 if (!ctx->initial) {
109 1 ctx->initial = av_buffer_alloc(read + 12);
110
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (!ctx->initial)
111 ✗ return AVERROR(ENOMEM);
112 1 AV_WL64(ctx->initial->data, FF_JPEGXL_CONTAINER_SIGNATURE_LE);
113 1 AV_WL32(ctx->initial->data + 8, 0x0a870a0d);
114 } else {
115 /* this only should be happening zero or one times in practice */
116 ✗ if (av_buffer_realloc(&ctx->initial, ctx->initial->size + read) < 0)
117 ✗ return AVERROR(ENOMEM);
118 }
119 1 ff_jpegxl_collect_codestream_header(buf, read, head + headsize, sizeofhead - headsize, &copied);
120 1 memcpy(ctx->initial->data + (ctx->initial->size - read), buf, read);
121 1 headsize += copied;
122
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1 if (headsize >= sizeofhead || read < sizeof(buf))
123 break;
124 }
125 }
126
127 2 memset(head + headsize, 0, AV_INPUT_BUFFER_PADDING_SIZE);
128
129 /* offset in bits of the animation header */
130 2 ret = ff_jpegxl_parse_codestream_header(head, headsize, &meta, 0);
131
2/4
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 2 if (ret < 0 || meta.animation_offset <= 0)
132 ✗ return AVERROR_INVALIDDATA;
133
134 2 st = avformat_new_stream(s, NULL);
135
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (!st)
136 ✗ return AVERROR(ENOMEM);
137
138 2 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
139 2 st->codecpar->codec_id = AV_CODEC_ID_JPEGXL;
140 2 avpriv_set_pts_info(st, 1, meta.timebase.num, meta.timebase.den);
141 2 ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL;
142
143 2 return 0;
144 }
145
146 /* the decoder requires the full input file as a single packet */
147 7 static int jpegxl_anim_read_packet(AVFormatContext *s, AVPacket *pkt)
148 {
149 7 JXLAnimDemuxContext *ctx = s->priv_data;
150 7 AVIOContext *pb = s->pb;
151 int ret;
152 int64_t size;
153 7 size_t offset = 0;
154
155 7 size = avio_size(pb);
156
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
 7 if (size < 0)
157 ✗ return size;
158
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
 7 if (size > INT_MAX)
159 ✗ return AVERROR(EDOM);
160
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
 7 if (size == 0)
161 ✗ size = 4096;
162
163
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 7 if (ctx->initial && size < ctx->initial->size)
164 ✗ size = ctx->initial->size;
165
166 7 ret = av_new_packet(pkt, size);
167
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
 7 if (ret < 0)
168 ✗ return ret;
169
170
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 5 times.
 7 if (ctx->initial) {
171 2 offset = ctx->initial->size;
172 2 memcpy(pkt->data, ctx->initial->data, offset);
173 2 av_buffer_unref(&ctx->initial);
174 }
175
176 7 pkt->pos = avio_tell(pb) - offset;
177
178 7 ret = avio_read(pb, pkt->data + offset, size - offset);
179
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 2 times.
 7 if (ret < 0)
180 5 return ret;
181
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (ret < size - offset)
182 ✗ pkt->size = ret + offset;
183
184 2 return 0;
185 }
186
187 2 static int jpegxl_anim_close(AVFormatContext *s)
188 {
189 2 JXLAnimDemuxContext *ctx = s->priv_data;
190
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (ctx->initial)
191 ✗ av_buffer_unref(&ctx->initial);
192
193 2 return 0;
194 }
195
196 const FFInputFormat ff_jpegxl_anim_demuxer = {
197 .p.name = "jpegxl_anim",
198 .p.long_name = NULL_IF_CONFIG_SMALL("Animated JPEG XL"),
199 .p.flags = AVFMT_GENERIC_INDEX | AVFMT_NOTIMESTAMPS,
200 .p.mime_type = "image/jxl",
201 .p.extensions = "jxl",
202 .priv_data_size = sizeof(JXLAnimDemuxContext),
203 .read_probe = jpegxl_anim_probe,
204 .read_header = jpegxl_anim_read_header,
205 .read_packet = jpegxl_anim_read_packet,
206 .read_close = jpegxl_anim_close,
207 .flags_internal = FF_INFMT_FLAG_INIT_CLEANUP,
208 };
209