#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <sys/time.h>
#include <sys/select.h>
int main(void)
{
//1.创建套接字
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if( sockfd < 0)
{
perror("socket");
return -1;
}
//2.绑定端口和地址
struct sockaddr_in addr;
addr.sin_family=AF_INET;
addr.sin_port = htons(9527);
//转网络字节序
addr.sin_addr.s_addr = htonl(INADDR_ANY);
//绑定地址为INADDR_ANY,这个个服务器程序在那个ip地址上运行就绑定那个ip地址
int ret = bind(sockfd, (struct sockaddr*)&addr,sizeof(addr));
if(ret < 0)
{
perror("bind");
return -1;
}
//3.监听
ret = listen(sockfd, 4);
if(ret < 0)
{
perror("listen");
return -1;
}
//接受连接--accept阻塞
//接收客户端数据--recv阻塞
//select多路复用
fd_set readfds;
//定义读文件描述符集合
int cfds[50]={0};
//保存所有客户端的套接字
while(1)
{
int maxfd = sockfd;
//清空集合
FD_ZERO(&readfds);
FD_SET(sockfd,&readfds);
//把sockfd添加到readfds集合中
//把客户端的套接字也添加到集合中
for(int i=0; i<50; i++)
{
if(cfds[i] != 0)//如果不等于0就说明这个空间存储了套接字
{
FD_SET(cfds[i],&readfds);
maxfd = maxfd>cfds[i]?maxfd:cfds[i];
}
} //select用监听readfds里面的所有文件描述符,
//如果某一个或多个描述符有响应就跳出select函数
ret = select(maxfd+1,&readfds,NULL, NULL, NULL);
if(ret < 0)
{
perror("select");
}
//判断是否是sockfd有数据---说明有客户端连接
if(FD_ISSET(sockfd, &readfds))
{
//接受连接
int clientfd = accept(sockfd, NULL,NULL);
if(clientfd < 0)
{
perror("accept");
} //把客户端套接字保存到cfds数组中
for(int i=0; i<50; i++)
{
if(cfds[i] == 0)//判断放在没有使用的空间中
{
cfds[i] = clientfd;
break;
}
}
} //判断是否是客户端有数据
for(int i=0; i<50; i++)
{
//判断套接字是否存在,并且是否有数据到达
if(cfds[i] != 0 && FD_ISSET(cfds[i], &readfds))
{
//接收客户端数据
char buffer[128]={0};
int size = recv(cfds[i], buffer, 128, 0);
if(size<=0)
{
printf("有客户端掉线\n");
close(cfds[i]);
cfds[i] = 0;
}
//把接收到的信息转发给客户端自己
size = send(cfds[i],buffer,size,0);
}
}
}
printf("有客户端连接\n");
close(sockfd);
}
标签:addr,多路复用,int,TCP,readfds,sockfd,include,服务端,客户端
From: https://www.cnblogs.com/lwj294/p/18246842