当前位置:   article > 正文

网络:udp&&tcp套接字

网络:udp&&tcp套接字

目录

协议

网络传输基本流程

网络编程套接字

udp套接字编程

 udp相关代码实现

sock函数

bind函数

recvfrom函数

sendto函数

udp执行指令代码

popen函数

udp多线程版收发消息

tcp套接字编程

tcp套接字代码

listen函数

accept函数

read/write函数

connect函数

recv/send函数


协议

协议本质就是一种约定,网络上一定要硬件上有标志,软件上有协议

关于协议分层

1.操作系统要进行协议管理:先描述,在组织
2.协议本质就是软件,软件是可以”分层‘
3.协议在设计的时候,就是被层状的划分的
4.要划分为层状结构是因为: a. 场景复杂 b.功能解耦,便于人们进行各种维护

所以网络协议也是层状的

OSI七层模型:物理层、数据链路层、网络层、传输层、会话层、表示层、应用层这7层

但是, OSI七层模型既复杂又不实用; 所以我们按照TCP/IP四(五)层模型来讲解,即(物理层)、数据链路层、网络层、传输层、应用层

TCP/IP通讯协议采用了5层的层级结构,每一层都呼叫它的下一层所提供的网络来完成自己的需求,物理层我们不关心

传输层和网络层是在操作系统中实现的

对于一台主机, 它的操作系统内核实现了从传输层到网络层
对于一台路由器, 它实现了从网络层到物理层
对于一台交换机, 它实现了从数据链路层到物理层
对于集线器, 它只实现了物理层


网络传输基本流程

下面几层对应的代表性的协议

网络通信的过程中:

在逻辑上:在应用层相互直接传输
在物理上: 必须得自顶向下(从应用层->链路层)将数据报文交付到最底层,然后在以太网上传输,被你的主机收到后,再自底向上(从链路层->应用层)交付到应用层

每层都有自己的协议定制方案,
每层协议都要有自己的协议报头
从上到下交付数据的时候,要添加报头
从下到上递交数据的时候,要去掉报头

协议报头就是收到的报文当中,多出来的数据内容,可以理解为同层协议提前约定好的共识字段

把自顶向下交付的过程,称之为封装的过程,封装的本质:添加报头

自底向上交付的时候,需要将报头有效载荷做分离,分析报头的字段,去决定将它的有效载荷向上应该交付给哪一个协议

把自底向上交付的过程,称之为解包的过程,解包的本质:去掉报头,展开分析


在局域网中两台主机是可以直接通信的

例如有5台主机都在一个局域网里,主机1和主机2通信,表面上是这两个主机单独通信,实际其他主机也都能看到他们通信的内容,只是每个主机对比自己的MAC地址,看是不是再和自己通信,如果不是,将当前报文丢弃即可

通过MAC地址来标识局域网中主机的唯一性

IP地址和MAC地址的区别:

IP就类似于旅游时,想从陕西去新疆去,所以此时的源IP就是陕西,目的IP就是新疆

在去的路途中,加入当前在甘肃省,当地人说下一站应该去青海省,所以此时的源MAC地址就是甘肃,下一站MAC地址就是青海

将数据在主机间转发仅仅是手段,机器收到之后,需要将数据交付给指定的进程
因此真正的网络通信过程,本质其实是进程间通信


在使用TCP/IP协议的网络中,IP及其向上的协议,看到的报文都是一样的。

应用层数据通过协议栈发到网络上时,每层协议都要加上一个数据首部,称为封装

数据封装成帧后发到传输介质上,到达目的主机后每层协议再剥掉相应的首部, 根据首部中的 "上层协议 字段" 将数据交给对应的上层协议处理.


网络编程套接字

网络4件套,只要将下面四种头文件都包含,网络中所需要的就都能使用了

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

#include <arpa/inet.h>


IP协议有两个版本, IPv4和IPv6,没有特殊说明的, 默认都是指IPv4,IPv4来说, IP地址是一个4字节, 32位的整数

IP地址,就例如:192.168.0.1,用点分割的每一个数字表示一个字节, 范围是 0 - 255;所以理论上,表示一个IP地址,其实4字节就够了,而192.168.0.1这个字符串却需要11个字节,所以点分十进制字符串风格的IP地址,需要和4字节的IP地址互相转换

IP地址(公网IP),它标定了主机的唯一性

端口号一般是16位的整数

端口号是标识特定主机上的网络进程的唯一性

一个进程可以绑定多个端口号; 但是一个端口号不能被多个进程绑定

所以IP+端口号就可以得到全网唯一的一个进程了

我们把IP地址+端口号,称之为套接字


 UDP协议

UDP协议叫做用户数据报协议,它的特点:无连接、不可靠传输、面向数据报

无连接:在使用UDP通信时,不用写代码时建立连接,直接发送数据即可

不可靠传输:UDP可能出现网络丢包的问题,或是数据包乱序重复等问题


TCP协议

TCP协议叫做用户数据报协议,它的特点:有连接、可靠传输、面向字节流

可靠传输:指TCP不会出现上述UDP的网络丢包等问题


那么UDP是不可靠传输,而TCP是可靠传输,为什么我们还需要学UDP呢,直接学TCP就好了呀

其实很简单,TCP保证可靠性,就一定决定了它在数据通信的过程中,要为了保证可靠性,而设计更多的策略,各种制度机制,都要靠协议自己完成,往往更复杂,成本更高

而UDP只要将数据发出去,就不用管可不可靠的问题了,UDP更为简单,成本更低,TCP则更为安全,所以UDP/TCP是各有所长的

因此在应用场景中,除非场景非常适合UDP,那么我们都选择TCP协议

像直播、视频一类的使用UDP更好,例如在看直播时,突然人物卡顿一下,掉帧一下,都是因为使用的UDP,导致的丢包问题,但是并不会影响我们观看直播


网络字节序

假设两个主机通信,一个主机是大端存储,一个主机是小段存储,那么在通信时,肯定会发生错误,因为两个主机存储的方式都不一样,肯定无法正确的通信

所以计算机网络规定,所有网络数据,都必须是大端存储的

操作系统提供了4个接口,头文件是arpa/inet.h

uint32_t htonl(uint32_t hostlong) ;
uint16_t htons(uint16_t hostshort) ;
uint32_t ntohl(uint32_t netlong) ;
uint16_t ntohs(uint16_t netshort) ;

其中h表示host,表示本地;n表示network,表示网络;l表示long,4字节;s表示short,2字节

所以hton就表示主机转网络,即无论主机是什么序列都会转成大端

ntoh就表示网络转主机


常见套接字:

域间socket、原始socket、网络socket

其中网络socket是使用的最多

理论上,这三种套接字是三种应用场景,对应的应该是三套接口,而Linux为了使用更方便,不想设计过多的接口,而是使用同一套接口,如下所示:

创建 socket 文件描述符 (TCP/UDP, 客户端 + 服务器)
int socket(int domain, int type, int protocol);

绑定端口号 (TCP/UDP, 服务器)
int bind(int socket, const struct sockaddr *address, socklen_t address_len);

开始监听socket (TCP, 服务器)
int listen(int socket, int backlog);

接收请求 (TCP, 服务器)
int accept(int socket, struct sockaddr* address, socklen_t* address_len);

建立连接 (TCP, 客户端)
int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

通过观察可以发现,每一个接口都有一个共同的结构sockaddr

其中网络socket是sockaddr_in结构,它的前两个字节是AF_INET

域间socket是sockaddr_un结构,它的前两个字节是AF_UNIX

那么这个公共的结构sockaddr,前两个字节就可以用来辨别是网络socket还是域间socket,sockaddr就如同我们C语言中的void*的概念


