libmpdemux/demux_nsv.c

   1 /*
   2  * Nullsoft Streaming Video demuxer
   3  * copyright (c) 2004 by Reza Jelveh <reza.jelveh@tuhh.de>
   4  * Based on A'rpis G2 work
   5  *
   6  * seeking and PCM audio not yet supported
   7  * PCM needs extra audio chunk "miniheader" parsing
   8  *
   9  * This file is part of MPlayer.
  10  *
  11  * MPlayer is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * MPlayer is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License along
  22  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  23  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  24  */
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <unistd.h>
  29
  30 #include "config.h"
  31 #include "mp_msg.h"
  32 #include "stream/stream.h"
  33 #include "demuxer.h"
  34 #include "stheader.h"
  35
  36 typedef struct {
  37     float   v_pts;
  38     int video_pack_no;
  39     unsigned int a_format;
  40     unsigned int v_format;
  41     unsigned char fps;
  42 } nsv_priv_t;
  43
  44 #define HEADER_SEARCH_SIZE 256000
  45
  46
  47 /**
  48  * Seeking still to be implemented
  49  */
  50 static void demux_seek_nsv ( demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags )
  51 {
  52 // seeking is not yet implemented
  53 }
  54
  55
  56 static int demux_nsv_fill_buffer ( demuxer_t *demuxer, demux_stream_t *ds )
  57 {
  58     unsigned char hdr[17];
  59     // for the extra data
  60     unsigned char aux[6];
  61     int i_aux = 0;
  62     // videolen = audio chunk length, audiolen = video chunk length
  63     int videolen,audiolen;
  64
  65     sh_video_t *sh_video = demuxer->video->sh;
  66     sh_audio_t *sh_audio = demuxer->audio->sh;
  67
  68     nsv_priv_t * priv = demuxer->priv;
  69
  70     // if the audio/video chunk has no new header the first 2 bytes will be discarded 0xBEEF
  71     // or rather 0xEF 0xBE
  72     stream_read(demuxer->stream,hdr,7);
  73     if(stream_eof(demuxer->stream)) return 0;
  74     // sometimes instead of 0xBEEF as described for the next audio/video chunk we get
  75     // a whole new header
  76
  77     mp_dbg(MSGT_DEMUX,MSGL_DBG2,"demux_nsv: %08X %08X\n",hdr[0]<<8|hdr[1],stream_tell(demuxer->stream));
  78     switch(hdr[0]<<8|hdr[1]) {
  79         case 0x4E53:
  80             if(hdr[2]==0x56 && hdr[3]==0x73){
  81                 // NSVs
  82                 // get the header since there is no more metaheader after the first one
  83                 // there is no more need to skip that
  84                 stream_read(demuxer->stream,hdr+7,17-7);
  85                 stream_read(demuxer->stream,hdr,7);
  86             }
  87             break;
  88
  89         case 0xEFBE:
  90             break;
  91
  92         default:
  93             mp_dbg(MSGT_DEMUX,MSGL_WARN,"demux_nsv: sync lost\n");
  94             break;
  95     }
  96
  97     if (sh_video)
  98     sh_video->pts = priv->v_pts =demuxer->video->pts=  priv->video_pack_no *
  99          (float)sh_video->frametime;
 100     else
 101         priv->v_pts = priv->video_pack_no;
 102
 103     demuxer->filepos=stream_tell(demuxer->stream);
 104
 105
 106     mp_dbg(MSGT_DEMUX,MSGL_DBG2,"demux_nsv: %08X: %02X %02X | %02X %02X %02X | %02X %02X  \n",
 107     (int)demuxer->filepos, hdr[0],hdr[1],hdr[2],hdr[3],hdr[4],hdr[5],hdr[6]);
 108
 109     // read video:
 110     videolen=(hdr[2]>>4)|(hdr[3]<<4)|(hdr[4]<<0xC);
 111     //check if we got extra data like subtitles here
 112     if( (hdr[2]&0x0f) != 0x0 ) {
 113         stream_read( demuxer->stream, aux, 6);
 114
 115         i_aux = aux[0]|aux[1]<<8;
 116         // We skip this extra data
 117         stream_skip( demuxer->stream, i_aux );
 118         i_aux+=6;
 119         videolen -= i_aux;
 120     }
 121
 122
 123     // we need to return an empty packet when the
 124     // video frame is empty otherwise the stream will fasten up
 125     if(sh_video) {
 126         if( (hdr[2]&0x0f) != 0x0 )
 127             ds_read_packet(demuxer->video,demuxer->stream,videolen,priv->v_pts,demuxer->filepos-i_aux,0);
 128         else
 129             ds_read_packet(demuxer->video,demuxer->stream,videolen,priv->v_pts,demuxer->filepos,0);
 130     }
 131     else
 132         stream_skip(demuxer->stream,videolen);
 133
 134     // read audio:
 135     audiolen=(hdr[5])|(hdr[6]<<8);
 136     // we need to return an empty packet when the
 137     // audio frame is empty otherwise the stream will fasten up
 138     if(sh_audio) {
 139         ds_read_packet(demuxer->audio,demuxer->stream,audiolen,priv->v_pts,demuxer->filepos+videolen,0);
 140     }
 141     else
 142         stream_skip(demuxer->stream,audiolen);
 143
 144     ++priv->video_pack_no;
 145
 146     return 1;
 147
 148 }
 149
 150
 151 static demuxer_t* demux_open_nsv ( demuxer_t* demuxer )
 152 {
 153     // last 2 bytes 17 and 18 are unknown but right after that comes the length
 154     unsigned char hdr[17];
 155     int videolen,audiolen;
 156     unsigned char buf[10];
 157     sh_video_t *sh_video = NULL;
 158     sh_audio_t *sh_audio = NULL;
 159
 160
 161     nsv_priv_t * priv = malloc(sizeof(nsv_priv_t));
 162     demuxer->priv=priv;
 163     priv->video_pack_no=0;
 164
 165       /* disable seeking yet to be fixed*/
 166     demuxer->seekable = 0;
 167
 168     stream_read(demuxer->stream,hdr,4);
 169     if(stream_eof(demuxer->stream)) return 0;
 170
 171     if(hdr[0]==0x4E && hdr[1]==0x53 && hdr[2]==0x56){
 172         // NSV header!
 173         if(hdr[3]==0x73){
 174             // NSVs
 175             stream_read(demuxer->stream,hdr+4,17-4);
 176         }
 177
 178         if(hdr[3]==0x66){
 179             // NSVf
 180             int len=stream_read_dword_le(demuxer->stream);
 181             // TODO: parse out metadata!!!!
 182             stream_skip(demuxer->stream,len-8);
 183
 184             // NSVs
 185             stream_read(demuxer->stream,hdr,17);
 186             if (stream_eof(demuxer->stream) || strncmp(hdr, "NSVs", 4))
 187                 return 0;
 188         }
 189
 190         // dummy debug message
 191         mp_msg(MSGT_DEMUX,MSGL_V,"demux_nsv: Header: %.12s\n",hdr);
 192
 193         //   bytes 8-11   audio codec fourcc
 194         // PCM fourcc needs extra parsing for every audio chunk, yet to implement
 195         if((demuxer->audio->id != -2) && strncmp(hdr+8,"NONE", 4)){//&&strncmp(hdr+8,"VLB ", 4)){
 196             sh_audio = new_sh_audio ( demuxer, 0 );
 197             demuxer->audio->id = 0;
 198             demuxer->audio->sh = sh_audio;
 199             sh_audio->format=mmioFOURCC(hdr[8],hdr[9],hdr[10],hdr[11]);
 200             sh_audio->ds = demuxer->audio;
 201             priv->a_format=mmioFOURCC(hdr[8],hdr[9],hdr[10],hdr[11]);
 202         }
 203
 204         // store hdr fps
 205         priv->fps=hdr[16];
 206
 207         if ((demuxer->video->id != -2) && strncmp(hdr+4,"NONE", 4)) {
 208             /* Create a new video stream header */
 209             sh_video = new_sh_video ( demuxer, 0 );
 210
 211             /* Make sure the demuxer knows about the new video stream header
 212              * (even though new_sh_video() ought to take care of it)
 213              */
 214             demuxer->video->sh = sh_video;
 215
 216             /* Make sure that the video demuxer stream header knows about its
 217              * parent video demuxer stream (this is getting wacky), or else
 218              * video_read_properties() will choke
 219              */
 220             sh_video->ds = demuxer->video;
 221
 222             //   bytes 4-7  video codec fourcc
 223             priv->v_format = sh_video->format=mmioFOURCC(hdr[4],hdr[5],hdr[6],hdr[7]);
 224
 225             // new video stream! parse header
 226             sh_video->disp_w=hdr[12]|(hdr[13]<<8);
 227             sh_video->disp_h=hdr[14]|(hdr[15]<<8);
 228             sh_video->bih=calloc(1,sizeof(BITMAPINFOHEADER));
 229             sh_video->bih->biSize=sizeof(BITMAPINFOHEADER);
 230             sh_video->bih->biPlanes=1;
 231             sh_video->bih->biBitCount=24;
 232             sh_video->bih->biWidth=hdr[12]|(hdr[13]<<8);
 233             sh_video->bih->biHeight=hdr[14]|(hdr[15]<<8);
 234             memcpy(&sh_video->bih->biCompression,hdr+4,4);
 235             sh_video->bih->biSizeImage=sh_video->bih->biWidth*sh_video->bih->biHeight*3;
 236
 237             // here we search for the correct keyframe
 238             // vp6 keyframe is when the 2nd byte of the vp6 header is
 239             // 0x36 for VP61 and 0x46 for VP62
 240             if((priv->v_format==mmioFOURCC('V','P','6','1')) ||
 241                (priv->v_format==mmioFOURCC('V','P','6','2')) ||
 242                (priv->v_format==mmioFOURCC('V','P','3','1'))) {
 243                 stream_read(demuxer->stream,buf,10);
 244                 if (((((priv->v_format>>16) & 0xff) == '6') && ((buf[8]&0x0e)!=0x06)) ||
 245                     ((((priv->v_format>>16) & 0xff) == '3') && (buf[8]!=0x00 || buf[9]!=0x08))) {
 246                     mp_msg(MSGT_DEMUX,MSGL_V,"demux_nsv: searching %.4s keyframe...\n", (char*)&priv->v_format);
 247                     while(((((priv->v_format>>16) & 0xff) == '6') && ((buf[8]&0x0e)!=0x06)) ||
 248                           ((((priv->v_format>>16) & 0xff) == '3') && (buf[8]!=0x00 || buf[9]!=0x08))){
 249                         mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_nsv: %.4s block skip.\n", (char*)&priv->v_format);
 250                         videolen=(buf[2]>>4)|(buf[3]<<4)|(buf[4]<<0xC);
 251                         audiolen=(buf[5])|(buf[6]<<8);
 252                         stream_skip(demuxer->stream, videolen+audiolen-3);
 253                         stream_read(demuxer->stream,buf,10);
 254                         if(stream_eof(demuxer->stream)) return 0;
 255                         if(buf[0]==0x4E){
 256                             mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_nsv: Got NSVs block.\n");
 257                             stream_skip(demuxer->stream,7);
 258                             stream_read(demuxer->stream,buf,10);
 259                         }
 260                     }
 261                 }
 262
 263                 // data starts 10 bytes before current pos but later
 264                 // we seek 17 backwards
 265                 stream_skip(demuxer->stream,7);
 266             }
 267
 268         switch(priv->fps){
 269         case 0x80:
 270             sh_video->fps=30;
 271             break;
 272         case 0x81:
 273             sh_video->fps=(float)30000.0/1001.0;
 274             break;
 275         case 0x82:
 276             sh_video->fps=25;
 277             break;
 278         case 0x83:
 279             sh_video->fps=(float)24000.0/1001.0;
 280             break;
 281         case 0x85:
 282             sh_video->fps=(float)15000.0/1001.0;
 283             break;
 284         case 0x89:
 285             sh_video->fps=(float)10000.0/1001.0;
 286             break;
 287         default:
 288             sh_video->fps = (float)priv->fps;
 289         }
 290         sh_video->frametime = (float)1.0 / (float)sh_video->fps;
 291        }
 292     }
 293
 294     // seek to start of NSV header
 295     stream_seek(demuxer->stream,stream_tell(demuxer->stream)-17);
 296
 297     return demuxer;
 298 }
 299
 300 static int nsv_check_file ( demuxer_t* demuxer )
 301 {
 302     uint32_t hdr = 0;
 303     int i;
 304
 305     mp_msg ( MSGT_DEMUX, MSGL_V, "Checking for Nullsoft Streaming Video\n" );
 306
 307     for (i = 0; i < HEADER_SEARCH_SIZE; i++) {
 308         uint8_t c = stream_read_char(demuxer->stream);
 309         if (stream_eof(demuxer->stream))
 310             return 0;
 311         if (hdr == mmioFOURCC('s', 'V', 'S', 'N') ||
 312             (hdr == mmioFOURCC('f', 'V', 'S', 'N') && !c)) {
 313             stream_seek(demuxer->stream,stream_tell(demuxer->stream)-5);
 314             return DEMUXER_TYPE_NSV;
 315         }
 316         hdr = (hdr << 8) | c;
 317     }
 318
 319     return 0;
 320 }
 321
 322 static void demux_close_nsv(demuxer_t* demuxer) {
 323     nsv_priv_t* priv = demuxer->priv;
 324
 325     if(!priv)
 326         return;
 327     free(priv);
 328
 329 }
 330
 331
 332 const demuxer_desc_t demuxer_desc_nsv = {
 333   "NullsoftVideo demuxer",
 334   "nsv",
 335   "Nullsoft Streaming Video",
 336   "Reza Jelveh",
 337   "nsv and nsa streaming files",
 338   DEMUXER_TYPE_NSV,
 339   0, // safe but expensive autodetect
 340   nsv_check_file,
 341   demux_nsv_fill_buffer,
 342   demux_open_nsv,
 343   demux_close_nsv,
 344   demux_seek_nsv,
 345   NULL
 346 };