小言_互联网的博客

<UDP网络编程>——《计算机网络》

473人阅读  评论(0)

目录

1. 网络基础知识

1.1 理解源IP地址和目的IP地址

1.2 认识端口号

1.3 理解 "端口号" 和 "进程ID"

1.3.1 理解源端口号和目的端口号

1.4 认识TCP协议

1.5 认识UDP协议

1.6 网络字节序

2. socket编程接口

2.1 socket 常见API

2.2 sockaddr结构

2.3 sockaddr 结构

2.4 sockaddr_in 结构

2.5 in_addr结构

3. 简单的UDP网络程序

3.1 封装 UdpSocket

3.2 UDP通用服务器

3.3 实现英译汉服务器

3.4 UDP通用客户端

3.5 实现英译汉客户端  

3.6 地址转换函数

3.7 关于inet_ntoa

4. udp网络编程

4.1 网络接口函数及测试

4.2 UDP程序完整代码

后记:●由于作者水平有限,文章难免存在谬误之处,敬请读者斧正,俚语成篇,恳望指教!

                                                                           ——By 作者:新晓·故知


本节重点

  • 认识IP地址, 端口号, 网络字节序等网络编程中的基本概念;
  • 学习socket api的基本用法;
  • 能够实现一个简单的udp客户端/服务器;

1. 网络基础知识

1.1 理解源IP地址和目的IP地址

在IP数据包头部中, 有两个IP地址, 分别叫做源IP地址, 和目的IP地址.
思考: 我们光有IP地址就可以完成通信了嘛? 想象一下发qq消息的例子, 有了IP地址能够把消息发送到对方的机器上, 但是还需要有一个其他的标识来区分出, 这个数据要给哪个程序进行解析。

 

在进行网络通信的时候,不仅需要考虑两台主机间互相交互数据,本质上,进行数据交互是用户与用户在进行交互!而用户的身份通常书程序体现的,程序在运行中就是在进程中,

主机间通信的本质是:在各自的主机上的进程在交互数据!

IP地址可以完成主机与主机的通信,而主机上各自的通信进程,才是发送和接受数据的一方!

1.2 认识端口号

端口号(port)是传输层协议的内容.
  • 端口号是一个2字节16位的整数;
  • 端口号用来标识一个进程, 告诉操作系统, 当前的这个数据要交给哪一个进程来处理;
  • IP地址 + 端口号能够标识网络上的某一台主机的某一个进程;
  • 一个端口号只能被一个进程占用.

IP:确保主机的唯一性
port:确保该主机上的进程的唯一性

IP-PORT: 标识互联网中唯一的一个进程       ——> socket  ——>网络通信的本质也是进程间通信!

1.3 理解 "端口号" 和 "进程ID"

我们之前在学习系统编程的时候, 学习了 pid 表示唯一一个进程; 此处我们的端口号也是唯一表示一个进程. 那么这两者之间是怎样的关系?
另外, 一个进程可以绑定多个端口号; 但是一个端口号不能被多个进程绑定;

1.3.1 理解源端口号和目的端口号

送快递例子
传输层协议(TCP和UDP)的数据段中有两个端口号, 分别叫做源端口号和目的端口号. 就是在描述 "数据是谁发的, 要发给谁";

 

1.4 认识TCP协议

此处我们先对TCP(Transmission Control Protocol 传输控制协议)有一个直观的认识; 后面我们再详细讨论TCP的一些细节问题.
  • 传输层协议
  • 有连接
  • 可靠传输
  • 面向字节流

1.5 认识UDP协议

此处我们也是对UDP(User Datagram Protocol 用户数据报协议)有一个直观的认识; 后面再详细讨论.
  • 传输层协议
  • 无连接
  • 不可靠传输
  • 面向数据报

 

1.6 网络字节序

我们已经知道,内存中的多字节数据相对于内存地址有大端和小端之分, 磁盘文件中的多字节数据相对于文件中的偏移地址也有大端小端之分, 网络数据流同样有大端小端之分. 那么如何定义网络数据流的地址呢?
  • 发送主机通常将发送缓冲区中的数据按内存地址从低到高的顺序发出;
  • 接收主机把从网络上接到的字节依次保存在接收缓冲区中,也是按内存地址从低到高的顺序保存;
  • 因此,网络数据流的地址应这样规定:先发出的数据是低地址,后发出的数据是高地址.
  • TCP/IP协议规定,网络数据流应采用大端字节序,即低地址高字节.
  • 不管这台主机是大端机还是小端机, 都会按照这个TCP/IP规定的网络字节序来发送/接收数据;
  • 如果当前发送主机是小端, 就需要先将数据转成大端; 否则就忽略, 直接发送即可
