FFmpeg: libavformat/psxstr.c Source File

FFmpeg
psxstr.c
Go to the documentation of this file.
1 /*
2  * Sony Playstation (PSX) STR File Demuxer
3  * Copyright (c) 2003 The FFmpeg project
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 /**
23  * @file
24  * PSX STR file demuxer
25  * by Mike Melanson (melanson@pcisys.net)
26  * This module handles streams that have been ripped from Sony Playstation
27  * CD games. This demuxer can handle either raw STR files (which are just
28  * concatenations of raw compact disc sectors) or STR files with 0x2C-byte
29  * RIFF headers, followed by CD sectors.
30  */
31 
32 #include "libavutil/channel_layout.h"
33 #include "libavutil/internal.h"
34 #include "libavutil/intreadwrite.h"
35 #include "avformat.h"
36 #include "demux.h"
37 #include "internal.h"
38 
39  #define RIFF_TAG MKTAG('R', 'I', 'F', 'F')
40  #define CDXA_TAG MKTAG('C', 'D', 'X', 'A')
41 
42  #define RAW_CD_SECTOR_SIZE 2352
43  #define RAW_CD_SECTOR_DATA_SIZE 2304
44  #define VIDEO_DATA_CHUNK_SIZE 0x7E0
45  #define VIDEO_DATA_HEADER_SIZE 0x38
46  #define RIFF_HEADER_SIZE 0x2C
47 
48  #define CDXA_TYPE_MASK 0x0E
49  #define CDXA_TYPE_DATA 0x08
50  #define CDXA_TYPE_AUDIO 0x04
51  #define CDXA_TYPE_VIDEO 0x02
52  #define CDXA_TYPE_EMPTY 0x00
53 
54  #define STR_MAGIC (0x80010160)
55 
56  typedef struct StrChannel {
57  /* video parameters */
58   int video_stream_index;
59   AVPacket tmp_pkt;
60 
61  /* audio parameters */
62   int audio_stream_index;
63 } StrChannel;
64 
65  typedef struct StrDemuxContext {
66 
67  /* a STR file can contain up to 32 channels of data */
68   StrChannel channels[32];
69 } StrDemuxContext;
70 
71  static const uint8_t sync_header[12] = {0x00,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0x00};
72 
73  static int str_probe(const AVProbeData *p)
74 {
75  const uint8_t *sector= p->buf;
76  const uint8_t *end= sector + p->buf_size;
77  int aud=0, vid=0;
78 
79  if (p->buf_size < RAW_CD_SECTOR_SIZE)
80  return 0;
81 
82  if ((AV_RL32(&p->buf[0]) == RIFF_TAG) &&
83  (AV_RL32(&p->buf[8]) == CDXA_TAG)) {
84 
85  /* RIFF header seen; skip 0x2C bytes */
86  sector += RIFF_HEADER_SIZE;
87  }
88 
89  while (end - sector >= RAW_CD_SECTOR_SIZE) {
90  /* look for CD sync header (00, 0xFF x 10, 00) */
91  if (memcmp(sector,sync_header,sizeof(sync_header)))
92  return 0;
93 
94  if (sector[0x11] >= 32)
95  return 0;
96 
97  switch (sector[0x12] & CDXA_TYPE_MASK) {
98  case CDXA_TYPE_DATA:
99  case CDXA_TYPE_VIDEO: {
100  int current_sector = AV_RL16(&sector[0x1C]);
101  int sector_count = AV_RL16(&sector[0x1E]);
102  int frame_size = AV_RL32(&sector[0x24]);
103 
104  if(!( frame_size>=0
105  && current_sector < sector_count
106  && sector_count*VIDEO_DATA_CHUNK_SIZE >=frame_size)){
107  return 0;
108  }
109  vid++;
110 
111  }
112  break;
113  case CDXA_TYPE_AUDIO:
114  if(sector[0x13]&0x2A)
115  return 0;
116  aud++;
117  break;
118  default:
119  if(sector[0x12] & CDXA_TYPE_MASK)
120  return 0;
121  }
122  sector += RAW_CD_SECTOR_SIZE;
123  }
124  /* MPEG files (like those ripped from VCDs) can also look like this;
125  * only return half certainty */
126  if(vid+aud > 3) return AVPROBE_SCORE_EXTENSION;
127  else if(vid+aud) return 1;
128  else return 0;
129 }
130 
131  static int str_read_header(AVFormatContext *s)
132 {
133  AVIOContext *pb = s->pb;
134  StrDemuxContext *str = s->priv_data;
135  unsigned char sector[RAW_CD_SECTOR_SIZE];
136  int start;
137  int i;
138 
139  /* skip over any RIFF header */
140  if (avio_read(pb, sector, RIFF_HEADER_SIZE) != RIFF_HEADER_SIZE)
141  return AVERROR(EIO);
142  if (AV_RL32(&sector[0]) == RIFF_TAG)
143  start = RIFF_HEADER_SIZE;
144  else
145  start = 0;
146 
147  avio_seek(pb, start, SEEK_SET);
148 
149  for(i=0; i<32; i++){
150  str->channels[i].video_stream_index=
151  str->channels[i].audio_stream_index= -1;
152  }
153 
154  s->ctx_flags |= AVFMTCTX_NOHEADER;
155 
156  return 0;
157 }
158 
159  static int str_read_packet(AVFormatContext *s,
160  AVPacket *ret_pkt)
161 {
162  AVIOContext *pb = s->pb;
163  StrDemuxContext *str = s->priv_data;
164  unsigned char sector[RAW_CD_SECTOR_SIZE];
165  int channel, ret;
166  AVPacket *pkt;
167  AVStream *st;
168 
169  while (1) {
170  int read = avio_read(pb, sector, RAW_CD_SECTOR_SIZE);
171 
172  if (read == AVERROR_EOF)
173  return AVERROR_EOF;
174 
175  if (read != RAW_CD_SECTOR_SIZE)
176  return AVERROR(EIO);
177 
178  channel = sector[0x11];
179  if (channel >= 32)
180  return AVERROR_INVALIDDATA;
181 
182  switch (sector[0x12] & CDXA_TYPE_MASK) {
183 
184  case CDXA_TYPE_DATA:
185  case CDXA_TYPE_VIDEO:
186  {
187 
188  int current_sector = AV_RL16(&sector[0x1C]);
189  int sector_count = AV_RL16(&sector[0x1E]);
190  int frame_size = AV_RL32(&sector[0x24]);
191 
192  if(!( frame_size>=0
193  && current_sector < sector_count
194  && sector_count*VIDEO_DATA_CHUNK_SIZE >=frame_size)){
195  av_log(s, AV_LOG_ERROR, "Invalid parameters %d %d %d\n", current_sector, sector_count, frame_size);
196  break;
197  }
198 
199  if(str->channels[channel].video_stream_index < 0){
200  /* allocate a new AVStream */
201  st = avformat_new_stream(s, NULL);
202  if (!st)
203  return AVERROR(ENOMEM);
204  avpriv_set_pts_info(st, 64, 1, 15);
205 
206  str->channels[channel].video_stream_index = st->index;
207 
208  st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
209  st->codecpar->codec_id = AV_CODEC_ID_MDEC;
210  st->codecpar->codec_tag = 0; /* no fourcc */
211  st->codecpar->width = AV_RL16(&sector[0x28]);
212  st->codecpar->height = AV_RL16(&sector[0x2A]);
213  }
214 
215  /* if this is the first sector of the frame, allocate a pkt */
216  pkt = &str->channels[channel].tmp_pkt;
217 
218  if(pkt->size != sector_count*VIDEO_DATA_CHUNK_SIZE){
219  if(pkt->data)
220  av_log(s, AV_LOG_ERROR, "mismatching sector_count\n");
221  av_packet_unref(pkt);
222  ret = av_new_packet(pkt, sector_count * VIDEO_DATA_CHUNK_SIZE);
223  if (ret < 0)
224  return ret;
225  memset(pkt->data, 0, sector_count*VIDEO_DATA_CHUNK_SIZE);
226 
227  pkt->pos= avio_tell(pb) - RAW_CD_SECTOR_SIZE;
228  pkt->stream_index =
229  str->channels[channel].video_stream_index;
230  }
231 
232  memcpy(pkt->data + current_sector*VIDEO_DATA_CHUNK_SIZE,
233  sector + VIDEO_DATA_HEADER_SIZE,
234  VIDEO_DATA_CHUNK_SIZE);
235 
236  if (current_sector == sector_count-1) {
237  pkt->size= frame_size;
238  *ret_pkt = *pkt;
239  pkt->data= NULL;
240  pkt->size= -1;
241  pkt->buf = NULL;
242  return 0;
243  }
244 
245  }
246  break;
247 
248  case CDXA_TYPE_AUDIO:
249  if(str->channels[channel].audio_stream_index < 0){
250  int fmt = sector[0x13];
251  /* allocate a new AVStream */
252  st = avformat_new_stream(s, NULL);
253  if (!st)
254  return AVERROR(ENOMEM);
255 
256  str->channels[channel].audio_stream_index = st->index;
257 
258  st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
259  st->codecpar->codec_id = AV_CODEC_ID_ADPCM_XA;
260  st->codecpar->codec_tag = 0; /* no fourcc */
261  av_channel_layout_default(&st->codecpar->ch_layout, (fmt & 1) + 1);
262  st->codecpar->sample_rate = (fmt&4)?18900:37800;
263  // st->codecpar->bit_rate = 0; //FIXME;
264  st->codecpar->block_align = 128;
265 
266  avpriv_set_pts_info(st, 64, 18 * 224 / st->codecpar->ch_layout.nb_channels,
267  st->codecpar->sample_rate);
268  st->start_time = 0;
269  }
270  pkt = ret_pkt;
271  if ((ret = av_new_packet(pkt, 2304)) < 0)
272  return ret;
273  memcpy(pkt->data,sector+24,2304);
274 
275  pkt->stream_index =
276  str->channels[channel].audio_stream_index;
277  pkt->duration = 1;
278  return 0;
279  case CDXA_TYPE_EMPTY: /* CD-ROM XA, May 1991, 4.3.2.3 */
280  /* NOTE this also catches 0x80 (EOF bit) because of CDXA_TYPE_MASK */
281  /* TODO consider refactoring so as to explicitly handle each case? */
282  break;
283  default:
284  av_log(s, AV_LOG_WARNING, "Unknown sector type %02X\n", sector[0x12]);
285  /* drop the sector and move on */
286  break;
287  }
288 
289  if (avio_feof(pb))
290  return AVERROR(EIO);
291  }
292 }
293 
294  static int str_read_close(AVFormatContext *s)
295 {
296  StrDemuxContext *str = s->priv_data;
297  int i;
298  for(i=0; i<32; i++){
299  if(str->channels[i].tmp_pkt.data)
300  av_packet_unref(&str->channels[i].tmp_pkt);
301  }
302 
303  return 0;
304 }
305 
306  const FFInputFormat ff_str_demuxer = {
307  .p.name = "psxstr",
308  .p.long_name = NULL_IF_CONFIG_SMALL("Sony Playstation STR"),
309  .p.flags = AVFMT_NO_BYTE_SEEK,
310  .priv_data_size = sizeof(StrDemuxContext),
311  .read_probe = str_probe,
312  .read_header = str_read_header,
313  .read_packet = str_read_packet,
314  .read_close = str_read_close,
315 };
av_packet_unref
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:427
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:186
AVFMT_NO_BYTE_SEEK
#define AVFMT_NO_BYTE_SEEK
Format does not allow seeking by bytes.
Definition: avformat.h:487
ff_str_demuxer
const FFInputFormat ff_str_demuxer
Definition: psxstr.c:306
AVERROR
Filter the word "frame" indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:51
avformat_new_stream
AVStream * avformat_new_stream(AVFormatContext *s, const struct AVCodec *c)
Add a new stream to a media file.
AVERROR_EOF
#define AVERROR_EOF
End of file.
Definition: error.h:57
StrChannel::audio_stream_index
int audio_stream_index
Definition: psxstr.c:62
AVPacket::data
uint8_t * data
Definition: packet.h:522
AVPacket::duration
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:540
AVCodecParameters::codec_tag
uint32_t codec_tag
Additional information about the codec (corresponds to the AVI FOURCC).
Definition: codec_par.h:59
AVProbeData::buf_size
int buf_size
Size of buf except extra allocated bytes.
Definition: avformat.h:454
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:313
RIFF_HEADER_SIZE
#define RIFF_HEADER_SIZE
Definition: psxstr.c:46
avpriv_set_pts_info
void avpriv_set_pts_info(AVStream *st, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: avformat.c:853
read_close
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:143
avio_tell
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:494
str_read_packet
static int str_read_packet(AVFormatContext *s, AVPacket *ret_pkt)
Definition: psxstr.c:159
AV_CODEC_ID_MDEC
@ AV_CODEC_ID_MDEC
Definition: codec_id.h:89
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
read_packet
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_read_callback.c:41
intreadwrite.h
sync_header
static const uint8_t sync_header[12]
Definition: psxstr.c:71
s
#define s(width, name)
Definition: cbs_vp9.c:198
av_new_packet
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:98
str_probe
static int str_probe(const AVProbeData *p)
Definition: psxstr.c:73
AVInputFormat::name
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:553
AVProbeData::buf
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:453
frame_size
int frame_size
Definition: mxfenc.c:2422
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
AVCodecParameters::width
int width
Video only.
Definition: codec_par.h:134
StrDemuxContext
Definition: psxstr.c:65
channels
channels
Definition: aptx.h:31
CDXA_TYPE_VIDEO
#define CDXA_TYPE_VIDEO
Definition: psxstr.c:51
AV_RL16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:94
StrChannel
Definition: psxstr.c:56
AVFormatContext
Format I/O context.
Definition: avformat.h:1255
internal.h
AVPacket::buf
AVBufferRef * buf
A reference to the reference-counted buffer where the packet data is stored.
Definition: packet.h:505
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:766
read_header
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:550
NULL
#define NULL
Definition: coverity.c:32
AVFMTCTX_NOHEADER
#define AVFMTCTX_NOHEADER
signal that no header is present (streams are added dynamically)
Definition: avformat.h:1206
AVProbeData
This structure contains the data a format has to probe a file.
Definition: avformat.h:451
CDXA_TAG
#define CDXA_TAG
Definition: psxstr.c:40
AVCodecParameters::ch_layout
AVChannelLayout ch_layout
Audio only.
Definition: codec_par.h:180
AVPROBE_SCORE_EXTENSION
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:461
AVCodecParameters::sample_rate
int sample_rate
Audio only.
Definition: codec_par.h:184
AV_CODEC_ID_ADPCM_XA
@ AV_CODEC_ID_ADPCM_XA
Definition: codec_id.h:375
aud
static int FUNC() aud(CodedBitstreamContext *ctx, RWContext *rw, H264RawAUD *current)
Definition: cbs_h264_syntax_template.c:841
AVIOContext
Bytestream IO Context.
Definition: avio.h:160
AVPacket::size
int size
Definition: packet.h:523
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:106
StrChannel::tmp_pkt
AVPacket tmp_pkt
Definition: psxstr.c:59
RAW_CD_SECTOR_SIZE
#define RAW_CD_SECTOR_SIZE
Definition: psxstr.c:42
CDXA_TYPE_MASK
#define CDXA_TYPE_MASK
Definition: psxstr.c:48
StrChannel::video_stream_index
int video_stream_index
Definition: psxstr.c:58
FFInputFormat::p
AVInputFormat p
The public AVInputFormat.
Definition: demux.h:41
CDXA_TYPE_AUDIO
#define CDXA_TYPE_AUDIO
Definition: psxstr.c:50
VIDEO_DATA_HEADER_SIZE
#define VIDEO_DATA_HEADER_SIZE
Definition: psxstr.c:45
VIDEO_DATA_CHUNK_SIZE
#define VIDEO_DATA_CHUNK_SIZE
Definition: psxstr.c:44
av_channel_layout_default
void av_channel_layout_default(AVChannelLayout *ch_layout, int nb_channels)
Get the default channel layout for a given number of channels.
Definition: channel_layout.c:830
RIFF_TAG
#define RIFF_TAG
Definition: psxstr.c:39
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
internal.h
AVCodecParameters::height
int height
Definition: codec_par.h:135
AVCodecParameters::block_align
int block_align
Audio only.
Definition: codec_par.h:191
demux.h
ret
ret
Definition: filter_design.txt:187
AVStream
Stream structure.
Definition: avformat.h:743
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:230
avformat.h
AV_RL32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:92
AVStream::index
int index
stream index in AVFormatContext
Definition: avformat.h:749
channel_layout.h
CDXA_TYPE_EMPTY
#define CDXA_TYPE_EMPTY
Definition: psxstr.c:52
avio_read
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:611
AVPacket::stream_index
int stream_index
Definition: packet.h:524
str_read_header
static int str_read_header(AVFormatContext *s)
Definition: psxstr.c:131
StrDemuxContext::channels
StrChannel channels[32]
Definition: psxstr.c:68
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
read_probe
static int read_probe(const AVProbeData *p)
Definition: cdg.c:30
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:55
AVPacket
This structure stores compressed data.
Definition: packet.h:499
AVPacket::pos
int64_t pos
byte position in stream, -1 if unknown
Definition: packet.h:542
FFInputFormat
Definition: demux.h:37
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
AVStream::start_time
int64_t start_time
Decoding: pts of the first frame of the stream in presentation order, in stream time base.
Definition: avformat.h:792
channel
channel
Definition: ebur128.h:39
read
static uint32_t BS_FUNC() read(BSCTX *bc, unsigned int n)
Return n bits from the buffer, n has to be in the 0-32 range.
Definition: bitstream_template.h:231
CDXA_TYPE_DATA
#define CDXA_TYPE_DATA
Definition: psxstr.c:49
avio_feof
int avio_feof(AVIOContext *s)
Similar to feof() but also returns nonzero on read errors.
Definition: aviobuf.c:345
str_read_close
static int str_read_close(AVFormatContext *s)
Definition: psxstr.c:294
Generated on Thu Sep 26 2024 23:15:50 for FFmpeg by   doxygen 1.8.17

AltStyle によって変換されたページ (->オリジナル) /