IO多路复用-----select

服务器

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include <sys/types.h>          
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <sys/select.h>
#include <sys/time.h>
#include <unistd.h>
#include <arpa/inet.h>

//打印错误信息
#define ERR_MSG(msg) do{\
	perror(msg);\
	printf("line : __%d__\n",__LINE__);\
}while(0)

#define PORT 8888		//本机端口号
#define IP "192.168.0.66"	//本机IP
#define BACKLOG 128			//未完成连接队伍容量

int main(int argc,const char *argv[])
{
	//创建套接字
	int sfd;
	sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0){
		ERR_MSG("socket");
		return -1;
	}

	//定义本机地址信息结构体，填充本机IP和端口号
	struct sockaddr_in sin;
	sin.sin_family			= AF_INET;
	sin.sin_port			= htons(PORT);
	sin.sin_addr.s_addr		= inet_addr(IP);
	//绑定主机信息
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){
		ERR_MSG("bind");
		return -1;
	}

	//将套接字设置为监听状态
	if(listen(sfd,BACKLOG) < 0){
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	//存储链接成功的客户端地址
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	//存储所有连接成功的客户端地址信息
	struct sockaddr_in savecin[1020];

	//创建读集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	
	//定义一个最大的文件描述符
	int maxfd = sfd;
	int newsfd;
	int res;
	char buf[128] = "";
	int sndfd;
	while(1){
		tempfds = readfds;
		//监测就绪的文件描述符
		res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(res < 0){
			ERR_MSG("select");
			return -1;
		}else if(res == 0){
			printf("time out......\n");
			break;
		}
		
		for(int i=0;i<=maxfd;i++){
			if(!FD_ISSET(i,&tempfds)){
				continue;
			}
			//运行到此说明文件描述符 i 存在于读集合
			if(0 == i)
			{
				printf("触发键盘输入事件>>> ");
				//从终端获取数据然后发送给指定客户端
				res = scanf("%d %s",&sndfd,buf);
				while(getchar() != 10);
				if(res != 2){
					fprintf(stderr,"输入格式错误，eg：文件描述符 内容\n");
					continue;
				}
				if(sndfd<=sfd || sndfd>=1024 || !FD_ISSET(sndfd,&readfds)){
					fprintf(stderr,"文件描述符非法\n");
					continue;
				}
				if(send(sndfd,buf,sizeof(buf),0) < 0){
					ERR_MSG("send");
					return -1;
				}

			}
			else if(sfd == i)
			{
				printf("触发客户端连接事件>>> ");
				//获取新的套接字
				newsfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(newsfd < 0){
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s : %d] newsfd = %d 客户端连接成功\n"\
						,inet_ntoa(cin.sin_addr),htons(cin.sin_port),newsfd);
				//將新的文件描述符添加到读集合中
				FD_SET(newsfd,&readfds);
				//更新最大文件描述符
				maxfd = maxfd>newsfd?maxfd:newsfd;
				//存储地址信息
				savecin[newsfd-4] = cin;
			}
			else
			{
				bzero(buf,sizeof(buf));
				res = recv(i,buf,sizeof(buf),0);
				if(res < 0){
					ERR_MSG("recv");
					return -1;
				}else if(res == 0){
					printf("[%s : %d] newsfd = %d 客户端已下线\n"\
							,inet_ntoa(savecin[i-4].sin_addr),htons(savecin[i-4].sin_port),i);
					//关闭文件描述符
					close(i);
					//将文件描述符剔除
					FD_CLR(i,&readfds);
					//更新最大文件描述符
					int j = -1;
					for(j=maxfd;j>=0;j--){
						if(FD_ISSET(j,&readfds)){
							maxfd = j;
							break;
						}
					}
					if(-1 == j)
						maxfd = -1;
					continue;
				}
				printf("[%s : %d] newsfd = %d : %s\n"\
						,inet_ntoa(savecin[i-4].sin_addr),htons(savecin[i-4].sin_port),i,buf);
				//将接收到的数据拼接好再发送回去
				strcat(buf,"*_*");
				if(send(i,buf,sizeof(buf),0) < 0){
					ERR_MSG("send");
					return -1;
				}
			}
		}
	}
	//关闭套接字
	close(sfd);
	return 0;
}

客户端

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

//打印错误信息及行号
#define MSG_ERR(msg) do{\
	fprintf(stderr,"line : __%d__\n",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888       //服务器端口号
#define IP "192.168.0.66"  //服务器IP   

int main(int argc,const char *argv[])
{
	//创建一个流式套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);   //网络层协议：IPv4,传输层协议：TCP
	if(cfd < 0){
		MSG_ERR("socket");
		return -1;
	}
	printf("cfd = %d\n",cfd);

	//填充服务器真实地址信息
	struct sockaddr_in sin;
	sin.sin_family      =  AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){
		MSG_ERR("connect");
		return -1;
	}
	printf("connext success   __%d__\n",__LINE__);

	//创建集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//将文件描述符放进读集合中
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);
	
	int res;
	char buf[128] = "";
	ssize_t ret;
	while(1){
		tempfds = readfds;
		res = select(cfd+1,&tempfds,NULL,NULL,NULL);
		if(res < 0){
			MSG_ERR("select");
			return -1;
		}else if(0 == res){
			printf("time out....\n");
			break;
		}
		if(FD_ISSET(0,&tempfds)){
			//发送
			printf("触发键盘输入事件>>> ");
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1] = 0;
			res = send(cfd,buf,sizeof(buf),0);
			if(res < 0){
				MSG_ERR("send");
				return -1;
			}
		}
		if(FD_ISSET(cfd,&tempfds)){
			//接收
			bzero(buf,sizeof(buf));
			res = recv(cfd,buf,sizeof(buf),0);
			if(res < 0){
				MSG_ERR("recv");
				return -1;
			}else if(0 == res){
				printf("cfd = %d 服务器已下线   __%d__\n",cfd,__LINE__);
				break;
			}
			printf("[%s ; %d] %s\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);

		}
	}
	/*
	   while(1){
	}
	*/

	close(cfd);
	return 0;
}