udp套接字编程

 udp相关代码实现

下面初步实现服务端与客户端链接,客户端发消息,服务端可以回复消息,如下所示:

 下面是几个需要使用的函数接口:

sock函数

sock函数是计算机网络给我们提供的系统调用接口,对传输层做了文件系统级别封装的接口

socket函数需要包含头文件sys/types.h和sys/socket.h

函数返回值返回一个文件描述符,返回-1就表示失败,errno就表明出错的原因

函数参数

第一个参数domain:一般表示套接字的域,有AF_INET(网络通信)、AF_UNIX(本地通信)等等

第二个参数type:type表示类型,代表你想创建这个套接字的通信种类是什么,一般填的是SOCK_DGRAM,表示用户数据报套接字

第三个参数protocol:protocol表示协议,一般前两个参数确定了,协议也就确定了,所以一般是0

使用示例如下:


bind函数

bind函数:将用户设置的ip和port在内核中和我们当前的进程强关联

bind函数需要包含头文件sys/types.h和sys/socket.h

函数返回值:成功返回0,否则返回-1,errno就表明出错的原因

函数参数

第一个参数sockfd:是刚刚创建的套接字,需要将ip和port与套接字关联起来

第二个参数addr:上面提到了填充ip和port信息的结构体

第三个参数addrlen:是addr的大小,用sizeof就能计算出来

使用示例如下:

因为bind接口的第二个参数类型是struct sockaddr*的,而我们创建的local的类型是struct sockaddr_in的,所以需要强转一下


recvfrom函数

recvfrom函数是读取数据的函数

函数返回值:ssize_t是有符号整数,失败会返回-1

函数参数:

第一个参数:表示刚刚创建成功的套接字

第二/三个参数:表示一段缓冲区,len表示大小

第四个参数:表示读取的方式,默认为0,表示以阻塞方式读取

第五/六个参数:表示两个输出型参数,需要我们传入特定的结构体,及特定结构体的其他信息,方便我们把客户端的IP和端口提上来,服务器就能够知道是谁发的消息,进而把消息再转发回去

除了拿到数据,我们也想知道是谁给我发的消息,即得到src_ip(源IP),src_port(源端口号),即客户端的IP和端口号,当client发送给server时,server就需要得到client_ip和client_port

使用示例如下:


sendto函数

sendto函数是返回消息的接口

第一个参数:表示创建的套接字

第二/三个参数:表示要发送的数据buf缓冲区,和它的长度len

第三个参数:flag默认设置为0

第四/五个参数:表示需要把缓冲区发给谁,长度是多少

使用示例如下:


关于127.0.0.1,这个IP地址称为本地环回,client和server发送数据只在本地协议栈中进行数据流动,不会吧我们的数据发送到网络中

通常用于本地网络服务器的测试,如果在127.0.0.1这里测试通过了,即客户端发消息,server能收到,那如果联通网络,再进行测试时,发现联不通,这时的问题大概率就是网络的问题


在云服务器中:

云服务器无法bind公网IP,也不建议bind公网IP

对于server来讲,我们也不推荐bind确定的IP

推荐使用任意IP的方案,如下

所以服务器服务端,写IP地址填充的时候,都是INADDR_ANY,当然如果未来有特殊需求,也可以在构造函数中给指定的IP,这样就bind指定的IP了


代码如下:

makefile:

  1. .PHONY:all
  2. all:udp_client udp_server
  3. udp_client:udp_client.cc
  4. g++ -o $@ $^ -std=c++11
  5. udp_server:udp_server.cc
  6. g++ -o $@ $^ -std=c++11
  7. .PHONY:clean
  8. clean:
  9. rm -f udp_client udp_server

udp_client.cc:

  1. #include <iostream>
  2. #include <cstdio>
  3. #include <cstring>
  4. #include <string>
  5. #include <unistd.h>
  6. #include <sys/types.h>
  7. #include <sys/socket.h>
  8. #include <netinet/in.h>
  9. #include <arpa/inet.h>
  10. static void usage(std::string proc)
  11. {
  12. std::cout << "\nUsage: " << proc << " ServerIp ServerPort\n" << std::endl;
  13. }
  14. int main(int argc, char* argv[])
  15. {
  16. if(argc != 3)
  17. {
  18. usage(argv[0]);
  19. exit(1);
  20. }
  21. //创建客户端套接字
  22. int sock = socket(AF_INET, SOCK_DGRAM, 0);
  23. if(sock < 0)
  24. {
  25. std::cerr << "socket error" << std::endl;
  26. exit(2);
  27. }
  28. // client当然也需要bind
  29. //但是一般client不会显示bind,即程序员不需要自己bind
  30. //client是一个客户端,是被普通人下载安装使用的
  31. //如果程序员自己bind了,client就bind了一个固定的port
  32. //万一其他客户端提前占用了这个port,该client就不会启动了
  33. //client一般不需要显示的bind指定port,而是让0S自动随机选择
  34. std::string message;
  35. struct sockaddr_in server;
  36. memset(&server,0,sizeof(server));//初始化为0
  37. server.sin_family = AF_INET;
  38. server.sin_port = htons(atoi(argv[2]));
  39. //inet_addr将点分十进制字符串转为4字节,再将4字节主机序列->网络序列
  40. server.sin_addr.s_addr = inet_addr(argv[1]);
  41. char buffer[1024];
  42. while(true)
  43. {
  44. std::cout << "请输入你的信息# ";
  45. std::getline(std::cin, message);
  46. if(message == "quit")
  47. break;
  48. //当client首次发送消息给服务器的时候,OS会自动给client bind他自己的IP和port
  49. sendto(sock,message.c_str(),message.size(),0,(struct sockaddr*)&server,sizeof(server));
  50. //读数据
  51. struct sockaddr_in temp;
  52. socklen_t len = sizeof(temp);
  53. ssize_t s = recvfrom(sock, buffer, sizeof(buffer), 0, (struct sockaddr*)&temp, &len);
  54. if(s > 0)
  55. {
  56. buffer[s] = 0;
  57. std::cout << "server echo# " << buffer << std::endl;
  58. }
  59. }
  60. close(sock);//关闭套接字
  61. return 0;
  62. }

