46 lines
1.6 KiB
C++
46 lines
1.6 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 RTSP_UDPSERVER_H_
|
|
#define RTSP_UDPSERVER_H_
|
|
|
|
#include <stdint.h>
|
|
#include <mutex>
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#include "Network/Socket.h"
|
|
|
|
namespace mediakit {
|
|
|
|
class UDPServer : public std::enable_shared_from_this<UDPServer> {
|
|
public:
|
|
using onRecvData = std::function<bool(int intervaled, const toolkit::Buffer::Ptr &buffer, struct sockaddr *peer_addr)> ;
|
|
~UDPServer();
|
|
static UDPServer &Instance();
|
|
toolkit::Socket::Ptr getSock(toolkit::SocketHelper &helper, const char *local_ip, int interleaved, uint16_t local_port = 0);
|
|
void listenPeer(const char *peer_ip, void *obj, const onRecvData &cb);
|
|
void stopListenPeer(const char *peer_ip, void *obj);
|
|
|
|
private:
|
|
UDPServer();
|
|
void onRecv(int interleaved, const toolkit::Buffer::Ptr &buf, struct sockaddr *peer_addr);
|
|
void onErr(const std::string &strKey, const toolkit::SockException &err);
|
|
|
|
private:
|
|
std::mutex _mtx_udp_sock;
|
|
std::mutex _mtx_on_recv;
|
|
std::unordered_map<std::string, toolkit::Socket::Ptr> _udp_sock_map;
|
|
std::unordered_map<std::string, std::unordered_map<void *, onRecvData> > _on_recv_map;
|
|
};
|
|
|
|
} /* namespace mediakit */
|
|
|
|
#endif /* RTSP_UDPSERVER_H_ */
|