You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

774 lines
23KB

  1. /*
  2. * "Real" compatible mux and demux.
  3. * Copyright (c) 2000, 2001 Fabrice Bellard.
  4. *
  5. * This library is free software; you can redistribute it and/or
  6. * modify it under the terms of the GNU Lesser General Public
  7. * License as published by the Free Software Foundation; either
  8. * version 2 of the License, or (at your option) any later version.
  9. *
  10. * This library is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13. * Lesser General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU Lesser General Public
  16. * License along with this library; if not, write to the Free Software
  17. * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  18. */
  19. #include "avformat.h"
  20. /* in ms */
  21. #define BUFFER_DURATION 0
  22. typedef struct {
  23. int nb_packets;
  24. int packet_total_size;
  25. int packet_max_size;
  26. /* codec related output */
  27. int bit_rate;
  28. float frame_rate;
  29. int nb_frames; /* current frame number */
  30. int total_frames; /* total number of frames */
  31. int num;
  32. AVCodecContext *enc;
  33. } StreamInfo;
  34. typedef struct {
  35. StreamInfo streams[2];
  36. StreamInfo *audio_stream, *video_stream;
  37. int data_pos; /* position of the data after the header */
  38. int nb_packets;
  39. } RMContext;
  40. static void put_str(ByteIOContext *s, const char *tag)
  41. {
  42. put_be16(s,strlen(tag));
  43. while (*tag) {
  44. put_byte(s, *tag++);
  45. }
  46. }
  47. static void put_str8(ByteIOContext *s, const char *tag)
  48. {
  49. put_byte(s, strlen(tag));
  50. while (*tag) {
  51. put_byte(s, *tag++);
  52. }
  53. }
  54. static void rv10_write_header(AVFormatContext *ctx,
  55. int data_size, int index_pos)
  56. {
  57. RMContext *rm = ctx->priv_data;
  58. ByteIOContext *s = &ctx->pb;
  59. StreamInfo *stream;
  60. unsigned char *data_offset_ptr, *start_ptr;
  61. const char *desc, *mimetype;
  62. int nb_packets, packet_total_size, packet_max_size, size, packet_avg_size, i;
  63. int bit_rate, v, duration, flags, data_pos;
  64. start_ptr = s->buf_ptr;
  65. put_tag(s, ".RMF");
  66. put_be32(s,18); /* header size */
  67. put_be16(s,0);
  68. put_be32(s,0);
  69. put_be32(s,4 + ctx->nb_streams); /* num headers */
  70. put_tag(s,"PROP");
  71. put_be32(s, 50);
  72. put_be16(s, 0);
  73. packet_max_size = 0;
  74. packet_total_size = 0;
  75. nb_packets = 0;
  76. bit_rate = 0;
  77. duration = 0;
  78. for(i=0;i<ctx->nb_streams;i++) {
  79. StreamInfo *stream = &rm->streams[i];
  80. bit_rate += stream->bit_rate;
  81. if (stream->packet_max_size > packet_max_size)
  82. packet_max_size = stream->packet_max_size;
  83. nb_packets += stream->nb_packets;
  84. packet_total_size += stream->packet_total_size;
  85. /* select maximum duration */
  86. v = (int) (1000.0 * (float)stream->total_frames / stream->frame_rate);
  87. if (v > duration)
  88. duration = v;
  89. }
  90. put_be32(s, bit_rate); /* max bit rate */
  91. put_be32(s, bit_rate); /* avg bit rate */
  92. put_be32(s, packet_max_size); /* max packet size */
  93. if (nb_packets > 0)
  94. packet_avg_size = packet_total_size / nb_packets;
  95. else
  96. packet_avg_size = 0;
  97. put_be32(s, packet_avg_size); /* avg packet size */
  98. put_be32(s, nb_packets); /* num packets */
  99. put_be32(s, duration); /* duration */
  100. put_be32(s, BUFFER_DURATION); /* preroll */
  101. put_be32(s, index_pos); /* index offset */
  102. /* computation of data the data offset */
  103. data_offset_ptr = s->buf_ptr;
  104. put_be32(s, 0); /* data offset : will be patched after */
  105. put_be16(s, ctx->nb_streams); /* num streams */
  106. flags = 1 | 2; /* save allowed & perfect play */
  107. if (url_is_streamed(s))
  108. flags |= 4; /* live broadcast */
  109. put_be16(s, flags);
  110. /* comments */
  111. put_tag(s,"CONT");
  112. size = strlen(ctx->title) + strlen(ctx->author) + strlen(ctx->copyright) +
  113. strlen(ctx->comment) + 4 * 2 + 10;
  114. put_be32(s,size);
  115. put_be16(s,0);
  116. put_str(s, ctx->title);
  117. put_str(s, ctx->author);
  118. put_str(s, ctx->copyright);
  119. put_str(s, ctx->comment);
  120. for(i=0;i<ctx->nb_streams;i++) {
  121. int codec_data_size;
  122. stream = &rm->streams[i];
  123. if (stream->enc->codec_type == CODEC_TYPE_AUDIO) {
  124. desc = "The Audio Stream";
  125. mimetype = "audio/x-pn-realaudio";
  126. codec_data_size = 73;
  127. } else {
  128. desc = "The Video Stream";
  129. mimetype = "video/x-pn-realvideo";
  130. codec_data_size = 34;
  131. }
  132. put_tag(s,"MDPR");
  133. size = 10 + 9 * 4 + strlen(desc) + strlen(mimetype) + codec_data_size;
  134. put_be32(s, size);
  135. put_be16(s, 0);
  136. put_be16(s, i); /* stream number */
  137. put_be32(s, stream->bit_rate); /* max bit rate */
  138. put_be32(s, stream->bit_rate); /* avg bit rate */
  139. put_be32(s, stream->packet_max_size); /* max packet size */
  140. if (stream->nb_packets > 0)
  141. packet_avg_size = stream->packet_total_size /
  142. stream->nb_packets;
  143. else
  144. packet_avg_size = 0;
  145. put_be32(s, packet_avg_size); /* avg packet size */
  146. put_be32(s, 0); /* start time */
  147. put_be32(s, BUFFER_DURATION); /* preroll */
  148. /* duration */
  149. if (url_is_streamed(s) || !stream->total_frames)
  150. put_be32(s, (int)(3600 * 1000));
  151. else
  152. put_be32(s, (int)(stream->total_frames * 1000 / stream->frame_rate));
  153. put_str8(s, desc);
  154. put_str8(s, mimetype);
  155. put_be32(s, codec_data_size);
  156. if (stream->enc->codec_type == CODEC_TYPE_AUDIO) {
  157. int coded_frame_size, fscode, sample_rate;
  158. sample_rate = stream->enc->sample_rate;
  159. coded_frame_size = (stream->enc->bit_rate *
  160. stream->enc->frame_size) / (8 * sample_rate);
  161. /* audio codec info */
  162. put_tag(s, ".ra");
  163. put_byte(s, 0xfd);
  164. put_be32(s, 0x00040000); /* version */
  165. put_tag(s, ".ra4");
  166. put_be32(s, 0x01b53530); /* stream length */
  167. put_be16(s, 4); /* unknown */
  168. put_be32(s, 0x39); /* header size */
  169. switch(sample_rate) {
  170. case 48000:
  171. case 24000:
  172. case 12000:
  173. fscode = 1;
  174. break;
  175. default:
  176. case 44100:
  177. case 22050:
  178. case 11025:
  179. fscode = 2;
  180. break;
  181. case 32000:
  182. case 16000:
  183. case 8000:
  184. fscode = 3;
  185. }
  186. put_be16(s, fscode); /* codec additional info, for AC3, seems
  187. to be a frequency code */
  188. /* special hack to compensate rounding errors... */
  189. if (coded_frame_size == 557)
  190. coded_frame_size--;
  191. put_be32(s, coded_frame_size); /* frame length */
  192. put_be32(s, 0x51540); /* unknown */
  193. put_be32(s, 0x249f0); /* unknown */
  194. put_be32(s, 0x249f0); /* unknown */
  195. put_be16(s, 0x01);
  196. /* frame length : seems to be very important */
  197. put_be16(s, coded_frame_size);
  198. put_be32(s, 0); /* unknown */
  199. put_be16(s, stream->enc->sample_rate); /* sample rate */
  200. put_be32(s, 0x10); /* unknown */
  201. put_be16(s, stream->enc->channels);
  202. put_str8(s, "Int0"); /* codec name */
  203. put_str8(s, "dnet"); /* codec name */
  204. put_be16(s, 0); /* title length */
  205. put_be16(s, 0); /* author length */
  206. put_be16(s, 0); /* copyright length */
  207. put_byte(s, 0); /* end of header */
  208. } else {
  209. /* video codec info */
  210. put_be32(s,34); /* size */
  211. put_tag(s,"VIDORV10");
  212. put_be16(s, stream->enc->width);
  213. put_be16(s, stream->enc->height);
  214. put_be16(s, (int) stream->frame_rate); /* frames per seconds ? */
  215. put_be32(s,0); /* unknown meaning */
  216. put_be16(s, (int) stream->frame_rate); /* unknown meaning */
  217. put_be32(s,0); /* unknown meaning */
  218. put_be16(s, 8); /* unknown meaning */
  219. /* Seems to be the codec version: only use basic H263. The next
  220. versions seems to add a diffential DC coding as in
  221. MPEG... nothing new under the sun */
  222. put_be32(s,0x10000000);
  223. //put_be32(s,0x10003000);
  224. }
  225. }
  226. /* patch data offset field */
  227. data_pos = s->buf_ptr - start_ptr;
  228. rm->data_pos = data_pos;
  229. data_offset_ptr[0] = data_pos >> 24;
  230. data_offset_ptr[1] = data_pos >> 16;
  231. data_offset_ptr[2] = data_pos >> 8;
  232. data_offset_ptr[3] = data_pos;
  233. /* data stream */
  234. put_tag(s,"DATA");
  235. put_be32(s,data_size + 10 + 8);
  236. put_be16(s,0);
  237. put_be32(s, nb_packets); /* number of packets */
  238. put_be32(s,0); /* next data header */
  239. }
  240. static void write_packet_header(AVFormatContext *ctx, StreamInfo *stream,
  241. int length, int key_frame)
  242. {
  243. int timestamp;
  244. ByteIOContext *s = &ctx->pb;
  245. stream->nb_packets++;
  246. stream->packet_total_size += length;
  247. if (length > stream->packet_max_size)
  248. stream->packet_max_size = length;
  249. put_be16(s,0); /* version */
  250. put_be16(s,length + 12);
  251. put_be16(s, stream->num); /* stream number */
  252. timestamp = (1000 * (float)stream->nb_frames) / stream->frame_rate;
  253. put_be32(s, timestamp); /* timestamp */
  254. put_byte(s, 0); /* reserved */
  255. put_byte(s, key_frame ? 2 : 0); /* flags */
  256. }
  257. static int rm_write_header(AVFormatContext *s)
  258. {
  259. RMContext *rm = s->priv_data;
  260. StreamInfo *stream;
  261. int n;
  262. AVCodecContext *codec;
  263. for(n=0;n<s->nb_streams;n++) {
  264. s->streams[n]->id = n;
  265. codec = &s->streams[n]->codec;
  266. stream = &rm->streams[n];
  267. memset(stream, 0, sizeof(StreamInfo));
  268. stream->num = n;
  269. stream->bit_rate = codec->bit_rate;
  270. stream->enc = codec;
  271. switch(codec->codec_type) {
  272. case CODEC_TYPE_AUDIO:
  273. rm->audio_stream = stream;
  274. stream->frame_rate = (float)codec->sample_rate / (float)codec->frame_size;
  275. /* XXX: dummy values */
  276. stream->packet_max_size = 1024;
  277. stream->nb_packets = 0;
  278. stream->total_frames = stream->nb_packets;
  279. break;
  280. case CODEC_TYPE_VIDEO:
  281. rm->video_stream = stream;
  282. stream->frame_rate = (float)codec->frame_rate / (float)FRAME_RATE_BASE;
  283. /* XXX: dummy values */
  284. stream->packet_max_size = 4096;
  285. stream->nb_packets = 0;
  286. stream->total_frames = stream->nb_packets;
  287. break;
  288. default:
  289. av_abort();
  290. }
  291. }
  292. rv10_write_header(s, 0, 0);
  293. put_flush_packet(&s->pb);
  294. return 0;
  295. }
  296. static int rm_write_audio(AVFormatContext *s, UINT8 *buf, int size)
  297. {
  298. UINT8 *buf1;
  299. RMContext *rm = s->priv_data;
  300. ByteIOContext *pb = &s->pb;
  301. StreamInfo *stream = rm->audio_stream;
  302. int i;
  303. /* XXX: suppress this malloc */
  304. buf1= (UINT8*) av_malloc( size * sizeof(UINT8) );
  305. write_packet_header(s, stream, size, stream->enc->key_frame);
  306. /* for AC3, the words seems to be reversed */
  307. for(i=0;i<size;i+=2) {
  308. buf1[i] = buf[i+1];
  309. buf1[i+1] = buf[i];
  310. }
  311. put_buffer(pb, buf1, size);
  312. put_flush_packet(pb);
  313. stream->nb_frames++;
  314. av_free(buf1);
  315. return 0;
  316. }
  317. static int rm_write_video(AVFormatContext *s, UINT8 *buf, int size)
  318. {
  319. RMContext *rm = s->priv_data;
  320. ByteIOContext *pb = &s->pb;
  321. StreamInfo *stream = rm->video_stream;
  322. int key_frame = stream->enc->key_frame;
  323. /* XXX: this is incorrect: should be a parameter */
  324. /* Well, I spent some time finding the meaning of these bits. I am
  325. not sure I understood everything, but it works !! */
  326. #if 1
  327. write_packet_header(s, stream, size + 7, key_frame);
  328. /* bit 7: '1' if final packet of a frame converted in several packets */
  329. put_byte(pb, 0x81);
  330. /* bit 7: '1' if I frame. bits 6..0 : sequence number in current
  331. frame starting from 1 */
  332. if (key_frame) {
  333. put_byte(pb, 0x81);
  334. } else {
  335. put_byte(pb, 0x01);
  336. }
  337. put_be16(pb, 0x4000 | (size)); /* total frame size */
  338. put_be16(pb, 0x4000 | (size)); /* offset from the start or the end */
  339. #else
  340. /* full frame */
  341. write_packet_header(s, size + 6);
  342. put_byte(pb, 0xc0);
  343. put_be16(pb, 0x4000 | size); /* total frame size */
  344. put_be16(pb, 0x4000 + packet_number * 126); /* position in stream */
  345. #endif
  346. put_byte(pb, stream->nb_frames & 0xff);
  347. put_buffer(pb, buf, size);
  348. put_flush_packet(pb);
  349. stream->nb_frames++;
  350. return 0;
  351. }
  352. static int rm_write_packet(AVFormatContext *s, int stream_index,
  353. UINT8 *buf, int size, int force_pts)
  354. {
  355. if (s->streams[stream_index]->codec.codec_type ==
  356. CODEC_TYPE_AUDIO)
  357. return rm_write_audio(s, buf, size);
  358. else
  359. return rm_write_video(s, buf, size);
  360. }
  361. static int rm_write_trailer(AVFormatContext *s)
  362. {
  363. RMContext *rm = s->priv_data;
  364. int data_size, index_pos, i;
  365. ByteIOContext *pb = &s->pb;
  366. if (!url_is_streamed(&s->pb)) {
  367. /* end of file: finish to write header */
  368. index_pos = url_fseek(pb, 0, SEEK_CUR);
  369. data_size = index_pos - rm->data_pos;
  370. /* index */
  371. put_tag(pb, "INDX");
  372. put_be32(pb, 10 + 10 * s->nb_streams);
  373. put_be16(pb, 0);
  374. for(i=0;i<s->nb_streams;i++) {
  375. put_be32(pb, 0); /* zero indices */
  376. put_be16(pb, i); /* stream number */
  377. put_be32(pb, 0); /* next index */
  378. }
  379. /* undocumented end header */
  380. put_be32(pb, 0);
  381. put_be32(pb, 0);
  382. url_fseek(pb, 0, SEEK_SET);
  383. for(i=0;i<s->nb_streams;i++)
  384. rm->streams[i].total_frames = rm->streams[i].nb_frames;
  385. rv10_write_header(s, data_size, index_pos);
  386. } else {
  387. /* undocumented end header */
  388. put_be32(pb, 0);
  389. put_be32(pb, 0);
  390. }
  391. put_flush_packet(pb);
  392. return 0;
  393. }
  394. /***************************************************/
  395. static void get_str(ByteIOContext *pb, char *buf, int buf_size)
  396. {
  397. int len, i;
  398. char *q;
  399. len = get_be16(pb);
  400. q = buf;
  401. for(i=0;i<len;i++) {
  402. if (i < buf_size - 1)
  403. *q++ = get_byte(pb);
  404. }
  405. *q = '\0';
  406. }
  407. static void get_str8(ByteIOContext *pb, char *buf, int buf_size)
  408. {
  409. int len, i;
  410. char *q;
  411. len = get_byte(pb);
  412. q = buf;
  413. for(i=0;i<len;i++) {
  414. if (i < buf_size - 1)
  415. *q++ = get_byte(pb);
  416. }
  417. *q = '\0';
  418. }
  419. static int rm_read_header(AVFormatContext *s, AVFormatParameters *ap)
  420. {
  421. RMContext *rm = s->priv_data;
  422. AVStream *st;
  423. ByteIOContext *pb = &s->pb;
  424. unsigned int tag, v;
  425. int tag_size, size, codec_data_size, i;
  426. INT64 codec_pos;
  427. unsigned int h263_hack_version;
  428. char buf[128];
  429. int flags = 0;
  430. if (get_le32(pb) != MKTAG('.', 'R', 'M', 'F'))
  431. return -EIO;
  432. get_be32(pb); /* header size */
  433. get_be16(pb);
  434. get_be32(pb);
  435. get_be32(pb); /* number of headers */
  436. for(;;) {
  437. if (url_feof(pb))
  438. goto fail;
  439. tag = get_le32(pb);
  440. tag_size = get_be32(pb);
  441. get_be16(pb);
  442. #if 0
  443. printf("tag=%c%c%c%c (%08x) size=%d\n",
  444. (tag) & 0xff,
  445. (tag >> 8) & 0xff,
  446. (tag >> 16) & 0xff,
  447. (tag >> 24) & 0xff,
  448. tag,
  449. tag_size);
  450. #endif
  451. if (tag_size < 10)
  452. goto fail;
  453. switch(tag) {
  454. case MKTAG('P', 'R', 'O', 'P'):
  455. /* file header */
  456. get_be32(pb); /* max bit rate */
  457. get_be32(pb); /* avg bit rate */
  458. get_be32(pb); /* max packet size */
  459. get_be32(pb); /* avg packet size */
  460. get_be32(pb); /* nb packets */
  461. get_be32(pb); /* duration */
  462. get_be32(pb); /* preroll */
  463. get_be32(pb); /* index offset */
  464. get_be32(pb); /* data offset */
  465. get_be16(pb); /* nb streams */
  466. flags = get_be16(pb); /* flags */
  467. break;
  468. case MKTAG('C', 'O', 'N', 'T'):
  469. get_str(pb, s->title, sizeof(s->title));
  470. get_str(pb, s->author, sizeof(s->author));
  471. get_str(pb, s->copyright, sizeof(s->copyright));
  472. get_str(pb, s->comment, sizeof(s->comment));
  473. break;
  474. case MKTAG('M', 'D', 'P', 'R'):
  475. st = av_mallocz(sizeof(AVStream));
  476. if (!st)
  477. goto fail;
  478. s->streams[s->nb_streams++] = st;
  479. st->id = get_be16(pb);
  480. get_be32(pb); /* max bit rate */
  481. st->codec.bit_rate = get_be32(pb); /* bit rate */
  482. get_be32(pb); /* max packet size */
  483. get_be32(pb); /* avg packet size */
  484. get_be32(pb); /* start time */
  485. get_be32(pb); /* preroll */
  486. get_be32(pb); /* duration */
  487. get_str8(pb, buf, sizeof(buf)); /* desc */
  488. get_str8(pb, buf, sizeof(buf)); /* mimetype */
  489. codec_data_size = get_be32(pb);
  490. codec_pos = url_ftell(pb);
  491. v = get_be32(pb);
  492. if (v == MKTAG(0xfd, 'a', 'r', '.')) {
  493. /* ra type header */
  494. get_be32(pb); /* version */
  495. get_be32(pb); /* .ra4 */
  496. get_be32(pb);
  497. get_be16(pb);
  498. get_be32(pb); /* header size */
  499. get_be16(pb); /* add codec info */
  500. get_be32(pb); /* coded frame size */
  501. get_be32(pb); /* ??? */
  502. get_be32(pb); /* ??? */
  503. get_be32(pb); /* ??? */
  504. get_be16(pb); /* 1 */
  505. get_be16(pb); /* coded frame size */
  506. get_be32(pb);
  507. st->codec.sample_rate = get_be16(pb);
  508. get_be32(pb);
  509. st->codec.channels = get_be16(pb);
  510. get_str8(pb, buf, sizeof(buf)); /* desc */
  511. get_str8(pb, buf, sizeof(buf)); /* desc */
  512. st->codec.codec_type = CODEC_TYPE_AUDIO;
  513. if (!strcmp(buf, "dnet")) {
  514. st->codec.codec_id = CODEC_ID_AC3;
  515. } else {
  516. st->codec.codec_id = CODEC_ID_NONE;
  517. pstrcpy(st->codec.codec_name, sizeof(st->codec.codec_name),
  518. buf);
  519. }
  520. } else {
  521. if (get_le32(pb) != MKTAG('V', 'I', 'D', 'O')) {
  522. fail1:
  523. fprintf(stderr, "Unsupported video codec\n");
  524. goto fail;
  525. }
  526. st->codec.codec_tag = get_le32(pb);
  527. if (st->codec.codec_tag != MKTAG('R', 'V', '1', '0'))
  528. goto fail1;
  529. st->codec.width = get_be16(pb);
  530. st->codec.height = get_be16(pb);
  531. st->codec.frame_rate = get_be16(pb) * FRAME_RATE_BASE;
  532. st->codec.codec_type = CODEC_TYPE_VIDEO;
  533. get_be32(pb);
  534. get_be16(pb);
  535. get_be32(pb);
  536. get_be16(pb);
  537. /* modification of h263 codec version (!) */
  538. h263_hack_version = get_be32(pb);
  539. switch(h263_hack_version) {
  540. case 0x10000000:
  541. case 0x10003000:
  542. case 0x10003001:
  543. st->codec.sub_id = h263_hack_version;
  544. st->codec.codec_id = CODEC_ID_RV10;
  545. break;
  546. default:
  547. /* not handled */
  548. st->codec.codec_id = CODEC_ID_NONE;
  549. break;
  550. }
  551. }
  552. /* skip codec info */
  553. size = url_ftell(pb) - codec_pos;
  554. url_fskip(pb, codec_data_size - size);
  555. break;
  556. case MKTAG('D', 'A', 'T', 'A'):
  557. goto header_end;
  558. default:
  559. /* unknown tag: skip it */
  560. url_fskip(pb, tag_size - 10);
  561. break;
  562. }
  563. }
  564. header_end:
  565. rm->nb_packets = get_be32(pb); /* number of packets */
  566. if (!rm->nb_packets && (flags & 4))
  567. rm->nb_packets = 3600 * 25;
  568. get_be32(pb); /* next data header */
  569. return 0;
  570. fail:
  571. for(i=0;i<s->nb_streams;i++) {
  572. av_free(s->streams[i]);
  573. }
  574. return -EIO;
  575. }
  576. static int get_num(ByteIOContext *pb, int *len)
  577. {
  578. int n, n1;
  579. n = get_be16(pb);
  580. (*len)-=2;
  581. if (n >= 0x4000) {
  582. return n - 0x4000;
  583. } else {
  584. n1 = get_be16(pb);
  585. (*len)-=2;
  586. return (n << 16) | n1;
  587. }
  588. }
  589. static int rm_read_packet(AVFormatContext *s, AVPacket *pkt)
  590. {
  591. RMContext *rm = s->priv_data;
  592. ByteIOContext *pb = &s->pb;
  593. AVStream *st;
  594. int len, num, timestamp, i, tmp, j;
  595. UINT8 *ptr;
  596. int flags;
  597. redo:
  598. if (rm->nb_packets == 0)
  599. return -EIO;
  600. get_be16(pb);
  601. len = get_be16(pb);
  602. if (len < 12)
  603. return -EIO;
  604. num = get_be16(pb);
  605. timestamp = get_be32(pb);
  606. get_byte(pb); /* reserved */
  607. flags = get_byte(pb); /* flags */
  608. rm->nb_packets--;
  609. len -= 12;
  610. st = NULL;
  611. for(i=0;i<s->nb_streams;i++) {
  612. st = s->streams[i];
  613. if (num == st->id)
  614. break;
  615. }
  616. if (i == s->nb_streams) {
  617. /* skip packet if unknown number */
  618. url_fskip(pb, len);
  619. goto redo;
  620. }
  621. if (st->codec.codec_type == CODEC_TYPE_VIDEO) {
  622. int full_frame, h, pic_num;
  623. h= get_byte(pb);
  624. if ((h & 0xc0) == 0xc0) {
  625. int len2, pos;
  626. full_frame = 1;
  627. len2= get_num(pb, &len);
  628. pos = get_num(pb, &len);
  629. //printf("pos:%d\n",len);
  630. len -= 2;
  631. } else {
  632. int seq, frame_size, pos;
  633. full_frame = 0;
  634. seq = get_byte(pb);
  635. frame_size = get_num(pb, &len);
  636. pos = get_num(pb, &len);
  637. //printf("seq:%d, size:%d, pos:%d\n",seq,frame_size,pos);
  638. len -= 3;
  639. }
  640. /* picture number */
  641. pic_num= get_byte(pb);
  642. //XXX/FIXME/HACK, demuxer should be fixed to send complete frames ...
  643. if(st->codec.slice_offset==NULL) st->codec.slice_offset= (int*)malloc(sizeof(int));
  644. st->codec.slice_count= full_frame;
  645. st->codec.slice_offset[0]= 0;
  646. }
  647. av_new_packet(pkt, len);
  648. pkt->stream_index = i;
  649. get_buffer(pb, pkt->data, len);
  650. /* for AC3, needs to swap bytes */
  651. if (st->codec.codec_id == CODEC_ID_AC3) {
  652. ptr = pkt->data;
  653. for(j=0;j<len;j+=2) {
  654. tmp = ptr[0];
  655. ptr[0] = ptr[1];
  656. ptr[1] = tmp;
  657. ptr += 2;
  658. }
  659. }
  660. return 0;
  661. }
  662. static int rm_read_close(AVFormatContext *s)
  663. {
  664. return 0;
  665. }
  666. static int rm_probe(AVProbeData *p)
  667. {
  668. /* check file header */
  669. if (p->buf_size <= 32)
  670. return 0;
  671. if (p->buf[0] == '.' && p->buf[1] == 'R' &&
  672. p->buf[2] == 'M' && p->buf[3] == 'F' &&
  673. p->buf[4] == 0 && p->buf[5] == 0)
  674. return AVPROBE_SCORE_MAX;
  675. else
  676. return 0;
  677. }
  678. static AVInputFormat rm_iformat = {
  679. "rm",
  680. "rm format",
  681. sizeof(RMContext),
  682. rm_probe,
  683. rm_read_header,
  684. rm_read_packet,
  685. rm_read_close,
  686. };
  687. static AVOutputFormat rm_oformat = {
  688. "rm",
  689. "rm format",
  690. "audio/x-pn-realaudio",
  691. "rm,ra",
  692. sizeof(RMContext),
  693. CODEC_ID_AC3,
  694. CODEC_ID_RV10,
  695. rm_write_header,
  696. rm_write_packet,
  697. rm_write_trailer,
  698. };
  699. int rm_init(void)
  700. {
  701. av_register_input_format(&rm_iformat);
  702. av_register_output_format(&rm_oformat);
  703. return 0;
  704. }