7.10作业

#include
#include
#include
#include
#include
#include
#include
#define ERR_MSG(msg) {\
    fprintf(stderr,"__%d__",__LINE__);\
    perror(msg);\
}
#define IP "192.168.122.138"
#define PORT 8888

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

    //定义地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family      =  AF_INET;
    sin.sin_port        =  htons(PORT);
    sin.sin_addr,s_addr =  inet_addr(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");
    fd_set readfds,tempfds;
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);
    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);

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

    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);
    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__);
        //能运行到当前位置，则代表集合中所有文件描述符准备就绪
        //判断哪个文件描述符准备就绪，走对应的处理函数即可
        //

            FD_SET(newfd,&readfds);
            maxfd = maxfd>newfd ? maxfd:newfd;
        }
        
        //通过循环遍历集合中所有文件描述符
        //由于sfd和newfd的值不确定，所以从0开始遍历
        for(int i=0;i<=maxfd;i++)
        {
            if(FD_ISSET(i,&tempfds) == 0)
            {
                continue;
            }
            //能运行到当前位置，则说明i所代表的文件描述符在tempfds里
            if(0 == i)//键盘输入事件
            {
              printf("触发键盘输入事件\n");
              int sndfd;//储存要发送给哪个客户端

              res = scanf("%d %s",&sndfd,buf);
              while(getchar()!=10);//循环吸收垃圾字符
              if(res!=2)
              {
                  fprintf(stderr,"sndfd=%d 输入的数据格式错误\n");
                  continue;
              }

              //能运行到当前位置，说明输入的格式是正确的
              //需要排除非法的文件描述符
              if(sndfd<=2 || sndfd>=1024 || !FD_ISSET(sndfd,&readfds))
              {
                  fprintf(stderr,"sndfd=%d 文件描述符错误\n",sndfd);
                  continue；
              }
                
              if(send(sndfd,buf,sizeof(buf),0)<0)
              {
                  ERR_MSG("send");
                  continue;
              }
              printf("发送成功");
            else if(sfd == i)//客户端连接事件
            {
             printf("触发客户端连接事件\n");
             newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
             if(newfd<0)
             {
                 ERR_MSG("accept");
                 continue;
             }
             printf("[%s : %d] newfd=%d 客户端连接成功\n",\
                     inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
            }
             saveCin[newfd-3] = cin;//将客户端信息另存为
             FD_SET(newfd,&readfds);//将newfd添加到集合readfds中
             maxfd = maxfd>newfd ? maxfd:newfd;//更新maxfd
            else//客户端交互事件
            {
                printf("触发客户端交互事件\n");
                bzero(buf,sizeof(buf));
                //接收数据
                res = recv(i,buf,sizeof(buf),0);
                if(res<0)
                {
                    ERR_MSG("recv");
                    return continue;
                }
                else if(0 == res)
                {
                   printf("[%s : %d] newfd=%d  客户端下线\n",\
                           inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i);
                   
                   close(i);//关闭文件描述符

                   FD_CLR(i,&readfds);//将i所代表的文件描述符删除

                   //更新maxfd-->有可能会将集合中最大的文件描述符剔除
                   //从最大的文件描述符开始遍历，直到文件描述符在集合中
                   //则该文件描述符
                   int j = maxfd;
                   for( ; j>=0;j--)
                   {
                     if(FD_ISSET(j,&readfds))
                         break;
                   }
                   maxfd = j;
                   //while(!FD_ISSET(maxfd,&readfds) && maxfd-->=0);
                   
                   
                   continue;
                }
                printf("[%s : %d] newfd=%d : %s\n",\
                        inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf);
                 strcat(buf,"*__*");

                 //发送数据
                 if(send(i,buf,sizeof(buf),0)<0)
                 {
                     ERR_MSG("send");
                     continue;
                 }
                 printf("发送成功");
            }

        }
    }
    if(close(sfd)<0)
    {
        ERR_MSG("close");
        return -1;
    }
    return 0;
}