10.1select并发服务器以及客户端

服务器：

#include

//do-while只是为了不让花括号单独存在，并不循环
#define ERR_MSG(msg) do{\
	fprintf(stderr,"%d:",__LINE__);\
	perror(msg);\
}while(0);

#define PORT 8888//端口号1024-49151
#define IP "192.168.2.54"//本机ＩＰ，终端输入ifconfig可得

int main(int argc, const char *argv[])
{
	//创建流式套接字 socket
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0){
		//printf("%d:",__LINE__);
		//perror("socket error");
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",sfd);

	//允许端口快速的被复用
	int reuse = 1;                                                            
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速的被复用成功\n");

	//填充地址信息结构体给bind函数绑定
	//真是的地址信息结构体根据地址族制定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;//必须填AF_INET
	sin.sin_port   		= htons(PORT);//端口号的网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);//本机IP
	//绑定服务器的地址信息  -->必须绑定    bind
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("bind");
		return -1;
	}//指针类型的强转不会有数据丢失
	printf("bind success\n");

	//将套接字设置为被动监听状态 listen
	if(listen(sfd,128)<0){
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	//定义文件描述符集合
	fd_set readfds,tmpfds;

	//将文件描述符集合清空
	FD_ZERO(&readfds);
	//将文件描述符放入集合之中
	FD_SET(sfd,&readfds);
	FD_SET(0, &readfds);

	int maxfd = sfd;
	int res;
	char buf[128] = "";
	struct sockaddr_in cin;  //存储客户端的地址信息
	socklen_t addrlen = sizeof(cin);  //真实的地址信息结构体的大小
	//获取一个已经完成的客户端信息，生成一个新的文件描述符 accept
	int newfd;
	struct sockaddr_in saveCin[1024];  //备份连接成功的客户端的地址信息，用下标来对应地址信息 
	while(1){
		tmpfds = readfds;

		//调用select函数
		res = select(maxfd+1,&tmpfds, NULL, NULL, NULL);
		if(res < 0){
			ERR_MSG("select error");
			break;
		}else if(0 == res){
			printf("time out\n");
			break;
		}else{
			for(int i=0; i<=maxfd; i++){
				if(0==FD_ISSET(i, &tmpfds)){
					continue;
				}else if(-1==FD_ISSET(i, &tmpfds)){
					ERR_MSG("FD_ISSET error");
					break;
				}

				if(0==i){
					//终端输入
					printf("从终端读取成功::");
					fgets(buf,sizeof(buf),stdin);

					buf[strlen(buf)-1] = 0;

					printf("%s\n",buf);
				}
				else if(sfd == i){
					if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){
						ERR_MSG("accept");
						return -1;


					}
					printf("[%s:%d]客户端链接成功 newfd = %d\n",\
							inet_ntoa(cin.sin_addr),\
							ntohs(cin.sin_port),newfd);

					saveCin[newfd] = cin;  //将cin另存到下标为文件描述符的位置

					//将newfd添加到集合中
					FD_SET(newfd, &readfds);

					//更新maxfd
					maxfd = maxfd>newfd?maxfd:newfd;


				}else{
					if(FD_ISSET(i, &readfds)){
						//清空字符串
						bzero(buf,sizeof(buf)); //memset

						//接收
						res = recv(i, buf, sizeof(buf) ,0);
						if(res<0){
							ERR_MSG("recv");
							return -1;
						}else if(0==res){
							printf("[%s:%d]客户端下线　newfd=%d\n",\
									inet_ntoa(saveCin[i].sin_addr),\
									ntohs(saveCin[i].sin_port),i);
							//将文件描述符从集合中踢出
							FD_CLR(i, &readfds);
							//由于踢出的文件描述符可能是最大文件描述符,更新maxfd
							/*		for(; maxfd>=0; maxfd--){
									if(FD_ISSET(maxfd, &readfds)){
									break;
									}
									}	*/

							while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);

							//关闭文件描述符
							close(i);
							continue;
						}
						printf("[%s:%d]客户端消息　newfd=%d:%s\n",\
								inet_ntoa(saveCin[i].sin_addr),\
								ntohs(saveCin[i].sin_port),i,buf);	


						//发送
						strcat(buf,">_<");
						if(send(i,buf,strlen(buf),0)<0){
							ERR_MSG("send");
							return -1;
						}
						printf("send success\n");			
					}				


				}
			}
/*
			//如果是终端输入就绪
			if(FD_ISSET(0,&tmpfds)){
				printf("从终端读取成功::");
				fgets(buf,sizeof(buf),stdin);

				buf[strlen(buf)-1] = 0;

				printf("%s\n",buf);
			}

			//如果是客户端连接就绪
			if(FD_ISSET(sfd,&tmpfds)){
				if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){
					ERR_MSG("accept");
					return -1;


				}
				printf("[%s:%d]客户端链接成功 newfd = %d\n",\
						inet_ntoa(cin.sin_addr),\
						ntohs(cin.sin_port),newfd);

				saveCin[newfd] = cin;  //将cin另存到下标为文件描述符的位置

				//将newfd添加到集合中
				FD_SET(newfd, &readfds);

				//更新maxfd
				maxfd = maxfd>newfd?maxfd:newfd;
			}
			for(int i=4; i<=maxfd; i++){
				if(FD_ISSET(i, &readfds)){
					//清空字符串
					bzero(buf,sizeof(buf)); //memset

					//接收
					res = recv(i, buf, sizeof(buf) ,0);
					if(res<0){
						ERR_MSG("recv");
						return -1;
					}else if(0==res){
						printf("[%s:%d]客户端下线　newfd=%d\n",\
								inet_ntoa(saveCin[i].sin_addr),\
								ntohs(saveCin[i].sin_port),i);
						//将文件描述符从集合中踢出
						FD_CLR(i, &readfds);
						//由于踢出的文件描述符可能是最大文件描述符,更新maxfd
						for(; maxfd>=0; maxfd--){
							if(FD_ISSET(maxfd, &readfds)){
								break;
							}
						}	

						while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);

						//关闭文件描述符
						close(i);
						continue;
					}
					printf("[%s:%d]客户端消息　newfd=%d:%s\n",\
							inet_ntoa(saveCin[i].sin_addr),\
							ntohs(saveCin[i].sin_port),i,buf);	


					//发送
					strcat(buf,">_<");
					if(send(i,buf,strlen(buf),0)<0){
						ERR_MSG("send");
						return -1;
					}
					printf("send success\n");			
				}
			}*/
		}
	}


	//关闭文件描述符
	close(newfd);
	if(close(sfd)<0){
		ERR_MSG("close");
		return -1;
	}


	return 0;
}

