【计算机网络学习之路】TCP socket编程

文章目录

  • 前言
  • 一. 服务器
    • 1. 初始化服务器
    • 2. 启动服务器
  • 二. 客户端
  • 三. 多进程服务器
  • 结束语

前言

本系列文章是计算机网络学习的笔记，欢迎大佬们阅读，纠错，分享相关知识。希望可以与你共同进步。

本篇博客基于UDP socket基础，介绍TCP socket编程接口和细节

UDP socket编程可参看【计算机网络学习之路】UDP socket编程

本次编写的服务器和客户端依然是最简单的echo服务器

一. 服务器

服务器的基本框架：

tcp_server.hpp

#pragma once#include <iostream>
#include <string>
#include <cstring>
#include <cerrno>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>namespace ns_server
{const uint16_t default_port = 8888;class TcpServer{public:TcpServer(uint16_t port = default_port) : _port(port){}void InitServer(){}//初始化服务器void Start(){}//启动服务器~TcpServer(){}private:int _sock; // 监听套接字uint16_t _port;  // 端口号};
}

tcp_server.cc

#include"tcp_server.hpp"
#include<memory>using namespace std;
using namespace ns_server;static void usage(char*argv)
{cout<<"Usage\n\t"<<argv<<" serverPort"<<endl;
}
int main(int argc,char*argv[])
{if(argc!=2){usage(argv[0]);exit(USAGE_ERR);}uint16_t port=atoi(argv[1]);unique_ptr<TcpServer> usvr(new TcpServer(echo,port));usvr->InitServer();usvr->Start();return 0;
}

1. 初始化服务器

服务器的初始化，还是一样的

  1. 创建套接字
  2. 绑定套接字
 void InitServer()
{// 1.创建套接字_sock = socket(AF_INET, SOCK_STREAM, 0);if (_listensock < 0){std::cerr << "create sock error," << strerror(errno) << std::endl;exit(1);}std::cout << "create listensock success: " << _sock << std::endl;// 2.绑定套接字struct sockaddr_in local;local.sin_family = AF_INET;local.sin_port = htons(_port);local.sin_addr.s_addr = INADDR_ANY;if (bind(_sock , (struct sockaddr *)&local, sizeof(local)) < 0){std::cerr << "bind error," << strerror(errno) << std::endl;exit(2);}
}

需要注意的是，socket的第二个参数为SOCK_STREAM面向字节流

TCP与UDP不同的地方是，TCP是面向连接的，UDP是无连接的
所以TCP还需要listen

在这里插入图片描述

返回值：成功返回0，失败返回-1并设置错误码

backlog参数需要在后续TCP详解中学习，先定义大小为32

const int backlog = 32;
void InitServer()
{// 1.创建套接字_sock = socket(AF_INET, SOCK_STREAM, 0);if (_listensock < 0){std::cerr << "create sock error," << strerror(errno) << std::endl;exit(1);}std::cout << "create listensock success: " << _sock << std::endl;// 2.绑定套接字struct sockaddr_in local;local.sin_family = AF_INET;local.sin_port = htons(_port);local.sin_addr.s_addr = INADDR_ANY;if (bind(_sock , (struct sockaddr *)&local, sizeof(local)) < 0){std::cerr << "bind error," << strerror(errno) << std::endl;exit(2);}// 3.监听if (listen(_listensock, backlog) < 0){std::cerr << "listen error," << strerror(errno) << std::endl;exit(3);}
}

初始化到此就结束了
接下来是启动服务器

2. 启动服务器

TCP通过accept获取客户端连接

在这里插入图片描述

  • sockfd：socket返回的文件描述符
  • addr：输入输出型参数，客户端信息的结构体
  • addrlen：输入输出型参数，结构体大小。注意：需要传入addr的大小
  • 返回值是网络文件描述符

在TCP中，socket返回的网络文件可以理解为连接文件，内部保存了连接信息
而accept是从连接文件中获取连接，然后创建套接字，网络文件。
真正通信的是connect创建的网络文件

我们将私有成员的_sock改为_listensock

void Start()
{while (true){struct sockaddr_in client;memset(&client, 0, sizeof(client));socklen_t len = sizeof(client);int sock = accept(_listensock, (struct sockaddr *)&client, &len);if (sock < 0){std::cerr << "accept error" << std::endl;continue;}// 提取客户端信息std::string clientIp = inet_ntoa(client.sin_addr);uint16_t clientPort = ntohs(client.sin_port);std::string name = "[" + clientIp + ":" + std::to_string(clientPort) + "]";std::cout << "create sock " << sock << " from " << _listensock << std::endl;}
}

接下来就可以在connect返回的套接字中读写数据了。
本次使用read和write

void Start()
{while (true){struct sockaddr_in client;memset(&client, 0, sizeof(client));socklen_t len = sizeof(client);int sock = accept(_listensock, (struct sockaddr *)&client, &len);if (sock < 0){std::cerr << "accept error" << std::endl;continue;}// 提取客户端信息std::string clientIp = inet_ntoa(client.sin_addr);uint16_t clientPort = ntohs(client.sin_port);std::string name = "[" + clientIp + ":" + std::to_string(clientPort) + "]";std::cout << "create sock " << sock << " from " << _listensock << std::endl;char buffer[1024];while (true){int n = read(sock, buffer, sizeof(buffer) - 1);if (n > 0){buffer[n] = '\0';std::cout << name << "# " << buffer << std::endl;std::string responce = buffer;//返回收到的数据int m = write(sock, responce.c_str(), responce.size());}else if (n == 0){// 写端关闭std::cout << name << " quit,me to" << std::endl;close(sock);break;}else{// 读数据异常std::cerr << "read error" << std::endl;break;}}}
}

完整代码：
tcp_server.hpp

#pragma once#include <iostream>
#include <string>
#include <cstring>
#include <cerrno>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>namespace ns_server
{const uint16_t default_port = 8888;const int backlog = 32;class TcpServer{public:TcpServer(func_t func, uint16_t port = default_port) : _port(port), _func(func){}void InitServer(){// 1.创建套接字_listensock = socket(AF_INET, SOCK_STREAM, 0);if (_listensock < 0){std::cerr << "create sock error," << strerror(errno) << std::endl;exit(1);}std::cout << "create listensock success: " << _listensock << std::endl;// 2.绑定套接字struct sockaddr_in local;local.sin_family = AF_INET;local.sin_port = htons(_port);local.sin_addr.s_addr = INADDR_ANY;if (bind(_listensock, (struct sockaddr *)&local, sizeof(local)) < 0){std::cerr << "bind error," << strerror(errno) << std::endl;exit(2);}// 3.监听if (listen(_listensock, backlog) < 0){std::cerr << "listen error," << strerror(errno) << std::endl;exit(3);}}void Start(){while (true){struct sockaddr_in client;memset(&client, 0, sizeof(client));socklen_t len = sizeof(client);int sock = accept(_listensock, (struct sockaddr *)&client, &len);if (sock < 0){std::cerr << "accept error" << std::endl;continue;}// 提取客户端信息std::string clientIp = inet_ntoa(client.sin_addr);uint16_t clientPort = ntohs(client.sin_port);std::string name = "[" + clientIp + ":" + std::to_string(clientPort) + "]";std::cout << "create sock " << sock << " from " << _listensock << std::endl;char buffer[1024];while (true){int n = read(sock, buffer, sizeof(buffer) - 1);if (n > 0){buffer[n] = '\0';std::cout << name << "# " << buffer << std::endl;std::string responce = buffer;int m = write(sock, responce.c_str(), responce.size());}else if (n == 0){// 写端关闭std::cout << name << " quit,me to" << std::endl;close(sock);break;}else{// 读数据异常std::cerr << "read error" << std::endl;break;}}}}~TcpServer(){}private:int _listensock; // 监听套接字uint16_t _port;  // 端口号};
}

PS：上述的服务器是单进程，所以只能同时处理一个客户端，读者可以尝试添加一下多进程，多线程或者线程池

本篇博客最后会贴出多进程的方案

二. 客户端

客户端就不作封装了

最开始也是要创建套接字
然后TCP的客户端需要connect服务器

在这里插入图片描述

  • sockfd：socket返回的文件描述符
  • addr：服务器信息的结构体
  • addrlen：结构体大小。
  • 返回值：成功返回0，失败返回-1并设置错误码

注意：connect时OS会bind客户端
UDP是在发送数据时才会bind

#include <iostream>
#include <string>
#include <unistd.h>
#include <cstring>
#include <cerrno>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>using namespace std;static void usage(char *argv)
{cout << "Usage:\n\t" << argv << " serverIp serverPort" << endl;
}int main(int argc, char *argv[])
{if (argc != 3){usage(argv[0]);exit(USAGE_ERR);}string serverIp = argv[1];uint16_t serverPort = atoi(argv[2]);// 1.创建套接字int sock = socket(AF_INET, SOCK_STREAM, 0);if (sock < 0){cerr << "create sock error," << strerror(errno) << endl;exit(SOCKET_ERR);}cout<<"create sock sucess:"<<sock<<endl;// 2. 连接struct sockaddr_in server;memset(&server, 0, sizeof(server));server.sin_family = AF_INET;server.sin_addr.s_addr = inet_addr(serverIp.c_str());server.sin_port = htons(serverPort);int cnt = 5; // 记录重连次数// connect时会bindwhile (connect(sock, (struct sockaddr *)&server, sizeof(server)) < 0){cout << "正在重连...还有" << cnt-- << "次" << endl;if (cnt <= 0){   cerr<<"连接失败"<<endl;exit(CONNECT_ERR);}sleep(1);}// 连接成功string name = "["+serverIp + ":" + to_string(serverPort)+"]";cout << "connect " << name << " sucess" << endl;return 0;
}

然后也可以开始读写数据了

完整代码：

tcp_client.cc

#include <iostream>
#include <string>
#include <unistd.h>
#include <cstring>
#include <cerrno>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>using namespace std;static void usage(char *argv)
{cout << "Usage:\n\t" << argv << " serverIp serverPort" << endl;
}int main(int argc, char *argv[])
{if (argc != 3){usage(argv[0]);exit(USAGE_ERR);}string serverIp = argv[1];uint16_t serverPort = atoi(argv[2]);// 1.创建套接字int sock = socket(AF_INET, SOCK_STREAM, 0);if (sock < 0){cerr << "create sock error," << strerror(errno) << endl;exit(SOCKET_ERR);}cout<<"create sock sucess:"<<sock<<endl;// 2. 连接struct sockaddr_in server;memset(&server, 0, sizeof(server));server.sin_family = AF_INET;server.sin_addr.s_addr = inet_addr(serverIp.c_str());server.sin_port = htons(serverPort);int cnt = 5; // 记录重连次数// connect时会bindwhile (connect(sock, (struct sockaddr *)&server, sizeof(server)) < 0){cout << "正在重连...还有" << cnt-- << "次" << endl;if (cnt <= 0){   cerr<<"连接失败"<<endl;exit(CONNECT_ERR);}sleep(1);}// 连接成功string name = "["+serverIp + ":" + to_string(serverPort)+"]";cout << "connect " << name << " sucess" << endl;// 发送消息while (true){cout << "please enter your message# ";string message;getline(cin, message);int n = write(sock, message.c_str(), message.size());if (n < 0){cerr << "write error," << strerror(errno) << endl;break;}else if (n == 0){cout << "读端关闭，停止写" << endl;break;}char buffer[1024];int m = read(sock, buffer, sizeof(buffer) - 1);if (m > 0){buffer[n] = '\0';cout<<name<<" echo "<<buffer<<endl;}else if (m == 0){// 写端关闭std::cout << name << " quit,me to" << std::endl;close(sock);break;}else{// 读数据异常std::cerr << "read error" << std::endl;break;}}return 0;
}

三. 多进程服务器

tcp_server.hpp

#pragma once#include <iostream>
#include <string>
#include <cstring>
#include <cerrno>
#include <unistd.h>
#include <signal.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>namespace ns_server
{const uint16_t default_port = 8888;const int backlog = 32;class TcpServer{public:TcpServer(uint16_t port = default_port) : _port(port){}void InitServer(){// 1.创建套接字_listensock = socket(AF_INET, SOCK_STREAM, 0);if (_listensock < 0){std::cerr << "create sock error," << strerror(errno) << std::endl;exit(1);}std::cout << "create listensock success: " << _listensock << std::endl;// 2.绑定套接字struct sockaddr_in local;local.sin_family = AF_INET;local.sin_port = htons(_port);local.sin_addr.s_addr = INADDR_ANY;if (bind(_listensock, (struct sockaddr *)&local, sizeof(local)) < 0){std::cerr << "bind error," << strerror(errno) << std::endl;exit(2);}// 3.监听if (listen(_listensock, backlog) < 0){std::cerr << "listen error," << strerror(errno) << std::endl;exit(3);}}void Start(){//忽略子进程的信号，不需要等待子进程退出（推荐！！！）signal(SIGCHLD,SIG_IGN);while (true){struct sockaddr_in client;memset(&client, 0, sizeof(client));socklen_t len = sizeof(client);int sock = accept(_listensock, (struct sockaddr *)&client, &len);if (sock < 0){std::cerr << "accept error" << std::endl;continue;}std::cout << "create sock " << sock << " from " << _listensock << std::endl;// 多进程pid_t id = fork();if (id < 0){close(sock);continue;}else if (id == 0){//子进程close(_listensock);//建议关掉不需要的fdif(fork()>0)exit(0);//子进程退掉，后续为孙子进程// 提取客户端信息std::string clientIp = inet_ntoa(client.sin_addr);uint16_t clientPort = ntohs(client.sin_port);service(sock, clientIp, clientPort);exit(0);}//父进程//一定关掉不需要的fd，防止fd泄露close(sock);//pid_t ret=waitpid(id,nullptr,0);//默认为阻塞等待//pid_t ret=waitpid(id,nullptr,WNOHANG);//非阻塞//if(ret==id) std::cout<<"wait "<<id<<" sucess"<<std::endl;}}void service(int sock, std::string &clientIp, uint16_t&clientPort){std::string name = "[" + clientIp + ":" + std::to_string(clientPort) + "]";char buffer[1024];while (true){int n = read(sock, buffer, sizeof(buffer) - 1);if (n > 0){buffer[n] = '\0';std::cout << name << "# " << buffer << std::endl;std::string responce = buffer;int m = write(sock, responce.c_str(), responce.size());}else if (n == 0){// 写端关闭std::cout << name << " quit,me to" << std::endl;close(sock);break;}else{// 读数据异常std::cerr << "read error" << std::endl;break;}}}~TcpServer(){}private:int _listensock; // 监听套接字uint16_t _port;  // 端口号};
}

结束语

本篇博客到此结束，感谢看到此处。
欢迎大家纠错和补充
如果觉得本篇文章对你有所帮助的话，不妨点个赞支持一下博主，拜托啦，这对我真的很重要。
在这里插入图片描述

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/207201.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

C++ vector 使用类作为模板参数/C++多态展示

C++ vector 使用类作为模板参数/C++多态展示

C vector 使用类作为模板参数 #include<iostream> #include<string> #include<vector>class vector3D {public:float x,y,z; vector3D(int x_,int y_,int z_):x(x_),y(y_),z(z_){}public:float add(){return (xyz);}};int main(){int *a new int[10];a[0] …
阅读更多...
基于原子轨道搜索算法优化概率神经网络PNN的分类预测 - 附代码

基于原子轨道搜索算法优化概率神经网络PNN的分类预测 - 附代码

基于原子轨道搜索算法优化概率神经网络PNN的分类预测 - 附代码 文章目录 基于原子轨道搜索算法优化概率神经网络PNN的分类预测 - 附代码1.PNN网络概述2.变压器故障诊街系统相关背景2.1 模型建立 3.基于原子轨道搜索优化的PNN网络5.测试结果6.参考文献7.Matlab代码 摘要&#xf…
阅读更多...
五大资源之Service（可以固定IP）

五大资源之Service（可以固定IP）

Service可以看作是一组同类Pod对外访问接口，借助Service应用可以方便的实现服务发现与负载均衡 创建集群内部可以访问Service #暴露Service（也创建在了namespace dev下） [root@master ~]# kubectl expose deployment（pod控制器） nginx --name=svc-nginx1 --type=Cluste…
阅读更多...
基于天鹰算法优化概率神经网络PNN的分类预测 - 附代码

基于天鹰算法优化概率神经网络PNN的分类预测 - 附代码

基于天鹰算法优化概率神经网络PNN的分类预测 - 附代码 文章目录 基于天鹰算法优化概率神经网络PNN的分类预测 - 附代码1.PNN网络概述2.变压器故障诊街系统相关背景2.1 模型建立 3.基于天鹰优化的PNN网络5.测试结果6.参考文献7.Matlab代码 摘要&#xff1a;针对PNN神经网络的光滑…
阅读更多...
Java架构师软件架构风格

Java架构师软件架构风格

目录 1 数据流风格1.1 管道过滤器1.2 数据流风格的优点2 调用返回风格2.1 面向对象风格2.2 调用返回风格总结3 独立构件风格3.1 事件驱动系统风格的主要特点3.2 独立构件风格总结4 虚拟机风格4.1 虚拟机风格总结5 仓库风格5.1 仓库风格总结想学习架构师构建流程请跳转：Java架构…
阅读更多...
第一百七十六回 如何创建渐变色边角

第一百七十六回 如何创建渐变色边角

文章目录 1. 概念介绍2. 实现方法3. 代码与细节3.1 示例代码3.2 代码细节 4. 内容总结 我们在上一章回中介绍了"如何创建放射形状渐变背景"相关的内容&#xff0c;本章回中将介绍"如何创建渐变色边角".闲话休提&#xff0c;让我们一起Talk Flutter吧。 1.…
阅读更多...
LeetCode（32）串联所有单词的子串【滑动窗口】【困难】（含图解）

LeetCode（32）串联所有单词的子串【滑动窗口】【困难】（含图解）

目录 1.题目2.答案3.提交结果截图4.图解 链接&#xff1a; 串联所有单词的子串 1.题目 给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。 s 中的 串联子串 是指一个包含 words 中所有字符串以任意顺序排列连接起来的子串。 例如&#xff0c;如果 w…
阅读更多...
sonar对webgoat进行静态扫描

sonar对webgoat进行静态扫描

安装sonar并配置 docker安装sonarqube&#xff0c;sonarQube静态代码扫描 - Joson6350 - 博客园 (cnblogs.com) 对webgoat进行sonar扫描 扫描结果 bugs Change this condition so that it does not always evaluate to "false" 意思是这里的else if语句不会执行…
阅读更多...
计算机网络之物理层（数据通信有关）

计算机网络之物理层（数据通信有关）

一、概述 1.1物理层引入的目的 屏蔽掉传输介质的多样性&#xff0c;导致数据传输方式的不同&#xff1b;物理层的引入使得高层看到的数据都是统一的0,1构成的比特流 1.2.物理层如何实现屏蔽 物理层靠定义的不同的通信协议&#xff08;一般称通信规程&#xff09; 这些协议…
阅读更多...
电动汽车充放电V2G模型MATLAB代码

电动汽车充放电V2G模型MATLAB代码

微❤关注“电气仔推送”获得资料&#xff08;专享优惠&#xff09; 主要内容&#xff1a; 本程序主要建立电动汽车充放电V2G模型&#xff0c;采用粒子群算法&#xff0c;在保证电动汽车用户出行需求的前提下&#xff0c;为了使工作区域电动汽车尽可能多的消纳供给商场基础负荷…
阅读更多...
使用paddleocr进行OCR文字识别

使用paddleocr进行OCR文字识别

1 OCR介绍 OCR&#xff08;Optical Character Recognition&#xff09;即光学字符识别&#xff0c;是一种将不同类型的文档&#xff08;如扫描的纸质文件、PDF文件或图像文件中的文本&#xff09;转换成可编辑和可搜索的数据的技术。OCR技术能够识别和转换印刷或手写文字&…
阅读更多...
关于Flink的旁路缓存与异步操作

关于Flink的旁路缓存与异步操作

1. 旁路缓存 1. 什么是旁路缓存? 将数据库中的数据,比较经常访问的数据,保存起来,以减少和硬盘数据库的交互 比如: 我们使用mysql时 经常查询一个表 , 而这个表又一般不会变化,就可以放在内存中,查找时直接对内存进行查找,而不需要再和mysql交互 2. 旁路缓存例子使用 dim层…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023