1fba95740f7a096857d2365b9627a61a2d184abc
[ffmpeg.git] / libavformat / dtshddec.c
1 /*
2  * Raw DTS-HD demuxer
3  * Copyright (c) 2012 Paul B Mahol
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/intreadwrite.h"
23 #include "libavutil/dict.h"
24 #include "avformat.h"
25
26 #define AUPR_HDR 0x415550522D484452
27 #define AUPRINFO 0x41555052494E464F
28 #define BITSHVTB 0x4249545348565442
29 #define BLACKOUT 0x424C41434B4F5554
30 #define BRANCHPT 0x4252414E43485054
31 #define BUILDVER 0x4255494C44564552
32 #define CORESSMD 0x434F524553534D44
33 #define DTSHDHDR 0x4454534844484452
34 #define EXTSS_MD 0x45585453535f4d44
35 #define FILEINFO 0x46494C45494E464F
36 #define NAVI_TBL 0x4E4156492D54424C
37 #define STRMDATA 0x5354524D44415441
38 #define TIMECODE 0x54494D45434F4445
39
40 typedef struct DTSHDDemuxContext {
41     uint64_t    data_start;
42     uint64_t    data_end;
43 } DTSHDDemuxContext;
44
45 static int dtshd_probe(AVProbeData *p)
46 {
47     if (AV_RB64(p->buf) == DTSHDHDR)
48         return AVPROBE_SCORE_MAX;
49     return 0;
50 }
51
52 static int dtshd_read_header(AVFormatContext *s)
53 {
54     DTSHDDemuxContext *dtshd = s->priv_data;
55     AVIOContext *pb = s->pb;
56     uint64_t chunk_type, chunk_size;
57     AVStream *st;
58     int ret;
59     char *value;
60
61     st = avformat_new_stream(s, NULL);
62     if (!st)
63         return AVERROR(ENOMEM);
64     st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
65     st->codecpar->codec_id   = AV_CODEC_ID_DTS;
66     st->need_parsing         = AVSTREAM_PARSE_FULL_RAW;
67
68     for (;;) {
69         chunk_type = avio_rb64(pb);
70         chunk_size = avio_rb64(pb);
71
72         if (avio_feof(pb))
73             break;
74
75         if (chunk_size < 4) {
76             av_log(s, AV_LOG_ERROR, "chunk size too small\n");
77             return AVERROR_INVALIDDATA;
78         }
79         if (chunk_size > ((uint64_t)1 << 61)) {
80             av_log(s, AV_LOG_ERROR, "chunk size too big\n");
81             return AVERROR_INVALIDDATA;
82         }
83
84         switch (chunk_type) {
85         case STRMDATA:
86             dtshd->data_start = avio_tell(pb);
87             dtshd->data_end = dtshd->data_start + chunk_size;
88             if (dtshd->data_end <= chunk_size)
89                 return AVERROR_INVALIDDATA;
90             if (!pb->seekable)
91                 return 0;
92             goto skip;
93             break;
94         case FILEINFO:
95             if (chunk_size > INT_MAX)
96                 goto skip;
97             value = av_malloc(chunk_size);
98             if (!value)
99                 goto skip;
100             avio_read(pb, value, chunk_size);
101             value[chunk_size - 1] = 0;
102             av_dict_set(&s->metadata, "fileinfo", value,
103                         AV_DICT_DONT_STRDUP_VAL);
104             break;
105         default:
106 skip:
107             ret = avio_skip(pb, chunk_size);
108             if (ret < 0)
109                 return ret;
110         };
111     }
112
113     if (!dtshd->data_end)
114         return AVERROR_EOF;
115
116     avio_seek(pb, dtshd->data_start, SEEK_SET);
117
118     return 0;
119 }
120
121 static int raw_read_packet(AVFormatContext *s, AVPacket *pkt)
122 {
123     DTSHDDemuxContext *dtshd = s->priv_data;
124     int64_t size, left;
125     int ret;
126
127     left = dtshd->data_end - avio_tell(s->pb);
128     size = FFMIN(left, 1024);
129     if (size <= 0)
130         return AVERROR_EOF;
131
132     ret = av_get_packet(s->pb, pkt, size);
133     if (ret < 0)
134         return ret;
135
136     pkt->stream_index = 0;
137
138     return ret;
139 }
140
141 AVInputFormat ff_dtshd_demuxer = {
142     .name           = "dtshd",
143     .long_name      = NULL_IF_CONFIG_SMALL("raw DTS-HD"),
144     .priv_data_size = sizeof(DTSHDDemuxContext),
145     .read_probe     = dtshd_probe,
146     .read_header    = dtshd_read_header,
147     .read_packet    = raw_read_packet,
148     .flags          = AVFMT_GENERIC_INDEX,
149     .extensions     = "dtshd",
150     .raw_codec_id   = AV_CODEC_ID_DTS,
151 };