一、作业
通过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多路复用)