0821作业+思维导图

一、作业

通过select完成一个服务器与两个客户端间的相互通信

1、服务器端

#include<myhead.h>
//增加数据到数组函数
void insert_client(int* cin_arr,int* len,int newfd)
{
	cin_arr[*len] = newfd;
	(*len)++;
}
//查找数组中数据函数
int find_client(int* cin_arr,int len, int newfd)
{
	for(int i=0;i<len;i++)
	{
		if(cin_arr[i] == newfd)
		{
			return i;
		}
	}
	return -1;
}
//删除数组中数据函数
void delete_client(int *cin_arr,int *len,int newfd)
{
	//查找数据位置
	int tar = find_client(cin_arr,*len,newfd);
	if(tar == -1)
	{
		return;
	}
	int i=-1;
	//删除数据
	for(int i=tar;i<(*len-1);i++)
	{
		cin_arr[i] = cin_arr[i+1];
	}
	cin_arr[i] = 0;
	(*len)--;
}

/***********主程序************/
int main(int argc, const char *argv[])
{
	//服务器端准备
	if(argc != 2)
	{
		printf("请输入正确的端口号\n");
		return -1;
	}
	int port = atoi(argv[1]);
	//创建服务器套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	//参数1：表示ipv4的网络通信
	//参数2：表示使用的是TCP通信方式
	//参数3：表示默认使用一个协议
	if(sfd == -1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success, sfd = %d\n",sfd);

	//端口号快速重用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt error");
		return -1;
	}
	//为套接字绑定ip地址和端口号
	//填充地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;        //通信域
	sin.sin_port = htons(port);  //端口号
	sin.sin_addr.s_addr = inet_addr("192.168.0.105");  //ip地址
	//绑定工作
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

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

	//selsct监听模型准备
	fd_set readfds;//准备一个描述符集合
	FD_ZERO(&readfds);//初始化描述符集合
	FD_SET(sfd,&readfds);//把服务器套接字文件描述符添加进描述符集合
	FD_SET(0,&readfds);//把标准输入流添加进描述符集合
	int cin_arr[128] = {0};//数组用来存储多个客户端
	int client_count = 0;//表示数组内有几个数据

	//开始监听
	while(1)
	{
		//创建一个描述符集合每次循环都刷新其中的文件描述符内容
		fd_set temp = readfds;
		//阻塞件套
		select(FD_SETSIZE,&temp,0,0,0);
		//判断服务器是否被激活
		if(FD_ISSET(sfd,&temp))
		{
			//链接客户端
			struct sockaddr_in cin;//用于接收地址
			socklen_t addrlen = sizeof(cin);//接收长度
		    int newfd = -1;
			if((newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen))==-1)
			{
			perror("accept error");
			return -1;
			}
			printf("连接到一个[%s:%d]的客户端\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
			FD_SET(newfd,&readfds);
			//将新连接的客户端加入到数组
			insert_client(cin_arr,&client_count,newfd);
			printf("%d\n",client_count);
		}
		//遍历数组判断客户端是否被激活
		for(int i=0;i<client_count;i++)
		{
			int client = cin_arr[i];
			if(FD_ISSET(client,&temp))
			{
				//激活可能是发来消息也可能是断开连接
				char buf[128] = "";
				int res = read(client,buf,sizeof(buf));
				if(res == 0)
				{
					//说明客户端断开链接
					printf("客户端断开链接\n");
					//从监视列表删除
					FD_CLR(client,&readfds);
					//从数组中删除
					delete_client(cin_arr,&client_count,client);
					//关闭套接字文件
					close(client);
				}else{
					printf("客户端发来消息：%s\n",buf);
				}
			}
		}
		if(FD_ISSET(0,&temp))
		{
			//标准输入流激活,把消息发送给所有客户端
			char buf[128] = "";
			read(0,buf,sizeof(buf));
			buf[strlen(buf)-1] = 0;
			for(int i=0;i<client_count;i++)
			{
				int client = cin_arr[i];
				send(client,buf,strlen(buf),0);
				printf("发送成功\n");
			}
		}
	}
	close(sfd);
	return 0;
}

2、客户端

#include<myhead.h>
int main(int argc, const char *argv[])
{
	if(argc != 2)
	{
		printf("请输入服务器端口号\n");
		return -1;
	}
	int port = atoi(argv[1]);
	//创建用于通信的套接字文件描述符
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd == -1)
	{
		perror("socket error");
		return -1;
	}
	printf("cfd = %d\n",cfd);   //3
	
	//端口号快速重用
	int reuse = 1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt error");
		return -1;
	}


	//连接到服务器
	//填充服务器地址结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(port);
	sin.sin_addr.s_addr = inet_addr("192.168.0.105");
	//链接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("connect error");
		return -1;
	}
	printf("链接服务器成功\n");

	//select监听模型准备
	fd_set readfds;
	FD_ZERO(&readfds);
	FD_SET(cfd,&readfds);
	FD_SET(0,&readfds);

	//数据收发
	char buf[128] = "";
	char bbuf[128] = "";
	while(1)
	{
		//阻塞监听
		fd_set temp = readfds;
		select(FD_SETSIZE,&temp,0,0,0);
		if(FD_ISSET(cfd,&readfds))
		{
			//接收服务器消息
			int res = read(cfd,bbuf,sizeof(bbuf));
			if(res !=0)
			{
				printf("服务器发来消息:%s\n",bbuf);
			}

		}
		if(FD_ISSET(0,&temp))
		{
			//把消息发送给服务端
			read(0,buf,sizeof(buf));
			buf[strlen(buf)-1] = 0;
			//将数据发送给服务器
			printf("111\n");
			send(cfd,buf,strlen(buf),0);
			printf("发送成功\n");
		}

	}
	//关闭套接字
	close(cfd);
	return 0;
}

3、实现截图

二、思维导图（IO多路复用）