C++套接字编程(聊天程序)

创建一个基于对话框的MFC程序MultiThread，首先设置界面，并设置控件ID
接收数据是组合框，下面编辑框
发送数据是组合框，下面有IP地址框，编辑框，发送按钮，ID号如下

#define IDC_EDIT_RECV 1000
#define IDC_IPADDRESS1 1001
#define IDC_EDIT_SEND 1002
#define IDC_BTN_SEND 1003



1、首先加载套接字库，在InitStance()中调用AfxSocketInit()，在stdafx.h中加入“Afxsock.h”,AfxsocketInit()的调用一定在创建窗口之前，否者出现加载套接字失败的情况。

if (!AfxSocketInit())
{
AfxMessageBox(_T("加载套接字失败"));
return false;
}

2、在MFCmultithread.h中创建一个结构体，包含套接字和窗口的句柄，主要是投送消息时可以将两个消息同时传送。

struct RECVPARAM{
SOCKET socket;
HWND hWnd;
};

3、在MFCmultithread类中增加成员变量m_socket和成员方法InitSocket()，InitSocket()完成m_socket的初始化和绑定。

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

int retval;
retval = bind(m_socket, (SOCKADDR*)&addrSock, sizeof(SOCKADDR));
if (SOCKET_ERROR == retval)
{
closesocket(m_socket);
AfxMessageBox(_T("绑定失败！"));
return false;

}
return true;
}

4、创建一个线程函数，要么是全局的，要么是静态的，或者是类的成员方法，在这里创建为类的静态方法，它不属于具体的某一个类成员，属于类。

在MFCmultithread.h中添加静态的成员方法

static DWORD WINAPI RecvProc(LPVOID lpParameter);

DWORD WINAPI  CMFCmultithreadDlg::RecvProc(LPVOID lpParameter)
{
SOCKET socket = ((RECVPARAM*)lpParameter)->socket;
HWND hWnd = ((RECVPARAM*)lpParameter)->hWnd;//窗口句柄
SOCKADDR_IN addrRecv;
int len = sizeof(SOCKADDR);
char recvBuf[100];
char tempBuf[100];
int recvRst;
CString recvStr;
while (TRUE)
{
//接收数据
recvRst = recvfrom(socket, recvBuf, 100, 0, (SOCKADDR*)&addrRecv, &len);
if (SOCKET_ERROR == recvRst)
{
break;
}
sprintf(tempBuf, "%s 说:%s\r\n", inet_ntoa(addrRecv.sin_addr), recvBuf);
//
//recvStr += tempBuf;//这种方式更加简单不用使用消息
//
//::SetDlgItemText(hWnd, IDC_EDIT_RECV, recvStr);
//使用自定义消息的方式来对控件填充内容
::PostMessage(hWnd, WM_RECVDATA, 0, (LPARAM)tempBuf);
}
return 0;
}

5、在OnInitDialog中调用InitSocket完成套接字初始化

InitSocket();

RECVPARAM *pRecvParm = new RECVPARAM();
pRecvParm->socket = m_socket;
pRecvParm->hWnd = m_hWnd;
HANDLE hThreadRecv = CreateThread(NULL, 0, RecvProc, (LPVOID)pRecvParm, 0, NULL);
CloseHandle(hThreadRecv);

6、添加一个自定义消息响应函数

#define WM_RECVDATA WM_USER+1

afx_msg LRESULT OnRecvData(WPARAM wParam,LPARAM lParam);

ON_MESSAGE(WM_RECVDATA, OnRecvData)

在类中添加OnRecvData函数

LRESULT CMFCmultithreadDlg::OnRecvData(WPARAM wParam, LPARAM lParam)
{
CString str((char*)lParam);
CString strTemp;
GetDlgItemText(IDC_EDIT_RECV, strTemp);
str += "\r\n";
str += strTemp;
SetDlgItemText(IDC_EDIT_RECV, str);
return true;
}

7、添加发送按钮的响应函数

void CMFCmultithreadDlg::OnBtnSend()
{
// TODO:  在此添加控件通知处理程序代码
DWORD dwIP;
((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);
SOCKADDR_IN addrTo;
addrTo.sin_family = AF_INET;
addrTo.sin_port = htons(6000);
addrTo.sin_addr.S_un.S_addr = htonl(dwIP);
CString strSend;
GetDlgItemText(IDC_EDIT_SEND, strSend);
sendto(m_socket, strSend, strSend.GetLength() + 1, 0,
(SOCKADDR*)&addrTo, sizeof(SOCKADDR));
SetDlgItemText(IDC_EDIT_SEND, "");
}