Linux下利用epoll实现两台机器之间的UDP传输丢包统计

公司机房到阿里云的数据传输有时候会出现丢包，所以需要程序来进行查询和排查，公司前辈写了一个程序来进行机器之间的丢包统计。
服务器：pong.cpp

#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#define MAXLINE 1500
#define SERV_PORT 2019    //服务器端口号


void do_echo(int sockfd, struct sockaddr *pcliaddr, socklen_t clilen)
{
    int n, jzq_num = 0;
    socklen_t len;
    char mesg[MAXLINE];
    int i ;


    for(;;)
    {
        len = clilen;
        /* waiting for receive data */
        n = recvfrom(sockfd, mesg, MAXLINE, 0, pcliaddr, &len);
        mesg[n]=0;
        printf("recvfrom data:%d %d\n",n, ++jzq_num);
        printf("%s\n",mesg);
        /* sent data back to client */
        sendto(sockfd, mesg, n, 0, pcliaddr, len);
    }
}


int main(int argc,char *argv[])
{
    int sockfd;
    int m_port = SERV_PORT;
    int result, buf_len;
    int len = sizeof (buf_len);
    struct sockaddr_in servaddr, cliaddr;

    if (argc >=2) sscanf(argv[1],"%d",&m_port);

    sockfd = socket(AF_INET, SOCK_DGRAM, 0); /* create a socket */


    /* init servaddr */
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(m_port);

    printf("port:%d\n", m_port);


    result = getsockopt((unsigned int)sockfd, SOL_SOCKET, SO_RCVBUF, (char *)&buf_len, (socklen_t *)&len);
    printf ("one:getsock_recvbuf-> result:%s buf_len:%d\n", result?"NO":"YES", buf_len);

    if(result == 0 && buf_len < 1024*1024)
    {
        buf_len = 1024*1024;

        setsockopt((unsigned int)sockfd, SOL_SOCKET, SO_RCVBUF, (char *)&buf_len, len);  //接收缓冲区大小为1MB
        printf ("setsock_recvbuf-> result:%s buf_len:%d\n", result?"NO":"YES", buf_len);

        result = getsockopt((unsigned int)sockfd, SOL_SOCKET, SO_RCVBUF, (char *)&buf_len, (socklen_t *)&len);
        printf ("two:getsock_recvbuf-> result:%s buf_len:%d\n", result?"NO":"YES", buf_len);
    }

    /* bind address and port to socket */
    if(bind(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1)
    {
        perror("bind error");
        exit(1);
    }
    printf("UPD servese is bind success and do_echo now\n");
    do_echo(sockfd, (struct sockaddr *)&cliaddr, sizeof(cliaddr));


    return 0;
}

测试程序check.cpp

#include <netinet/in.h>
#include <sys/socket.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/epoll.h>
#include <time.h>
#include <unistd.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <sys/time.h>

#define MAXSIZE     1024
#define FDSIZE      1024
#define EPOLLEVENTS 20

#define SERV_PORT   (9999)//你需要连接的服务器端口号
#define LOCAL_PORT  (0)    //UDP传输端口，0表示系统分配
#define WAIT_TIME   (100)

static int handle_connection(int sockfd, struct sockaddr_in *servaddr, int flag);
static int handle_events(int epollfd,struct epoll_event *events,int num,int sockfd,char *buf, struct sockaddr_in *servaddr);
static int do_read(int epollfd,int fd,int sockfd,char *buf, struct sockaddr_in *servaddr);
static void add_event(int epollfd,int fd,int state);
static void delete_event(int epollfd,int fd,int state);
static void modify_event(int epollfd,int fd,int state);


char send_buf[MAXSIZE];
char ip[50];

int check(char *ip, int port, int flag)
{
    int sockfd;
    struct sockaddr_in  servaddr;
    struct sockaddr_in  clientaddr;
    int res ;

    sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    servaddr.sin_addr.s_addr = inet_addr(ip);

    bzero(&clientaddr, sizeof(clientaddr));
    clientaddr.sin_family = AF_INET;
    clientaddr.sin_port = htons(LOCAL_PORT);
    clientaddr.sin_addr.s_addr = htonl(INADDR_ANY);

    if(bind(sockfd, (struct sockaddr *)&clientaddr, sizeof(clientaddr)) == -1) 
    {   
        perror("bind error");
        exit(1);
    }

    sockaddr_in addrMy;
    memset(&addrMy,0,sizeof(addrMy));
    int leng = sizeof(addrMy);

    if (getsockname(sockfd, (sockaddr*)&addrMy, (socklen_t*)&leng) != 0)
    {
       puts("Getsockname Error!");
       return 0;
    }
    printf("Current Socket PORT->%d ",ntohs(addrMy.sin_port)); 
    fflush(stdout);
    res = handle_connection(sockfd, &servaddr, flag);
    close(sockfd);
    return res;
}

void print_cur_time()
{
     time_t timep;
     struct tm *p;
     time(&timep);
     p = localtime(&timep); //取得当地时间
     printf ("%d-%02d-%02d %02d:%02d:%02d ",(1900+p->tm_year), (1+p->tm_mon), p->tm_mday, p->tm_hour, p->tm_min, p->tm_sec);
     fflush(stdout);
}

int main(int argc,char *argv[])
{
  FILE *fp;
  int num = 0, lost_num = 0;
  while(1)
  {
    fp = fopen("./home/jzq/Check/poxy.txt*/", "r");//自己创目录文件，poxy.txt里是要连的ip地址
    if(fp == NULL)
    {
        printf("The poxy.txt not exit\n");
       return 0;
    }
    while(~fscanf(fp, "%s", ip))
    {
        print_cur_time();
        printf("ALY -> %s ", ip);
        fflush(stdout);
        check(ip, SERV_PORT, 2);
        sleep(0.1);
    }
    fclose(fp);
  }
  return 0;
}


static int handle_connection(int sockfd, struct sockaddr_in *servaddr, int flag)
{
    int epollfd;
    struct epoll_event events[EPOLLEVENTS];
    int ret, len, num = 30;
    int lost_num =0;
    epollfd = epoll_create(FDSIZE);
    add_event(epollfd,sockfd,EPOLLIN);
    while(num--)
    {
        len = 20;              //包长，可以按需要增大，但要注意传输时间要相应增大
        memset(send_buf, '6', len);
        send_buf[len] = 0;

        if(sendto(sockfd, send_buf, len, 0, (struct sockaddr*)servaddr, sizeof(struct sockaddr_in)) <= 0)
    {
        printf("sendto error");
        continue;
    }

        ret = epoll_wait(epollfd, events, epollevents, wait_time);
        if(ret > 0 && (len = handle_events(epollfd, events, ret, sockfd, send_buf, servaddr)) > 0)
    {
            if(!len)
                printf("unkonw\n");
    }
    else
        {
            lost_num++; 
            printf(".");
            fflush(stdout);
        }
    }
    close(epollfd);
    printf("lost:%d%\n", lost_num*100/30);
    return lost_num;
}

static int handle_events(int epollfd,struct epoll_event *events,int num,int sockfd,char *buf, struct sockaddr_in *servaddr)
{
    int fd;
    int i;
    for (i = 0;i < num;i++)
    {
        fd = events[i].data.fd;
        if ((events[i].events & EPOLLIN) && (fd == sockfd))
            return do_read(epollfd,fd,sockfd,buf, servaddr);
    }
    return -1;
}

static int do_read(int epollfd,int fd,int sockfd,char *buf, struct sockaddr_in *servaddr)
{
    socklen_t len = sizeof(struct sockaddr_in);
    return recvfrom(sockfd, buf, MAXSIZE, MSG_DONTWAIT, (struct sockaddr *)servaddr, &len);
}

static void add_event(int epollfd,int fd,int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&ev);
}

static void delete_event(int epollfd,int fd,int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epollfd,EPOLL_CTL_DEL,fd,&ev);
}

static void modify_event(int epollfd,int fd,int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&ev);
}