为使网络程序具有可移植性,使同样的C代码在大端和小端计算机上编译后都能正常运行,可以调用以下库函数做网络字节序和主机字节序的转换
  • 这些函数名很好记,h表示host,n表示network,l表示32位长整数,s表示16位短整数。
  • 例如htonl表示将32位的长整数从主机字节序转换为网络字节序,例如将IP地址转换后准备发送。
  • 如果主机是小端字节序,这些函数将参数做相应的大小端转换然后返回;
  • 如果主机是大端字节序,这些 函数不做转换,将参数原封不动地返回。

2. socket编程接口

2.1 socket 常见API


    
  1. // 创建 socket 文件描述符 (TCP/UDP, 客户端 + 服务器)
  2. int socket(int domain, int type, int protocol);
  3. // 绑定端口号 (TCP/UDP, 服务器)      
  4. int bind(int socket, const struct sockaddr* address, socklen_t address_len);
  5. // 开始监听socket (TCP, 服务器)
  6. int listen(int socket, int backlog);
  7. // 接收请求 (TCP, 服务器)
  8. int accept(int socket, struct sockaddr* address, socklen_t* address_len);
  9. // 建立连接 (TCP, 客户端)
  10. int connect(int sockfd, const struct sockaddr* addr, socklen_t addrlen);

2.2 sockaddr结构

socket API是一层抽象的网络编程接口,适用于各种底层网络协议,如IPv4、IPv6,以及后面要讲的UNIX Domain Socket. 然而, 各种网络协议的地址格式并不相同。
  • IPv4和IPv6的地址格式定义在netinet/in.h中,IPv4地址用sockaddr_in结构体表示,包括16位地址类型, 16位端口号和32位IP地址.
  • IPv4、IPv6地址类型分别定义为常数AF_INET、AF_INET6. 这样,只要取得某种sockaddr结构体的首地址, 不需要知道具体是哪种类型的sockaddr结构体,就可以根据地址类型字段确定结构体中的内容.
  • socket API可以都用struct sockaddr *类型表示, 在使用的时候需要强制转化成sockaddr_in; 这样的好处是程序的通用性, 可以接收IPv4, IPv6, 以及UNIX Domain Socket各种类型的sockaddr结构体指针做为参数;

2.3 sockaddr 结构

2.4 sockaddr_in 结构

虽然socket api的接口是sockaddr, 但是我们真正在基于IPv4编程时, 使用的数据结构是sockaddr_in; 这个结构里主
要有三部分信息: 地址类型, 端口号, IP地址.

2.5 in_addr结构

in_addr用来表示一个IPv4的IP地址. 其实就是一个32位的整数

 

3. 简单的UDP网络程序

实现一个简单的英译汉的功能
备注: 代码中会用到 地址转换函数 . 参考接下来的章节

3.1 封装 UdpSocket

udp_socket.hpp:


   
  1. #pragma once
  2. #include <stdio.h>
  3. #include <string.h>
  4. #include <stdlib.h>
  5. #include <cassert>
  6. #include <string>
  7. #include <unistd.h>
  8. #include <sys/socket.h>
  9. #include <netinet/in.h>
  10. #include <arpa/inet.h>
  11. typedef struct sockaddr sockaddr;
  12. typedef struct sockaddr_in sockaddr_in;
  13. class UdpSocket
  14. {
  15. public:
  16. UdpSocket() : fd_( -1)
  17. {
  18. }
  19. bool Socket()
  20. {
  21. fd_ = socket(AF_INET, SOCK_DGRAM, 0);
  22. if (fd_ < 0)
  23. {
  24. perror( "socket");
  25. return false;
  26. }
  27. return true;
  28. }
  29. bool Close()
  30. {
  31. close(fd_);
  32. return true;
  33. }
  34. bool Bind(const std::string &ip, uint16_t port)
  35. {
  36. sockaddr_in addr;
  37. addr.sin_family = AF_INET;
  38. addr.sin_addr.s_addr = inet_addr(ip. c_str());
  39. addr.sin_port = htons(port);
  40. int ret = bind(fd_, (sockaddr *)&addr, sizeof(addr));
  41. if (ret < 0)
  42. {
  43. perror( "bind");
  44. return false;
  45. }
  46. return true;
  47. }
  48. bool RecvFrom(std::string *buf, std::string *ip = NULL, uint16_t *port = NULL)
  49. {
  50. char tmp[ 1024 * 10] = { 0};
  51. sockaddr_in peer;
  52. socklen_t len = sizeof(peer);
  53. ssize_t read_size = recvfrom(fd_, tmp, sizeof(tmp) - 1, 0, (sockaddr *)&peer, &len);
  54. if (read_size < 0)
  55. {
  56. perror( "recvfrom");
  57. return false;
  58. }
  59. // 将读到的缓冲区内容放到输出参数中
  60. buf-> assign(tmp, read_size);
  61. if (ip != NULL)
  62. {
  63. *ip = inet_ntoa(peer.sin_addr);
  64. }
  65. if (port != NULL)
  66. {
  67. *port = ntohs(peer.sin_port);
  68. }
  69. return true;
  70. }
  71. bool SendTo(const std::string &buf, const std::string &ip, uint16_t port)
  72. {
  73. sockaddr_in addr;
  74. addr.sin_family = AF_INET;
  75. addr.sin_addr.s_addr = inet_addr(ip. c_str());
  76. addr.sin_port = htons(port);
  77. ssize_t write_size = sendto(fd_, buf. data(), buf. size(), 0, (sockaddr *)&addr, sizeof(addr));
  78. if (write_size < 0)
  79. {
  80. perror( "sendto");
  81. return false;
  82. }
  83. return true;
  84. }
  85. private:
  86. int fd_;
  87. };

