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

服务器端代码

#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);
}