From 5ce6085328a841e270c7ce1ddfdd36efabf0b095 Mon Sep 17 00:00:00 2001 From: xingzilong <xingzilong@454eff88-639b-444f-9e54-f578c98de674> Date: 星期四, 17 八月 2017 11:46:03 +0800 Subject: [PATCH] h264裸数据读取 --- RtspFace/PL_RTSPServer.cpp | 38 ++++++++++++++++++++++++++------------ 1 files changed, 26 insertions(+), 12 deletions(-) diff --git a/RtspFace/PL_RTSPServer.cpp b/RtspFace/PL_RTSPServer.cpp index 6c6394b..4f140ca 100644 --- a/RtspFace/PL_RTSPServer.cpp +++ b/RtspFace/PL_RTSPServer.cpp @@ -1,8 +1,9 @@ #include "PL_RTSPServer.h" #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" @@ -95,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) @@ -108,11 +118,12 @@ pBuffer += 4; newBufferSize -= 4; } - + *FrameBuffer = pBuffer; *FrameSize = newBufferSize; - printf("send frame size=%u\n", in.buffSize); + LOG_DEBUG << "send frame size=" << in.buffSize << std::endl; + return 1; } virtual char ReleaseFrame() @@ -124,7 +135,7 @@ int ret = pthread_mutex_unlock(in.frame_mutex); if(ret != 0) { - printf("pthread_mutex_unlock frame_mutex: %s/n", strerror(ret)); + LOG_WARN << "pthread_mutex_unlock frame_mutex: " << strerror(ret) << std::endl; return 0; } } @@ -142,7 +153,7 @@ int ret = pthread_mutex_lock(in.frame_mutex); if(ret != 0) { - printf("pthread_mutex_lock frame_mutex: %s/n", strerror(ret)); + LOG_WARN << "pthread_mutex_lock frame_mutex: " << strerror(ret) << std::endl; return; } } @@ -196,7 +207,7 @@ int ret = pthread_create(&(in->live_daemon_thid), NULL, live_daemon_thd, in); if(ret != 0) { - printf("pthread_create: %s/n", strerror(ret)); + LOG_ERROR << "pthread_create: " << strerror(ret) << std::endl; return false; } @@ -214,22 +225,24 @@ { RTSPServer_Internal* in = (RTSPServer_Internal*)internal; - if (pm.buffer == nullptr || pm.buffSize <= 0) + if (pm.buffer == nullptr) return false; if (pm.type != PipeMaterial::PMT_FRAME) { - printf("PL_RTSPServer::pay only support PMT_FRAME\n"); + LOG_ERROR << "PL_RTSPServer::pay only support PMT_FRAME" << std::endl; return false; } if (in->buffSize > 0) - printf("PL_RTSPServer::pay may lost data size=%u\n", 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; @@ -241,6 +254,7 @@ { RTSPServer_Internal* in = (RTSPServer_Internal*)internal; + pm.type = PipeMaterial::PMT_NONE; pm.buffer = nullptr; pm.buffSize = 0; pm.former = this; -- Gitblit v1.8.0