3.2 UDP通用服务器

udp_server.hpp:

 


   
  1. #pragma once
  2. #include "udp_socket.hpp"
  3. // C 式写法
  4. // typedef void (*Handler)(const std::string& req, std::string* resp);
  5. // C++ 11 式写法, 能够兼容函数指针, 仿函数, 和 lambda
  6. #include <functional>
  7. typedef std::function< void( const std::string &, std::string *resp)> Handler;
  8. class UdpServer
  9. {
  10. public:
  11. UdpServer()
  12. {
  13. assert(sock_. Socket());
  14. }
  15. ~ UdpServer()
  16. {
  17. sock_. Close();
  18. }
  19. bool Start(const std::string &ip, uint16_t port, Handler handler)
  20. {
  21. // 1. 创建 socket
  22. // 2. 绑定端口号
  23. bool ret = sock_. Bind(ip, port);
  24. if (!ret)
  25. {
  26. return false;
  27. }
  28. // 3. 进入事件循环
  29. for (;;)
  30. {
  31. // 4. 尝试读取请求
  32. std::string req;
  33. std::string remote_ip;
  34. uint16_t remote_port = 0;
  35. bool ret = sock_. RecvFrom(&req, &remote_ip, &remote_port);
  36. if (!ret)
  37. {
  38. continue;
  39. }
  40. std::string resp;
  41. // 5. 根据请求计算响应
  42. handler(req, &resp);
  43. // 6. 返回响应给客户端
  44. sock_. SendTo(resp, remote_ip, remote_port);
  45. printf( "[%s:%d] req: %s, resp: %s\n", remote_ip. c_str(), remote_port, req. c_str(), resp. c_str());
  46. }
  47. sock_. Close();
  48. return true;
  49. }
  50. private:
  51. UdpSocket sock_;
  52. };

3.3 实现英译汉服务器

以上代码是对 udp 服务器进行通用接口的封装. 基于以上封装, 实现一个查字典的服务器就很容易了.
dict_server.cc:

    
  1. #include "udp_server.hpp"
  2. #include <unordered_map>
  3. #include <iostream>
  4. std::unordered_map<std::string, std::string> g_dict;
  5. void Translate(const std::string &req, std::string *resp)
  6. {
  7. auto it = g_dict. find(req);
  8. if (it == g_dict. end())
  9. {
  10. *resp = "未查到!";
  11. return;
  12. }
  13. *resp = it->second;
  14. }
  15. int main(int argc, char *argv[])
  16. {
  17. if (argc != 3)
  18. {
  19. printf( "Usage ./dict_server [ip] [port]\n");
  20. return 1;
  21. }
  22. // 1. 数据初始化
  23. g_dict. insert(std:: make_pair( "hello", "你好"));
  24. g_dict. insert(std:: make_pair( "world", "世界"));
  25. g_dict. insert(std:: make_pair( "c++", "一种面向对象程序语言"));
  26. g_dict. insert(std:: make_pair( "Linux", "一种操作系统"));
  27. // 2. 启动服务器
  28. UdpServer server;
  29. server. Start(argv[ 1], atoi(argv[ 2]), Translate);
  30. return 0;
  31. }

3.4 UDP通用客户端

 udp_client.hpp:


    
  1. #pragma once
  2. #include "udp_socket.hpp"
  3. class UdpClient
  4. {
  5. public:
  6. UdpClient( const std::string &ip, uint16_t port) : ip_(ip), port_(port)
  7. {
  8. assert(sock_. Socket());
  9. }
  10. ~ UdpClient()
  11. {
  12. sock_. Close();
  13. }
  14. bool RecvFrom(std::string *buf)
  15. {
  16. return sock_. RecvFrom(buf);
  17. }
  18. bool SendTo(const std::string &buf)
  19. {
  20. return sock_. SendTo(buf, ip_, port_);
  21. }
  22. private:
  23. UdpSocket sock_;
  24. // 服务器端的 IP 和 端口号
  25. std::string ip_;
  26. uint16_t port_;
  27. };

