houxiao
2017-08-17 0250edfd9c4d453f25a2f9827698ccf87ff5afff
RtspFace/PL_RTSPServer.cpp
@@ -2,8 +2,8 @@
#include "MaterialBuffer.h"
#include "logger.h"
#include <liveMedia.hh>
#include <BasicUsageEnvironment.hh>
#include <liveMedia/liveMedia.hh>
#include <BasicUsageEnvironment/BasicUsageEnvironment.hh>
#include "FFmpegRTSPServer/IEncoder.h"
#include "FFmpegRTSPServer/LiveRTSPServer.h"
@@ -96,9 +96,18 @@
         ReleaseFrame();
         return 0;
      }
      //static size_t f = 0;
      //static FILE *pFile = fopen("/data/bb.264", "wb");
      //fwrite(in.buffer, sizeof(char), in.buffSize, pFile);
      //if (++f > 400){
      //   fclose(pFile);
      //   exit(0);
      //}
      uint8_t* pBuffer = in.buffer;
      size_t newBufferSize = in.buffSize;
      if (in.config.payWithAux)
      {
         if (newBufferSize <= 4)
@@ -109,11 +118,12 @@
         pBuffer += 4;
         newBufferSize -= 4;
      }
      *FrameBuffer = pBuffer;
      *FrameSize = newBufferSize;
      LOG(DEBUG) << "send frame size=" << in.buffSize;
      LOG_DEBUG << "send frame size=" << in.buffSize << std::endl;
      return 1;
   }
   
   virtual char ReleaseFrame()
@@ -125,7 +135,7 @@
         int ret = pthread_mutex_unlock(in.frame_mutex);
         if(ret != 0)
         {
            LOG(WARN) << "pthread_mutex_unlock frame_mutex: " << strerror(ret);
            LOG_WARN << "pthread_mutex_unlock frame_mutex: " << strerror(ret) << std::endl;
            return 0;
         }
      }
@@ -143,7 +153,7 @@
         int ret = pthread_mutex_lock(in.frame_mutex);
         if(ret != 0)
         {
            LOG(WARN) << "pthread_mutex_lock frame_mutex: " << strerror(ret);
            LOG_WARN << "pthread_mutex_lock frame_mutex: " << strerror(ret) << std::endl;
            return;
         }
      }
@@ -197,7 +207,7 @@
   int ret = pthread_create(&(in->live_daemon_thid), NULL, live_daemon_thd, in);
   if(ret != 0)
   {
      LOG(ERROR) << "pthread_create: " << strerror(ret);
      LOG_ERROR << "pthread_create: " << strerror(ret) << std::endl;
      return false;
   }
@@ -220,19 +230,19 @@
   
   if (pm.type != PipeMaterial::PMT_FRAME)
   {
      LOG(ERROR) << "PL_RTSPServer::pay only support PMT_FRAME";
      LOG_ERROR << "PL_RTSPServer::pay only support PMT_FRAME" << std::endl;
      return false;
   }
   
   if (in->buffSize > 0)
      LOG(WARN) << "PL_RTSPServer::pay may lost data size=" << in->buffSize;
      LOG_WARN << "PL_RTSPServer::pay may lost data size=" << in->buffSize << std::endl;
   
   MB_Frame* frame = (MB_Frame*)pm.buffer;
   if (frame->buffer == nullptr)
      return false;
   memcpy(in->buffer, frame->buffer, frame->buffSize);
   in->buffSize = frame->buffSize;
   if (in->encoderStub == nullptr)
      return false;