udp_server.hpp:

  1. #ifndef _UDP_SERVER_HPP
  2. #define _UDP_SERVER_HPP
  3. #include "log.hpp"
  4. #include <iostream>
  5. #include <string>
  6. #include <cstdio>
  7. #include <memory>
  8. #include <cstring>
  9. #include <strings.h>
  10. #include <cerrno>
  11. #include <cstdlib>
  12. #include <sys/types.h>
  13. #include <sys/socket.h>
  14. #include <netinet/in.h>
  15. #include <arpa/inet.h>
  16. #include <unistd.h>
  17. #define SIZE 1024
  18. class UdpServer
  19. {
  20. public:
  21. UdpServer(uint16_t port, std::string ip=""):
  22. _port(port),_ip(ip),_sock(-1)
  23. {}
  24. //初始化服务器
  25. bool initServer()
  26. {
  27. //从这里开始,就是新的系统调用,来完成网络功能
  28. // 1.创建套接字
  29. _sock = socket(AF_INET, SOCK_DGRAM, 0);
  30. if(_sock < 0)//_sock < 0表示创建套接字失败
  31. {
  32. //strerror将错误码转化为错误码描述
  33. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  34. exit(2);
  35. }
  36. // 2. bind:将用户设置的ip和port在内核中和我们当前的进程强关联
  37. //sockaddr_in是结构体封装的整数
  38. struct sockaddr_in local;
  39. //结构体需要进行清0,使用bzero函数
  40. bzero(&local, sizeof(local));
  41. //上面传入的参数是AF_INET,这里的sin_family也就是AF_INET
  42. local.sin_family = AF_INET;
  43. //服务器的IP和端口未来也是要发送给对方主机的->先要将数据发送到网络
  44. //所以需要将_port主机转网络
  45. local.sin_port = htons(_port);
  46. //1.同上,先要将点分十进制字符串风格的IP地址-> 4字节
  47. //2. 4字节主机序列->网络序列
  48. //有一套接口inet_addr,可以一次帮我们做完上述这两件事情
  49. //宏INADDR_ANY就是0,让服务器在工作过程中,可以从任意IP中获取数据
  50. local.sin_addr.s_addr = _ip.empty() ? INADDR_ANY : inet_addr(_ip.c_str());
  51. //绑定
  52. if(bind(_sock,(struct sockaddr*)&local, sizeof(local)) < 0)
  53. {
  54. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  55. exit(2);
  56. }
  57. logMessage(NORMAL, "init udp server done %s", strerror(errno));
  58. //初始化结束
  59. return true;
  60. }
  61. //启动服务器
  62. void Start()
  63. {
  64. //作为一款网络服务器,永远不退出的
  65. //服务器启动->常驻进程->永远在内存中存在,除非挂了,所以要小心内存泄漏的问题
  66. //echo server: client给我们发送消息,我们原封不动返回即可
  67. //创建缓冲区
  68. char buffer[SIZE];
  69. for( ; ; )
  70. {
  71. //倒数第二个参数peer是纯输出型参数
  72. struct sockaddr_in peer;
  73. bzero(&peer,sizeof(peer));
  74. //最后一个参数,既可以输入又可以输出
  75. //输入时:peer缓冲区的大小
  76. //输出时:实际读到的的peer
  77. socklen_t len = sizeof(peer);
  78. //start : 读取数据
  79. ssize_t s = recvfrom(_sock, buffer, sizeof(buffer)-1, 0, (struct sockaddr*)&peer, &len);
  80. //s>0表示读取数据成功
  81. if(s > 0)
  82. {
  83. buffer[s] = 0;//我们目前数据当做字符串
  84. // 1.输出发送的信息 2. 知道是谁发送的
  85. //从网络中来,所以需要网络转主机,即ntoh
  86. uint16_t cli_port = ntohs(peer.sin_port);
  87. //4字节的网络序列的IP -> 本主机的点分十进制风格的IP,使用inet_ntoa函数
  88. std::string cli_ip = inet_ntoa(peer.sin_addr);
  89. printf("[%s:%d]# %s\n", cli_ip.c_str(), cli_port, buffer);
  90. }
  91. //分析和处理数据
  92. //end : 写回数据
  93. sendto(_sock, buffer, strlen(buffer), 0, (struct sockaddr*)&peer, len);
  94. }
  95. }
  96. ~UdpServer()
  97. {
  98. if(_sock >= 0)
  99. close(_sock);//关闭套接字
  100. }
  101. private:
  102. //一个服务器,一般必须需要ip地址和port(16位的整数)
  103. uint16_t _port;
  104. std::string _ip;
  105. int _sock;//通信的套接字
  106. };
  107. #endif

udp_server.cc:

  1. #include "udp_server.hpp"
  2. #include <memory>
  3. #include <cstdlib>
  4. static void usage(std::string proc)
  5. {
  6. std::cout << "\nUsage:" << proc << " port\n" << std::endl;
  7. }
  8. // ./udp_server port
  9. //命令行参数,argc是2
  10. //argv[0]/[1]分别是./udp_server/port
  11. int main(int argc, char* argv[])
  12. {
  13. if(argc != 2)
  14. {
  15. usage(argv[0]);
  16. exit(1);
  17. }
  18. //std::string ip = argv[1];
  19. uint16_t port = atoi(argv[1]);
  20. std::unique_ptr<UdpServer> svr(new UdpServer(port));
  21. //初始化服务器、启动服务器
  22. svr->initServer();
  23. svr->Start();
  24. return 0;
  25. }

以及使用的以前实现的log.hpp:

  1. #pragma once
  2. #include <iostream>
  3. #include <cstdio>
  4. #include <cstdarg>
  5. #include <string>
  6. #include <ctime>
  7. //日志等级
  8. #define DEBUG 0
  9. #define NORMAL 1
  10. #define WARNING 2
  11. #define ERROR 3
  12. #define FATAL 4
  13. //日志等级的映射表
  14. const char* gLevelMap[]={
  15. "DEBUG",
  16. "NORMAL",
  17. "WARNING",
  18. "ERROR",
  19. "FATAL"
  20. };
  21. //日志功能至少:日志等级 时间;支持用户自定义(日志内容等等)
  22. //format是输出格式例如%s之类的
  23. void logMessage(int level, const char* format, ...)
  24. {
  25. //条件编译,如果定义了DEBUG_SHOW这个宏,在打印时就正常打印
  26. //如果没有定义这个DEBUG_SHOW宏,所以level是DEBUG的语句就不再打印了
  27. //我们可以在makefile命令行中定义宏,加上-D选项即可
  28. #ifndef DEBUG_SHOW
  29. if(level == DEBUG) return;
  30. #endif
  31. // vprintf/vfprintf/vsprintf/ vsnprintf
  32. //是把传入的参数按照可变的方式分别进行显示到 显示器 / 文件 / 字符串 /指定长度的字符串
  33. //stdBuffer是日志的标准部分,如日志等级 时间
  34. //logBuffer是日志的自定义部分,如日志内容等
  35. char stdBuffer[1024];
  36. char logBuffer[1024];
  37. //这里的时间采用较为简单的时间戳表示
  38. time_t tm = time(nullptr);
  39. snprintf(stdBuffer, sizeof stdBuffer, "[%s] [%ld]", gLevelMap[level], tm);
  40. va_list args;
  41. va_start(args, format);
  42. vsnprintf(logBuffer,sizeof logBuffer, format, args);
  43. va_end(args);
  44. //拼接两个字符串的内容,一块打印出来
  45. printf("%s %s\n", stdBuffer, logBuffer);
  46. }


udp执行指令代码

如果是上面的聊天,发送的就是文本,如果是指令例如ls -a -l,我们想把命令执行完,把执行的结果再返回,下面实现远程实现命令的功能:

popen函数

popen可以执行传入的command指令,先建立pipe管道,然后再fork(),子进程执行(exec*,即exec系列的接口)command的命令,

返回值FILE *:可以将执行结果通过FILE*指针进行读取


