udpserver.h

UDP的服务器与客户端基类

#ifndef UDP_SERVER_H
#define UDP_SERVER_H

#include "Config.h"

typedef struct _thread_pack
{
	void *obj;
	void *buf;
	struct sockaddr_in addr;
} THREAD_PACK;

class CUdpServer
{
public:
	CUdpServer(short pt);
	virtual ~CUdpServer(void);

	int NonBlockStart(void);
	int BlockStart(void);
	int Stop(void);
	int Restart(void);

	void SetPort(short pt);

	virtual void Engine(void *buf, int len, struct sockaddr_in &dest) = 0;
	__int32_t WriteTo(void *buf, __u_int len, struct sockaddr_in &dest);

private:
	int Init(void);

	static DWORD WINAPI WorkerThread(LPVOID arg);
	static DWORD WINAPI ListenThread(LPVOID arg);

private:
	short port;
	bool  over;
	WSADATA wsa;
	WORD    WinVersion;
	SOCKET  sock;
	HANDLE  listen_thread;

};

#endif