课后作业:
1.将select的TCP服务器重新搭建
2.搭建select的TCP客户端
3.搭建poll客户端
4.上传的代码完成
4没能做出来
服务器端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <IOhead.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#define PROT 1225
#define IP "192.168.122.130"
int main(int argc, const char *argv[])
{
//创建流式套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("sfd = %d\n",sfd);
int reuse = 1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_MSG("reuse");
return -1;
}
printf("允许端口快速重用成功\n");
//填充地址信息结构体,真实地址根据地质族制定
//AF_INET: man 7 IP
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必填AF_INET
sin.sin_port = htons(PROT);//端口号:1024~49151
sin.sin_addr.s_addr = inet_addr(IP);//本机IP
//绑定服务器IP和端口
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success \n");
//将套接字设置为被动监听状态
if(listen(sfd,128)<0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success \n");
//创建读集合
fd_set readfds,tempfds;
//fd_set本质上为结构体 是long类型数组
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//将文件描述符添加入集合
FD_SET(0,&readfds);
FD_SET(sfd,&readfds);
int maxfd=sfd;
int s_res = -1;
int newfd = -1;
char buf[128]="";
ssize_t res = 0;
struct sockaddr_in cin;
socklen_t addrlen = sizeof(cin);
while(1)
{
tempfds = readfds;
//调用IO多路复用函数select()
s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(s_res<0)
{
ERR_MSG("select");
return -1;
}
else if(0 == s_res)
{
printf("time out\n");
break;
}
printf("__%d__\n",__LINE__);
if(FD_ISSET(0,&tempfds))
{
char buf[128] = "";
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf) - 1] = 0;
for (int i=4;i<=maxfd;i++)
{
if(FD_ISSET(i,&readfds) && i!=sfd)
{
send(i,buf,strlen(buf),0);
}
}
}
if(FD_ISSET(sfd,&tempfds))
{
printf("触发连接\n");
newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
if(newfd<0)
{
ERR_MSG("accept");
return -1;
}
printf("[%s:%d] newfd = %d 客户端连接成功\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
//将FD添加到集合
FD_SET(newfd,&readfds);
//更新maxfd
maxfd = maxfd>newfd ? maxfd:newfd;
}
for(int i=4;i<=maxfd;i++)
{
if(FD_ISSET(i,&tempfds))
{
printf("触发客户端交互\n");
bzero(buf,sizeof(buf));
//接收数据
res = recv(i,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("res");
return -1;
}
else if(0==res)
{
printf("newfd = %d 客户端下线\n",i);
close(i);
FD_CLR(i, &readfds);
while(FD_ISSET(maxfd,&readfds)==0 &&maxfd -->=0);
continue;
}
printf("[%s:%d] newfd = %d :%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf);
/* //发送数据
strcat (buf,"swtsb\n");
if(send(newfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return -1;
}
printf("[%s:%d] newfd = %d 客户端连接成功\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
printf("send success \n");*/
}
}
}
close(newfd);
close(sfd);
return 0;
}
客户端
root@linux:~/TCP# cat sel1.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <IOhead.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#define PROT 1225
#define IP "192.168.122.130"
int main(int argc, const char *argv[])
{
//创建流式套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("sfd = %d\n",sfd);
int reuse = 1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_MSG("reuse");
return -1;
}
printf("允许端口快速重用成功\n");
//填充地址信息结构体,真实地址根据地质族制定
//AF_INET: man 7 IP
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必填AF_INET
sin.sin_port = htons(PROT);//端口号:1024~49151
sin.sin_addr.s_addr = inet_addr(IP);//本机IP
if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("connect");
return -1;
}
printf("connect success \n");
//创建读集合
fd_set readfds,tempfds;
//fd_set本质上为结构体 是long类型数组
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//将文件描述符添加入集合
FD_SET(0,&readfds);
FD_SET(sfd,&readfds);
int maxfd=sfd;
int s_res = -1;
int newfd = -1;
char buf[128]="";
ssize_t res = 0;
struct sockaddr_in cin;
socklen_t addrlen = sizeof(cin);
while(1)
{
tempfds = readfds;
//调用IO多路复用函数select()
s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(s_res<0)
{
ERR_MSG("select");
return -1;
}
else if(0 == s_res)
{
printf("time out\n");
break;
}
printf("__%d__\n",__LINE__);
if(FD_ISSET(0,&tempfds))
{
char buf[128] = "";
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf) - 1] = 0;
send(sfd, buf, strlen(buf), 0);
}
if(FD_ISSET(sfd,&tempfds))
{
char buf[128] = "";
ssize_t res = recv(sfd, buf, sizeof(buf), 0);
if (res < 0)
{
perror("recv");
break;
}
else if (res == 0)
{
printf("服务器关闭\n");
break;
}
else
{
buf[res] = '\0';
printf("%s\n", buf);
}
}
}
close(newfd);
close(sfd);
return 0;
}
poll客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <IOhead.h>
#include <poll.h>
#define PORT 1225
#define IP "192.168.122.130"
int main(int argc, const char *argv[])
{
int cfd = socket(AF_INET,SOCK_STREAM,0);
if(cfd<0)
{
ERR_MSG("SOCKET");
return -1;
}
printf("cfd = %d\n",cfd);
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("CONNECT");
return -1;
}
printf("连接成功");
struct pollfd fds[2];
fds[0].fd=0;
fds[0].events = POLLIN;
fds[1].fd=cfd;
fds[1].events = POLLIN;
char buf[128]="";
int p_res = -1;
ssize_t res = 0;
while(1)
{
p_res = poll(fds,2,-1);
if(p_res<0)
{
ERR_MSG("POLL");
return -1;
}
else if(0==p_res)
{
printf("time out");
break;
}
if((fds[0].revents & POLLIN) !=0)
{
printf("触发输入");
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
if(send(cfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("SEND");
return -1;
}
printf("发送成功");
}
if((fds[1].revents & POLLIN)!=0)
{
printf("触发交互");
bzero(buf,sizeof(buf));
res = recv(cfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("RECV");
return -1;
}
else if(0==res)
{
printf("掉线");
break;
}
printf("[%s:%d] cfd=%d : %s \n",IP,PORT,cfd,buf);
}
}
close(cfd);
return 0;
}
思维导图: