网络编程 day6

最新推荐文章于 2024-09-21 22:16:03 发布

wangzonghao-vm

最新推荐文章于 2024-09-21 22:16:03 发布

阅读量747

点赞数 14

文章标签：网络

本文链接：https://blog.csdn.net/2401_83238851/article/details/137902600

版权

用select实现tcp并发多路复用服务器
#define SER_IP "192.168.71.128" //服务器地址
#define SER_PORT 8888 //服务器端口

int main(int argc, const char *argv[])
{
   //创建套接字用来接收客户端消息
   int sfd=socket(AF_INET,SOCK_STREAM,0);
   if(sfd==-1)
   {
       perror("socket");
       return -1;
   }
   printf("sfd=%d\n",sfd);//3 0 1 2 STD

   //设置端口号快速重启
   int reuse = 1;
   if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1)
   {
       perror("setsockopt error");
       return -1;
   }
   printf("端口号快速重用成功\n");

   //用bind函数给sfd创建IP和端口号
   //填充结构体消息
   struct sockaddr_in sin;
   sin.sin_family = AF_INET;
   sin.sin_port = htons(SER_PORT);
   sin.sin_addr.s_addr=inet_addr(SER_IP);
   //绑定工作
   if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
   {
       perror("bind");
       return -1;
   }
   printf("bind success\n");

   //设置被动监听状态
   if(listen(sfd,128)==-1)
   {
       perror("listen");
       return -1;
   }
   printf("listen success\n");

   //阻塞等待客户；连接请求
   struct sockaddr_in cin;
   socklen_t socklen = sizeof(cin);
   int newfd=-1;

//创建一个文件描述符检测的集合
fd_set readfds,tempfds;

//清空集合
FD_ZERO(&readfds);

   //将sfd和0放进集合里面
   FD_SET(0,&readfds);
   FD_SET(sfd,&readfds);

int maxfd=sfd; //记录检测容器中最大文件描述符的值

struct sockaddr_in cin_arr[1024]; //定义一个地址信息结构体容器，主要是为了显示不同服务器连接时的端口号

   while(1)
   {
       tempfds=readfds;

       //阻塞检测文件是否有事件产生
       int res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
       if(res==0)
       {
           printf("time out\n");
           return -1;
       }else if(res==-1)
       {
           perror("celect");
           return -1;
       }

       //程序执行到此，就要判断文件描述符是否还在集合中
       for(int i=0;i<=maxfd;i++)
       {
           //判断事件是否产生
           if(!FD_ISSET(i,&tempfds))
           {
               continue;
           }

//程序执行到此，说明触发了事件，则需要判断i是那个文件描述符

           if(i==sfd)
           {
               newfd = accept(sfd,(struct sockaddr*)&cin,&socklen);
               if(newfd==-1)
               {
                   perror("accept");
                   return -1;
               }

printf("[%s:%d]:已连接,newfd=%d\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

               //将newfd放入检测集合readfds中
               FD_SET(newfd,&readfds);
               cin_arr[newfd]=cin;

               //更新maxfd
               if(maxfd<newfd)
               {
                   maxfd=newfd;
               }

}

           //判断文件描述符是否在集合中
           else if(i==0)
           {
               //另一个阻塞事件
               char buf[128]="";
               fgets(buf,sizeof(buf),stdin);
               buf[strlen(buf)-1]=0;
               printf("触发了键盘输入事件：%s\n",buf);

               //将buf中的所有消息发送给所以客户端
               for(int i=4;i<=maxfd;i++)
               {
                   send(i,buf,strlen(buf),0);

}

           }
           //判断客户端是否发来消息
           else
           {
               //数据通信
               char rbuf[128]="";

bzero(rbuf,sizeof(rbuf));

               int res=read(i,rbuf,sizeof(rbuf));
               if(res==0)
               {
                   perror("客户端已下线");
                   close(i); //关闭下线的客户端

//移除文件描述符
FD_CLR(i,&readfds);

                   //更新maxfd
                   for(int j=maxfd;j>sfd;j--)
                   {
                       if(FD_ISSET(j,&readfds))
                       {
                           maxfd=j;
                           break;
                       }
                   }

continue ;
}

printf("[%s:%d]:%s\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),rbuf);

               //加个笑脸会回去
               strcat(rbuf,"*_*");
               write(newfd,rbuf,strlen(rbuf));
               printf("发送成功");

               //将收到的消息广播给所有客户端
               for(int i=4;i<=maxfd;i++)
               {
                   send(i,rbuf,strlen(rbuf),0);

}

}
}

close(sfd);
return 0;
}

用poll实现客户端

#define SER_IP "192.168.71.128" //服务器地址
#define SER_PORT 8888 //服务器端口
#define CLI_IP "192.168.71.128"
#define CLI_PORT 9999

int main(int argc, const char *argv[])
{
   int cfd=socket(AF_INET,SOCK_STREAM,0);
   if(cfd==-1)
   {
       perror("socket");
       return -1;
   }
   printf("socket success cfd=%d\n",cfd);

//设置端口号快速重用
int reuse = 1;
if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1)
{
perror("setsockopt error");
return -1;
}
printf("端口号快速重用成功\n");

   //绑定ip和客户端
   struct sockaddr_in cin ;
   cin.sin_family=AF_INET;
   cin.sin_port=htons(CLI_PORT);
   cin.sin_addr.s_addr=inet_addr(CLI_IP);

   if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
   {
       perror("bind");
       return -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地址

   if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
   {
       perror("connect");
       return -1;
   }
   printf("connect success\n");

   //创建一个用于检测文件描述符的容器
   struct pollfd pfds[2];

pfds[0].fd=0; //表示第一个元素检测0号文件描述符
pfds[0].events = POLLIN; // 要检测0号文件描述符的读事件

pfds[1].fd=cfd; //表示第二个元素检测1号文件描述符
pfds[1].events = POLLIN; //要检测1号文件描述符的读事件

   //数据收发
   char wbuf[128]="";
   while(1)
   {
       int res = poll(pfds,2,-1); //阻塞文件描述符集合是否有事件产生
       if(res==0)
       {
           printf("time out\n");
           return -1 ;
       }else if(res==-1)
       {
           perror("poll");
           return -1;
       }

//程序执行至此，说明集合中有事件生产，根据不同的事件执行不同的函数

       if(pfds[0].events == POLLIN) //判断是否为终端输入
       {
           //从终端获取数据
           fgets(wbuf,strlen(wbuf),stdin);
           wbuf[strlen(wbuf)-1]='\0';
           if(strcat(wbuf,"quit")==0)
           {
               break;
           }
           //将数据发送给服务器
           send(cfd,wbuf,strlen(wbuf),0);
       }

       //判断是否为服务器发来的消息
       if(pfds[1].events == POLLIN)
       {
           bzero(wbuf,sizeof(wbuf));
           recv(cfd,wbuf,sizeof(wbuf),0);
           printf("收到的服务器消息为：%s\n",wbuf);
       }