FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavformat/dsfdec.c
Date: 2024-04-25 15:36:26
Exec Total Coverage
Lines: 3 109 2.8%
Functions: 1 4 25.0%
Branches: 2 61 3.3%
Line Branch Exec Source
1 /*
2 * DSD Stream File (DSF) demuxer
3 * Copyright (c) 2014 Peter Ross
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 #include "libavutil/channel_layout.h"
23 #include "libavutil/intreadwrite.h"
24 #include "avformat.h"
25 #include "demux.h"
26 #include "internal.h"
27 #include "id3v2.h"
28
29 typedef struct {
30 uint64_t data_end;
31 uint64_t audio_size;
32 uint64_t data_size;
33 } DSFContext;
34
35 7128 static int dsf_probe(const AVProbeData *p)
36 {
37
2/6
✓ Branch 0 taken 7128 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7128 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 7128 if (p->buf_size < 12 || memcmp(p->buf, "DSD ", 4) || AV_RL64(p->buf + 4) != 28)
38 7128 return 0;
39 ✗ return AVPROBE_SCORE_MAX;
40 }
41
42 static const AVChannelLayout dsf_channel_layout[] = {
43 { .order = AV_CHANNEL_ORDER_UNSPEC },
44 AV_CHANNEL_LAYOUT_MONO,
45 AV_CHANNEL_LAYOUT_STEREO,
46 AV_CHANNEL_LAYOUT_SURROUND,
47 AV_CHANNEL_LAYOUT_QUAD,
48 AV_CHANNEL_LAYOUT_4POINT0,
49 AV_CHANNEL_LAYOUT_5POINT0_BACK,
50 AV_CHANNEL_LAYOUT_5POINT1_BACK,
51 };
52
53 ✗ static void read_id3(AVFormatContext *s, uint64_t id3pos)
54 {
55 ID3v2ExtraMeta *id3v2_extra_meta;
56 ✗ if (avio_seek(s->pb, id3pos, SEEK_SET) < 0)
57 ✗ return;
58
59 ✗ ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta, 0);
60 ✗ if (id3v2_extra_meta) {
61 ✗ ff_id3v2_parse_apic(s, id3v2_extra_meta);
62 ✗ ff_id3v2_parse_chapters(s, id3v2_extra_meta);
63 }
64 ✗ ff_id3v2_free_extra_meta(&id3v2_extra_meta);
65 }
66
67 ✗ static int dsf_read_header(AVFormatContext *s)
68 {
69 ✗ DSFContext *dsf = s->priv_data;
70 ✗ AVIOContext *pb = s->pb;
71 AVStream *st;
72 uint64_t id3pos;
73 unsigned int channel_type;
74 int channels;
75
76 ✗ avio_skip(pb, 4);
77 ✗ if (avio_rl64(pb) != 28)
78 ✗ return AVERROR_INVALIDDATA;
79
80 /* create primary stream before any id3 coverart streams */
81 ✗ st = avformat_new_stream(s, NULL);
82 ✗ if (!st)
83 ✗ return AVERROR(ENOMEM);
84
85 ✗ avio_skip(pb, 8);
86 ✗ id3pos = avio_rl64(pb);
87 ✗ if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
88 ✗ read_id3(s, id3pos);
89 ✗ avio_seek(pb, 28, SEEK_SET);
90 }
91
92 /* fmt chunk */
93
94 ✗ if (avio_rl32(pb) != MKTAG('f', 'm', 't', ' ') || avio_rl64(pb) != 52)
95 ✗ return AVERROR_INVALIDDATA;
96
97 ✗ if (avio_rl32(pb) != 1) {
98 ✗ avpriv_request_sample(s, "unknown format version");
99 ✗ return AVERROR_INVALIDDATA;
100 }
101
102 ✗ if (avio_rl32(pb)) {
103 ✗ avpriv_request_sample(s, "unknown format id");
104 ✗ return AVERROR_INVALIDDATA;
105 }
106
107 ✗ channel_type = avio_rl32(pb);
108 ✗ if (channel_type < FF_ARRAY_ELEMS(dsf_channel_layout))
109 ✗ st->codecpar->ch_layout = dsf_channel_layout[channel_type];
110 ✗ if (!st->codecpar->ch_layout.nb_channels)
111 ✗ avpriv_request_sample(s, "channel type %i", channel_type);
112
113 ✗ st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
114 ✗ channels = avio_rl32(pb);
115 ✗ if (!st->codecpar->ch_layout.nb_channels) {
116 ✗ st->codecpar->ch_layout.nb_channels = channels;
117 ✗ } else if (channels != st->codecpar->ch_layout.nb_channels) {
118 ✗ av_log(s, AV_LOG_ERROR, "Channel count mismatch\n");
119 ✗ return AVERROR(EINVAL);
120 }
121 ✗ st->codecpar->sample_rate = avio_rl32(pb) / 8;
122
123 ✗ if (st->codecpar->ch_layout.nb_channels <= 0)
124 ✗ return AVERROR_INVALIDDATA;
125
126 ✗ switch(avio_rl32(pb)) {
127 ✗ case 1: st->codecpar->codec_id = AV_CODEC_ID_DSD_LSBF_PLANAR; break;
128 ✗ case 8: st->codecpar->codec_id = AV_CODEC_ID_DSD_MSBF_PLANAR; break;
129 ✗ default:
130 ✗ avpriv_request_sample(s, "unknown most significant bit");
131 ✗ return AVERROR_INVALIDDATA;
132 }
133
134 ✗ dsf->audio_size = avio_rl64(pb) / 8 * st->codecpar->ch_layout.nb_channels;
135 ✗ st->codecpar->block_align = avio_rl32(pb);
136 ✗ if (st->codecpar->block_align > INT_MAX / st->codecpar->ch_layout.nb_channels ||
137 ✗ st->codecpar->block_align <= 0) {
138 ✗ avpriv_request_sample(s, "block_align invalid");
139 ✗ return AVERROR_INVALIDDATA;
140 }
141 ✗ st->codecpar->block_align *= st->codecpar->ch_layout.nb_channels;
142 ✗ st->codecpar->bit_rate = st->codecpar->ch_layout.nb_channels * 8LL * st->codecpar->sample_rate;
143 ✗ avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
144 ✗ avio_skip(pb, 4);
145
146 /* data chunk */
147
148 ✗ dsf->data_end = avio_tell(pb);
149 ✗ if (avio_rl32(pb) != MKTAG('d', 'a', 't', 'a'))
150 ✗ return AVERROR_INVALIDDATA;
151 ✗ dsf->data_size = avio_rl64(pb) - 12;
152 ✗ dsf->data_end += dsf->data_size + 12;
153
154 ✗ return 0;
155 }
156
157 ✗ static int dsf_read_packet(AVFormatContext *s, AVPacket *pkt)
158 {
159 ✗ FFFormatContext *const si = ffformatcontext(s);
160 ✗ DSFContext *dsf = s->priv_data;
161 ✗ AVIOContext *pb = s->pb;
162 ✗ AVStream *st = s->streams[0];
163 ✗ int64_t pos = avio_tell(pb);
164 ✗ int channels = st->codecpar->ch_layout.nb_channels;
165 int ret;
166
167 ✗ if (pos >= dsf->data_end)
168 ✗ return AVERROR_EOF;
169
170 ✗ if (dsf->data_size > dsf->audio_size) {
171 ✗ int last_packet = pos == (dsf->data_end - st->codecpar->block_align);
172
173 ✗ if (last_packet) {
174 ✗ int64_t data_pos = pos - si->data_offset;
175 ✗ int64_t packet_size = dsf->audio_size - data_pos;
176 ✗ int64_t skip_size = dsf->data_size - data_pos - packet_size;
177 uint8_t *dst;
178 int ch, ret;
179
180 ✗ if (packet_size <= 0 || skip_size <= 0)
181 ✗ return AVERROR_INVALIDDATA;
182
183 ✗ if ((ret = av_new_packet(pkt, packet_size)) < 0)
184 ✗ return ret;
185 ✗ dst = pkt->data;
186 ✗ for (ch = 0; ch < st->codecpar->ch_layout.nb_channels; ch++) {
187 ✗ ret = avio_read(pb, dst, packet_size / st->codecpar->ch_layout.nb_channels);
188 ✗ if (ret < packet_size / st->codecpar->ch_layout.nb_channels)
189 ✗ return AVERROR_EOF;
190
191 ✗ dst += ret;
192 ✗ avio_skip(pb, skip_size / st->codecpar->ch_layout.nb_channels);
193 }
194
195 ✗ pkt->pos = pos;
196 ✗ pkt->stream_index = 0;
197 ✗ pkt->pts = (pos - si->data_offset) / channels;
198 ✗ pkt->duration = packet_size / channels;
199 ✗ return 0;
200 }
201 }
202 ✗ ret = av_get_packet(pb, pkt, FFMIN(dsf->data_end - pos, st->codecpar->block_align));
203 ✗ if (ret < 0)
204 ✗ return ret;
205
206 ✗ pkt->stream_index = 0;
207 ✗ pkt->pts = (pos - si->data_offset) / channels;
208 ✗ pkt->duration = st->codecpar->block_align / channels;
209
210 ✗ return 0;
211 }
212
213 const FFInputFormat ff_dsf_demuxer = {
214 .p.name = "dsf",
215 .p.long_name = NULL_IF_CONFIG_SMALL("DSD Stream File (DSF)"),
216 .p.flags = AVFMT_GENERIC_INDEX | AVFMT_NO_BYTE_SEEK,
217 .priv_data_size = sizeof(DSFContext),
218 .read_probe = dsf_probe,
219 .read_header = dsf_read_header,
220 .read_packet = dsf_read_packet,
221 };
222