first commit for srt intergrate

2026-07-06 19:58:10 +08:00 · 2022-06-03 13:25:32 +08:00
parent 59dcd03b70
commit aa2ff01d9f
18 changed files with 2484 additions and 0 deletions
--- a/srt/SrtTransport.hpp
+++ b/srt/SrtTransport.hpp
@@ -0,0 +1,143 @@
+#ifndef ZLMEDIAKIT_SRT_TRANSPORT_H
+#define ZLMEDIAKIT_SRT_TRANSPORT_H
+
+#include <mutex>
+#include <chrono>
+#include <memory>
+#include <atomic>
+
+#include "Network/Session.h"
+#include "Poller/EventPoller.h"
+#include "Util/TimeTicker.h"
+
+#include "Common.hpp"
+#include "Packet.hpp"
+#include "PacketQueue.hpp"
+#include "Statistic.hpp"
+
+namespace SRT {
+using namespace toolkit;
+
+extern const std::string kPort;
+extern const std::string kTimeOutSec;
+
+class SrtTransport : public std::enable_shared_from_this<SrtTransport> {
+public:
+    friend class SrtSession;
+    using Ptr = std::shared_ptr<SrtTransport>;
+
+    SrtTransport(const EventPoller::Ptr &poller);
+    virtual ~SrtTransport();
+    const EventPoller::Ptr &getPoller() const;
+    void setSession(Session::Ptr session);
+    const Session::Ptr &getSession() const;
+    /**
+     * socket收到udp数据
+     * @param buf 数据指针
+     * @param len 数据长度
+     * @param addr 数据来源地址
+     */
+    void inputSockData(uint8_t *buf, int len, struct sockaddr_storage *addr);
+
+    std::string getIdentifier();
+    
+    void unregisterSelfHandshake();
+    void unregisterSelf();
+protected:
+    virtual void onHandShakeFinished(std::string& streamid){};
+    virtual void onSRTData(DataPacket::Ptr pkt){};
+    virtual void onShutdown(const SockException &ex);
+
+private:
+    void registerSelfHandshake();
+    void registerSelf();
+
+    void switchToOtherTransport(uint8_t *buf, int len,uint32_t socketid, struct sockaddr_storage *addr);
+
+    void handleHandshake(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleHandshakeInduction(HandshakePacket& pkt,struct sockaddr_storage *addr);
+    void handleHandshakeConclusion(HandshakePacket& pkt,struct sockaddr_storage *addr);
+
+    void handleKeeplive(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleACK(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleACKACK(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleNAK(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleCongestionWarning(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleShutDown(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleDropReq(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleUserDefinedType(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handlePeerError(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    void handleDataPacket(uint8_t *buf, int len, struct sockaddr_storage *addr);
+    
+    void sendNAKPacket(std::list<PacketQueue::LostPair>& lost_list);
+    void sendACKPacket();
+    void sendLightACKPacket();
+protected:
+    void sendDataPacket(DataPacket::Ptr pkt,char* buf,int len,bool flush = false);
+    void sendControlPacket(ControlPacket::Ptr pkt,bool  flush = true);
+    void sendPacket(Buffer::Ptr pkt,bool flush =  true);
+private:
+    //当前选中的udp链接
+    Session::Ptr _selected_session;
+    //链接迁移前后使用过的udp链接
+    std::unordered_map<Session *, std::weak_ptr<Session> > _history_sessions;
+
+    EventPoller::Ptr _poller;
+
+    uint32_t _peer_socket_id;
+    uint32_t _socket_id = 0;
+
+    TimePoint _start_timestamp;
+
+    uint32_t _mtu = 1500;
+    uint32_t _max_window_size = 8192;
+    uint32_t  _init_seq_number = 0;
+
+    std::string _stream_id;
+    uint32_t _sync_cookie = 0;
+
+    PacketQueue::Ptr _recv_buf;
+    uint32_t _rtt = 100*1000;
+    uint32_t _rtt_variance =50*1000;
+    uint32_t _light_ack_pkt_count = 0;
+    uint32_t _ack_number_count = 0;
+    Ticker _ack_ticker;
+    std::map<uint32_t,TimePoint> _ack_send_timestamp;
+
+    PacketRecvRateContext _pkt_recv_rate_context;
+    EstimatedLinkCapacityContext _estimated_link_capacity_context;
+    RecvRateContext _recv_rate_context;
+
+    Ticker _nak_ticker;
+
+    //保持发送的握手消息，防止丢失重发
+    HandshakePacket::Ptr _handleshake_res;
+
+    ResourcePool<BufferRaw> _packet_pool;
+
+};
+
+class SrtTransportManager {
+public:
+    static SrtTransportManager &Instance();
+    SrtTransport::Ptr getItem(const std::string &key);
+    void addItem(const std::string &key, const SrtTransport::Ptr &ptr);
+    void removeItem(const std::string &key);
+
+    void addHandshakeItem(const std::string &key, const SrtTransport::Ptr &ptr);
+    void removeHandshakeItem(const std::string &key);
+    SrtTransport::Ptr getHandshakeItem(const std::string &key);
+private:
+    SrtTransportManager() = default;
+
+private:
+    std::mutex _mtx;
+    std::unordered_map<std::string, std::weak_ptr<SrtTransport>> _map;
+
+    std::mutex _handshake_mtx;
+    std::unordered_map<std::string, std::weak_ptr<SrtTransport>> _handshake_map;
+};
+
+} // namespace SRT
+
+#endif // ZLMEDIAKIT_SRT_TRANSPORT_H