122 lines
3.7 KiB
C++
122 lines
3.7 KiB
C++
/*
|
||
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT-like 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.
|
||
*/
|
||
|
||
#ifndef ZLMEDIAKIT_RTPSERVER_H
|
||
#define ZLMEDIAKIT_RTPSERVER_H
|
||
|
||
#if defined(ENABLE_RTPPROXY)
|
||
#include <memory>
|
||
#include "Network/Socket.h"
|
||
#include "Network/TcpServer.h"
|
||
#include "Network/UdpServer.h"
|
||
#include "RtpSession.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class RtcpHelper;
|
||
|
||
/**
|
||
* RTP服务器，支持UDP/TCP
|
||
* RTP server, supports UDP/TCP
|
||
|
||
* [AUTO-TRANSLATED:03e7daba]
|
||
*/
|
||
class RtpServer : public std::enable_shared_from_this<RtpServer> {
|
||
public:
|
||
using Ptr = std::shared_ptr<RtpServer>;
|
||
using onRecv = std::function<void(const toolkit::Buffer::Ptr &buf)>;
|
||
enum TcpMode { NONE = 0, PASSIVE, ACTIVE };
|
||
|
||
~RtpServer();
|
||
|
||
/**
|
||
* 开启服务器，可能抛异常
|
||
* @param local_port 本地端口，0时为随机端口
|
||
* @param local_ip 绑定的本地网卡ip
|
||
* @param stream_id 流id，置空则使用ssrc
|
||
* @param tcp_mode tcp服务模式
|
||
* @param re_use_port 是否设置socket为re_use属性
|
||
* @param ssrc 指定的ssrc
|
||
* @param multiplex 多路复用
|
||
* Start the server, may throw an exception
|
||
* @param local_port Local port, 0 for random port
|
||
* @param local_ip Local network interface ip to bind
|
||
* @param stream_id Stream id, use ssrc if empty
|
||
* @param tcp_mode TCP service mode
|
||
* @param re_use_port Whether to set the socket to re_use property
|
||
* @param ssrc Specified ssrc
|
||
* @param multiplex Multiplexing
|
||
|
||
* [AUTO-TRANSLATED:cb9e0717]
|
||
*/
|
||
void start(uint16_t local_port, const char *local_ip = "::", const MediaTuple &tuple = MediaTuple{DEFAULT_VHOST, kRtpAppName, "", ""}, TcpMode tcp_mode = PASSIVE,
|
||
bool re_use_port = true, uint32_t ssrc = 0, int only_track = 0, bool multiplex = false);
|
||
|
||
/**
|
||
* 连接到tcp服务(tcp主动模式)
|
||
* @param url 服务器地址
|
||
* @param port 服务器端口
|
||
* @param cb 连接服务器是否成功的回调
|
||
* Connect to the tcp service (tcp active mode)
|
||
* @param url Server address
|
||
* @param port Server port
|
||
* @param cb Callback whether the connection to the server is successful
|
||
|
||
* [AUTO-TRANSLATED:3de57bc0]
|
||
*/
|
||
void connectToServer(const std::string &url, uint16_t port, const std::function<void(const toolkit::SockException &ex)> &cb);
|
||
|
||
/**
|
||
* 获取绑定的本地端口
|
||
* Get the bound local port
|
||
|
||
* [AUTO-TRANSLATED:beed15d6]
|
||
*/
|
||
uint16_t getPort();
|
||
|
||
/**
|
||
* 设置RtpProcess onDetach事件回调
|
||
* Set RtpProcess onDetach event callback
|
||
|
||
* [AUTO-TRANSLATED:a316d07e]
|
||
*/
|
||
void setOnDetach(RtpProcess::onDetachCB cb);
|
||
|
||
/**
|
||
* 更新ssrc
|
||
* Update ssrc
|
||
|
||
* [AUTO-TRANSLATED:dae5e276]
|
||
*/
|
||
void updateSSRC(uint32_t ssrc);
|
||
|
||
private:
|
||
// tcp主动模式连接服务器成功回调 [AUTO-TRANSLATED:0775844e]
|
||
// tcp active mode connection server success callback
|
||
void onConnect();
|
||
|
||
protected:
|
||
toolkit::Socket::Ptr _rtp_socket;
|
||
toolkit::UdpServer::Ptr _udp_server;
|
||
toolkit::TcpServer::Ptr _tcp_server;
|
||
std::shared_ptr<uint32_t> _ssrc;
|
||
std::shared_ptr<RtcpHelper> _rtcp_helper;
|
||
std::function<void()> _on_cleanup;
|
||
|
||
int _only_track = 0;
|
||
// 用于tcp主动模式 [AUTO-TRANSLATED:faedf05c]
|
||
// Used for tcp active mode
|
||
TcpMode _tcp_mode = NONE;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif//defined(ENABLE_RTPPROXY)
|
||
#endif //ZLMEDIAKIT_RTPSERVER_H
|