此功能实现,仅仅改变了udp_server.hpp,其他文件没有变化:

  1. #ifndef _UDP_SERVER_HPP
  2. #define _UDP_SERVER_HPP
  3. #include "log.hpp"
  4. #include <iostream>
  5. #include <string>
  6. #include <cstdio>
  7. #include <memory>
  8. #include <cstring>
  9. #include <strings.h>
  10. #include <cerrno>
  11. #include <cstdlib>
  12. #include <sys/types.h>
  13. #include <sys/socket.h>
  14. #include <netinet/in.h>
  15. #include <arpa/inet.h>
  16. #include <unistd.h>
  17. #define SIZE 1024
  18. class UdpServer
  19. {
  20. public:
  21. UdpServer(uint16_t port, std::string ip=""):
  22. _port(port),_ip(ip),_sock(-1)
  23. {}
  24. //初始化服务器
  25. bool initServer()
  26. {
  27. //从这里开始,就是新的系统调用,来完成网络功能
  28. // 1.创建套接字
  29. _sock = socket(AF_INET, SOCK_DGRAM, 0);
  30. if(_sock < 0)//_sock < 0表示创建套接字失败
  31. {
  32. //strerror将错误码转化为错误码描述
  33. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  34. exit(2);
  35. }
  36. // 2. bind:将用户设置的ip和port在内核中和我们当前的进程强关联
  37. //sockaddr_in是结构体封装的整数
  38. struct sockaddr_in local;
  39. //结构体需要进行清0,使用bzero函数
  40. bzero(&local, sizeof(local));
  41. //上面传入的参数是AF_INET,这里的sin_family也就是AF_INET
  42. local.sin_family = AF_INET;
  43. //服务器的IP和端口未来也是要发送给对方主机的->先要将数据发送到网络
  44. //所以需要将_port主机转网络
  45. local.sin_port = htons(_port);
  46. //1.同上,先要将点分十进制字符串风格的IP地址-> 4字节
  47. //2. 4字节主机序列->网络序列
  48. //有一套接口inet_addr,可以一次帮我们做完上述这两件事情
  49. //宏INADDR_ANY就是0,让服务器在工作过程中,可以从任意IP中获取数据
  50. local.sin_addr.s_addr = _ip.empty() ? INADDR_ANY : inet_addr(_ip.c_str());
  51. //绑定
  52. if(bind(_sock,(struct sockaddr*)&local, sizeof(local)) < 0)
  53. {
  54. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  55. exit(2);
  56. }
  57. logMessage(NORMAL, "init udp server done %s", strerror(errno));
  58. //初始化结束
  59. return true;
  60. }
  61. //启动服务器
  62. void Start()
  63. {
  64. //作为一款网络服务器,永远不退出的
  65. //服务器启动->常驻进程->永远在内存中存在,除非挂了,所以要小心内存泄漏的问题
  66. //echo server: client给我们发送消息,我们原封不动返回即可
  67. //创建缓冲区
  68. char buffer[SIZE];
  69. for( ; ; )
  70. {
  71. //倒数第二个参数peer是纯输出型参数
  72. struct sockaddr_in peer;
  73. bzero(&peer,sizeof(peer));
  74. //最后一个参数,既可以输入又可以输出
  75. //输入时:peer缓冲区的大小
  76. //输出时:实际读到的的peer
  77. socklen_t len = sizeof(peer);
  78. char result[256];//将fgets读到的都放到result中
  79. std::string cmd_echo;
  80. //start : 读取数据
  81. ssize_t s = recvfrom(_sock, buffer, sizeof(buffer)-1, 0, (struct sockaddr*)&peer, &len);
  82. //s>0表示读取数据成功
  83. if(s > 0)
  84. {
  85. buffer[s] = 0;//我们目前数据当做字符串
  86. //判断命令中不能有rm相关的指令
  87. if(strcasestr(buffer,"rm") != nullptr || strcasestr(buffer,"rmdir") != nullptr)
  88. {
  89. std::string err_message = "非法请求: ";
  90. std::cout << err_message << buffer << std::endl;
  91. sendto(_sock, err_message.c_str(), err_message.size(), 0, (struct sockaddr*)&peer, len);
  92. continue;
  93. }
  94. FILE* fp = popen(buffer, "r");
  95. if(fp == nullptr)
  96. {
  97. logMessage(ERROR,"%d:%s", errno, strerror(errno));
  98. continue;
  99. }
  100. while(fgets(result, sizeof(result), fp) != nullptr)//读取成功
  101. {
  102. cmd_echo += result;
  103. }
  104. fclose(fp);
  105. }
  106. //分析和处理数据
  107. //end : 写回数据
  108. //sendto(_sock, buffer, strlen(buffer), 0, (struct sockaddr*)&peer, len);
  109. sendto(_sock, cmd_echo.c_str(), cmd_echo.size(), 0, (struct sockaddr*)&peer, len);
  110. }
  111. }
  112. ~UdpServer()
  113. {
  114. if(_sock >= 0)
  115. close(_sock);//关闭套接字
  116. }
  117. private:
  118. //一个服务器,一般必须需要ip地址和port(16位的整数)
  119. uint16_t _port;
  120. std::string _ip;
  121. int _sock;//通信的套接字
  122. };
  123. #endif

udp多线程版收发消息

下面再加以改进,把所有访问当前服务器的人都记录下来,每次一个客户端发消息,另一个客户端能够收到别的客户端发送的消息

下面演示效果:有两个客户端使用mkfifo创建命名管道,分别是clientA和clientB,每一个客户端都设立两个窗口,一个窗口发数据,将收到的数据输出重定向到管道中,另一个窗口使用cat从管道读出来:

上述是4个窗口,上面两个是clientA的,下面两个是clientB的,左边的窗口发送数据,右边的窗口接收数据,并且可以根据端口号看出来是谁发的消息,例如端口号为32795的就是clientA,而端口号为44280的就是clientB

下面是服务端:

最开始运行./udp_server 8080后,直到clientA发送数据后,服务端就会打印init,key,add这三条语句,并且每当其中一个客户端发送数据,服务端都会将这条数据发送给每一个客户端,两个push语句就能够很好地体现这一点


下面是相关代码:

makefile的udp_client后面需要加-lpthread,因为要使用pthread库

并且这里引用了前面曾经实现过的thread.hpp


