linux_socket_thread.zip_linux 网络通信_linux 网络通信_多线程socket

#include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <sys/types.h> #include <netinet/in.h> #include <sys/socket.h> #include <sys/wait.h> #include <pthread.h> #include <sys/time.h> #include <unistd.h> #include <semaphore.h> #define SERVER_PORT 4000/*服务器端口号：4000*/ #define NUM_MAX_CLIENTS 10/*最多接受10个客户端发送消息*/ #define SEND_MAX_DATA_LENS 1024 #define RECV_MAX_DATA_LENS 1024 #define MAX_DATA_NUM 128 struct client_t/*client结构体,包含一个套接字的编号，发送线程和接受线程*/ { int sockid[NUM_MAX_CLIENTS]; pthread_t send_tid[NUM_MAX_CLIENTS]; pthread_t recv_tid[NUM_MAX_CLIENTS]; }; struct client_t g_client; int g_client_id = 0; struct msg{ char data[MAX_DATA_NUM]; }; void *send_msg(void *arg);//接收数据线程函数 void *recv_msg(void *arg);//发送数据线程函数 int get_id(){ for(int i=0;i<NUM_MAX_CLIENTS;i++){ if(g_client.sockid[i]==-1) return i; } return -1; } int main(int argc, char *argv[]) { int sock_fd; int current_client_id; pthread_attr_t attr;/*线程的属性*/ struct sockaddr_in server_addr; struct sockaddr_in client_addr; socklen_t sin_size; int err; int port_reuse = 1; for(int i=0;i<NUM_MAX_CLIENTS;i++){ g_client.sockid[i]=-1;//初始化所有的套接字编号 } /*创建分离线程*/ pthread_attr_init(&attr); pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); /*创建套接字*/ if((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1){ perror("can not get a socket!\n"); exit(1); } server_addr.sin_family = AF_INET; server_addr.sin_port = htons(SERVER_PORT); server_addr.sin_addr.s_addr = INADDR_ANY; memset(server_addr.sin_zero, '\0', sizeof(server_addr.sin_zero)); if((setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &port_reuse, sizeof(int))) == -1){ perror("addr port reuse error!\n"); exit(1); } /*捆绑socket描述符、服务器地址和端口号*/ if(bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1){ perror("bind error\n"); exit(1); } /*监听端口*/ if(listen(sock_fd, NUM_MAX_CLIENTS) == -1){ perror("listen error\n"); exit(1); } printf("服务器监听端口%d...\n",SERVER_PORT); /*等待客户端连接,每连接一个客户端,为这个客户端分别建立一个发送和接收线程*/ while(1){ sin_size = sizeof(struct sockaddr); /*接收新的连接*/ g_client_id=get_id(); { int a=accept(sock_fd, (struct sockaddr *)&client_addr, &sin_size); g_client.sockid[g_client_id] = a; if(g_client.sockid[g_client_id] == -1){ perror("accept error"); } current_client_id = g_client_id; printf("新连接来自客户端[%d]:%s\n\n",current_client_id,inet_ntoa(client_addr.sin_addr)); /*创建接受线程*/ err = pthread_create(&g_client.recv_tid[current_client_id], &attr, recv_msg, (void *)&current_client_id); if(0 != err){ printf("can't create recv_msg thread!\n"); exit(1); } /*创建发送线程*/ err = pthread_create(&g_client.send_tid[current_client_id], &attr, send_msg, (void *)&current_client_id); if(0 != err){ printf("can't create send_msg thread!\n"); exit(1); } } } return 0; } /*接收线程*/ void *recv_msg(void *arg) { int client_id; struct msg server_recv_msg; struct timeval timeout; fd_set recvfds; int numbytes; char buf[RECV_MAX_DATA_LENS]; client_id = *(int *)arg; timeout.tv_sec = 100;//秒 timeout.tv_usec =0;//微秒 int flag=1; while(flag){ FD_ZERO(&recvfds); FD_SET(g_client.sockid[client_id], &recvfds); switch(select(g_client.sockid[client_id] + 1, &recvfds, NULL, NULL, &timeout)){ case -1: printf("Select error!\n"); flag=0; break; case 0: printf("服务器没有任何输入信息，并且客户端%d也没有信息到来，关闭与该客户端的连接...\n",client_id); close(g_client.sockid[client_id]); g_client.sockid[client_id]=-1; flag=0; break;//超时 default: if(FD_ISSET(g_client.sockid[client_id], &recvfds) > 0) { if((numbytes = recv(g_client.sockid[client_id], buf, RECV_MAX_DATA_LENS, 0)) == -1){ perror("receive error\n"); close(g_client.sockid[client_id]); exit(1); } if(0 == numbytes){ //若客户端关闭,一关掉相应的客户端的套接字,二,关掉相应的发送线程,三,关掉本线程 printf("client[%d] socket is closed!\n", client_id); g_client_id=client_id; close(g_client.sockid[client_id]); g_client.sockid[client_id]=-1; flag=0; break; } else{ printf("接收来自客户端[%d]数据:%s", client_id, buf); printf("长度:%d个字节\n\n",strlen(buf)-1); } } } } } /*发送线程*/ void *send_msg(void *arg) { struct msg server_send_msg; int client_id; char number[NUM_MAX_CLIENTS]; while(1){ memset(server_send_msg.data,'\0',sizeof(server_send_msg.data)); fgets(server_send_msg.data,sizeof(server_send_msg.data),stdin); client_id = *(int *)arg; printf("请输入要发送消息的客户端号:"); fgets(number,sizeof(number),stdin); client_id=atoi(number); if(send(g_client.sockid[client_id], (void *)&server_send_msg, sizeof(server_send_msg)-1, 0) == -1){ perror("send server message error!\n"); close(g_client.sockid[client_id]); pthread_exit(NULL); } else printf("长度是:%d\n\n",strlen(server_send_msg.data)-1); } }