00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022 #include "libavutil/avstring.h"
00023 #include "libavutil/intreadwrite.h"
00024 #include "avformat.h"
00025 #include "rm.h"
00026
00027 struct RMStream {
00028 AVPacket pkt;
00029 int videobufsize;
00030 int videobufpos;
00031 int curpic_num;
00032 int cur_slice, slices;
00033 int64_t pktpos;
00034
00035 int64_t audiotimestamp;
00036 int sub_packet_cnt;
00037 int sub_packet_size, sub_packet_h, coded_framesize;
00038 int audio_framesize;
00039 int sub_packet_lengths[16];
00040 };
00041
00042 typedef struct {
00043 int nb_packets;
00044 int old_format;
00045 int current_stream;
00046 int remaining_len;
00047 int audio_stream_num;
00048 int audio_pkt_cnt;
00049 } RMDemuxContext;
00050
00051 static inline void get_strl(ByteIOContext *pb, char *buf, int buf_size, int len)
00052 {
00053 int i;
00054 char *q, r;
00055
00056 q = buf;
00057 for(i=0;i<len;i++) {
00058 r = get_byte(pb);
00059 if (i < buf_size - 1)
00060 *q++ = r;
00061 }
00062 if (buf_size > 0) *q = '\0';
00063 }
00064
00065 static void get_str8(ByteIOContext *pb, char *buf, int buf_size)
00066 {
00067 get_strl(pb, buf, buf_size, get_byte(pb));
00068 }
00069
00070 static void rm_read_metadata(AVFormatContext *s, int wide)
00071 {
00072 char buf[1024];
00073 int i;
00074 for (i=0; i<FF_ARRAY_ELEMS(ff_rm_metadata); i++) {
00075 int len = wide ? get_be16(s->pb) : get_byte(s->pb);
00076 get_strl(s->pb, buf, sizeof(buf), len);
00077 av_metadata_set(&s->metadata, ff_rm_metadata[i], buf);
00078 }
00079 }
00080
00081 RMStream *ff_rm_alloc_rmstream (void)
00082 {
00083 RMStream *rms = av_mallocz(sizeof(RMStream));
00084 rms->curpic_num = -1;
00085 return rms;
00086 }
00087
00088 void ff_rm_free_rmstream (RMStream *rms)
00089 {
00090 av_free_packet(&rms->pkt);
00091 }
00092
00093 static int rm_read_audio_stream_info(AVFormatContext *s, ByteIOContext *pb,
00094 AVStream *st, RMStream *ast, int read_all)
00095 {
00096 char buf[256];
00097 uint32_t version;
00098
00099
00100 version = get_be32(pb);
00101 if (((version >> 16) & 0xff) == 3) {
00102 int64_t startpos = url_ftell(pb);
00103 url_fskip(pb, 14);
00104 rm_read_metadata(s, 0);
00105 if ((startpos + (version & 0xffff)) >= url_ftell(pb) + 2) {
00106
00107 get_byte(pb);
00108 get_str8(pb, buf, sizeof(buf));
00109 }
00110
00111 if ((startpos + (version & 0xffff)) > url_ftell(pb))
00112 url_fskip(pb, (version & 0xffff) + startpos - url_ftell(pb));
00113 st->codec->sample_rate = 8000;
00114 st->codec->channels = 1;
00115 st->codec->codec_type = CODEC_TYPE_AUDIO;
00116 st->codec->codec_id = CODEC_ID_RA_144;
00117 } else {
00118 int flavor, sub_packet_h, coded_framesize, sub_packet_size;
00119
00120 get_be32(pb);
00121 get_be32(pb);
00122 get_be16(pb);
00123 get_be32(pb);
00124 flavor= get_be16(pb);
00125 ast->coded_framesize = coded_framesize = get_be32(pb);
00126 get_be32(pb);
00127 get_be32(pb);
00128 get_be32(pb);
00129 ast->sub_packet_h = sub_packet_h = get_be16(pb);
00130 st->codec->block_align= get_be16(pb);
00131 ast->sub_packet_size = sub_packet_size = get_be16(pb);
00132 get_be16(pb);
00133 if (((version >> 16) & 0xff) == 5) {
00134 get_be16(pb); get_be16(pb); get_be16(pb);
00135 }
00136 st->codec->sample_rate = get_be16(pb);
00137 get_be32(pb);
00138 st->codec->channels = get_be16(pb);
00139 if (((version >> 16) & 0xff) == 5) {
00140 get_be32(pb);
00141 get_buffer(pb, buf, 4);
00142 buf[4] = 0;
00143 } else {
00144 get_str8(pb, buf, sizeof(buf));
00145 get_str8(pb, buf, sizeof(buf));
00146 }
00147 st->codec->codec_type = CODEC_TYPE_AUDIO;
00148 if (!strcmp(buf, "dnet")) {
00149 st->codec->codec_id = CODEC_ID_AC3;
00150 st->need_parsing = AVSTREAM_PARSE_FULL;
00151 } else if (!strcmp(buf, "28_8")) {
00152 st->codec->codec_id = CODEC_ID_RA_288;
00153 st->codec->extradata_size= 0;
00154 ast->audio_framesize = st->codec->block_align;
00155 st->codec->block_align = coded_framesize;
00156
00157 if(ast->audio_framesize >= UINT_MAX / sub_packet_h){
00158 av_log(s, AV_LOG_ERROR, "ast->audio_framesize * sub_packet_h too large\n");
00159 return -1;
00160 }
00161
00162 av_new_packet(&ast->pkt, ast->audio_framesize * sub_packet_h);
00163 } else if ((!strcmp(buf, "cook")) || (!strcmp(buf, "atrc")) || (!strcmp(buf, "sipr"))) {
00164 int codecdata_length;
00165 get_be16(pb); get_byte(pb);
00166 if (((version >> 16) & 0xff) == 5)
00167 get_byte(pb);
00168 codecdata_length = get_be32(pb);
00169 if(codecdata_length + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)codecdata_length){
00170 av_log(s, AV_LOG_ERROR, "codecdata_length too large\n");
00171 return -1;
00172 }
00173
00174 if(sub_packet_size <= 0){
00175 av_log(s, AV_LOG_ERROR, "sub_packet_size is invalid\n");
00176 return -1;
00177 }
00178
00179 if (!strcmp(buf, "cook")) st->codec->codec_id = CODEC_ID_COOK;
00180 else if (!strcmp(buf, "sipr")) st->codec->codec_id = CODEC_ID_SIPR;
00181 else st->codec->codec_id = CODEC_ID_ATRAC3;
00182 st->codec->extradata_size= codecdata_length;
00183 st->codec->extradata= av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
00184 get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
00185 ast->audio_framesize = st->codec->block_align;
00186 st->codec->block_align = ast->sub_packet_size;
00187
00188 if(ast->audio_framesize >= UINT_MAX / sub_packet_h){
00189 av_log(s, AV_LOG_ERROR, "rm->audio_framesize * sub_packet_h too large\n");
00190 return -1;
00191 }
00192
00193 av_new_packet(&ast->pkt, ast->audio_framesize * sub_packet_h);
00194 } else if (!strcmp(buf, "raac") || !strcmp(buf, "racp")) {
00195 int codecdata_length;
00196 get_be16(pb); get_byte(pb);
00197 if (((version >> 16) & 0xff) == 5)
00198 get_byte(pb);
00199 st->codec->codec_id = CODEC_ID_AAC;
00200 codecdata_length = get_be32(pb);
00201 if(codecdata_length + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)codecdata_length){
00202 av_log(s, AV_LOG_ERROR, "codecdata_length too large\n");
00203 return -1;
00204 }
00205 if (codecdata_length >= 1) {
00206 st->codec->extradata_size = codecdata_length - 1;
00207 st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
00208 get_byte(pb);
00209 get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
00210 }
00211 } else {
00212 st->codec->codec_id = CODEC_ID_NONE;
00213 av_strlcpy(st->codec->codec_name, buf, sizeof(st->codec->codec_name));
00214 }
00215 if (read_all) {
00216 get_byte(pb);
00217 get_byte(pb);
00218 get_byte(pb);
00219 rm_read_metadata(s, 0);
00220 }
00221 }
00222 return 0;
00223 }
00224
00225 int
00226 ff_rm_read_mdpr_codecdata (AVFormatContext *s, ByteIOContext *pb,
00227 AVStream *st, RMStream *rst, int codec_data_size)
00228 {
00229 unsigned int v;
00230 int size;
00231 int64_t codec_pos;
00232
00233 av_set_pts_info(st, 64, 1, 1000);
00234 codec_pos = url_ftell(pb);
00235 v = get_be32(pb);
00236 if (v == MKTAG(0xfd, 'a', 'r', '.')) {
00237
00238 if (rm_read_audio_stream_info(s, pb, st, rst, 0))
00239 return -1;
00240 } else {
00241 int fps, fps2;
00242 if (get_le32(pb) != MKTAG('V', 'I', 'D', 'O')) {
00243 fail1:
00244 av_log(st->codec, AV_LOG_ERROR, "Unsupported video codec\n");
00245 goto skip;
00246 }
00247 st->codec->codec_tag = get_le32(pb);
00248
00249 if ( st->codec->codec_tag != MKTAG('R', 'V', '1', '0')
00250 && st->codec->codec_tag != MKTAG('R', 'V', '2', '0')
00251 && st->codec->codec_tag != MKTAG('R', 'V', '3', '0')
00252 && st->codec->codec_tag != MKTAG('R', 'V', '4', '0')
00253 && st->codec->codec_tag != MKTAG('R', 'V', 'T', 'R'))
00254 goto fail1;
00255 st->codec->width = get_be16(pb);
00256 st->codec->height = get_be16(pb);
00257 st->codec->time_base.num= 1;
00258 fps= get_be16(pb);
00259 st->codec->codec_type = CODEC_TYPE_VIDEO;
00260 get_be32(pb);
00261 fps2= get_be16(pb);
00262 get_be16(pb);
00263
00264 st->codec->extradata_size= codec_data_size - (url_ftell(pb) - codec_pos);
00265
00266 if(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)st->codec->extradata_size){
00267
00268 av_log(s, AV_LOG_ERROR, "st->codec->extradata_size too large\n");
00269 return -1;
00270 }
00271 st->codec->extradata= av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
00272 if (!st->codec->extradata)
00273 return AVERROR(ENOMEM);
00274 get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
00275
00276
00277 st->codec->time_base.den = fps * st->codec->time_base.num;
00278 switch(((uint8_t*)st->codec->extradata)[4]>>4){
00279 case 1: st->codec->codec_id = CODEC_ID_RV10; break;
00280 case 2: st->codec->codec_id = CODEC_ID_RV20; break;
00281 case 3: st->codec->codec_id = CODEC_ID_RV30; break;
00282 case 4: st->codec->codec_id = CODEC_ID_RV40; break;
00283 default: goto fail1;
00284 }
00285 }
00286
00287 skip:
00288
00289 size = url_ftell(pb) - codec_pos;
00290 url_fskip(pb, codec_data_size - size);
00291
00292 return 0;
00293 }
00294
00295
00296 static int rm_read_header_old(AVFormatContext *s, AVFormatParameters *ap)
00297 {
00298 RMDemuxContext *rm = s->priv_data;
00299 AVStream *st;
00300
00301 rm->old_format = 1;
00302 st = av_new_stream(s, 0);
00303 if (!st)
00304 return -1;
00305 st->priv_data = ff_rm_alloc_rmstream();
00306 return rm_read_audio_stream_info(s, s->pb, st, st->priv_data, 1);
00307 }
00308
00309 static int rm_read_header(AVFormatContext *s, AVFormatParameters *ap)
00310 {
00311 RMDemuxContext *rm = s->priv_data;
00312 AVStream *st;
00313 ByteIOContext *pb = s->pb;
00314 unsigned int tag;
00315 int tag_size;
00316 unsigned int start_time, duration;
00317 char buf[128];
00318 int flags = 0;
00319
00320 tag = get_le32(pb);
00321 if (tag == MKTAG('.', 'r', 'a', 0xfd)) {
00322
00323 return rm_read_header_old(s, ap);
00324 } else if (tag != MKTAG('.', 'R', 'M', 'F')) {
00325 return AVERROR(EIO);
00326 }
00327
00328 get_be32(pb);
00329 get_be16(pb);
00330 get_be32(pb);
00331 get_be32(pb);
00332
00333 for(;;) {
00334 if (url_feof(pb))
00335 return -1;
00336 tag = get_le32(pb);
00337 tag_size = get_be32(pb);
00338 get_be16(pb);
00339 #if 0
00340 printf("tag=%c%c%c%c (%08x) size=%d\n",
00341 (tag) & 0xff,
00342 (tag >> 8) & 0xff,
00343 (tag >> 16) & 0xff,
00344 (tag >> 24) & 0xff,
00345 tag,
00346 tag_size);
00347 #endif
00348 if (tag_size < 10 && tag != MKTAG('D', 'A', 'T', 'A'))
00349 return -1;
00350 switch(tag) {
00351 case MKTAG('P', 'R', 'O', 'P'):
00352
00353 get_be32(pb);
00354 get_be32(pb);
00355 get_be32(pb);
00356 get_be32(pb);
00357 get_be32(pb);
00358 get_be32(pb);
00359 get_be32(pb);
00360 get_be32(pb);
00361 get_be32(pb);
00362 get_be16(pb);
00363 flags = get_be16(pb);
00364 break;
00365 case MKTAG('C', 'O', 'N', 'T'):
00366 rm_read_metadata(s, 1);
00367 break;
00368 case MKTAG('M', 'D', 'P', 'R'):
00369 st = av_new_stream(s, 0);
00370 if (!st)
00371 return AVERROR(ENOMEM);
00372 st->id = get_be16(pb);
00373 get_be32(pb);
00374 st->codec->bit_rate = get_be32(pb);
00375 get_be32(pb);
00376 get_be32(pb);
00377 start_time = get_be32(pb);
00378 get_be32(pb);
00379 duration = get_be32(pb);
00380 st->start_time = start_time;
00381 st->duration = duration;
00382 get_str8(pb, buf, sizeof(buf));
00383 get_str8(pb, buf, sizeof(buf));
00384 st->codec->codec_type = CODEC_TYPE_DATA;
00385 st->priv_data = ff_rm_alloc_rmstream();
00386 if (ff_rm_read_mdpr_codecdata(s, s->pb, st, st->priv_data,
00387 get_be32(pb)) < 0)
00388 return -1;
00389 break;
00390 case MKTAG('D', 'A', 'T', 'A'):
00391 goto header_end;
00392 default:
00393
00394 url_fskip(pb, tag_size - 10);
00395 break;
00396 }
00397 }
00398 header_end:
00399 rm->nb_packets = get_be32(pb);
00400 if (!rm->nb_packets && (flags & 4))
00401 rm->nb_packets = 3600 * 25;
00402 get_be32(pb);
00403 return 0;
00404 }
00405
00406 static int get_num(ByteIOContext *pb, int *len)
00407 {
00408 int n, n1;
00409
00410 n = get_be16(pb);
00411 (*len)-=2;
00412 n &= 0x7FFF;
00413 if (n >= 0x4000) {
00414 return n - 0x4000;
00415 } else {
00416 n1 = get_be16(pb);
00417 (*len)-=2;
00418 return (n << 16) | n1;
00419 }
00420 }
00421
00422
00423 #define RAW_PACKET_SIZE 1000
00424
00425 static int sync(AVFormatContext *s, int64_t *timestamp, int *flags, int *stream_index, int64_t *pos){
00426 RMDemuxContext *rm = s->priv_data;
00427 ByteIOContext *pb = s->pb;
00428 int len, num, res, i;
00429 AVStream *st;
00430 uint32_t state=0xFFFFFFFF;
00431
00432 while(!url_feof(pb)){
00433 *pos= url_ftell(pb) - 3;
00434 if(rm->remaining_len > 0){
00435 num= rm->current_stream;
00436 len= rm->remaining_len;
00437 *timestamp = AV_NOPTS_VALUE;
00438 *flags= 0;
00439 }else{
00440 state= (state<<8) + get_byte(pb);
00441
00442 if(state == MKBETAG('I', 'N', 'D', 'X')){
00443 len = get_be16(pb) - 6;
00444 if(len<0)
00445 continue;
00446 goto skip;
00447 }
00448
00449 if(state > (unsigned)0xFFFF || state < 12)
00450 continue;
00451 len=state;
00452 state= 0xFFFFFFFF;
00453
00454 num = get_be16(pb);
00455 *timestamp = get_be32(pb);
00456 res= get_byte(pb);
00457 *flags = get_byte(pb);
00458
00459
00460 len -= 12;
00461 }
00462 for(i=0;i<s->nb_streams;i++) {
00463 st = s->streams[i];
00464 if (num == st->id)
00465 break;
00466 }
00467 if (i == s->nb_streams) {
00468 skip:
00469
00470 url_fskip(pb, len);
00471 rm->remaining_len -= len;
00472 continue;
00473 }
00474 *stream_index= i;
00475
00476 return len;
00477 }
00478 return -1;
00479 }
00480
00481 static int rm_assemble_video_frame(AVFormatContext *s, ByteIOContext *pb,
00482 RMDemuxContext *rm, RMStream *vst,
00483 AVPacket *pkt, int len)
00484 {
00485 int hdr, seq, pic_num, len2, pos;
00486 int type;
00487
00488 hdr = get_byte(pb); len--;
00489 type = hdr >> 6;
00490
00491 if(type != 3){
00492 seq = get_byte(pb); len--;
00493 }
00494 if(type != 1){
00495 len2 = get_num(pb, &len);
00496 pos = get_num(pb, &len);
00497 pic_num = get_byte(pb); len--;
00498 }
00499 if(len<0)
00500 return -1;
00501 rm->remaining_len = len;
00502 if(type&1){
00503 if(type == 3)
00504 len= len2;
00505 if(rm->remaining_len < len)
00506 return -1;
00507 rm->remaining_len -= len;
00508 if(av_new_packet(pkt, len + 9) < 0)
00509 return AVERROR(EIO);
00510 pkt->data[0] = 0;
00511 AV_WL32(pkt->data + 1, 1);
00512 AV_WL32(pkt->data + 5, 0);
00513 get_buffer(pb, pkt->data + 9, len);
00514 return 0;
00515 }
00516
00517
00518 if((seq & 0x7F) == 1 || vst->curpic_num != pic_num){
00519 vst->slices = ((hdr & 0x3F) << 1) + 1;
00520 vst->videobufsize = len2 + 8*vst->slices + 1;
00521 av_free_packet(&vst->pkt);
00522 if(av_new_packet(&vst->pkt, vst->videobufsize) < 0)
00523 return AVERROR(ENOMEM);
00524 vst->videobufpos = 8*vst->slices + 1;
00525 vst->cur_slice = 0;
00526 vst->curpic_num = pic_num;
00527 vst->pktpos = url_ftell(pb);
00528 }
00529 if(type == 2)
00530 len = FFMIN(len, pos);
00531
00532 if(++vst->cur_slice > vst->slices)
00533 return 1;
00534 AV_WL32(vst->pkt.data - 7 + 8*vst->cur_slice, 1);
00535 AV_WL32(vst->pkt.data - 3 + 8*vst->cur_slice, vst->videobufpos - 8*vst->slices - 1);
00536 if(vst->videobufpos + len > vst->videobufsize)
00537 return 1;
00538 if (get_buffer(pb, vst->pkt.data + vst->videobufpos, len) != len)
00539 return AVERROR(EIO);
00540 vst->videobufpos += len;
00541 rm->remaining_len-= len;
00542
00543 if(type == 2 || (vst->videobufpos) == vst->videobufsize){
00544 vst->pkt.data[0] = vst->cur_slice-1;
00545 *pkt= vst->pkt;
00546 vst->pkt.data= NULL;
00547 vst->pkt.size= 0;
00548 if(vst->slices != vst->cur_slice)
00549 memmove(pkt->data + 1 + 8*vst->cur_slice, pkt->data + 1 + 8*vst->slices,
00550 vst->videobufpos - 1 - 8*vst->slices);
00551 pkt->size = vst->videobufpos + 8*(vst->cur_slice - vst->slices);
00552 pkt->pts = AV_NOPTS_VALUE;
00553 pkt->pos = vst->pktpos;
00554 return 0;
00555 }
00556
00557 return 1;
00558 }
00559
00560 static inline void
00561 rm_ac3_swap_bytes (AVStream *st, AVPacket *pkt)
00562 {
00563 uint8_t *ptr;
00564 int j;
00565
00566 if (st->codec->codec_id == CODEC_ID_AC3) {
00567 ptr = pkt->data;
00568 for (j=0;j<pkt->size;j+=2) {
00569 FFSWAP(int, ptr[0], ptr[1]);
00570 ptr += 2;
00571 }
00572 }
00573 }
00574
00575 int
00576 ff_rm_parse_packet (AVFormatContext *s, ByteIOContext *pb,
00577 AVStream *st, RMStream *ast, int len, AVPacket *pkt,
00578 int *seq, int *flags, int64_t *timestamp)
00579 {
00580 RMDemuxContext *rm = s->priv_data;
00581
00582 if (st->codec->codec_type == CODEC_TYPE_VIDEO) {
00583 rm->current_stream= st->id;
00584 if(rm_assemble_video_frame(s, pb, rm, ast, pkt, len))
00585 return -1;
00586 } else if (st->codec->codec_type == CODEC_TYPE_AUDIO) {
00587 if ((st->codec->codec_id == CODEC_ID_RA_288) ||
00588 (st->codec->codec_id == CODEC_ID_COOK) ||
00589 (st->codec->codec_id == CODEC_ID_ATRAC3) ||
00590 (st->codec->codec_id == CODEC_ID_SIPR)) {
00591 int x;
00592 int sps = ast->sub_packet_size;
00593 int cfs = ast->coded_framesize;
00594 int h = ast->sub_packet_h;
00595 int y = ast->sub_packet_cnt;
00596 int w = ast->audio_framesize;
00597
00598 if (*flags & 2)
00599 y = ast->sub_packet_cnt = 0;
00600 if (!y)
00601 ast->audiotimestamp = *timestamp;
00602
00603 switch(st->codec->codec_id) {
00604 case CODEC_ID_RA_288:
00605 for (x = 0; x < h/2; x++)
00606 get_buffer(pb, ast->pkt.data+x*2*w+y*cfs, cfs);
00607 break;
00608 case CODEC_ID_ATRAC3:
00609 case CODEC_ID_COOK:
00610 for (x = 0; x < w/sps; x++)
00611 get_buffer(pb, ast->pkt.data+sps*(h*x+((h+1)/2)*(y&1)+(y>>1)), sps);
00612 break;
00613 }
00614
00615 if (++(ast->sub_packet_cnt) < h)
00616 return -1;
00617 else {
00618 ast->sub_packet_cnt = 0;
00619 rm->audio_stream_num = st->index;
00620 rm->audio_pkt_cnt = h * w / st->codec->block_align - 1;
00621
00622 av_new_packet(pkt, st->codec->block_align);
00623 memcpy(pkt->data, ast->pkt.data, st->codec->block_align);
00624 *timestamp = ast->audiotimestamp;
00625 *flags = 2;
00626 }
00627 } else if (st->codec->codec_id == CODEC_ID_AAC) {
00628 int x;
00629 rm->audio_stream_num = st->index;
00630 ast->sub_packet_cnt = (get_be16(pb) & 0xf0) >> 4;
00631 if (ast->sub_packet_cnt) {
00632 for (x = 0; x < ast->sub_packet_cnt; x++)
00633 ast->sub_packet_lengths[x] = get_be16(pb);
00634
00635 rm->audio_pkt_cnt = ast->sub_packet_cnt - 1;
00636 av_get_packet(pb, pkt, ast->sub_packet_lengths[0]);
00637 *flags = 2;
00638 }
00639 } else {
00640 av_get_packet(pb, pkt, len);
00641 rm_ac3_swap_bytes(st, pkt);
00642 }
00643 } else
00644 av_get_packet(pb, pkt, len);
00645
00646 if( (st->discard >= AVDISCARD_NONKEY && !(*flags&2))
00647 || st->discard >= AVDISCARD_ALL){
00648 av_free_packet(pkt);
00649 return -1;
00650 }
00651
00652 pkt->stream_index = st->index;
00653
00654 #if 0
00655 if (st->codec->codec_type == CODEC_TYPE_VIDEO) {
00656 if(st->codec->codec_id == CODEC_ID_RV20){
00657 int seq= 128*(pkt->data[2]&0x7F) + (pkt->data[3]>>1);
00658 av_log(s, AV_LOG_DEBUG, "%d %"PRId64" %d\n", *timestamp, *timestamp*512LL/25, seq);
00659
00660 seq |= (*timestamp&~0x3FFF);
00661 if(seq - *timestamp > 0x2000) seq -= 0x4000;
00662 if(seq - *timestamp < -0x2000) seq += 0x4000;
00663 }
00664 }
00665 #endif
00666
00667 pkt->pts= *timestamp;
00668 if (*flags & 2)
00669 pkt->flags |= PKT_FLAG_KEY;
00670
00671 return st->codec->codec_type == CODEC_TYPE_AUDIO ? rm->audio_pkt_cnt : 0;
00672 }
00673
00674 int
00675 ff_rm_retrieve_cache (AVFormatContext *s, ByteIOContext *pb,
00676 AVStream *st, RMStream *ast, AVPacket *pkt)
00677 {
00678 RMDemuxContext *rm = s->priv_data;
00679
00680 assert (rm->audio_pkt_cnt > 0);
00681
00682 if (st->codec->codec_id == CODEC_ID_AAC)
00683 av_get_packet(pb, pkt, ast->sub_packet_lengths[ast->sub_packet_cnt - rm->audio_pkt_cnt]);
00684 else {
00685 av_new_packet(pkt, st->codec->block_align);
00686 memcpy(pkt->data, ast->pkt.data + st->codec->block_align *
00687 (ast->sub_packet_h * ast->audio_framesize / st->codec->block_align - rm->audio_pkt_cnt),
00688 st->codec->block_align);
00689 }
00690 rm->audio_pkt_cnt--;
00691 pkt->flags = 0;
00692 pkt->stream_index = st->index;
00693
00694 return rm->audio_pkt_cnt;
00695 }
00696
00697 static int rm_read_packet(AVFormatContext *s, AVPacket *pkt)
00698 {
00699 RMDemuxContext *rm = s->priv_data;
00700 ByteIOContext *pb = s->pb;
00701 AVStream *st;
00702 int i, len;
00703 int64_t timestamp, pos;
00704 int flags;
00705
00706 if (rm->audio_pkt_cnt) {
00707
00708 st = s->streams[rm->audio_stream_num];
00709 ff_rm_retrieve_cache(s, s->pb, st, st->priv_data, pkt);
00710 } else if (rm->old_format) {
00711 RMStream *ast;
00712
00713 st = s->streams[0];
00714 ast = st->priv_data;
00715 if (st->codec->codec_id == CODEC_ID_RA_288) {
00716 int x, y;
00717
00718 for (y = 0; y < ast->sub_packet_h; y++)
00719 for (x = 0; x < ast->sub_packet_h/2; x++)
00720 if (get_buffer(pb, ast->pkt.data+x*2*ast->audio_framesize+y*ast->coded_framesize, ast->coded_framesize) <= 0)
00721 return AVERROR(EIO);
00722 rm->audio_stream_num = 0;
00723 rm->audio_pkt_cnt = ast->sub_packet_h * ast->audio_framesize / st->codec->block_align - 1;
00724
00725 av_new_packet(pkt, st->codec->block_align);
00726 memcpy(pkt->data, ast->pkt.data, st->codec->block_align);
00727 pkt->flags |= PKT_FLAG_KEY;
00728 pkt->stream_index = 0;
00729 } else {
00730
00731 len = RAW_PACKET_SIZE;
00732 len= av_get_packet(pb, pkt, len);
00733 pkt->stream_index = 0;
00734 if (len <= 0) {
00735 return AVERROR(EIO);
00736 }
00737 pkt->size = len;
00738 }
00739 rm_ac3_swap_bytes(st, pkt);
00740 } else {
00741 int seq=1;
00742 resync:
00743 len=sync(s, ×tamp, &flags, &i, &pos);
00744 if(len<0)
00745 return AVERROR(EIO);
00746 st = s->streams[i];
00747
00748 if (ff_rm_parse_packet (s, s->pb, st, st->priv_data, len, pkt,
00749 &seq, &flags, ×tamp) < 0)
00750 goto resync;
00751
00752 if((flags&2) && (seq&0x7F) == 1)
00753 av_add_index_entry(st, pos, timestamp, 0, 0, AVINDEX_KEYFRAME);
00754 }
00755
00756 return 0;
00757 }
00758
00759 static int rm_read_close(AVFormatContext *s)
00760 {
00761 int i;
00762
00763 for (i=0;i<s->nb_streams;i++)
00764 ff_rm_free_rmstream(s->streams[i]->priv_data);
00765
00766 return 0;
00767 }
00768
00769 static int rm_probe(AVProbeData *p)
00770 {
00771
00772 if ((p->buf[0] == '.' && p->buf[1] == 'R' &&
00773 p->buf[2] == 'M' && p->buf[3] == 'F' &&
00774 p->buf[4] == 0 && p->buf[5] == 0) ||
00775 (p->buf[0] == '.' && p->buf[1] == 'r' &&
00776 p->buf[2] == 'a' && p->buf[3] == 0xfd))
00777 return AVPROBE_SCORE_MAX;
00778 else
00779 return 0;
00780 }
00781
00782 static int64_t rm_read_dts(AVFormatContext *s, int stream_index,
00783 int64_t *ppos, int64_t pos_limit)
00784 {
00785 RMDemuxContext *rm = s->priv_data;
00786 int64_t pos, dts;
00787 int stream_index2, flags, len, h;
00788
00789 pos = *ppos;
00790
00791 if(rm->old_format)
00792 return AV_NOPTS_VALUE;
00793
00794 url_fseek(s->pb, pos, SEEK_SET);
00795 rm->remaining_len=0;
00796 for(;;){
00797 int seq=1;
00798 AVStream *st;
00799
00800 len=sync(s, &dts, &flags, &stream_index2, &pos);
00801 if(len<0)
00802 return AV_NOPTS_VALUE;
00803
00804 st = s->streams[stream_index2];
00805 if (st->codec->codec_type == CODEC_TYPE_VIDEO) {
00806 h= get_byte(s->pb); len--;
00807 if(!(h & 0x40)){
00808 seq = get_byte(s->pb); len--;
00809 }
00810 }
00811
00812 if((flags&2) && (seq&0x7F) == 1){
00813
00814 av_add_index_entry(st, pos, dts, 0, 0, AVINDEX_KEYFRAME);
00815 if(stream_index2 == stream_index)
00816 break;
00817 }
00818
00819 url_fskip(s->pb, len);
00820 }
00821 *ppos = pos;
00822 return dts;
00823 }
00824
00825 AVInputFormat rm_demuxer = {
00826 "rm",
00827 NULL_IF_CONFIG_SMALL("RealMedia format"),
00828 sizeof(RMDemuxContext),
00829 rm_probe,
00830 rm_read_header,
00831 rm_read_packet,
00832 rm_read_close,
00833 NULL,
00834 rm_read_dts,
00835 };
00836
00837 AVInputFormat rdt_demuxer = {
00838 "rdt",
00839 NULL_IF_CONFIG_SMALL("RDT demuxer"),
00840 sizeof(RMDemuxContext),
00841 NULL,
00842 NULL,
00843 NULL,
00844 rm_read_close,
00845 };