Linux中的TCP编程接口基本利用
TCP编程接口基本利用本篇先容
在UDP编程接口基本利用已经先容过UDP编程相关的接口，本篇开始先容TCP编程相关的接口。有了UDP编程的基础，明白TCP相关的接口会更加容易，下面将按照两个方向利用TCP编程接口：
[*]基本利用TCP编程接口实现服务端和客户端通讯
[*]利用TCP编程实现客户端控制服务器执行相关命令的步调
创建并封装服务端
创建服务器类
与UDP一样，起首创建服务器类的基本框架，本次计划的服务器一旦启动就不再关闭，除非手动关闭，所以可以提供两个接口：
[*]start：启动服务器
[*]stop：制止服务器
基本结构如下：
class TcpServer
{
public:
TcpServer()
{
}
// 启动服务器
void start()
{
}
// 停止服务器
void stop()
{
}
~TcpServer()
{
}
};
创建服务器套接字
创建方式与UDP基本同等，只是在socket接口的第二个参数利用SOCK_STREAM而不再是SOCK_DGRAM，代码如下：
class TcpServer
{
public:
TcpServer()
: _socketfd(-1)
{
// 创建服务器套接字
_socketfd = socket(AF_INET, SOCK_STREAM, 0);
if (_socketfd < 0)
{
LOG(LogLevel::FATAL) << "Server initiated error: " << strerror(errno);
exit(static_cast<int>(ErrorNumber::SocketFail));
}
LOG(LogLevel::INFO) << "Server initated: " << _socketfd;
}
// ...
private:
int _socketfd;// 服务器套接字
};
绑定服务器IP地址和端口
绑定方式与UDP基本同等，先利用原生的方式而不是直接利用封装后的sockaddr_in结构。在UDP编程接口基本利用部分已经提到过服务器不需要指定IP地址，所以本次一步到位，代码如下：
// 默认端口
const uint16_t default_port = 8080;
class TcpServer
{
public:
TcpServer(uint16_t port = default_port)
: // ...
, _port(port)
{
// ...
struct sockaddr_in server;
server.sin_family = AF_INET;
server.sin_port = htons(_port);
server.sin_addr.s_addr = INADDR_ANY;
int ret = bind(_socketfd, reinterpret_cast<const struct sockaddr *>(&server), sizeof(server));
if (ret < 0)
{
LOG(LogLevel::FATAL) << "Bind error：" << strerror(errno);
exit(static_cast<int>(ErrorNumber::BindSocketFail));
}
LOG(LogLevel::INFO) << "Bind Success";
}
// ...
private:
int _socketfd;// 服务器套接字
uint16_t _port; // 服务器端口
};
开启服务器监听
在UDP部分，走完上面的步骤就已经完成了基本工作，一旦服务器启动就会等候连接。但是在TCP部分则不可，由于TCP是面向连接的，也就是说，利用客户端需要连接利用TCP的客户端必须先建立连接，只有连接建立完成了才可以开始通讯。为了可以让客户端和服务端乐成建立连接，起首需要让服务器处于监听状态，此时服务器只会一直等候客户端发起连接请求
在Linux中，实现服务器监听可以利用listen接口，其原型如下：
int listen(int sockfd, int backlog);
该接口的第一个参数表现当前需要作为传输的套接字，第二个参数表现等候中的客户端的最大个数。之所以会有第二个参数是由于一旦请求连接的客户端太多但是服务器又无法快速得做出响应就会导致用户一直处于等候连接状态从而造成不必要的损失。一般情况下第二个参数不发起设置比较大，而是由于应该根据实际情况决定，但是一定不能为0，本次大小定为8
当监听乐成，该接口会返回0，否则返回-1并设置对应的错误码
在TCP中，服务器一旦被创建那么久意味着其需要开始进行监听，所以本次考虑将监听放在构造中：
// 默认最大支持排队等待连接的客户端个数
const int max_backlog = 8;
class TcpServer
{
public:
TcpServer(uint16_t port = default_port)
: _socketfd(-1), _port(port)
{
// ...
ret = listen(_socketfd, max_backlog);
if (ret < 0)
{
LOG(LogLevel::ERROR) << "Listen error：" << strerror(errno);
exit(static_cast<int>(ErrorNumber::ListenFail));
}
LOG(LogLevel::INFO) << "Listen Success";
}
// ...
};
启动服务器
在TCP中，启动服务器的逻辑和UDP的逻辑有一点差别，由于TCP服务器在启动之前先要进行监听，所以实际上此时服务器并没有进入IO状态，所以一旦启动服务器后，起首要做的就是一旦乐成建立连接就需要进入收发消息的状态
起首判断服务器是否启动，假如服务器本身已经启动就不需要再次启动，所以还是利用一个_isRunning变量作为判断条件，基本逻辑如下：
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
}
}
}
接着就是在监听乐成的情况下进入IO状态，这里利用的接口就是accept，其原型如下：
int accept(int sockfd, struct sockaddr * addr, socklen_t * addrlen);
该接口的第一个参数表现需要绑定的服务器套接字，第二个参数表现对方的套接字结构，第二个参数表现对方套接字结构的大小，此中第二个参数和第三个参数均为输出型参数
需要注意的是该接口的返回值，当函数执行乐成时，该接口会返回一个套接字，这个套接字与前面通过socket接口获取到的套接字差别。在UDP中，只有一个套接字，就是socket的返回值，但是在TCP中，由于起首需要先监听，此时需要用到的实际上是监听套接字，一旦监听乐成，才会给定用于IO的套接字。所以实际上，在TCP中，socket接口的返回值对应的是listen用的套接字，而accept的套接字就是用于IO的套接字
基于上面的概念，现在对前面的代码进行一定的修正：对于前面的成员_socketfd，应该修改为_listen_socketfd：
class TcpServer
{
public:
TcpServer(uint16_t port = default_port)
: _listen_socketfd(-1), _port(port), _isRunning(false)
{
// 创建服务器套接字
_listen_socketfd = socket(AF_INET, SOCK_STREAM, 0);
if (_listen_socketfd < 0)
{
LOG(LogLevel::FATAL) << "Server initiated error: " << strerror(errno);
exit(static_cast<int>(ErrorNumber::SocketFail));
}
LOG(LogLevel::INFO) << "Server initated: " << _listen_socketfd;
int ret = bind(_listen_socketfd, reinterpret_cast<const struct sockaddr *>(&server), sizeof(server));
// ...
ret = listen(_listen_socketfd, max_backlog);
// ...
}
// ...
private:
int _listen_socketfd; // 服务器监听套接字
// ...
};
接着，对于吸收乐成也可以创建一个成员变量_ac_socketfd，并用其吸收accept接口的返回值：
class TcpServer
{
public:
TcpServer(uint16_t port = default_port)
: // ...
, _ac_socketfd(-1)
{
// ...
}
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
struct sockaddr_in peer;
socklen_t length = sizeof(peer);
_ac_socketfd = accept(_listen_socketfd, reinterpret_cast<struct sockaddr *>(&peer), &length);
if (_ac_socketfd < 0)
{
LOG(LogLevel::WARNING) << "Accept failed：" << strerror(errno);
exit(static_cast<int>(ErrorNumber::AcceptFail));
}
LOG(LogLevel::INFO) << "Accept Success: " << _ac_socketfd;
}
}
}
// ...
private:
// ...
int _ac_socketfd; // 服务器接收套接字
// ...
};
后续的代码与UDP思绪类似，但是具体实现有些差别。由于UDP是面向数据包的，所以只能「整发整取」，但是TCP是面向字节省的，所以可以「按照需求读取」而不需要「一定完整读取」，而在文件部分，读取和写入文件也是面向字节省的，所以在TCP中，读取和写入就可以直接利用文件的读写接口。但是需要注意，由于读写不是一次性的，所以需要一个循环控制持续读和写：
// 启动服务器
void start()
{
if (!_isRunning)
{
while (true)
{
// ...
while (true)
{
// 读取客户端消息
char buffer = {0};
ssize_t ret = read(_ac_socketfd, buffer, sizeof(buffer) - 1);
if (ret > 0)
{
LOG(LogLevel::INFO) << "Client: " << inet_ntoa(peer.sin_addr) << ":" << std::to_string(ntohs(peer.sin_port)) << " send: " << buffer;
// 向客户端回消息
ret = write(_ac_socketfd, buffer, sizeof(buffer));
}
}
}
}
}
制止服务器
制止服务器和UDP思绪同等，但是需要注意，除了要关闭吸收套接字以外还需要关闭监听套接字，此处不再赘述：
=== “制止服务器函数”
// 停止服务器
void stop()
{
if (_isRunning)
{
close(_listen_socketfd);
close(_ac_socketfd);
}
}
=== “析构函数”
~TcpServer()
{
stop();
}
创建并封装客户端
创建客户端类
与UDP同等，代码如下：
class TcpClient
{
public:
TcpClient()
{
}
// 启动客户端
void start()
{
}
// 停止客户端
void stop()
{
}
~TcpClient()
{
}
};
创建客户端套接字
与UDP同等，此处不再赘述：
class TcpClient
{
public:
TcpClient()
: _socketfd(-1)
{
_socketfd = socket(AF_INET, SOCK_STREAM, 0);
if (_socketfd < 0)
{
LOG(LogLevel::FATAL) << "Client initiated error：" << strerror(errno);
exit(static_cast<int>(ErrorNumber::SocketFail));
}
LOG(LogLevel::INFO) << "Client initiated";
}
// ...
private:
int _socketfd;
};
启动客户端
由于当前是TCP，所以客户端必须先与服务端建立连接才可以进行数据传输。在Linux中，让客户端连接服务端的接口是connect，其原型如下：
int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
该接口的第一个参数表现传送数据需要的套接字，第二个参数表现服务器的套接字结构，第三个参数表现第二个参数的大小
假如该接口连接乐成大概绑定乐成，则返回0，否则返回-1并且设置错误码
需要注意，该接口会在乐成连接后自动绑定端口和IP地址，与UDP一样不需要用户手动设置客户端的IP地址和端口
由于需要用到服务器的端口和IP地址，所以在创建客户端对象时需要让用户转达IP地址和端口，所以基本代码如下：
// 默认服务器端口和IP地址
const std::string default_ip = "127.0.0.1";
const uint16_t default_port = 8080;
class TcpClient
{
public:
TcpClient(const std::string &ip = default_ip, uint16_t port = default_port)
: // ...
, _isRunning(false), _ip(ip), _port(port)
{
// ...
}
// 启动客户端
void start()
{
if (!_isRunning)
{
_isRunning = true;
// 启动后就进行connect
struct sockaddr_in server;
server.sin_family = AF_INET;
server.sin_addr.s_addr = inet_addr(_ip.c_str());
server.sin_port = htons(_port);
int ret = connect(_socketfd, reinterpret_cast<const struct sockaddr *>(&server), sizeof(server));
if (ret < 0)
{
LOG(LogLevel::WARNING) << "Connect failed" << strerror(errno);
exit(static_cast<int>(ErrorNumber::ConnectFail));
}
LOG(LogLevel::INFO) << "Connect Success: " << _socketfd;
while (true)
{
// ...
}
}
}
// ...
private:
// ...
std::string _ip; // 服务器IP地址
uint16_t _port;// 服务器端口
bool _isRunning; // 判断是否正在运行
};
在上面的代码中需要注意，不要把connect放在循环里，由于建立连接需要一次而不需要每一次发送都建立连接
接着就是写入和读取消息，基本思绪与UDP相同，代码如下：
// 启动客户端
void start()
{
if (!_isRunning)
{
// ...
while (true)
{
// 向服务器写入
std::string message;
std::cout << "请输入消息：";
std::getline(std::cin, message);
ssize_t ret = write(_socketfd, message.c_str(), message.size());
// 收到消息
char buffer = {0};
ret = read(_socketfd, buffer, sizeof(buffer));
if (ret > 0)
LOG(LogLevel::INFO) << "收到服务器消息：" << buffer;
}
}
}
制止客户端
制止客户端的思绪与UDP同等，此处不再赘述：
=== “制止客户端函数”
// 停止客户端
void stop()
{
if (_isRunning)
close(_socketfd);
}
=== “析构函数”
~TcpClient()
{
stop();
}
本地通讯测试
测试步骤：
[*]先启动服务端，再启动客户端
[*]客户端向服务器端发送信息
测试目的：
[*]客户端可以正常向服务器端发送信息
[*]服务端可以正常显示客户端信息并正常向客户端返回客户端发送的信息
[*]客户端可以正常显示服务端回复的信息
测试代码如下：
=== “客户端”
#include "tcp_client.hpp"
#include <memory>
using namespace TcpClientModule;
int main(int argc, char *argv[])
{
std::shared_ptr<TcpClient> tcp_client;
if (argc == 1)
{
// 使用默认端口和IP地址
tcp_client = std::make_shared<TcpClient>();
}
else if (argc == 3)
{
std::string ip = argv;
std::uint16_t port = std::stoi(argv);
// 使用自定义端口和IP地址
tcp_client = std::make_shared<TcpClient>(ip, port);
}
else
{
LOG(LogLevel::ERROR) << "错误使用，正确使用为：" << argv << " IP地址 端口号（或者二者都不存在）";
exit(7);
}
tcp_client->start();
tcp_client->stop();
return 0;
}
=== “服务端”
#include "tcp_server.hpp"
#include <memory>
using namespace TcpServerModule;
int main(int argc, char *argv[])
{
std::shared_ptr<TcpServer> tcp_server;
if (argc == 1)
{
// 使用默认的端口
tcp_server = std::make_shared<TcpServer>();
}
else if (argc == 2)
{
// 使用自定义端口
std::string port = argv;
tcp_server = std::make_shared<TcpServer>(port);
}
else
{
LOG(LogLevel::ERROR) << "错误使用，正确方式：" << argv << " 端口（可以省略）";
exit(6);
}
tcp_server->start();
tcp_server->stop();
return 0;
}
本次计划的客户端支持用户从命令行输入端口和IP地址，否则就直接利用默认，下面是一种结果：
https://i-blog.csdnimg.cn/direct/cd0513458cdd4b33aef3f0135c6749f9.png
客户端退出但服务端没有退出的问题
在UDP中，假如客户端退出但服务端没有退出，下一次客户端再连接该服务端时不会出现问题。但是在TCP中就并不是这样，例如：
https://i-blog.csdnimg.cn/direct/dac67066eff84c4f9c2360cf112f3ee6.png
从上图可以看到，假如客户端连接后断开再连接就会出现第二次连接发送消息无法得到回应。之所以出现这个问题就是由于服务器卡在了读写死循环中，办理这个问题的方式很简单，只需要判断read接口返回值是否为0，假如为0，阐明当前服务器并没有读取到任何内容，直接退出即可：
// 启动服务器
void start()
{
if (!_isRunning)
{
// ...
while (true)
{
// ...
while(true)
{
// 读取客户端消息
char buffer = {0};
ssize_t ret = read(_ac_socketfd, buffer, sizeof(buffer) - 1);
if (ret > 0)
{
// ...
}
else if (ret == 0)
{
LOG(LogLevel::INFO) << "Client disconnected: " << _ac_socketfd;
break;
}
}
}
}
}
此时便可以办理上面的问题：
https://i-blog.csdnimg.cn/direct/40b7b4f1336d45af92f2e1c01f94bed1.png
文件描述符走漏问题
在上面的测试结果中可以发现，当客户端退出后再重新连接服务端，此时的文件描述符由4变成了5，但是实际上文件描述符是非常有限的，对于一般的用户机来说，文件描述符最大为1024，而服务器一般为65535，利用下面的指令可以查看：
ulimit -a
在结果中的open files一栏即可看到值
既然客户端已经退出了，那么对应的文件描述符就应该关闭而不是持续被占用着，此时就出现了文件描述符走漏问题。办理这个问题很简答，只需要在判断读取结果小于0时关闭文件描述符再退出即可：
// 启动服务器
void start()
{
if (!_isRunning)
{
// ...
while (true)
{
// ...
// ...
close(_ac_socketfd);
}
}
}
测试云服务器与本地进行通讯
相同操作体系（客户端和服务端均为Linux）
测试云服务器与本地进行通讯最直接的步骤如下：
[*]将服务端步调拷贝到云服务器
[*]本地作为客户端，通过云服务器的公网IP地址连接云服务器的服务端
[*]客户端向云服务器发送信息
具体操作步骤与UDP类似，下面直接展示结果：
https://i-blog.csdnimg.cn/direct/a8cc1bbb828641e584824465b234d44e.png
与UDP一样需要注意安全组的问题，以阿里云为例，设置结果如下：
https://i-blog.csdnimg.cn/direct/c40af2c249ac43a888006c7e83883f85.png
差别操作体系（客户端为Windows，服务端为Linux）
由于Windows中利用接口和Linux中差不多，所以不会再详细先容，下面直接给出Windows客户端代码：
#include <winsock2.h>
#include <iostream>
#include <string>
#pragma warning(disable : 4996)
#pragma comment(lib, "ws2_32.lib")
std::string serverip = "47.113.217.80";// 填写云服务器IP地址
uint16_t serverport = 8888; // 填写云服务开放的端口号
int main()
{
WSADATA wsaData;
int result = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (result != 0)
{
std::cerr << "WSAStartup failed: " << result << std::endl;
return 1;
}
SOCKET clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (clientSocket == INVALID_SOCKET)
{
std::cerr << "socket failed" << std::endl;
WSACleanup();
return 1;
}
sockaddr_in serverAddr;
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(serverport); // 替换为服务器端口
serverAddr.sin_addr.s_addr = inet_addr(serverip.c_str()); // 替换为服务器IP地址
result = connect(clientSocket, (SOCKADDR *)&serverAddr, sizeof(serverAddr));
if (result == SOCKET_ERROR)
{
std::cerr << "connect failed" << std::endl;
closesocket(clientSocket);
WSACleanup();
return 1;
}
while (true)
{
std::string message;
std::cout << "Please Enter@ ";
std::getline(std::cin, message);
if(message.empty()) continue;
send(clientSocket, message.c_str(), message.size(), 0);
char buffer = {0};
int bytesReceived = recv(clientSocket, buffer, sizeof(buffer) - 1, 0);
if (bytesReceived > 0)
{
buffer = '\0'; // 确保字符串以 null 结尾
std::cout << "Received from server: " << buffer << std::endl;
}
else
{
std::cerr << "recv failed" << std::endl;
}
}
closesocket(clientSocket);
WSACleanup();
return 0;
}
运行结果如下：
https://i-blog.csdnimg.cn/direct/264f5a00d230457583274f1772f7e259.png
多个客户端同时连接服务器
在上面已经测试过一个客户端连接一个服务端，接下来测试多个客户端连接服务端
基本现象
利用本地虚拟机和云服务器的客户端本地连接云服务器的服务端：
先利用虚拟机大概云服务器的客户端连接服务端：
https://i-blog.csdnimg.cn/direct/1bc655e766c74decaa6f14daa01a66cb.png
可以看到正常连接，但是此时假如云服务器本地客户端连接云服务器的服务端：
https://i-blog.csdnimg.cn/direct/aebbd6edc9214b78bfe5699524b82d45.png
此时就会发现，尽管云服务器客户端提示连接乐成，但是服务器却没有显示吸收。假如云服务器的客户端向服务器发送消息也不回得到回应：
https://i-blog.csdnimg.cn/direct/b858e3cd78ab4d8fb232eadfe123de7d.png
假如终断虚拟机的连接，此时服务器又会显示连接乐成：
https://i-blog.csdnimg.cn/direct/11c5731927a1434589e0a5879af36e28.png
之所以会出现这个问题就是由于在上面的逻辑中：只有吸收乐成了才会发送消息，而一旦吸收乐成后，就在写入和读取中死循环，此时就导致accept不能继承吸收。办理这个问题就需要考虑到利用子进程大概新线程，将吸收和读写分别放在两个执行进程大概执行流中，根据这个思绪下面提供三种办理方案：
[*]子进程版本
[*]新线程版本
[*]线程池版本
子进程版本
计划子进程版本的本质就是让子进程执行读写方法，先将读写逻辑抽离到一个函数中：
=== “读写函数”
// 读写函数
void read_write_msg(struct sockaddr_in peer)
{
while (true)
{
// 读取客户端消息
char buffer = {0};
ssize_t ret = read(_ac_socketfd, buffer, sizeof(buffer) - 1);
if (ret > 0)
{
LOG(LogLevel::INFO) << "Client: " << inet_ntoa(peer.sin_addr) << ":" << std::to_string(ntohs(peer.sin_port)) << " send: " << buffer;
// 向客户端回消息
ret = write(_ac_socketfd, buffer, sizeof(buffer));
}
else if (ret == 0)
{
LOG(LogLevel::INFO) << "Client disconnected: " << _ac_socketfd;
break;
}
}
close(_ac_socketfd);
}
=== “启动服务器函数”
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
struct sockaddr_in peer;
socklen_t length = sizeof(peer);
_ac_socketfd = accept(_listen_socketfd, reinterpret_cast<struct sockaddr *>(&peer), &length);
if (_ac_socketfd < 0)
{
LOG(LogLevel::WARNING) << "Accept failed：" << strerror(errno);
exit(static_cast<int>(ErrorNumber::AcceptFail));
}
LOG(LogLevel::INFO) << "Accept Success: " << _ac_socketfd;
// 读写逻辑
read_write_msg(peer);
}
}
}
接着，为了让子进程执行对应的任务，起首就是创建一个子进程，此处直接利用原生接口：
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
// ...
// 创建子进程
pid_t pid = fork();
if (pid == 0)
{
// 子进程
// 读写逻辑
read_write_msg(peer);
exit(0);
}
}
}
}
但是，这样写还不敷以办理问题，在Linux进程间通讯提到子进程会拷贝父进程描述符表，此时同样会导致文件描述符走漏问题，所以父进程和子进程都需要关闭本身不需要的文件描述符：对于父进程来说，其需要关闭读写用的文件描述符，由于写入和读取交给了子进程；对于子进程来说，其需要关闭监听用的文件描述符，由于继承监听其他客户端的连接由父进程进行
基于上面的思绪，代码如下：
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
// ...
// 创建子进程
pid_t pid = fork();
if (pid == 0)
{
// 子进程
// 关闭监听文件描述符
close(_listen_socketfd);
// ...
}
// 父进程关闭读写描述符
close(_ac_socketfd);
}
}
}
一旦创建了子进程，父进程就需要对其进行等候并回收，假如不回收就会导致内存走漏问题，回收子进程的方式目前有下面两种：
[*]利用wait和waitpid接口进行等候
[*]借助子进程退出时发送的SIGCHILD信号，利用SIG_IGN行为
但是本次不利用上面的任意一种，而是考虑让子进程再创建一个子进程，一旦创建乐成就让当前子进程退出，而让新创建的子进程（孙子进程）继承执行后续的代码，由于当前子进程已经退出并且退出前并没有回收新创建的子进程（孙子进程），所以当前孙子进程就会被操作体系托管变成孤儿进程，一旦孙子进程走到了读写逻辑下面的exit(0)就会退出，此时操作体系就会自动回收这个孙子进程。这个思绪也被称为「双重fork（大概保卫进程化））」。所以，代码如下：
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
// ...
// 创建子进程
pid_t pid = fork();
if (pid == 0)
{
// 子进程
// ...
// 创建孙子进程
if (fork())
exit(0); // 当前子进程执行exit(0)
// 孙子进程从此处继续向后执行
// 读写逻辑
read_write_msg(peer);
exit(0);
}
// ...
}
}
}
现在，再进行上面的测试可以发现问题已包办理：
https://i-blog.csdnimg.cn/direct/b000523b861a4e7c9c6f7fc2e3cb159b.png
https://i-blog.csdnimg.cn/direct/ec796e224d44432f8a81eed0317ec24e.png
新线程版本
由于所有线程共享一个文件描述符表，所以不需要手动关闭一些文件描述符，下面利用前面封装的线程进行演示：
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
// ...
// // 父进程关闭读写描述符
// close(_ac_socketfd);
// 创建新线程
Thread t(std::bind(&TcpServer::read_write_msg, this, peer));
t.start();
}
}
}
测试之后可以发现和子进程测试的效果一样，此处不再展示
线程池版本
线程池版本和新线程版本的思绪非常类似，给出代码不再演示：
using task_t = std::function<void()>;
// ...
class TcpServer
{
public:
TcpServer(uint16_t port = default_port)
: _listen_socketfd(-1), _port(port), _isRunning(false), _ac_socketfd(-1)
{
// 创建服务器套接字
// ...
// 绑定
// ...
// 创建线程池
_tp = ThreadPool<task_t>::getInstance();
// 启动线程池
_tp->startThreads();
// 监听
// ...
}
// ...
// 启动服务器
void start()
{
if (!_isRunning)
{
_isRunning = true;
while (true)
{
// ...
// version-3
_tp->pushTasks(std::bind(&TcpServer::read_write_msg, this, peer));
}
}
}
// 停止服务器
void stop()
{
if (_isRunning)
{
_tp->stopThreads();
// ...
}
}
~TcpServer()
{
stop();
}
private:
// ...
std::shared_ptr<ThreadPool<task_t>> _tp;
// ...
};
客户端控制服务器执行相关命令的步调
思绪分析
既然需要客户端控制服务器执行命令就必须要履历下面的步骤：
[*]客户端将命令字符串发送给服务端
[*]服务端创建子进程，利用进程间通讯将分析后的命令交给子进程，子进程调用exec家族函数将命令执行的结果通过服务器发送给客户端
实现
由于服务端本身就是进行吸收和返回结果，所以考虑将命令执行单独作为一个类来描述，本次为了执行的安全，考虑只允许用户执行部分命令，并且提供判断命令是否是合法命令，所以少不了需要查询的接口，为了更快速的查询，可以利用set集合。另外，由于要执行命令，所以需要一个成员函数executeCommand执行对应的命令，所以基本结构如下：
class Command
{
Command()
{
// 构造可以执行的一些命令
_commands.insert("ls");
_commands.insert("pwd");
_commands.insert("ll");
_commands.insert("touch");
_commands.insert("who");
_commands.insert("whoami");
}
// 判断命令是否合法
bool isValid(std::string cmd)
{
auto pos = _commands.find(cmd);
if (pos == _commands.end())
return false;
return true;
}
// 执行命令
std::string executeCommand(const std::string &cmd)
{
}
~Command()
{
}
private:
std::set<std::string> _commands;
};
接着，改变服务端的读写任务的接口，此处不再利用文件的read和write接口，而是利用recv和send接口，这两个接口只是比read和write多了flags，其余都一样，并且目前情况下flags设置为0即可：
// 读写函数
void read_write_msg(struct sockaddr_in peer)
{
while (true)
{
// 读取客户端消息
char buffer = {0};
ssize_t ret = recv(_ac_socketfd, buffer, sizeof(buffer) - 1, 0);
if (ret > 0)
{
LOG(LogLevel::INFO) << "Client: " << inet_ntoa(peer.sin_addr) << ":" << std::to_string(ntohs(peer.sin_port)) << " send: " << buffer;
// 向客户端回消息
Command cmd;
if (cmd.isValid(buffer))
{
// 命令合法可以执行命令
std::string ret = cmd.executeCommand(buffer);
send(_ac_socketfd, ret.c_str(), ret.size(), 0);
}
else
{
send(_ac_socketfd, "错误指令", sizeof("错误指令"), 0);
}
}
// ...
}
// ...
}
接下来就是实现执行命令函数，根据前面的分析需要创建子进程调用exec家族函数执行对应的命令，但是在尺度库中有对应的接口已经实现了这个功能：popen，其原型如下：
FILE *popen(const char *command, const char *type);
对应的接口就是pclose接口，原型如下：
int pclose(FILE *stream);
对于popen接口来说，其会对传入的命令进行分析并创建子进程执行，将执行结果放到返回值中，由于FILE是文件结构，所以只需要利用文件的读写接口即可读取到此中的内容，这个接口第二个参数表现读模式大概写模式，由于是执行命令，所以只需要填入"r"即可
结合上面的接口即可完成对应的执行命令函数：
std::string executeCommand(const std::string &cmd)
{
FILE *fp = popen(cmd.c_str(), "r");
if (fp == nullptr)
return std::string();
char buffer;
std::string result;
while (fgets(buffer, sizeof(buffer), fp))
{
result += buffer;
}
pclose(fp);
return result;
}
!!! note
fgets会自动添加\0，不需要预留\0的位置
测试
服务端主函数代码和客户端主函数代码稳定，下面是测试结果：
https://i-blog.csdnimg.cn/direct/48dcdae063e14e40919806e40ef0ac6f.png
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]