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