3.16作业

这篇文章展示了如何使用select和poll系统调用来实现IO多路复用,分别构建了一个select型的服务器和客户端,以及一个poll型的客户端。服务器监听并接受多个客户端连接,处理交互事件,而客户端则负责连接服务器并进行数据收发。代码中包含了错误处理和套接字操作。
摘要由CSDN通过智能技术生成

io多路复用

head.h

#ifndef __HEAD_H__
#define __HEAD_H__
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <pthread.h>
#include <sys/select.h>
#include <sys/time.h>
#include <poll.h>


#define ERR_MSG(msg) do{\
    fprintf(stderr,"line:%d ",__LINE__);\
    perror(msg);\
}while(0)

#define ERR_MSG_1(msg) ERR_MSG(msg);return -1

#define PORT 2222
#define IP "192.168.1.33"
#endif

select型服务器

#include "head.h"
int main(int argc, const char *argv[])
{
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if(sfd<0)
    {
        ERR_MSG_1("socket");
    }

    int resue=1;
    if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&resue,sizeof(resue))<0)
    {
        ERR_MSG_1("setsockopt");
    }

    struct sockaddr_in sin;
    sin.sin_family=AF_INET;
    sin.sin_port=htons(PORT);
    sin.sin_addr.s_addr=inet_addr(IP);

    if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG_1("bind");
    }

    if(listen(sfd,128)<0)
    {
        ERR_MSG_1("listen");
    }

    fd_set readfds,tempfds;
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);

    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);


    int s_res=0;
    char buf[128]="";

    struct sockaddr_in saveCin[1024-4];
    struct sockaddr_in cin;
    socklen_t addrlen=sizeof(cin);

    int newfd=-1;
    ssize_t res=-1;

    int maxfd=sfd;

    while(1)
    {
        tempfds=readfds;

        s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
        if(s_res<0)
        {
            ERR_MSG_1("select");
        }
        else if(0==s_res)
        {
            printf("time out...\n");
            break;
        }

        for(int i=0;i<=maxfd;i++)
        {
            if(FD_ISSET(i,&tempfds)==0) 
            {
                continue;
            }
            if(0==i)
            {
                int sndfd;
                res = scanf("%d %s",&sndfd,buf);
                while(getchar()!=10);
                if(res!=2)
                {
                    fprintf(stderr,"请输入正确格式:文件描述符内容\n");
                    continue;
                }
                if(sndfd>=1024||sndfd<=sfd||!FD_ISSET(sndfd,&readfds))
                {
                    fprintf(stderr,"sndfd=%d没有对应的客户端\n", sndfd);
                    continue;
                }
                if(send(sndfd,buf,sizeof(buf),0)<0)
                {
                    ERR_MSG_1("send");
                }
                printf("send success\n");
            }
            else if(sfd==i)
            {
                printf("触发客户端连接事件\n");
                newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
                if(newfd<0)
                {
                    ERR_MSG_1("accept");
                }
                printf("[%s : %d] newfd=%d 客户端连接成功\n",\
                        inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

                saveCin[newfd-4]=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");
                }
                else if(0==res)
                {
                    fprintf(stderr,"[%s : %d] newfd=%d 客户端下线\n",\
                            inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[i-4].sin_port),i);
                    close(i);
                    FD_CLR(i,&readfds);
                    for(int j=maxfd;j>sfd;j--)
                    {
                        if(FD_ISSET(j,&readfds))
                        {
                            maxfd=j;
                            break;
                        }
                    }
                }
                else
                {
                    printf("[%s : %d] newfd=%d : %s\n",\
                            inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[i-4].sin_port),i,buf);
                    strcat(buf,"*_*");
                    if(send(i,buf,sizeof(buf),0)<0)
                    {
                        ERR_MSG_1("send");
                    }
                    printf("send success\n");
                }
            }
        }
    }

    close(sfd);
    return 0;
}

select型客户端

#include "head.h"

int main(int argc, const char *argv[])
{
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if (sfd<0)
    {
        ERR_MSG_1("socket");
    }

    //填写服务器地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family=AF_INET;
    sin.sin_port=htons(PORT);
    sin.sin_addr.s_addr=inet_addr(IP);

    if (connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG_1("connect");
    }

    fd_set readfds,tempfds;
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);

    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);

    char buf[128]="";
    ssize_t res=-1;
    int s_res=-1;

    while (1)
    {
        tempfds=readfds;

        s_res=select(sfd+1,&tempfds,NULL,NULL,NULL);
        if (s_res<0)
        {
            ERR_MSG_1("select");
        }
        else if (0==s_res)
        {
            puts("time out");
            break;
        }

        if (FD_ISSET(0,&tempfds))
        {
            bzero(buf,sizeof(buf));
            fgets(buf,sizeof(buf),stdin);
            buf[strlen(buf)-1]=0;

            if (send(sfd,buf,sizeof(buf),0)<0)
            {
                ERR_MSG_1("send");
            }
            puts("send success");
        }

        if (FD_ISSET(sfd,&tempfds))
        {
            bzero(buf,sizeof(buf));
            res=recv(sfd,buf,sizeof(buf),0);
            if (res<0)
            {
                ERR_MSG_1("recv");
            }
            else if(0==res)
            {
                puts("服务器下线");
                break;
            }
            printf(":%s\n",buf);
        }
    }

    close(sfd);
    return 0;
}

poll型客户端

#include "head.h"

int main(int argc, const char *argv[])
{
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if (sfd<0)
    {
        ERR_MSG_1("socket");
    }

    //填写服务器地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family=AF_INET;
    sin.sin_port=htons(PORT);
    sin.sin_addr.s_addr=inet_addr(IP);

    if (connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG_1("connect");
    }

    struct pollfd fds[2];
    fds[0].fd=0;
    fds[0].events=POLLIN;
    fds[1].fd=sfd;
    fds[1].events=POLLIN;

    char buf[128]="";
    ssize_t res=-1;
    int s_res=-1;

    while (1)
    {
        s_res=poll(fds,2,-1);

        if (s_res<0)
        {
            ERR_MSG_1("poll");
        }
        else if (0==s_res)
        {
            puts("time out");
            break;
        }

        if (fds[0].revents&POLLIN)
        {
            bzero(buf,sizeof(buf));
            fgets(buf,sizeof(buf),stdin);
            buf[strlen(buf)-1]=0;

            if (send(sfd,buf,sizeof(buf),0)<0)
            {
                ERR_MSG_1("send");
            }
            puts("send success");
        }

        if (fds[1].revents&POLLIN)
        {
            bzero(buf,sizeof(buf));
            res=recv(sfd,buf,sizeof(buf),0);
            if (res<0)
            {
                ERR_MSG_1("recv");
            }
            else if(0==res)
            {
                puts("服务器下线");
                break;
            }
            printf(":%s\n",buf);
        }

    }

    close(sfd);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值