stream/stream.c

   1
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <unistd.h>
   5
   6 #include <sys/types.h>
   7 #include <sys/stat.h>
   8 #ifndef __MINGW32__
   9 #include <sys/ioctl.h>
  10 #include <sys/wait.h>
  11 #endif
  12 #include <fcntl.h>
  13 #include <signal.h>
  14 #include <strings.h>
  15
  16 #include "config.h"
  17
  18 #ifndef HAVE_WINSOCK2
  19 #define closesocket close
  20 #else
  21 #include <winsock2.h>
  22 #endif
  23
  24 #include "mp_msg.h"
  25 #include "help_mp.h"
  26 #include "osdep/shmem.h"
  27
  28 #include "stream.h"
  29 #include "libmpdemux/demuxer.h"
  30
  31 #include "m_option.h"
  32 #include "m_struct.h"
  33
  34
  35 void cache_uninit(stream_t *s); // defined in cache2.c
  36
  37 //#include "vcd_read_bincue.h"
  38
  39 #ifdef HAVE_VCD
  40 extern stream_info_t stream_info_vcd;
  41 #endif
  42 #ifdef HAVE_CDDA
  43 extern stream_info_t stream_info_cdda;
  44 #endif
  45 #ifdef MPLAYER_NETWORK
  46 extern stream_info_t stream_info_netstream;
  47 extern stream_info_t stream_info_pnm;
  48 extern stream_info_t stream_info_asf;
  49 extern stream_info_t stream_info_rtsp;
  50 extern stream_info_t stream_info_rtp;
  51 extern stream_info_t stream_info_udp;
  52 extern stream_info_t stream_info_http1;
  53 extern stream_info_t stream_info_http2;
  54 #endif
  55 #ifdef HAS_DVBIN_SUPPORT
  56 extern stream_info_t stream_info_dvb;
  57 #endif
  58 #ifdef USE_TV
  59 extern stream_info_t stream_info_tv;
  60 #endif
  61 #ifdef USE_RADIO
  62 extern stream_info_t stream_info_radio;
  63 #endif
  64 #ifdef HAVE_PVR
  65 extern stream_info_t stream_info_pvr;
  66 #endif
  67 #ifdef HAVE_FTP
  68 extern stream_info_t stream_info_ftp;
  69 #endif
  70 #ifdef HAVE_VSTREAM
  71 extern stream_info_t stream_info_vstream;
  72 #endif
  73 #ifdef USE_DVDNAV
  74 extern stream_info_t stream_info_dvdnav;
  75 #endif
  76 #ifdef LIBSMBCLIENT
  77 extern stream_info_t stream_info_smb;
  78 #endif
  79 #ifdef STREAMING_LIVE555
  80 extern stream_info_t stream_info_sdp;
  81 extern stream_info_t stream_info_rtsp_sip;
  82 #endif
  83
  84 extern stream_info_t stream_info_cue;
  85 extern stream_info_t stream_info_null;
  86 extern stream_info_t stream_info_mf;
  87 extern stream_info_t stream_info_file;
  88 #ifdef HAVE_DVD
  89 extern stream_info_t stream_info_dvd;
  90 #endif
  91
  92 stream_info_t* auto_open_streams[] = {
  93 #ifdef HAVE_VCD
  94   &stream_info_vcd,
  95 #endif
  96 #ifdef HAVE_CDDA
  97   &stream_info_cdda,
  98 #endif
  99 #ifdef MPLAYER_NETWORK
 100   &stream_info_netstream,
 101   &stream_info_http1,
 102   &stream_info_asf,
 103   &stream_info_pnm,
 104   &stream_info_rtsp,
 105 #ifdef STREAMING_LIVE555
 106   &stream_info_sdp,
 107   &stream_info_rtsp_sip,
 108 #endif
 109   &stream_info_rtp,
 110   &stream_info_udp,
 111   &stream_info_http2,
 112 #endif
 113 #ifdef HAS_DVBIN_SUPPORT
 114   &stream_info_dvb,
 115 #endif
 116 #ifdef USE_TV
 117   &stream_info_tv,
 118 #endif
 119 #ifdef USE_RADIO
 120   &stream_info_radio,
 121 #endif
 122 #ifdef HAVE_PVR
 123   &stream_info_pvr,
 124 #endif
 125 #ifdef HAVE_FTP
 126   &stream_info_ftp,
 127 #endif
 128 #ifdef HAVE_VSTREAM
 129   &stream_info_vstream,
 130 #endif
 131 #ifdef LIBSMBCLIENT
 132   &stream_info_smb,
 133 #endif
 134   &stream_info_cue,
 135 #ifdef HAVE_DVD
 136   &stream_info_dvd,
 137 #endif
 138 #ifdef USE_DVDNAV
 139   &stream_info_dvdnav,
 140 #endif
 141
 142   &stream_info_null,
 143   &stream_info_mf,
 144   &stream_info_file,
 145   NULL
 146 };
 147
 148 stream_t* open_stream_plugin(stream_info_t* sinfo,char* filename,int mode,
 149                              char** options, int* file_format, int* ret) {
 150   void* arg = NULL;
 151   stream_t* s;
 152   m_struct_t* desc = (m_struct_t*)sinfo->opts;
 153
 154   // Parse options
 155   if(desc) {
 156     arg = m_struct_alloc(desc);
 157     if(sinfo->opts_url) {
 158       m_option_t url_opt =
 159         { "stream url", arg , CONF_TYPE_CUSTOM_URL, 0, 0 ,0, sinfo->opts };
 160       if(m_option_parse(&url_opt,"stream url",filename,arg,M_CONFIG_FILE) < 0) {
 161         mp_msg(MSGT_OPEN,MSGL_ERR, "URL parsing failed on url %s\n",filename);
 162         m_struct_free(desc,arg);
 163         return NULL;
 164       }
 165     }
 166     if(options) {
 167       int i;
 168       for(i = 0 ; options[i] != NULL ; i += 2) {
 169         mp_msg(MSGT_OPEN,MSGL_DBG2, "Set stream arg %s=%s\n",
 170                options[i],options[i+1]);
 171         if(!m_struct_set(desc,arg,options[i],options[i+1]))
 172           mp_msg(MSGT_OPEN,MSGL_WARN, "Failed to set stream option %s=%s\n",
 173                  options[i],options[i+1]);
 174       }
 175     }
 176   }
 177   s = new_stream(-2,-2);
 178   s->url=strdup(filename);
 179   s->flags |= mode;
 180   *ret = sinfo->open(s,mode,arg,file_format);
 181   if((*ret) != STREAM_OK) {
 182     free(s->url);
 183     free(s);
 184     return NULL;
 185   }
 186   if(s->type <= -2)
 187     mp_msg(MSGT_OPEN,MSGL_WARN, "Warning streams need a type !!!!\n");
 188   if(s->flags & STREAM_SEEK && !s->seek)
 189     s->flags &= ~STREAM_SEEK;
 190   if(s->seek && !(s->flags & STREAM_SEEK))
 191     s->flags |= STREAM_SEEK;
 192
 193
 194   mp_msg(MSGT_OPEN,MSGL_V, "STREAM: [%s] %s\n",sinfo->name,filename);
 195   mp_msg(MSGT_OPEN,MSGL_V, "STREAM: Description: %s\n",sinfo->info);
 196   mp_msg(MSGT_OPEN,MSGL_V, "STREAM: Author: %s\n", sinfo->author);
 197   mp_msg(MSGT_OPEN,MSGL_V, "STREAM: Comment: %s\n", sinfo->comment);
 198
 199   return s;
 200 }
 201
 202
 203 stream_t* open_stream_full(char* filename,int mode, char** options, int* file_format) {
 204   int i,j,l,r;
 205   stream_info_t* sinfo;
 206   stream_t* s;
 207
 208   for(i = 0 ; auto_open_streams[i] ; i++) {
 209     sinfo = auto_open_streams[i];
 210     if(!sinfo->protocols) {
 211       mp_msg(MSGT_OPEN,MSGL_WARN, "Stream type %s has protocols == NULL, it's a bug\n", sinfo->name);
 212       continue;
 213     }
 214     for(j = 0 ; sinfo->protocols[j] ; j++) {
 215       l = strlen(sinfo->protocols[j]);
 216       // l == 0 => Don't do protocol matching (ie network and filenames)
 217       if((l == 0) || ((strncmp(sinfo->protocols[j],filename,l) == 0) &&
 218                       (strncmp("://",filename+l,3) == 0))) {
 219         *file_format = DEMUXER_TYPE_UNKNOWN;
 220         s = open_stream_plugin(sinfo,filename,mode,options,file_format,&r);
 221         if(s) return s;
 222         if(r != STREAM_UNSUPORTED) {
 223           mp_msg(MSGT_OPEN,MSGL_ERR, MSGTR_FailedToOpen,filename);
 224           return NULL;
 225         }
 226         break;
 227       }
 228     }
 229   }
 230
 231   mp_msg(MSGT_OPEN,MSGL_ERR, "No stream found to handle url %s\n",filename);
 232   return NULL;
 233 }
 234
 235 //=================== STREAMER =========================
 236
 237 int stream_fill_buffer(stream_t *s){
 238   int len;
 239   if (/*s->fd == NULL ||*/ s->eof) { s->buf_pos = s->buf_len = 0; return 0; }
 240   switch(s->type){
 241   case STREAMTYPE_STREAM:
 242 #ifdef MPLAYER_NETWORK
 243     if( s->streaming_ctrl!=NULL ) {
 244             len=s->streaming_ctrl->streaming_read(s->fd,s->buffer,STREAM_BUFFER_SIZE, s->streaming_ctrl);break;
 245     } else {
 246       len=read(s->fd,s->buffer,STREAM_BUFFER_SIZE);break;
 247     }
 248 #else
 249     len=read(s->fd,s->buffer,STREAM_BUFFER_SIZE);break;
 250 #endif
 251   case STREAMTYPE_DS:
 252     len = demux_read_data((demux_stream_t*)s->priv,s->buffer,STREAM_BUFFER_SIZE);
 253     break;
 254
 255
 256   default:
 257     len= s->fill_buffer ? s->fill_buffer(s,s->buffer,STREAM_BUFFER_SIZE) : 0;
 258   }
 259   if(len<=0){ s->eof=1; s->buf_pos=s->buf_len=0; return 0; }
 260   s->buf_pos=0;
 261   s->buf_len=len;
 262   s->pos+=len;
 263 //  printf("[%d]",len);fflush(stdout);
 264   return len;
 265 }
 266
 267 int stream_seek_long(stream_t *s,off_t pos){
 268 off_t newpos=0;
 269
 270 //  if( mp_msg_test(MSGT_STREAM,MSGL_DBG3) ) printf("seek_long to 0x%X\n",(unsigned int)pos);
 271
 272   s->buf_pos=s->buf_len=0;
 273
 274   switch(s->type){
 275   case STREAMTYPE_STREAM:
 276 #ifdef _LARGEFILE_SOURCE
 277     newpos=pos&(~((long long)STREAM_BUFFER_SIZE-1));break;
 278 #else
 279     newpos=pos&(~(STREAM_BUFFER_SIZE-1));break;
 280 #endif
 281   default:
 282     // Round on sector size
 283     if(s->sector_size)
 284       newpos=(pos/s->sector_size)*s->sector_size;
 285     else { // Otherwise on the buffer size
 286 #ifdef _LARGEFILE_SOURCE
 287       newpos=pos&(~((long long)STREAM_BUFFER_SIZE-1));break;
 288 #else
 289       newpos=pos&(~(STREAM_BUFFER_SIZE-1));break;
 290 #endif
 291     }
 292     break;
 293   }
 294
 295 if( mp_msg_test(MSGT_STREAM,MSGL_DBG3) ){
 296   mp_msg(MSGT_STREAM,MSGL_DBG3, "s->pos=%"PRIX64"  newpos=%"PRIX64"  new_bufpos=%"PRIX64"  buflen=%X  \n",
 297     (int64_t)s->pos,(int64_t)newpos,(int64_t)pos,s->buf_len);
 298 }
 299   pos-=newpos;
 300
 301 if(newpos==0 || newpos!=s->pos){
 302   switch(s->type){
 303   case STREAMTYPE_STREAM:
 304     //s->pos=newpos; // real seek
 305     // Some streaming protocol allow to seek backward and forward
 306     // A function call that return -1 can tell that the protocol
 307     // doesn't support seeking.
 308 #ifdef MPLAYER_NETWORK
 309     if(s->seek) { // new stream seek is much cleaner than streaming_ctrl one
 310       if(!s->seek(s,newpos)) {
 311         mp_msg(MSGT_STREAM,MSGL_ERR, "Seek failed\n");
 312         return 0;
 313       }
 314       break;
 315     }
 316
 317     if( s->streaming_ctrl!=NULL && s->streaming_ctrl->streaming_seek ) {
 318       if( s->streaming_ctrl->streaming_seek( s->fd, pos, s->streaming_ctrl )<0 ) {
 319         mp_msg(MSGT_STREAM,MSGL_INFO,"Stream not seekable!\n");
 320         return 1;
 321       }
 322     }
 323 #else
 324     if(newpos<s->pos){
 325       mp_msg(MSGT_STREAM,MSGL_INFO,"Cannot seek backward in linear streams!\n");
 326       return 1;
 327     }
 328     while(s->pos<newpos){
 329       if(stream_fill_buffer(s)<=0) break; // EOF
 330     }
 331 #endif
 332     break;
 333   default:
 334     // This should at the beginning as soon as all streams are converted
 335     if(!s->seek)
 336       return 0;
 337     // Now seek
 338     if(!s->seek(s,newpos)) {
 339       mp_msg(MSGT_STREAM,MSGL_ERR, "Seek failed\n");
 340       return 0;
 341     }
 342   }
 343 //   putchar('.');fflush(stdout);
 344 //} else {
 345 //   putchar('%');fflush(stdout);
 346 }
 347
 348 while(stream_fill_buffer(s) > 0 && pos >= 0) {
 349   if(pos<=s->buf_len){
 350     s->buf_pos=pos; // byte position in sector
 351     return 1;
 352   }
 353   pos -= s->buf_len;
 354 }
 355
 356 //  if(pos==s->buf_len) printf("XXX Seek to last byte of file -> EOF\n");
 357
 358   mp_msg(MSGT_STREAM,MSGL_V,"stream_seek: WARNING! Can't seek to 0x%"PRIX64" !\n",(int64_t)(pos+newpos));
 359   return 0;
 360 }
 361
 362
 363 void stream_reset(stream_t *s){
 364   if(s->eof){
 365     s->pos=0; //ftell(f);
 366 //    s->buf_pos=s->buf_len=0;
 367     s->eof=0;
 368   }
 369   if(s->control) s->control(s,STREAM_CTRL_RESET,NULL);
 370   //stream_seek(s,0);
 371 }
 372
 373 int stream_control(stream_t *s, int cmd, void *arg){
 374   if(!s->control) return STREAM_UNSUPORTED;
 375   return s->control(s, cmd, arg);
 376 }
 377
 378 stream_t* new_memory_stream(unsigned char* data,int len){
 379   stream_t *s;
 380
 381   if(len < 0)
 382     return NULL;
 383   s=malloc(sizeof(stream_t)+len);
 384   memset(s,0,sizeof(stream_t));
 385   s->fd=-1;
 386   s->type=STREAMTYPE_MEMORY;
 387   s->buf_pos=0; s->buf_len=len;
 388   s->start_pos=0; s->end_pos=len;
 389   stream_reset(s);
 390   s->pos=len;
 391   memcpy(s->buffer,data,len);
 392   return s;
 393 }
 394
 395 stream_t* new_stream(int fd,int type){
 396   stream_t *s=malloc(sizeof(stream_t));
 397   if(s==NULL) return NULL;
 398   memset(s,0,sizeof(stream_t));
 399
 400 #ifdef HAVE_WINSOCK2
 401   {
 402     WSADATA wsdata;
 403     int temp = WSAStartup(0x0202, &wsdata); // there might be a better place for this (-> later)
 404     mp_msg(MSGT_STREAM,MSGL_V,"WINSOCK2 init: %i\n", temp);
 405   }
 406 #endif
 407
 408   s->fd=fd;
 409   s->type=type;
 410   s->buf_pos=s->buf_len=0;
 411   s->start_pos=s->end_pos=0;
 412   s->priv=NULL;
 413   s->url=NULL;
 414   s->cache_pid=0;
 415   stream_reset(s);
 416   return s;
 417 }
 418
 419 void free_stream(stream_t *s){
 420 //  printf("\n*** free_stream() called ***\n");
 421 #ifdef USE_STREAM_CACHE
 422   if(s->cache_pid) {
 423     cache_uninit(s);
 424   }
 425 #endif
 426   if(s->close) s->close(s);
 427   if(s->fd>0){
 428     /* on unix we define closesocket to close
 429        on windows however we have to distinguish between
 430        network socket and file */
 431     if(s->url && strstr(s->url,"://"))
 432       closesocket(s->fd);
 433     else close(s->fd);
 434   }
 435 #ifdef HAVE_WINSOCK2
 436   mp_msg(MSGT_STREAM,MSGL_V,"WINSOCK2 uninit\n");
 437   WSACleanup(); // there might be a better place for this (-> later)
 438 #endif
 439   // Disabled atm, i don't like that. s->priv can be anything after all
 440   // streams should destroy their priv on close
 441   //if(s->priv) free(s->priv);
 442   if(s->url) free(s->url);
 443   free(s);
 444 }
 445
 446 stream_t* new_ds_stream(demux_stream_t *ds) {
 447   stream_t* s = new_stream(-1,STREAMTYPE_DS);
 448   s->priv = ds;
 449   return s;
 450 }