一、作业
通过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、实现截图
![](https://i-blog.csdnimg.cn/direct/8edfb6d238014549a01290da1856ffd8.png)
二、思维导图(IO多路复用)
![](https://i-blog.csdnimg.cn/direct/4ff256a0b91548bea92c54f70db32384.png)