mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-01 17:11:31 +08:00
83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
/*
|
|
* RtpBroadCaster.h
|
|
*
|
|
* Created on: 2016年12月23日
|
|
* Author: xzl
|
|
*/
|
|
|
|
#ifndef SRC_RTSP_RTPBROADCASTER_H_
|
|
#define SRC_RTSP_RTPBROADCASTER_H_
|
|
|
|
#include <arpa/inet.h>
|
|
#include <netinet/in.h>
|
|
#include <mutex>
|
|
#include <memory>
|
|
#include <unordered_set>
|
|
#include <unordered_map>
|
|
#include "config.h"
|
|
#include "RtspMediaSource.h"
|
|
#include "Util/mini.h"
|
|
#include "Network/Socket.h"
|
|
|
|
using namespace std;
|
|
using namespace ZL::Util;
|
|
using namespace ZL::Network;
|
|
|
|
namespace ZL {
|
|
namespace Rtsp {
|
|
|
|
class MultiCastAddressMaker
|
|
{
|
|
public:
|
|
static MultiCastAddressMaker &Instance(){
|
|
static MultiCastAddressMaker instance;
|
|
return instance;
|
|
}
|
|
static bool isMultiCastAddress(uint32_t iAddr){
|
|
static uint32_t addrMin = mINI::Instance()[Config::MultiCast::kAddrMin].as<uint32_t>();
|
|
static uint32_t addrMax = mINI::Instance()[Config::MultiCast::kAddrMax].as<uint32_t>();
|
|
return iAddr >= addrMin && iAddr <= addrMax;
|
|
}
|
|
static string toString(uint32_t iAddr){
|
|
iAddr = htonl(iAddr);
|
|
return ::inet_ntoa((struct in_addr &)(iAddr));
|
|
}
|
|
virtual ~MultiCastAddressMaker(){}
|
|
std::shared_ptr<uint32_t> obtain(uint32_t iTry = 10);
|
|
private:
|
|
MultiCastAddressMaker(){};
|
|
void release(uint32_t iAddr);
|
|
uint32_t m_iAddr = mINI::Instance()[Config::MultiCast::kAddrMin].as<uint32_t>();
|
|
recursive_mutex m_mtx;
|
|
unordered_set<uint32_t> m_setBadAddr;
|
|
};
|
|
class RtpBroadCaster {
|
|
public:
|
|
typedef std::shared_ptr<RtpBroadCaster> Ptr;
|
|
typedef function<void()> onDetach;
|
|
virtual ~RtpBroadCaster();
|
|
static Ptr get(const string &strLocalIp,const string &strApp,const string &strStream);
|
|
void setDetachCB(void *listener,const onDetach &cb);
|
|
uint16_t getPort(int iTrackId);
|
|
string getIP();
|
|
private:
|
|
static recursive_mutex g_mtx;
|
|
static unordered_map<string , weak_ptr<RtpBroadCaster> > g_mapBroadCaster;
|
|
static Ptr make(const string &strLocalIp,const string &strApp,const string &strStream);
|
|
|
|
std::shared_ptr<uint32_t> m_multiAddr;
|
|
recursive_mutex m_mtx;
|
|
unordered_map<void * , onDetach > m_mapDetach;
|
|
RtspMediaSource::RingType::RingReader::Ptr m_pReader;
|
|
Socket::Ptr m_apUdpSock[2];
|
|
struct sockaddr_in m_aPeerUdpAddr[2];
|
|
|
|
RtpBroadCaster(const string &strLocalIp,const string &strApp,const string &strStream);
|
|
|
|
};
|
|
|
|
} /* namespace Rtsp */
|
|
} /* namespace ZL */
|
|
|
|
#endif /* SRC_RTSP_RTPBROADCASTER_H_ */
|