服务器：

#include

//do-while只是为了不让花括号单独存在，并不循环
#define ERR_MSG(msg) do{\
	fprintf(stderr,"%d:",__LINE__);\
	perror(msg);\
}while(0);

#define PORT 8888//端口号1024-49151
#define IP "192.168.2.54"//本机ＩＰ，终端输入ifconfig可得

int main(int argc, const char *argv[])
{
	//创建流式套接字 socket
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0){
		//printf("%d:",__LINE__);
		//perror("socket error");
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",cfd);

	//填充服务器的地址信息结构体给connect函数连接
	//相连接哪个服务器就输入哪个服务器的地址信息
	//真是的地址信息结构体根据地址族制定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;//必须填AF_INET
	sin.sin_port   		= htons(PORT);//服务器绑定的端口号的网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);//服务器绑定的IP
	socklen_t addrlen    = sizeof(sin);
	//连接指定服务器 connect
	if(connect(cfd,(struct sockaddr*)&sin,addrlen)<0){
		ERR_MSG("connect");
		return -1;
	}

	char buf[128] = "";
	ssize_t res = 0;

	//定义文件描述符集合
	fd_set readfds,tmpfds;
	//初始化集合
	FD_ZERO(&readfds);
	//把cfd加入集合
	FD_SET(cfd, &readfds);
	//把标准输入加入集合
	FD_SET(0, &readfds);
	int ret = -1;
	while(1){
		tmpfds = readfds;
		ret = select(cfd+1, &tmpfds, NULL, NULL, NULL);

		if(ret<0){
			ERR_MSG("select error");
			break;
		}else if(0 == ret){
			printf("time out\n");
			break;
		}else{
			if(FD_ISSET(0, &tmpfds)){
				//说明有来自终端的输入，需要发送
				//清空字符串
				bzero(buf,sizeof(buf)); //memset
				printf("请输入>> ");
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1] = 0;

				//发送
				strcat(buf,">_<");
				if(send(cfd,buf,strlen(buf),0)<0){
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}

			if(FD_ISSET(cfd, &tmpfds)){
				//套接字文件的接收缓冲区有数据，需要接收
				//说明有来自终端的输入，需要发送
				//清空字符串
				bzero(buf,sizeof(buf)); //memset

				//接收
				res = recv(cfd, buf, sizeof(buf) ,0);
				if(res<0){
					ERR_MSG("recv");
					return -1;
				}else if(0==res){
					printf("[%s:%d]服务器下线　cfd=%d\n",\
							inet_ntoa(sin.sin_addr),\
							ntohs(sin.sin_port),cfd);
					break;
				}
				printf("[%s:%d]服务器消息　cfd=%d:%s\n",\
						inet_ntoa(sin.sin_addr),\
						ntohs(sin.sin_port),cfd,buf);
			}

		}

	
	}


/*
	while(1){
		//清空字符串
		bzero(buf,sizeof(buf)); //memset
		printf("请输入>> ");
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1] = 0;

		//发送
		strcat(buf,">_<");
		if(send(cfd,buf,strlen(buf),0)<0){
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");

		//接收
		res = recv(cfd, buf, sizeof(buf) ,0);
		if(res<0){
			ERR_MSG("recv");
			return -1;
		}else if(0==res){
			printf("[%s:%d]服务器下线　cfd=%d\n",\
					inet_ntoa(sin.sin_addr),\
					ntohs(sin.sin_port),cfd);
			break;
		}
		printf("[%s:%d]服务器消息　cfd=%d:%s\n",\
					inet_ntoa(sin.sin_addr),\
					ntohs(sin.sin_port),cfd,buf);		
	}
*/
	//关闭文件描述符
	//close(newfd);
	if(close(cfd)<0){
		ERR_MSG("close");
		return -1;
	}


	return 0;
}

效果：