301 lines
9.6 KiB
Markdown
301 lines
9.6 KiB
Markdown
|
## 添加用户注册功能
|
|||
|
|
|
|||
|
|
#### 要求
|
|||
|
|
|
|||
|
|
在之前实验的服务器客户端代码中增添用户注册的功能，在客户端为用户提供注册的选项，并将用户注册的账号、密码发送到服务器端，并保存在服务器的 MySQL 数据库中。
|
|||
|
|
|
|||
|
|
1. 在服务器本地 Mysql 中创建新数据库 ChatProject，库中有一张表叫 USER，该表中有账号 NAME 和密码 PASSWORD 两项属性。
|
|||
|
|
2. 为客户端添加注册功能，让用户注册账号密码，客户端将注册信息发送到服务器端，注意：当用户注册时应输入两次密码，如果密码不一致需要重新输入。
|
|||
|
|
3. 服务器端接收来自客户端的注册信息，将用户的账号、密码写入本地 Mysql 数据库。
|
|||
|
|
4. 要面向对象编程，进行类封装。
|
|||
|
|
|
|||
|
|
#### 具体实现
|
|||
|
|
|
|||
|
|
首先在 MySQL 控制台创建数据库 ChatProject，如下：
|
|||
|
|
|
|||
|
|
```sql
|
|||
|
|
create database ChatProject;
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
接下来先用 use 命令切换到新建的数据库。
|
|||
|
|
|
|||
|
|
```sql
|
|||
|
|
use ChatProject;
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
然后新建一张表格叫 USER 用来保存账号信息，表中有账号 NAME 和密码 PASSWORD 两项属性，都为 VARCHAR 可变长度字符串类型，且将账号 NAME 设为 PRIMARY KEY 主键，主键不允许重复保证了账号的唯一性，而且主键能自动建立索引加快查询速度。
|
|||
|
|
|
|||
|
|
```sql
|
|||
|
|
CREATE TABLE USER(
|
|||
|
|
NAME VARCHAR(20) PRIMARY KEY,
|
|||
|
|
PASSWORD VARCHAR(20)
|
|||
|
|
);
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
建好表之后我们可以查看当前数据库中所有的表格。
|
|||
|
|
|
|||
|
|
```sql
|
|||
|
|
show tables;
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
最后输入 `exit` 即可离开 MySQL 控制台回到终端。
|
|||
|
|
|
|||
|
|
创建好了表格之后，我们就可以开始为 Client 类和 Server 类添加注册账号的代码了。
|
|||
|
|
|
|||
|
|
首先修改 `client.h` 头文件，添加一个函数 HandleClient，该函数将在与服务器连接建立之后开始工作，与用户进行交互并处理各项事务。
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
#ifndef CLIENT_H
|
|||
|
|
#define CLIENT_H
|
|||
|
|
|
|||
|
|
#include "global.h"
|
|||
|
|
|
|||
|
|
class client{
|
|||
|
|
private:
|
|||
|
|
int server_port;
|
|||
|
|
string server_ip;
|
|||
|
|
int sock;
|
|||
|
|
public:
|
|||
|
|
client(int port,string ip);
|
|||
|
|
~client();
|
|||
|
|
void run();
|
|||
|
|
static void SendMsg(int conn);
|
|||
|
|
static void RecvMsg(int conn);
|
|||
|
|
void HandleClient(int conn);
|
|||
|
|
};
|
|||
|
|
#endif
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
然后修改 `client.cpp` 文件中的 run 函数，连接完服务器之后就调用 HandleClient，并将文件描述符作为参数传入，如下：
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
void client::run(){
|
|||
|
|
|
|||
|
|
//定义sockfd
|
|||
|
|
sock = socket(AF_INET,SOCK_STREAM, 0);
|
|||
|
|
|
|||
|
|
//定义sockaddr_in
|
|||
|
|
struct sockaddr_in servaddr;
|
|||
|
|
memset(&servaddr, 0, sizeof(servaddr));
|
|||
|
|
servaddr.sin_family = AF_INET;
|
|||
|
|
servaddr.sin_port = htons(server_port); //服务器端口
|
|||
|
|
servaddr.sin_addr.s_addr = inet_addr(server_ip.c_str()); //服务器ip
|
|||
|
|
|
|||
|
|
//连接服务器，成功返回0，错误返回-1
|
|||
|
|
if (connect(sock, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0)
|
|||
|
|
{
|
|||
|
|
perror("connect");
|
|||
|
|
exit(1);
|
|||
|
|
}
|
|||
|
|
cout<<"连接服务器成功\n";
|
|||
|
|
|
|||
|
|
HandleClient(sock);
|
|||
|
|
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
接下来设计 `client.cpp` 文件中的 HandleClient 函数，开头先打印一段信息指示用户操作，然后开始处理事务，如果用户输入 2 就进入注册模块，注册时输入两次密码，密码不一致就要重新输入。最后将注册的账号和密码格式化成“name:xxxpass:yyy”（xxx 代表用户名，yyy 代表密码）发送到服务器端，格式化之后可以方便服务器提取出用户名和密码。
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
void client::HandleClient(int conn){
|
|||
|
|
int choice;
|
|||
|
|
string name,pass,pass1;
|
|||
|
|
//bool if_login=false;//记录是否登录成功
|
|||
|
|
|
|||
|
|
cout<<" ------------------\n";
|
|||
|
|
cout<<"| |\n";
|
|||
|
|
cout<<"| 请输入你要的选项:|\n";
|
|||
|
|
cout<<"| 0:退出 |\n";
|
|||
|
|
cout<<"| 1:登录 |\n";
|
|||
|
|
cout<<"| 2:注册 |\n";
|
|||
|
|
cout<<"| |\n";
|
|||
|
|
cout<<" ------------------ \n\n";
|
|||
|
|
|
|||
|
|
//开始处理各种事务
|
|||
|
|
while(1){
|
|||
|
|
//if(if_login)
|
|||
|
|
// break;
|
|||
|
|
cin>>choice;
|
|||
|
|
if(choice==0)
|
|||
|
|
break;
|
|||
|
|
//注册
|
|||
|
|
else if(choice==2){
|
|||
|
|
cout<<"注册的用户名:";
|
|||
|
|
cin>>name;
|
|||
|
|
while(1){
|
|||
|
|
cout<<"密码:";
|
|||
|
|
cin>>pass;
|
|||
|
|
cout<<"确认密码:";
|
|||
|
|
cin>>pass1;
|
|||
|
|
if(pass==pass1)
|
|||
|
|
break;
|
|||
|
|
else
|
|||
|
|
cout<<"两次密码不一致!\n\n";
|
|||
|
|
}
|
|||
|
|
name="name:"+name;
|
|||
|
|
pass="pass:"+pass;
|
|||
|
|
string str=name+pass;
|
|||
|
|
send(conn,str.c_str(),str.length(),0);
|
|||
|
|
cout<<"注册成功！\n";
|
|||
|
|
cout<<"\n继续输入你要的选项:";
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
接下来开始设计服务器端，因为服务器要连接 MySQL，所以我们先在 `global.h` 中添加 mysql 头文件，如下：
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
#ifndef _GLOBAL_H
|
|||
|
|
#define _GLOBAL_H
|
|||
|
|
|
|||
|
|
#include <sys/types.h>
|
|||
|
|
#include <sys/socket.h>
|
|||
|
|
#include <stdio.h>
|
|||
|
|
#include <netinet/in.h>
|
|||
|
|
#include <arpa/inet.h>
|
|||
|
|
#include <unistd.h>
|
|||
|
|
#include <string.h>
|
|||
|
|
#include <stdlib.h>
|
|||
|
|
#include <fcntl.h>
|
|||
|
|
#include <sys/shm.h>
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <thread>
|
|||
|
|
#include <vector>
|
|||
|
|
#include <mysql/mysql.h> //新添加
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
然后编写 `server.h`，添加静态函数 HandleRequest 的定义。
|
|||
|
|
|
|||
|
|
server.h:
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
#ifndef SERVER_H
|
|||
|
|
#define SERVER_H
|
|||
|
|
|
|||
|
|
#include "global.h"
|
|||
|
|
|
|||
|
|
class server{
|
|||
|
|
private:
|
|||
|
|
int server_port;
|
|||
|
|
int server_sockfd;
|
|||
|
|
string server_ip;
|
|||
|
|
static vector<bool> sock_arr;
|
|||
|
|
public:
|
|||
|
|
server(int port,string ip);
|
|||
|
|
~server();
|
|||
|
|
void run();
|
|||
|
|
static void RecvMsg(int conn);
|
|||
|
|
static void HandleRequest(int conn,string str);
|
|||
|
|
};
|
|||
|
|
#endif
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
修改 `server.cpp` 中 RecvMsg 的代码，让其接收到信息之后调用 HandleRequest 函数，并传入文件描述符和接收到的字符串作为参数。
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
void server::RecvMsg(int conn){
|
|||
|
|
//接收缓冲区
|
|||
|
|
char buffer[1000];
|
|||
|
|
//不断接收数据
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
memset(buffer,0,sizeof(buffer));
|
|||
|
|
int len = recv(conn, buffer, sizeof(buffer),0);
|
|||
|
|
//客户端发送exit或者异常结束时，退出
|
|||
|
|
if(strcmp(buffer,"exit")==0 || len<=0){
|
|||
|
|
close(conn);
|
|||
|
|
sock_arr[conn]=false;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
cout<<"收到套接字描述符为"<<conn<<"发来的信息："<<buffer<<endl;
|
|||
|
|
string str(buffer);
|
|||
|
|
HandleRequest(conn,str);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
接下来在 `server.cpp` 中给出 HandleRequest 具体实现，首先连接 MySQL 数据库，然后判断字符串是否含有“name:”，含有就说明是注册信息，通过字符串的 find、substr 函数提取出用户名和密码，然后组合成 SQL 插入语句并通过 mysql_query 执行该 SQL 语句即可。这里我们插入数据用的 SQL 语句为“INSERT INTO USER VALUES ("xxx","yyy");”（xxx 和 yyy 为具体的用户名、密码）。
|
|||
|
|
|
|||
|
|
```cpp
|
|||
|
|
void server::HandleRequest(int conn,string str){
|
|||
|
|
char buffer[1000];
|
|||
|
|
string name,pass;
|
|||
|
|
// bool if_login=false;//记录当前服务对象是否成功登录
|
|||
|
|
//string login_name;//记录当前服务对象的名字
|
|||
|
|
//string target_name;//记录发送信息时目标用户的名字
|
|||
|
|
//int group_num;//记录群号
|
|||
|
|
|
|||
|
|
|
|||
|
|
//连接MYSQL数据库
|
|||
|
|
MYSQL *con=mysql_init(NULL);
|
|||
|
|
mysql_real_connect(con,"127.0.0.1","root","","ChatProject",0,NULL,CLIENT_MULTI_STATEMENTS);
|
|||
|
|
|
|||
|
|
if(str.find("name:")!=str.npos){
|
|||
|
|
int p1=str.find("name:"),p2=str.find("pass:");
|
|||
|
|
name=str.substr(p1+5,p2-5);
|
|||
|
|
pass=str.substr(p2+5,str.length()-p2-4);
|
|||
|
|
string search="INSERT INTO USER VALUES (\"";
|
|||
|
|
search+=name;
|
|||
|
|
search+="\",\"";
|
|||
|
|
search+=pass;
|
|||
|
|
search+="\");";
|
|||
|
|
cout<<"sql语句:"<<search<<endl<<endl;
|
|||
|
|
mysql_query(con,search.c_str());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
最后修改 `makefile`，因为服务器要连接 MySQL，所以要加上 `-l` 来动态链接。
|
|||
|
|
|
|||
|
|
```bash
|
|||
|
|
all: test_server.cpp server.cpp server.h test_client.cpp client.cpp client.h global.h
|
|||
|
|
g++ -o test_client test_client.cpp client.cpp -lpthread
|
|||
|
|
g++ -o test_server test_server.cpp server.cpp -lpthread -lmysqlclient
|
|||
|
|
test_server: test_server.cpp server.cpp server.h global.h
|
|||
|
|
g++ -o test_server test_server.cpp server.cpp -lpthread -lmysqlclient
|
|||
|
|
test_client: test_client.cpp client.cpp client.h global.h
|
|||
|
|
g++ -o test_client test_client.cpp client.cpp -lpthread
|
|||
|
|
clean:
|
|||
|
|
rm test_server
|
|||
|
|
rm test_client
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
然后用 `make` 编译即可，接下来进行测试：
|
|||
|
|
|
|||
|
|
```bash
|
|||
|
|
make
|
|||
|
|
./test_server
|
|||
|
|
# 新开一个终端，执行如下命令
|
|||
|
|
./test_client
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
然后另开一个终端看看数据库中的 USER 表是否已经有用户数据：
|
|||
|
|
|
|||
|
|
```bash
|
|||
|
|
mysql -u root
|
|||
|
|
> use ChatProject;
|
|||
|
|
> select * from USER;
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
可见成功注册。
|
|||
|
|
|
|||
|
|
最后提交 git：
|
|||
|
|
|
|||
|
|
|