udp_server.hpp:

  1. #ifndef _UDP_SERVER_HPP
  2. #define _UDP_SERVER_HPP
  3. #include "log.hpp"
  4. #include <iostream>
  5. #include <string>
  6. #include <unordered_map>
  7. #include <cstdio>
  8. #include <memory>
  9. #include <cstring>
  10. #include <strings.h>
  11. #include <cerrno>
  12. #include <cstdlib>
  13. #include <sys/types.h>
  14. #include <sys/socket.h>
  15. #include <netinet/in.h>
  16. #include <arpa/inet.h>
  17. #include <unistd.h>
  18. #define SIZE 1024
  19. class UdpServer
  20. {
  21. public:
  22. UdpServer(uint16_t port, std::string ip=""):
  23. _port(port),_ip(ip),_sock(-1)
  24. {}
  25. //初始化服务器
  26. bool initServer()
  27. {
  28. //从这里开始,就是新的系统调用,来完成网络功能
  29. // 1.创建套接字
  30. _sock = socket(AF_INET, SOCK_DGRAM, 0);
  31. if(_sock < 0)//_sock < 0表示创建套接字失败
  32. {
  33. //strerror将错误码转化为错误码描述
  34. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  35. exit(2);
  36. }
  37. // 2. bind:将用户设置的ip和port在内核中和我们当前的进程强关联
  38. //sockaddr_in是结构体封装的整数
  39. struct sockaddr_in local;
  40. //结构体需要进行清0,使用bzero函数
  41. bzero(&local, sizeof(local));
  42. //上面传入的参数是AF_INET,这里的sin_family也就是AF_INET
  43. local.sin_family = AF_INET;
  44. //服务器的IP和端口未来也是要发送给对方主机的->先要将数据发送到网络
  45. //所以需要将_port主机转网络
  46. local.sin_port = htons(_port);
  47. //1.同上,先要将点分十进制字符串风格的IP地址-> 4字节
  48. //2. 4字节主机序列->网络序列
  49. //有一套接口inet_addr,可以一次帮我们做完上述这两件事情
  50. //宏INADDR_ANY就是0,让服务器在工作过程中,可以从任意IP中获取数据
  51. local.sin_addr.s_addr = _ip.empty() ? INADDR_ANY : inet_addr(_ip.c_str());
  52. //绑定
  53. if(bind(_sock,(struct sockaddr*)&local, sizeof(local)) < 0)
  54. {
  55. logMessage(FATAL,"%d:%s", errno, strerror(errno));
  56. exit(2);
  57. }
  58. logMessage(NORMAL, "init udp server done %s", strerror(errno));
  59. //初始化结束
  60. return true;
  61. }
  62. //启动服务器
  63. void Start()
  64. {
  65. //作为一款网络服务器,永远不退出的
  66. //服务器启动->常驻进程->永远在内存中存在,除非挂了,所以要小心内存泄漏的问题
  67. //echo server: client给我们发送消息,我们原封不动返回即可
  68. //创建缓冲区
  69. char buffer[SIZE];
  70. for( ; ; )
  71. {
  72. //倒数第二个参数peer是纯输出型参数
  73. struct sockaddr_in peer;
  74. bzero(&peer,sizeof(peer));
  75. //最后一个参数,既可以输入又可以输出
  76. //输入时:peer缓冲区的大小
  77. //输出时:实际读到的的peer
  78. socklen_t len = sizeof(peer);
  79. char key[64];
  80. char result[256];//将fgets读到的都放到result中
  81. std::string cmd_echo;
  82. //start : 读取数据
  83. ssize_t s = recvfrom(_sock, buffer, sizeof(buffer)-1, 0, (struct sockaddr*)&peer, &len);
  84. //s>0表示读取数据成功
  85. if(s > 0)
  86. {
  87. buffer[s] = 0;//我们目前数据当做字符串
  88. //提取ip和port
  89. uint16_t cli_port = ntohs(peer.sin_port);
  90. //4字节的网络序列的IP -> 本主机的点分十进制风格的IP,使用inet_ntoa函数
  91. std::string cli_ip = inet_ntoa(peer.sin_addr);
  92. snprintf(key,sizeof(key),"%s-%u",cli_ip.c_str(),cli_port);
  93. logMessage(NORMAL, "key:%s", key);
  94. auto it = _users.find(key);//查找key
  95. if(it == _users.end())
  96. {
  97. logMessage(NORMAL,"add new user : %s", key);
  98. //说明用户不存在
  99. _users.insert({key,peer});
  100. }
  101. }
  102. //分析和处理数据
  103. //end : 写回数据
  104. //sendto(_sock, buffer, strlen(buffer), 0, (struct sockaddr*)&peer, len);
  105. for(auto& iter : _users)
  106. {
  107. std::string sendMessage = key;
  108. sendMessage += "# ";
  109. sendMessage += buffer;
  110. //因为直接发送消息,并不知道是谁发的消息,所以在消息前加上发送消息的 “IP+port#”
  111. logMessage(NORMAL,"push message to %s", iter.first.c_str());
  112. sendto(_sock, sendMessage.c_str(), sendMessage.size(), 0, (struct sockaddr*)&(iter.second), sizeof(iter.second));
  113. }
  114. }
  115. }
  116. ~UdpServer()
  117. {
  118. if(_sock >= 0)
  119. close(_sock);//关闭套接字
  120. }
  121. private:
  122. //一个服务器,一般必须需要ip地址和port(16位的整数)
  123. uint16_t _port;
  124. std::string _ip;
  125. int _sock;//通信的套接字
  126. std::unordered_map<std::string, struct sockaddr_in> _users;
  127. };
  128. #endif

udp_server.cc:

  1. #include "udp_server.hpp"
  2. #include <memory>
  3. #include <cstdlib>
  4. static void usage(std::string proc)
  5. {
  6. std::cout << "\nUsage:" << proc << " port\n" << std::endl;
  7. }
  8. // ./udp_server port
  9. //命令行参数,argc是2
  10. //argv[0]/[1]分别是./udp_server/port
  11. int main(int argc, char* argv[])
  12. {
  13. if(argc != 2)
  14. {
  15. usage(argv[0]);
  16. exit(1);
  17. }
  18. //std::string ip = argv[1];
  19. uint16_t port = atoi(argv[1]);
  20. std::unique_ptr<UdpServer> svr(new UdpServer(port));
  21. //初始化服务器、启动服务器
  22. svr->initServer();
  23. svr->Start();
  24. return 0;
  25. }

udP_client.cc:

  1. #include <iostream>
  2. #include <cstdio>
  3. #include <cstring>
  4. #include <string>
  5. #include <memory>
  6. #include <unistd.h>
  7. #include <sys/types.h>
  8. #include <sys/socket.h>
  9. #include <netinet/in.h>
  10. #include <arpa/inet.h>
  11. #include <pthread.h>
  12. #include "thread.hpp"
  13. //我们发现:无论是多线程读还是写,用的sock都是一个,sock代表就是文件,UDP是全双工的->可以同时进行收发而不受干扰
  14. uint16_t serverport = 0;
  15. std::string serverip;
  16. static void usage(std::string proc)
  17. {
  18. std::cout << "\nUsage: " << proc << " ServerIp ServerPort\n"
  19. << std::endl;
  20. }
  21. // 发送方
  22. static void *udpSend(void *args)
  23. {
  24. // 拿到sock和线程名name
  25. int sock = *(int *)((ThreadData *)args)->args_;
  26. std::string name = ((ThreadData *)args)->name_;
  27. std::string message;
  28. struct sockaddr_in server;
  29. memset(&server, 0, sizeof(server)); // 初始化为0
  30. server.sin_family = AF_INET;
  31. server.sin_port = htons(serverport);
  32. // inet_addr将点分十进制字符串转为4字节,再将4字节主机序列->网络序列
  33. server.sin_addr.s_addr = inet_addr(serverip.c_str());
  34. // 不断的给对方发送数据
  35. while (true)
  36. {
  37. std::cerr << "请输入你的信息# "; //cerr标准错误,向文件描述符2打印
  38. std::getline(std::cin, message);
  39. if (message == "quit")
  40. break;
  41. // 当client首次发送消息给服务器的时候,OS会自动给client bind他自己的IP和port
  42. sendto(sock, message.c_str(), message.size(), 0, (struct sockaddr *)&server, sizeof(server));
  43. }
  44. return nullptr;
  45. }
  46. //收消息
  47. static void *udpRecv(void *args)
  48. {
  49. int sock = *(int *)((ThreadData *)args)->args_;
  50. std::string name = ((ThreadData *)args)->name_;
  51. char buffer[1024];
  52. while (true)
  53. {
  54. memset(buffer, 0, sizeof(buffer));
  55. // 读数据
  56. struct sockaddr_in temp;
  57. socklen_t len = sizeof(temp);
  58. ssize_t s = recvfrom(sock, buffer, sizeof(buffer), 0, (struct sockaddr *)&temp, &len);
  59. if (s > 0)
  60. {
  61. buffer[s] = 0;
  62. std::cout << buffer << std::endl;
  63. }
  64. }
  65. }
  66. int main(int argc, char *argv[])
  67. {
  68. if (argc != 3)
  69. {
  70. usage(argv[0]);
  71. exit(1);
  72. }
  73. // 创建客户端套接字
  74. int sock = socket(AF_INET, SOCK_DGRAM, 0);
  75. if (sock < 0)
  76. {
  77. std::cerr << "socket error" << std::endl;
  78. exit(2);
  79. }
  80. serverport = atoi(argv[2]);
  81. serverip = argv[1];
  82. // 两个线程,一个发送数据,一个接收数据
  83. std::unique_ptr<Thread> sender(new Thread(1, udpSend, (void *)&sock));
  84. std::unique_ptr<Thread> recver(new Thread(1, udpRecv, (void *)&sock));
  85. sender->start();
  86. recver->start();
  87. sender->join();
  88. recver->join();
  89. close(sock); // 关闭套接字
  90. // client当然也需要bind
  91. // 但是一般client不会显示bind,即程序员不需要自己bind
  92. // client是一个客户端,是被普通人下载安装使用的
  93. // 如果程序员自己bind了,client就bind了一个固定的port
  94. // 万一其他客户端提前占用了这个port,该client就不会启动了
  95. // client一般不需要显示的bind指定port,而是让0S自动随机选择
  96. // std::string message;
  97. // struct sockaddr_in server;
  98. // memset(&server,0,sizeof(server));//初始化为0
  99. // server.sin_family = AF_INET;
  100. // server.sin_port = htons(atoi(argv[2]));
  101. // //inet_addr将点分十进制字符串转为4字节,再将4字节主机序列->网络序列
  102. // server.sin_addr.s_addr = inet_addr(argv[1]);
  103. // char buffer[1024];
  104. // while(true)
  105. // {
  106. // //多线程
  107. // std::cout << "请输入你的信息# ";
  108. // std::getline(std::cin, message);
  109. // if(message == "quit")
  110. // break;
  111. // //当client首次发送消息给服务器的时候,OS会自动给client bind他自己的IP和port
  112. // sendto(sock,message.c_str(),message.size(),0,(struct sockaddr*)&server,sizeof(server));
  113. // //读数据
  114. // struct sockaddr_in temp;
  115. // socklen_t len = sizeof(temp);
  116. // ssize_t s = recvfrom(sock, buffer, sizeof(buffer), 0, (struct sockaddr*)&temp, &len);
  117. // if(s > 0)
  118. // {
  119. // buffer[s] = 0;
  120. // std::cout << "server echo# " << buffer << std::endl;
  121. // }
  122. // }
  123. return 0;
  124. }

