MFC网络通信

服务器端：

#include "stdafx.h"
#include "dataHandle.h"
#include "mysql.h"
#include <WinSock2.h>
#pragma comment(lib, "ws2_32")
void SockInit();
int _tmain(int argc, _TCHAR* argv[])
{


MYSQL m_sqlCon;
mysql_init(&m_sqlCon);//初始化数据库对象
if (!mysql_real_connect(&m_sqlCon, "localhost", "root", "123456", "student", 3306, NULL, 0))
//localhost:服务器地址，可以直接填入IP;root:账号;123:密码;test:数据库名;3306:网络端口  
{
AfxMessageBox(_T("数据库连接失败!"));
return FALSE;
}
else//连接成功则继续访问数据库，之后的相关操作代码基本是放在这里面的
{
AfxMessageBox(_T("数据库连接成功!"));
}


SockInit();


SOCKET lisSock = socket(AF_INET, SOCK_STREAM, 0);
sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(8090);
addr.sin_addr.S_un.S_addr = ADDR_ANY;
bind(lisSock, (sockaddr*)&addr, sizeof(sockaddr));
listen(lisSock, 5);

fd_set fds;
FD_ZERO(&fds);

FD_SET(lisSock, &fds);  //将监听套接字加入数组中


timeval timeout;
timeout.tv_sec = 5;
timeout.tv_usec = 0;


while (1)
{
fd_set tmSet = fds;  //临时数组
//移除掉没有事件发生的套接字


// SELECT 在编程的过程中，经常会遇到许多阻塞的函数，
//好像read和网络编程时使用的recv, recvfrom函数都是阻塞的函数，
//当函数不能成功执行的时候，程序就会一直阻塞在这里，无法执行下面的代码。
//这是就需要用到非阻塞的编程方式，使用select函数就可以实现非阻塞编程。
select(0, &tmSet, 0, 0, &timeout);


SOCKET s[5];
int j = 0;


//剩下的都是有事件发生的套接字
for (int i = 0; i < tmSet.fd_count; i++)
{
if (tmSet.fd_array[i] == lisSock) //如果是监听监听套接字
{
SOCKET cliSock = accept(lisSock, 0, 0);
printf("新连接.\n");
FD_SET(cliSock, &fds);  //将已连接的套接字加入数组中
}
else
{
char recvBuf[1024] = { 0 };
int recvLen = recv(tmSet.fd_array[i], recvBuf, 1024, 0);

if (recvLen > 0)
{
//接收到客户端的数据后进行处理
HandleData(recvBuf,recvLen,m_sqlCon, tmSet.fd_array[i]);
}
else  //错误发生，或者客户端断开连接 
{
printf("客户已断开\n");
FD_CLR(tmSet.fd_array[i], &fds);  //从数组中移除已端口的客户端
}
}
}
}
return 0;
}


void SockInit()
{
WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD(2, 2);
err = WSAStartup(wVersionRequested, &wsaData);
if (err != 0) {
return;
}
if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
{
WSACleanup();
return;
}

}

客户端：

（1）在BOOL CclientApp::InitInstance()中添加如下代码：

        AfxSocketInit();//网络初始化
clientSock sock;
sock.Create();
bool conRet = sock.Connect(_T("127.0.0.1"), 8090);

if (conRet == false) {
int errCode = GetLastError();
CString err;
err.Format(_T("连接服务器失败 : %d，请重试!"), errCode);
MessageBox(0, err, _T("错误"), MB_OK);
return FALSE;

}

其中clientSock继承系统类CSocket，clientSock.h

class clientSock : public CSocket //继承MFC的类
{
public:
clientSock();
virtual ~clientSock();
virtual void OnReceive(int nErrorCode);

};

clientSock.cpp中，实现系统方法OnReceive：

void clientSock::OnReceive(int nErrorCode)
{
// TODO: 在此添加专用代码和/或调用基类

char recvData[1024] = {0};
memset(recvData, 0, sizeof(recvData));
int nLen = Receive(recvData, sizeof(recvData));


short msgID = *(short*)recvData;
switch (msgID)
{


case LOGIN_RESULT_MSGID:
{
LOGIN_RESULT loginMsg;
memset(&loginMsg, 0, sizeof(LOGIN_RESULT));
memcpy(&loginMsg,recvData, sizeof(loginMsg));//注意sizeof中要是转换的结构体大小
if (strcmp(loginMsg.result, "loginTrue") == 0) {
AfxMessageBox("登录成功!");
::PostMessage(login_hwnd, LOGINMSG, 0, (LPARAM)recvData);//自定义消息,用于将内容传到对话框
}
else
{
AfxMessageBox("登录失败!");
}


}
break;


case REGISTER_RESULT_MSGID:
{
REGISTER_RESULT registerMsg;
memset(&registerMsg, 0, sizeof(LOGIN_RESULT));
memcpy(&registerMsg, recvData, sizeof(registerMsg));//注意sizeof中要是转换的结构体大小
if (strcmp(registerMsg.result, "registerTrue") == 0) {
AfxMessageBox("注册成功!");
}
else
{
AfxMessageBox("注册失败!");
}


}
break;


case ALLCOURSE_RESULT_MSGID://显示所有课程
{
::PostMessage(student_hwnd, SHOW_COURSE_MSG, 0, (LPARAM)recvData);//自定义消息
}
break;


case STUDENT_COURSE_RESULT_MSGID://显示学生已选课程
{
::PostMessage(student_hwnd, SHOW_MYCOURSE_MSG, 0, (LPARAM)recvData);//自定义消息
}
break;


case STUDENT_ADD_COURSE_RESULT_MSGID://学生添加课程结果
{
STUDENT_ADD_COURSE_RESULT addMsg;
memset(&addMsg, 0, sizeof(STUDENT_ADD_COURSE_RESULT));
memcpy(&addMsg, recvData, sizeof(addMsg));//注意sizeof中要是转换的结构体大小
if (strcmp(addMsg.result, "addCourseTrue") == 0) {
::PostMessage(student_hwnd, UPDATE_DATA_MSG, 0, 0);//自定义消息
AfxMessageBox("选课成功!");
}
else
{
AfxMessageBox("选课失败!");
}
}
break;


case STUDENT_DELETE_COURSE_RESULT_MSGID://学生添加课程结果
{
STUDENT_DELETE_COURSE_RESULT deleteMsg;
memset(&deleteMsg, 0, sizeof(STUDENT_DELETE_COURSE_RESULT));
memcpy(&deleteMsg, recvData, sizeof(deleteMsg));//注意sizeof中要是转换的结构体大小
if (strcmp(deleteMsg.result, "deleteCourseTrue") == 0) {
::PostMessage(student_hwnd, UPDATE_DATA_MSG, 0, 0);//自定义消息
AfxMessageBox("删除课程成功!");
}
else
{
AfxMessageBox("删除课程失败!");
}
}
break;
}


CSocket::OnReceive(nErrorCode);
}