#include <stdlib.h>
#include <stdio.h>/*定義輸入/輸出函數(shù)*/
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include  <string.h>/*字符串處理*/
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>

#define MAXDATASIZE 256

#define SERVPORT 4444	/*服務(wù)器監(jiān)聽(tīng)端口號(hào)*/
#define BACKLOG 1     /*最大同時(shí)連接請(qǐng)求數(shù)*/
#define STDIN 0        /*標(biāo)準(zhǔn)輸入文件描述符*/
	
int main(void)
{
	FILE *fp;		/*定義文件類型指針fp*/
	int sockfd,client_fd;	/*監(jiān)聽(tīng)socket.sock_fd,數(shù)據(jù)傳輸socket.new_fd*/
	int sin_size;
	struct sockaddr_in my_addr, remote_addr;	/*本機(jī)地址信息,客戶地址信息*/	
		char buf[256];		/*用于聊天的緩沖區(qū)*/
       
	 char buff[256];	/*用于輸入用戶名的緩沖區(qū)*/
	char send_str[256];		/*最多發(fā)出的字符不能超過(guò)256*/
	int recvbytes;			
	fd_set rfd_set, wfd_set, efd_set;/*被select()監(jiān)視的讀,寫(xiě),異常處理的文件描述符集合*/
	struct timeval timeout;		/*本次select的超時(shí)結(jié)束時(shí)間*/
	int ret;			/*與client連接的結(jié)果*/

	if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { 	/*錯(cuò)誤檢測(cè)*/
		perror("socket");
		exit(1);
	}
	/* 端填充 sockaddr結(jié)構(gòu)  */
	bzero(&my_addr, sizeof(struct sockaddr_in));
	my_addr.sin_family=AF_INET;	/*地址族*/
	my_addr.sin_port=htons(SERVPORT);	/*端口號(hào)為4444*/
	inet_aton("127.0.0.1", &my_addr.sin_addr);

	if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr)) == -1) { 			/*錯(cuò)誤檢測(cè)*/
		perror("bind");
		exit(1);
	}
	if (listen(sockfd, BACKLOG) == -1) {	 /*錯(cuò)誤檢測(cè)*/
		perror("listen");
		exit(1);
	}

	sin_size = sizeof(struct sockaddr_in);
	if ((client_fd = accept(sockfd, (struct sockaddr *)&remote_addr, &sin_size)) == -1) {
 /*錯(cuò)誤檢測(cè)*/
		perror("accept");
		exit(1);
	}

	fcntl(client_fd, F_SETFD, O_NONBLOCK);/* 服務(wù)器設(shè)為非阻塞*/
	recvbytes=recv(client_fd, buff, MAXDATASIZE, 0);
/*接收從客戶端傳來(lái)的用戶名*/	
	buff[recvbytes] = '\0';
	fflush(stdout);
	/*強(qiáng)制立即內(nèi)容*/
	if((fp=fopen("name.txt","a+"))==NULL)
			{
					printf("can not open file,exit...\n");
					return -1;
			}
			fprintf(fp,"%s\n",buff);
/*將用戶名寫(xiě)入name.txt中*/	
	while (1) {
		FD_ZERO(&rfd_set);/*將select()監(jiān)視的讀的文件描述符集合清除*/
		FD_ZERO(&wfd_set);/*將select()監(jiān)視的寫(xiě)的文件描述符集合清除*/
		FD_ZERO(&efd_set);/*將select()監(jiān)視的異常的文件描述符集合清除*/
		FD_SET(STDIN, &rfd_set);
/*將標(biāo)準(zhǔn)輸入文件描述符加到seletct()監(jiān)視的讀的文件描述符集合中*/
		FD_SET(client_fd, &rfd_set);
/*將新建的描述符加到seletct()監(jiān)視的讀的文件描述符集合中*/
		FD_SET(client_fd, &wfd_set);
/*將新建的描述符加到seletct()監(jiān)視的寫(xiě)的文件描述符集合中*/
		FD_SET(client_fd, &efd_set);
/*將新建的描述符加到seletct()監(jiān)視的異常的文件描述符集合中*/
		timeout.tv_sec = 10;/*select在被監(jiān)視窗口等待的秒數(shù)*/
		timeout.tv_usec = 0;/*select在被監(jiān)視窗口等待的微秒數(shù)*/
		ret = select(client_fd + 1, &rfd_set, &wfd_set, &efd_set, &timeout);
		if (ret == 0) {
			continue;
		}
			if (ret < 0) {
			perror("select error: ");
			exit(-1);
		}

		if(FD_ISSET(STDIN, &rfd_set))
		 {
			fgets(send_str, 256, stdin);
			send_str[strlen(send_str)-1] = '\0';
			if (strncmp("quit", send_str, 4) == 0) {	
				close(client_fd);
				close(sockfd);	
				exit(0);
			}
			send(client_fd, send_str, strlen(send_str), 0);
		}
/*判斷是否已將新建的描述符加到seletct()監(jiān)視的讀的文件描述符集合中*/
		if (FD_ISSET(client_fd, &rfd_set))
	 {
			recvbytes=recv(client_fd, buf, MAXDATASIZE, 0);/*接收從客戶端傳來(lái)的聊天內(nèi)容*/
			if (recvbytes == 0) {
				close(client_fd);
				close(sockfd);	/*關(guān)閉套接字*/
				exit(0);
			}
			buf[recvbytes] = '\0';
			printf("%s:%s\n",buff,buf);
			printf("Server: ");
			fflush(stdout);
		}
/*判斷是否已將新建的描述符加到seletct()監(jiān)視的異常的文件描述符集合中*/
		if (FD_ISSET(client_fd, &efd_set)) {
			close(client_fd);	/*關(guān)閉套接字*/
			exit(0);
		}
	}
}
15031426256