手撕测试tcp服务器效率工具——以epoll和io_uring对比为例

#include<stdio.h>
#include<string.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<unistd.h>
#include<sys/time.h>
#include<pthread.h>
#include<arpa/inet.h>
//设置该结构体用于存储线程函数所需的参数
typedef struct test_context_s{
  
  char serverip[16]; //服务器ip
  int port; //服务器端口
  int threadnum;  //线程数量
  int connection; //连接数量，此案例中与线程数量一致
  int requestion; //请求数量，也就是报文数量
  
  int failed; //统计发送失败的次数，有个大概的数就行，所以没用原子变量
}test_context_t;
typedef struct test_context_s test_context_t;
//与服务器建立tcp连接，常规的socket然后connect
int connect_tcpserver(const char* ip,unsigned short port){
  
  int connfd = socket(AF_INET,SOCK_STREAM,0);
  struct sockaddr_in tcpserver_addr;
  memset(&tcpserver_addr,0,sizeof(struct sockaddr_in));
  tcpserver_addr.sin_family = AF_INET;
  tcpserver_addr.sin_addr.s_addr = inet_addr(ip);
  tcpserver_addr.sin_port = htons(port);
  int ret = connect(connfd,(struct sockaddr*)&tcpserver_addr,sizeof(struct sockaddr_in));
  if(ret){
    perror("connect");
    return -1;
  }
  return connfd;
}
#define TIME_SUB_MS(tv1,tv2) ((tv1.tv_sec - tv2.tv_sec)*1000 + (tv1.tv_usec - tv2.tv_usec)/1000)
//要发送给客户端的数据的基本单位
#define TEST_MESSAGE   "ABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890abcdefghijklmnopqrstuvwxyz\r\n"
#define RBUFFER_LENGTH 2048 //读数据的空间的大小，不一定用满
#define WBUFFER_LENGTH 2048 //写数据的空间的大小，不一定用满
//客户端发送并接收数据
int send_recv_tcppkt(int fd){
  char wbuffer[WBUFFER_LENGTH] = {0};
  //设置每次发送的报文包含多少个基本单位
  int i = 0;
  for(i=0;i<8;i++){
    strcpy(wbuffer+i *strlen(TEST_MESSAGE),TEST_MESSAGE);
  }
  //发送报文
  int res = send(fd,wbuffer,strlen(wbuffer),0);
  if(res<0){
    exit(1);
  }
  //接收报文
  char rbuffer[RBUFFER_LENGTH] ={0};
  res = recv(fd,rbuffer,RBUFFER_LENGTH,0);
  if(res<=0){
    exit(1);
  }
  if(strcmp(rbuffer,wbuffer)!=0){
    return -1;
  }
  
}
//线程函数，主要作用是建立连接、发送接收报文
static void *test_qps_entry(void* arg){
  
  test_context_t *pctx = (test_context_t*)arg;
  //建立连接
  int connfd = connect_tcpserver(pctx->serverip,pctx->port);
  if(connfd<0){
    printf("connect_tcpserver failed!\n");
    return NULL;
  } 
  //每个线程要发送的报文数量
  int count = pctx->requestion/pctx->threadnum;
  //发送报文
  int i=0;
  int res;
  while(i++<count){
    res = send_recv_tcppkt(connfd);
    if(res!=0){
      printf("send_recv_tcppkt failed\n");
      pctx->failed++;
      continue;
    }
  }
  return NULL;
}
int main(int argc,char *argv[]){
  
  int ret =0;
  test_context_t ctx ={0};
  
  int opt;
  
  //getopt函数可以一次解析出带有名称的输入参数
  //注意这个函数是线程不安全的，
  while((opt = getopt(argc,argv,"s:p:t:c:n:?"))!=-1){
    switch(opt){
      case 's':
        printf("-s:%s\n",optarg);
        strcpy(ctx.serverip,optarg);//服务器IP
        break;    
       case 'p':
        printf("-p:%s\n",optarg);
        ctx.port = atoi(optarg);//服务器端口
        break;
      case 't':    
        printf("-t:%s\n",optarg);
        ctx.threadnum = atoi(optarg);//线程数
        break;
      case 'c':
        printf("-c:%s\n",optarg);
        ctx.connection = ctx.threadnum;//还是和线程数一致吧  
        break;
      case 'n':
        printf("-n:%s\n",optarg);
        ctx.requestion = atoi(optarg);
        break;
      default:
        return -1;          
    }
  }
  //线程数组
  pthread_t *ptid = malloc(ctx.threadnum *sizeof("pthread_t"));
  //开始大规模发送发送报文
  struct timeval tv_begin;  //记录报文开始发送的时间
  gettimeofday(&tv_begin,NULL);
  int i = 0;
  for(i=0;i<ctx.threadnum;i++){ //建立线程运行线程函数
    pthread_create(&ptid[i],NULL,test_qps_entry,&ctx);
  }
  for(i=0;i<ctx.threadnum;i++){
    pthread_join(ptid[i],NULL);
  }
  struct timeval tv_end;  //记录报文全部发送并接收完毕的时间
  gettimeofday(&tv_end,NULL);
  int time_used = TIME_SUB_MS(tv_end,tv_begin);//计算用时
  printf("success: %d, failed: %d, time_used: %d, qps: %d\n", ctx.requestion-ctx.failed, 
    ctx.failed, time_used, ctx.requestion * 1000 / time_used);
  return 0;
}