#pragma warning(disable: 4530)
#pragma warning(disable: 4786)
#include <map>
#include <cassert>
#include <iostream>
#include <fstream>
#include <vector>
#include <string>
#include <algorithm>
#include <exception>
#include <iomanip>
#include <cstdio>
#include <cstdlib>
#include <tchar.h>
#include <ctime>
using namespace std;
#include <winsock2.h>
#include <windows.h>
#include "thread.h" //见网络同步校时TCP服务器端SDK代码末尾
char * GetErrorMessage(char *szBuffer, DWORD dwSize)
{
FormatMessage(
FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
szBuffer,
dwSize - 1,
NULL
);
return szBuffer;
}
class RFC868UDPServer : public Thread
{
long _lWorkCount; //发送时间计数
bool _fRunFlag; //循环运行标志
SOCKET _S37; //socket
u_long _uIP; //ip
public:
RFC868UDPServer() : _S37(INVALID_SOCKET), _lWorkCount(0), _fRunFlag(true)
{
_uIP = htonl(INADDR_ANY);//默认绑定本地IP
}
~RFC868UDPServer()
{
if(_S37 != INVALID_SOCKET)
closesocket(_S37);
}
//结束线程
inline bool End()
{
_fRunFlag = false;
Sleep(100);
return Thread::End();
}
inline bool WaitEnd(DWORD dwWaitTime = INFINITE)
{
_fRunFlag = false;
return !Wait(dwWaitTime) ? Thread::End() : true;
}
//打印信息
inline void LogMessage(char *szStr)
{
Thread::Lock();
cout << szStr << endl;
Thread::UnLock();
}
//打印错误信息
inline void LogError()
{
char Buffer[256];
LogMessage(GetErrorMessage(Buffer, 256));
}
inline long GetWorkCount()
{
return _lWorkCount;
}
//设置绑定的ip
inline void SetIP(char *szIPStr)
{
_uIP = inet_addr(szIPStr);
}
inline void SetIP(u_long ip)
{
_uIP = ip;
}
inline virtual bool Begin()
{
struct sockaddr_in server;
server.sin_family = AF_INET;
server.sin_port = htons(37);
server.sin_addr.s_addr = _uIP;
if((_S37 = socket(AF_INET, SOCK_DGRAM, 0)) == INVALID_SOCKET)
return false;
unsigned long NonBlock = 1;
if(ioctlsocket(_S37, FIONBIO, &NonBlock) == SOCKET_ERROR)
return false;
if(bind(_S37, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR)
return false;
return Thread::Begin();
}
//线程工作函数
virtual void WorkProc() //if error return 1
{
char Buffer[256];
struct sockaddr_in client;
int c_len = sizeof(client), ret;
FD_SET read_set;
timeval tv = {0, 100000}; //tv.tv_sec = 0, tv.tv_usec = 100000;
while(_fRunFlag)
{
FD_ZERO(&read_set);
FD_SET(_S37, &read_set);
//select模型
if(select(0, &read_set, NULL, NULL, &tv) == SOCKET_ERROR)
{
LogMessage(GetErrorMessage(Buffer, 256));
continue;
}
if(FD_ISSET(_S37, &read_set))//等到客户端
{
if(recvfrom(_S37, NULL, 0, 0, (struct sockaddr *)&client, &c_len) != SOCKET_ERROR)
{
time_t cur_time = time(NULL) + 2208988800;
char s_buf[4], *tp = ((char *)&cur_time) + 3;
for(int i=0; i<4; i++)
s_buf[i] = *tp--;
if(sendto(_S37, s_buf, 4, 0, (struct sockaddr *)&client, c_len) == SOCKET_ERROR)
{
LogMessage(GetErrorMessage(Buffer, 256));
continue;
}
sprintf(Buffer, "udp recvfrom client %s (%d) a null datagram \n",
inet_ntoa(client.sin_addr), ntohs(client.sin_port));
LogMessage(Buffer);
InterlockedIncrement(&_lWorkCount);
}
else
LogMessage(GetErrorMessage(Buffer, 256));
}
}
}
};
int main(int argc, char *argv[])
{
try
{
char buf[256];
WORD wVersionRequested;
WSADATA wsaData;
wVersionRequested = MAKEWORD(2, 2);
if(WSAStartup(wVersionRequested, &wsaData))
{
cout << GetErrorMessage(buf, sizeof(buf)) << endl;
return 1;
}
RFC868UDPServer udp;
udp.Begin();
for(int i=0; ; i++)
{
Sleep(200);
if(udp.GetWorkCount() >= 5)//完成5客户端的时间发送就退出
break;
}
udp.WaitEnd(200);
WSACleanup();
}
catch(exception &e)
{
cout << e.what() << endl;
}
return 0;
}
