从0实现基于Linux socket聊天室-增加数据加密功能-6

之前更新过从0实现聊天室的5篇文章，但是数据在网络中是裸奔状态，全部是明文，本文一口君继续在原有代码基础上增加数据加密功能。

前面文章链接：

《从0实现基于Linux socket聊天室-多线程服务器模型-1》?

《从0实现基于Linux socket聊天室-多线程服务器一个很隐晦的错误-2》?

《从0实现基于Linux socket聊天室-实现聊天室的登录、注册功能-3》?

《从0实现基于Linux socket聊天室-增加公聊、私聊-4》

《从0实现基于Linux socket聊天室-增加数据库sqlite功能-5》

本文需要增加数据库功能，关于数据库的基础知识点，表的创建、增删改查等操作，以及对应的库函数的使用请参考以下3篇文章：

《嵌入式数据库sqlite3【基础篇】-基本命令操作，小白一看就懂》

《嵌入式数据库sqlite3【进阶篇】-子句和函数的使用，小白一文入门》

好了让我们开始吧。

一. 调整目录结构

为了方便编译，现在我们将前面文章的代码结构做如下调整。

root@ubuntu:/mnt/hgfs/code/chat#?tree?.
.
├──?chat_client
│???├──?include
│???├──?Makefile
│???├──?obj
│???│???└──?Makefile
│???└──?src
│???????├──?client.c
│???????└──?Makefile
├──?chat.h
├──?chat_server
│???├──?bin
│???│???└──?server
│???├──?data
│???├──?include
│???├──?Makefile
│???├──?obj
│???│???└──?server.o
│???└──?src
│???????├──?Makefile
│???????└──?server.c
└──?gcc.sh

10?directories,?15?files

服务端代码放置到chat_server目录下；客户端代码放置到chat_client目录下；

数据库相关代码放在chat_server/data下。

chat.h是所有客户端和服务器都会用到的头文件，所以放置在根目录下。

后续增加功能后，新增的头文件和C文件分别添加到对应工程目录的include和src目录下即可。

二、模块设计

1. 设计数据库表

我们之前维护的所有客户端的信息是用一个全局数组，并且没有保存功能，现在我们要把所有客户端的信息全部保存到数据库中。

数据库存储的目录

chat_server/data

数据库名：

user.db

存储用户信息的表名：

user

表user格式如下：

2. 下发keygen流程

3. 客户端发送数据

4. 客户端接收数据

三、 主要功能操作的语句及函数

数据库操作最重要的就是语句，下面讲解针对不同的功能对应的实现语句

1 创建表user

CREATE?TABLE?IF?NOT?EXISTS?user(name?TEXT?PRIMARY?KEY??NOT?NULL,passwd?TEXT?NOT?NULL,fd?INT?NOT?NULL,regist?INT??NOT?NULL);

2 增加一个用户

客户端发送注册请求后，服务器端注册用户信息到数据库中

数据库操作语句如下：

insert?into?user?values('一口Linux',?'123456',-1,?1)

功能函数如下：

int?db_add_user(char?name[],char?passwd[])

功能：
增加一个用户，执行该函数前需要先判断该用户名是否存在

参数：
name：用户名
passwd：密码

返回值：
-1：失败
1：成功

3 判断用户是否在线

客户端发送登陆命令后，服务器通过该函数判断该用户是否已经登陆成功

数据库操作语句如下：

select?fd?from?user?where?name='一口Linux'

功能函数如下：

int?db_user_if_online(char?*name,char?*passwd)

功能：
判断用户是否在线，该函数主要根据fd的值来判断用户是否在线

参数：
name：用户名
passwd：密码??

返回值：
1：在线
-1：不在线
-2：用户不存在

4 判断某个用户名是否注册

用户发送注册命令，服务器需要判断该用户名是否已经被注册过

数据库操作语句如下：

select?regist?from?user?where?name='一口Linux'

功能函数如下：

int?db_user_if_reg(char?*name)

功能：
判断某个用户名是否注册过

参数：
name：用户名

返回值：
?1：注册过
-1：没有注册过

5 判断用户名密码是否正确

用户发送登陆命令，需要判断用户名密码是否正确

数据库操作语句如下：

select?*?from?user?where?name='一口Linux'?and?passwd='123456'

功能函数如下：

int?db_user_pwd_corrct(char?*name,char*?passwd)

功能：
判断客户端发送的用户名密码是否正确

参数：
name：用户名
passwd：密码

返回值：
?1：正确
-1：用户名或者密码不正确

6 用户上线、下线

用户登陆成功后，或者发送下线申请，或者异常掉线，需要更新数据库的状态。

数据库操作语句如下：

UPDATE??user?set?fd=-1?where?name='一口Linux'

fd的值是套接字描述符，下线设置为-1，上线设置为对应的>0的值

功能函数如下：

int?db_user_on_off(int?fd,char?*name,unsigned?int?on_off)?

功能：
更新数据库中用户的fd字段

参数：
fd：套接字描述符
name：用户名
on_off：上线还是下线

返回值：
?1：正确
-1：失败

四、运行结果

界面与之前操作一致！

可以参考前面系列文章！

下面是操作视频：

五、代码说明

本实例代码已经上传到gitee仓库，
地址：https://gitee.com/wx_98fa5ee790/chat

如果代码对你有所帮助，还请给一口君一个小心心！

云盘下载的代码，log如下：

linux@ubuntu:/mnt/hgfs/share/code$?git?log
commit?478e60fc788ff9ffedf890f164e7a61a9a650d14
Author:?peng?<peng@ubuntu.(none)>
Date:???Sun?Nov?7?05:41:58?2021?-0800

??????? 1.服务器端增加数据库操作功能，数据库名称：user.db 用户表：user
????????2.基于数据库操作的注册、登录、上线、下线功能已经测试通过
????????3.增加clean功能的脚本clean.sh
????????????????????????学习Linux嵌入式请关注公众号：一口Linux

commit?597330ae0a183c9db8f68b7c9f60df94f8965778
Author:?root?<root@ubuntu.(none)>
Date:???Sat?Nov?6?09:15:40?2021?-0700

????????????????这是聊天室的初始版本V0.1
????????????????该版本包含登录、注册、公聊、私聊等功能
????????????????请关注公众号：一口Linux

切换到没有数据库的版本，执行下面命令即可。

git?reset?--hard??597330ae0a183c9db8f68b7c9f60df94f8965778

要切回有数据库的版本执行下面的命令：

git?reset?--hard?478e60fc788ff9ffedf890f164e7a61a9a650d14

获得完整代码，直接访问https://gitee.com/wx_98fa5ee790/chat