bug fixed gb mem leak

zhangmeng

2019-10-28 f2fb5eb58ae50bb35f2d0ae2673a596adc06cb00

 csrc/ffmpeg/format/FormatIn.cpp

@@ -38,38 +38,37 @@

   FormatIn::~FormatIn()
   {
      if(io_ctx_){
         if(read_io_buff_){
            // av_free(read_io_buff_);
            read_io_buff_ = NULL;
         }
         avio_context_free(&io_ctx_);
         io_ctx_ = NULL;
      }
      if(ctx_){
         if (!ctx_->oformat){
            avformat_free_context(ctx_);
         }else{
         if (!(ctx_->flags & AVFMT_FLAG_CUSTOM_IO)){
            avformat_close_input(&ctx_);
         }else{
            avformat_free_context(ctx_);
         }

         ctx_ = NULL;
         if(dec_ctx_){
            avcodec_close(dec_ctx_);
            dec_ctx_ = NULL;
         }
			
      }
		
      if (handle_gb28181){
         delete handle_gb28181;
      }
      if(read_io_buff_){
         av_free(read_io_buff_);
         read_io_buff_ = NULL;
      }
      if(io_ctx_){
         avio_context_free(&io_ctx_);
         io_ctx_ = NULL;
      }

   }

////////////////////////////////////////////////////////////////////////
   int FormatIn::openWithCustomIO(void *opaque, read_packet fn, AVDictionary **options/*=NULL*/){
      ctx_ = avformat_alloc_context();
      if(!ctx_){
         logIt("open with custom io create format error\n");
         return -1;
      }
      read_io_buff_ = (uint8_t*)av_malloc(read_io_buff_size_);
      if(!read_io_buff_){
         logIt("open with custom io alloc read io buff error\n");
@@ -81,12 +80,18 @@
         logIt("open with custom io create custom avio error\n");
         return -1;
      }
		
      ctx_ = avformat_alloc_context();
      if(!ctx_){
         logIt("open with custom io create format error\n");
         return -1;
      }

      ctx_->pb = io_ctx_;

      auto err = av_probe_input_buffer(ctx_->pb, &ctx_->iformat, NULL, NULL, 0, read_io_buff_size_);
      auto err = av_probe_input_buffer(ctx_->pb, &ctx_->iformat, NULL, NULL, 0, 0);
      if(err != 0){
         logIt("open with custom io prob input buffer error:%d\n", err);
            logIt("custom io failed:%s", getAVErrorDesc(err).c_str());
         logIt("open with custom io prob input buffer error:%d err: %s\n", err, getAVErrorDesc(err).c_str());
         return -1;
      }

@@ -154,11 +159,11 @@
            vs_idx_ = i;

            auto in = ctx_->streams[i];
            if(in->avg_frame_rate.num >=1 && in->avg_frame_rate.den >= 1){
                   fps_ = av_q2d(in->avg_frame_rate);
               }else if(in->r_frame_rate.num >=1 && in->r_frame_rate.den >= 1){
            if(in->r_frame_rate.num >=1 && in->r_frame_rate.den >= 1){
                   fps_ = av_q2d(in->r_frame_rate);
               }
               }else if(in->avg_frame_rate.num >=1 && in->avg_frame_rate.den >= 1){
                   fps_ = av_q2d(in->avg_frame_rate);
               } 
            logIt("in stream video fps %f, time_base: %d : %d", fps_, in->time_base.num, in->time_base.den);
         }
         if (type == AVMEDIA_TYPE_AUDIO){
@@ -172,6 +177,10 @@
      }

      return true;
   }

   const bool FormatIn::IsHEVC()const{
      return ctx_->streams[vs_idx_]->codecpar->codec_id == AV_CODEC_ID_HEVC;
   }

   bool FormatIn::openCodec(AVDictionary **options){
@@ -298,7 +307,9 @@
   
   int FormatIn::readPacket(AVPacket *pkt_out){

      return av_read_frame(ctx_, pkt_out);
      auto flag = av_read_frame(ctx_, pkt_out);
		
      return flag;
   }

   int FormatIn::decode(AVFrame* frame, AVPacket *pkt){