3.5 实现英译汉客户端  

dict_client.cc:

 

 

3.6 地址转换函数

本节只介绍基于IPv4的socket网络编程,sockaddr_in中的成员struct in_addr sin_addr表示32位 的IP 地址但是我们通常用点分十进制的字符串表示IP 地址,以下函数可以在字符串表示 和in_addr表示之间转换;

其中inet_pton和inet_ntop不仅可以转换IPv4的in_addr,还可以转换IPv6的in6_addr,因此函数接口是void *addrptr。
 

3.7 关于inet_ntoa

inet_ntoa这个函数返回了一个char*, 很显然是这个函数自己在内部为我们申请了一块内存来保存ip的结果. 那么是否需要调用者手动释放呢?
man手册上说, inet_ntoa函数, 是把这个返回结果放到了静态存储区. 这个时候不需要我们手动进行释放.
那么问题来了, 如果我们调用多次这个函数, 会有什么样的效果呢? 参见如下代码:

运行结果如下:

 

因为inet_ntoa把结果放到自己内部的一个静态存储区, 这样第二次调用时的结果会覆盖掉上一次的结果.
  • 思考: 如果有多个线程调用 inet_ntoa, 是否会出现异常情况呢?
  • 在APUE中, 明确提出inet_ntoa不是线程安全的函数;
  • 但是在centos7上测试, 并没有出现问题, 可能内部的实现加了互斥锁;
  • 同学们课后自己写程序验证一下在自己的机器上inet_ntoa是否会出现多线程的问题;
  • 在多线程环境下, 推荐使用inet_ntop, 这个函数由调用者提供一个缓冲区保存结果, 可以规避线程安全问题;
多线程调用inet_ntoa代码示例如下:

 


       
  1. #include <stdio.h>
  2. #include <unistd.h>
  3. #include <sys/socket.h>
  4. #include <netinet/in.h>
  5. #include <arpa/inet.h>
  6. #include <pthread.h>
  7. using namespace std;
  8. void *Func1(void *p)
  9. {
  10. struct sockaddr_in *addr = ( struct sockaddr_in *)p;
  11. while ( 1)
  12. {
  13. char *ptr = inet_ntoa(addr->sin_addr);
  14. printf( "addr1: %s\n", ptr);
  15. }
  16. return NULL;
  17. }
  18. void *Func2(void *p)
  19. {
  20. struct sockaddr_in *addr = ( struct sockaddr_in *)p;
  21. while ( 1)
  22. {
  23. char *ptr = inet_ntoa(addr->sin_addr);
  24. printf( "addr2: %s\n", ptr);
  25. }
  26. return NULL;
  27. }
  28. int main()
  29. {
  30. pthread_t tid1 = 0;
  31. struct sockaddr_in addr1;
  32. struct sockaddr_in addr2;
  33. addr1.sin_addr.s_addr = 0;
  34. addr2.sin_addr.s_addr = 0xffffffff;
  35. pthread_create(&tid1, NULL, Func1, &addr1);
  36. pthread_t tid2 = 0;
  37. pthread_create(&tid2, NULL, Func2, &addr2);
  38. pthread_join(tid1, NULL);
  39. pthread_join(tid2, NULL);
  40. return 0;
  41. }

makefile: 


       
  1. .PHONY:all
  2. all:dictclient dictserver inetntoa
  3. dictclient:dict_client.cc
  4. g++ -o $@ $^ -std=c++ 11
  5. dictserver:dict_server.cc
  6. g++ -o $@ $^ -std=c++ 11
  7. inetntoa:inet_ntoa.cc
  8. g++ -o $@ $^ -lpthread -std=c++ 11
  9. .PHONY:clean
  10. clean:
  11. rm -f dictclient dictserver inetntoa

 

4. udp网络编程

4.1 网络接口函数及测试

