简单UDP聊天模型
下面是自己写的一个UDP的对等聊天模型，将客户端与服务器整合到一起，可以用于在局域网内聊天。
基本思想：
在主线程中创建一个线程，在一个约定的端口进行绑定，该线程用于接收并打印聊天信息。在主线程中进行聊天输入处理
具体代码如下：
#include "iostream"
#include <winsock2.h>
#include <Windows.h>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
#define Const_BufLength 1024
#define Const_Port 12345
BOOL InitServerSocket(void);
BOOL InitClientSocket(void);
WSADATA wsaData;
SOCKET m_socket;
SOCKET m_client_socket;
sockaddr_in SerAdd,CliAdd;
int iLength=sizeof(CliAdd);
int retval;
char strSendMsg[Const_BufLength];
char strRecvMsg[Const_BufLength];
void ReceiveData();
void SendData();
int main()
{
if(!InitServerSocket())
{
cout<<"服务套接字初始化出错，即将退出"<<endl;
WSACleanup();
system("pause");
return -1;
}
if(!InitClientSocket())
{
cout<<"客户套接字初始化出错，即将退出"<<endl;
WSACleanup();
system("pause");
return -1;
}
char ip[20];
cout << "input ip" << endl;
cin >> ip;
CliAdd.sin_addr.s_addr = inet_addr(ip);
HANDLE handle1 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)ReceiveData, NULL, 0, NULL);
while(1)
{
memset(strSendMsg,0,sizeof(strSendMsg));
cout<<"回复："<<endl;
cin >> strSendMsg;
if(strcmp(strSendMsg,"exit")==0)
{
strcpy(strSendMsg,"收到客户下线的消息，我是服务器，我也下线了");
retval=sendto(m_socket,strSendMsg,sizeof(strSendMsg),0,(SOCKADDR*)&CliAdd,sizeof(CliAdd));
closesocket(m_socket);
WSACleanup();
return 0;
}
CliAdd.sin_family = AF_INET;
retval=sendto(m_client_socket,strSendMsg,sizeof(strSendMsg),0,(SOCKADDR*)&CliAdd,sizeof(CliAdd));
if(retval==SOCKET_ERROR)
{
cout<<"发送失败，错误ID为："<<WSAGetLastError()<<endl;
break;
}
else if(retval==0)
{
break;
}
else
{
cout<<"发送成功!"<<endl;
}
}
CloseHandle(handle1);
closesocket(m_socket);
WSACleanup();
return 0;
}
void ReceiveData()
{
sockaddr_in socktmp;
while(1)
{
memset(strRecvMsg,0,sizeof(strRecvMsg));
retval=recvfrom(m_socket,strRecvMsg,Const_BufLength,0,(SOCKADDR*)&socktmp,&iLength);
if(retval==SOCKET_ERROR)
{
cout<<"接受失败，错误ID为："<<WSAGetLastError()<<endl;
break;
}
else if(retval==0)
{
break;
}
else
{
cout<<"IP:Port/t"<<inet_ntoa(CliAdd.sin_addr)<<":"<<ntohs(CliAdd.sin_port)<<'/t'<<strRecvMsg<<endl;
}
}
}
BOOL InitServerSocket(void)
{
if(WSAStartup(MAKEWORD(2,2),&wsaData)!=0)
{
cout<<"加载winsock失败，错误ID为："<<WSAGetLastError()<<endl;
exit(0);
}
m_socket=socket(AF_INET,SOCK_DGRAM,0);
SerAdd.sin_family=AF_INET;
SerAdd.sin_port=htons(Const_Port);
SerAdd.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
int retval;
retval=bind(m_socket,(SOCKADDR*)&SerAdd,sizeof(SOCKADDR));
if(SOCKET_ERROR==retval)
{
closesocket(m_socket);
cout<<"端口绑定失败! 错误ID为："<<WSAGetLastError()<<endl;
return FALSE;
}
return TRUE;
}
BOOL InitClientSocket(void)
{
if(WSAStartup(MAKEWORD(2,2),&wsaData)!=0)
{
cout<<"加载winsock失败，错误ID为："<<WSAGetLastError()<<endl;
exit(0);
}
m_client_socket=socket(AF_INET,SOCK_DGRAM,0);
if(INVALID_SOCKET==m_client_socket)
{
cout<<"套接字创建失败！错误ID为："<<WSAGetLastError()<<endl;
return FALSE;
}
CliAdd.sin_addr.s_addr = inet_addr("127.0.0.1");
CliAdd.sin_port = htons(12345);
return TRUE;
}