网络编程_Day05 select/poll搭建服务器/客户端

使用select的TCP服务器

#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/select.h>
#include<sys/time.h>

#define IP "192.168.31.165"  //本机IP

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


int main(int argc, char const *argv[])
{
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd < 0)
    {
        ERR_MSG("socket");    
        return -1;
    }



    //在绑定端口之前,允许该端口能被快速释放。
    //允许端口快速被重复使用,当检测到该端口号没有被进程正常占用的时候
    //允许端口快速被新的进程占用覆盖
    int reuse = 1;
    if( setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重用\n" );



    //填充服务器的地址信息结构体
    //该真实结构体根据地址族制定:AF_INET: man 7 ip
    struct sockaddr_in sin;
    sin.sin_family  = AF_INET;  //必须填充AF_INET
    sin.sin_port    = htons(9999);//端口号的网络字节序1024~49151
    sin.sin_addr.s_addr = inet_addr(IP); //IP,本机ip: ifconfig

    //绑定
    //将服务器的IP地址和端口号绑定到服务器套接字文件中
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);

    //将套接字设置为被动监听状态
    if(listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success __%d__\n", __LINE__);


    //创建一个读集合,fd_set是一个结构体,里面只有long int类型的数组
    //由于readfds中需要存储要检测的文件描述符
    //所以不能让readfds中的数组中为随机值,需要清空
    fd_set readfds, tempfds;
    FD_ZERO(&readfds); //清空
    
    //将需要的文件描述符添加到集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);
    
    int maxfd = sfd;        //存储集合中最大的文件描述符

    
    struct sockaddr_in cin; //存储连接成功的客户端地址信息
    socklen_t addrlen = sizeof(cin);
    
    int newfd = -1;   //存储连接成功的客户端地址信息
    char buf[128] = "";
    ssize_t res = 0;
    int s_res = 0;
    while(1)
    {
        //每次循环更新readfds集合,因为select运行后会只保留触发的文件描述符
        tempfds = readfds;
        //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;
        }
        printf("__%d__\n", __LINE__);


        //遍历触发集合中的文件描述符,并做相应操作
        for(int i = 0; i <= maxfd; i++)
        {
            printf("__%d__\n", __LINE__);
            if(0 == FD_ISSET(i, &tempfds)) //i所代表的文件描述符不在集合中
            {
                continue;  //进入下一次循环
            }

            //能运行到当前位置,表示i对应的文件描述符在集合中
            //判断i是什么描述符即可
            if(0 == i)
            {
                printf("触发键盘输入事件>>>");
                fflush(stdout);

                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf)-1] = 0;
                printf("%s\n", buf);

            }

            else if(sfd == i)   //i == 3
            {
                printf("触发客户端连接事件>>>");
                fflush(stdout);

                // 从已完成连接的队列头中获取一个客户端信息, 生产一个新的文件描述符    
                newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
                if (newfd < 0)
                {
                    ERR_MSG("accept");
                    return -1;
                }
                printf("[%s : %d] newfd = %d 连接成功__%d__\n",\
                    inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);
                
                //将newfd添加到集合
                FD_SET(newfd, &readfds);

                //更新maxfd
                maxfd = maxfd>newfd?maxfd:newfd;
            }
            else if(i > 3)
            {
                fprintf(stderr, "触发客户端交互事件>>>");
                //对于字符串的操作,操作之前或者操作之后要清空,
                //防止之前的数据对这次操作有干扰
                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 客户端下线__%d__\n",\
                        inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i, __LINE__);
                    //关闭文件描述符
                    close(i);
                    //从集合中剔除文件描述符
                    FD_CLR(i, &readfds);

                    //更新maxfd,j从大到小依次遍历直到有一个文件描述符在集合中
                    int j = maxfd;
                    for( ; j>=0; j--)
                    {
                        if(FD_ISSET(j, &readfds))
                        {
                            break;
                        }
                    }
                    maxfd = j;
                    continue;  //某个客户端下线,不能break出循环,否则会使得其他客户端发送的数据不能被接收,需要将break改成continue继续循环;
                }
                //客户端发送数据
                printf("[%s : %d] newfd = %d : %s__%d__\n",\
                        inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i, buf, __LINE__);
                
                //发送--->谁发给我数据,我就发还给谁
                strcat(buf, "*_*");
                if(send(i, buf, sizeof(buf), 0) < 0)
                {
                    ERR_MSG("send");
                    return -1;
                }
                printf("send success\n");
            }
        }
    }
    
    //关闭文件描述符
    close(sfd);
    return 0;
}

使用poll()的TCP客户端

#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<poll.h>


#define IP "192.168.31.165"  //服务器IP

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


