【Linux】Socket编程-TCP构建自己的C++服务器

ops/2025/1/18 6:56:49/

🌈 个人主页:Zfox_
🔥 系列专栏:Linux

目录

  • 一:🔥 Socket 编程 TCP
    • 🦋 TCP socket API 详解
    • 🦋 多线程远程命令执行
    • 🦋 网络版计算器(应用层自定义协议与序列化)
  • 二:🔥 共勉

一:🔥 Socket 编程 TCP

🦋 TCP socket API 详解

下面介绍程序中用到的 socket API,这些函数都在 sys/socket.h 中
socket

#include <sys/types.h>
#include <sys/socket.h>// 创建 socket 文件描述符 (TCP/UDP, 客户端 + 服务器)
int socket(int domain, int type, int protocol);domain:/ 协议家族AF_INET      IPv4 Internet protocolsAF_INET6     IPv6 Internet protocolstype: 报文类型SOCK_DGRAM      Supports datagrams (connectionless, unreliable messages of a fixed maximum length).SOCK_STREAM     Provides sequenced, reliable, two-way, connection-based byte streams.  An out-of-band data transmission mechanism may be supported.protocol: 传输层类型默认为0On success, a file descriptor for the new socket is returned.  On error, -1 is returned, and errno is set appropriately.

bind

#include <sys/types.h>
#include <sys/socket.h>// 绑定端口号 (TCP/UDP, 服务器)
int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);On success, zero is returned.  On error, -1 is returned, and errno is set appropriately.// 2. 填充网络信息,并bind绑定
// 2.1 没有把socket信息设置进入内核
struct sockaddr_in local;
bzero(&local, sizeof(local));       // string.h
local.sin_family = AF_INET;
local.sin_port = ::htons(_port);   // 要被发送给对方,既要发送到网络中!   主机序列转换为网络序列 大小端转换 网络中都是大端 #include <arpa/inet.h>
local.sin_addr.s_addr = ::inet_addr(_ip.c_str());    // 1. string ip -> 4bytes 2. network order   #include <sys/socket.h>    #include <netinet/in.h>  #include <arpa/inet.h>
local.sin_addr.s_addr = INADDR_ANY;// 2.1 bind    这里设置进入内核
int n = ::bind(_sockfd, _addr.NetAddr(), _addr.NetAddrLen());

📚 我们之前在调用socket的时候,明明已经填充了一次 AF_INET, 为什么这里还需要一次呢?

创建套接字的时候填充的 AF_INET 是给操作系统文件系统里的网络文件接口,告诉我们的操作系统我们要创建一个网络的套接字。

这里则是用来填充 sockaddr_in 网络信息,只有套接字的结构和这里的结构一样,操作系统才能绑定成功。

📚 必带四件套

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

listen

#include <sys/types.h>          
#include <sys/socket.h>int listen(int sockfd, int backlog);sockfd:			 指定的套接字
backlog				 等待连接队列的最大长度。On success, zero is returned.  On error, -1 is returned, and errno is set appropriately.

listen() 声明 sockfd 处于监听状态, 并且最多允许有 backlog 个客户端处于连接等待状态, 如果接收到更多的连接请求就忽略, 这里设置不会太大 (一般是 5)

accept

#include <sys/types.h>      
#include <sys/socket.h>int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);sockfd:			 指定的套接字
  • 三次握手完成后, 服务器调用 accept() 接受连接;
  • 如果服务器调用 accept()时还没有客户端的连接请求,就阻塞等待直到有客户端连接上来;
  • addr 是一个传出参数,accept()返回时传出客户端的地址和端口号;
  • 如果给 addr 参数传 NULL,表示不关心客户端的地址;
  • addrlen 参数是一个传入传出参数(value-result argument), 传入的是调用者提供的, 缓冲区 addr 的长度以避免缓冲区溢出问题, 传出的是客户端地址结构体的实际长度(有可能没有占满调用者提供的缓冲区);

connect

#include <sys/types.h>       
#include <sys/socket.h>int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);sockfd:			 指定的套接字
  • 客户端需要调用 connect()连接服务器;
  • connect 和 bind 的参数形式一致, 区别在于 bind 的参数是自己的地址, 而 connect 的参数是对方的地址;
  • connect() 成功返回 0,出错返回-1

🦋 多线程远程命令执行

📚 代码结构

C++
CommandExec.hpp  Common.hpp  Cond.hpp  InetAddr.hpp  Log.hpp  Makefile  
Mutex.hpp  TcpClient.cc  TcpServer.cc  TcpServer.hpp  Thread.hpp  ThreadPool.hpp

