思维导图
多路复用
selsect
ser
#include<myhead.h>
#define SER_PORT 8888 //服务器端口号
#define SER_IP "192.168.125.50" //服务器IP地址
int main(int argc, const char *argv[])
{
//1、创建用于连接的套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if(sfd == -1)
{
perror("socket error");
return -1;
}
printf("sfd = %d\n", sfd);
//将端口号快速重用函数
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
{
perror("setsockopt error");
return -1;
}
printf("端口号快速重用成功\n");
//2、给当前套接字绑定IP地址和端口号
//2.1 填充要绑定的地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET; //通信域
sin.sin_port = htons(SER_PORT); //端口号
sin.sin_addr.s_addr = inet_addr(SER_IP); //IP地址
//2.2 绑定
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
{
perror("bind error");
return -1;
}
printf("bind success %s %s %d\n", __FILE__, __func__, __LINE__);
//3、将套接字设置成监听状态
if(listen(sfd, 128) == -1)
{
perror("listen error");
return -1;
}
printf("listen success %s %s %d\n", __FILE__, __func__, __LINE__);
//4、阻塞等待客户端的连接请求
//4.1 定义容器接收客户端的地址信息
struct sockaddr_in cin; //用于接收地址信息
socklen_t socklen = sizeof(cin); //用于接收地址信息的大小
//4.2接收客户端的连接
int newfd = -1;
//11、准备一个文件描述符容器
fd_set readfds, tempfds;
//22、清空容器
FD_ZERO(&readfds);
//33、将要检测的文件描述符放入集合
FD_SET(0 ,&readfds);
FD_SET(sfd ,&readfds);
//定义一个变量存储容器中最大的文件描述符
int maxfd = sfd;
//定义一个地址信息结构体,每一个元素对应一个客户端文件描述符
struct sockaddr_in cin_arr[1024];
while(1)
{
tempfds = readfds; //存储一份数据来使用
//使用select函数对容器中的文件描述符进行检测
int res = select(maxfd+1, &tempfds, NULL, NULL, NULL);
if(res == -1)
{
perror("select error");
return -1;
}else if(res == 0)
{
printf("timeout\n");
return -1;
}
//判断是否是文件描述符触发事件
for(int cli=0; cli<=maxfd; cli++)
{
//去掉不存在于集合中的文件描述符
if(!FD_ISSET(cli, &tempfds))
{
continue;
}
if(cli == sfd)
{
newfd = accept(sfd, (struct sockaddr*)&cin, &socklen);
if(newfd == -1)
{
perror("accept error");
return -1;
}
printf("[%s:%d]发来连接请求 %s %s %d\n",\
inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), __FILE__, __func__, __LINE__);
//更新地址信息结构体数组
cin_arr[newfd] = cin;
//将newfd放入readfds中
FD_SET(newfd, &readfds);
//更新maxfd
if(newfd > maxfd)
{
maxfd = newfd;
}
}else if(cli == 0) //判断0号文件描述符是否在集合中
{
//终端输入
char wbuf[128] = "";
scanf("%s", wbuf);
printf("触发了终端输入事件...\n");
if(strcmp(wbuf, "quit") == 0)
{
goto END;
}
for(int i=4; i<=maxfd; i++)
{
sendto(i, wbuf, sizeof(wbuf), 0, (struct sockaddr*)&cin_arr[i], sizeof(cin_arr[i]));
}
}else
{
//5、跟客户端进行消息通信
char buf[128] = "";
//清空数组
bzero(buf, sizeof(buf));
//读取客户端发来的消息
int res = recv(cli, buf, sizeof(buf), 0);
if(res == 0)
{
printf("客户端已经下线\n");
//关闭当前通信套接字
close(cli);
//将当前文件描述符从文件描述符集合中移除
FD_CLR(cli, &readfds);
//更新maxfd
for(int i= maxfd; i>=sfd; i--)
{
if(FD_ISSET(i, &readfds))
{
maxfd = i;
break;
}
}
continue;
}
printf("[%s:%d] %s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), buf);
//给客户端发送消息
strcat(buf, "123");
send(cli, buf, sizeof(buf), 0);
printf("发送成功\n");
}
}
}
END:
//6、关闭套接字
close(sfd);
return 0;
}
cli
#include<myhead.h>
#define SER_PORT 8888 //服务器端口号
#define SER_IP "192.168.125.50" //服务器IP地址
#define CLI_PORT 6666 //客户端的端口号
#define CLI_IP "192.168.125.50" //客户端IP地址
int main(int argc, const char *argv[])
{
//1、创建用于通信的套接字文件描述符
int cfd = socket(AF_INET, SOCK_STREAM, 0);
if(cfd == -1)
{
perror("socket error");
return -1;
}
printf("cfd = %d\n", cfd);
//2、绑定(可选)
//2.1 填充地址信息结构体
struct sockaddr_in cin;
cin.sin_family = AF_INET;
cin.sin_port = htons(CLI_PORT);
cin.sin_addr.s_addr = inet_addr(CLI_IP);
//2.2 绑定
if(bind(cfd, (struct sockaddr*)&cin, sizeof(cin)) == -1)
{
perror("bind error");
return -1;
}
printf("bind success\n");
//3、连接服务器
//3.1 填充服务器地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(SER_PORT);
sin.sin_addr.s_addr = inet_addr(SER_IP);
//3.2 连接
if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
{
perror("connect error");
return -1;
}
printf("connect success\n");
//11、准备一个文件描述符容器
fd_set readfds, tempfds;
//22、清空容器
FD_ZERO(&readfds);
//33、将要检测的文件描述符放入集合
FD_SET(0, &readfds);
FD_SET(cfd, &readfds);
//4、收发数据
char buf[128]="";
while(1)
{
//存储一份数据
tempfds = readfds;
//清空数组
bzero(buf, sizeof(buf));
//使用select函数对容器中的文件描述符进行检测
int res = select(cfd+1, &tempfds, NULL, NULL, NULL);
if(res == -1)
{
perror("select error");
return -1;
}else if(res == 0)
{
printf("timeout\n");
return -1;
}
if(FD_ISSET(0, &tempfds))
{
printf("请输入>>>");
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf)-1] = '\0';
//发送给服务器
send(cfd, buf, sizeof(buf), 0);
printf("发送成功\n");
if(strcmp(buf, "quit") == 0)
{
break;
}
}
if(FD_ISSET(cfd, &tempfds))
{
//接收服务器发来的消息
recv(cfd, buf, sizeof(buf), 0);
printf("[%s:%d]:%s\n", SER_IP, SER_PORT, buf);
}
}
//5、关闭套接字
close(cfd);
return 0;
}
poll
ser
#include<myhead.h>
#define SER_PORT 8888 //服务器端口号
#define SER_IP "192.168.125.50" //服务器IP地址
int main(int argc, const char *argv[])
{
//1、创建用于连接的套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if(sfd == -1)
{
perror("socket error");
return -1;
}
printf("sfd = %d\n", sfd);
//将端口号快速重用函数
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
{
perror("setsockopt error");
return -1;
}
printf("端口号快速重用成功\n");
//2、给当前套接字绑定IP地址和端口号
//2.1 填充要绑定的地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET; //通信域
sin.sin_port = htons(SER_PORT); //端口号
sin.sin_addr.s_addr = inet_addr(SER_IP); //IP地址
//2.2 绑定
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
{
perror("bind error");
return -1;
}
printf("bind success %s %s %d\n", __FILE__, __func__, __LINE__);
//3、将套接字设置成监听状态
if(listen(sfd, 128) == -1)
{
perror("listen error");
return -1;
}
printf("listen success %s %s %d\n", __FILE__, __func__, __LINE__);
//4、阻塞等待客户端的连接请求
//4.1 定义容器接收客户端的地址信息
struct sockaddr_in cin; //用于接收地址信息
socklen_t socklen = sizeof(cin); //用于接收地址信息的大小
//4.2接收客户端的连接
int newfd = -1;
//11、定义一个等待文件描述符结构体数组
struct pollfd pfd[1024];
//22、填充要等待的文件描述符及事件
pfd[0].fd = 0;
pfd[0].events = POLLIN;
pfd[sfd].fd = sfd;
pfd[sfd].events = POLLIN;
//定义一个地址信息结构体,每一个元素对应一个客户端文件描述符
struct sockaddr_in sin_arr[1024];
//定义一个变量存储需要检测的最大个数
int size = 4;
//5、跟客户端进行消息通信
char buf[128] = "";
while(1)
{
//阻塞检测集合中是否有事件产生
int res = poll(pfd, size, -1);
if(res == -1)
{
perror("poll error");
return -1;
}else if(res == 0)
{
printf("timeout\n");
return -1;
}
//到此有事件发生
for(int index=0; index<1024; index++)
{
//判断连接事件
if(pfd[index].revents != POLLIN)
{
continue;
}
if(pfd[sfd].revents == POLLIN)
{
newfd = accept(sfd, (struct sockaddr*)&cin, &socklen);
if(newfd == -1)
{
perror("accept error");
return -1;
}
printf("[%s:%d]发来连接请求 %s %s %d\n",\
inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), __FILE__, __func__, __LINE__);
sin_arr[newfd] = sin;
pfd[newfd].fd = newfd;
pfd[newfd].events = POLLIN;
size++;
}else if((pfd[index].revents == POLLIN) && pfd[index].fd == 0)
{
//清空数组
bzero(buf, sizeof(buf));
//终端输入
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf)-1] = 0;
printf("服务器输入为:%s", buf);
}else
{
//清空数组
bzero(buf, sizeof(buf));
int res = recv(pfd[index].fd, buf, sizeof(buf), 0);
if(res == 0)
{
printf("客户端下线\n");
close(pfd[index].fd);
continue;
}
printf("[%s:%d]发送:%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), buf);
//给客户端发送消息
strcat(buf, "123");
send(newfd, buf, sizeof(buf), 0);
printf("发送成功\n");
}
}
}
END:
//6、关闭套接字
close(sfd);
return 0;
}
cli
#include<myhead.h>
#define SER_PORT 8888 //服务器端口号
#define SER_IP "192.168.125.50" //服务器IP地址
#define CLI_PORT 6666 //客户端的端口号
#define CLI_IP "192.168.125.50" //客户端IP地址
int main(int argc, const char *argv[])
{
//1、创建用于通信的套接字文件描述符
int cfd = socket(AF_INET, SOCK_STREAM, 0);
if(cfd == -1)
{
perror("socket error");
return -1;
}
printf("cfd = %d\n", cfd);
//2、绑定(可选)
//2.1 填充地址信息结构体
struct sockaddr_in cin;
cin.sin_family = AF_INET;
cin.sin_port = htons(CLI_PORT);
cin.sin_addr.s_addr = inet_addr(CLI_IP);
//2.2 绑定
if(bind(cfd, (struct sockaddr*)&cin, sizeof(cin)) == -1)
{
perror("bind error");
return -1;
}
printf("bind success\n");
//3、连接服务器
//3.1 填充服务器地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(SER_PORT);
sin.sin_addr.s_addr = inet_addr(SER_IP);
//3.2 连接
if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) == -1)
{
perror("connect error");
return -1;
}
printf("connect success\n");
//11、定义一个等待文件描述符结构体数组
struct pollfd pfd[2];
//2、填充要等待的文件描述符及事件
pfd[0].fd = 0; //将0号文件描述符放入检测集合中
pfd[0].events = POLLIN; //表示检测该文件描述符的读事件
pfd[1].fd = cfd;
pfd[1].events = POLLIN;
//4、收发数据
char buf[128]="";
while(1)
{
//阻塞检测集合中是否有事件产生
int res = poll(pfd, 2, -1);
if(res == -1)
{
perror("poll error");
return -1;
}else if(res == 0)
{
printf("timeout\n");
return -1;
}
//程序执行至此,表示已经有事件产生
//判断是0号文件描述符产生的事件
if(pfd[0].revents == POLLIN)
{
//清空数组
bzero(buf, sizeof(buf));
printf("请输入>>>");
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf)-1] = '\0';
//发送给服务器
send(cfd, buf, sizeof(buf), 0);
printf("发送成功\n");
if(strcmp(buf, "quit") == 0)
{
break;
}
}
//判断是cfd产生的事件
if(pfd[1].revents == POLLIN)
{
//接收服务器发来的消息
recv(cfd, buf, sizeof(buf), 0);
printf("[%s:%d]:%s\n", SER_IP, SER_PORT, buf);
}
}
//5、关闭套接字
close(cfd);
return 0;
}