linux网络编程：服务器与客户机模拟聊天（多线程）

灰太狼 2022-05-31 00:48 191阅读 0赞

服务器端代码

#include <stdio.h>
    #include <stdlib.h>
    #include <stdbool.h>
    #include <errno.h>
    #include <string.h>
    #include <sys/types.h>
    #include <netinet/in.h>
    #include <sys/socket.h>
    #include <sys/wait.h>
    #include <sys/types.h>
    #include <sys/stat.h>
    #include <sys/fcntl.h>
    #include <pthread.h>
    #include <unistd.h>
    
    #define SERVER_PORT 4321
    #define BUF_SIZE 1024
    #define LISNUM 5
    int recv_flag = 1;
    int main(){
    	int sock_fd, new_fd;
    	struct sockaddr_in server_addr;
    	struct sockaddr_in client_addr;
    
    	int sin_size,myport,lisnum;
    	int n;
    	char buff[BUF_SIZE+1];
    	//char temp[BUF_SIZE+1];
    	char temp[BUF_SIZE+1];
    	bool usb_connect,net_connect ;
    	int on = 1;
    	pthread_t s_usb2net_id,s_net2usb_id;
    
    
    	if ((sock_fd = socket(AF_INET,SOCK_STREAM,0)) == -1){
    		perror("socket");
    		exit(1);
    	}
    
    	/* Enable address reuse */
    	setsockopt(sock_fd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on));
    
    	bzero(&server_addr,sizeof(struct sockaddr_in));
    	server_addr.sin_family = AF_INET;
    	server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    	server_addr.sin_port = htons(SERVER_PORT);
    
    	if(bind(sock_fd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr)) == -1){
    		perror("bind");
    		exit(1);
    	}
    	
    	printf("Bind success, welcome to server!\n");
    	
    	if(listen(sock_fd,LISNUM) == -1){
    		perror("listen");
    		exit(1);
    	}
    	printf("Listening....\n");
    	while(1){
    		//recv_flag = 0;
    		sin_size = sizeof(struct sockaddr_in);
    
    		if ((new_fd = accept(sock_fd,(struct sockaddr *)&(client_addr),&sin_size))==-1) {
    			perror("accept");
    //			continue;
    			exit(1);
    		}else{
    				if(send(new_fd,"connect to server success!",30,0) == -1){
    					perror("send");
    					close(new_fd);
    				    exit(0);
    				}
    		}
    
    		
    		
    
    		/*the read and send thread */
    		void *send_thread(void *arg){
    			
    		
    			while(1){
    				scanf("%s", temp);
    				if(send(new_fd,temp,strlen(temp),0) == -1){
    					perror("send");
    					
    				}else{
    					printf("server send success!\n");
    				}
    			}
    		}
    
    		/*the receive and write thread */
       		void *recv_thread(void *arg){
    			while(1){
    				bzero(buff,BUF_SIZE+1);
    				if ( (n = recv(new_fd,buff,BUF_SIZE,0 ))== -1) {
    					perror("server recv ERROR");
    				}
    				buff[n] = 0;
    				if(n > 0){
    					if(recv_flag%2 == 0)
    					printf("Server receive a massage from client:%s\n",buff);
    					recv_flag++;
    					//printf("------%d\n", recv_flag);
    					//break;
    					continue;
    				}
    			}
    		}
    
    
    		/* create the receive and write thread */
    		if(pthread_create(&s_net2usb_id,NULL,recv_thread,NULL)){
    			printf("recv_thread is created failed!\n");
    		}
    
    
    		/* create the read  and send thread */
    		if(pthread_create(&s_usb2net_id,NULL,send_thread,NULL)){
    			printf("send_thread is created failed!\n");
    		}
    		
    		if((pthread_join(s_net2usb_id,NULL)) != 0){
    			perror("pthread_join ERROR");
    		}
    		if((pthread_join(s_usb2net_id,NULL)) != 0){
    			perror("pthread_join ERROR");
    		}
    		
    
    
    	}
    	close(sock_fd);
    	exit(0);
    }

客户端代码

#include <stdio.h>
    #include <stdlib.h>
    #include <stdbool.h>
    #include <errno.h>
    #include <string.h>
    #include <sys/types.h>
    #include <netinet/in.h>
    #include <sys/socket.h>
    #include <netdb.h>
    #include <pthread.h>
    
    #define MAXDATASIZE 1024
    #define PORT	4321
    int main(int argc, char *argv[]){
    	int sockfd, portnumber,nbytes;
    	char buff[MAXDATASIZE];
    	char temp[MAXDATASIZE];
    	bool conn = true,sendthread = true;
    	struct hostent *he;
    	struct sockaddr_in server_addr;
    	int i;
    	pthread_t c_usb2net_id,c_net2usb_id;
    	
    	if(argc != 3){
    		fprintf(stderr,"USAGE: ./client Hostname(or ip address) Text\n");
    		exit(1);
    	}
    
    	/*if ( (portnumber = atoi(argv[2])) < 0 ) {
    		fprintf(stderr,"Usage:%s  hostname portnumber\a\n",argv[0]);
    		exit(1);
    
    	}*/
    
    	if ( (he = gethostbyname(argv[1])) == NULL ) {
    		perror("gethostbyname");
    		exit(1);
    	}
    
    	if ( (sockfd = socket(AF_INET,SOCK_STREAM,0))== -1) {
    		perror("socket");
    		exit(1);
    	}
    
    	bzero(&server_addr,sizeof(server_addr));
    	server_addr.sin_family = AF_INET;
    	server_addr.sin_port = htons(PORT);
    	server_addr.sin_addr = *((struct in_addr *)he->h_addr);
    	
    	while(1){
    
    		while(conn){
    			if(connect(sockfd, (struct sockaddr *)&server_addr,sizeof(struct sockaddr)) == -1){
    				perror("connect");
    				exit(1);
    			}
    			conn = false;
    		}
    
    		/*****************   the send thread function ********************/
    		void *send_thread(void *arg){
    			i=0;i++;
    			printf("send thread ------>%d\n",i);
    			while(1){
    				scanf("%s", buff);
    				if(send(sockfd,buff,strlen(buff),0) == -1){
    					perror("send");
    					//				  close(sockfd);
    				}else if(send(sockfd,buff,strlen(buff),0) == 0){
    					printf("----> clinet is ranking and waiting!\n");
    				}else{
    					printf("----> clinet send success!\n");
    				}
    			}
    		}
    		
    		/* create the thread */
    		while(sendthread){
    			if(pthread_create(&c_usb2net_id,NULL,send_thread,NULL)){
    				printf("C_UsbtoNet_thread is created failed!\n");
    				sendthread = true;
    			}
    			sendthread = false;
    		}
    		
    
    		/**************** the receive action in the main thread ***********/
    		bzero(temp,MAXDATASIZE+1);
    		if ( (nbytes = recv(sockfd,temp,MAXDATASIZE,0 ))== -1) {
    			perror("server recv ERROR");
    			//exit(1);
    		}
    		temp[nbytes] = 0;
    		if(nbytes > 0){
    			printf("clinet receive a massage from server:%s\n",temp);
    		}
    
    	}
    	close(sockfd);
    	exit(0);
    }