【网络编程】demo版TCP网络服务器实现
文章目录
- 一、引入
- 二、服务端实现
-
- 2.1 创建套接字socket
- 2.2 绑定bind
- 2.3 设置监听状态listen
- 2.4 获取新链接accept
- 2.5 获取信息与返回信息（文件操作）
- 三、客户端实现
-
- 3.1 创建套接字socket
- 3.2 绑定问题
- 3.3 发起链接connect
- 3.4 客户端并行
-
- 3.4.1 多进程版
- 3.4.2 多线程版
- 3.4.3 线程池版
- 四、总结
一、引入
UDP和TCP的区别：
对于TCP协议有几个特点：
1️⃣ 传输层协议
2️⃣ 有连接（正式通信前要先建立连接）
3️⃣ 可靠传输（在内部帮我们做可靠传输工作）
4️⃣ 面向字节流
对于UDP协议有几个特点：
1️⃣ 传输层协议
2️⃣ 无连接
3️⃣ 不可靠传输
4️⃣ 面向数据报
可以看到TCP对比UDP会建立链接。
其他的接口跟UDP其实没什么区别：【网络编程】demo版UDP网络服务器实现
二、服务端实现
2.1 创建套接字socket
在通信之前要先把网卡文件打开。
#include 这个函数的作用是打开一个文件，把文件和网卡关联起来。
参数介绍：
domain：一个域，标识了这个套接字的通信类型（网络或者本地）。
只用关注上面两个类，第一个AF_UNIX表示本地通信，而AF_INET表示网络通信。
type：套接字提供服务的类型。
这一章我们讲的式TCP，所以使用SOCK_STREAM。
protocol：想使用的协议，默认为0即可，因为前面的两个参数决定了，就已经决定了是TCP还是UDP协议了。
返回值：
成功则返回打开的文件描述符（指向网卡文件），失败返回-1。
而从这里我们就联想到系统中的文件操作，未来各种操作都要通过这个文件描述符，所以在服务端类中还需要一个成员变量表示文件描述符。
#pragma once
#include 2.2 绑定bind
#include 参数介绍：
socket：创建套接字的返回值。
address：通用结构体（【网络编程】socket套接字有详细介绍）。
address_len：传入结构体的长度。
所以我们要先定义一个sockaddr_in结构体填充数据，在传递进去。
然后就是跟UDP一样，先初始化结构体，再处理IP和端口。
要注意IP要绑定任意IP也就是INADDR_ANY。
至于为什么再上一章【网络编程】demo版UDP网络服务器实现有过详细讲解。
void InitServer()
{
_sock = socket(AF_INET, SOCK_STREAM, 0);
if(_sockfd == -1)
{
std::cerr << "create socket error" << std::endl;
exit(1);
}
std::cout << "create socket success" << std::endl;
struct sockaddr_in si;
// 初始化结构体
bzero(&si, sizeof si);
si.sin_family = AF_INET;
si.sin_port = htons(_port);// 主机转网络序列
si.sin_addr.s_addr = INADDR_ANY;
if(bind(_sock, (struct sockaddr*)&si, sizeof si) < 0)
{
std::cout << "bind socket error" << std::endl;
exit(1);
}
std::cout << "bind socket success" << std::endl;
}
2.3 设置监听状态listen
TCP跟UDP的不同在这里就体现了出来。
要把socket套接字的状态设置为listen状态。只有这样才能一直获取新链接，接收新的链接请求。
举个例子：
我们买东西如果出现了问题会去找客服，如果客服不在那么就回复不了，所以规定了客服在工作的时候必须要时刻接收回复消息，这个客服所处的状态就叫做监听状态。
#include 关于第二个参数backlog后边讲TCP协议的时候介绍，目前先直接用。
static const int gbacklog = 10;
void InitServer()
{
_sock = socket(AF_INET, SOCK_STREAM, 0);
if(_sock == -1)
{
std::cerr << "create socket error" << std::endl;
exit(1);
}
std::cout << "create socket success" << std::endl;
struct sockaddr_in si;
// 初始化结构体
bzero(&si, sizeof si);
si.sin_family = AF_INET;
si.sin_port = htons(_port);// 主机转网络序列
si.sin_addr.s_addr = INADDR_ANY;
if(bind(_sock, (struct sockaddr*)&si, sizeof si) < 0)
{
std::cout << "bind socket error" << std::endl;
exit(1);
}
std::cout << "bind socket success" << std::endl;
// 设置监听状态
if(listen(_sock, gbacklog) < 0)
{
std::cout << "listen socket error" << std::endl;
exit(1);
}
std::cout << "listen socket success" << std::endl;
}
2.4 获取新链接accept
上面初始化完毕，现在开始就是要运行服务端，而TCP不能直接发数据，因为它是面向链接的，必须要先建立链接。
#include 参数介绍：
sockfd文件描述符，找到套接字
addr输入输出型参数，是一个结构体，用来获取客户端的信息。
addrlen输入输出型参数，客户端传过来的结构体大小。
返回值：
成功返回一个文件描述符
失败返回-1
而我们知道sockfd本来就是一个文件描述符，那么这个返回的文件描述符是什么呢？
举个例子：
我们去吃饭的时候会发现每个店铺门口都会有人来招揽顾客，这个人把我们领进去门店后，然后他就会继续站在门口继续招揽顾客，而我们会有里面的服务员来招待我们，给我们提供服务。
这里的揽客的人就是sockfd，而服务员就是返回值的文件描述符。
意思就是sockfd的作用就是把链接从底层获取上来，返回值的作用就是跟客户端通信。
从这里就知道了成员变量中的_sock并不是通信用的套接字，而是获取链接的套接字。为了方便观察，我们可以把所有的_sock换成_listensock。
void start()
{
while(1)
{
// 获取新链接
struct sockaddr_in si;
socklen_t len = sizeof si;
int sock = accept(_listensock, (struct sockaddr*)&si, &len);
if(sock < 0)
{
// 获取链接失败无影响，继续获取即可
std::cout << "accept error, continue" << std::endl;
continue;
}
std::cout << "accept a new link success" << std::endl;
std::cout << "sock: " << sock << std::endl;
}
}
2.5 获取信息与返回信息（文件操作）
上面获取到了通信用的套接字sock，而因为TCP通信是面向字节流的，所以后续通信全部是用文件操作（IO），因为文件也是面向字节流的。
IO的操作我们可以封装一个函数。
void ServerIO(int sock)
{
char buf[1024];
// 接收消息
while(1)
{
ssize_t n = read(sock, buf, sizeof buf - 1);
if(n > 0)
{
buf[n] = '\0';
std::cout << "read a message: " << buf << std::endl;
// 把消息发送回去
std::string outbuf;
outbuf += "Server[echo]#";
outbuf += buf;
write(sock, outbuf.c_str(), outbuf.size());
}
else if(n == 0)
{
// 代表客户端退出
std::cout << "Client quit" << std::endl;
break;
}
}
}
当IO完后要记得关闭文件描述符sock，不然会导致可用描述符越来越少。
验证发现可以运行
三、客户端实现
3.1 创建套接字socket
Socket可以看成在两个程序进行通讯连接中的一个端点，一个程序将一段信息写入Socket中，该Socket将这段信息发送给另外一个Socket中，使这段信息能传送到其他程序中。
所以客户端也需要一个套接字。
void initClient()
{
_sock = socket(AF_INET, SOCK_STREAM, 0);
if(_sock < 0)
{
std::cout << "create socket error" << std::endl;
exit(1);
}
std::cout << "create socket success" << std::endl;
}
3.2 绑定问题
跟上一章一样要绑定，但是不能显示绑定。
3.3 发起链接connect
#include 参数说明：
这里的
addr和addrlen填入的是服务端信息。
在UDP通信中，客户端在sendto的时候会自动绑定IP和port，TCP这里就是在connect的时候绑定。
void start()
{
struct sockaddr_in si;
bzero(&si, sizeof si);
si.sin_family = AF_INET;
si.sin_port = htons(_serverport);
si.sin_addr.s_addr = inet_addr(_serverip.c_str());
if(connect(_sock, (struct sockaddr*)&si, sizeof si) < 0)
{
std::cout << "connect socket error" << std::endl;
}
else
{
std::string msg;
while(1)
{
std::cout << "Please Enter#";
std::getline(std::cin, msg);
write(_sock, msg.c_str(), msg.size());
// 从服务端读取数据
char buf[1024];
int n = read(_sock, buf, sizeof buf - 1);
if(n > 0)
{
buf[n] = '\0';
std::cout << buf << std::endl;
}
else
{
break;
}
}
}
}
最后在析构的时候要关掉文件描述符。
~TCPClient()
{
if(_sock >= 0)
{
close(_sock);
}
}
服务端：
客户端：
但此时如果我们在用一个客户端链接，会发现无法通信，除非第一个客户端退出。
这是因为获取新链接后会进入ServerIo中死循环。只要这个客户端不退出，就会一直给这个客户端提供服务。
那怎么保证多个客户端并行呢？
3.4 客户端并行
3.4.1 多进程版
因为fork后子进程会复制父进程的文件描述符。
这里注意子进程并不需要_listensock文件描述符，所以最好关闭。
pid_t id = fork();
if(id == 0)// child
{
close(_listensock);
ServerIO(sock);
close(sock);
exit(1);
}
接下来父进程怎么办呢？是等待吗？
如果父进程等待的话又会导致上面的情况，子进程不退出父进程就一直等待。
子进程退出时，会给父进程发送一个SIGCHLD，17号信号。所以有一种解决办法就是用signal函数，在回调函数中把waitpid的参数设置为-1（等待任意进程），就可以回收。
现在我们不用这种办法，我们可以这么写：
pid_t id = fork();
if(id == 0)// child
{
close(_listensock);
if(fork() > 0) exit(1);
ServerIO(sock);
close(sock);
exit(1);
}
// father
pid_t ret = waitpid(id, nullptr, 0);
if(ret > 0)
{
std::cout << "wait success" << ret << std::endl;
}
这里的意思就是创建孙子进程，父进程直接退出，让孙子进程执行ServerIO，此时孙子进程就会被操作系统收养，不用我们管，而父进程退出，外边的父进程也等待成功了。
结果演示：
客户端：
服务端：
其实下面的等待可以不用等待，因为SIGCHLD信号默认的处理方式是忽略。
这里看到客户端退出了但是文件描述符并没有被回收。
这里的原因是我们只关闭了子进程的文件描述符，没有关闭父进程：
3.4.2 多线程版
struct ThreadData
{
TCPServer* _self;
int _sock;
};
void start()
{
while(1)
{
// 获取新链接
struct sockaddr_in si;
socklen_t len = sizeof si;
int sock = accept(_listensock, (struct sockaddr*)&si, &len);
if(sock < 0)
{
// 获取链接失败无影响，继续获取即可
std::cout << "accept error, continue" << std::endl;
continue;
}
std::cout << "accept a new link success" << std::endl;
std::cout << "sock: " << sock << std::endl;
// 多线程
pthread_t tid;
ThreadData* td = new ThreadData({this, sock});
pthread_create(&tid, nullptr, thread_start, td);
}
}
static void* thread_start(void* args)
{
// 线程分离
pthread_detach(pthread_self());
ThreadData* tp = static_cast<ThreadData*>(args);
tp->_self->ServerIO(tp->_sock);
close(tp->_sock);
delete tp;
}
3.4.3 线程池版
前面我们写过线程池【linux】基于单例模式实现线程池，这里直接拿来用即可。
这里就要修改一下代码，因为ServerIO可以不属于类，所以可以把ServerIO放在任务Task.hpp中。
// Task.hpp
#pragma once
#include 四、总结
对比UDP服务器，TCP服务器多了获取新链接和监听的操作，而因为TCP是面向字节流的，所以接收和发送数据都是IO操作，也就是文件操作。