带界面的UDP聊天程序

BEGIN_MESSAGE_MAP(CMFCApplication38Dlg, CDialogEx)
ON_MESSAGE(WM_USER+1,OnRecvData)
ON_BN_CLICKED(IDC_BTN_SEND, &CMFCApplication38Dlg::OnBnClickedBtnSend)
END_MESSAGE_MAP()

// CMFCApplication38Dlg 消息处理程序

BOOL CMFCApplication38Dlg::OnInitDialog()
{
CDialogEx::OnInitDialog();

if(!AfxSocketInit())
{
AfxMessageBox(_T("加载套接字失败！"));
return FALSE;
}
if(!InitSocket())
{
return FALSE;
}
// TODO: 在此添加额外的初始化代码
RECVPARAM *param = new RECVPARAM;
param->hwnd = m_hWnd;
param->socket = m_socket;
handle = CreateThread(NULL,0,RecvThread,(LPVOID)param,0,NULL);
CloseHandle(handle);
return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

BOOL CMFCApplication38Dlg::InitSocket()
{
m_socket = socket(AF_INET,SOCK_DGRAM,0);
if(INVALID_SOCKET == m_socket)
{
AfxMessageBox(_T("套接字创建失败!"));
return FALSE;
}
SOCKADDR_IN socketaddr;
socketaddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
socketaddr.sin_family = AF_INET;
socketaddr.sin_port = htons(6000);

int retvel;

retvel = bind(m_socket,(SOCKADDR*)&socketaddr,sizeof(SOCKADDR));

if(SOCKET_ERROR == retvel)
{
closesocket(m_socket);
AfxMessageBox(_T("绑定失败!"));
return FALSE;
}
return TRUE;
}

DWORD WINAPI CMFCApplication38Dlg::RecvThread(LPVOID lpParam)
{
SOCKET socket = ((RECVPARAM*)lpParam)->socket;
HWND hwnd = ((RECVPARAM*)lpParam)->hwnd;
delete lpParam;
SOCKADDR_IN socketaddr;
int len = sizeof(SOCKADDR);
char RecvBuf[200];
char tempBuf[200];
int retval;
while(TRUE)
{
retval = recvfrom(socket,RecvBuf,200,0,(SOCKADDR*)&socketaddr,&len);

sprintf_s(tempBuf,"%s:%s",inet_ntoa(socketaddr.sin_addr),RecvBuf);
::PostMessage(hwnd,WM_USER+1,0,(LPARAM)tempBuf);
}
return 0;
}

LRESULT CMFCApplication38Dlg::OnRecvData(WPARAM wparam,LPARAM lparam)
{
CString str((char *)lparam);
CString strTemp;
GetDlgItemText(IDC_EDIT_RECV,strTemp);
//strTemp +="\r\n";
str = strTemp + str;
str += "\r\n";
SetDlgItemText(IDC_EDIT_RECV,str);
return 0;
}

void CMFCApplication38Dlg::OnBnClickedBtnSend()
{
// TODO: 在此添加控件通知处理程序代码
DWORD dwIP;
((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);
SOCKADDR_IN addrTo;
addrTo.sin_addr.S_un.S_addr = htonl(dwIP);
addrTo.sin_family = AF_INET;
addrTo.sin_port = htons(6000);
CString strSend;
GetDlgItemText(IDC_EDIT_SEND,strSend);

char SendBuf[200];
int i;
for(i=0;i<strSend.GetLength();i++)
{
SendBuf[i] = strSend[i];
}
SendBuf[i] ='\0';
sendto(m_socket,SendBuf,strSend.GetLength()+1,0,(SOCKADDR*)&addrTo,sizeof(SOCKADDR));
SetDlgItemText(IDC_EDIT_SEND,_T(""));
}