套接字编程（二）UDP服务端与客户端的通信模拟实现

3、发送数据

① ssize_t sendto(int sockfd, void *buf, size_t dlen,

int flag, struct sockaddr *peer, socklen_t alen);

② sockfd socket返回的套接字描述符

③ buf 发送数据的起始地址

④ dlen 发送数据的长度从buf地址开始发送dlen长度的数据

⑤ flag 默认为0阻塞发送发送缓冲区满了就等着当发送缓冲区有空间了进行发送

⑥ peer 对端地址信息描述了数据要发送给谁

描述了信息送往哪里这里存放的是接收方的信息

⑦ alen 对端地址信息长度

⑧ 返回值成功返回实际发送数据字节长度失败返回-1

4、接收数据

① ssize_t recvfrom(int sockfd, void *buf, size_t dlen,
int flag, struct sockaddr *peer, socklen_t *alen);

② sockfd socket创建的套接字描述符

③ 一块地址空间里面用来存放我接收得到的数据

④ dlen 接收数据的字节长度

⑤ flag 默认为0当socket接收缓冲区为空的时候进行等待有数据了就进行接收

⑥ peer 对端地址信息描述了接收的数据来自哪里。这里存放的是来自发送方的地址信息

⑦ alen 地址信息长度里面即存放了想要接收的数据长度也存放了实际接收的数据长度

⑧ 成功返回实际接收的数据长度失败返回-1

5、功能函数代码

  1 #include<iostream>
  2 #include<string>
  3 #include<unistd.h>
  4 #include<arpa/inet.h> // 字节序转换接口头文件
  5 #include<netinet/in.h>// 地址结构类型以及协议类型头文件
  6 #include<sys/socket.h>// 套接字接口文件
  7 
  8 class UdpSocket
  9 {
 10   private:
 11     int _sockfd;
 12   public:
 13     UdpSocket():_sockfd(-1){}
 14     ~UdpSocket(){Close();}
 15     bool Sockte() // 创建套接字
 16     {
 17       _sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
 18       if(_sockfd<0)
 19       {
 20         perror("socket error\n");
 21         return false;
 22       }
 23       return true;
 24     }
 25     bool Bind(const std::string& ip, uint16_t port)
 26     {
 27         struct sockaddr_in addr; // 定义一个ipv4socket结构体
 28         addr.sin_family = AF_INET;// 初始化addr结构体中的sin_family
 29         addr.sin_port = htons(port);
 30         addr.sin_addr.s_addr = inet_addr(ip.c_str());
 31         socklen_t len = sizeof(struct sockaddr_in);
 32         int ret = bind(_sockfd, (struct sockaddr*)&addr, len);
 33         if(ret<0){
 34           perror("bind error\n");
 35           return false;
 36         }                                                                                
 37         return true;
 38     }
 39     bool Recv(std::string *body, std::string *peer_ip, uint16_t *peer_port)
 40     {
 41         struct sockaddr_in peer;
 42         socklen_t len = sizeof(struct sockaddr_in);
 43         char tmp[4096]={0};
 44         ssize_t ret = recvfrom(_sockfd, tmp, 4096, 0, (struct sockaddr*)&peer, &len);
 45         if(ret<0)
 46         {
 47           perror("recv error");
 48           return false;                                                                  
 49         }
 50         *peer_ip = inet_ntoa(peer.sin_addr);
 51         *peer_port = ntohs(peer.sin_port);
 52         body->assign(tmp, ret); // 从tmp中取出ret长度数据放入body中
 53         return true;
 54     }
 55     bool Send(const std::string &body, const std::string &peer_ip, uint16_t peer_port)
 56     {
 57         struct sockaddr_in addr;
 58         addr.sin_family = AF_INET;
 59         addr.sin_port = htons(peer_port);
 60         addr.sin_addr.s_addr = inet_addr(peer_ip.c_str());
 61         socklen_t len = sizeof(struct sockaddr_in);
 62         ssize_t ret = sendto(_sockfd, body.c_str(), body.size(), 0, (struct sockaddr*)&addr, len);
 63         if(ret < 0){
 64           perror("send error");
 65           return false;
 66         }
 67         return true;
 68     }
 69     bool Close()
 70     {
 71         if(_sockfd != -1)
 72         {
 73           close(_sockfd);
 74           _sockfd = -1;
 75         }
 76         return true;
 77     }
 78 };

6、主main函数

   81 int main(int argc, char* argv[])
   82 {
   83     if(argc != 3)
   84     {
   85       std::cout<<"usage: ./client.cpp 192.168.2.2 9000\n";                             
   86       return -1;
   87     }
   88     std::string srv_ip = argv[1];
   89     uint16_t srv_port = std::atoi(argv[2]);
   90     UdpSocket socket;
   91     // 1、创建套接字
   92     assert(socket.Sockte() == true);
   93     // 2、绑定套接字(不推荐)
   94     while(1){
   95     // 3、发送数据
   96         std::string data;
   97         std::cout<<"client say:";
   98         fflush(stdout);
   99         std::cin>>data;
  100         assert(socket.Send(data, srv_ip, srv_port) == true);
  101         
  102     // 4、接收数据
  103         data.clear();
  104         //assert(socket.Recv(&data, &srv_ip, &srv_port) == true);
  105         assert(socket.Recv(&data) == true);
  106     }
  107     socket.Close();
  108     return 0;
  109 }