day4 IO模型

    1.将select的TCP服务器重新搭建

#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <head.h>

#define PORT 10000
#define IP "192.168.122.74"

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("setsockopt");
        return -1;
    }
    printf("允许端口快速复用\n");
    //填充地址信息结构体,真实的地址信息结构体根据地址族制定
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;   //必须填AF_INET
    sin.sin_port = htons(PORT);//端口号范围为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");
    //从已完成连接的队列中获取一个客户端信息,生成一个新的文件描述符
    struct sockaddr_in cin;                                                      
    socklen_t addrlen = sizeof(cin);
    struct sockaddr_in savecin[1024];

    //创建一个读集合
    fd_set readfds,tempfds; //fd_set本质上是一个结构体,成员是一个long类型的数组
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);
    //将需要的文件描述符添加到读集合中
    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);
    int maxfd=sfd;
    char buf[128]="";
    ssize_t res=0;
    int newfd=-1;
    int s_res=-1;
    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;
        }
        //运行到当前位置,则代表集合中有文件描述符准备就绪
        //此时集合中会只剩下触发事件的文件描述符
        //例如:
        //当0号文件描述符触发事件,则集合中只剩下0
        //当sfd文件描述符触发事件,则集合中只剩下sfd
        //若同时触发事件,则集合中剩下0和sfd
        //判断是哪个文件描述符准备就绪,走对应处理函数即可
        for(int i=0;i<=maxfd;i++)
        {
            if(FD_ISSET(i,&tempfds)==0)
                continue;
            if(0==i)                                                           
            {
                printf("触发键盘输入事件\n");
                int sndfd=-1;
                int a=scanf("%d %s",&sndfd,buf);
                while(getchar()!=10);
                if(a!=2)
                {
                    printf("请输入正确的格式:fd string\n");
                    continue;
                }
                if(sndfd<3 || !FD_ISSET(sndfd,&readfds))
                {
                    printf("该客户端不存在\n");
                    continue;
                }
                if(send(sndfd,buf,sizeof(buf),0)<0)
                {
                    ERR_MSG("send");
                    return -1;
                }
            }
            else if(i==sfd)
            {
                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);

                savecin[newfd]=cin;
                FD_SET(newfd,&readfds);
                maxfd=maxfd>newfd ? maxfd:newfd;
            }
            else
            {
                printf("触发客户端输入事件\n");
                bzero(buf,sizeof(buf));
                //接收数据                                                                             
                res=recv(i,buf,sizeof(buf),0);
                if(res<0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if(0==res)
                {
                    printf("[%s:%d] newfd=%d 客户端下线\n",\
                            inet_ntoa(savecin[i].sin_addr),ntohs(savecin[i].sin_port),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(savecin[i].sin_addr),ntohs(savecin[i].sin_port),i,buf);
            }
        }
    }
    //关闭文件描述符
    close(sfd);
    return 0;
}


    2.搭建select的TCP客户端

#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <head.h>

#define PORT 10000
#define IP "192.168.122.74"

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;   //必须填AF_INET
    sin.sin_port = htons(PORT);//端口号范围为1024到49151
    sin.sin_addr.s_addr = inet_addr(IP);    //本机IP

    //连接服务器
    if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("服务器连接成功\n");

                                                                                    
    //创建一个读集合
    fd_set readfds,tempfds; //fd_set本质上是一个结构体,成员是一个long类型的数组
    FD_ZERO(&readfds);
    //将需要的文件描述符添加到读集合中
    FD_SET(0,&readfds);
    FD_SET(cfd,&readfds);
    char buf[128]="";
    ssize_t res=0;
    int newfd=-1;
     int s_res=-1;
     while(1)
     {
         tempfds=readfds;                                                        
         //调用IO多路复用函数 select()
         s_res=select(cfd+1,&tempfds,NULL,NULL,NULL);
         if(s_res<0)
         {
             ERR_MSG("select");
             return -1;
         }
         else if(0==s_res)
         {
             printf("time out..\n");
             break;
         }
         //运行到当前位置,则代表集合中有文件描述符准备就绪
         //此时集合中会只剩下触发事件的文件描述符
         //例如:
         //当0号文件描述符触发事件,则集合中只剩下0
         //当sfd文件描述符触发事件,则集合中只剩下sfd
         //若同时触发事件,则集合中剩下0和sfd
         //判断是哪个文件描述符准备就绪,走对应处理函数即可
             if(FD_ISSET(0,&tempfds))
             {
                 printf("触发键盘输入事件\n");
                 fgets(buf,sizeof(buf),stdin);
                 buf[strlen(buf)-1]=0;
                 if(send(cfd,buf,sizeof(buf),0)<0)
                 {
                     ERR_MSG("send");
                     return -1;
                 }
                 printf("发送成功\n");
             }
             if(FD_ISSET(cfd,&tempfds))
             {
                 printf("触发客户端输入事件\n");
                 bzero(buf,sizeof(buf));
                 //接收数据
                 res=recv(cfd,buf,sizeof(buf),0);
                 if(res<0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if(0==res)
                {
                    printf("[%s:%d] cfd=%d 服务器掉线\n",IP,PORT,cfd);
                    break;
                }
                printf("[%s:%d] cfd=%d :%s \n",IP,PORT,cfd,buf);
            }
        }
    //关闭文件描述符
    close(cfd);
    return 0;
}


    3.搭建poll客户端

#include <stdio.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <head.h>

#define PORT 10000
#define IP "192.168.122.74"

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;   //必须填AF_INET
    sin.sin_port = htons(PORT);//端口号范围为1024到49151
    sin.sin_addr.s_addr = inet_addr(IP);    //本机IP

    //连接服务器
    if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("服务器连接成功\n");


    //创建一个读集合
    struct pollfd fds[2];
                                                                            
    fds[0].fd=0;
    fds[0].events = POLLIN;

    fds[1].fd=cfd;
    fds[1].events = POLLIN;

    char buf[128]="";
    ssize_t res=0;
    int p_res=-1;
    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..\n");
            break;
        }
        //运行到当前位置,则代表集合中有文件描述符准备就绪
        //此时集合中会只剩下触发事件的文件描述符
        //例如:
        //当0号文件描述符触发事件,则集合中只剩下0
        //当sfd文件描述符触发事件,则集合中只剩下sfd
        //若同时触发事件,则集合中剩下0和sfd
        //判断是哪个文件描述符准备就绪,走对应处理函数即可
            if(fds[0].revents & POLLIN)
            {
                printf("触发键盘输入事件\n");
                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("发送成功\n");
            }
            if(fds[1].revents & POLLIN)
            {
                printf("触发客户端输入事件\n");
                bzero(buf,sizeof(buf));
                //接收数据
                res=recv(cfd,buf,sizeof(buf),0);
                if(res<0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if(0==res)
                {
                    printf("[%s:%d] cfd=%d 服务器掉线\n",IP,PORT,cfd);
                    break;
                }
                printf("[%s:%d] cfd=%d :%s \n",IP,PORT,cfd,buf);
            }
        }
    //关闭文件描述符
    close(cfd);
    return 0;
}

4.思维导图

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值