FFmpeg  4.4.4
rmdec.c
Go to the documentation of this file.
1 /*
2  * "Real" compatible demuxer.
3  * Copyright (c) 2000, 2001 Fabrice Bellard
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 <inttypes.h>
23 
24 #include "libavutil/avassert.h"
25 #include "libavutil/avstring.h"
27 #include "libavutil/internal.h"
28 #include "libavutil/intreadwrite.h"
29 #include "libavutil/dict.h"
30 #include "avformat.h"
31 #include "avio_internal.h"
32 #include "internal.h"
33 #include "rmsipr.h"
34 #include "rm.h"
35 
36 #define DEINT_ID_GENR MKTAG('g', 'e', 'n', 'r') ///< interleaving for Cooker/ATRAC
37 #define DEINT_ID_INT0 MKTAG('I', 'n', 't', '0') ///< no interleaving needed
38 #define DEINT_ID_INT4 MKTAG('I', 'n', 't', '4') ///< interleaving for 28.8
39 #define DEINT_ID_SIPR MKTAG('s', 'i', 'p', 'r') ///< interleaving for Sipro
40 #define DEINT_ID_VBRF MKTAG('v', 'b', 'r', 'f') ///< VBR case for AAC
41 #define DEINT_ID_VBRS MKTAG('v', 'b', 'r', 's') ///< VBR case for AAC
42 
43 struct RMStream {
44  AVPacket pkt; ///< place to store merged video frame / reordered audio data
45  int videobufsize; ///< current assembled frame size
46  int videobufpos; ///< position for the next slice in the video buffer
47  int curpic_num; ///< picture number of current frame
49  int64_t pktpos; ///< first slice position in file
50  /// Audio descrambling matrix parameters
51  int64_t audiotimestamp; ///< Audio packet timestamp
52  int sub_packet_cnt; // Subpacket counter, used while reading
53  int sub_packet_size, sub_packet_h, coded_framesize; ///< Descrambling parameters from container
54  int audio_framesize; /// Audio frame size from container
55  int sub_packet_lengths[16]; /// Length of each subpacket
56  int32_t deint_id; ///< deinterleaver used in audio stream
57 };
58 
59 typedef struct RMDemuxContext {
64  int audio_stream_num; ///< Stream number for audio packets
65  int audio_pkt_cnt; ///< Output packet counter
66  int data_end;
68 
69 static int rm_read_close(AVFormatContext *s);
70 
71 static inline void get_strl(AVIOContext *pb, char *buf, int buf_size, int len)
72 {
73  int read = avio_get_str(pb, len, buf, buf_size);
74 
75  if (read > 0)
76  avio_skip(pb, len - read);
77 }
78 
79 static void get_str8(AVIOContext *pb, char *buf, int buf_size)
80 {
81  get_strl(pb, buf, buf_size, avio_r8(pb));
82 }
83 
85 {
86  if (size >= 1<<24) {
87  av_log(s, AV_LOG_ERROR, "extradata size %u too large\n", size);
88  return -1;
89  }
90  return ff_get_extradata(s, par, pb, size);
91 }
92 
93 static void rm_read_metadata(AVFormatContext *s, AVIOContext *pb, int wide)
94 {
95  char buf[1024];
96  int i;
97 
98  for (i=0; i<FF_ARRAY_ELEMS(ff_rm_metadata); i++) {
99  int len = wide ? avio_rb16(pb) : avio_r8(pb);
100  if (len > 0) {
101  get_strl(pb, buf, sizeof(buf), len);
102  av_dict_set(&s->metadata, ff_rm_metadata[i], buf, 0);
103  }
104  }
105 }
106 
108 {
109  RMStream *rms = av_mallocz(sizeof(RMStream));
110  if (!rms)
111  return NULL;
112  rms->curpic_num = -1;
113  return rms;
114 }
115 
117 {
118  if (!rms)
119  return;
120 
121  av_packet_unref(&rms->pkt);
122 }
123 
125  AVStream *st, RMStream *ast, int read_all)
126 {
127  char buf[256];
128  uint32_t version;
129  int ret;
130 
131  /* ra type header */
132  version = avio_rb16(pb); /* version */
133  if (version == 3) {
134  unsigned bytes_per_minute;
135  int header_size = avio_rb16(pb);
136  int64_t startpos = avio_tell(pb);
137  avio_skip(pb, 8);
138  bytes_per_minute = avio_rb16(pb);
139  avio_skip(pb, 4);
140  rm_read_metadata(s, pb, 0);
141  if ((startpos + header_size) >= avio_tell(pb) + 2) {
142  // fourcc (should always be "lpcJ")
143  avio_r8(pb);
144  get_str8(pb, buf, sizeof(buf));
145  }
146  // Skip extra header crap (this should never happen)
147  if ((startpos + header_size) > avio_tell(pb))
148  avio_skip(pb, header_size + startpos - avio_tell(pb));
149  if (bytes_per_minute)
150  st->codecpar->bit_rate = 8LL * bytes_per_minute / 60;
151  st->codecpar->sample_rate = 8000;
152  st->codecpar->channels = 1;
156  ast->deint_id = DEINT_ID_INT0;
157  } else {
158  int flavor, sub_packet_h, coded_framesize, sub_packet_size;
159  int codecdata_length;
160  unsigned bytes_per_minute;
161  /* old version (4) */
162  avio_skip(pb, 2); /* unused */
163  avio_rb32(pb); /* .ra4 */
164  avio_rb32(pb); /* data size */
165  avio_rb16(pb); /* version2 */
166  avio_rb32(pb); /* header size */
167  flavor= avio_rb16(pb); /* add codec info / flavor */
168  coded_framesize = avio_rb32(pb); /* coded frame size */
169  if (coded_framesize < 0)
170  return AVERROR_INVALIDDATA;
171  ast->coded_framesize = coded_framesize;
172 
173  avio_rb32(pb); /* ??? */
174  bytes_per_minute = avio_rb32(pb);
175  if (version == 4) {
176  if (bytes_per_minute)
177  st->codecpar->bit_rate = 8LL * bytes_per_minute / 60;
178  }
179  avio_rb32(pb); /* ??? */
180  ast->sub_packet_h = sub_packet_h = avio_rb16(pb); /* 1 */
181  st->codecpar->block_align= avio_rb16(pb); /* frame size */
182  ast->sub_packet_size = sub_packet_size = avio_rb16(pb); /* sub packet size */
183  avio_rb16(pb); /* ??? */
184  if (version == 5) {
185  avio_rb16(pb); avio_rb16(pb); avio_rb16(pb);
186  }
187  st->codecpar->sample_rate = avio_rb16(pb);
188  avio_rb32(pb);
189  st->codecpar->channels = avio_rb16(pb);
190  if (version == 5) {
191  ast->deint_id = avio_rl32(pb);
192  avio_read(pb, buf, 4);
193  buf[4] = 0;
194  } else {
195  AV_WL32(buf, 0);
196  get_str8(pb, buf, sizeof(buf)); /* desc */
197  ast->deint_id = AV_RL32(buf);
198  get_str8(pb, buf, sizeof(buf)); /* desc */
199  }
201  st->codecpar->codec_tag = AV_RL32(buf);
203  st->codecpar->codec_tag);
204 
205  switch (st->codecpar->codec_id) {
206  case AV_CODEC_ID_AC3:
208  break;
209  case AV_CODEC_ID_RA_288:
210  st->codecpar->extradata_size= 0;
211  av_freep(&st->codecpar->extradata);
213  st->codecpar->block_align = coded_framesize;
214  break;
215  case AV_CODEC_ID_COOK:
217  case AV_CODEC_ID_ATRAC3:
218  case AV_CODEC_ID_SIPR:
219  if (read_all) {
220  codecdata_length = 0;
221  } else {
222  avio_rb16(pb); avio_r8(pb);
223  if (version == 5)
224  avio_r8(pb);
225  codecdata_length = avio_rb32(pb);
226  if((unsigned)codecdata_length > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE){
227  av_log(s, AV_LOG_ERROR, "codecdata_length too large\n");
228  return -1;
229  }
230  }
231 
233  if (st->codecpar->codec_id == AV_CODEC_ID_SIPR) {
234  if (flavor > 3) {
235  av_log(s, AV_LOG_ERROR, "bad SIPR file flavor %d\n",
236  flavor);
237  return -1;
238  }
239  st->codecpar->block_align = ff_sipr_subpk_size[flavor];
241  } else {
242  if(sub_packet_size <= 0){
243  av_log(s, AV_LOG_ERROR, "sub_packet_size is invalid\n");
244  return -1;
245  }
247  }
248  if ((ret = rm_read_extradata(s, pb, st->codecpar, codecdata_length)) < 0)
249  return ret;
250 
251  break;
252  case AV_CODEC_ID_AAC:
253  avio_rb16(pb); avio_r8(pb);
254  if (version == 5)
255  avio_r8(pb);
256  codecdata_length = avio_rb32(pb);
257  if((unsigned)codecdata_length > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE){
258  av_log(s, AV_LOG_ERROR, "codecdata_length too large\n");
259  return -1;
260  }
261  if (codecdata_length >= 1) {
262  avio_r8(pb);
263  if ((ret = rm_read_extradata(s, pb, st->codecpar, codecdata_length - 1)) < 0)
264  return ret;
265  }
266  break;
267  }
268  switch (ast->deint_id) {
269  case DEINT_ID_INT4:
270  if (ast->coded_framesize > ast->audio_framesize ||
271  sub_packet_h <= 1 ||
272  ast->coded_framesize * (uint64_t)sub_packet_h > (2 + (sub_packet_h & 1)) * ast->audio_framesize)
273  return AVERROR_INVALIDDATA;
274  if (ast->coded_framesize * (uint64_t)sub_packet_h != 2*ast->audio_framesize) {
275  avpriv_request_sample(s, "mismatching interleaver parameters");
276  return AVERROR_INVALIDDATA;
277  }
278  break;
279  case DEINT_ID_GENR:
280  if (ast->sub_packet_size <= 0 ||
281  ast->sub_packet_size > ast->audio_framesize)
282  return AVERROR_INVALIDDATA;
283  if (ast->audio_framesize % ast->sub_packet_size)
284  return AVERROR_INVALIDDATA;
285  break;
286  case DEINT_ID_SIPR:
287  case DEINT_ID_INT0:
288  case DEINT_ID_VBRS:
289  case DEINT_ID_VBRF:
290  break;
291  default:
292  av_log(s, AV_LOG_ERROR ,"Unknown interleaver %"PRIX32"\n", ast->deint_id);
293  return AVERROR_INVALIDDATA;
294  }
295  if (ast->deint_id == DEINT_ID_INT4 ||
296  ast->deint_id == DEINT_ID_GENR ||
297  ast->deint_id == DEINT_ID_SIPR) {
298  if (st->codecpar->block_align <= 0 ||
299  ast->audio_framesize * (uint64_t)sub_packet_h > (unsigned)INT_MAX ||
300  ast->audio_framesize * sub_packet_h < st->codecpar->block_align)
301  return AVERROR_INVALIDDATA;
302  if (av_new_packet(&ast->pkt, ast->audio_framesize * sub_packet_h) < 0)
303  return AVERROR(ENOMEM);
304  }
305 
306  if (read_all) {
307  avio_r8(pb);
308  avio_r8(pb);
309  avio_r8(pb);
310  rm_read_metadata(s, pb, 0);
311  }
312  }
313  return 0;
314 }
315 
317  AVStream *st, RMStream *rst,
318  unsigned int codec_data_size, const uint8_t *mime)
319 {
320  unsigned int v;
321  int size;
322  int64_t codec_pos;
323  int ret;
324 
325  if (codec_data_size > INT_MAX)
326  return AVERROR_INVALIDDATA;
327  if (codec_data_size == 0)
328  return 0;
329 
330  // Duplicate tags
333  return AVERROR_INVALIDDATA;
334 
335  avpriv_set_pts_info(st, 64, 1, 1000);
336  codec_pos = avio_tell(pb);
337  v = avio_rb32(pb);
338 
339  if (v == MKTAG(0xfd, 'a', 'r', '.')) {
340  /* ra type header */
341  if (rm_read_audio_stream_info(s, pb, st, rst, 0))
342  return -1;
343  } else if (v == MKBETAG('L', 'S', 'D', ':')) {
344  avio_seek(pb, -4, SEEK_CUR);
345  if ((ret = rm_read_extradata(s, pb, st->codecpar, codec_data_size)) < 0)
346  return ret;
347 
351  st->codecpar->codec_tag);
352  } else if(mime && !strcmp(mime, "logical-fileinfo")){
353  int stream_count, rule_count, property_count, i;
354  ff_free_stream(s, st);
355  if (avio_rb16(pb) != 0) {
356  av_log(s, AV_LOG_WARNING, "Unsupported version\n");
357  goto skip;
358  }
359  stream_count = avio_rb16(pb);
360  avio_skip(pb, 6*stream_count);
361  rule_count = avio_rb16(pb);
362  avio_skip(pb, 2*rule_count);
363  property_count = avio_rb16(pb);
364  for(i=0; i<property_count; i++){
365  uint8_t name[128], val[128];
366  avio_rb32(pb);
367  if (avio_rb16(pb) != 0) {
368  av_log(s, AV_LOG_WARNING, "Unsupported Name value property version\n");
369  goto skip; //FIXME skip just this one
370  }
371  get_str8(pb, name, sizeof(name));
372  switch(avio_rb32(pb)) {
373  case 2: get_strl(pb, val, sizeof(val), avio_rb16(pb));
374  av_dict_set(&s->metadata, name, val, 0);
375  break;
376  default: avio_skip(pb, avio_rb16(pb));
377  }
378  }
379  } else {
380  int fps;
381  if (avio_rl32(pb) != MKTAG('V', 'I', 'D', 'O')) {
382  fail1:
383  av_log(s, AV_LOG_WARNING, "Unsupported stream type %08x\n", v);
384  goto skip;
385  }
386  st->codecpar->codec_tag = avio_rl32(pb);
388  st->codecpar->codec_tag);
389  av_log(s, AV_LOG_TRACE, "%"PRIX32" %X\n",
390  st->codecpar->codec_tag, MKTAG('R', 'V', '2', '0'));
391  if (st->codecpar->codec_id == AV_CODEC_ID_NONE)
392  goto fail1;
393  st->codecpar->width = avio_rb16(pb);
394  st->codecpar->height = avio_rb16(pb);
395  avio_skip(pb, 2); // looks like bits per sample
396  avio_skip(pb, 4); // always zero?
399  fps = avio_rb32(pb);
400 
401  if ((ret = rm_read_extradata(s, pb, st->codecpar, codec_data_size - (avio_tell(pb) - codec_pos))) < 0)
402  return ret;
403 
404  if (fps > 0) {
406  0x10000, fps, (1 << 30) - 1);
407 #if FF_API_R_FRAME_RATE
408  st->r_frame_rate = st->avg_frame_rate;
409 #endif
410  } else if (s->error_recognition & AV_EF_EXPLODE) {
411  av_log(s, AV_LOG_ERROR, "Invalid framerate\n");
412  return AVERROR_INVALIDDATA;
413  }
414  }
415 
416 skip:
417  /* skip codec info */
418  size = avio_tell(pb) - codec_pos;
419  if (codec_data_size >= size) {
420  avio_skip(pb, codec_data_size - size);
421  } else {
422  av_log(s, AV_LOG_WARNING, "codec_data_size %u < size %d\n", codec_data_size, size);
423  }
424 
425  return 0;
426 }
427 
428 /** this function assumes that the demuxer has already seeked to the start
429  * of the INDX chunk, and will bail out if not. */
431 {
432  AVIOContext *pb = s->pb;
433  unsigned int size, n_pkts, str_id, next_off, n, pos, pts;
434  AVStream *st;
435 
436  do {
437  if (avio_rl32(pb) != MKTAG('I','N','D','X'))
438  return -1;
439  size = avio_rb32(pb);
440  if (size < 20)
441  return -1;
442  avio_skip(pb, 2);
443  n_pkts = avio_rb32(pb);
444  str_id = avio_rb16(pb);
445  next_off = avio_rb32(pb);
446  for (n = 0; n < s->nb_streams; n++)
447  if (s->streams[n]->id == str_id) {
448  st = s->streams[n];
449  break;
450  }
451  if (n == s->nb_streams) {
453  "Invalid stream index %d for index at pos %"PRId64"\n",
454  str_id, avio_tell(pb));
455  goto skip;
456  } else if ((avio_size(pb) - avio_tell(pb)) / 14 < n_pkts) {
458  "Nr. of packets in packet index for stream index %d "
459  "exceeds filesize (%"PRId64" at %"PRId64" = %"PRId64")\n",
460  str_id, avio_size(pb), avio_tell(pb),
461  (avio_size(pb) - avio_tell(pb)) / 14);
462  goto skip;
463  }
464 
465  for (n = 0; n < n_pkts; n++) {
466  if (avio_feof(pb))
467  return AVERROR_INVALIDDATA;
468  avio_skip(pb, 2);
469  pts = avio_rb32(pb);
470  pos = avio_rb32(pb);
471  avio_skip(pb, 4); /* packet no. */
472 
474  }
475 
476 skip:
477  if (next_off && avio_tell(pb) < next_off &&
478  avio_seek(pb, next_off, SEEK_SET) < 0) {
480  "Non-linear index detected, not supported\n");
481  return -1;
482  }
483  } while (next_off);
484 
485  return 0;
486 }
487 
489 {
490  RMDemuxContext *rm = s->priv_data;
491  AVStream *st;
492 
493  rm->old_format = 1;
494  st = avformat_new_stream(s, NULL);
495  if (!st)
496  return -1;
498  if (!st->priv_data)
499  return AVERROR(ENOMEM);
500  return rm_read_audio_stream_info(s, s->pb, st, st->priv_data, 1);
501 }
502 
504  AVStream *st, char *mime)
505 {
506  int number_of_streams = avio_rb16(pb);
507  int number_of_mdpr;
508  int i, ret;
509  unsigned size2;
510  for (i = 0; i<number_of_streams; i++)
511  avio_rb16(pb);
512  number_of_mdpr = avio_rb16(pb);
513  if (number_of_mdpr != 1) {
514  avpriv_request_sample(s, "MLTI with multiple (%d) MDPR", number_of_mdpr);
515  }
516  for (i = 0; i < number_of_mdpr; i++) {
517  AVStream *st2;
518  if (i > 0) {
519  st2 = avformat_new_stream(s, NULL);
520  if (!st2) {
521  ret = AVERROR(ENOMEM);
522  return ret;
523  }
524  st2->id = st->id + (i<<16);
525  st2->codecpar->bit_rate = st->codecpar->bit_rate;
526  st2->start_time = st->start_time;
527  st2->duration = st->duration;
530  if (!st2->priv_data)
531  return AVERROR(ENOMEM);
532  } else
533  st2 = st;
534 
535  size2 = avio_rb32(pb);
536  ret = ff_rm_read_mdpr_codecdata(s, s->pb, st2, st2->priv_data,
537  size2, NULL);
538  if (ret < 0)
539  return ret;
540  }
541  return 0;
542 }
543 
545 {
546  RMDemuxContext *rm = s->priv_data;
547  AVStream *st;
548  AVIOContext *pb = s->pb;
549  unsigned int tag;
550  int tag_size;
551  unsigned int start_time, duration;
552  unsigned int data_off = 0, indx_off = 0;
553  char buf[128], mime[128];
554  int flags = 0;
555  int ret;
556  unsigned size, v;
557  int64_t codec_pos;
558 
559  tag = avio_rl32(pb);
560  if (tag == MKTAG('.', 'r', 'a', 0xfd)) {
561  /* very old .ra format */
562  return rm_read_header_old(s);
563  } else if (tag != MKTAG('.', 'R', 'M', 'F')) {
564  return AVERROR(EIO);
565  }
566 
567  tag_size = avio_rb32(pb);
568  if (tag_size < 0)
569  return AVERROR_INVALIDDATA;
570  avio_skip(pb, tag_size - 8);
571 
572  for(;;) {
573  ret = AVERROR_INVALIDDATA;
574  if (avio_feof(pb))
575  goto fail;
576  tag = avio_rl32(pb);
577  tag_size = avio_rb32(pb);
578  avio_rb16(pb);
579  av_log(s, AV_LOG_TRACE, "tag=%s size=%d\n",
580  av_fourcc2str(tag), tag_size);
581  if (tag_size < 10 && tag != MKTAG('D', 'A', 'T', 'A'))
582  goto fail;
583  switch(tag) {
584  case MKTAG('P', 'R', 'O', 'P'):
585  /* file header */
586  avio_rb32(pb); /* max bit rate */
587  avio_rb32(pb); /* avg bit rate */
588  avio_rb32(pb); /* max packet size */
589  avio_rb32(pb); /* avg packet size */
590  avio_rb32(pb); /* nb packets */
591  duration = avio_rb32(pb); /* duration */
592  s->duration = av_rescale(duration, AV_TIME_BASE, 1000);
593  avio_rb32(pb); /* preroll */
594  indx_off = avio_rb32(pb); /* index offset */
595  data_off = avio_rb32(pb); /* data offset */
596  avio_rb16(pb); /* nb streams */
597  flags = avio_rb16(pb); /* flags */
598  break;
599  case MKTAG('C', 'O', 'N', 'T'):
600  rm_read_metadata(s, pb, 1);
601  break;
602  case MKTAG('M', 'D', 'P', 'R'):
603  st = avformat_new_stream(s, NULL);
604  if (!st) {
605  ret = AVERROR(ENOMEM);
606  goto fail;
607  }
608  st->id = avio_rb16(pb);
609  avio_rb32(pb); /* max bit rate */
610  st->codecpar->bit_rate = avio_rb32(pb); /* bit rate */
611  avio_rb32(pb); /* max packet size */
612  avio_rb32(pb); /* avg packet size */
613  start_time = avio_rb32(pb); /* start time */
614  avio_rb32(pb); /* preroll */
615  duration = avio_rb32(pb); /* duration */
616  st->start_time = start_time;
617  st->duration = duration;
618  if(duration>0)
619  s->duration = AV_NOPTS_VALUE;
620  get_str8(pb, buf, sizeof(buf)); /* desc */
621  get_str8(pb, mime, sizeof(mime)); /* mimetype */
624  if (!st->priv_data) {
625  ret = AVERROR(ENOMEM);
626  goto fail;
627  }
628 
629  size = avio_rb32(pb);
630  codec_pos = avio_tell(pb);
631 
632  ffio_ensure_seekback(pb, 4);
633  v = avio_rb32(pb);
634  if (v == MKBETAG('M', 'L', 'T', 'I')) {
635  ret = rm_read_multi(s, s->pb, st, mime);
636  if (ret < 0)
637  goto fail;
638  avio_seek(pb, codec_pos + size, SEEK_SET);
639  } else {
640  avio_skip(pb, -4);
641  ret = ff_rm_read_mdpr_codecdata(s, s->pb, st, st->priv_data,
642  size, mime);
643  if (ret < 0)
644  goto fail;
645  }
646 
647  break;
648  case MKTAG('D', 'A', 'T', 'A'):
649  goto header_end;
650  default:
651  /* unknown tag: skip it */
652  avio_skip(pb, tag_size - 10);
653  break;
654  }
655  }
656  header_end:
657  rm->nb_packets = avio_rb32(pb); /* number of packets */
658  if (!rm->nb_packets && (flags & 4))
659  rm->nb_packets = 3600 * 25;
660  avio_rb32(pb); /* next data header */
661 
662  if (!data_off)
663  data_off = avio_tell(pb) - 18;
664  if (indx_off && (pb->seekable & AVIO_SEEKABLE_NORMAL) &&
665  !(s->flags & AVFMT_FLAG_IGNIDX) &&
666  avio_seek(pb, indx_off, SEEK_SET) >= 0) {
667  rm_read_index(s);
668  avio_seek(pb, data_off + 18, SEEK_SET);
669  }
670 
671  return 0;
672 
673 fail:
674  rm_read_close(s);
675  return ret;
676 }
677 
678 static int get_num(AVIOContext *pb, int *len)
679 {
680  int n, n1;
681 
682  n = avio_rb16(pb);
683  (*len)-=2;
684  n &= 0x7FFF;
685  if (n >= 0x4000) {
686  return n - 0x4000;
687  } else {
688  n1 = avio_rb16(pb);
689  (*len)-=2;
690  return (n << 16) | n1;
691  }
692 }
693 
694 /* multiple of 20 bytes for ra144 (ugly) */
695 #define RAW_PACKET_SIZE 1000
696 
697 static int rm_sync(AVFormatContext *s, int64_t *timestamp, int *flags, int *stream_index, int64_t *pos){
698  RMDemuxContext *rm = s->priv_data;
699  AVIOContext *pb = s->pb;
700  AVStream *st;
701  uint32_t state=0xFFFFFFFF;
702 
703  while(!avio_feof(pb)){
704  int len, num, i;
705  int mlti_id;
706  *pos= avio_tell(pb) - 3;
707  if(rm->remaining_len > 0){
708  num= rm->current_stream;
709  mlti_id = 0;
710  len= rm->remaining_len;
711  *timestamp = AV_NOPTS_VALUE;
712  *flags= 0;
713  }else{
714  state= (state<<8) + avio_r8(pb);
715 
716  if(state == MKBETAG('I', 'N', 'D', 'X')){
717  int n_pkts;
718  int64_t expected_len;
719  len = avio_rb32(pb);
720  avio_skip(pb, 2);
721  n_pkts = avio_rb32(pb);
722  expected_len = 20 + n_pkts * 14LL;
723 
724  if (len == 20 && expected_len <= INT_MAX)
725  /* some files don't add index entries to chunk size... */
726  len = expected_len;
727  else if (len != expected_len)
729  "Index size %d (%d pkts) is wrong, should be %"PRId64".\n",
730  len, n_pkts, expected_len);
731  if(len < 14)
732  continue;
733  len -= 14; // we already read part of the index header
734  goto skip;
735  } else if (state == MKBETAG('D','A','T','A')) {
737  "DATA tag in middle of chunk, file may be broken.\n");
738  }
739 
740  if(state > (unsigned)0xFFFF || state <= 12)
741  continue;
742  len=state - 12;
743  state= 0xFFFFFFFF;
744 
745  num = avio_rb16(pb);
746  *timestamp = avio_rb32(pb);
747  mlti_id = (avio_r8(pb) >> 1) - 1;
748  mlti_id = FFMAX(mlti_id, 0) << 16;
749  *flags = avio_r8(pb); /* flags */
750  }
751  for(i=0;i<s->nb_streams;i++) {
752  st = s->streams[i];
753  if (mlti_id + num == st->id)
754  break;
755  }
756  if (i == s->nb_streams) {
757 skip:
758  /* skip packet if unknown number */
759  avio_skip(pb, len);
760  rm->remaining_len = 0;
761  continue;
762  }
763  *stream_index= i;
764 
765  return len;
766  }
767  return -1;
768 }
769 
771  RMDemuxContext *rm, RMStream *vst,
772  AVPacket *pkt, int len, int *pseq,
773  int64_t *timestamp)
774 {
775  int hdr;
776  int seq = 0, pic_num = 0, len2 = 0, pos = 0; //init to silence compiler warning
777  int type;
778  int ret;
779 
780  hdr = avio_r8(pb); len--;
781  type = hdr >> 6;
782 
783  if(type != 3){ // not frame as a part of packet
784  seq = avio_r8(pb); len--;
785  }
786  if(type != 1){ // not whole frame
787  len2 = get_num(pb, &len);
788  pos = get_num(pb, &len);
789  pic_num = avio_r8(pb); len--;
790  }
791  if(len<0) {
792  av_log(s, AV_LOG_ERROR, "Insufficient data\n");
793  return -1;
794  }
795  rm->remaining_len = len;
796  if(type&1){ // frame, not slice
797  if(type == 3){ // frame as a part of packet
798  len= len2;
799  *timestamp = pos;
800  }
801  if(rm->remaining_len < len) {
802  av_log(s, AV_LOG_ERROR, "Insufficient remaining len\n");
803  return -1;
804  }
805  rm->remaining_len -= len;
806  if ((ret = av_new_packet(pkt, len + 9)) < 0)
807  return ret;
808  pkt->data[0] = 0;
809  AV_WL32(pkt->data + 1, 1);
810  AV_WL32(pkt->data + 5, 0);
811  if ((ret = avio_read(pb, pkt->data + 9, len)) != len) {
813  av_log(s, AV_LOG_ERROR, "Failed to read %d bytes\n", len);
814  return ret < 0 ? ret : AVERROR(EIO);
815  }
816  return 0;
817  }
818  //now we have to deal with single slice
819 
820  *pseq = seq;
821  if((seq & 0x7F) == 1 || vst->curpic_num != pic_num){
822  if (len2 > ffio_limit(pb, len2)) {
823  av_log(s, AV_LOG_ERROR, "Impossibly sized packet\n");
824  return AVERROR_INVALIDDATA;
825  }
826  vst->slices = ((hdr & 0x3F) << 1) + 1;
827  vst->videobufsize = len2 + 8*vst->slices + 1;
828  av_packet_unref(&vst->pkt); //FIXME this should be output.
829  if ((ret = av_new_packet(&vst->pkt, vst->videobufsize)) < 0)
830  return ret;
831  vst->videobufpos = 8*vst->slices + 1;
832  vst->cur_slice = 0;
833  vst->curpic_num = pic_num;
834  vst->pktpos = avio_tell(pb);
835  }
836  if(type == 2)
837  len = FFMIN(len, pos);
838 
839  if(++vst->cur_slice > vst->slices) {
840  av_log(s, AV_LOG_ERROR, "cur slice %d, too large\n", vst->cur_slice);
841  return 1;
842  }
843  if(!vst->pkt.data)
844  return AVERROR(ENOMEM);
845  AV_WL32(vst->pkt.data - 7 + 8*vst->cur_slice, 1);
846  AV_WL32(vst->pkt.data - 3 + 8*vst->cur_slice, vst->videobufpos - 8*vst->slices - 1);
847  if(vst->videobufpos + len > vst->videobufsize) {
848  av_log(s, AV_LOG_ERROR, "outside videobufsize\n");
849  return 1;
850  }
851  if (avio_read(pb, vst->pkt.data + vst->videobufpos, len) != len)
852  return AVERROR(EIO);
853  vst->videobufpos += len;
854  rm->remaining_len-= len;
855 
856  if (type == 2 || vst->videobufpos == vst->videobufsize) {
857  vst->pkt.data[0] = vst->cur_slice-1;
858  av_packet_move_ref(pkt, &vst->pkt);
859  if(vst->slices != vst->cur_slice) //FIXME find out how to set slices correct from the begin
860  memmove(pkt->data + 1 + 8*vst->cur_slice, pkt->data + 1 + 8*vst->slices,
861  vst->videobufpos - 1 - 8*vst->slices);
862  av_shrink_packet(pkt, vst->videobufpos + 8*(vst->cur_slice - vst->slices));
864  pkt->pos = vst->pktpos;
865  vst->slices = 0;
866  return 0;
867  }
868 
869  return 1;
870 }
871 
872 static inline void
874 {
875  uint8_t *ptr;
876  int j;
877 
878  if (st->codecpar->codec_id == AV_CODEC_ID_AC3) {
879  ptr = pkt->data;
880  for (j=0;j<pkt->size;j+=2) {
881  FFSWAP(int, ptr[0], ptr[1]);
882  ptr += 2;
883  }
884  }
885 }
886 
887 static int readfull(AVFormatContext *s, AVIOContext *pb, uint8_t *dst, int n) {
888  int ret = avio_read(pb, dst, n);
889  if (ret != n) {
890  if (ret >= 0) memset(dst + ret, 0, n - ret);
891  else memset(dst , 0, n);
892  av_log(s, AV_LOG_ERROR, "Failed to fully read block\n");
893  }
894  return ret;
895 }
896 
897 int
899  AVStream *st, RMStream *ast, int len, AVPacket *pkt,
900  int *seq, int flags, int64_t timestamp)
901 {
902  RMDemuxContext *rm = s->priv_data;
903  int ret;
904 
905  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
906  rm->current_stream= st->id;
907  ret = rm_assemble_video_frame(s, pb, rm, ast, pkt, len, seq, &timestamp);
908  if(ret)
909  return ret < 0 ? ret : -1; //got partial frame or error
910  } else if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
911  if ((ast->deint_id == DEINT_ID_GENR) ||
912  (ast->deint_id == DEINT_ID_INT4) ||
913  (ast->deint_id == DEINT_ID_SIPR)) {
914  int x;
915  int sps = ast->sub_packet_size;
916  int cfs = ast->coded_framesize;
917  int h = ast->sub_packet_h;
918  int y = ast->sub_packet_cnt;
919  int w = ast->audio_framesize;
920 
921  if (flags & 2)
922  y = ast->sub_packet_cnt = 0;
923  if (!y)
924  ast->audiotimestamp = timestamp;
925 
926  switch (ast->deint_id) {
927  case DEINT_ID_INT4:
928  for (x = 0; x < h/2; x++)
929  readfull(s, pb, ast->pkt.data+x*2*w+y*cfs, cfs);
930  break;
931  case DEINT_ID_GENR:
932  for (x = 0; x < w/sps; x++)
933  readfull(s, pb, ast->pkt.data+sps*(h*x+((h+1)/2)*(y&1)+(y>>1)), sps);
934  break;
935  case DEINT_ID_SIPR:
936  readfull(s, pb, ast->pkt.data + y * w, w);
937  break;
938  }
939 
940  if (++(ast->sub_packet_cnt) < h)
941  return -1;
942  if (ast->deint_id == DEINT_ID_SIPR)
944 
945  ast->sub_packet_cnt = 0;
946  rm->audio_stream_num = st->index;
947  if (st->codecpar->block_align <= 0) {
948  av_log(s, AV_LOG_ERROR, "Invalid block alignment %d\n", st->codecpar->block_align);
949  return AVERROR_INVALIDDATA;
950  }
951  rm->audio_pkt_cnt = h * w / st->codecpar->block_align;
952  } else if ((ast->deint_id == DEINT_ID_VBRF) ||
953  (ast->deint_id == DEINT_ID_VBRS)) {
954  int x;
955  rm->audio_stream_num = st->index;
956  ast->sub_packet_cnt = (avio_rb16(pb) & 0xf0) >> 4;
957  if (ast->sub_packet_cnt) {
958  for (x = 0; x < ast->sub_packet_cnt; x++)
959  ast->sub_packet_lengths[x] = avio_rb16(pb);
960  rm->audio_pkt_cnt = ast->sub_packet_cnt;
961  ast->audiotimestamp = timestamp;
962  } else
963  return -1;
964  } else {
965  ret = av_get_packet(pb, pkt, len);
966  if (ret < 0)
967  return ret;
968  rm_ac3_swap_bytes(st, pkt);
969  }
970  } else {
971  ret = av_get_packet(pb, pkt, len);
972  if (ret < 0)
973  return ret;
974  }
975 
976  pkt->stream_index = st->index;
977 
978  pkt->pts = timestamp;
979  if (flags & 2)
981 
982  return st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO ? rm->audio_pkt_cnt : 0;
983 }
984 
985 int
987  AVStream *st, RMStream *ast, AVPacket *pkt)
988 {
989  RMDemuxContext *rm = s->priv_data;
990  int ret;
991 
992  av_assert0 (rm->audio_pkt_cnt > 0);
993 
994  if (ast->deint_id == DEINT_ID_VBRF ||
995  ast->deint_id == DEINT_ID_VBRS) {
996  ret = av_get_packet(pb, pkt, ast->sub_packet_lengths[ast->sub_packet_cnt - rm->audio_pkt_cnt]);
997  if (ret < 0)
998  return ret;
999  } else {
1000  ret = av_new_packet(pkt, st->codecpar->block_align);
1001  if (ret < 0)
1002  return ret;
1003  memcpy(pkt->data, ast->pkt.data + st->codecpar->block_align * //FIXME avoid this
1004  (ast->sub_packet_h * ast->audio_framesize / st->codecpar->block_align - rm->audio_pkt_cnt),
1005  st->codecpar->block_align);
1006  }
1007  rm->audio_pkt_cnt--;
1008  if ((pkt->pts = ast->audiotimestamp) != AV_NOPTS_VALUE) {
1011  } else
1012  pkt->flags = 0;
1013  pkt->stream_index = st->index;
1014 
1015  return rm->audio_pkt_cnt;
1016 }
1017 
1019 {
1020  RMDemuxContext *rm = s->priv_data;
1021  AVStream *st = NULL; // init to silence compiler warning
1022  int i, res, seq = 1;
1023  int64_t timestamp, pos, len;
1024  int flags;
1025 
1026  for (;;) {
1027  if (rm->audio_pkt_cnt) {
1028  // If there are queued audio packet return them first
1029  st = s->streams[rm->audio_stream_num];
1030  res = ff_rm_retrieve_cache(s, s->pb, st, st->priv_data, pkt);
1031  if(res < 0)
1032  return res;
1033  flags = 0;
1034  } else {
1035  if (rm->old_format) {
1036  RMStream *ast;
1037 
1038  st = s->streams[0];
1039  ast = st->priv_data;
1040  timestamp = AV_NOPTS_VALUE;
1041  len = !ast->audio_framesize ? RAW_PACKET_SIZE :
1042  ast->coded_framesize * (int64_t)ast->sub_packet_h / 2;
1043  if (len > INT_MAX)
1044  return AVERROR_INVALIDDATA;
1045  flags = (seq++ == 1) ? 2 : 0;
1046  pos = avio_tell(s->pb);
1047  } else {
1048  len = rm_sync(s, &timestamp, &flags, &i, &pos);
1049  if (len > 0)
1050  st = s->streams[i];
1051  }
1052 
1053  if (avio_feof(s->pb))
1054  return AVERROR_EOF;
1055  if (len <= 0)
1056  return AVERROR(EIO);
1057 
1058  res = ff_rm_parse_packet (s, s->pb, st, st->priv_data, len, pkt,
1059  &seq, flags, timestamp);
1060  if (res < -1)
1061  return res;
1062  if((flags&2) && (seq&0x7F) == 1)
1063  av_add_index_entry(st, pos, timestamp, 0, 0, AVINDEX_KEYFRAME);
1064  if (res)
1065  continue;
1066  }
1067 
1068  if( (st->discard >= AVDISCARD_NONKEY && !(flags&2))
1069  || st->discard >= AVDISCARD_ALL){
1071  } else
1072  break;
1073  }
1074 
1075  return 0;
1076 }
1077 
1079 {
1080  int i;
1081 
1082  for (i=0;i<s->nb_streams;i++)
1083  ff_rm_free_rmstream(s->streams[i]->priv_data);
1084 
1085  return 0;
1086 }
1087 
1088 static int rm_probe(const AVProbeData *p)
1089 {
1090  /* check file header */
1091  if ((p->buf[0] == '.' && p->buf[1] == 'R' &&
1092  p->buf[2] == 'M' && p->buf[3] == 'F' &&
1093  p->buf[4] == 0 && p->buf[5] == 0) ||
1094  (p->buf[0] == '.' && p->buf[1] == 'r' &&
1095  p->buf[2] == 'a' && p->buf[3] == 0xfd))
1096  return AVPROBE_SCORE_MAX;
1097  else
1098  return 0;
1099 }
1100 
1101 static int64_t rm_read_dts(AVFormatContext *s, int stream_index,
1102  int64_t *ppos, int64_t pos_limit)
1103 {
1104  RMDemuxContext *rm = s->priv_data;
1105  int64_t pos, dts;
1106  int stream_index2, flags, len, h;
1107 
1108  pos = *ppos;
1109 
1110  if(rm->old_format)
1111  return AV_NOPTS_VALUE;
1112 
1113  if (avio_seek(s->pb, pos, SEEK_SET) < 0)
1114  return AV_NOPTS_VALUE;
1115 
1116  rm->remaining_len=0;
1117  for(;;){
1118  int seq=1;
1119  AVStream *st;
1120 
1121  len = rm_sync(s, &dts, &flags, &stream_index2, &pos);
1122  if(len<0)
1123  return AV_NOPTS_VALUE;
1124 
1125  st = s->streams[stream_index2];
1126  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
1127  h= avio_r8(s->pb); len--;
1128  if(!(h & 0x40)){
1129  seq = avio_r8(s->pb); len--;
1130  }
1131  }
1132 
1133  if((flags&2) && (seq&0x7F) == 1){
1134  av_log(s, AV_LOG_TRACE, "%d %d-%d %"PRId64" %d\n",
1135  flags, stream_index2, stream_index, dts, seq);
1136  av_add_index_entry(st, pos, dts, 0, 0, AVINDEX_KEYFRAME);
1137  if(stream_index2 == stream_index)
1138  break;
1139  }
1140 
1141  avio_skip(s->pb, len);
1142  }
1143  *ppos = pos;
1144  return dts;
1145 }
1146 
1147 static int rm_read_seek(AVFormatContext *s, int stream_index,
1148  int64_t pts, int flags)
1149 {
1150  RMDemuxContext *rm = s->priv_data;
1151 
1152  if (ff_seek_frame_binary(s, stream_index, pts, flags) < 0)
1153  return -1;
1154  rm->audio_pkt_cnt = 0;
1155  return 0;
1156 }
1157 
1158 
1160  .name = "rm",
1161  .long_name = NULL_IF_CONFIG_SMALL("RealMedia"),
1162  .priv_data_size = sizeof(RMDemuxContext),
1163  .read_probe = rm_probe,
1167  .read_timestamp = rm_read_dts,
1169 };
1170 
1172  .name = "rdt",
1173  .long_name = NULL_IF_CONFIG_SMALL("RDT demuxer"),
1174  .priv_data_size = sizeof(RMDemuxContext),
1176  .flags = AVFMT_NOFILE,
1177 };
1178 
1179 static int ivr_probe(const AVProbeData *p)
1180 {
1181  if (memcmp(p->buf, ".R1M\x0\x1\x1", 7) &&
1182  memcmp(p->buf, ".REC", 4))
1183  return 0;
1184 
1185  return AVPROBE_SCORE_MAX;
1186 }
1187 
1189 {
1190  unsigned tag, type, len, tlen, value;
1191  int i, j, n, count, nb_streams = 0, ret;
1192  uint8_t key[256], val[256];
1193  AVIOContext *pb = s->pb;
1194  AVStream *st;
1195  int64_t pos, offset=0, temp;
1196 
1197  pos = avio_tell(pb);
1198  tag = avio_rl32(pb);
1199  if (tag == MKTAG('.','R','1','M')) {
1200  if (avio_rb16(pb) != 1)
1201  return AVERROR_INVALIDDATA;
1202  if (avio_r8(pb) != 1)
1203  return AVERROR_INVALIDDATA;
1204  len = avio_rb32(pb);
1205  avio_skip(pb, len);
1206  avio_skip(pb, 5);
1207  temp = avio_rb64(pb);
1208  while (!avio_feof(pb) && temp) {
1209  offset = temp;
1210  temp = avio_rb64(pb);
1211  }
1212  if (offset <= 0)
1213  return AVERROR_INVALIDDATA;
1214  avio_skip(pb, offset - avio_tell(pb));
1215  if (avio_r8(pb) != 1)
1216  return AVERROR_INVALIDDATA;
1217  len = avio_rb32(pb);
1218  avio_skip(pb, len);
1219  if (avio_r8(pb) != 2)
1220  return AVERROR_INVALIDDATA;
1221  avio_skip(pb, 16);
1222  pos = avio_tell(pb);
1223  tag = avio_rl32(pb);
1224  }
1225 
1226  if (tag != MKTAG('.','R','E','C'))
1227  return AVERROR_INVALIDDATA;
1228 
1229  if (avio_r8(pb) != 0)
1230  return AVERROR_INVALIDDATA;
1231  count = avio_rb32(pb);
1232  for (i = 0; i < count; i++) {
1233  if (avio_feof(pb))
1234  return AVERROR_INVALIDDATA;
1235 
1236  type = avio_r8(pb);
1237  tlen = avio_rb32(pb);
1238  avio_get_str(pb, tlen, key, sizeof(key));
1239  len = avio_rb32(pb);
1240  if (type == 5) {
1241  avio_get_str(pb, len, val, sizeof(val));
1242  av_log(s, AV_LOG_DEBUG, "%s = '%s'\n", key, val);
1243  } else if (type == 4) {
1244  av_log(s, AV_LOG_DEBUG, "%s = '0x", key);
1245  for (j = 0; j < len; j++) {
1246  if (avio_feof(pb))
1247  return AVERROR_INVALIDDATA;
1248  av_log(s, AV_LOG_DEBUG, "%X", avio_r8(pb));
1249  }
1250  av_log(s, AV_LOG_DEBUG, "'\n");
1251  } else if (len == 4 && type == 3 && !strncmp(key, "StreamCount", tlen)) {
1252  nb_streams = value = avio_rb32(pb);
1253  } else if (len == 4 && type == 3) {
1254  value = avio_rb32(pb);
1255  av_log(s, AV_LOG_DEBUG, "%s = %d\n", key, value);
1256  } else {
1257  av_log(s, AV_LOG_DEBUG, "Skipping unsupported key: %s\n", key);
1258  avio_skip(pb, len);
1259  }
1260  }
1261 
1262  for (n = 0; n < nb_streams; n++) {
1263  if (!(st = avformat_new_stream(s, NULL)) ||
1264  !(st->priv_data = ff_rm_alloc_rmstream())) {
1265  ret = AVERROR(ENOMEM);
1266  goto fail;
1267  }
1268 
1269  if (avio_r8(pb) != 1)
1270  goto invalid_data;
1271 
1272  count = avio_rb32(pb);
1273  for (i = 0; i < count; i++) {
1274  if (avio_feof(pb))
1275  goto invalid_data;
1276 
1277  type = avio_r8(pb);
1278  tlen = avio_rb32(pb);
1279  avio_get_str(pb, tlen, key, sizeof(key));
1280  len = avio_rb32(pb);
1281  if (type == 5) {
1282  avio_get_str(pb, len, val, sizeof(val));
1283  av_log(s, AV_LOG_DEBUG, "%s = '%s'\n", key, val);
1284  } else if (type == 4 && !strncmp(key, "OpaqueData", tlen)) {
1285  ret = ffio_ensure_seekback(pb, 4);
1286  if (ret < 0)
1287  goto fail;
1288  if (avio_rb32(pb) == MKBETAG('M', 'L', 'T', 'I')) {
1289  ret = rm_read_multi(s, pb, st, NULL);
1290  } else {
1291  if (avio_feof(pb))
1292  goto invalid_data;
1293  avio_seek(pb, -4, SEEK_CUR);
1294  ret = ff_rm_read_mdpr_codecdata(s, pb, st, st->priv_data, len, NULL);
1295  }
1296 
1297  if (ret < 0)
1298  goto fail;
1299  } else if (type == 4) {
1300  int j;
1301 
1302  av_log(s, AV_LOG_DEBUG, "%s = '0x", key);
1303  for (j = 0; j < len; j++) {
1304  if (avio_feof(pb))
1305  goto invalid_data;
1306  av_log(s, AV_LOG_DEBUG, "%X", avio_r8(pb));
1307  }
1308  av_log(s, AV_LOG_DEBUG, "'\n");
1309  } else if (len == 4 && type == 3 && !strncmp(key, "Duration", tlen)) {
1310  st->duration = avio_rb32(pb);
1311  } else if (len == 4 && type == 3) {
1312  value = avio_rb32(pb);
1313  av_log(s, AV_LOG_DEBUG, "%s = %d\n", key, value);
1314  } else {
1315  av_log(s, AV_LOG_DEBUG, "Skipping unsupported key: %s\n", key);
1316  avio_skip(pb, len);
1317  }
1318  }
1319  }
1320 
1321  if (avio_r8(pb) != 6)
1322  goto invalid_data;
1323  avio_skip(pb, 12);
1324  avio_seek(pb, avio_rb64(pb) + pos, SEEK_SET);
1325  if (avio_r8(pb) != 8)
1326  goto invalid_data;
1327  avio_skip(pb, 8);
1328 
1329  return 0;
1330 invalid_data:
1331  ret = AVERROR_INVALIDDATA;
1332 fail:
1333  rm_read_close(s);
1334  return ret;
1335 }
1336 
1338 {
1339  RMDemuxContext *rm = s->priv_data;
1340  int ret = AVERROR_EOF, opcode;
1341  AVIOContext *pb = s->pb;
1342  unsigned size, index;
1343  int64_t pos, pts;
1344 
1345  if (avio_feof(pb) || rm->data_end)
1346  return AVERROR_EOF;
1347 
1348  pos = avio_tell(pb);
1349 
1350  for (;;) {
1351  if (rm->audio_pkt_cnt) {
1352  // If there are queued audio packet return them first
1353  AVStream *st;
1354 
1355  st = s->streams[rm->audio_stream_num];
1356  ret = ff_rm_retrieve_cache(s, pb, st, st->priv_data, pkt);
1357  if (ret < 0) {
1358  return ret;
1359  }
1360  } else {
1361  if (rm->remaining_len) {
1362  avio_skip(pb, rm->remaining_len);
1363  rm->remaining_len = 0;
1364  }
1365 
1366  if (avio_feof(pb))
1367  return AVERROR_EOF;
1368 
1369  opcode = avio_r8(pb);
1370  if (opcode == 2) {
1371  AVStream *st;
1372  int seq = 1;
1373 
1374  pts = avio_rb32(pb);
1375  index = avio_rb16(pb);
1376  if (index >= s->nb_streams)
1377  return AVERROR_INVALIDDATA;
1378 
1379  avio_skip(pb, 4);
1380  size = avio_rb32(pb);
1381  avio_skip(pb, 4);
1382 
1383  if (size < 1 || size > INT_MAX/4) {
1384  av_log(s, AV_LOG_ERROR, "size %u is invalid\n", size);
1385  return AVERROR_INVALIDDATA;
1386  }
1387 
1388  st = s->streams[index];
1389  ret = ff_rm_parse_packet(s, pb, st, st->priv_data, size, pkt,
1390  &seq, 0, pts);
1391  if (ret < -1) {
1392  return ret;
1393  } else if (ret) {
1394  continue;
1395  }
1396 
1397  pkt->pos = pos;
1398  pkt->pts = pts;
1399  pkt->stream_index = index;
1400  } else if (opcode == 7) {
1401  pos = avio_rb64(pb);
1402  if (!pos) {
1403  rm->data_end = 1;
1404  return AVERROR_EOF;
1405  }
1406  } else {
1407  av_log(s, AV_LOG_ERROR, "Unsupported opcode=%d at %"PRIX64"\n", opcode, avio_tell(pb) - 1);
1408  return AVERROR(EIO);
1409  }
1410  }
1411 
1412  break;
1413  }
1414 
1415  return ret;
1416 }
1417 
1419  .name = "ivr",
1420  .long_name = NULL_IF_CONFIG_SMALL("IVR (Internet Video Recording)"),
1421  .priv_data_size = sizeof(RMDemuxContext),
1422  .read_probe = ivr_probe,
1426  .extensions = "ivr",
1427 };
static double val(void *priv, double ch)
Definition: aeval.c:76
uint8_t
int32_t
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
#define AV_EF_EXPLODE
abort decoding on minor error detection
Definition: avcodec.h:1656
Main libavformat public API header.
#define AVINDEX_KEYFRAME
Definition: avformat.h:811
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:453
#define AVFMT_FLAG_IGNIDX
Ignore index.
Definition: avformat.h:1365
#define AVFMT_NOFILE
Demuxer will use avio_open, no opened file should be provided by the caller.
Definition: avformat.h:458
int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
Allocate and read the payload of a packet and initialize its fields with default values.
Definition: utils.c:310
@ AVSTREAM_PARSE_TIMESTAMPS
full parsing and interpolation of timestamps for frames not starting on a packet boundary
Definition: avformat.h:796
@ AVSTREAM_PARSE_FULL_RAW
full parsing and repack with timestamp and position generation by parser for raw this assumes that ea...
Definition: avformat.h:798
@ AVSTREAM_PARSE_HEADERS
Only parse headers, do not repack.
Definition: avformat.h:795
@ AVSTREAM_PARSE_FULL
full parsing and repack
Definition: avformat.h:794
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:253
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:40
uint64_t avio_rb64(AVIOContext *s)
Definition: aviobuf.c:902
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:342
unsigned int avio_rb16(AVIOContext *s)
Definition: aviobuf.c:766
int avio_feof(AVIOContext *s)
Similar to feof() but also returns nonzero on read errors.
Definition: aviobuf.c:364
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:337
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:633
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:750
int avio_get_str(AVIOContext *pb, int maxlen, char *buf, int buflen)
Read a string from pb into buf.
Definition: aviobuf.c:860
unsigned int avio_rb32(AVIOContext *s)
Definition: aviobuf.c:781
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:624
int ffio_ensure_seekback(AVIOContext *s, int64_t buf_size)
Ensures that the requested seekback buffer size will be available.
Definition: aviobuf.c:998
int ffio_limit(AVIOContext *s, int size)
Definition: utils.c:244
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
#define AV_RL32
Definition: intreadwrite.h:146
#define flags(name, subs,...)
Definition: cbs_av1.c:561
static int FUNC() sps(CodedBitstreamContext *ctx, RWContext *rw, H264RawSPS *current)
#define s(width, name)
Definition: cbs_vp9.c:257
audio channel layout utility functions
static struct @321 state
#define fail()
Definition: checkasm.h:133
#define FFSWAP(type, a, b)
Definition: common.h:108
#define FFMIN(a, b)
Definition: common.h:105
#define MKTAG(a, b, c, d)
Definition: common.h:478
#define MKBETAG(a, b, c, d)
Definition: common.h:479
#define FFMAX(a, b)
Definition: common.h:103
#define NULL
Definition: coverity.c:32
Public dictionary API.
double value
Definition: eval.c:98
static int64_t start_time
Definition: ffplay.c:332
static int nb_streams
Definition: ffprobe.c:283
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:545
#define AV_CH_LAYOUT_MONO
@ AV_CODEC_ID_SIPR
Definition: codec_id.h:465
@ AV_CODEC_ID_NONE
Definition: codec_id.h:47
@ AV_CODEC_ID_RA_288
Definition: codec_id.h:411
@ AV_CODEC_ID_COOK
Definition: codec_id.h:444
@ AV_CODEC_ID_ATRAC3
Definition: codec_id.h:455
@ AV_CODEC_ID_AAC
Definition: codec_id.h:426
@ AV_CODEC_ID_AC3
Definition: codec_id.h:427
@ AV_CODEC_ID_RA_144
Definition: codec_id.h:410
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding.
Definition: avcodec.h:215
@ AVDISCARD_ALL
discard all
Definition: avcodec.h:236
@ AVDISCARD_NONKEY
discard all frames except keyframes
Definition: avcodec.h:235
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:634
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:410
void av_packet_move_ref(AVPacket *dst, AVPacket *src)
Move every field in src to dst and reset src.
Definition: avpacket.c:690
void av_shrink_packet(AVPacket *pkt, int size)
Reduce packet size, correctly zeroing padding.
Definition: avpacket.c:114
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:99
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4509
int av_add_index_entry(AVStream *st, int64_t pos, int64_t timestamp, int size, int distance, int flags)
Add an index entry into a sorted list.
Definition: utils.c:2013
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:70
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
#define AVERROR_EOF
End of file.
Definition: error.h:55
#define AVERROR(e)
Definition: error.h:43
#define AV_LOG_TRACE
Extremely verbose debugging, useful for libav* development.
Definition: log.h:220
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:215
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:200
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
int av_reduce(int *dst_num, int *dst_den, int64_t num, int64_t den, int64_t max)
Reduce a fraction.
Definition: rational.c:35
int64_t av_rescale(int64_t a, int64_t b, int64_t c)
Rescale a 64-bit integer with rounding to nearest.
Definition: mathematics.c:129
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:237
#define av_fourcc2str(fourcc)
Definition: avutil.h:348
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
@ AVMEDIA_TYPE_DATA
Opaque data information usually continuous.
Definition: avutil.h:203
@ AVMEDIA_TYPE_UNKNOWN
Usually treated as AVMEDIA_TYPE_DATA.
Definition: avutil.h:200
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
#define AV_TIME_BASE
Internal time base represented as integer.
Definition: avutil.h:254
int index
Definition: gxfenc.c:89
cl_device_type type
const char * key
int i
Definition: input.c:407
#define AV_WL32(p, v)
Definition: intreadwrite.h:426
int ff_seek_frame_binary(AVFormatContext *s, int stream_index, int64_t target_ts, int flags)
Perform a binary search using av_index_search_timestamp() and AVInputFormat.read_timestamp().
Definition: utils.c:2145
void ff_free_stream(AVFormatContext *s, AVStream *st)
Definition: utils.c:4428
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4945
int ff_get_extradata(AVFormatContext *s, AVCodecParameters *par, AVIOContext *pb, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0 and f...
Definition: utils.c:3332
enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
Definition: utils.c:3131
static int read_probe(const AVProbeData *pd)
Definition: jvdec.c:55
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:145
static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)
Definition: libcdio.c:153
version
Definition: libkvazaar.c:326
uint8_t w
Definition: llviddspenc.c:39
uint32_t tag
Definition: movenc.c:1611
const char * name
Definition: qsvenc.c:46
const char *const ff_rm_metadata[4]
Definition: rm.c:24
const AVCodecTag ff_rm_codec_tags[]
Definition: rm.c:31
static void rm_read_metadata(AVFormatContext *s, AVIOContext *pb, int wide)
Definition: rmdec.c:93
#define DEINT_ID_INT0
no interleaving needed
Definition: rmdec.c:37
AVInputFormat ff_rm_demuxer
Definition: rmdec.c:1159
static int rm_read_header(AVFormatContext *s)
Definition: rmdec.c:544
static int rm_probe(const AVProbeData *p)
Definition: rmdec.c:1088
static int rm_assemble_video_frame(AVFormatContext *s, AVIOContext *pb, RMDemuxContext *rm, RMStream *vst, AVPacket *pkt, int len, int *pseq, int64_t *timestamp)
Definition: rmdec.c:770
static int rm_read_index(AVFormatContext *s)
this function assumes that the demuxer has already seeked to the start of the INDX chunk,...
Definition: rmdec.c:430
static int get_num(AVIOContext *pb, int *len)
Definition: rmdec.c:678
#define DEINT_ID_VBRS
VBR case for AAC.
Definition: rmdec.c:41
static int rm_read_audio_stream_info(AVFormatContext *s, AVIOContext *pb, AVStream *st, RMStream *ast, int read_all)
Definition: rmdec.c:124
static int rm_read_close(AVFormatContext *s)
Definition: rmdec.c:1078
#define RAW_PACKET_SIZE
Definition: rmdec.c:695
static int64_t rm_read_dts(AVFormatContext *s, int stream_index, int64_t *ppos, int64_t pos_limit)
Definition: rmdec.c:1101
RMStream * ff_rm_alloc_rmstream(void)
Definition: rmdec.c:107
#define DEINT_ID_VBRF
VBR case for AAC.
Definition: rmdec.c:40
static int rm_sync(AVFormatContext *s, int64_t *timestamp, int *flags, int *stream_index, int64_t *pos)
Definition: rmdec.c:697
static int readfull(AVFormatContext *s, AVIOContext *pb, uint8_t *dst, int n)
Definition: rmdec.c:887
static int rm_read_extradata(AVFormatContext *s, AVIOContext *pb, AVCodecParameters *par, unsigned size)
Definition: rmdec.c:84
#define DEINT_ID_SIPR
interleaving for Sipro
Definition: rmdec.c:39
void ff_rm_free_rmstream(RMStream *rms)
Definition: rmdec.c:116
AVInputFormat ff_ivr_demuxer
Definition: rmdec.c:1418
int ff_rm_retrieve_cache(AVFormatContext *s, AVIOContext *pb, AVStream *st, RMStream *ast, AVPacket *pkt)
Retrieve one cached packet from the rm-context.
Definition: rmdec.c:986
static int ivr_probe(const AVProbeData *p)
Definition: rmdec.c:1179
static int rm_read_header_old(AVFormatContext *s)
Definition: rmdec.c:488
#define DEINT_ID_GENR
interleaving for Cooker/ATRAC
Definition: rmdec.c:36
AVInputFormat ff_rdt_demuxer
Definition: rmdec.c:1171
static void get_strl(AVIOContext *pb, char *buf, int buf_size, int len)
Definition: rmdec.c:71
static int ivr_read_header(AVFormatContext *s)
Definition: rmdec.c:1188
static int rm_read_multi(AVFormatContext *s, AVIOContext *pb, AVStream *st, char *mime)
Definition: rmdec.c:503
static int ivr_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: rmdec.c:1337
int ff_rm_parse_packet(AVFormatContext *s, AVIOContext *pb, AVStream *st, RMStream *ast, int len, AVPacket *pkt, int *seq, int flags, int64_t timestamp)
Parse one rm-stream packet from the input bytestream.
Definition: rmdec.c:898
int ff_rm_read_mdpr_codecdata(AVFormatContext *s, AVIOContext *pb, AVStream *st, RMStream *rst, unsigned int codec_data_size, const uint8_t *mime)
Read the MDPR chunk, which contains stream-specific codec initialization parameters.
Definition: rmdec.c:316
static void rm_ac3_swap_bytes(AVStream *st, AVPacket *pkt)
Definition: rmdec.c:873
static int rm_read_seek(AVFormatContext *s, int stream_index, int64_t pts, int flags)
Definition: rmdec.c:1147
#define DEINT_ID_INT4
interleaving for 28.8
Definition: rmdec.c:38
static void get_str8(AVIOContext *pb, char *buf, int buf_size)
Definition: rmdec.c:79
static int rm_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: rmdec.c:1018
void ff_rm_reorder_sipr_data(uint8_t *buf, int sub_packet_h, int framesize)
Perform 4-bit block reordering for SIPR data.
Definition: rmsipr.c:41
const unsigned char ff_sipr_subpk_size[4]
Definition: rmsipr.c:25
#define FF_ARRAY_ELEMS(a)
unsigned int pos
Definition: spdifenc.c:412
This struct describes the properties of an encoded stream.
Definition: codec_par.h:52
int extradata_size
Size of the extradata content in bytes.
Definition: codec_par.h:78
uint64_t channel_layout
Audio only.
Definition: codec_par.h:162
int channels
Audio only.
Definition: codec_par.h:166
int width
Video only.
Definition: codec_par.h:126
int64_t bit_rate
The average bitrate of the encoded data (in bits per second).
Definition: codec_par.h:89
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:56
int block_align
Audio only.
Definition: codec_par.h:177
uint32_t codec_tag
Additional information about the codec (corresponds to the AVI FOURCC).
Definition: codec_par.h:64
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: codec_par.h:74
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
int sample_rate
Audio only.
Definition: codec_par.h:170
Format I/O context.
Definition: avformat.h:1232
Bytestream IO Context.
Definition: avio.h:161
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:260
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:645
This structure stores compressed data.
Definition: packet.h:346
int stream_index
Definition: packet.h:371
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:375
int size
Definition: packet.h:370
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:362
uint8_t * data
Definition: packet.h:369
int64_t pos
byte position in stream, -1 if unknown
Definition: packet.h:389
This structure contains the data a format has to probe a file.
Definition: avformat.h:441
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:443
int num
Numerator.
Definition: rational.h:59
int den
Denominator.
Definition: rational.h:60
Stream structure.
Definition: avformat.h:873
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1038
enum AVDiscard discard
Selects which packets can be discarded at will and do not need to be demuxed.
Definition: avformat.h:928
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:922
void * priv_data
Definition: avformat.h:888
int id
Format-specific stream ID.
Definition: avformat.h:880
int index
stream index in AVFormatContext
Definition: avformat.h:874
int64_t start_time
Decoding: pts of the first frame of the stream in presentation order, in stream time base.
Definition: avformat.h:912
AVRational avg_frame_rate
Average framerate.
Definition: avformat.h:946
enum AVStreamParseType need_parsing
Definition: avformat.h:1081
AVRational r_frame_rate
Real base framerate of the stream.
Definition: avformat.h:1015
int data_end
Definition: rmdec.c:66
int audio_stream_num
Stream number for audio packets.
Definition: rmdec.c:64
int current_stream
Definition: rmdec.c:62
int remaining_len
Definition: rmdec.c:63
int nb_packets
Definition: rmdec.c:60
int old_format
Definition: rmdec.c:61
int audio_pkt_cnt
Output packet counter.
Definition: rmdec.c:65
Definition: rmdec.c:43
int64_t audiotimestamp
Audio descrambling matrix parameters.
Definition: rmdec.c:51
int videobufsize
current assembled frame size
Definition: rmdec.c:45
int coded_framesize
Descrambling parameters from container.
Definition: rmdec.c:53
int slices
Definition: rmdec.c:48
int sub_packet_cnt
Definition: rmdec.c:52
int sub_packet_size
Definition: rmdec.c:53
int cur_slice
Definition: rmdec.c:48
int audio_framesize
Definition: rmdec.c:54
int32_t deint_id
Length of each subpacket.
Definition: rmdec.c:56
int sub_packet_h
Definition: rmdec.c:53
int curpic_num
picture number of current frame
Definition: rmdec.c:47
AVPacket pkt
place to store merged video frame / reordered audio data
Definition: rmdec.c:44
int sub_packet_lengths[16]
Audio frame size from container.
Definition: rmdec.c:55
int videobufpos
position for the next slice in the video buffer
Definition: rmdec.c:46
int64_t pktpos
first slice position in file
Definition: rmdec.c:49
#define avpriv_request_sample(...)
#define av_freep(p)
#define av_log(a,...)
int64_t duration
Definition: movenc.c:64
AVPacket * pkt
Definition: movenc.c:59
static int64_t pts
int size
else temp
Definition: vf_mcdeint.c:259
if(ret< 0)
Definition: vf_mcdeint.c:282
static const uint8_t offset[127][2]
Definition: vf_spp.c:107
int len