--- /dev/null
+/*
+ * NUT (de)muxer based on initial draft
+ * Copyright (c) 2003 Alex Beregszaszi
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ * NUT DRAFT can be found in MPlayer CVS at DOCS/tech/mpcf.txt
+ *
+ * Compatible with draft version 20030906
+ *
+ */
+
+/*
+ * TODO:
+ * - checksumming
+ * - correct rate denom/nom and sample_mul
+ * - correct timestamp handling
+ * - index writing
+ * - info and index packet reading support
+ * - startcode searching for broken streams
+ * - subpacket support
+ * - handling of codec specific headers
+*/
+
+//#define DEBUG 1
+
+#include "avformat.h"
+#include "mpegaudio.h"
+#include "avi.h"
+
+//from /dev/random
+
+#define MAIN_STARTCODE (0xF9526A6200000000ULL + ('N'<<24) + ('U'<<16) + ('T'<<8) + 'M')
+#define STREAM_STARTCODE (0xD667773F00000000ULL + ('N'<<24) + ('U'<<16) + ('T'<<8) + 'S')
+#define KEYFRAME_STARTCODE (0xCB86308700000000ULL + ('N'<<24) + ('U'<<16) + ('T'<<8) + 'K')
+#define INDEX_STARTCODE (0xEBFCDE0E00000000ULL + ('N'<<24) + ('U'<<16) + ('T'<<8) + 'X')
+#define INFO_STARTCODE (0xA37B643500000000ULL + ('N'<<24) + ('U'<<16) + ('T'<<8) + 'I')
+
+typedef struct {
+ int curr_frame_start;
+ int last_frame_size;
+ int curr_frame_size;
+} NUTContext;
+
+static int bytes_left(ByteIOContext *bc)
+{
+ return bc->buf_end - bc->buf_ptr;
+}
+
+static uint64_t get_v(ByteIOContext *bc)
+{
+ uint64_t val = 0;
+
+// for (; bytes_left(s)*8 > 0; )
+ for(; bytes_left(bc) > 0; )
+ {
+ int tmp = get_byte(bc);
+
+ if (tmp&0x80)
+ val= (val<<7) + tmp - 0x80;
+ else
+ return (val<<7) + tmp;
+ }
+ return -1;
+}
+
+static int64_t get_s(ByteIOContext *bc)
+{
+ int64_t v = get_v(bc) + 1;
+
+ if (v&1)
+ return -(v>>1);
+ else
+ return (v>>1);
+}
+
+static int get_b(ByteIOContext *bc, char *data, int maxlen)
+{
+ int i, len;
+
+ len = get_v(bc);
+ for (i = 0; i < len && i < maxlen; i++)
+ data[i] = get_byte(bc);
+ if (i < len)
+ {
+ len = i;
+ for (i = 0; i < len; i++)
+ get_byte(bc);
+ }
+
+ return 0;
+}
+
+static int get_bi(ByteIOContext *bc)
+{
+ int i, len, val;
+
+ len = get_v(bc);
+ if(len > 4) return -1;
+
+ val = 0;
+ for (i = 0; i < len; i++) {
+ val |= get_byte(bc) << (i * 8);
+ }
+
+ return val;
+}
+
+static int get_packetheader(NUTContext *nut, ByteIOContext *bc)
+{
+ nut->curr_frame_start = url_ftell(bc);
+ nut->curr_frame_size = get_v(bc);
+ nut->last_frame_size = get_v(bc);
+ dprintf("Packet: fwd: %d bwd: %d\n",
+ nut->curr_frame_size, nut->last_frame_size);
+
+ return 0;
+}
+
+/**
+ *
+ */
+static int get_length(uint64_t val){
+ int i;
+
+ for (i=7; ; i+=7)
+ if ((val>>i) == 0)
+ return i;
+
+ return 7; //not reached
+}
+
+static int put_v(ByteIOContext *bc, uint64_t val)
+{
+ int i;
+
+// if (bytes_left(s)*8 < 9)
+// return -1;
+
+ if (bytes_left(bc) < 1)
+ return -1;
+
+ val &= 0x7FFFFFFFFFFFFFFFULL; // FIXME can only encode upto 63 bits currently
+ i= get_length(val);
+
+ for (i-=7; i>0; i-=7){
+ put_byte(bc, 0x80 | (val>>i));
+ }
+
+ put_byte(bc, val&0x7f);
+
+ return 0;
+}
+
+static int put_s(ByteIOContext *bc, uint64_t val)
+{
+ if (val<=0)
+ return put_v(bc, -2*val);
+ else
+ return put_v(bc, 2*val-1);
+}
+
+static int put_b(ByteIOContext *bc, char *data, int len)
+{
+ int i;
+
+ put_v(bc, len);
+ for (i = 0; i < len; i++)
+ put_byte(bc, data[i]);
+
+ return 0;
+}
+
+static int put_bi(ByteIOContext *bc, int val)
+{
+ put_v(bc, 4);
+ put_le32(bc, val);
+ return 0;
+}
+
+static int put_packetheader(NUTContext *nut, ByteIOContext *bc, int max_size)
+{
+ put_flush_packet(bc);
+ nut->curr_frame_start = url_ftell(bc);
+ nut->curr_frame_size = max_size;
+
+ /* packet header */
+ put_v(bc, nut->curr_frame_size); /* forward ptr */
+ put_v(bc, nut->last_frame_size); /* backward ptr */
+ dprintf("Packet: fwd: %d, bwd: %d\n",
+ nut->curr_frame_size, nut->last_frame_size);
+
+ nut->last_frame_size = nut->curr_frame_size;
+
+ return 0;
+}
+
+static int update_packetheader(NUTContext *nut, ByteIOContext *bc, int additional_size){
+ offset_t start= nut->curr_frame_start;
+ offset_t cur= url_ftell(bc);
+ int size= cur - start + additional_size;
+
+ assert( size <= nut->curr_frame_size );
+
+ url_fseek(bc, start, SEEK_SET);
+ put_v(bc, size);
+ if(get_length(size) < get_length(nut->curr_frame_size))
+ put_byte(bc, 0x80);
+ nut->curr_frame_size= size;
+ dprintf("Packet update: size: %d\n", size);
+
+ url_fseek(bc, cur, SEEK_SET);
+
+ return 0;
+}
+
+static int nut_write_header(AVFormatContext *s)
+{
+ NUTContext *nut = s->priv_data;
+ ByteIOContext *bc = &s->pb;
+ AVCodecContext *codec;
+ int i;
+ int stream_length = 0;
+
+ for (i = 0; i < s->nb_streams; i++)
+ {
+ if (stream_length < (s->streams[i]->duration * (AV_TIME_BASE / 1000)))
+ stream_length = s->streams[i]->duration * (AV_TIME_BASE / 1000);
+ }
+
+ /* main header */
+ put_be64(bc, MAIN_STARTCODE);
+ put_packetheader(nut, bc, 120);
+ put_v(bc, 0); /* version */
+ put_v(bc, s->nb_streams);
+ put_v(bc, 0); /* file size */
+ put_v(bc, stream_length); /* len in msec */
+ put_be32(bc, 0); /* FIXME: checksum */
+
+ update_packetheader(nut, bc, 0);
+
+ /* stream headers */
+ for (i = 0; i < s->nb_streams; i++)
+ {
+ codec = &s->streams[i]->codec;
+
+ put_be64(bc, STREAM_STARTCODE);
+ put_packetheader(nut, bc, 120);
+ put_v(bc, i /*s->streams[i]->index*/);
+ put_v(bc, (codec->codec_type == CODEC_TYPE_AUDIO) ? 32 : 0);
+ if (codec->codec_tag)
+ put_bi(bc, codec->codec_tag);
+ else if (codec->codec_type == CODEC_TYPE_VIDEO)
+ {
+ int tmp = codec_get_bmp_tag(codec->codec_id);
+ put_bi(bc, tmp);
+ }
+ else if (codec->codec_type == CODEC_TYPE_AUDIO)
+ {
+ int tmp = codec_get_wav_tag(codec->codec_id);
+ put_bi(bc, tmp);
+ }
+ put_v(bc, codec->bit_rate);
+ put_v(bc, 0); /* no language code */
+ put_v(bc, codec->frame_rate_base);
+ put_v(bc, codec->frame_rate);
+ put_v(bc, 0); /* timestamp_shift */
+ put_v(bc, 0); /* shuffle type */
+ put_byte(bc, 0); /* flags: 0x1 - fixed_fps, 0x2 - index_present */
+
+ put_v(bc, 0); /* no codec specific headers */
+
+ switch(codec->codec_type)
+ {
+ case CODEC_TYPE_AUDIO:
+ put_v(bc, codec->sample_rate / (double)(codec->frame_rate_base / codec->frame_rate));
+ put_v(bc, codec->channels);
+ put_be32(bc, 0); /* FIXME: checksum */
+ break;
+ case CODEC_TYPE_VIDEO:
+ put_v(bc, codec->width);
+ put_v(bc, codec->height);
+ put_v(bc, 0); /* aspected w */
+ put_v(bc, 0); /* aspected h */
+ put_v(bc, 0); /* csp type -- unknown */
+ put_be32(bc, 0); /* FIXME: checksum */
+ break;
+ default:
+ break;
+ }
+ update_packetheader(nut, bc, 0);
+ }
+
+#if 0
+ /* info header */
+ put_be64(bc, INFO_STARTCODE);
+ put_packetheader(nut, bc, 16+strlen(s->author)+strlen(s->title)+
+ strlen(s->comment)+strlen(s->copyright));
+ if (s->author[0])
+ {
+ put_v(bc, 5); /* type */
+ put_b(bc, s->author, strlen(s->author));
+ }
+ if (s->title[0])
+ {
+ put_v(bc, 6); /* type */
+ put_b(bc, s->title, strlen(s->title));
+ }
+ if (s->comment[0])
+ {
+ put_v(bc, 7); /* type */
+ put_b(bc, s->comment, strlen(s->comment));
+ }
+ if (s->copyright[0])
+ {
+ put_v(bc, 8); /* type */
+ put_b(bc, s->copyright, strlen(s->copyright));
+ }
+ /* encoder */
+ put_v(bc, 9); /* type */
+ put_b(bc, LIBAVFORMAT_IDENT "\0", strlen(LIBAVFORMAT_IDENT));
+
+ put_v(bc, 0); /* eof info */
+
+ put_be32(bc, 0); /* FIXME: checksum */
+ update_packetheader(nut, bc, 0);
+#endif
+
+ put_flush_packet(bc);
+
+ return 0;
+}
+
+static int nut_write_packet(AVFormatContext *s, int stream_index,
+ const uint8_t *buf, int size, int64_t pts)
+{
+ NUTContext *nut = s->priv_data;
+ ByteIOContext *bc = &s->pb;
+ int key_frame = 0;
+ int flags;
+ AVCodecContext *enc;
+
+ if (stream_index > s->nb_streams)
+ return 1;
+
+ enc = &s->streams[stream_index]->codec;
+ key_frame = enc->coded_frame->key_frame;
+
+ if (key_frame)
+ put_be64(bc, KEYFRAME_STARTCODE);
+
+ flags=0;
+ flags<<=2; flags|=1; //priority
+ flags<<=1; flags|=0; //checksum
+ flags<<=1; flags|=0; //msb_timestamp_flag
+ flags<<=2; flags|=1; //subpacket_type
+ flags<<=1; flags|=0; //reserved
+
+ put_byte(bc, flags);
+
+ put_packetheader(nut, bc, size+20);
+ put_v(bc, stream_index);
+ put_s(bc, pts); /* lsb_timestamp */
+ update_packetheader(nut, bc, size);
+
+ put_buffer(bc, buf, size);
+
+ put_flush_packet(bc);
+
+ return 0;
+}
+
+static int nut_write_trailer(AVFormatContext *s)
+{
+ ByteIOContext *bc = &s->pb;
+#if 0
+ int i;
+
+ /* WRITE INDEX */
+
+ for (i = 0; s->nb_streams; i++)
+ {
+ put_be64(bc, INDEX_STARTCODE);
+ put_packetheader(nut, bc, 64);
+ put_v(bc, s->streams[i]->id);
+ put_v(bc, ...);
+ put_be32(bc, 0); /* FIXME: checksum */
+ update_packetheader(nut, bc, 0);
+ }
+#endif
+
+ put_flush_packet(bc);
+
+ return 0;
+}
+
+static int nut_probe(AVProbeData *p)
+{
+ int i;
+ uint64_t code;
+
+ code = 0xff;
+ for (i = 0; i < p->buf_size; i++) {
+ int c = p->buf[i];
+ code = (code << 8) | c;
+ if (code == MAIN_STARTCODE)
+ return AVPROBE_SCORE_MAX;
+ }
+ return 0;
+}
+
+static int nut_read_header(AVFormatContext *s, AVFormatParameters *ap)
+{
+ NUTContext *nut = s->priv_data;
+ ByteIOContext *bc = &s->pb;
+ uint64_t tmp;
+ int cur_stream, nb_streams;
+
+ /* main header */
+ tmp = get_be64(bc);
+ if (tmp != MAIN_STARTCODE)
+ fprintf(stderr, "damaged? startcode!=1 (%Ld)\n", tmp);
+ get_packetheader(nut, bc);
+
+ tmp = get_v(bc);
+ if (tmp != 0)
+ fprintf(stderr, "bad version (%Ld)\n", tmp);
+
+ nb_streams = get_v(bc);
+
+ s->file_size = get_v(bc);
+ s->duration = get_v(bc) / (AV_TIME_BASE / 1000);
+
+ get_be32(bc); /* checkusm */
+
+ s->bit_rate = 0;
+
+ /* stream header */
+ for (cur_stream = 0; cur_stream < nb_streams; cur_stream++)
+ {
+ int class;
+ AVStream *st;
+
+ tmp = get_be64(bc);
+ if (tmp != STREAM_STARTCODE)
+ fprintf(stderr, "damaged? startcode!=1 (%Ld)\n", tmp);
+ get_packetheader(nut, bc);
+ st = av_new_stream(s, get_v(bc));
+ if (!st)
+ return AVERROR_NOMEM;
+ class = get_v(bc);
+ tmp = get_bi(bc);
+ switch(class)
+ {
+ case 0:
+ st->codec.codec_type = CODEC_TYPE_VIDEO;
+ st->codec.codec_id = codec_get_bmp_id(tmp);
+ if (st->codec.codec_id == CODEC_ID_NONE)
+ fprintf(stderr, "Unknown codec?!\n");
+ break;
+ case 32:
+ st->codec.codec_type = CODEC_TYPE_AUDIO;
+ st->codec.codec_id = codec_get_wav_id(tmp);
+ if (st->codec.codec_id == CODEC_ID_NONE)
+ fprintf(stderr, "Unknown codec?!\n");
+ break;
+ default:
+ fprintf(stderr, "Unknown stream class (%d)\n", class);
+ return -1;
+ }
+ s->bit_rate += get_v(bc);
+ get_b(bc, NULL, 0); /* language code */
+ st->codec.frame_rate_base = get_v(bc);
+ st->codec.frame_rate = get_v(bc);
+ get_v(bc); /* FIXME: msb timestamp base */
+ get_v(bc); /* shuffle type */
+ get_byte(bc); /* flags */
+
+ get_v(bc); /* FIXME: codec specific data headers */
+
+ if (class == 0) /* VIDEO */
+ {
+ st->codec.width = get_v(bc);
+ st->codec.height = get_v(bc);
+ get_v(bc); /* aspected w */
+ get_v(bc); /* aspected h */
+ get_v(bc); /* csp type */
+ get_be32(bc); /* checksum */
+ }
+ if (class == 32) /* AUDIO */
+ {
+ st->codec.sample_rate = get_v(bc) * (double)(st->codec.frame_rate_base / st->codec.frame_rate);
+ st->codec.channels = get_v(bc);
+ get_be32(bc); /* checksum */
+ }
+ }
+
+ return 0;
+}
+
+static int nut_read_packet(AVFormatContext *s, AVPacket *pkt)
+{
+ NUTContext *nut = s->priv_data;
+ ByteIOContext *bc = &s->pb;
+ int id, timestamp, size;
+ int key_frame = 0;
+ uint64_t tmp;
+
+
+ if (url_feof(bc))
+ return -1;
+
+ tmp = get_byte(bc);
+ if (tmp & 0x80) /* zero bit set? */
+ {
+ tmp<<=8 ; tmp |= get_byte(bc);
+ tmp<<=16; tmp |= get_be16(bc);
+ tmp<<=32; tmp |= get_be32(bc);
+ if (tmp == KEYFRAME_STARTCODE)
+ {
+ key_frame = 1;
+ tmp = get_byte(bc); /* flags */
+ }
+ else
+ fprintf(stderr, "error in zero bit / startcode %LX\n", tmp);
+ }
+ get_packetheader(nut, bc);
+#if 0
+ if (((tmp & 0x60)>>5) > 3) /* priority <= 3 */
+ fprintf(stderr, "sanity check failed!\n");
+#endif
+ id = get_v(bc);
+ timestamp = get_s(bc);
+
+ size = (nut->curr_frame_size - (url_ftell(bc)-nut->curr_frame_start));
+ dprintf("flags: 0x%Lx, timestamp: %d, packet size: %d\n", tmp, timestamp, size);
+
+ if (size < 0)
+ return -1;
+
+ av_new_packet(pkt, size);
+ get_buffer(bc, pkt->data, size);
+ pkt->stream_index = id;
+ if (key_frame)
+ pkt->flags |= PKT_FLAG_KEY;
+ pkt->pts = timestamp;
+
+ return 0;
+}
+
+static AVInputFormat nut_iformat = {
+ "nut",
+ "nut format",
+ sizeof(NUTContext),
+ nut_probe,
+ nut_read_header,
+ nut_read_packet,
+// nut_read_close,
+// nut_read_seek,
+ .extensions = "nut",
+};
+
+static AVOutputFormat nut_oformat = {
+ "nut",
+ "nut format",
+ "video/x-nut",
+ "nut",
+ sizeof(NUTContext),
+#ifdef CONFIG_VORBIS
+ CODEC_ID_VORBIS,
+#elif defined(CONFIG_MP3LAME)
+ CODEC_ID_MP3,
+#else
+ CODEC_ID_MP2, /* AC3 needs liba52 decoder */
+#endif
+ CODEC_ID_MPEG4,
+ nut_write_header,
+ nut_write_packet,
+ nut_write_trailer,
+};
+
+int nut_init(void)
+{
+ av_register_input_format(&nut_iformat);
+ av_register_output_format(&nut_oformat);
+ return 0;
+}