FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavformat/aacdec.c
Date: 2022-12-05 03:11:11
Exec Total Coverage
Lines: 93 105 88.6%
Functions: 5 5 100.0%
Branches: 46 60 76.7%

Line Branch Exec Source
1 /*
2 * raw ADTS AAC demuxer
3 * Copyright (c) 2008 Michael Niedermayer <michaelni@gmx.at>
4 * Copyright (c) 2009 Robert Swain ( rob opendot cl )
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 "libavutil/avassert.h"
24 #include "libavutil/intreadwrite.h"
25 #include "avformat.h"
26 #include "avio_internal.h"
27 #include "internal.h"
28 #include "id3v1.h"
29 #include "id3v2.h"
30 #include "apetag.h"
31
32 #define ADTS_HEADER_SIZE 7
33
34 6782 static int adts_aac_probe(const AVProbeData *p)
35 {
36 6782 int max_frames = 0, first_frames = 0;
37 int fsize, frames;
38 6782 const uint8_t *buf0 = p->buf;
39 const uint8_t *buf2;
40 const uint8_t *buf;
41 6782 const uint8_t *end = buf0 + p->buf_size - 7;
42
43 6782 buf = buf0;
44
45
2/2
✓ Branch 0 taken 92973949 times.
✓ Branch 1 taken 6782 times.
92980731 for (; buf < end; buf = buf2 + 1) {
46 92973949 buf2 = buf;
47
48
2/2
✓ Branch 0 taken 92986659 times.
✓ Branch 1 taken 1908 times.
92988567 for (frames = 0; buf2 < end; frames++) {
49 92986659 uint32_t header = AV_RB16(buf2);
50
2/2
✓ Branch 0 taken 92971198 times.
✓ Branch 1 taken 15461 times.
92986659 if ((header & 0xFFF6) != 0xFFF0) {
51
2/2
✓ Branch 0 taken 92964444 times.
✓ Branch 1 taken 6754 times.
92971198 if (buf != buf0) {
52 // Found something that isn't an ADTS header, starting
53 // from a position other than the start of the buffer.
54 // Discard the count we've accumulated so far since it
55 // probably was a false positive.
56 92964444 frames = 0;
57 }
58 92971198 break;
59 }
60 15461 fsize = (AV_RB32(buf2 + 3) >> 13) & 0x1FFF;
61
2/2
✓ Branch 0 taken 843 times.
✓ Branch 1 taken 14618 times.
15461 if (fsize < 7)
62 843 break;
63 14618 fsize = FFMIN(fsize, end - buf2);
64 14618 buf2 += fsize;
65 }
66 92973949 max_frames = FFMAX(max_frames, frames);
67
2/2
✓ Branch 0 taken 6782 times.
✓ Branch 1 taken 92967167 times.
92973949 if (buf == buf0)
68 6782 first_frames = frames;
69 }
70
71
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 6756 times.
6782 if (first_frames >= 3)
72 26 return AVPROBE_SCORE_EXTENSION + 1;
73
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6756 times.
6756 else if (max_frames > 100)
74 return AVPROBE_SCORE_EXTENSION;
75
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6756 times.
6756 else if (max_frames >= 3)
76 return AVPROBE_SCORE_EXTENSION / 2;
77
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 6754 times.
6756 else if (first_frames >= 1)
78 2 return 1;
79 else
80 6754 return 0;
81 }
82
83 32 static int adts_aac_resync(AVFormatContext *s)
84 {
85 uint16_t state;
86 32 int64_t start_pos = avio_tell(s->pb);
87
88 // skip data until an ADTS frame is found
89 32 state = avio_r8(s->pb);
90
2/2
✓ Branch 1 taken 402 times.
✓ Branch 2 taken 1 times.
403 while (!avio_feof(s->pb) &&
91
1/2
✓ Branch 1 taken 402 times.
✗ Branch 2 not taken.
402 (avio_tell(s->pb) - start_pos) < s->probesize) {
92 402 state = (state << 8) | avio_r8(s->pb);
93
2/2
✓ Branch 0 taken 371 times.
✓ Branch 1 taken 31 times.
402 if ((state >> 4) != 0xFFF)
94 371 continue;
95 31 avio_seek(s->pb, -2, SEEK_CUR);
96 31 break;
97 }
98
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 31 times.
32 if (s->pb->eof_reached)
99 1 return AVERROR_EOF;
100
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if ((state >> 4) != 0xFFF)
101 return AVERROR_INVALIDDATA;
102
103 31 return 0;
104 }
105
106 31 static int adts_aac_read_header(AVFormatContext *s)
107 {
108 AVStream *st;
109 int ret;
110
111 31 st = avformat_new_stream(s, NULL);
112
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (!st)
113 return AVERROR(ENOMEM);
114
115 31 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
116 31 st->codecpar->codec_id = s->iformat->raw_codec_id;
117 31 ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL_RAW;
118
119 31 ff_id3v1_read(s);
120
3/4
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 30 times.
✓ Branch 3 taken 1 times.
62 if ((s->pb->seekable & AVIO_SEEKABLE_NORMAL) &&
121 31 !av_dict_get(s->metadata, "", NULL, AV_DICT_IGNORE_SUFFIX)) {
122 30 int64_t cur = avio_tell(s->pb);
123 30 ff_ape_parse_tag(s);
124 30 avio_seek(s->pb, cur, SEEK_SET);
125 }
126
127 31 ret = adts_aac_resync(s);
128
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (ret < 0)
129 return ret;
130
131 // LCM of all possible ADTS sample rates
132 31 avpriv_set_pts_info(st, 64, 1, 28224000);
133
134 31 return 0;
135 }
136
137 3 static int handle_id3(AVFormatContext *s, AVPacket *pkt)
138 {
139 3 AVDictionary *metadata = NULL;
140 FFIOContext pb;
141 ID3v2ExtraMeta *id3v2_extra_meta;
142 int ret;
143
144 3 ret = av_append_packet(s->pb, pkt, ff_id3v2_tag_len(pkt->data) - pkt->size);
145
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (ret < 0) {
146 return ret;
147 }
148
149 3 ffio_init_context(&pb, pkt->data, pkt->size, 0, NULL, NULL, NULL, NULL);
150 3 ff_id3v2_read_dict(&pb.pub, &metadata, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta);
151
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 if ((ret = ff_id3v2_parse_priv_dict(&metadata, id3v2_extra_meta)) < 0)
152 goto error;
153
154
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (metadata) {
155
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
3 if ((ret = av_dict_copy(&s->metadata, metadata, 0)) < 0)
156 goto error;
157 3 s->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
158 }
159
160 error:
161 3 av_packet_unref(pkt);
162 3 ff_id3v2_free_extra_meta(&id3v2_extra_meta);
163 3 av_dict_free(&metadata);
164
165 3 return ret;
166 }
167
168 5903 static int adts_aac_read_packet(AVFormatContext *s, AVPacket *pkt)
169 {
170 int ret, fsize;
171
172 5906 retry:
173 5906 ret = av_get_packet(s->pb, pkt, ADTS_HEADER_SIZE);
174
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 5861 times.
5906 if (ret < 0)
175 45 return ret;
176
177
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5861 times.
5861 if (ret < ADTS_HEADER_SIZE) {
178 return AVERROR(EIO);
179 }
180
181
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 5857 times.
5861 if ((AV_RB16(pkt->data) >> 4) != 0xfff) {
182 // Parse all the ID3 headers between frames
183 4 int append = ID3v2_HEADER_SIZE - ADTS_HEADER_SIZE;
184
185 av_assert2(append > 0);
186 4 ret = av_append_packet(s->pb, pkt, append);
187
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
4 if (ret != append) {
188 return AVERROR(EIO);
189 }
190
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 3 times.
4 if (!ff_id3v2_match(pkt->data, ID3v2_DEFAULT_MAGIC)) {
191 1 av_packet_unref(pkt);
192 1 ret = adts_aac_resync(s);
193 } else
194 3 ret = handle_id3(s, pkt);
195
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
4 if (ret < 0)
196 1 return ret;
197
198 3 goto retry;
199 }
200
201 5857 fsize = (AV_RB32(pkt->data + 3) >> 13) & 0x1FFF;
202
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5857 times.
5857 if (fsize < ADTS_HEADER_SIZE) {
203 return AVERROR_INVALIDDATA;
204 }
205
206 5857 ret = av_append_packet(s->pb, pkt, fsize - pkt->size);
207
208 5857 return ret;
209 }
210
211 const AVInputFormat ff_aac_demuxer = {
212 .name = "aac",
213 .long_name = NULL_IF_CONFIG_SMALL("raw ADTS AAC (Advanced Audio Coding)"),
214 .read_probe = adts_aac_probe,
215 .read_header = adts_aac_read_header,
216 .read_packet = adts_aac_read_packet,
217 .flags = AVFMT_GENERIC_INDEX,
218 .extensions = "aac",
219 .mime_type = "audio/aac,audio/aacp,audio/x-aac",
220 .raw_codec_id = AV_CODEC_ID_AAC,
221 };
222