void HeadersTest::caseInsensitivity()
{
QFETCH(QByteArray, qbytearray1);
QFETCH(QByteArray, qbytearray2);
const IByteArray ibytearray1 = IByteArray{qbytearray1.constData()};
const IByteArray ibytearray2 = qbytearray2;
Headers headers;
headers.insert(ibytearray1, qbytearray2);
headers.insert(ibytearray2, qbytearray2);
QCOMPARE(headers.count(ibytearray1), 2);
QCOMPARE(headers.count(ibytearray2), 2);
QVERIFY(headers.values(ibytearray1)[0] == headers.values(ibytearray2)[1]);
/* ====================================================================== */
headers.clear();
headers.insert(ibytearray1, qbytearray2);
headers.replace(ibytearray1, qbytearray1);
headers.insert(ibytearray2, qbytearray2);
QCOMPARE(headers.count(ibytearray1), 2);
QCOMPARE(headers.count(ibytearray2), 2);
QVERIFY(headers.values(ibytearray1)[0] != headers.values(ibytearray2)[1]);
}
void QPalringoConnection::onPingReceived( const Headers, const QByteArray /* body */ )
{
Headers h;
if( protocolVersion_ == 2 )
{
h.insert( qpHeaderAttribute::PS, packetSeq_ );
}
sendCmd( qpCommand::PING, h, "");
}
/**
* @brief
*
* @param headers
* @param payload
* @param statusCode
* @param statusText
*
* @return
*/
bool RtspConnection::sendResponse( Headers &headers, std::string payload, int statusCode, std::string statusText )
{
std::string response;
response = stringtf( "RTSP/1.0 %d %s\r\n", statusCode, statusText.c_str() );
//headers.insert( Headers::value_type( "Server", "ZoneMinder Streamer V2.0" ) );
headers.insert( Headers::value_type( "Date", datetimeInet() ) );
for ( Headers::iterator iter = headers.begin(); iter != headers.end(); iter++ )
{
response += iter->first+": "+iter->second+"\r\n";
}
response += "\r\n";
if ( payload.size() )
{
response += payload;
}
Debug( 2, "Sending RTSP response: %s", response.c_str() );
if ( mSocket->send( response.c_str(), response.size() ) != (int)response.length() )
{
Error( "Unable to send response '%s': %s", response.c_str(), strerror(errno) );
return( false );
}
#if 0
// Not currently used. If RTSP over HTTP ever required then will need to do something like this
if ( mMethod == RTP_OZ_RTSP_HTTP )
{
response = base64Encode( response );
Debug( 2, "Sending encoded RTSP response: %s", response.c_str() );
if ( mRtspSocket2.send( response.c_str(), response.size() ) != (int)response.length() )
{
Error( "Unable to send response '%s': %s", response.c_str(), strerror(errno) );
return( false );
}
}
else
{
if ( mRtspSocket.send( response.c_str(), response.size() ) != (int)response.length() )
{
Error( "Unable to send response '%s': %s", response.c_str(), strerror(errno) );
return( false );
}
}
#endif
return( true );
}
void getResponseHeaders(Headers& headers) {
while (true) {
istream line(input.readLine());
Header newHeader;
getline(line, newHeader.first, ':'); // Only read up until the ':'
trim(newHeader.first);
to_lower(newHeader.first);
if (newHeader.first.empty())
break;
getline(line, newHeader.second); // Read the rest of the line
trim(newHeader.second);
headers.insert(newHeader);
}
}
void addHeader( Header const & header )
{
assert( !fromCache() );
includedHeaders_.insert( header );
}
/**
* @brief
*
* @param request
*
* @return
*/
bool RtspConnection::handleRequest( const std::string &request )
{
Debug( 2, "Handling RTSP request: %s (%zd bytes)", request.c_str(), request.size() );
StringTokenList lines( request, "\r\n" );
if ( lines.size() <= 0 )
{
Error( "Unable to split request '%s' into tokens", request.c_str() );
return( false );
}
StringTokenList parts( lines[0], " " );
if ( parts.size() != 3 )
{
Error( "Unable to split request part '%s' into tokens", lines[0].c_str() );
return( false );
}
std::string requestType = parts[0];
Debug( 4, "Got request '%s'", requestType.c_str() );
std::string requestUrl = parts[1];
Debug( 4, "Got requestUrl '%s'", requestUrl.c_str() );
std::string requestVer = parts[2];
Debug( 4, "Got requestVer '%s'", requestVer.c_str() );
if ( requestVer != "RTSP/1.0" )
{
Error( "Unexpected RTSP version '%s'", requestVer.c_str() );
return( false );
}
// Extract headers from request
Headers requestHeaders;
for ( int i = 1; i < lines.size(); i++ )
{
StringTokenList parts( lines[i], ": " );
if ( parts.size() != 2 )
{
Error( "Unable to split request header '%s' into tokens", lines[i].c_str() );
return( false );
}
Debug( 4, "Got header '%s', value '%s'", parts[0].c_str(), parts[1].c_str() );
requestHeaders.insert( Headers::value_type( parts[0], parts[1] ) );
}
if ( requestHeaders.find("CSeq") == requestHeaders.end() )
{
Error( "No CSeq header found" );
return( false );
}
Debug( 4, "Got sequence number %s", requestHeaders["CSeq"].c_str() );
uint32_t session = 0;
if ( requestHeaders.find("Session") != requestHeaders.end() )
{
Debug( 4, "Got session header, '%s', passing to session", requestHeaders["Session"].c_str() );
session = strtol( requestHeaders["Session"].c_str(), NULL, 16 );
}
Headers responseHeaders;
responseHeaders.insert( Headers::value_type( "CSeq", requestHeaders["CSeq"] ) );
if ( requestType == "OPTIONS" )
{
responseHeaders.insert( Headers::value_type( "Public", "DESCRIBE, SETUP, PLAY, GET_PARAMETER, TEARDOWN" ) );
return( sendResponse( responseHeaders ) );
}
else if ( requestType == "DESCRIBE" )
{
FeedProvider *provider = validateRequestUrl( requestUrl );
if ( !provider )
{
sendResponse( responseHeaders, "", 404, "Not Found" );
return( false );
}
const VideoProvider *videoProvider = dynamic_cast<const VideoProvider *>(provider);
int codec = AV_CODEC_ID_MPEG4;
int width = videoProvider->width();
int height = videoProvider->height();
FrameRate frameRate = 15;
//FrameRate frameRate = videoProvider->frameRate();
int bitRate = 90000;
int quality = 70;
std::string sdpFormatString =
"v=0\r\n"
"o=- %jd %jd IN IP4 %s\r\n"
"s=ZoneMinder Stream\r\n"
"i=Media Streamers\r\n"
"c=IN IP4 0.0.0.0\r\n"
"t=0 0\r\n"
"a=control:*\r\n"
"a=range:npt=0.000000-\r\n";
uint64_t now64 = time64();
char hostname[HOST_NAME_MAX] = "";
if ( gethostname( hostname, sizeof(hostname) ) < 0 )
Fatal( "Can't gethostname: %s", strerror(errno) );
std::string sdpString = stringtf( sdpFormatString, now64, now64, hostname );
if ( codec == AV_CODEC_ID_H264 )
{
if ( provider->cl4ss() == "RawH264Input" )
{
std::string encoderKey = H264Relay::getPoolKey( provider->identity(), width, height, frameRate, bitRate, quality );
if ( !(mEncoder = Encoder::getPooledEncoder( encoderKey )) )
{
H264Relay *h264Relay = NULL;
mEncoder = h264Relay = new H264Relay( provider->identity(), width, height, frameRate, bitRate, quality );
mEncoder->registerProvider( *provider );
Encoder::poolEncoder( mEncoder );
h264Relay->start();
}
sdpString += mEncoder->sdpString( 1 ); // XXX - Should be variable
responseHeaders.insert( Headers::value_type( "Content-length", stringtf( "%zd", sdpString.length() ) ) );
}
else
{
std::string encoderKey = H264Encoder::getPoolKey( provider->identity(), width, height, frameRate, bitRate, quality );
if ( !(mEncoder = Encoder::getPooledEncoder( encoderKey )) )
{
H264Encoder *h264Encoder = NULL;
mEncoder = h264Encoder = new H264Encoder( provider->identity(), width, height, frameRate, bitRate, quality );
mEncoder->registerProvider( *provider );
Encoder::poolEncoder( mEncoder );
h264Encoder->start();
}
sdpString += mEncoder->sdpString( 1 ); // XXX - Should be variable
responseHeaders.insert( Headers::value_type( "Content-length", stringtf( "%zd", sdpString.length() ) ) );
}
}
else if ( codec == AV_CODEC_ID_MPEG4 )
{
std::string encoderKey = MpegEncoder::getPoolKey( provider->identity(), width, height, frameRate, bitRate, quality );
if ( !(mEncoder = Encoder::getPooledEncoder( encoderKey )) )
{
MpegEncoder *mpegEncoder = NULL;
mEncoder = mpegEncoder = new MpegEncoder( provider->identity(), width, height, frameRate, bitRate, quality );
mEncoder->registerProvider( *provider );
Encoder::poolEncoder( mEncoder );
mpegEncoder->start();
}
sdpString += mEncoder->sdpString( 1 ); // XXX - Should be variable
responseHeaders.insert( Headers::value_type( "Content-length", stringtf( "%zd", sdpString.length() ) ) );
}
return( sendResponse( responseHeaders, sdpString ) );
}
else if ( requestType == "SETUP" )
{
// These commands are handled by RTSP session so pass them on and send any required responses
RtspSession *rtspSession = 0;
if ( session )
{
rtspSession = mRtspController->getSession( session );
}
else
{
rtspSession = mRtspController->newSession( this, mEncoder );
}
if ( rtspSession->recvRequest( requestType, requestUrl, requestHeaders, responseHeaders ) )
return( sendResponse( responseHeaders ) );
return( false );
}
else if ( requestType == "PLAY" || requestType == "GET_PARAMETER" || requestType == "TEARDOWN" )
{
// These commands are handled by RTSP session so pass them on and send any required responses
RtspSession *rtspSession = 0;
if ( session )
{
rtspSession = mRtspController->getSession( session );
if ( rtspSession && rtspSession->recvRequest( requestType, requestUrl, requestHeaders, responseHeaders ) )
return( sendResponse( responseHeaders ) );
}
return( sendResponse( responseHeaders, "", 454, "Session Not Found" ) );
}
Error( "Unrecognised RTSP command '%s'", requestType.c_str() );
return( sendResponse( responseHeaders, "", 405, "Method not implemented" ) );
}