int main(int argc, char const *argv[])
{
    //创建流式套接字
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if(cfd < 0)
    {
        ERR_MSG("socket");    
        return -1;
    }


    //绑定==>非必须绑定
    //功能:将客户端的IP地址和端口号绑定到客户端套接字文件中
    //如果不绑定,会默认将本机IP和随机端口(从临时端口中随机取)
    //一般不建议自己绑定,所以此处省略


    //填充服务器的地址信息结构体,该结构体用于给connect函数使用
    //该真实结构体根据地址族制定:AF_INET: man 7 ip
    struct sockaddr_in sin;
    sin.sin_family  = AF_INET;  //必须填充AF_INET
    sin.sin_port    = htons(9999);//端口号的网络字节序1024~49151
    sin.sin_addr.s_addr = inet_addr(IP); //IP,本机ip: ifconfig


    //连接服务器
    if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    //创建文件描述符结构体集合
    struct pollfd pfds[2];

    pfds[0].fd = 0;
    pfds[0].events = POLLIN;

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

    int p_res;

    char buf[128] = "";
    ssize_t res = 0;
    while(1)
    {
        printf("%d\n", __LINE__);
        p_res = poll(pfds, 2, -1);
        printf("%d\n", __LINE__);
        if(p_res < 0)
        {
            ERR_MSG("poll");
            return -1;
        }
        else if(0 == p_res)
        {
            printf("time out\n");
            break;
        }
        printf("%d\n", __LINE__);
        if(pfds[0].revents & POLLIN)
        {
            //对于字符串的操作,操作之前或者操作之后要清空,
            //防止之前的数据对这次操作有干扰
            bzero(buf, sizeof(buf));

            printf("触发键盘输入事件>>>");
            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(pfds[1].revents & POLLIN)
        {
            //对于字符串的操作,操作之前或者操作之后要清空,
            //防止之前的数据对这次操作有干扰
            printf("触发服务器交互事件");
            fflush(stdout);
            bzero(buf, sizeof(buf));
            //接收
            res = recv(cfd, buf, sizeof(buf), 0);
            if(res < 0)
            {
                ERR_MSG("recv");
                return -1;
            }
            else if(0 == res)
            {
                printf("服务器下线\n");
                break;
            }
            printf(" : %s__%d__\n", buf, __LINE__);
        }

    }
    

    //关闭文件描述符
    close(cfd);
    return 0;
}

使用select()的TCP客户端

#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/select.h>
#include <sys/time.h>

#define IP "192.168.31.165"  //服务器IP

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


int main(int argc, char const *argv[])
{
    //创建流式套接字
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if(cfd < 0)
    {
        ERR_MSG("socket");    
        return -1;
    }


    //绑定==>非必须绑定
    //功能:将客户端的IP地址和端口号绑定到客户端套接字文件中
    //如果不绑定,会默认将本机IP和随机端口(从临时端口中随机取)
    //一般不建议自己绑定,所以此处省略


    //填充服务器的地址信息结构体,该结构体用于给connect函数使用
    //该真实结构体根据地址族制定:AF_INET: man 7 ip
    struct sockaddr_in sin;
    sin.sin_family  = AF_INET;  //必须填充AF_INET
    sin.sin_port    = htons(9999);//端口号的网络字节序1024~49151
    sin.sin_addr.s_addr = inet_addr(IP); //IP,本机ip: ifconfig


    //连接服务器
    if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    fd_set readfds, tempfds;  //创建一个文件描述符集合及其副本
    FD_ZERO(&readfds); //清空集合
    //将stdin和cfd加入集合
    FD_SET(0, &readfds);
    FD_SET(cfd, &readfds);


    int s_res;
    char buf[128] = "";
    ssize_t res = 0;
    // printf("__%d__\n", __LINE__);
    while(1)
    {
        tempfds = readfds;
        s_res = select(cfd+1, &readfds, NULL, NULL, NULL);
        //意外情况处理
        if(s_res<0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(0 == s_res)
        {
            printf("time out ...\n");
            break;
        }
        //正常情况处理
        for(int i = 0; i<=cfd; i++)
        {
            if(0 == FD_ISSET(i, &tempfds))
            {
                continue;
            }
            if(0 == i) //键盘有输入
            {
                //对于字符串的操作,操作之前或者操作之后要清空,
                //防止之前的数据对这次操作有干扰
                bzero(buf, sizeof(buf));
                printf("触发键盘输入事件>>>");
                fflush(stdout);

                //从stdin读取字符串到buf
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf)-1] = 0;
                printf("%s\n", buf);

                //向服务器发送字符串
                if(send(cfd, buf, sizeof(buf), 0) < 0)
                {
                    ERR_MSG("send");
                    return -1;
                }
                printf("发送成功\n");

            }
            else if(cfd == i)
            {
                fprintf(stderr, "触发服务器交互事件>>>");
                //对于字符串的操作,操作之前或者操作之后要清空,
                //防止之前的数据对这次操作有干扰
                bzero(buf, sizeof(buf));
                //接收
                res = recv(cfd, buf, sizeof(buf), 0);
                if(res < 0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if(0 == res)
                {
                    printf("服务器下线\n");
                    
                    FD_CLR(i, &readfds);//从集合中剔除
                    //关闭文件描述符
                    close(i);
                    return 0;
                }
                printf(" : %s__%d__\n", buf, __LINE__);
            }
        }
    }
    
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值