5.22作业

最新推荐文章于 2024-10-05 22:19:52 发布

肥嘟嘟左卫门66

最新推荐文章于 2024-10-05 22:19:52 发布

阅读量359

点赞数 3

文章标签：算法

本文链接：https://blog.csdn.net/2402_83485638/article/details/139127048

版权

SELECT服务器

#include<header.h>
#define SER_PORT 8886
#define SER_IP "192.168.126.54"   

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

    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 error");
        return -1;
    }
    printf("bind success\n");

    if(listen(sfd, 128)==-1)
    {
        perror("listen error");
        return -1;
    }
    printf("listen success\n");

    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);

    fd_set readfds,tempfds;
    FD_ZERO(&readfds);
    FD_SET(sfd,&readfds);
    FD_SET(0,&readfds);
    int newfd;
    int maxfd = sfd;
    struct sockaddr_in cin_addr[1024];

    while(1)
    {

        tempfds = readfds;
        int res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
        if(res == -1)
        {
            perror("select");
            return -1;
        }else if(res ==0)
        {
            printf("time out\n");
            return -1;
        }

        for(int i = 0; i<=maxfd;i++)
        {                                                                                                                                                   
            if(!FD_ISSET(i,&tempfds))
            {
                continue;
            }
            if(i == sfd)
            {
                newfd = accept(i, (struct sockaddr*)&cin, &addrlen);
                if(newfd == -1)
                {
                    perror("accept error");
                    return -1;
                }
                printf("[%s:%d]:发来连接请求\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
                FD_SET(newfd,&readfds);
                if(newfd >= maxfd)
                {
                    maxfd = newfd;
                }
                cin_addr[newfd] = cin;


            }else if(i == 0)
            {
                char buf[128] = "";
                fgets(buf,sizeof(buf),stdin);
                buf[strlen(buf) -1] = 0;
                printf("键盘输入:%s\n",buf);
                for(int k = 4 ; k <=maxfd ; k++)
                {
                    send(k,buf,strlen(buf),0);
                }

            }else
            {
                char rbuf[128] = "";
                bzero(rbuf, sizeof(rbuf));

                int res = recv(i, rbuf, sizeof(rbuf), 0);
                if(res == 0)
                {
                    printf("客户端已经下线\n");
                    close(i);
                    FD_CLR(i,&readfds);

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

                printf("[%s:%d]:%s\n", inet_ntoa(cin_addr[i].sin_addr), ntohs(cin_addr[i].sin_port), rbuf);
                for(int k = 4 ; k <= maxfd ; k++)
                {
                    if(k != i)
                    {
                        send(k,rbuf,strlen(rbuf),0);
                    }
                }
              
            }
        }
    }
    close(sfd);


    return 0;
}

POLL客户端

#include<header.h>
#define SER_PORT 8886
#define SER_IP "192.168.126.54"
#define CLI_PORT 7777
#define CLI_IP "192.168.126.54"

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

    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 error");
        return -1;
    }
    printf("bind success\n");

    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(connect(cfd, (struct sockaddr*)&sin, sizeof(sin))==-1)
    {
        perror("connect error");
        return -1;
    }
    printf("connect success\n");

    struct pollfd pfd[2];
    pfd[0].fd = 0;
    pfd[0].events = POLLIN;

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

    char wbuf[128] = "";
    char rbuf[128] = "";
    while(1)
    {
        int res = poll(pfd, 2, -1);
        if(res == -1)
        {
            perror("poll error");
            return -1;
        }else if(res == 0)
        {
            printf("time out\n");
            return -1;
        }

        if(pfd[0].revents == POLLIN)
        {
            fgets(wbuf, sizeof(wbuf), stdin);
            wbuf[strlen(wbuf)-1] = '\0';
            send(cfd, wbuf, strlen(wbuf), 0);
            printf("发送成功\n");
        }

        if(pfd[1].revents == POLLIN)
        {
            bzero(rbuf, sizeof(rbuf));
            recv(cfd, rbuf, sizeof(rbuf), 0);
            printf("服务器发来的消息为：%s\n", rbuf);
        }
    }

    close(cfd);
    return 0;
}