linux下搭建能同时接收UDP和TCP连接的socket并发多线程服务器

由于UDP和TCP函数接口不同，所以服务器的处理线程需要能分清此时客户端的链接方式，这里通过传递简单一个pthread_arg.protocol变量来识别

==============================================================

//include.h

…………

…………省略若干头文件、参数、函数声明

…………

struct pthread_arg{  //供线程使用的参数

     char protocol;// protocol =='U'为udp连接 ; protocol == 'T'则为tcp连接；

     int fd;

     struct sockaddr_in sock;

};

==============================================================

//server.c

#include "include.h"

int main(int argc, char ** argv)

{

     struct sockaddr_in server_addr, client_addr;

     struct pthread_arg ptr_arg;//供线程使用的参数

     fd_set fdset;

     pthread_t pthid;

     int tcp_socketfd, tcp_connectfd, udp_connectfd, maxfd;

     int client_len;

     char local_port[32], local_ip[32];

     int reuse = 1;//进程结束后端口允许重用，否则在调试程序时得等系统回收端口，浪费时间

     void *retval;

// struct timeval timeout = { 3, 0};

//--------------------- TCP ----------------------------------------------------------

     if((tcp_socketfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){

          printf("fail to socket!!\n");

          exit(-1);

     }

//  printf("socket success!!\n");

     setsockopt(tcp_socketfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(int));

     bzero(&server_addr, sizeof(server_addr));

     server_addr.sin_family = AF_INET;

     sprintf(local_ip, "%s", LOCAL_IP);

     server_addr.sin_addr.s_addr = inet_addr(local_ip);

     printf("[server ip = %s]\n", local_ip);

     sprintf(local_port, "%d", LOCAL_PORT);

     server_addr.sin_port = htons(atoi(local_port));

     printf("[server port = %s]\n", local_port);

    

     if(bind(tcp_socketfd,  (struct sockaddr *)&server_addr,  sizeof(server_addr)) < 0){

          perror("fail to bind!!");

          exit(-1);

     }

    

     if(listen(tcp_socketfd, 5) < 0){

          perror("fail to listen!!");

          exit(-1);

     }

//--------------------- UDP --------------------------------------------------------------------- 

     if((udp_connectfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0){

          perror("Error : udp fail to socket!!\n");

          exit(-1);

     }

     if(bind(udp_connectfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0){

          perror("Error : udp fail to bind!!\n");

          exit(-1);

     }

     maxfd = (tcp_connectfd > udp_connectfd ? tcp_connectfd : udp_connectfd);

     while(1){

          FD_ZERO(&fdset);

          FD_SET(tcp_socketfd, &fdset);

          FD_SET(udp_connectfd, &fdset);

          if(select(maxfd+1, &fdset, NULL, NULL, NULL)< 0){

               perror("Error: select error!!\n");

               exit(-1);

          }

          //tcp连接------------------------------- 

          //为每一个tcp客户端创建独立的处理线程，线程处理函数为deal_send_pth，参数为ptr_arg

          if(FD_ISSET(tcp_socketfd, &fdset)){

              //循环接受客户端的链接请求

               client_len = sizeof(client_addr);

               if((tcp_connectfd = accept(tcp_socketfd, (struct sockaddr *)&client_addr, (socklen_t *)&(client_len))) < 0){

                    perror("fail to accept!!");

                    return -1;

               }

              

               ptr_arg.protocol = 'T';

               ptr_arg.fd = tcp_connectfd;

               strcpy((char *)&ptr_arg.sock, (const char *)&client_addr);

               if(pthread_create(&pthid, NULL, deal_send_pth, (void *)&ptr_arg) < 0){

                    perror("fail to pthread_create!!");

                    return -1;

               }

               pthread_detach(pthid);//回收线程、非阻塞

          }

          //udp连接--------------------------------

          //为每一个udp客户端创建独立的处理线程，线程处理函数同样为deal_send_pth，参数为ptr_arg 

          if(FD_ISSET(udp_connectfd, &fdset)){

               ptr_arg.protocol = 'U';

               ptr_arg.fd = udp_connectfd;

               strcpy((char *)&ptr_arg.sock, (const char *)&client_addr);

               if(pthread_create(&pthid, NULL, deal_send_pth, (void *)&ptr_arg) < 0){

                    perror("fail to pthread_create!!");

                    return -1;

               }

               pthread_join(pthid, &retval);//回收线程、非阻塞

          }

     }

 return 0;

}

//deal_send_pth（）中可以拿到pth_arg,判断下protocol成员，后面就不多说了！！