测试1:


   
  1. #include <iostream>
  2. #include <string>
  3. #include <cstring>
  4. #include <unistd.h>
  5. #include <stdlib.h>
  6. #include <ctype.h>
  7. #include <unordered_map>
  8. #include <sys/types.h>
  9. #include <sys/socket.h>
  10. #include <netinet/in.h>
  11. #include <arpa/inet.h>
  12. #include "Log.hpp"
  13. class UdpServer
  14. {
  15. public:
  16. UdpServer()
  17. {}
  18. ~ UdpServer()
  19. {}
  20. public:
  21. void init()
  22. {}
  23. void start()
  24. {}
  25. private:
  26. int sockfd_;
  27. };
  28. int main(int argc, char *argv[])
  29. {
  30. // UdpServer svr;
  31. // svr.init();
  32. // svr.start();
  33. int fd= socket(AF_INET,SOCK_DGRAM, 0);
  34. if(fd < 0)
  35. {
  36. logMessage(FATAL, "%s:%d", strerror(errno),fd);
  37. exit( 1);
  38. }
  39. logMessage(DEBUG, "socket create success! fd: %d",fd);
  40. return 0;
  41. }

 


   
  1. #pragma once
  2. #include <cstdio>
  3. #include <ctime>
  4. #include <cstdarg>
  5. #include <cassert>
  6. #include <cstring>
  7. #include <cerrno>
  8. #include <stdlib.h>
  9. #define DEBUG 0
  10. #define NOTICE 1
  11. #define WARINING 2
  12. #define FATAL 3
  13. const char *log_level[]={ "DEBUG", "NOTICE", "WARINING", "FATAL"};
  14. // logMessage(DEBUG, "%d", 10);
  15. void logMessage(int level, const char *format, ...)
  16. {
  17. assert(level >= DEBUG);
  18. assert(level <= FATAL);
  19. char *name = getenv( "USER");
  20. char logInfo[ 1024];
  21. va_list ap; // ap -> char*
  22. va_start(ap, format);
  23. vsnprintf(logInfo, sizeof(logInfo) -1, format, ap);
  24. va_end(ap); // ap = NULL
  25. FILE *out = (level == FATAL) ? stderr:stdout;
  26. fprintf(out, "%s | %u | %s | %s\n", \
  27. log_level[level], \
  28. ( unsigned int) time( nullptr),\
  29. name == nullptr ? "unknow":name,\
  30. logInfo);
  31. // char *s = format;
  32. // while(s){
  33. // case '%':
  34. // if(*(s+1) == 'd') int x = va_arg(ap, int);
  35. // break;
  36. // }
  37. }

测试2:

 


   
  1. #include <iostream>
  2. #include <string>
  3. #include <cstring>
  4. #include <unistd.h>
  5. #include <stdlib.h>
  6. #include <ctype.h>
  7. #include <unordered_map>
  8. #include <sys/types.h>
  9. #include <sys/socket.h>
  10. #include <netinet/in.h>
  11. #include <arpa/inet.h>
  12. #include "Log.hpp"
  13. static void Usage(const std::string porc)
  14. {
  15. std::cout << "Usage:\n\t" << porc << " port [ip]" << std::endl;
  16. }
  17. /// @brief 我们想写一个简单的udpSever
  18. /// 云服务器有一些特殊情况:
  19. /// 1. 禁止你bind云服务器上的任何确定IP, 只能使用INADDR_ANY,如果你是虚拟机,随意
  20. class UdpServer
  21. {
  22. public:
  23. UdpServer( int port, std::string ip = "") : port_(( uint16_t)port), ip_(ip), sockfd_( -1)
  24. {
  25. }
  26. ~ UdpServer()
  27. {
  28. }
  29. public:
  30. void init()
  31. {
  32. // 1. 创建socket套接字
  33. sockfd_ = socket(AF_INET, SOCK_DGRAM, 0); // 就是打开了一个文件
  34. if (sockfd_ < 0)
  35. {
  36. logMessage(FATAL, "socket:%s:%d", strerror(errno), sockfd_);
  37. exit( 1);
  38. }
  39. logMessage(DEBUG, "socket create success! sockfd_: %d", sockfd_);
  40. // 2. 绑定网络信息,指明ip+port
  41. // 2.1 先填充基本信息到 struct sockaddr_in
  42. struct sockaddr_in local; // local在哪里开辟的空间? 用户栈 -> 临时变量 -> 写入内核中
  43. bzero(&local, sizeof(local)); // memset
  44. // 填充协议家族,域
  45. local.sin_family = AF_INET;
  46. // 填充服务器对应的端口号信息,一定是会发给对方的,port_一定会到网络中
  47. local.sin_port = htons(port_);
  48. // 服务器都必须具有IP地址,"xx.yy.zz.aaa",字符串风格点分十进制 -> 4字节IP -> uint32_t ip
  49. // INADDR_ANY(0): 程序员不关心会bind到哪一个ip, 任意地址bind,强烈推荐的做法,所有服务器一般的做法
  50. // inet_addr: 指定填充确定的IP,特殊用途,或者测试时使用,除了做转化,还会自动给我们进行 h—>n
  51. local.sin_addr.s_addr = ip_. empty() ? htonl(INADDR_ANY) : inet_addr(ip_. c_str());
  52. // 2.2 bind 网络信息
  53. if ( bind(sockfd_, ( const struct sockaddr *)&local, sizeof(local)) == -1)
  54. {
  55. logMessage(FATAL, "bind: %s:%d", strerror(errno), sockfd_);
  56. exit( 2);
  57. }
  58. logMessage(DEBUG, "socket bind success! sockfd_: %d", sockfd_);
  59. // done
  60. }
  61. void start()
  62. {
  63. // 服务器设计的时候,服务器都是死循环
  64. // demo1
  65. while ( true)
  66. {
  67. logMessage(NOTICE, "server 提供 service 中....");
  68. sleep( 1);
  69. }
  70. }
  71. private:
  72. // 服务器必须得有端口号信息
  73. uint16_t port_;
  74. // 服务器必须得有ip地址
  75. std::string ip_;
  76. // 服务器的socket fd信息
  77. int sockfd_;
  78. // onlineuser
  79. std::unordered_map<std::string, struct sockaddr_in> users;
  80. };
  81. // struct client{
  82. // struct sockaddr_in peer;
  83. // uint64_t when; //peer如果在when之前没有再给我发消息,我就删除这用户
  84. // }
  85. // ./udpServer port [ip]
  86. int main(int argc, char *argv[])
  87. {
  88. if (argc != 2 && argc != 3) //反面:argc == 2 || argc == 3
  89. {
  90. Usage(argv[ 0]);
  91. exit( 3);
  92. }
  93. uint16_t port = atoi(argv[ 1]);
  94. std::string ip;
  95. if (argc == 3)
  96. {
  97. ip = argv[ 2];
  98. }
  99. UdpServer svr(port, ip);
  100. svr. init();
  101. svr. start();
  102. return 0;
  103. }

 

 

 

 

 

