3.16作业

最新推荐文章于 2024-08-10 15:48:22 发布

m0_65619859

最新推荐文章于 2024-08-10 15:48:22 发布

阅读量33

点赞数

文章标签：网络服务器 Powered by 金山文档

本文链接：https://blog.csdn.net/m0_65619859/article/details/129595094

版权

这篇文章展示了如何使用select函数创建一个TCP服务器，监听指定端口并处理多个客户端连接。服务器端接收到客户端请求后，可以进行数据的收发。同时，文章也给出了一个TCP客户端的示例，该客户端能随时收发数据，通过select函数监控键盘输入和服务器响应。

摘要由CSDN通过智能技术生成

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

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

int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd=socket(AF_INET,SOCK_SRTEAM);
    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;
    }
    //填充服务器地址的信息结构体
    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("bind");
        return -1;
    }
    //将套接字设置为被动监听状态
    if(listen(sfd,10)<0)
    {
        ERR_MSG("listen");
        return -1;
    }

    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 cin;     //存储连接成功的客户端的地址信息
    socklen_t addrlen=sizeof(cin);
    int newfd=-1;
    ssize_t res=-1;

    int  maxfd=sfd;
    while(1)
    {
        tempfds=readfds;

        //让内核去监测集合中的文件描述符是否准备就绪
        s_res=select(sfd+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++)
        {    
            if(FD_ISSET(i,&tempfds)==0)
            {
                continue;
            }
            //能运行到当前位置，则说明i所代表的文件描述符在集合中
            //所以需要判断i所代表的文件描述符是什么，如果为0，则走fgets，如果为sfd，则走accept
            //取余走recv
            if(0==i)
            {
                printf("触发键盘输入事件>>>");
                fflush(stdout);

                fgets(buf,sizeof(buf),stdin);
                buf[strlen(buf)-1]=0;
                printf("%s\n",buf);
            }
              else if(sfd==i)
            {
                printf("触发客户端连接事件\n");
                newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
                if(newfd<0)
                {
                    ERR_MSG("accept");
                    return -1;
                }
           
            //将新生成的文件描述符newfd，添加到集合中
            FD_SET(newfd,&readfds);
            //更新maxfd
            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),
                    //关闭文件描述符
                    close(i);
                    //将该文件描述符从集合中剔除
                    FD_CLR(i, &readfds);
                    //更新maxfd , 从大往小判断最大的文件描述符
                    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),
                    //发送 -- 将数据拼接一个 *_* 发送回去
                    strcat(buf, "*_*");
                    if(send(i, buf, sizeof(buf), 0) < 0)
                    {
                        ERR_MSG("send");
                        return -1;
                    }
                    printf("send success\n");
                }

            }
        }
    }
    //关闭所有套接字文件描述符
    close(sfd);
    return 0;
}

2.使用select函数完成TCP客户端，实现可随时收发。

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

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

#define PORT 6666               
#define IP  "192.168.31.123"    

int main(int argc, const char *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;
    }
    //填充服务器地址的信息结构体
    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("bind");
        return -1;
    }
    //连接服务器
    if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("与服务器连接成功\n");

    //设置读集合
    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]="";
    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("select");
            return -1;
        }
        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)
             {
                 printf("触发键盘输入事件>>>");
                 bzero(buf,sizeof(buf));

                 fgets(buf,sizeof(buf),stdin);
                 buf[strlen(buf)-1]='\0';
                 strcat(buf,"---Cli");
                 if(send(sfd,buf,sizeof(buf),0)<0)
                 {
                     ERR_MSG("send");
                     return -1;
                 }
                 printf("发送成功\n");
             }  
             //接收
             else if(sfd==i)
             {
                 printf("服务器交互事件>>>");
                 bzero(buf,sizeof(buf));
                 res=recv(sfd,buf,sizeof(buf),0);
                 if(res<0)
                 {
                     ERR_MSG("recv");
                     return -1;
                 }
                 else if(0==res)
                 {
                     fprintf(stderr,"newfd=%d 客户端下线\n",sfd);
                     close(i);
                     break;
                 }
                 printf("接受：newfd=%d:%s\n",sfd,buf);
             }
        }
    }
    close(sfd);
    return 0;
}