Mercurial > mplayer.hg
view libmpdemux/demuxer.c @ 22113:50c9dc00154d
Add timeout to tcp connections, avoid hanging forever.
Based on patch for bugzilla #673
author | rtogni |
---|---|
date | Sun, 04 Feb 2007 14:18:56 +0000 |
parents | 2c473e083c70 |
children | 84f95595f31f |
line wrap: on
line source
//=================== DEMUXER v2.5 ========================= #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/types.h> #include <sys/stat.h> #include "config.h" #include "mp_msg.h" #include "help_mp.h" #include "m_config.h" #include "libvo/fastmemcpy.h" #include "stream.h" #include "demuxer.h" #include "stheader.h" #include "mf.h" #include "libaf/af_format.h" extern void resync_audio_stream(sh_audio_t *sh_audio); // Demuxer list extern demuxer_desc_t demuxer_desc_rawaudio; extern demuxer_desc_t demuxer_desc_rawvideo; extern demuxer_desc_t demuxer_desc_tv; extern demuxer_desc_t demuxer_desc_mf; extern demuxer_desc_t demuxer_desc_avi; extern demuxer_desc_t demuxer_desc_y4m; extern demuxer_desc_t demuxer_desc_asf; extern demuxer_desc_t demuxer_desc_nuv; extern demuxer_desc_t demuxer_desc_real; extern demuxer_desc_t demuxer_desc_smjpeg; extern demuxer_desc_t demuxer_desc_matroska; extern demuxer_desc_t demuxer_desc_realaudio; extern demuxer_desc_t demuxer_desc_vqf; extern demuxer_desc_t demuxer_desc_mov; extern demuxer_desc_t demuxer_desc_vivo; extern demuxer_desc_t demuxer_desc_fli; extern demuxer_desc_t demuxer_desc_film; extern demuxer_desc_t demuxer_desc_roq; extern demuxer_desc_t demuxer_desc_gif; extern demuxer_desc_t demuxer_desc_ogg; extern demuxer_desc_t demuxer_desc_avs; extern demuxer_desc_t demuxer_desc_pva; extern demuxer_desc_t demuxer_desc_nsv; extern demuxer_desc_t demuxer_desc_mpeg_ts; extern demuxer_desc_t demuxer_desc_lmlm4; extern demuxer_desc_t demuxer_desc_mpeg_ps; extern demuxer_desc_t demuxer_desc_mpeg_pes; extern demuxer_desc_t demuxer_desc_mpeg_es; extern demuxer_desc_t demuxer_desc_mpeg_gxf; extern demuxer_desc_t demuxer_desc_mpeg4_es; extern demuxer_desc_t demuxer_desc_h264_es; extern demuxer_desc_t demuxer_desc_rawdv; extern demuxer_desc_t demuxer_desc_mpc; extern demuxer_desc_t demuxer_desc_audio; extern demuxer_desc_t demuxer_desc_xmms; extern demuxer_desc_t demuxer_desc_mpeg_ty; extern demuxer_desc_t demuxer_desc_rtp; extern demuxer_desc_t demuxer_desc_lavf; extern demuxer_desc_t demuxer_desc_aac; extern demuxer_desc_t demuxer_desc_nut; demuxer_desc_t* demuxer_list[] = { &demuxer_desc_rawaudio, &demuxer_desc_rawvideo, #ifdef USE_TV &demuxer_desc_tv, #endif &demuxer_desc_mf, &demuxer_desc_avi, &demuxer_desc_y4m, &demuxer_desc_asf, &demuxer_desc_nsv, &demuxer_desc_nuv, &demuxer_desc_real, &demuxer_desc_smjpeg, &demuxer_desc_matroska, &demuxer_desc_realaudio, &demuxer_desc_vqf, &demuxer_desc_mov, &demuxer_desc_vivo, &demuxer_desc_fli, &demuxer_desc_film, &demuxer_desc_roq, #ifdef HAVE_GIF &demuxer_desc_gif, #endif #ifdef HAVE_OGGVORBIS &demuxer_desc_ogg, #endif #ifdef USE_WIN32DLL &demuxer_desc_avs, #endif &demuxer_desc_pva, &demuxer_desc_mpeg_ts, &demuxer_desc_lmlm4, &demuxer_desc_mpeg_ps, &demuxer_desc_mpeg_pes, &demuxer_desc_mpeg_es, &demuxer_desc_mpeg_gxf, &demuxer_desc_mpeg4_es, &demuxer_desc_h264_es, #ifdef MUSEPACK &demuxer_desc_mpc, #endif &demuxer_desc_audio, &demuxer_desc_mpeg_ty, #ifdef STREAMING_LIVE555 &demuxer_desc_rtp, #endif #if defined(USE_LIBAVFORMAT) || defined(USE_LIBAVFORMAT_SO) &demuxer_desc_lavf, #endif #ifdef HAVE_LIBDV095 &demuxer_desc_rawdv, #endif &demuxer_desc_aac, #ifdef HAVE_LIBNUT &demuxer_desc_nut, #endif #ifdef HAVE_XMMS &demuxer_desc_xmms, #endif NULL }; void free_demuxer_stream(demux_stream_t *ds){ ds_free_packs(ds); free(ds); } demux_stream_t* new_demuxer_stream(struct demuxer_st *demuxer,int id){ demux_stream_t* ds=malloc(sizeof(demux_stream_t)); ds->buffer_pos=ds->buffer_size=0; ds->buffer=NULL; ds->pts=0; ds->pts_bytes=0; ds->eof=0; ds->pos=0; ds->dpos=0; ds->pack_no=0; //--------------- ds->packs=0; ds->bytes=0; ds->first=ds->last=ds->current=NULL; ds->id=id; ds->demuxer=demuxer; //---------------- ds->asf_seq=-1; ds->asf_packet=NULL; //---------------- ds->ss_mul=ds->ss_div=0; //---------------- ds->sh=NULL; return ds; } /** * Get demuxer description structure for a given demuxer type * * @param file_format type of the demuxer * @return structure for the demuxer, NULL if not found */ static demuxer_desc_t* get_demuxer_desc_from_type(int file_format) { int i; for (i = 0; demuxer_list[i]; i++) if (file_format == demuxer_list[i]->type) return demuxer_list[i]; return NULL; } demuxer_t* new_demuxer(stream_t *stream,int type,int a_id,int v_id,int s_id,char *filename){ demuxer_t *d=malloc(sizeof(demuxer_t)); memset(d,0,sizeof(demuxer_t)); d->stream=stream; d->stream_pts = MP_NOPTS_VALUE; d->movi_start=stream->start_pos; d->movi_end=stream->end_pos; d->seekable=1; d->synced=0; d->filepos=0; d->audio=new_demuxer_stream(d,a_id); d->video=new_demuxer_stream(d,v_id); d->sub=new_demuxer_stream(d,s_id); d->type=type; if(type) if (!(d->desc = get_demuxer_desc_from_type(type))) mp_msg(MSGT_DEMUXER,MSGL_ERR,"BUG! Invalid demuxer type in new_demuxer(), big troubles ahead."); if(filename) // Filename hack for avs_check_file d->filename=strdup(filename); stream_reset(stream); stream_seek(stream,stream->start_pos); return d; } sh_sub_t *new_sh_sub_sid(demuxer_t *demuxer, int id, int sid) { if (id > MAX_S_STREAMS - 1 || id < 0) { mp_msg(MSGT_DEMUXER,MSGL_WARN,"Requested sub stream id overflow (%d > %d)\n", id, MAX_S_STREAMS); return NULL; } if (demuxer->s_streams[id]) mp_msg(MSGT_DEMUXER, MSGL_WARN, "Sub stream %i redefined\n", id); else { sh_sub_t *sh = calloc(1, sizeof(sh_sub_t)); demuxer->s_streams[id] = sh; sh->sid = sid; mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SUBTITLE_ID=%d\n", sid); } return demuxer->s_streams[id]; } sh_audio_t* new_sh_audio_aid(demuxer_t *demuxer,int id,int aid){ if(id > MAX_A_STREAMS-1 || id < 0) { mp_msg(MSGT_DEMUXER,MSGL_WARN,"Requested audio stream id overflow (%d > %d)\n", id, MAX_A_STREAMS); return NULL; } if(demuxer->a_streams[id]){ mp_msg(MSGT_DEMUXER,MSGL_WARN,MSGTR_AudioStreamRedefined,id); } else { sh_audio_t *sh; mp_msg(MSGT_DEMUXER,MSGL_V,MSGTR_FoundAudioStream,id); demuxer->a_streams[id]=calloc(1, sizeof(sh_audio_t)); sh = demuxer->a_streams[id]; // set some defaults sh->samplesize=2; sh->sample_format=AF_FORMAT_S16_NE; sh->audio_out_minsize=8192;/* default size, maybe not enough for Win32/ACM*/ sh->pts=MP_NOPTS_VALUE; mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_AUDIO_ID=%d\n", aid); } ((sh_audio_t *)demuxer->a_streams[id])->aid = aid; return demuxer->a_streams[id]; } void free_sh_audio(demuxer_t *demuxer, int id) { sh_audio_t *sh = demuxer->a_streams[id]; demuxer->a_streams[id] = NULL; mp_msg(MSGT_DEMUXER,MSGL_DBG2,"DEMUXER: freeing sh_audio at %p\n",sh); if(sh->wf) free(sh->wf); free(sh); } sh_video_t* new_sh_video_vid(demuxer_t *demuxer,int id,int vid){ if(id > MAX_V_STREAMS-1 || id < 0) { mp_msg(MSGT_DEMUXER,MSGL_WARN,"Requested video stream id overflow (%d > %d)\n", id, MAX_V_STREAMS); return NULL; } if(demuxer->v_streams[id]){ mp_msg(MSGT_DEMUXER,MSGL_WARN,MSGTR_VideoStreamRedefined,id); } else { mp_msg(MSGT_DEMUXER,MSGL_V,MSGTR_FoundVideoStream,id); demuxer->v_streams[id]=calloc(1, sizeof(sh_video_t)); mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VIDEO_ID=%d\n", vid); } ((sh_video_t *)demuxer->v_streams[id])->vid = vid; return demuxer->v_streams[id]; } void free_sh_video(sh_video_t* sh){ mp_msg(MSGT_DEMUXER,MSGL_DBG2,"DEMUXER: freeing sh_video at %p\n",sh); if(sh->bih) free(sh->bih); free(sh); } void free_demuxer(demuxer_t *demuxer){ int i; mp_msg(MSGT_DEMUXER,MSGL_DBG2,"DEMUXER: freeing demuxer at %p\n",demuxer); if(demuxer->desc->close) demuxer->desc->close(demuxer); // Very ugly hack to make it behave like old implementation if (demuxer->desc->type == DEMUXER_TYPE_DEMUXERS) goto skip_streamfree; // free streams: for(i = 0; i < MAX_A_STREAMS; i++) if(demuxer->a_streams[i]) free_sh_audio(demuxer, i); for(i = 0; i < MAX_V_STREAMS; i++) if(demuxer->v_streams[i]) free_sh_video(demuxer->v_streams[i]); // free demuxers: free_demuxer_stream(demuxer->audio); free_demuxer_stream(demuxer->video); free_demuxer_stream(demuxer->sub); skip_streamfree: if(demuxer->info) { for(i=0;demuxer->info[i] != NULL; i++) free(demuxer->info[i]); free(demuxer->info); } if(demuxer->filename) free(demuxer->filename); if (demuxer->chapters) { for (i=0; i<demuxer->num_chapters; i++) if (demuxer->chapters[i].name) free(demuxer->chapters[i].name); free(demuxer->chapters); } free(demuxer); } void ds_add_packet(demux_stream_t *ds,demux_packet_t* dp){ // demux_packet_t* dp=new_demux_packet(len); // stream_read(stream,dp->buffer,len); // dp->pts=pts; //(float)pts/90000.0f; // dp->pos=pos; // append packet to DS stream: ++ds->packs; ds->bytes+=dp->len; if(ds->last){ // next packet in stream ds->last->next=dp; ds->last=dp; } else { // first packet in stream ds->first=ds->last=dp; } mp_dbg(MSGT_DEMUXER,MSGL_DBG2,"DEMUX: Append packet to %s, len=%d pts=%5.3f pos=%u [packs: A=%d V=%d]\n", (ds==ds->demuxer->audio)?"d_audio":"d_video", dp->len,dp->pts,(unsigned int)dp->pos,ds->demuxer->audio->packs,ds->demuxer->video->packs); } void ds_read_packet(demux_stream_t *ds, stream_t *stream, int len, double pts, off_t pos, int flags) { demux_packet_t* dp=new_demux_packet(len); len = stream_read(stream,dp->buffer,len); resize_demux_packet(dp, len); dp->pts=pts; //(float)pts/90000.0f; dp->pos=pos; dp->flags=flags; // append packet to DS stream: ds_add_packet(ds,dp); } // return value: // 0 = EOF or no stream found or invalid type // 1 = successfully read a packet int demux_fill_buffer(demuxer_t *demux,demux_stream_t *ds){ // Note: parameter 'ds' can be NULL! // printf("demux->type=%d\n",demux->type); return demux->desc->fill_buffer(demux, ds); } // return value: // 0 = EOF // 1 = succesfull int ds_fill_buffer(demux_stream_t *ds){ demuxer_t *demux=ds->demuxer; if(ds->current) free_demux_packet(ds->current); if( mp_msg_test(MSGT_DEMUXER,MSGL_DBG3) ){ if(ds==demux->audio) mp_dbg(MSGT_DEMUXER,MSGL_DBG3,"ds_fill_buffer(d_audio) called\n");else if(ds==demux->video) mp_dbg(MSGT_DEMUXER,MSGL_DBG3,"ds_fill_buffer(d_video) called\n");else if(ds==demux->sub) mp_dbg(MSGT_DEMUXER,MSGL_DBG3,"ds_fill_buffer(d_sub) called\n");else mp_dbg(MSGT_DEMUXER,MSGL_DBG3,"ds_fill_buffer(unknown 0x%X) called\n",(unsigned int)ds); } while(1){ if(ds->packs){ demux_packet_t *p=ds->first; // copy useful data: ds->buffer=p->buffer; ds->buffer_pos=0; ds->buffer_size=p->len; ds->pos=p->pos; ds->dpos+=p->len; // !!! ++ds->pack_no; if (p->pts != (correct_pts ? MP_NOPTS_VALUE : 0)) { ds->pts=p->pts; ds->pts_bytes=0; } ds->pts_bytes+=p->len; // !!! if(p->stream_pts != MP_NOPTS_VALUE) demux->stream_pts=p->stream_pts; ds->flags=p->flags; // unlink packet: ds->bytes-=p->len; ds->current=p; ds->first=p->next; if(!ds->first) ds->last=NULL; --ds->packs; return 1; //ds->buffer_size; } if(demux->audio->packs>=MAX_PACKS || demux->audio->bytes>=MAX_PACK_BYTES){ mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_TooManyAudioInBuffer,demux->audio->packs,demux->audio->bytes); mp_msg(MSGT_DEMUXER,MSGL_HINT,MSGTR_MaybeNI); break; } if(demux->video->packs>=MAX_PACKS || demux->video->bytes>=MAX_PACK_BYTES){ mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_TooManyVideoInBuffer,demux->video->packs,demux->video->bytes); mp_msg(MSGT_DEMUXER,MSGL_HINT,MSGTR_MaybeNI); break; } if(!demux_fill_buffer(demux,ds)){ mp_dbg(MSGT_DEMUXER,MSGL_DBG2,"ds_fill_buffer()->demux_fill_buffer() failed\n"); break; // EOF } } ds->buffer_pos=ds->buffer_size=0; ds->buffer=NULL; ds->current=NULL; mp_msg(MSGT_DEMUXER,MSGL_V,"ds_fill_buffer: EOF reached (stream: %s) \n",ds==demux->audio?"audio":"video"); ds->eof=1; return 0; } int demux_read_data(demux_stream_t *ds,unsigned char* mem,int len){ int x; int bytes=0; while(len>0){ x=ds->buffer_size-ds->buffer_pos; if(x==0){ if(!ds_fill_buffer(ds)) return bytes; } else { if(x>len) x=len; if(mem) memcpy(mem+bytes,&ds->buffer[ds->buffer_pos],x); bytes+=x;len-=x;ds->buffer_pos+=x; } } return bytes; } int demux_read_data_pack(demux_stream_t *ds,unsigned char* mem,int len){ int x; int bytes=0; while(len>0){ x=ds->buffer_size-ds->buffer_pos; if(x==0){ if(!ds_fill_buffer(ds)) return bytes; } else { if(x>len) x=len; if(mem) memcpy(mem+bytes,&ds->buffer[ds->buffer_pos],x); bytes+=x;len-=x;ds->buffer_pos+=x; return bytes; // stop at end of package! (for correct timestamping) } } return bytes; } /** * \brief read data until the given 3-byte pattern is encountered, up to maxlen * \param mem memory to read data into, may be NULL to discard data * \param maxlen maximum number of bytes to read * \param read number of bytes actually read * \param pattern pattern to search for (lowest 8 bits are ignored) * \return whether pattern was found */ int demux_pattern_3(demux_stream_t *ds, unsigned char *mem, int maxlen, int *read, uint32_t pattern) { register uint32_t head = 0xffffff00; register uint32_t pat = pattern & 0xffffff00; int total_len = 0; do { register unsigned char *ds_buf = &ds->buffer[ds->buffer_size]; int len = ds->buffer_size - ds->buffer_pos; register long pos = -len; if (unlikely(pos >= 0)) { // buffer is empty ds_fill_buffer(ds); continue; } do { head |= ds_buf[pos]; head <<= 8; } while (++pos && head != pat); len += pos; if (total_len + len > maxlen) len = maxlen - total_len; len = demux_read_data(ds, mem ? &mem[total_len] : NULL, len); total_len += len; } while ((head != pat || total_len < 3) && total_len < maxlen && !ds->eof); if (read) *read = total_len; return total_len >= 3 && head == pat; } void ds_free_packs(demux_stream_t *ds){ demux_packet_t *dp=ds->first; while(dp){ demux_packet_t *dn=dp->next; free_demux_packet(dp); dp=dn; } if(ds->asf_packet){ // free unfinished .asf fragments: free(ds->asf_packet->buffer); free(ds->asf_packet); ds->asf_packet=NULL; } ds->first=ds->last=NULL; ds->packs=0; // !!!!! ds->bytes=0; if(ds->current) free_demux_packet(ds->current); ds->current=NULL; ds->buffer=NULL; ds->buffer_pos=ds->buffer_size; ds->pts=0; ds->pts_bytes=0; } int ds_get_packet(demux_stream_t *ds,unsigned char **start){ while(1){ int len; if(ds->buffer_pos>=ds->buffer_size){ if(!ds_fill_buffer(ds)){ // EOF *start = NULL; return -1; } } len=ds->buffer_size-ds->buffer_pos; *start = &ds->buffer[ds->buffer_pos]; ds->buffer_pos+=len; return len; } } int ds_get_packet_pts(demux_stream_t *ds,unsigned char **start, double *pts) { int len; *pts = MP_NOPTS_VALUE; if(ds->buffer_pos>=ds->buffer_size){ if (!ds_fill_buffer(ds)) { // EOF *start = NULL; return -1; } } // Should use MP_NOPTS_VALUE for "unknown pts" in the packets too // Return pts unless this read starts from the middle of a packet if (!ds->buffer_pos && (correct_pts || ds->current->pts)) *pts = ds->current->pts; len=ds->buffer_size-ds->buffer_pos; *start = &ds->buffer[ds->buffer_pos]; ds->buffer_pos+=len; return len; } int ds_get_packet_sub(demux_stream_t *ds,unsigned char **start){ while(1){ int len; if(ds->buffer_pos>=ds->buffer_size){ *start = NULL; if(!ds->packs) return -1; // no sub if(!ds_fill_buffer(ds)) return -1; // EOF } len=ds->buffer_size-ds->buffer_pos; *start = &ds->buffer[ds->buffer_pos]; ds->buffer_pos+=len; return len; } } double ds_get_next_pts(demux_stream_t *ds) { demuxer_t* demux = ds->demuxer; while(!ds->first) { if(demux->audio->packs>=MAX_PACKS || demux->audio->bytes>=MAX_PACK_BYTES){ mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_TooManyAudioInBuffer,demux->audio->packs,demux->audio->bytes); mp_msg(MSGT_DEMUXER,MSGL_HINT,MSGTR_MaybeNI); return MP_NOPTS_VALUE; } if(demux->video->packs>=MAX_PACKS || demux->video->bytes>=MAX_PACK_BYTES){ mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_TooManyVideoInBuffer,demux->video->packs,demux->video->bytes); mp_msg(MSGT_DEMUXER,MSGL_HINT,MSGTR_MaybeNI); return MP_NOPTS_VALUE; } if(!demux_fill_buffer(demux,ds)) return MP_NOPTS_VALUE; } return ds->first->pts; } // ==================================================================== // commandline options, flags: extern int force_ni; extern int pts_from_bps; //extern int audio_id; //extern int video_id; //extern int dvdsub_id; void demuxer_help(void) { int i; mp_msg(MSGT_DEMUXER, MSGL_INFO, "Available demuxers:\n"); mp_msg(MSGT_DEMUXER, MSGL_INFO, " demuxer: type info: (comment)\n"); mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_DEMUXERS\n"); for (i = 0; demuxer_list[i]; i++) { if (demuxer_list[i]->type > DEMUXER_TYPE_MAX) // Don't display special demuxers continue; if (demuxer_list[i]->comment && strlen(demuxer_list[i]->comment)) mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s %2d %s (%s)\n", demuxer_list[i]->name, demuxer_list[i]->type, demuxer_list[i]->info, demuxer_list[i]->comment); else mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s %2d %s\n", demuxer_list[i]->name, demuxer_list[i]->type, demuxer_list[i]->info); } } /** * Get demuxer type for a given demuxer name * * @param demuxer_name string with demuxer name of demuxer number * @param force will be set if demuxer should be forced. * May be NULL. * @return DEMUXER_TYPE_xxx, -1 if error or not found */ int get_demuxer_type_from_name(char *demuxer_name, int *force) { int i; long type_int; char *endptr; if (!demuxer_name || !demuxer_name[0]) return DEMUXER_TYPE_UNKNOWN; if (force) *force = demuxer_name[0] == '+'; if (demuxer_name[0] == '+') demuxer_name = &demuxer_name[1]; for (i = 0; demuxer_list[i]; i++) { if (demuxer_list[i]->type > DEMUXER_TYPE_MAX) // Can't select special demuxers from commandline continue; if (strcmp(demuxer_name, demuxer_list[i]->name) == 0) return demuxer_list[i]->type; } // No match found, try to parse name as an integer (demuxer number) type_int = strtol(demuxer_name, &endptr, 0); if (*endptr) // Conversion failed return -1; if ((type_int > 0) && (type_int <= DEMUXER_TYPE_MAX)) return (int)type_int; return -1; } int extension_parsing=1; // 0=off 1=mixed (used only for unstable formats) int correct_pts=0; /* NOTE : Several demuxers may be opened at the same time so demuxers should NEVER rely on an external var to enable them self. If a demuxer can't do any autodection it should only use file_format. The user can explictly set file_format with the -demuxer option so there is really no need for another extra var. For conivence an option can be added to set file_format directly to the right type (ex: rawaudio,rawvideo). Also the stream can override the file_format so a demuxer wich rely on a special stream type can set file_format at the stream level (ex: tv,mf). */ static demuxer_t* demux_open_stream(stream_t *stream, int file_format, int force, int audio_id, int video_id, int dvdsub_id, char* filename) { //int file_format=(*file_format_ptr); demuxer_t *demuxer=NULL; sh_video_t *sh_video=NULL; demuxer_desc_t *demuxer_desc; int fformat; int i; //printf("demux_open(%p,%d,%d,%d,%d) \n",stream,file_format,audio_id,video_id,dvdsub_id); // If somebody requested a demuxer check it if (file_format) { if ((demuxer_desc = get_demuxer_desc_from_type(file_format))) { demuxer = new_demuxer(stream,demuxer_desc->type,audio_id,video_id,dvdsub_id,filename); if (demuxer_desc->check_file) fformat = demuxer_desc->check_file(demuxer); if (force || !demuxer_desc->check_file) fformat = demuxer_desc->type; if (fformat != 0) { if (fformat == demuxer_desc->type) { demuxer_t *demux2 = demuxer; // Move messages to demuxer detection code? mp_msg(MSGT_DEMUXER, MSGL_INFO, MSGTR_Detected_XXX_FileFormat, demuxer_desc->shortdesc); file_format = demuxer_desc->type = fformat; if (!demuxer->desc->open || (demux2 = demuxer->desc->open(demuxer))) { demuxer = demux2; goto dmx_open; } } else { // Format changed after check, recurse free_demuxer(demuxer); return demux_open_stream(stream, fformat, force, audio_id, video_id, dvdsub_id, filename); } } // Check failed for forced demuxer, quit free_demuxer(demuxer); return NULL; } } // Test demuxers with safe file checks for (i = 0; (demuxer_desc = demuxer_list[i]); i++) { if (demuxer_desc->safe_check) { demuxer = new_demuxer(stream,demuxer_desc->type,audio_id,video_id,dvdsub_id,filename); if ((fformat = demuxer_desc->check_file(demuxer)) != 0) { if (fformat == demuxer_desc->type) { demuxer_t *demux2 = demuxer; mp_msg(MSGT_DEMUXER, MSGL_INFO, MSGTR_Detected_XXX_FileFormat, demuxer_desc->shortdesc); file_format = fformat; if (!demuxer->desc->open || (demux2 = demuxer->desc->open(demuxer))) { demuxer = demux2; goto dmx_open; } } else { if (fformat == DEMUXER_TYPE_PLAYLIST) return demuxer; // handled in mplayer.c // Format changed after check, recurse free_demuxer(demuxer); demuxer=demux_open_stream(stream, fformat, force, audio_id, video_id, dvdsub_id, filename); if(demuxer) return demuxer; // done! file_format = DEMUXER_TYPE_UNKNOWN; } } free_demuxer(demuxer); demuxer = NULL; } } // If no forced demuxer perform file extension based detection // Ok. We're over the stable detectable fileformats, the next ones are a bit // fuzzy. So by default (extension_parsing==1) try extension-based detection // first: if(file_format==DEMUXER_TYPE_UNKNOWN && filename && extension_parsing==1){ file_format=demuxer_type_by_filename(filename); if(file_format!=DEMUXER_TYPE_UNKNOWN){ // we like recursion :) demuxer=demux_open_stream(stream, file_format, force, audio_id, video_id, dvdsub_id, filename); if(demuxer) return demuxer; // done! file_format=DEMUXER_TYPE_UNKNOWN; // continue fuzzy guessing... mp_msg(MSGT_DEMUXER,MSGL_V,"demuxer: continue fuzzy content-based format guessing...\n"); } } // Try detection for all other demuxers for (i = 0; (demuxer_desc = demuxer_list[i]); i++) { if (!demuxer_desc->safe_check && demuxer_desc->check_file) { demuxer = new_demuxer(stream,demuxer_desc->type,audio_id,video_id,dvdsub_id,filename); if ((fformat = demuxer_desc->check_file(demuxer)) != 0) { if (fformat == demuxer_desc->type) { demuxer_t *demux2 = demuxer; mp_msg(MSGT_DEMUXER, MSGL_INFO, MSGTR_Detected_XXX_FileFormat, demuxer_desc->shortdesc); file_format = fformat; if (!demuxer->desc->open || (demux2 = demuxer->desc->open(demuxer))) { demuxer = demux2; goto dmx_open; } } else { if (fformat == DEMUXER_TYPE_PLAYLIST) return demuxer; // handled in mplayer.c // Format changed after check, recurse free_demuxer(demuxer); demuxer=demux_open_stream(stream, fformat, force, audio_id, video_id, dvdsub_id, filename); if(demuxer) return demuxer; // done! file_format = DEMUXER_TYPE_UNKNOWN; } } free_demuxer(demuxer); demuxer = NULL; } } return NULL; //====== File format recognized, set up these for compatibility: ========= dmx_open: demuxer->file_format=file_format; if ((sh_video=demuxer->video->sh) && sh_video->bih){ int biComp=le2me_32(sh_video->bih->biCompression); mp_msg(MSGT_DEMUX,MSGL_INFO,"VIDEO: [%.4s] %dx%d %dbpp %5.3f fps %5.1f kbps (%4.1f kbyte/s)\n", (char *)&biComp, sh_video->bih->biWidth, sh_video->bih->biHeight, sh_video->bih->biBitCount, sh_video->fps, sh_video->i_bps*0.008f, sh_video->i_bps/1024.0f ); } return demuxer; } char* audio_stream = NULL; char* sub_stream = NULL; int demuxer_type = 0; // used by rawaudio and rawvideo int audio_stream_cache = 0; char *demuxer_name = NULL; // parameter from -demuxer char *audio_demuxer_name = NULL; // parameter from -audio-demuxer char *sub_demuxer_name = NULL; // parameter from -sub-demuxer extern int hr_mp3_seek; extern float stream_cache_min_percent; extern float stream_cache_seek_min_percent; demuxer_t* demux_open(stream_t *vs,int file_format,int audio_id,int video_id,int dvdsub_id,char* filename){ stream_t *as = NULL,*ss = NULL; demuxer_t *vd,*ad = NULL,*sd = NULL; int afmt =DEMUXER_TYPE_UNKNOWN,sfmt = DEMUXER_TYPE_UNKNOWN ; int audio_demuxer_type = 0, sub_demuxer_type = 0; int demuxer_force = 0, audio_demuxer_force = 0, sub_demuxer_force = 0; if ((demuxer_type = get_demuxer_type_from_name(demuxer_name, &demuxer_force)) < 0) { mp_msg(MSGT_DEMUXER,MSGL_ERR,"-demuxer %s does not exist.\n",demuxer_name); } if ((audio_demuxer_type = get_demuxer_type_from_name(audio_demuxer_name, &audio_demuxer_force)) < 0) { mp_msg(MSGT_DEMUXER,MSGL_ERR,"-audio-demuxer %s does not exist.\n",audio_demuxer_name); } if ((sub_demuxer_type = get_demuxer_type_from_name(sub_demuxer_name, &sub_demuxer_force)) < 0) { mp_msg(MSGT_DEMUXER,MSGL_ERR,"-sub-demuxer %s does not exist.\n",sub_demuxer_name); } if(audio_stream) { as = open_stream(audio_stream,0,&afmt); if(!as) { mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_CannotOpenAudioStream,audio_stream); return NULL; } if(audio_stream_cache) { if(!stream_enable_cache(as,audio_stream_cache*1024,audio_stream_cache*1024*(stream_cache_min_percent / 100.0), audio_stream_cache*1024*(stream_cache_seek_min_percent / 100.0))) { free_stream(as); mp_msg(MSGT_DEMUXER,MSGL_ERR,"Can't enable audio stream cache\n"); return NULL; } } } if(sub_stream) { ss = open_stream(sub_stream,0,&sfmt); if(!ss) { mp_msg(MSGT_DEMUXER,MSGL_ERR,MSGTR_CannotOpenSubtitlesStream,sub_stream); return NULL; } } vd = demux_open_stream(vs, demuxer_type ? demuxer_type : file_format, demuxer_force, audio_stream ? -2 : audio_id, video_id, sub_stream ? -2 : dvdsub_id, filename); if(!vd) { if(as) free_stream(as); if(ss) free_stream(ss); return NULL; } if(as) { ad = demux_open_stream(as, audio_demuxer_type ? audio_demuxer_type : afmt, audio_demuxer_force, audio_id, -2, -2, audio_stream); if(!ad) { mp_msg(MSGT_DEMUXER,MSGL_WARN,MSGTR_OpeningAudioDemuxerFailed,audio_stream); free_stream(as); } else if(ad->audio->sh && ((sh_audio_t*)ad->audio->sh)->format == 0x55) // MP3 hr_mp3_seek=1; // Enable high res seeking } if(ss) { sd = demux_open_stream(ss, sub_demuxer_type ? sub_demuxer_type : sfmt, sub_demuxer_force, -2, -2, dvdsub_id, sub_stream); if(!sd) { mp_msg(MSGT_DEMUXER,MSGL_WARN,MSGTR_OpeningSubtitlesDemuxerFailed,sub_stream); free_stream(ss); } } if(ad && sd) return new_demuxers_demuxer(vd,ad,sd); else if(ad) return new_demuxers_demuxer(vd,ad,vd); else if(sd) return new_demuxers_demuxer(vd,vd,sd); else return vd; } int demux_seek(demuxer_t *demuxer,float rel_seek_secs,float audio_delay,int flags){ demux_stream_t *d_audio=demuxer->audio; demux_stream_t *d_video=demuxer->video; sh_audio_t *sh_audio=d_audio->sh; sh_video_t *sh_video=d_video->sh; double tmp = 0; double pts; if(!demuxer->seekable){ if(demuxer->file_format==DEMUXER_TYPE_AVI) mp_msg(MSGT_SEEK,MSGL_WARN,MSGTR_CantSeekRawAVI); #ifdef USE_TV else if (demuxer->file_format==DEMUXER_TYPE_TV) mp_msg(MSGT_SEEK,MSGL_WARN,MSGTR_TVInputNotSeekable); #endif else mp_msg(MSGT_SEEK,MSGL_WARN,MSGTR_CantSeekFile); return 0; } // clear demux buffers: if(sh_audio){ ds_free_packs(d_audio);sh_audio->a_buffer_len=0;} ds_free_packs(d_video); demuxer->stream->eof=0; // clear eof flag demuxer->video->eof=0; demuxer->audio->eof=0; #if 0 if(sh_audio) sh_audio->timer=sh_video->timer; #else if(sh_audio) sh_audio->delay=0; if(sh_video) sh_video->timer=0; // !!!!!! #endif if(flags & 1) // absolute seek pts = 0.0f; else { if(demuxer->stream_pts == MP_NOPTS_VALUE) goto dmx_seek; pts = demuxer->stream_pts; } if(flags & 2) { // percent seek if(stream_control(demuxer->stream, STREAM_CTRL_GET_TIME_LENGTH, &tmp) == STREAM_UNSUPORTED) goto dmx_seek; pts += tmp * rel_seek_secs; } else pts += rel_seek_secs; if(stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_TIME, &pts) != STREAM_UNSUPORTED) { demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL); return 1; } dmx_seek: if (demuxer->desc->seek) demuxer->desc->seek(demuxer,rel_seek_secs,audio_delay,flags); if (sh_audio) resync_audio_stream(sh_audio); return 1; } int demux_info_add(demuxer_t *demuxer, const char *opt, const char *param) { char **info = demuxer->info; int n = 0; for(n = 0; info && info[2*n] != NULL; n++) { if(!strcasecmp(opt,info[2*n])) { mp_msg(MSGT_DEMUX, MSGL_INFO,MSGTR_DemuxerInfoChanged,opt,param); free(info[2*n+1]); info[2*n+1] = strdup(param); return 0; } } info = demuxer->info = (char**)realloc(info,(2*(n+2))*sizeof(char*)); info[2*n] = strdup(opt); info[2*n+1] = strdup(param); memset(&info[2*(n+1)],0,2*sizeof(char*)); return 1; } int demux_info_print(demuxer_t *demuxer) { char **info = demuxer->info; int n; if(!info) return 0; mp_msg(MSGT_DEMUX, MSGL_INFO,MSGTR_ClipInfo); for(n = 0; info[2*n] != NULL ; n++) { mp_msg(MSGT_DEMUX, MSGL_INFO, " %s: %s\n",info[2*n],info[2*n+1]); mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_NAME%d=%s\n", n, info[2*n]); mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_VALUE%d=%s\n", n, info[2*n+1]); } mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_N=%d\n", n); return 0; } char* demux_info_get(demuxer_t *demuxer, char *opt) { int i; char **info = demuxer->info; for(i = 0; info && info[2*i] != NULL; i++) { if(!strcasecmp(opt,info[2*i])) return info[2*i+1]; } return NULL; } int demux_control(demuxer_t *demuxer, int cmd, void *arg) { if (demuxer->desc->control) return demuxer->desc->control(demuxer,cmd,arg); return DEMUXER_CTRL_NOTIMPL; } double demuxer_get_time_length(demuxer_t *demuxer){ double get_time_ans; sh_video_t *sh_video = demuxer->video->sh; sh_audio_t *sh_audio = demuxer->audio->sh; // <= 0 means DEMUXER_CTRL_NOTIMPL or DEMUXER_CTRL_DONTKNOW if (demux_control(demuxer, DEMUXER_CTRL_GET_TIME_LENGTH,(void *)&get_time_ans)<=0) { if (sh_video && sh_video->i_bps) get_time_ans = (double)(demuxer->movi_end-demuxer->movi_start)/sh_video->i_bps; else if (sh_audio && sh_audio->i_bps) get_time_ans = (double)(demuxer->movi_end-demuxer->movi_start)/sh_audio->i_bps; else get_time_ans=0; } return get_time_ans; } /** * \brief demuxer_get_current_time() returns the time of the current play in three possible ways: * either when the stream reader satisfies STREAM_CTRL_GET_CURRENT_TIME (e.g. dvd) * or using sh_video->pts when the former method fails * 0 otherwise * \return the current play time */ int demuxer_get_current_time(demuxer_t *demuxer){ double get_time_ans = 0; sh_video_t *sh_video = demuxer->video->sh; if(demuxer->stream_pts != MP_NOPTS_VALUE) get_time_ans = demuxer->stream_pts; else if(sh_video) get_time_ans = sh_video->pts; return (int) get_time_ans; } int demuxer_get_percent_pos(demuxer_t *demuxer){ int ans = 0; int res = demux_control(demuxer, DEMUXER_CTRL_GET_PERCENT_POS, &ans); int len = (demuxer->movi_end - demuxer->movi_start) / 100; if (res <= 0) { if (len > 0) ans = (demuxer->filepos - demuxer->movi_start) / len; else ans = 0; } if (ans < 0) ans = 0; if (ans > 100) ans = 100; return ans; } int demuxer_switch_audio(demuxer_t *demuxer, int index){ int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_AUDIO, &index); if (res == DEMUXER_CTRL_NOTIMPL) index = demuxer->audio->id; return index; } int demuxer_switch_video(demuxer_t *demuxer, int index){ int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_VIDEO, &index); if (res == DEMUXER_CTRL_NOTIMPL) index = demuxer->video->id; return index; } int demuxer_add_chapter(demuxer_t* demuxer, const char* name, uint64_t start, uint64_t end){ if (demuxer->chapters == NULL) demuxer->chapters = malloc (32*sizeof(*demuxer->chapters)); else if (!(demuxer->num_chapters % 32)) demuxer->chapters = realloc (demuxer->chapters, (demuxer->num_chapters + 32) * sizeof(*demuxer->chapters)); demuxer->chapters[demuxer->num_chapters].start = start; demuxer->chapters[demuxer->num_chapters].end = end; demuxer->chapters[demuxer->num_chapters].name = strdup(name); return demuxer->num_chapters ++; } /** * \brief demuxer_seek_chapter() seeks to a chapter in two possible ways: * either using the demuxer->chapters structure set by the demuxer * or asking help to the stream layer (e.g. dvd) * \param chapter - chapter number wished - 0-based * \param mode 0: relative to current main pts, 1: absolute * \param seek_pts set by the function to the pts to seek to (if demuxer->chapters is set) * \param num_chapters number of chapters present (set by this function is param is not null) * \param chapter_name name of chapter found (set by this function is param is not null) * \return -1 on error, current chapter if successful */ int demuxer_seek_chapter(demuxer_t *demuxer, int chapter, int mode, float *seek_pts, int *num_chapters, char **chapter_name) { int ris; int current, total; sh_video_t *sh_video = demuxer->video->sh; sh_audio_t *sh_audio = demuxer->audio->sh; if (!demuxer->num_chapters || !demuxer->chapters) { if(!mode) { ris = stream_control(demuxer->stream, STREAM_CTRL_GET_CURRENT_CHAPTER, ¤t); if(ris == STREAM_UNSUPORTED) return -1; chapter += current; } if(demuxer->video->sh) ds_free_packs(demuxer->video); if(demuxer->audio->sh) ds_free_packs(demuxer->audio); if(demuxer->sub->id >= 0) ds_free_packs(demuxer->sub); ris = stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_CHAPTER, &chapter); if(ris != STREAM_UNSUPORTED) demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL); if(sh_video) { ds_fill_buffer(demuxer->video); resync_video_stream(sh_video); } if(sh_audio) { ds_fill_buffer(demuxer->audio); resync_audio_stream(sh_audio); } //exit status may be ok, but main() doesn't have to seek itself (because e.g. dvds depend on sectors, not on pts) *seek_pts = -1.0; if(num_chapters) { if(stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_CHAPTERS, num_chapters) == STREAM_UNSUPORTED) *num_chapters = 0; } if(chapter_name) { char *tmp = malloc(16); *chapter_name = NULL; if(num_chapters && *num_chapters && tmp) { sprintf(tmp, " of %3d", *num_chapters); *chapter_name = tmp; } } return (ris != STREAM_UNSUPORTED ? chapter : -1); } else { //chapters structure is set in the demuxer total = demuxer->num_chapters; if (mode==1) { //absolute seeking current = chapter; } else { //relative seeking uint64_t now; now = (sh_video ? sh_video->pts : (sh_audio ? sh_audio->pts : 0.)) * 1000 + .5; for (current = total - 1; current >= 0; --current) { demux_chapter_t* chapter = demuxer->chapters + current; if (chapter->start <= now) break; } current += chapter; } if (current >= total) return -1; if (current < 0) current = 0; *seek_pts = demuxer->chapters[current].start / 1000.0; if(num_chapters) *num_chapters = demuxer->num_chapters; if(chapter_name) { if(demuxer->chapters[current].name) *chapter_name = strdup(demuxer->chapters[current].name); else *chapter_name = NULL; } return current; } }