码迷,mamicode.com
首页 > 编程语言 > 详细

聊天室(C++客户端+Pyhton服务器)_1.框架搭设

时间:2019-06-16 15:47:36      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:反馈   except   image   ons   保存   结果   需要   use   释放   

聊天室

一.客户端发送

用MFC可视化做个客户端登录界面。

技术图片

 技术图片

技术图片

技术图片

 

技术图片

 

 

 

 

技术图片

 

技术图片

先点击注册账号按钮,注册账号的时候就需要连接到服务器,

服务器需要查数据库,并做出相应的回应。

所以开始写C++客户端套接口类用来连接到服务器。

demosocket.cpp文件

#include "pch.h"
#include <WS2tcpip.h>
#include "DemoSocket.h"
?
?
// 构造函数,用于执行初始化和套接字的创建
DemoSocket::DemoSocket()
{
// 0. 使用必须的结构体
WSAData WsaData = { 0 };
?
// 1. 初始化网络环境并请求指定版本
if (WSAStartup(MAKEWORD(2, 2), &WsaData))
{
MessageBox(NULL, L"网络环境初始化失败!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
?
// 2. 判断版本信息是否匹配
if (MAKEWORD(2,2) != WsaData.wVersion)
{
MessageBox(NULL, L"套接字版本不匹配!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
?
// 3. 创建一个套接字
ClientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (ClientSocket == INVALID_SOCKET)
{
MessageBox(NULL, L"创建套接字失败!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
}
?
?
// 析构函数,主要用于清理网络环境和释放套接字
DemoSocket::~DemoSocket()
{
// 清理网络环境和释放套接字
closesocket(ClientSocket);
WSACleanup();
?
// 结束接收线程并关闭内核对象句柄
}
?
?
// 用于设置对应的主窗口
void DemoSocket::SetMainWnd(CDialogEx* MainDialog)
{
// 主窗口对象主要用于获取主窗口的句柄,并发送数据
this->MainWnd = MainDialog;
}
?
?
// 用于连接到对应的服务器
void DemoSocket::Connect(LPCSTR Ip, u_short Port)
{
// 1. 使用传入的 Ip 和 Port 填充结构体
sockaddr_in ServerAddr = { AF_INET };
ServerAddr.sin_port = htons(Port);
inet_pton(AF_INET, Ip, &ServerAddr.sin_addr);
?
// 2. 调用 connect 连接到服务器
if (SOCKET_ERROR == connect(ClientSocket, (sockaddr*)& ServerAddr, sizeof(ServerAddr)))
{
MessageBox(NULL, L"连接服务器失败!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
?

 

 

 

 

客户端点击注册的时候具体怎么实现连接到服务器呢。

把登录窗口作为主窗口,主窗口初始化的时候实例化套接口类连接服务器,并把套接字绑定在主窗口上。

在主窗口类构造函数直接初始化连接到服务器

CMainDialog::CMainDialog(CWnd* pParent /*=nullptr*/)
: CDialogEx(IDD_DIALOG1, pParent)
, m_UserName(_T("xiaoming"))
, m_PassWord(_T("xiaoming"))
{
// 让套接字指向当前主窗口
MySocket.SetMainWnd(this);
?
// 进行网络连接,传入地址和端口
MySocket.Connect("127.0.0.1", 6666);
}

 

客户端点击注册按钮后,需要发送信息给服务器,那么套接字类需要添加发送的函数,同时定义发送的信息的结构体长什么样。

 

信息的结构体

结构体枚举

结构体的联合应用

通过自定义消息发送到服务器

自定义消息宏定义未识别,所以要自己定义

发送函数

发送

//放基础信息  pch.h文件里面
// 拟定的消息类型
enum TYPE {
REGISTER = 0, LOGIN, ADDFRD, UPDATEFRD, FRDMSG,
CRTGRP = 5, ADDGRP, UPDATEGRP, GRPMSG, UPDATEGRPUSER
};
?
// 0. 注册的结构体
typedef struct _REGISTER_INFO
{
WCHAR UserName[32];
WCHAR PassWord[32];
} REGISTER_INFO, *PREGISTER_INFO;
?
// 用于保存发送的数据
typedef struct _SEND_INFO
{
// 1. 类型:标识发送的消息是什么
TYPE Type;
?
// 2. 句柄:表示消息由谁发送
HWND hWnd;
?
// 3. 联合体: 不同类型对应不同的结构体
union {
REGISTER_INFO RegisterInfo;
LOGIN_INFO LoginInfo;
ADDFRD_INFO AddFrdInfo;
UPDATEFRD_INFO UpdateFrdInfo;
FRDMSG_INFO FrdMsg;
CRTGRP_INFO CrtGrpInfo;
ADDGRP_INFO AddGrpInfo;
UPDATEGRP_INFO UpdateGrpInfo;
GRPMSG_INFO GrpMsg;
UPDATEGRPUSER_INFO UpdateGrpUserInfo;
};
}SEND_INFO, * PSEND_INFO;
?
//demosocket.cpp文件里的函数
// 用于发送数据给服务器的函数
void DemoSocket::Send(LPVOID Buffer, INT Size)
{
// 发送消息并判断是否成功
if (SOCKET_ERROR == send(ClientSocket, (PCHAR)Buffer, Size, 0))
{
MessageBox(NULL, L"发送消息失败!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
}
?
?
//点击注册按钮CMainDlg.cpp文件里面
// 注册按钮的响应
void CMainDialog::OnBnClickedRegister()
{
// 1. 获取用户输入的数据
UpdateData(TRUE);
?
// 2. 判断输入的数据是否为空
if (m_UserName.IsEmpty() || m_PassWord.IsEmpty())
{
MessageBox(L"请输入用户名和密码");
return;
}
?
// 3. 填充需要发送的数据内容
SEND_INFO SendInfo = { TYPE::REGISTER, GetSafeHwnd() };
memcpy(SendInfo.RegisterInfo.UserName, m_UserName.GetBuffer(), m_UserName.GetLength() * 2);
memcpy(SendInfo.RegisterInfo.PassWord, m_PassWord.GetBuffer(), m_PassWord.GetLength() * 2);
?
// 4. 将数据通过自定义消息发送到服务器
SendMessage(UM_SEND_MSG, (WPARAM)& SendInfo, sizeof(SEND_INFO));
}
?
//自定义消息,打开类向导-选择自定义消息-编辑UM_SEND_MSG
// 任何一个窗口都需要通过这个消息发送数据到服务器
#define UM_SEND_MSG WM_USER + 1
// 发送消息, wParam 是需要发送的数据, lPram 是大小
afx_msg LRESULT CMainDialog::OnUmSendMsg(WPARAM wParam, LPARAM lParam)
{
// 直接的发送消息
MySocket.Send((LPVOID)wParam, lParam);
return 0;
}

 

 

二.python服务器

接受到客户端发来的消息处理完再发回给客户端

Mysql

数据库 user_table(user,pswd)

账号密码需要保存在数据库。所以开始写数据库相关的东西。

两个方法,一个有返回值,一个没有。

import pymysql
?
?
# 用于执行数据库操作
class Mysql(object):
?
   # 构造函数,用于连接数据库
   def __init__(self):
       try:
           # 获取连接对象,用于操作连接数据库
           self.connect = pymysql.connect(host="127.0.0.1",
               port=3306, user="root", password="123456",db="chatroom")
           # 获取游标对象
           self.cursor = self.connect.cursor()
       except Exception as info:
           print(info)
?
   # 执行带返回值的指令
   def select(self, sql):
       try:
           # 执行sql指令
           self.cursor.execute(sql)
           # 获取执行的结果
           return self.cursor.fetchall()
       except Exception as info:
           # 回滚操作
           self.connect.rollback()
           # 输出错误信息
           print(info)
?
?
   # 执行不带返回值的指令
   def exec(self, sql):
       try:
           # 执行sql指令
           self.cursor.execute(sql)
           # 提交完成的操作
           self.connect.commit()
       except Exception as info:
           # 回滚操作
           self.connect.rollback()
           # 输出错误信息
           print(info)

 

首先是账号密码。

客户端连接上服务器,判断账号密码是否正确是在服务器上,所以数据库跟python对接,

# 构造函数,用于初始化网络以及创建套接字
   def __init__(self):
       # 1.创建套接字,是服务器的套接字
       self.server = socket.socket()
       # 2.绑定到指定的地址和端口,与客户端请求的一致
       self.server.bind(("127.0.0.1", 6666))
       # 3.设置当前套接字的监听状态
       self.server.listen(socket.SOMAXCONN)
       # 4. 初始化mysql对象
       self.mysql = mysql.Mysql()
           # 实例方法,用于接收客户端的连接
   def accept(self):
       while True:
           # 1. 直接调用函数接收客户端并保存套接字和地址
           client, address = self.server.accept()
           # 2. 输出显示客户端的连接消息
           print(address, "连接到了服务器")
           # 3. 为每一个客户端创建一个接收线程
           threading.Thread(target=self.recevice, args=(client,)).start()
?

 

 

C++点击登录之后,信息打包发给服务器,服务器要不断接受到信息,并根据不同消息分配不同函数。

 # 实例方法,用于分别接收每一个客户端的消息
   def recevice(self, client):
       while True:
           try:
               # 1. 获取了客户端发送的消息
               message = client.recv(2048)
               # 2. 解包获取其中的类型(返回的是元组)
               type, = struct.unpack("i", message[:4])
               # 3. 根据不同的类型调用不同的函数,第一个参数是self
               DemoChat.dict_func[type](self, client, message[4:])
           except Exception as info:
               # 如果产生了一场,说明客户端已经退出了
               for username, client_socket in self.dict_client.items():
                   # 如果套接字相同,就从字典中移除,说明退出登录
                   if client_socket == client:
                       self.dict_client.pop(username)
                       print(username, "退出了聊天室")
                       #################################
                       print(info)
                       #################################
                       break
               break
?
   

 

服务器做判断。根据数据建判断

# 定义一个枚举类型,标注消息类型
class Type(enum.Enum):
   REGISTER = 0
   LOGIN = 1
   ADDFRD = 2
   UPDATEFRD = 3
   FRDMSG = 4
   CRTGRP = 5
   ADDGRP = 6
   UPDATEGRP = 7
   GRPMSG = 8
# 创建一个字典,用于保存类型和与之匹配的处理函数
   dict_func = {
       Type.REGISTER.value: on_register,
       Type.LOGIN.value: on_login,
       Type.ADDFRD.value: on_addfrd,
       Type.UPDATEFRD.value: on_updatefrd,
       Type.FRDMSG.value: on_frdmsg,
       Type.CRTGRP.value: on_crtgrp,
       Type.ADDGRP.value: on_addgrp,
       Type.UPDATEGRP.value: on_updategrp,
       Type.GRPMSG.value: on_grpmsg
  }
?

 

 

分配到了注册的方法

根据数据库检查是否存在账号,不存在就注册

# 实例方式:用于处理注册消息
   def on_register(self, client, message):
       # 解包获取发送来的数据
       hwnd, username, password = struct.unpack("i64s64s", message[:132])
       # 将用户名和密码进行解码
       username = username.decode("UTF16").strip("\x00")
       password = password.decode("UTF16").strip("\x00")
       # 判断用户是否已存在
       if self.mysql.select("SELECT * FROM user_table WHERE user=‘%s‘" % username):
           # 发送的内容 => 类型 + 句柄 + 是否成功 + 服务器返回的信息
           client.send(struct.pack("iii40s", Type.REGISTER.value, hwnd, 0, "用户名已存在".encode("UTF16")))
       else:
           # 如果不存在,向数据库中添加一组数据
           self.mysql.exec("INSERT INTO user_table(user,pswd) VALUE(‘%s‘,MD5(‘%s‘))" % (username, password))
           # 将注册结果进行返回
           client.send(struct.pack("iii40s", Type.REGISTER.value, hwnd, 1, "注册成功".encode("UTF16")))
?
?

 

三.客户端接收到服务器发来的消息根据消息处理。

客户端发送消息结构体给服务器后,要循环接收服务器的反馈,所以在客户端要写个线程不断接受来自服务器的信息,
// 3. 创建线程用于接收服务器发过来的消息,线程函数必须是静态或全局的
Thread = CreateThread(NULL, NULL, WorkerThread, this, NULL, NULL);
if (NULL == Thread)
{
MessageBox(NULL, L"接收线程创建失败!", L"错误", MB_OK | MB_ICONERROR);
ExitProcess(0);
}
}
// 线程函数,用于接收消息并将消息发送给主窗口
DWORD WINAPI DemoSocket::WorkerThread(LPVOID lpThreadParameter)
{
// 1. 获取需要用到的所有数据
PCHAR Buffer = new CHAR[sizeof(RECV_INFO)]{ 0 };
DemoSocket* pMySocket = (DemoSocket*)lpThreadParameter;
?
// 2. 循环的接收服务器发送的数据
while (recv(pMySocket->ClientSocket, Buffer, sizeof(RECV_INFO), 0) > 0)
{
// 3. 将接收到的消息转交给主窗口进行分发
SendMessage(pMySocket->MainWnd->GetSafeHwnd(), UM_RECV_MSG, NULL,(LPARAM)Buffer);
}
?
// 4. 网络断开连接(自己的网\服务器)
MessageBox(NULL, L"网络连接已断开", L"错误", MB_OK | MB_ICONERROR);
delete[] Buffer; ExitProcess(0);
}

 

线程接受到的信息后,转发给主窗口响应UM_RECV_MSG消息

所以要自定义消息,打开类向导-选择自定义消息-编辑UM_RECV_MSG,

主窗口把注册的消息转发给了对应的消息,这里是UM_RECV_REGISTER

// 套接字使用这个消息将接收到的内容发送给主窗口
#define UM_RECV_MSG WM_USER + 2
// 接收到服务器发送的消息并进行处理(关键函数)
afx_msg LRESULT CMainDialog::OnUmRecvMsg(WPARAM wParam, LPARAM lParam)
{
// 1. 将 lParam 转换成对应的结构体
PRECV_INFO RecvInfo = (PRECV_INFO)lParam;
?
// 2. 根据不同的类型,转发消息给不同的窗口
switch (RecvInfo->Type)
{
// 响应注册
case TYPE::REGISTER:
{
// 对于注册是否成功的消息,wParam 没有用,lParam 指向 RECV_STATE
::SendMessage(RecvInfo->hWnd, UM_RECV_REGISTER, NULL, (LPARAM)&RecvInfo->RecvState);
break;
}
?
return 0;
}

通过自定义消息响应接受信息

自定义消息宏定义未识别,所以要自己定义

// 是否注册成功
#define UM_RECV_REGISTER WM_USER + 3
// 是否注册成功
afx_msg LRESULT CMainDialog::OnUmRecvRegister(WPARAM wParam, LPARAM lParam)
{
// 将 lParam 转换成对应的的结构
PRECV_STATE RecvState = (PRECV_STATE)lParam;
?
// 输出服务器返回的消息
MessageBox(RecvState->MsgInfo);
?
return 0;
}

接受信息的结构体

// 返回服务器处理的结果
typedef struct _RECV_STATE
{
BOOL IsSuccess; // 是否成功
WCHAR MsgInfo[20]; // 如果失败了,原因是什么
}RECV_STATE, *PRECV_STATE;
?
?
// 用于保存接收的数据
typedef struct _RECV_INFO
{
// 1. 类型:标识接收的消息是什么
TYPE Type;
?
// 2. 句柄:表示消息由谁发送
HWND hWnd;
?
// 3. 联合体: 不同类型对应不同的结构体
union {
RECV_STATE RecvState;
UPDATEFRD_INFO UpdateFrdInfo;
FRDMSG_INFO FrdMsg;
UPDATEGRP_INFO UpdateGrpInfo;
};
} RECV_INFO, * PRECV_INFO;

聊天室(C++客户端+Pyhton服务器)框架搭设完毕。

聊天室(C++客户端+Pyhton服务器)_1.框架搭设

标签:反馈   except   image   ons   保存   结果   需要   use   释放   

原文地址:https://www.cnblogs.com/ltyandy/p/11031474.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!