230 lines
7.6 KiB
C++
230 lines
7.6 KiB
C++
/* ---------------------------------------------------------------------------
|
|
**
|
|
** H264_DeviceSource.cpp
|
|
**
|
|
** H264 Live555 source
|
|
**
|
|
** -------------------------------------------------------------------------*/
|
|
|
|
#include "zm_rtsp_server_h264_device_source.h"
|
|
|
|
#include "zm_config.h"
|
|
#include "zm_logger.h"
|
|
#include "zm_rtsp_server_frame.h"
|
|
#include <iomanip>
|
|
#include <sstream>
|
|
|
|
#if HAVE_RTSP_SERVER
|
|
// live555
|
|
#include <Base64.hh>
|
|
|
|
// ---------------------------------
|
|
// H264 ZoneMinder FramedSource
|
|
// ---------------------------------
|
|
//
|
|
H264_ZoneMinderDeviceSource::H264_ZoneMinderDeviceSource(
|
|
UsageEnvironment& env,
|
|
std::shared_ptr<Monitor> monitor,
|
|
AVStream *stream,
|
|
unsigned int queueSize,
|
|
bool repeatConfig,
|
|
bool keepMarker)
|
|
: H26X_ZoneMinderDeviceSource(env, std::move(monitor), stream, queueSize, repeatConfig, keepMarker)
|
|
{
|
|
// extradata appears to simply be the SPS and PPS NAL's
|
|
this->splitFrames(m_stream->codecpar->extradata, m_stream->codecpar->extradata_size);
|
|
}
|
|
|
|
// split packet into frames
|
|
std::list< std::pair<unsigned char*, size_t> > H264_ZoneMinderDeviceSource::splitFrames(unsigned char* frame, unsigned frameSize) {
|
|
std::list< std::pair<unsigned char*, size_t> > frameList;
|
|
|
|
size_t bufSize = frameSize;
|
|
size_t size = 0;
|
|
unsigned char* buffer = this->extractFrame(frame, bufSize, size);
|
|
while ( buffer != nullptr ) {
|
|
switch ( m_frameType & 0x1F ) {
|
|
case 7:
|
|
Debug(1, "SPS_Size: %d bufSize %d", size, bufSize);
|
|
m_sps.assign((char*)buffer, size);
|
|
break;
|
|
case 8:
|
|
Debug(1, "PPS_Size: %d bufSize %d", size, bufSize);
|
|
m_pps.assign((char*)buffer, size);
|
|
break;
|
|
case 5:
|
|
Debug(1, "IDR_Size: %d bufSize %d", size, bufSize);
|
|
if ( m_repeatConfig && !m_sps.empty() && !m_pps.empty() ) {
|
|
frameList.push_back(std::pair<unsigned char*,size_t>((unsigned char*)m_sps.c_str(), m_sps.size()));
|
|
frameList.push_back(std::pair<unsigned char*,size_t>((unsigned char*)m_pps.c_str(), m_pps.size()));
|
|
}
|
|
break;
|
|
default:
|
|
Debug(1, "Unknown frametype!? %d %d", m_frameType, m_frameType & 0x1F);
|
|
break;
|
|
}
|
|
|
|
if ( !m_sps.empty() && !m_pps.empty() ) {
|
|
u_int32_t profile_level_id = 0;
|
|
if ( m_sps.size() >= 4 ) profile_level_id = (m_sps[1]<<16)|(m_sps[2]<<8)|m_sps[3];
|
|
|
|
char* sps_base64 = base64Encode(m_sps.c_str(), m_sps.size());
|
|
char* pps_base64 = base64Encode(m_pps.c_str(), m_pps.size());
|
|
|
|
std::ostringstream os;
|
|
os << "profile-level-id=" << std::hex << std::setw(6) << std::setfill('0') << profile_level_id;
|
|
os << ";sprop-parameter-sets=" << sps_base64 << "," << pps_base64;
|
|
m_auxLine.assign(os.str());
|
|
Debug(1, "auxLine: %s", m_auxLine.c_str());
|
|
|
|
delete [] sps_base64;
|
|
delete [] pps_base64;
|
|
}
|
|
frameList.push_back(std::pair<unsigned char*,size_t>(buffer, size));
|
|
|
|
buffer = this->extractFrame(&buffer[size], bufSize, size);
|
|
} // end while buffer
|
|
return frameList;
|
|
}
|
|
|
|
H265_ZoneMinderDeviceSource::H265_ZoneMinderDeviceSource(
|
|
UsageEnvironment& env,
|
|
std::shared_ptr<Monitor> monitor,
|
|
AVStream *stream,
|
|
unsigned int queueSize,
|
|
bool repeatConfig,
|
|
bool keepMarker)
|
|
: H26X_ZoneMinderDeviceSource(env, std::move(monitor), stream, queueSize, repeatConfig, keepMarker)
|
|
{
|
|
// extradata appears to simply be the SPS and PPS NAL's
|
|
this->splitFrames(m_stream->codecpar->extradata, m_stream->codecpar->extradata_size);
|
|
}
|
|
|
|
// split packet in frames
|
|
std::list< std::pair<unsigned char*,size_t> >
|
|
H265_ZoneMinderDeviceSource::splitFrames(unsigned char* frame, unsigned frameSize) {
|
|
std::list< std::pair<unsigned char*,size_t> > frameList;
|
|
|
|
size_t bufSize = frameSize;
|
|
size_t size = 0;
|
|
unsigned char* buffer = this->extractFrame(frame, bufSize, size);
|
|
while ( buffer != nullptr ) {
|
|
switch ((m_frameType&0x7E)>>1) {
|
|
case 32:
|
|
Debug(1, "VPS_Size: %d bufSize %d", size, bufSize);
|
|
m_vps.assign((char*)buffer,size);
|
|
break;
|
|
case 33:
|
|
Debug(1, "SPS_Size: %d bufSize %d", size, bufSize);
|
|
m_sps.assign((char*)buffer,size);
|
|
break;
|
|
case 34:
|
|
Debug(1, "PPS_Size: %d bufSize %d", size, bufSize);
|
|
m_pps.assign((char*)buffer,size);
|
|
break;
|
|
case 19:
|
|
case 20:
|
|
Debug(1, "IDR_Size: %d bufSize %d", size, bufSize);
|
|
if ( m_repeatConfig && !m_vps.empty() && !m_sps.empty() && !m_pps.empty() ) {
|
|
frameList.push_back(std::pair<unsigned char*,size_t>((unsigned char*)m_vps.c_str(), m_vps.size()));
|
|
frameList.push_back(std::pair<unsigned char*,size_t>((unsigned char*)m_sps.c_str(), m_sps.size()));
|
|
frameList.push_back(std::pair<unsigned char*,size_t>((unsigned char*)m_pps.c_str(), m_pps.size()));
|
|
}
|
|
break;
|
|
default:
|
|
Debug(1, "Unknown frametype!? %d %d", m_frameType, ((m_frameType & 0x7E) >> 1));
|
|
break;
|
|
}
|
|
|
|
if ( !m_vps.empty() && !m_sps.empty() && !m_pps.empty() ) {
|
|
char* vps_base64 = base64Encode(m_vps.c_str(), m_vps.size());
|
|
char* sps_base64 = base64Encode(m_sps.c_str(), m_sps.size());
|
|
char* pps_base64 = base64Encode(m_pps.c_str(), m_pps.size());
|
|
|
|
std::ostringstream os;
|
|
os << "sprop-vps=" << vps_base64;
|
|
os << ";sprop-sps=" << sps_base64;
|
|
os << ";sprop-pps=" << pps_base64;
|
|
m_auxLine.assign(os.str());
|
|
Debug(1, "Assigned auxLine to %s", m_auxLine.c_str());
|
|
|
|
delete [] vps_base64;
|
|
delete [] sps_base64;
|
|
delete [] pps_base64;
|
|
}
|
|
frameList.push_back(std::pair<unsigned char*,size_t>(buffer, size));
|
|
|
|
buffer = this->extractFrame(&buffer[size], bufSize, size);
|
|
} // end while buffer
|
|
if ( bufSize ) {
|
|
Debug(1, "%d bytes remaining", bufSize);
|
|
}
|
|
return frameList;
|
|
} // end H265_ZoneMinderDeviceSource::splitFrames(unsigned char* frame, unsigned frameSize)
|
|
|
|
unsigned char * H26X_ZoneMinderDeviceSource::findMarker(
|
|
unsigned char *frame, size_t size, size_t &length
|
|
) {
|
|
//Debug(1, "findMarker %p %d", frame, size);
|
|
unsigned char *start = nullptr;
|
|
for ( size_t i = 0; i < size-2; i += 1 ) {
|
|
//Debug(1, "%d: %d %d %d", i, frame[i], frame[i+1], frame[i+2]);
|
|
if ( (frame[i] == 0) and (frame[i+1]) == 0 and (frame[i+2] == 1) ) {
|
|
if ( i and (frame[i-1] == 0) ) {
|
|
start = frame + i - 1;
|
|
length = sizeof(H264marker);
|
|
} else {
|
|
start = frame + i;
|
|
length = sizeof(H264shortmarker);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
return start;
|
|
}
|
|
|
|
// extract a frame
|
|
unsigned char* H26X_ZoneMinderDeviceSource::extractFrame(unsigned char* frame, size_t& size, size_t& outsize) {
|
|
unsigned char *outFrame = nullptr;
|
|
Debug(1, "ExtractFrame: %p %d", frame, size);
|
|
outsize = 0;
|
|
size_t markerLength = 0;
|
|
size_t endMarkerLength = 0;
|
|
m_frameType = 0;
|
|
unsigned char *startFrame = nullptr;
|
|
if ( size >= 3 )
|
|
startFrame = this->findMarker(frame, size, markerLength);
|
|
if ( startFrame != nullptr ) {
|
|
Debug(1, "startFrame: %p marker Length %d", startFrame, markerLength);
|
|
m_frameType = startFrame[markerLength];
|
|
|
|
int remainingSize = size-(startFrame-frame+markerLength);
|
|
unsigned char *endFrame = nullptr;
|
|
if ( remainingSize > 3 ) {
|
|
endFrame = this->findMarker(startFrame+markerLength, remainingSize, endMarkerLength);
|
|
}
|
|
Debug(1, "endFrame: %p marker Length %d, remaining size %d", endFrame, endMarkerLength, remainingSize);
|
|
|
|
if ( m_keepMarker ) {
|
|
size -= startFrame-frame;
|
|
outFrame = startFrame;
|
|
} else {
|
|
size -= startFrame-frame+markerLength;
|
|
outFrame = &startFrame[markerLength];
|
|
}
|
|
|
|
if ( endFrame != nullptr ) {
|
|
outsize = endFrame - outFrame;
|
|
} else {
|
|
outsize = size;
|
|
}
|
|
size -= outsize;
|
|
Debug(1, "Have frame type: %d size %d, keepmarker %d", m_frameType, outsize, m_keepMarker);
|
|
} else if ( size >= sizeof(H264shortmarker) ) {
|
|
Info("No marker found");
|
|
}
|
|
|
|
return outFrame;
|
|
}
|
|
#endif // HAVE_RTSP_SERVER
|