TcpServer.hpp

#pragma once#include <iostream>
#include <cstring>
#include <string>
#include <cerrno>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/wait.h>
#include <signal.h>
#include <pthread.h>#include "Log.hpp"
#include "Common.hpp"
#include "InetAddr.hpp"
#include "ThreadPool.hpp"using namespace LogModule;
using namespace ThreadPoolModule;static const uint16_t gport = 8080;
using handler_t = std::function<std::string(std::string)>;#define BACKLOG 8class TcpServer
{using task_t = std::function<void()>;struct ThreadData{int sockfd;TcpServer *self;};
public:TcpServer(handler_t handler, int port = gport): _handler(handler), _port(port), _isrunning(false){}bool InitServer(){// 1. 创建tcp socket_listensockfd = ::socket(AF_INET, SOCK_STREAM, 0); // Tcp Socketif (_listensockfd < 0){LOG(LogLevel::FATAL) << "_listensockfd error";Die(SOCKET_ERR);}LOG(LogLevel::INFO) << "_listensockfd create success, _listensockfd is : " << _listensockfd;struct sockaddr_in local;memset(&local, 0, sizeof(local));local.sin_family = AF_INET;local.sin_port = htons(_port);local.sin_addr.s_addr = INADDR_ANY;// 2. bindint n = ::bind(_listensockfd, CONV(&local), sizeof(local));if (n < 0){LOG(LogLevel::FATAL) << "bind error";Die(BIND_ERR);}LOG(LogLevel::INFO) << "bind success, _listensockfd is : " << _listensockfd;// 3. cs tcp是面向连接的,就要求tcp随时随地等待被连接// tcp 需要将socket设置成为监听状态n = ::listen(_listensockfd, BACKLOG);if (n < 0){LOG(LogLevel::FATAL) << "listen error";Die(LISTEN_ERR);}LOG(LogLevel::INFO) << "listen success, _listensockfd is : " << _listensockfd;//::signal(SIGCHLD, SIG_IGN);     // 子进程退出,OS会自动回收资源,不用再wait了return true;}void HandlerRequest(int sockfd)   // TCP 也是全双工通信{LOG(LogLevel::INFO) << "HandlerRequest, sockfd is : " << sockfd;char inbuffer[4096];// 长任务while(true){ssize_t n = ::recv(sockfd, inbuffer, sizeof(inbuffer) - 1, 0);if(n > 0){LOG(LogLevel::INFO) << inbuffer;inbuffer[n] = 0;// std::string echo_str = "server echo# ";// echo_str += inbuffer; std::string cmd_result = _handler(inbuffer);::send(sockfd, cmd_result.c_str(), cmd_result.size(), 0);}else if(n == 0){// read 如果读取返回值是0,表示client退出LOG(LogLevel::INFO) << "client quit: " << sockfd;break;}else {// 读取失败了break;}}::close(sockfd);   // fd泄露问题}static void *ThreadEntry(void *args){pthread_detach(pthread_self());ThreadData* data = (ThreadData*)args;data->self->HandlerRequest(data->sockfd);delete data;return nullptr;}void Start(){_isrunning = true;while (_isrunning){// 不能直接读取数据// 1. 获取新连接struct sockaddr_in peer;socklen_t peerlen = sizeof(peer);LOG(LogLevel::DEBUG) << "accept ing ...";// 我们要获取客户端的信息:数据(sockfd) + client socket信息(accept)int sockfd = ::accept(_listensockfd, CONV(&peer), &peerlen);if (sockfd < 0){LOG(LogLevel::WARNING) << "accept error" << strerror(errno);continue;}// 获取连接成功了LOG(LogLevel::INFO) << "accept success, socket is : " << sockfd;InetAddr addr(peer);LOG(LogLevel::INFO) << "client info: " << addr.Addr();// version-0// HandlerRequest(sockfd);// version-1 多进程版本// pid_t id = fork();// if(id == 0)// {//     // child//     // 问题1: 父进程的文件描述符表子进程会继承 父子各一张共两张 //     // 1.关闭不需要的fd//     ::close(_listensockfd);//     if(fork() > 0) exit(0); // 子进程退出//     // 孙子进程 -> 孤儿进程 -> 1//     HandlerRequest(sockfd);//     exit(0);// }// ::close(sockfd);    // 父进程也关闭不需要的 已经交给子进程了// // 不会阻塞// pid_t rid = ::waitpid(id, nullptr, 0);// if(rid < 0)// {//     LOG(LogLevel::WARNING) << "waitpid error";// }// version-2 多线程版本// pthread_t tid;// ThreadData* data = new ThreadData;// data->sockfd = sockfd;// data->self = this;// pthread_create(&tid, nullptr, ThreadEntry, data);    // 主线程和新线程是如何看待,文件描述符表, 共享一张文件描述符表!!属于同一个进程 !// version-3 线程池版本  一般用于短任务(注册登录),少量用户// task_t f = std::bind(&TcpServer::HandlerRequest, this, sockfd);     // 构建任务// ThreadPool<task_t>::getInstance()->Equeue(f);ThreadPool<task_t>::getInstance()->Equeue([this, sockfd](){this->HandlerRequest(sockfd);});}}void Stop(){_isrunning = false;}~TcpServer(){}private:int _listensockfd; // 监听socketuint16_t _port;bool _isrunning;// 处理上层任务的入口handler_t _handler;
};

TcpServer.cc

#include "TcpServer.hpp"
#include "CommandExec.hpp"
#include <memory>using namespace LogModule;int main()
{ENABLE_CONSOLE_LOG();Command cmd;std::unique_ptr<TcpServer> tsvr = std::make_unique<TcpServer>([&cmd](std::string cmdstr){return cmd.Execute(cmdstr);});tsvr->InitServer();tsvr->Start();return 0;
}

TcpClient.cc

#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <cstring>// ./client_tcp serverip serverport
int main(int argc, char *argv[])
{if(argc != 3){std::cout << "Usage: " << argv[0] << " serverip serverport" << std::endl;return 1;}std::string serverip = argv[1];int server_port = std::stoi(argv[2]);int sockfd = socket(AF_INET, SOCK_STREAM, 0);if(sockfd < 0){std::cout << "Create socket failed." << std::endl;return 2;}struct sockaddr_in server_addr;memset(&server_addr, 0, sizeof(server_addr));server_addr.sin_family = AF_INET;server_addr.sin_port = htons(server_port);    server_addr.sin_addr.s_addr = inet_addr(serverip.c_str());// client 不需要显示的进行bind, tcp是面向连接的, connect 底层自动会进行bindint n = ::connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));if(n < 0){std::cout << "Connect to server failed." << std::endl;return 3;}// echo clientstd::string message;while(true){char inbuffer[1024];std::cout << "input message: ";std::getline(std::cin, message);n = ::write(sockfd, message.c_str(), message.size());if(n > 0){int m = ::read(sockfd, inbuffer, sizeof(inbuffer));if(m > 0){inbuffer[m] = 0;std::cout << inbuffer << std::endl;}else break;}else break;}::close(sockfd);return 0;   
}

CommandExec.hpp

#pragma once#include <iostream>
#include <string>
#include <cstdio>
#include <set>const int line_size = 1024;class Command
{
public:Command(){_white_list.insert("ls");_white_list.insert("pwd");_white_list.insert("ls -l");_white_list.insert("who");_white_list.insert("whoami");_white_list.insert("ll");}bool SafeCheck(const std::string& cmdstr){auto iter = _white_list.find(cmdstr);return iter == _white_list.end() ? false : true;}// 给你一个命令字符串"ls -l",执行它并返回执行结果std::string Execute(std::string cmdstr){// 1. pope// 2.fork + dup2(pipe[1], 1) + exec*, 执行结果给父进程, pipe[0]// 3. return // FILE *popen(const cahr *command, const char *type);// pclose(FILE *stream);if(!SafeCheck(cmdstr)){return std::string(cmdstr + "不支持");}FILE *fp = popen(cmdstr.c_str(), "r");if(fp == nullptr){return std::string("Failed");}char buffer[line_size];std::string result;while(true){char *ret = ::fgets(buffer, sizeof(buffer), fp);if(!ret) break;result += ret;}pclose(fp);return result.empty() ? std::string("Done") : result;}
private:std::set<std::string> _white_list;
};

CommandExec.hpp

#pragma once#include <iostream>
#include <string>
#include <cstdio>
#include <set>const int line_size = 1024;class Command
{
public:Command(){_white_list.insert("ls");_white_list.insert("pwd");_white_list.insert("ls -l");_white_list.insert("who");_white_list.insert("whoami");_white_list.insert("ll");}bool SafeCheck(const std::string& cmdstr){auto iter = _white_list.find(cmdstr);return iter == _white_list.end() ? false : true;}// 给你一个命令字符串"ls -l",执行它并返回执行结果std::string Execute(std::string cmdstr){// 1. pope// 2.fork + dup2(pipe[1], 1) + exec*, 执行结果给父进程, pipe[0]// 3. return // FILE *popen(const cahr *command, const char *type);// pclose(FILE *stream);if(!SafeCheck(cmdstr)){return std::string(cmdstr + "不支持");}FILE *fp = popen(cmdstr.c_str(), "r");if(fp == nullptr){return std::string("Failed");}char buffer[line_size];std::string result;while(true){char *ret = ::fgets(buffer, sizeof(buffer), fp);if(!ret) break;result += ret;}pclose(fp);return result.empty() ? std::string("Done") : result;}
private:std::set<std::string> _white_list;
};

🦋 网络版计算器(应用层自定义协议与序列化)

代码结构

C++
Calculator.hpp  Common.hpp  Cond.hpp  Deamon.hpp  InetAddr.hpp  Log.hpp  Makefile  Mutex.hpp  
Protocol.hpp  TcpClient.cc  TcpServer.cc  TcpServer.hpp  Thread.hpp  ThreadPool.hpp
// 简单起见, 可以直接采用自定义线程
// 直接 client<<->>server 通信, 这样可以省去编写没有干货的代码

网络版计算器(应用层自定义协议与序列化)

二:🔥 共勉

以上就是我对 【Linux】Socket编程-TCP构建自己的C++服务器 的理解,想要完整代码可以私信博主噢!觉得这篇博客对你有帮助的,可以点赞收藏关注支持一波~😉
在这里插入图片描述


http://www.ppmy.cn/ops/151022.html

相关文章

2025第3周 | json-server的基本使用

目录 1. json-server是什么&#xff1f;2. json-server怎么用&#xff1f;2.1 安装2.2 创建db.json2.3 启动服务2.4 查看效果 3. 前端进行模拟交互3.1 创建demo.html3.2 创建demo.js 2025&#xff0c;做想做的事&#xff0c;读想读的书&#xff0c;持续学习&#xff0c;自律生活…

【华为路由/交换机的telnet远程设置】

华为路由/交换机的telnet远程设置 R1&#xff1a;10.1.1.254 R2&#xff1a;10.1.1.1 R3&#xff1a;10.1.1.2 S1&#xff1a;空配 在R2上设置密码为huawei 用R1和R3远程后&#xff0c;在R2上可以查看已经登陆的用户信息 但是这种远程方式的权限非常低&#xff0c;vty用户界…

C语言中char str和char str[]的区别

char* str和char* str[]的区别&#xff1a;C语言中char *str[] 和char *str有什么区别-CSDN博客 char str 和 char str[] 在 C 语言中也有不同的含义和用途&#xff0c;以下是它们的区别&#xff1a; 1. char str 类型&#xff1a;这是一个单一的字符变量。 用途&#xff1a…

5步打造完善的物联网IoT测试体系

前言 智能硬件和软件测试规范涉及概述用于测试系统功能、性能和可靠性的详细说明和标准&#xff0c;以帮助确保系统的可靠性和质量。 请记住&#xff0c;测试是一个迭代过程&#xff0c;需要根据项目不断发展的性质进行调整。 一、物联网测试流程概览 测试基本流程 物联网测试…

怎么在iPhone手机上使用便签进行记录?

宝子们&#xff0c;在这个快节奏的时代&#xff0c;灵感的火花总是一闪而过&#xff0c;待办事项也常常让人应接不暇。好在咱们的 iPhone手机便签超给力&#xff0c;能满足各种记录需求&#xff01;今天就来给大家分享一下&#xff0c;如何在 iPhone 手机上巧用便签&#xff0c…

NLP DAY1: 文本数据读取

实验介绍 在自然语言处理的第一步&#xff0c;我们要面对的是各种各样以不同形式表现的文本数据&#xff0c;比如&#xff0c;有的是纯 txt 文档&#xff0c;有的是存储在 Excel 中的表格数据&#xff0c;还有的是无法直接打开的 pkl 文件等。刚开始接触编程的同学在这里就犯难…

通过ffmpeg将FLV文件转换为MP4

使用 ffmpeg 将 FLV 文件转换为 MP4 文件是一个常见的操作。ffmpeg 是一个强大的多媒体处理工具&#xff0c;支持多种格式的转换、剪辑、合并等操作。以下是详细的步骤和命令示例&#xff0c;帮助你完成这一任务。 安装 FFmpeg 如果你还没有安装 ffmpeg&#xff0c;可以根据你…

windows-本地部署Git仓库-安装Gitea

windows-本地部署Git仓库-安装Gitea 初始化MysQL数据库下载运行后关闭配置服务初始化打开防火墙指定端口入站规则 初始化MysQL数据库 create database gitea character set utf8mb4; 下载 运行后关闭 配置服务 初始化 打开防火墙指定端口入站规则