thread.hpp:

  1. #pragma once
  2. #include <iostream>
  3. #include <unistd.h>
  4. #include <vector>
  5. #include <pthread.h>
  6. #include <queue>
  7. #include <string>
  8. #include <cstdio>
  9. typedef void*(*func_t)(void*);
  10. //设置ThreadData是为了未来创建线程的时候,把名字也传进去
  11. class ThreadData
  12. {
  13. public:
  14. void* args_;
  15. std::string name_;
  16. };
  17. class Thread
  18. {
  19. public:
  20. Thread(int num, func_t callback, void* args):func_(callback)
  21. {
  22. //构造函数中初始化name_
  23. char namebuffer[64];
  24. snprintf(namebuffer, sizeof namebuffer, "thread->%d", num);
  25. name_ = namebuffer;
  26. tdata_.args_ = args;
  27. tdata_.name_ = name_;
  28. }
  29. //创建线程
  30. void start()
  31. {
  32. pthread_create(&tid_, nullptr, func_, (void*)&tdata_);
  33. }
  34. //线程等待
  35. void join()
  36. {
  37. pthread_join(tid_, nullptr);
  38. }
  39. std::string name()
  40. {
  41. return name_;
  42. }
  43. ~Thread()
  44. {}
  45. private:
  46. std::string name_;
  47. func_t func_;
  48. ThreadData tdata_;
  49. pthread_t tid_;
  50. };

tcp套接字编程

tcp套接字代码

tcp同样,需要socket先创建套接字:

与udp一样,第一个参数需要传AF_INET,因为是网络通信

第二个参数就与udp不同了,传入的是SOCK_STREAM,代表套接字类型是stream

第三个参数同样默认为0


listen函数

因为TCP是面向连接的,当我们正式通信的时候,需要先建立连接,所以就需要用到listen,阿金套接字状态设置为监听状态

函数参数:

第一个参数就是创建好的套接字

第二个参数叫做全链接队列长度

返回值:

成功返回0,失败返回-1,错误码被设置

此时执行完listen函数后,在一个窗口执行./tcp_server 8080后,另一个窗口使用netstat -antp,可以观察到./tcp_server的状态是LISTEN,表示此时的服务器已经运行,且处于监听状态


accept函数

accept函数用于获取新连接

函数参数:

第一个参数:是创建好的套接字

第二/三个参数:addr是输出型参数,addrlen是输入输出型参数,在recvfrom中使用方式一样,详细见上面的recvfrom函数

返回值:

成功的话返回一个合法的整数套接字,失败返回-1,错误码被设置


read/write函数

在udp中,读取数据使用的是recvfrom函数,而tcp中不使用这个函数,读使用read,写使用write

这里的read和write在读写文件时使用过,网络这里也可以直接使用(也可以使用后面的recv/send函数)

read:

使用方式如下:

注意这里的sizeof(buffer)-1,是将buffer后面的\0减掉了

write:

使用方式如下:

注意这里的strlen算的是buffer里的字符数量,没有减去\0 


connect函数

connect是进行连接的函数,connect函数会自动绑定当前客户端的IP和端口,能够进行后续通信

函数参数:

第一个参数:套接字

第二/三个参数:代表想要连接谁,和sendto的后两个参数一样

返回值:

成功返回0,失败返回-1,错误码被设置


recv/send函数

send函数也是基于tcp向目标服务器发送消息的

它的前三个参数和write一样, 最后一个flag参数默认设为0即可

同样和recvfrom的前三个参数一样,最后一个flag参数默认设为0即可


最终可以通过改变tcp_server.hpp中的原始的server函数,从而实现不同的功能,下面实现了通信、大小写转换、英汉互译等功能,以大小写转换为例,示范:

用户端:

服务端:


代码如下:

Threadpool中的Task.hpp改变如下:

  1. #pragma once
  2. #include "log.hpp"
  3. #include <iostream>
  4. #include <functional>
  5. #include <string>
  6. // typedef function<void(int, const std::string&, const uint16_t&)> fun_t;
  7. using fun_t = function<void(int, const std::string&, const uint16_t&, const std::string&)>;
  8. class Task
  9. {
  10. public:
  11. Task()
  12. {}
  13. Task(int sock,const std::string ip,uint16_t port,fun_t func)
  14. :_sock(sock),_ip(ip),_port(port),_func(func)
  15. {}
  16. //仿函数
  17. void operator()(const std::string& name)
  18. {
  19. _func(_sock, _ip, _port, name);
  20. }
  21. public:
  22. int _sock; //套接字
  23. std::string _ip;//ip
  24. uint16_t _port; //端口
  25. fun_t _func;
  26. };

makefile:

  1. .PHONY:all
  2. all:tcp_client tcp_server
  3. tcp_client:tcp_client.cc
  4. g++ -o $@ $^ -std=c++11
  5. tcp_server:tcp_server.cc
  6. g++ -o $@ $^ -std=c++11 -lpthread
  7. .PHONY:clean
  8. clean:
  9. rm -f tcp_client tcp_server