4.2 UDP程序完整代码

udpServer.cc:


   
  1. #include <iostream>
  2. #include <string>
  3. #include <cstring>
  4. #include <unistd.h>
  5. #include <stdlib.h>
  6. #include <ctype.h>
  7. #include <unordered_map>
  8. #include <sys/types.h>
  9. #include <sys/socket.h>
  10. #include <netinet/in.h>
  11. #include <arpa/inet.h>
  12. #include "Log.hpp"
  13. static void Usage(const std::string porc)
  14. {
  15. std::cout << "Usage:\n\t" << porc << " port [ip]" << std::endl;
  16. }
  17. /// @brief 我们想写一个简单的udpSever
  18. /// 云服务器有一些特殊情况:
  19. /// 1. 禁止你bind云服务器上的任何确定IP, 只能使用INADDR_ANY,如果你是虚拟机,随意
  20. class UdpServer
  21. {
  22. public:
  23. UdpServer( int port, std::string ip = "") : port_(( uint16_t)port), ip_(ip), sockfd_( -1)
  24. {
  25. }
  26. ~ UdpServer()
  27. {
  28. }
  29. public:
  30. void init()
  31. {
  32. // 1. 创建socket套接字
  33. sockfd_ = socket(AF_INET, SOCK_DGRAM, 0); // 就是打开了一个文件
  34. if (sockfd_ < 0)
  35. {
  36. logMessage(FATAL, "socket:%s:%d", strerror(errno), sockfd_);
  37. exit( 1);
  38. }
  39. logMessage(DEBUG, "socket create success: %d", sockfd_);
  40. // 2. 绑定网络信息,指明ip+port
  41. // 2.1 先填充基本信息到 struct sockaddr_in
  42. struct sockaddr_in local; // local在哪里开辟的空间? 用户栈 -> 临时变量 -> 写入内核中
  43. bzero(&local, sizeof(local)); // memset
  44. // 填充协议家族,域
  45. local.sin_family = AF_INET;
  46. // 填充服务器对应的端口号信息,一定是会发给对方的,port_一定会到网络中
  47. local.sin_port = htons(port_);
  48. // 服务器都必须具有IP地址,"xx.yy.zz.aaa",字符串风格点分十进制 -> 4字节IP -> uint32_t ip
  49. // INADDR_ANY(0): 程序员不关心会bind到哪一个ip, 任意地址bind,强烈推荐的做法,所有服务器一般的做法
  50. // inet_addr: 指定填充确定的IP,特殊用途,或者测试时使用,除了做转化,还会自动给我们进行 h—>n
  51. local.sin_addr.s_addr = ip_. empty() ? htonl(INADDR_ANY) : inet_addr(ip_. c_str());
  52. // 2.2 bind 网络信息
  53. if ( bind(sockfd_, ( const struct sockaddr *)&local, sizeof(local)) == -1)
  54. {
  55. logMessage(FATAL, "bind: %s:%d", strerror(errno), sockfd_);
  56. exit( 2);
  57. }
  58. logMessage(DEBUG, "socket bind success: %d", sockfd_);
  59. // done
  60. }
  61. void start()
  62. {
  63. // 服务器设计的时候,服务器都是死循环
  64. char inbuffer[ 1024]; //将来读取到的数据,都放在这里
  65. char outbuffer[ 1024]; //将来发送的数据,都放在这里
  66. while ( true)
  67. {
  68. struct sockaddr_in peer; //输出型参数
  69. socklen_t len = sizeof(peer); //输入输出型参数
  70. // demo2
  71. // UDP无连接的
  72. // 对方给你发了消息,你想不想给对方回消息?要的!后面的两个参数是输出型参数
  73. ssize_t s = recvfrom(sockfd_, inbuffer, sizeof(inbuffer) - 1, 0, ( struct sockaddr *)&peer, &len);
  74. if (s > 0)
  75. {
  76. inbuffer[s] = 0; //当做字符串
  77. }
  78. else if (s == -1)
  79. {
  80. logMessage(WARINING, "recvfrom: %s:%d", strerror(errno), sockfd_);
  81. continue;
  82. }
  83. // 读取成功的,除了读取到对方的数据,你还要读取到对方的网络地址[ip:port]
  84. std::string peerIp = inet_ntoa(peer.sin_addr); //拿到了对方的IP
  85. uint32_t peerPort = ntohs(peer.sin_port); // 拿到了对方的port
  86. checkOnlineUser(peerIp, peerPort, peer); //如果存在,什么都不做,如果不存在,就添加
  87. // 打印出来客户端给服务器发送过来的消息
  88. logMessage(NOTICE, "[%s:%d]# %s", peerIp. c_str(), peerPort, inbuffer);
  89. //大小写转化测试
  90. // for(int i = 0; i < strlen(inbuffer); i++)
  91. // {
  92. // if(isalpha(inbuffer[i]) && islower(inbuffer[i])) outbuffer[i] = toupper(inbuffer[i]);
  93. // else outbuffer[i] = toupper(inbuffer[i]);
  94. // }
  95. messageRoute(peerIp, peerPort,inbuffer); //消息路由
  96. // 线程池!
  97. // sendto(sockfd_, outbuffer, strlen(outbuffer), 0, (struct sockaddr*)&peer, len);
  98. // demo1
  99. // logMessage(NOTICE, "server 提供 service 中....");
  100. // sleep(1);
  101. }
  102. }
  103. void checkOnlineUser(std::string &ip, uint32_t port, struct sockaddr_in &peer)
  104. {
  105. std::string key = ip;
  106. key += ":";
  107. key += std:: to_string(port);
  108. auto iter = users. find(key);
  109. if(iter == users. end())
  110. {
  111. users. insert({key, peer});
  112. }
  113. else
  114. {
  115. // iter->first, iter->second->
  116. // do nothing
  117. }
  118. }
  119. void messageRoute(std::string ip, uint32_t port, std::string info)
  120. {
  121. std::string message = "[";
  122. message += ip;
  123. message += ":";
  124. message += std:: to_string(port);
  125. message += "]# ";
  126. message += info;
  127. for( auto &user : users)
  128. {
  129. sendto(sockfd_, message. c_str(), message. size(), 0, ( struct sockaddr*)&(user.second), sizeof(user.second));
  130. }
  131. }
  132. private:
  133. // 服务器必须得有端口号信息
  134. uint16_t port_;
  135. // 服务器必须得有ip地址
  136. std::string ip_;
  137. // 服务器的socket fd信息
  138. int sockfd_;
  139. // onlineuser
  140. std::unordered_map<std::string, struct sockaddr_in> users;
  141. };
  142. // struct client{
  143. // struct sockaddr_in peer;
  144. // uint64_t when; //peer如果在when之前没有再给我发消息,我就删除这用户
  145. // }
  146. // ./udpServer port [ip]
  147. int main(int argc, char *argv[])
  148. {
  149. if (argc != 2 && argc != 3) //反面:argc == 2 || argc == 3
  150. {
  151. Usage(argv[ 0]);
  152. exit( 3);
  153. }
  154. uint16_t port = atoi(argv[ 1]);
  155. std::string ip;
  156. if (argc == 3)
  157. {
  158. ip = argv[ 2];
  159. }
  160. UdpServer svr(port, ip);
  161. svr. init();
  162. svr. start();
  163. return 0;
  164. }
  165. // struct ip
  166. // {
  167. // uint32_t part1:8;
  168. // uint32_t part2:8;
  169. // uint32_t part3:8;
  170. // uint32_t part4:8;
  171. // }
  172. // struct ip ip_;
  173. // ip_.part1 = s.substr();

