作业要求：①使用IO多路复用中的select函数实现TCP并发服务器客户端
②使用IO多路复用中的poll函数实现TCP并发服务器的服务器端
一、
代码
#include <myhead.h>
#define SERPORT 8888 //服务器端口号
#define SERIP "192.168.114.113" //服务器IP地址
int main(int argc, const char *argv[])
{
//创建用于通信的套接字
int cfd = socket(AF_INET,SOCK_STREAM,0);
if(cfd == -1)
{
perror("socket error");
return -1;
}
//连接服务器
///填充服务器地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(SERPORT);
sin.sin_addr.s_addr = inet_addr(SERIP);
///连接服务器
if(connect(cfd,(struct sockaddr *)&sin,sizeof(sin)) == -1)
{
perror("connect error");
return -1;
}
//创建用于检测文件描述符的集合
fd_set readfds,tempfds;
//清空集合
FD_ZERO(&readfds);
//将要检测的文件描述符放入集合中
FD_SET(cfd,&readfds);
FD_SET(0,&readfds);
int res = 0; //接收select的返回值
int maxfd = cfd; //集合中值最大的文件描述符
//向服务器进行数据的收发
char buf[128] = "";
int ret = 0; //接收recv的返回值
while(1)
{
tempfds = readfds;
res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(res == -1)
{
perror("select error");
return -1;
}else if(res == 0)
{
printf("time out\n");
return -1;
}
//遍历集合中所有的文件描述符
for(int i = 0;i <= maxfd;i++)
{
//判断当前文件描述符是否在集合中
if(!FD_ISSET(i,&readfds))
{
continue;
}
//判断0号文件描述符是否还在集合中
if(0 == i)
{
//从标准输入中读取数据
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1] == 0;
//将数据发送到服务器
if(send(cfd,buf,sizeof(buf),0) == -1)
{
perror("send error");
return -1;
}
}else if(cfd == i) //判断cfd是否还在集合中
{
//接收来自服务器的消息
ret = recv(cfd,buf,sizeof(buf),0);
if(ret == -1)
{
perror("recv error");
return -1;
}else if(ret == 0)
{
printf("服务器已关闭\n");
return -1;
}
printf("服务器消息：%s\n",buf);
}
}
}
//关闭文件描述符
close(cfd);
return 0;
}
效果图
二、
代码
#include <myhead.h>
#define IP "192.168.114.118"
#define PORT 8888
int main(int argc, const char *argv[])
{
//创建用于连接的套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd == -1)
{
perror("socket error");
return -1;
}
//设置端口号快速重用
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(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");
//定义一个集合管理sfd和打开的通信用文件描述符
struct pollfd fds[1024];
int maxfd = 0;
//手动放入sfd
fds[0].fd = sfd;
fds[0].events = POLLIN; //表明为读事件
//将fds中其余元素初始化为-1
for(int i = 4;i <= 1024;i++)
{
fds[i].fd = -1;
}
//填充客户端地址信息结构体
struct sockaddr_in cin;
cin.sin_family = AF_INET;
socklen_t socklen = sizeof(cin);
char cbuf[128] = ""; //给客户端用的容器
int nfd;
int res = 0; //接收poll返回的结果
while(1)
{
res = poll(fds,maxfd+1,-1);
if(res == -1)
{
perror("select");
return -1;
}
else if(res == 0)
{
continue;;
}
else if(res > 0) //说明检测到了有文件描述符对应的缓冲区的数据发生了改变
{
if(fds[0].revents == POLLIN) //表明有新的客户连接进来了
{
int nfd = accept(sfd,(struct sockaddr*)&cin,&socklen); //阻塞在此处，直到有客户端连接上来
if(nfd == -1) //增加这些错误的判断非常重要，可以帮助找到出现问题的地方
{
perror("accept");
return -1;
}
//将新的文件描述符加入到集合中
for(int i = 1;i < 1024;i++)
{
if( fds[i].fd == -1)
{
fds[i].fd = nfd;
fds[i].events = POLLIN;
break;
}
}
//更新最大的文件描述符
if(nfd > maxfd)
{
maxfd = nfd;
}
}
for(int i = 1;i <= maxfd;i++) //轮询客户端对应的文件描述符
{
if(fds[i].revents == POLLIN) //说明此文件描述符对应的客户端发送来了数据
{
int ret = read(fds[i].fd,cbuf,sizeof(cbuf));
if(ret == -1)
{
perror("read");
exit(-1);
}
else if(ret == 0)
{
printf("client closed\n");
close(fds[i].fd); //关闭对应的文件描述符
fds[i].fd = -1; //在fds中清空对应的文件描述符
}
else if(ret > 0)
{
printf("read buf = %s\n",cbuf);
write(fds[i].fd,cbuf,strlen(cbuf)+1);
}
}
}
}
}
//关闭所有套接字
close(sfd);
return 0;
}
效果图