tcp_client.cc:

  1. #include <iostream>
  2. #include <string>
  3. #include <cstdio>
  4. #include <cstring>
  5. #include <unistd.h>
  6. #include <sys/types.h>
  7. #include <sys/socket.h>
  8. #include <netinet/in.h>
  9. #include <arpa/inet.h>
  10. static void usage(std::string proc)
  11. {
  12. std::cout << "\nUsage: " << proc << " serverIp serverPort\n"
  13. << std::endl;
  14. }
  15. //./tcp_client serverip serverport
  16. int main(int argc, char *argv[])
  17. {
  18. if (argc != 3)
  19. {
  20. usage(argv[0]);
  21. exit(1);
  22. }
  23. std::string serverip = argv[1];
  24. uint16_t serverport = atoi(argv[2]);
  25. bool alive = false;
  26. int sock = 0;
  27. std::string line;
  28. while (true)
  29. {
  30. //alive为false,表示需要创建套接字
  31. if (!alive)
  32. {
  33. // 创建套接字
  34. sock = socket(AF_INET, SOCK_STREAM, 0);
  35. if (sock < 0) // 创建套接字失败
  36. {
  37. std::cerr << "socket error" << std::endl;
  38. exit(2);
  39. }
  40. // client不需要手动bind,需要OS自动进行选择
  41. // client需要连接别人的能力
  42. struct sockaddr_in server;
  43. memset(&server, 0, sizeof(server)); // 置空
  44. server.sin_family = AF_INET;
  45. server.sin_port = htons(serverport);
  46. server.sin_addr.s_addr = inet_addr(serverip.c_str());
  47. // connec进行连接
  48. if (connect(sock, (struct sockaddr *)&server, sizeof(server)) < 0)
  49. {
  50. std::cerr << "connect error" << std::endl;
  51. exit(3);
  52. }
  53. // 连接成功
  54. std::cout << "connect success" << std::endl;
  55. alive = true;
  56. }
  57. std::cout << "请输入# ";
  58. std::getline(std::cin, line);
  59. if (line == "quit")
  60. break;
  61. ssize_t ret = send(sock, line.c_str(), line.size(), 0);
  62. //如果发送成功
  63. if (ret > 0)
  64. {
  65. char buffer[1024];
  66. ssize_t s = recv(sock, buffer, sizeof(buffer) - 1, 0);
  67. if (s > 0)
  68. {
  69. // 读取成功
  70. buffer[s] = 0; // 当做字符串处理
  71. std::cout << "server 回显# " << buffer << std::endl;
  72. }
  73. else if (s == 0)
  74. {
  75. alive = false;
  76. close(sock);
  77. }
  78. }
  79. //发送失败
  80. else
  81. {
  82. alive = false;
  83. close(sock);
  84. }
  85. // else if (s == 0)
  86. // {
  87. // break;
  88. // }
  89. // else
  90. // {
  91. // break;
  92. // }
  93. }
  94. return 0;
  95. }

tcp_server.cc:

  1. #include "tcp_server.hpp"
  2. #include <memory>
  3. static void usage(std::string proc)
  4. {
  5. std::cout << "\nUsage: " << proc << " port\n" << std::endl;
  6. }
  7. //./tcp_server port
  8. int main(int argc, char* argv[])
  9. {
  10. if(argc != 2)
  11. {
  12. usage(argv[0]);
  13. exit(1);
  14. }
  15. uint16_t port = atoi(argv[1]);
  16. std::unique_ptr<TcpServer> svr(new TcpServer(port));
  17. svr->initServer();
  18. svr->Start();
  19. return 0;
  20. }