udpClient.cc:


   
  1. #include <iostream>
  2. #include <string>
  3. #include <cstdlib>
  4. #include <cassert>
  5. #include <unistd.h>
  6. #include <strings.h>
  7. #include <netinet/in.h>
  8. #include <arpa/inet.h>
  9. #include <sys/types.h>
  10. #include <sys/socket.h>
  11. #include <pthread.h>
  12. struct sockaddr_in server;
  13. static void Usage(std::string name)
  14. {
  15. std::cout << "Usage:\n\t" << name << " server_ip server_port" << std::endl;
  16. }
  17. void *recverAndPrint(void *args)
  18. {
  19. while ( true)
  20. {
  21. int sockfd = *( int *)args;
  22. char buffer[ 1024];
  23. struct sockaddr_in temp;
  24. socklen_t len = sizeof(temp);
  25. ssize_t s = recvfrom(sockfd, buffer, sizeof(buffer), 0, ( struct sockaddr *)&temp, &len);
  26. if (s > 0)
  27. {
  28. buffer[s] = 0;
  29. std::cout << "server echo# " << buffer << std::endl;
  30. }
  31. }
  32. }
  33. // ./udpClient server_ip server_port
  34. // 如果一个客户端要连接server必须知道server对应的ip和port
  35. int main(int argc, char *argv[])
  36. {
  37. if (argc != 3)
  38. {
  39. Usage(argv[ 0]);
  40. exit( 1);
  41. }
  42. // 1. 根据命令行,设置要访问的服务器IP
  43. std::string server_ip = argv[ 1];
  44. uint16_t server_port = atoi(argv[ 2]);
  45. // 2. 创建客户端
  46. // 2.1 创建socket
  47. int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
  48. assert(sockfd > 0);
  49. // 2.2 client 需不需要bind??? 需要bind,但是不需要用户自己bind,而是os自动给你bind
  50. // 所谓的"不需要",指的是: 不需要用户自己bind端口信息!因为OS会自动给你绑定,你也最好这么做!
  51. // 如果我非要自己bind呢?可以!严重不推荐!
  52. // 所有的客户端软件 <-> 服务器 通信的时候,必须得有 client[ip:port] <-> server[ip:port]
  53. // 为什么呢??client很多,不能给客户端bind指定的port,port可能被别的client使用了,你的client就无法启动了
  54. // 那么server凭什么要bind呢??server提供的服务,必须被所有人知道!server不能随便改变!
  55. // 2.2 填写服务器对应的信息
  56. bzero(&server, sizeof server);
  57. server.sin_family = AF_INET;
  58. server.sin_port = htons(server_port);
  59. server.sin_addr.s_addr = inet_addr(server_ip. c_str());
  60. pthread_t t;
  61. pthread_create(&t, nullptr, recverAndPrint, ( void *)&sockfd);
  62. // 3. 通讯过程
  63. std::string buffer;
  64. while ( true)
  65. {
  66. std::cerr << "Please Enter# ";
  67. std:: getline(std::cin, buffer);
  68. // 发送消息给server
  69. // 首次调用sendto函数的时候,我们的client会自动bind自己的ip和port
  70. sendto(sockfd, buffer. c_str(), buffer. size(), 0, ( const struct sockaddr *)&server, sizeof(server));
  71. }
  72. close(sockfd);
  73. return 0;
  74. }

