]> git.sesse.net Git - ffmpeg/blob - libavformat/flacenc_header.c
avcodec/flacdec: Check for invalid vlcs
[ffmpeg] / libavformat / flacenc_header.c
1 /*
2  * raw FLAC muxer
3  * Copyright (C) 2009 Justin Ruggles
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
24 #include "libavcodec/flac.h"
25
26 #include "avformat.h"
27 #include "flacenc.h"
28
29 int ff_flac_write_header(AVIOContext *pb, uint8_t *extradata,
30                          int extradata_size, int last_block)
31 {
32     uint8_t header[8] = {
33         0x66, 0x4C, 0x61, 0x43, 0x00, 0x00, 0x00, 0x22
34     };
35
36     header[4] = last_block ? 0x80 : 0x00;
37
38     if (extradata_size < FLAC_STREAMINFO_SIZE)
39         return AVERROR_INVALIDDATA;
40
41     /* write "fLaC" stream marker and first metadata block header */
42     avio_write(pb, header, 8);
43
44     /* write STREAMINFO */
45     avio_write(pb, extradata, FLAC_STREAMINFO_SIZE);
46
47     return 0;
48 }
49
50 int ff_flac_is_native_layout(uint64_t channel_layout)
51 {
52     if (channel_layout == AV_CH_LAYOUT_MONO     ||
53         channel_layout == AV_CH_LAYOUT_STEREO   ||
54         channel_layout == AV_CH_LAYOUT_SURROUND ||
55         channel_layout == AV_CH_LAYOUT_QUAD     ||
56         channel_layout == AV_CH_LAYOUT_5POINT0  ||
57         channel_layout == AV_CH_LAYOUT_5POINT1  ||
58         channel_layout == AV_CH_LAYOUT_6POINT1  ||
59         channel_layout == AV_CH_LAYOUT_7POINT1)
60         return 1;
61     return 0;
62 }