tcp_server.hpp:

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <cstring>
  5. #include <unistd.h>
  6. #include <cassert>
  7. #include <unordered_map>
  8. #include <memory>
  9. #include <signal.h>
  10. #include <sys/types.h>
  11. #include <sys/socket.h>
  12. #include <netinet/in.h>
  13. #include <arpa/inet.h>
  14. #include <sys/wait.h>
  15. #include <pthread.h>
  16. #include <ctype.h>
  17. #include "threadpool/log.hpp"
  18. #include "threadpool/lockGuard.hpp"
  19. #include "threadpool/Task.hpp"
  20. #include "threadpool/threadPool.hpp"
  21. #include "threadpool/thread.hpp"
  22. // static void service(int sock, const std::string& clientip, const uint16_t& clientport)
  23. // {
  24. // //echo server
  25. // char buffer[1024];
  26. // while(true)
  27. // {
  28. // //read && write可以直接使用
  29. // ssize_t s = read(sock, buffer, sizeof(buffer)-1);
  30. // //read成功
  31. // if(s > 0)
  32. // {
  33. // buffer[s] = 0;//将发过来的数据当做字符串
  34. // std::cout << clientip << ":" << clientport << "#" << buffer << std::endl;
  35. // }
  36. // else if(s == 0) //对端关闭连接
  37. // {
  38. // logMessage(NORMAL, "%s:%d shutdown, me too", clientip.c_str(), clientport);
  39. // break;
  40. // }
  41. // else
  42. // {
  43. // logMessage(ERROR, "read socket error, %d:%s", errno, strerror(errno));
  44. // break;
  45. // }
  46. // write(sock, buffer, strlen(buffer));
  47. // }
  48. // close(sock);
  49. // }
  50. static void service(int sock, const std::string &clientip, const uint16_t &clientport, const std::string &thread_name)
  51. {
  52. // echo server
  53. // 所以,我们一般服务器进程业务处理,如果是从连上,到断开,要一直保持这个链接,长连接
  54. //
  55. char buffer[1024];
  56. while (true)
  57. {
  58. // read && write可以直接使用
  59. ssize_t s = read(sock, buffer, sizeof(buffer) - 1);
  60. // read成功
  61. if (s > 0)
  62. {
  63. buffer[s] = 0; // 将发过来的数据当做字符串
  64. std::cout << thread_name << "|" << clientip << ":" << clientport << "#" << buffer << std::endl;
  65. }
  66. else if (s == 0) // 对端关闭连接
  67. {
  68. logMessage(NORMAL, "%s:%d shutdown, me too", clientip.c_str(), clientport);
  69. break;
  70. }
  71. else
  72. {
  73. logMessage(ERROR, "read socket error, %d:%s", errno, strerror(errno));
  74. break;
  75. }
  76. write(sock, buffer, strlen(buffer));
  77. }
  78. close(sock);
  79. }
  80. // 改为大小写转换的方法
  81. static void change(int sock, const std::string &clientip, const uint16_t &clientport, const std::string &thread_name)
  82. {
  83. // echo server
  84. // 所以,我们一般服务器进程业务处理,如果是从连上,到断开,要一直保持这个链接,长连接
  85. //
  86. char buffer[1024];
  87. // read && write可以直接使用
  88. ssize_t s = read(sock, buffer, sizeof(buffer) - 1);
  89. // read成功
  90. if (s > 0)
  91. {
  92. buffer[s] = 0; // 将发过来的数据当做字符串
  93. std::cout << thread_name << "|" << clientip << ":" << clientport << "#" << buffer << std::endl;
  94. std::string message;
  95. char* start = buffer;
  96. while(*start)
  97. {
  98. char c;
  99. if(islower(*start)) c = toupper(*start);
  100. else c = *start;
  101. message.push_back(c);
  102. start++;
  103. }
  104. write(sock, message.c_str(), message.size());
  105. }
  106. else if (s == 0) // 对端关闭连接
  107. {
  108. logMessage(NORMAL, "%s:%d shutdown, me too", clientip.c_str(), clientport);
  109. }
  110. else
  111. {
  112. logMessage(ERROR, "read socket error, %d:%s", errno, strerror(errno));
  113. }
  114. close(sock);
  115. }
  116. // 改为英汉互译的方法
  117. static void dictOnline(int sock, const std::string &clientip, const uint16_t &clientport, const std::string &thread_name)
  118. {
  119. // echo server
  120. // 所以,我们一般服务器进程业务处理,如果是从连上,到断开,要一直保持这个链接,长连接
  121. char buffer[1024];
  122. static std::unordered_map<std::string, std::string> dict = {
  123. {"apple","苹果"},
  124. {"pencil","铅笔"},
  125. {"bus","公交车"},
  126. {"ruler","尺子"}
  127. };
  128. // read && write可以直接使用
  129. ssize_t s = read(sock, buffer, sizeof(buffer) - 1);
  130. // read成功
  131. if (s > 0)
  132. {
  133. buffer[s] = 0; // 将发过来的数据当做字符串
  134. std::cout << thread_name << "|" << clientip << ":" << clientport << "#" << buffer << std::endl;
  135. std::string message;
  136. auto iter = dict.find(buffer);
  137. if(iter == dict.end())
  138. message = "我不知道";
  139. else
  140. message = iter->second;
  141. write(sock, message.c_str(), message.size());
  142. }
  143. else if (s == 0) // 对端关闭连接
  144. {
  145. logMessage(NORMAL, "%s:%d shutdown, me too", clientip.c_str(), clientport);
  146. }
  147. else
  148. {
  149. logMessage(ERROR, "read socket error, %d:%s", errno, strerror(errno));
  150. }
  151. close(sock);
  152. }
  153. // class ThreadDate
  154. // {
  155. // public:
  156. // int _sock;
  157. // std::string _ip;
  158. // uint16_t _port;
  159. // };
  160. class TcpServer
  161. {
  162. private:
  163. const static int gbacklog = 20; // 一般不能太大,也不能太小,后面在进行详细解释
  164. // 类内会有默认参数this,不符合回调函数的参数类型,所以设置为static的
  165. // static void* threadRoutine(void* args)
  166. // {
  167. // //线程分离,设置分离状态后主线程就不需要再join线程了,线程退出时,自动释放线程资源
  168. // pthread_detach(pthread_self());
  169. // ThreadDate* td = static_cast<ThreadDate*>(args);
  170. // service(td->_sock, td->_ip, td->_port);
  171. // delete td;
  172. // return nullptr;
  173. // }
  174. public:
  175. TcpServer(uint16_t port, std::string ip = "")
  176. : _port(port), _ip(ip), _listensock(-1), _threadpool_ptr(ThreadPool<Task>::getThreadPool())
  177. {
  178. }
  179. void initServer()
  180. {
  181. // 1. socket创建套接字 进程 + 文件
  182. _listensock = socket(AF_INET, SOCK_STREAM, 0);
  183. if (socket < 0)
  184. {
  185. logMessage(FATAL, "create socket error,%d:%s", errno, strerror(errno));
  186. exit(2);
  187. }
  188. // 文件描述符012默认打开,所以再创建就是3
  189. logMessage(NORMAL, "create socket success, _listensock: %d", _listensock);
  190. // 2. bind 文件 + 网络
  191. struct sockaddr_in local;
  192. memset(&local, 0, sizeof(local));
  193. local.sin_family = AF_INET;
  194. local.sin_port = htons(_port);
  195. local.sin_addr.s_addr = _ip.empty() ? INADDR_ANY : inet_addr(_ip.c_str());
  196. // binf失败
  197. if (bind(_listensock, (struct sockaddr *)&local, sizeof(local)) < 0)
  198. {
  199. logMessage(FATAL, "bind error,%d:%s", errno, strerror(errno));
  200. exit(3);
  201. }
  202. // 3.因为TCP是面向连接的,当我们正式通信的时候,需要先建立连接
  203. if (listen(_listensock, gbacklog) < 0)
  204. {
  205. // 创建失败
  206. logMessage(FATAL, "listen error,%d:%s", errno, strerror(errno));
  207. exit(4);
  208. }
  209. logMessage(NORMAL, "init server success");
  210. }
  211. void Start()
  212. {
  213. _threadpool_ptr->run(); // 运行线程池
  214. // signal(SIGCHLD, SIG_IGN);//忽略SIG_CHLD信号,子进程退出时会自动释放自己的僵尸状态
  215. // 服务器需要周而复始的运行,所以while(true)
  216. while (true)
  217. {
  218. // 获取连接
  219. struct sockaddr_in src;
  220. socklen_t len = sizeof(src);
  221. // serversock和_listensock都是套接字,区别如下:
  222. // serversock(未来进行网络通信服务) vs _listensock(只是获取新连接)
  223. int serversock = accept(_listensock, (struct sockaddr *)&src, &len);
  224. if (serversock < 0)
  225. {
  226. // 获取连接失败,continue继续获取连接
  227. logMessage(ERROR, "accept error,%d : %s", errno, strerror(errno));
  228. continue;
  229. }
  230. // 获取连接成功
  231. uint16_t client_port = ntohs(src.sin_port);
  232. std::string client_ip = inet_ntoa(src.sin_addr);
  233. logMessage(NORMAL, "link success, serversock: %d | %s : %d | \n",
  234. serversock, client_ip.c_str(), client_port);
  235. // version 4 -- 线程池版本
  236. // Task t(serversock, client_ip, client_port, service);
  237. // Task t(serversock, client_ip, client_port, change);
  238. Task t(serversock, client_ip, client_port, dictOnline);
  239. _threadpool_ptr->pushTask(t);
  240. // //version 3 -- 多线程版本
  241. // ThreadDate* td = new ThreadDate();
  242. // td->_ip = client_ip;
  243. // td->_port = client_port;
  244. // td->_sock = serversock;
  245. // pthread_t tid;
  246. // pthread_create(&tid, nullptr, threadRoutine, td);
  247. // close(serversock)
  248. // version 2.1 -- 多进程版本 -- 创建子进程(不使用signal)
  249. // pid_t id = fork();
  250. // assert(id != -1);
  251. // if(id == 0)
  252. // {
  253. // //子进程
  254. // close(_listensock);
  255. // if(fork() > 0)
  256. // exit(0);//子进程本身直接退出
  257. // //下面的是孙子进程,孤儿进程,OS领养,OS在孤儿进程退出的时候,由OS自动回收孤儿进程
  258. // service(serversock, client_ip, client_port);
  259. // exit(0);
  260. // }
  261. // //父进程
  262. // waitpid(id, nullptr, 0);//不会阻塞,因为子进程再创建孙子进程时就exit退出了
  263. // close(serversock);
  264. // 开始进行通信服务
  265. // version 1.0 -- 单进程循环版本
  266. // 只能够进行一次处理一个客户端,处理完一个,才能处理下一个客户端
  267. // 很显然version 1 是不能直接被使用,因为是单进程的
  268. // 使用service函数实现
  269. // service(serversock, client_ip, client_port);
  270. // version 2.0 -- 多进程版本 -- 创建子进程
  271. // 让子进程给新的连接提供服务,子进程能够打开父进程曾经打开的文件fd
  272. // pid_t id = fork();
  273. // assert(id != -1);
  274. // if(id == 0)
  275. // {
  276. // //子进程会继承父进程打开的文件与文件fd
  277. // //子进程是来进行提供服务的,所以不需要知道监听socket,所以close掉监听socket
  278. // close(_listensock);
  279. // //子进程对新的连接提供服务,父进程则继续accept新连接
  280. // service(serversock, client_ip, client_port);
  281. // exit(0); // 子进程退出后变为僵尸状态
  282. // }
  283. // 父进程
  284. // 需要等待子进程,但是这里最好不要使用waitpid(),因为这时阻塞等待的,设置为非阻塞也比较麻烦
  285. // 所以这里采用信号的方式signal,子进程退出会主动向父进程发送SIG_CHLD信号,所以可以忽略SIG_CHLD信号
  286. // 子进程退出的时候,会自动释放自己的僵尸状态
  287. // close(serversock);//父进程只需要listensocket,不需要serversock,所以close掉serversock
  288. // 如果父进程关闭servicesock,会不会影响子进程?
  289. }
  290. }
  291. ~TcpServer()
  292. {
  293. }
  294. private:
  295. uint16_t _port; // 端口
  296. std::string _ip; // IP
  297. int _listensock; // 监听套接字
  298. std::unique_ptr<ThreadPool<Task>> _threadpool_ptr;
  299. };

以上即为udp、tcp套接字编程的全部内容


声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号