blob: 6bd18f29bd719f4e5542ab512f7b81c5be354a0c
1 | /* |
2 | * MSF demuxer |
3 | * Copyright (c) 2015 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 "avformat.h" |
24 | #include "internal.h" |
25 | |
26 | static int msf_probe(AVProbeData *p) |
27 | { |
28 | if (memcmp(p->buf, "MSF", 3)) |
29 | return 0; |
30 | |
31 | if (AV_RB32(p->buf+8) <= 0) |
32 | return 0; |
33 | |
34 | if (AV_RB32(p->buf+16) <= 0) |
35 | return 0; |
36 | |
37 | if (AV_RB32(p->buf+4) > 16) |
38 | return AVPROBE_SCORE_MAX / 5; //unsupported / unknown codec |
39 | |
40 | return AVPROBE_SCORE_MAX / 3 * 2; |
41 | } |
42 | |
43 | static int msf_read_header(AVFormatContext *s) |
44 | { |
45 | unsigned codec, size; |
46 | AVStream *st; |
47 | int ret; |
48 | |
49 | avio_skip(s->pb, 4); |
50 | |
51 | st = avformat_new_stream(s, NULL); |
52 | if (!st) |
53 | return AVERROR(ENOMEM); |
54 | |
55 | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
56 | codec = avio_rb32(s->pb); |
57 | st->codecpar->channels = avio_rb32(s->pb); |
58 | if (st->codecpar->channels <= 0 || st->codecpar->channels >= INT_MAX / 1024) |
59 | return AVERROR_INVALIDDATA; |
60 | size = avio_rb32(s->pb); |
61 | st->codecpar->sample_rate = avio_rb32(s->pb); |
62 | if (st->codecpar->sample_rate <= 0) |
63 | return AVERROR_INVALIDDATA; |
64 | // avio_rb32(s->pb); /* byte flags with encoder info */ |
65 | switch (codec) { |
66 | case 0: st->codecpar->codec_id = AV_CODEC_ID_PCM_S16BE; break; |
67 | case 1: st->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE; break; |
68 | case 3: st->codecpar->block_align = 16 * st->codecpar->channels; |
69 | st->codecpar->codec_id = AV_CODEC_ID_ADPCM_PSX; break; |
70 | case 4: |
71 | case 5: |
72 | case 6: st->codecpar->block_align = (codec == 4 ? 96 : codec == 5 ? 152 : 192) * st->codecpar->channels; |
73 | ret = ff_alloc_extradata(st->codecpar, 14); |
74 | if (ret < 0) |
75 | return ret; |
76 | memset(st->codecpar->extradata, 0, st->codecpar->extradata_size); |
77 | AV_WL16(st->codecpar->extradata, 1); /* version */ |
78 | AV_WL16(st->codecpar->extradata+2, 2048 * st->codecpar->channels); /* unknown size */ |
79 | AV_WL16(st->codecpar->extradata+6, codec == 4 ? 1 : 0); /* joint stereo */ |
80 | AV_WL16(st->codecpar->extradata+8, codec == 4 ? 1 : 0); /* joint stereo (repeat?) */ |
81 | AV_WL16(st->codecpar->extradata+10, 1); |
82 | st->codecpar->codec_id = AV_CODEC_ID_ATRAC3; break; |
83 | case 7: st->need_parsing = AVSTREAM_PARSE_FULL_RAW; |
84 | st->codecpar->codec_id = AV_CODEC_ID_MP3; break; |
85 | default: |
86 | avpriv_request_sample(s, "Codec %d", codec); |
87 | return AVERROR_PATCHWELCOME; |
88 | } |
89 | st->duration = av_get_audio_frame_duration2(st->codecpar, size); |
90 | avio_skip(s->pb, 0x40 - avio_tell(s->pb)); |
91 | avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); |
92 | |
93 | return 0; |
94 | } |
95 | |
96 | static int msf_read_packet(AVFormatContext *s, AVPacket *pkt) |
97 | { |
98 | AVCodecParameters *par = s->streams[0]->codecpar; |
99 | |
100 | return av_get_packet(s->pb, pkt, par->block_align ? par->block_align : 1024 * par->channels); |
101 | } |
102 | |
103 | AVInputFormat ff_msf_demuxer = { |
104 | .name = "msf", |
105 | .long_name = NULL_IF_CONFIG_SMALL("Sony PS3 MSF"), |
106 | .read_probe = msf_probe, |
107 | .read_header = msf_read_header, |
108 | .read_packet = msf_read_packet, |
109 | .extensions = "msf", |
110 | }; |
111 |