select实现客户端代码
#include<myhead.h>
#define SERIP "192.168.114.125"
#define SERPORT 8888
#define CLIIP "192.168.114,125"
#define CLIPORT 6666
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; //使用的是IPV4通信
cin.sin_port = htons(CLIPORT); //服务器端口号
cin.sin_addr.s_addr = inet_addr(CLIIP); //服务器IP地址
//2.2绑定工作
if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
{
perror("bind error");
return -1;
}
//3.连接服务器
//3.1、填充服务器地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET; //使用的是IPV4通信
sin.sin_port = htons(SERPORT); //服务器端口号
sin.sin_addr.s_addr = inet_addr(SERIP); //服务器IP地址
//3.2、连接服务器
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
{
perror("connect error");
return -1;
}
printf("connect success\n");
定义一个用来检测文件描述符的集合
fd_set readfds;
清空容器中的内容
FD_ZERO(&readfds);
将要检测的文件描述符放入集合中
FD_SET(cfd,&readfds);
FD_SET(0,&readfds);
int res=0;//接收select的返回值
//4.收发数据
char buf[128]="";
char rbuf[128]="";
while(1)
{
res=select(cfd+1,&readfds,NULL,NULL,NULL);
if(res==-1)
{
perror("select error");
return -1;
}else if(res==0)
{
printf("time out\n");
return -1;
}else
{
bzero(buf,sizeof(buf));
bzero(rbuf,sizeof(rbuf));
fgets(buf,sizeof(buf),stdin); //从标准输入中读取数据
buf[strlen(buf)-1]='\0';
//将数据发送给服务器
send(cfd,buf,sizeof(buf),0);
if(strcmp(buf,"quit")==0)
{
break;
}
//接收服务器发送来的消息
int res=recv(cfd,rbuf,sizeof(rbuf),0);
if(res==0)
{
printf("服务器已经关闭\n");
break;
}
printf("rbuf=%s\n",rbuf);
}
}
//5、关闭客户端
close(cfd);
return 0;
poll实现服务器端代码
#include<myhead.h>
#define PORT 8888
#define IP "192.168.114.125"
int main(int argc, const char *argv[])
{
//1.创建用于接收连接的套接字
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;
} //2.绑定端口和IP
//2.1地址结构体信息
struct sockaddr_in sin;
sin.sin_family=AF_INET;
sin.sin_port=htons(PORT);
sin.sin_addr.s_addr=inet_addr(IP);
//2.2
if( bind(sfd,(struct sockaddr *)&sin,sizeof(sin))==-1)
{
perror("bind error");
return -1;
}
//3.将套接字设置为被动监听
if( listen(sfd,128)==-1)
{
perror("listen error");
return -1;
}
//4.阻塞等待接收连接
struct sockaddr_in cin;
cin.sin_family=AF_INET;
socklen_t socklen=sizeof(cin);
int newfd;
//文件描述符集合数组
struct pollfd fds[1024];
fds[0].fd=0;
fds[0].events=POLLIN;
fds[1].fd=sfd;
fds[1].events=POLLIN;
struct sockaddr_in save[1024];
int res=0;
char buf[128]="";
int maxfd=sfd;
while(1)
{
res=poll(fds,maxfd+1,-1);
if(res == -1)
{
perror("poll error");
return -1;
}
else if(res == 0)
{
printf("timeout\n");
return -1;
}
if(fds[1].revents==POLLIN)
{ if((newfd=accept(sfd,(struct sockaddr *)&cin,&socklen))==-1)
{
perror("accept error");
return -1;
}
printf("accept success newfd=%d\n",newfd);
fds[newfd].fd=newfd;
fds[newfd].events=POLLIN;
save[newfd]=cin;
if(newfd>maxfd)
{
maxfd=newfd;
}
}
if(fds[0].revents==POLLIN)
{
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]='\0';
printf("触发终端输入:%s\n",buf);
for(int i=4;i<=maxfd;i++)
{send(fds[i].fd,buf,sizeof(buf),0);}
}
for(int i=4;i<maxfd+1;i++)
{ if(fds[i].revents==POLLIN)
{
char buf[128]="";
bzero(buf,sizeof(buf));
int res= recv(i,buf,sizeof(buf),0);
if(res == 0)
{
printf("客户端下线\n");
close(fds[i].fd);
fds[i].fd=-1;
for(int j=maxfd;j>3;j--)
{
if(fds[j].fd!=-1)
{
maxfd=fds[j].fd;
break;
}
}
continue;
}
else if(res<0)
{
perror("recv error");
return -1;
}
printf("%s:%d]:%s\n",inet_ntoa(save[i].sin_addr),ntohs(save[i].sin_port),buf);
}
}
}
//6.关闭套接字
close(sfd);
return 0;