select的TCP服務(wù)器模型
服務(wù)器端
#include <head.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define PORT 6666 //1024~49151
#define IP "192.168.122.120" //ifconfig查看本機(jī)IP
int main(int argc, const char *argv[])
{
//創(chuàng)建流式套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd < 0)
{
ERRO_MES("socket");
return -1;
}
printf("sfd = %d\n",sfd);
//綁定服務(wù)器的IP和端口--->必須綁定
//填充地址信息結(jié)構(gòu)體，真實的地址信息結(jié)構(gòu)體根據(jù)地質(zhì)族制定
//AF_INET： man 7 ip;
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必須填A(yù)F_INET;
sin.sin_port = htons(PORT); //端口號：1024~49151;
sin.sin_addr.s_addr = inet_addr(IP); //本機(jī)IP ifconfig查看;
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
{
ERRO_MES("bind");
return -1;
}
printf("bind success\n");
//將套接字設(shè)置為被動監(jiān)聽狀態(tài)
if(listen(sfd,10) < 0)
{
ERRO_MES("listen");
return -1;
}
printf("listen success\n");
//設(shè)置允許端口號被快速復(fù)用
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
{
ERRO_MES("setsockopt");
return -1;
}
printf("允許端口快速重用成功\n");
//從已完成鏈接的隊列中獲取一個客戶端信息，生成一個新的文件
//該文件描述符才是與客戶端通信的文件描述符
struct sockaddr_in cin;
socklen_t addrlen = sizeof(cin);
struct sockaddr_in saveCin[1024-3];
//創(chuàng)建一個讀集合
fd_set readfds,tempfds;
//初始化文件描述符
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//將需要的文件描述符添加到讀集合中
FD_SET(sfd,&readfds);
FD_SET(0,&readfds);
//確定最大文件描述符
int maxfd = sfd;
int s_res = -1;
char buf[128]="";
ssize_t res = 0;
int newfd = -1;
while(1)
{
tempfds = readfds;
//調(diào)用IO多路復(fù)用函數(shù) ，select()
s_res = select(maxfd+1, &tempfds, NULL, NULL, NULL);
if(s_res < 0)
{
ERRO_MES("select");
return -1;
}
else if(0 == s_res)
{
printf("超時操作\n");
break;
}
//判斷是哪個文件描述符準(zhǔn)備就緒，走對應(yīng)處理函數(shù)即可
for(int i=0; i<=maxfd; i++)
{
if(FD_ISSET(i, &tempfds) == 0)
continue;
if(0 == i)
{
printf("觸發(fā)鍵盤輸入事件\n");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1] = 0;
printf(": %s\n",buf);
}
else if(sfd == i)
{
printf("觸發(fā)客戶端連接事件\n");
newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
if(newfd < 0)
{
ERRO_MES("accept");
return -1;
}
printf("[%s : %d] newfd = %d,客戶端鏈接成功\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
saveCin[newfd-3] = cin;
//將newfd添加到集合
FD_SET(newfd, &readfds);
maxfd = newfd>sfd ? newfd : sfd;
}
else
{
printf("觸發(fā)客戶端交互事件\n");
bzero(buf,sizeof(buf));
//接收數(shù)據(jù)
res = recv(i,buf,sizeof(buf),0);
if(res < 0)
{
ERRO_MES("recv");
return -1;
}
else if(0 == res)
{
printf("[%s : %d] newfd = %d,客戶下線\n",\
inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i-3].sin_port),i);
//關(guān)閉下線客戶端的文件描述符
close(i);
//將下線客戶端的文件描述符從集合中剔除
FD_CLR(i,&readfds);
//更新maxfd
while(FD_ISSET(maxfd,&readfds) == 0 && maxfd-- >=0);
continue;
}
printf("[%s : %d] newfd :%d\t buf = %s\n"\
,inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i-3].sin_port),i,buf);
//發(fā)送數(shù)據(jù)
strcat(buf,"----");
if(send(newfd,buf,sizeof(buf),0) < 0)
{
ERRO_MES("send");
return -1;
}
printf("send success\n");
}
}
}
close(sfd);
close(newfd);
return 0;
}
客戶端
#include <head.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define PORT 6666 //1024~49151
#define IP "192.168.122.120" //ifconfig查看本機(jī)IP
int main(int argc, const char *argv[])
{
//創(chuàng)建流式套接字
int cfd = socket(AF_INET,SOCK_STREAM,0);
if(cfd < 0)
{
ERRO_MES("socket");
return -1;
}
printf("cfd = %d\n",cfd);
//綁定服務(wù)器的IP和端口--->非必須綁定
//填充地址信息結(jié)構(gòu)體，真實的地址信息結(jié)構(gòu)體根據(jù)地質(zhì)族制定
//AF_INET： man 7 ip;
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必須填A(yù)F_INET;
sin.sin_port = htons(PORT); //端口號：填服務(wù)器綁定的端口號;
sin.sin_addr.s_addr = inet_addr(IP);//IP地址;服務(wù)器綁定的IP地址
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
{
ERRO_MES("connect");
return -1;
}
printf("連接成功\n");
//創(chuàng)建一個讀集合
fd_set readfds,tempfds;
//初始化集合
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//將需要的文件描述符添加到讀集合
FD_SET(0,&readfds);
FD_SET(cfd,&readfds);
char buf[128]="";
ssize_t res = 0;
int s_res = -1;
int maxfd = cfd;
while(1)
{
tempfds = readfds;
s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(s_res < 0)
{
ERRO_MES("select");
return -1;
}
else if(0 == s_res)
{
printf("time out...\n");
break;
}
if(FD_ISSET(0,&tempfds))
{
printf("觸發(fā)鍵盤輸入事件\n");
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1] = 0;
printf(": %s\n",buf);
if(send(cfd,buf,sizeof(buf),0) < 0)
{
ERRO_MES("send");
return -1;
}
printf("send success\n");
}
if(FD_ISSET(cfd,&tempfds))
{
printf("觸發(fā)服務(wù)器交互事件\n");
bzero(buf,sizeof(buf));
//接收數(shù)據(jù)
res = recv(cfd,buf,sizeof(buf),0);
if(res < 0)
{
ERRO_MES("recv");
return -1;
}
else if(0 == res)
{
printf("[%s : %d] cfd = %d,客戶下線\n",IP,PORT,cfd);
break;
}
printf("[%s : %d] cfd :%d\t buf = %s\n",IP,PORT,cfd,buf);
}
}
close(cfd);
return 0;
}
poll客戶端
#include <head.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <poll.h>
#include <unistd.h>
#define PORT 6666
#define IP "192.168.122.120"
int main(int argc, const char *argv[])
{
//創(chuàng)建流式套接字
int cfd = socket(AF_INET,SOCK_STREAM,0);
if(cfd < 0)
{
ERRO_MES("socket");
return -1;
}
printf("cfd = %d\n",cfd);
//綁定客戶端的IP和端口 ----->客戶端(非必須)
//若不綁定，則操作系統(tǒng)會給客戶端綁定上客戶端所在的主機(jī)IP，以及隨機(jī)端口(49152~65535)
//填充服務(wù)器的地址信息結(jié)構(gòu)體，真實的地址信息結(jié)構(gòu)體根據(jù)地址族指定
//AF_INET：man 7 ip
//要連接哪個服務(wù)器，就填對應(yīng)服務(wù)器的IP和端口
struct sockaddr_in sin;
sin.sin_family = AF_INET; //必須填A(yù)F_INET
sin.sin_port = htons(PORT); //端口號：填服務(wù)器綁定端口號
sin.sin_addr.s_addr = inet_addr(IP); //IP地址：服務(wù)器綁定的IP地址
//連接服務(wù)器
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
{
ERRO_MES("connect");
return -1;
}
printf("connect success\n");
//定義集合
struct pollfd fds[2];
//將需要監(jiān)測的文件描述符添加到集合中
fds[0].fd = 0; //指定監(jiān)測0號文件描述符
fds[0].events = POLLIN; //監(jiān)測讀事件
fds[1].fd = cfd; //指定監(jiān)測cfd號文件描述符
fds[1].events = POLLIN; //監(jiān)測讀事件
char buf[128] = "";
ssize_t res = 0;
int p_res = -1;
while(1)
{
p_res = poll(fds,2,-1);
if(p_res < 0)
{
ERRO_MES("poll");
return -1;
}
else if(0 == p_res)
{
printf("time out...\n");
break;
}
//能運(yùn)行到當(dāng)前位置，代表集合中有文件描述符準(zhǔn)備就緒
//判斷是哪個文件描述符準(zhǔn)備就緒，走對應(yīng)處理函數(shù)
//判斷集合中的每個文件描述符的revents中是否有POLLIN
//從revents中將代表POLLIN的那一位單獨(dú)提取出來，判斷是1還是0
if((fds[0].revents & POLLIN) != 0)
{
printf("觸發(fā)鍵盤輸入事件\n");
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1] = 0;
//發(fā)送數(shù)據(jù)
if(send(cfd, buf,sizeof(buf),0) < 0)
{
ERRO_MES("send");
return -1;
}
printf("send success\n");
}
if(fds[1].revents & POLLIN)
{
printf("觸發(fā)服務(wù)器交互事件\n");
bzero(buf,sizeof(buf));
//接收數(shù)據(jù)
res = recv(cfd,buf,sizeof(buf),0);
if(res < 0)
{
ERRO_MES("recv");
return -1;
}
else if(0 == res)
{
printf("[%s : %d] cfd= %d 服務(wù)器掉線\n",IP,PORT,cfd);
break;
}
printf("[%s : %d] cfd= %d : %s\n",IP,PORT,cfd,buf);
}
}
close(cfd);
return 0;
}
文章來源地址http://www.zghlxwxcb.cn/news/detail-650162.html
文章來源：http://www.zghlxwxcb.cn/news/detail-650162.html
到了這里，關(guān)于IO多路復(fù)用中select的TCP服務(wù)器模型和poll服務(wù)模型的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
![網(wǎng)絡(luò)編程 IO多路復(fù)用 [select版] (TCP網(wǎng)絡(luò)聊天室)](https://imgs.yssmx.com/Uploads/2024/02/620278-1.png)
