58 lines
1.3 KiB
C++
58 lines
1.3 KiB
C++
#pragma once
|
|
#include <string>
|
|
|
|
#ifndef _WIN32_WINNT
|
|
#define _WIN32_WINNT 0x0500
|
|
#include <WinSock2.h>
|
|
#endif
|
|
|
|
#ifdef linux
|
|
#include<sys/types.h>
|
|
#include<sys/socket.h>
|
|
#include<arpa/inet.h>
|
|
#define EVENT__HAVE_PTHREADS
|
|
#endif
|
|
|
|
#include <stdint.h>
|
|
#include <iostream>
|
|
#include <mutex>
|
|
#include <thread>
|
|
#include <map>
|
|
#include <functional>
|
|
#include <functional>
|
|
|
|
|
|
class UdpDataGramLibevent {
|
|
public:
|
|
typedef std::function<void(const char* dat, int len, struct sockaddr_in)> OnReadDataHandle;
|
|
UdpDataGramLibevent(std::string ip, uint32_t port);
|
|
UdpDataGramLibevent(std::string ip, uint32_t port, std::string group_addr);
|
|
|
|
OnReadDataHandle OnReadHandle();
|
|
typedef enum {
|
|
RUNNING,
|
|
STOP,
|
|
FAIL
|
|
}STATUS;
|
|
friend void read_cb(int, short, void*);
|
|
void SetOnReadHandle(OnReadDataHandle);
|
|
void SendTo(const char* dat, uint32_t len, std::string ip, int port);
|
|
int SocketFD();
|
|
private:
|
|
int bind_socket(struct event* ev, const char* ip, uint16_t port);
|
|
int bind_socket_with_group(struct event* ev, const char* ip, uint16_t port, const char* udp_group);
|
|
|
|
uint32_t m_port; //
|
|
std::string m_bind_ip; //
|
|
uint16_t m_backlog;
|
|
int m_sock_fd;
|
|
struct sockaddr_in m_bind_addr; //
|
|
struct event* m_event;
|
|
STATUS m_status; //
|
|
std::thread* m_thread; //
|
|
intptr_t mSocketFD; // socket
|
|
OnReadDataHandle mOnRead;
|
|
};
|
|
|
|
|