mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-23 11:17:09 +08:00
96 lines
3.1 KiB
C++
96 lines
3.1 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#if defined(ENABLE_RTPPROXY)
|
||
#include "RtpSession.h"
|
||
#include "RtpSelector.h"
|
||
#include "Network/TcpServer.h"
|
||
namespace mediakit{
|
||
|
||
const string RtpSession::kStreamID = "stream_id";
|
||
|
||
void RtpSession::attachServer(const TcpServer &server) {
|
||
_stream_id = const_cast<TcpServer &>(server)[kStreamID];
|
||
}
|
||
|
||
RtpSession::RtpSession(const Socket::Ptr &sock) : TcpSession(sock) {
|
||
DebugP(this);
|
||
socklen_t addr_len = sizeof(addr);
|
||
getpeername(sock->rawFD(), &addr, &addr_len);
|
||
}
|
||
RtpSession::~RtpSession() {
|
||
DebugP(this);
|
||
if(_process){
|
||
RtpSelector::Instance().delProcess(_stream_id,_process.get());
|
||
}
|
||
}
|
||
|
||
void RtpSession::onRecv(const Buffer::Ptr &data) {
|
||
try {
|
||
RtpSplitter::input(data->data(), data->size());
|
||
} catch (SockException &ex) {
|
||
shutdown(ex);
|
||
} catch (std::exception &ex) {
|
||
shutdown(SockException(Err_other, ex.what()));
|
||
}
|
||
}
|
||
|
||
void RtpSession::onError(const SockException &err) {
|
||
WarnL << _stream_id << " " << err.what();
|
||
}
|
||
|
||
void RtpSession::onManager() {
|
||
if(_process && !_process->alive()){
|
||
shutdown(SockException(Err_timeout, "receive rtp timeout"));
|
||
}
|
||
|
||
if(!_process && _ticker.createdTime() > 10 * 1000){
|
||
shutdown(SockException(Err_timeout, "illegal connection"));
|
||
}
|
||
}
|
||
|
||
void RtpSession::onRtpPacket(const char *data, uint64_t len) {
|
||
if (len > 1024 * 10) {
|
||
throw SockException(Err_shutdown, StrPrinter << "rtp包长度异常(" << len << ")，发送端可能缓存溢出并覆盖");
|
||
}
|
||
if (!_process) {
|
||
uint32_t ssrc;
|
||
if (!RtpSelector::getSSRC(data, len, ssrc)) {
|
||
return;
|
||
}
|
||
if (_stream_id.empty()) {
|
||
//未指定流id就使用ssrc为流id
|
||
_stream_id = printSSRC(ssrc);
|
||
}
|
||
//tcp情况下，一个tcp链接只可能是一路流，不需要通过多个ssrc来区分，所以不需要频繁getProcess
|
||
_process = RtpSelector::Instance().getProcess(_stream_id, true);
|
||
_process->setListener(dynamic_pointer_cast<RtpSession>(shared_from_this()));
|
||
}
|
||
_process->inputRtp(false, getSock(), data, len, &addr);
|
||
_ticker.resetTime();
|
||
}
|
||
|
||
bool RtpSession::close(MediaSource &sender, bool force) {
|
||
//此回调在其他线程触发
|
||
if(!_process || (!force && _process->totalReaderCount())){
|
||
return false;
|
||
}
|
||
string err = StrPrinter << "close media:" << sender.getSchema() << "/" << sender.getVhost() << "/" << sender.getApp() << "/" << sender.getId() << " " << force;
|
||
safeShutdown(SockException(Err_shutdown,err));
|
||
return true;
|
||
}
|
||
|
||
int RtpSession::totalReaderCount(MediaSource &sender) {
|
||
//此回调在其他线程触发
|
||
return _process ? _process->totalReaderCount() : sender.totalReaderCount();
|
||
}
|
||
|
||
}//namespace mediakit
|
||
#endif//defined(ENABLE_RTPPROXY)
|