67 lines
2.0 KiB
C++
67 lines
2.0 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 SRC_RTSP_RTPBROADCASTER_H_
|
|
#define SRC_RTSP_RTPBROADCASTER_H_
|
|
|
|
#include <mutex>
|
|
#include <memory>
|
|
#include <unordered_set>
|
|
#include <unordered_map>
|
|
#include "RtspMediaSource.h"
|
|
#include "Network/Socket.h"
|
|
|
|
namespace mediakit{
|
|
|
|
class MultiCastAddressMaker {
|
|
public:
|
|
static MultiCastAddressMaker& Instance();
|
|
static bool isMultiCastAddress(uint32_t addr);
|
|
static std::string toString(uint32_t addr);
|
|
|
|
std::shared_ptr<uint32_t> obtain(uint32_t max_try = 10);
|
|
|
|
private:
|
|
MultiCastAddressMaker() = default;
|
|
void release(uint32_t addr);
|
|
|
|
private:
|
|
uint32_t _addr = 0;
|
|
std::recursive_mutex _mtx;
|
|
std::unordered_set<uint32_t> _used_addr;
|
|
};
|
|
|
|
class RtpMultiCaster {
|
|
public:
|
|
using Ptr = std::shared_ptr<RtpMultiCaster>;
|
|
using onDetach = std::function<void()>;
|
|
|
|
~RtpMultiCaster();
|
|
|
|
static Ptr get(toolkit::SocketHelper &helper, const std::string &local_ip, const MediaTuple &tuple, uint32_t multicast_ip = 0, uint16_t video_port = 0, uint16_t audio_port = 0);
|
|
void setDetachCB(void *listener,const onDetach &cb);
|
|
|
|
std::string getMultiCasterIP();
|
|
uint16_t getMultiCasterPort(TrackType trackType);
|
|
|
|
private:
|
|
RtpMultiCaster(toolkit::SocketHelper &helper, const std::string &local_ip, const MediaTuple &tuple, uint32_t multicast_ip, uint16_t video_port, uint16_t audio_port);
|
|
|
|
private:
|
|
std::recursive_mutex _mtx;
|
|
toolkit::Socket::Ptr _udp_sock[2];
|
|
std::shared_ptr<uint32_t> _multicast_ip;
|
|
std::unordered_map<void * , onDetach > _detach_map;
|
|
RtspMediaSource::RingType::RingReader::Ptr _rtp_reader;
|
|
};
|
|
|
|
}//namespace mediakit
|
|
#endif /* SRC_RTSP_RTPBROADCASTER_H_ */
|