Log.hpp:


   
  1. #pragma once
  2. #include <cstdio>
  3. #include <ctime>
  4. #include <cstdarg>
  5. #include <cassert>
  6. #include <cstring>
  7. #include <cerrno>
  8. #include <stdlib.h>
  9. #define DEBUG 0
  10. #define NOTICE 1
  11. #define WARINING 2
  12. #define FATAL 3
  13. const char *log_level[]={ "DEBUG", "NOTICE", "WARINING", "FATAL"};
  14. // logMessage(DEBUG, "%d", 10);
  15. void logMessage(int level, const char *format, ...)
  16. {
  17. assert(level >= DEBUG);
  18. assert(level <= FATAL);
  19. char *name = getenv( "USER");
  20. char logInfo[ 1024];
  21. va_list ap; // ap -> char*
  22. va_start(ap, format);
  23. vsnprintf(logInfo, sizeof(logInfo) -1, format, ap);
  24. va_end(ap); // ap = NULL
  25. FILE *out = (level == FATAL) ? stderr:stdout;
  26. fprintf(out, "%s | %u | %s | %s\n", \
  27. log_level[level], \
  28. ( unsigned int) time( nullptr),\
  29. name == nullptr ? "unknow":name,\
  30. logInfo);
  31. // char *s = format;
  32. // while(s){
  33. // case '%':
  34. // if(*(s+1) == 'd') int x = va_arg(ap, int);
  35. // break;
  36. // }
  37. }

makefile:

 


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

 

后记:
●由于作者水平有限,文章难免存在谬误之处,敬请读者斧正,俚语成篇,恳望指教!

                                                                           ——By 作者:新晓·故知

 


转载:https://blog.csdn.net/m0_57859086/article/details/128525896
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场