4月12日

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

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

#define IP_LOCAL     "192.168.8.151"
#define POST_LOCAL     8989

int main(int argc, const char *argv[])
{
    __sighandler_t s = signal(SIGCHLD, hangler);
    if (SIG_ERR == s)
    {
        ERR_MSG("signal");
        return -1;
    }

    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket success sfd = %d\n", sfd);

    int reuse = 1;
    if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口重用\n");

    struct sockaddr_in sin;
    sin.sin_family         = AF_INET;
    sin.sin_port         = htons(POST_LOCAL);
    sin.sin_addr.s_addr = inet_addr(IP_LOCAL);

    if (bind(sfd, (struct sockaddr*)&sin, (socklen_t)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\n");

    
    //设置一个读集合

    //tempfds用作readfds备份
    fd_set readfds, tempfds;   //vi -t da_set  集合中是一个长整形的数组

    //数组在创建时会填入随机值，且不能为文件描述符，所以要清空
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);

    //将文件描述符添加到集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);
    
    int newfd = -1;
    pid_t cpid = 0;
    char buf[32] = "";
    int s_res;
    int res = 0;
    
    //最大文件描述符中转变量
    int maxfd = sfd;

    //存储客户端信息结构体数组
    struct sockaddr_in save_cin_info[1024-4]; //0 1 2 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");
            return -1;
        }

        //能运行到当前位置，代表有文件描述符准备就绪
        //走触发事件的文件描述符对应的处理函数
        
        //当集合中有文件描述符准备就绪，集合中只会保留产生时间的文件描述符
        //多个文件描述符准备就绪，就会保留多个文件描述符
        //用FD_ISSET判断是否有文件描述符

        //***********************************************************************//
        for (int i=0; i<=maxfd; i++)
        {
            //如果i文件描述符不存在
            if (!FD_ISSET(i, &tempfds))
            {
                continue;
            }
            //到这步，文件描述符肯定存在
            if (0 == i)
            {
                printf("出发键盘输入事件>>> ");         //读写转换刷新缓冲区
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf)-1] = 0;
                printf("%s\n", buf);
            }
            //用else if 不用else 因为i只有以一种可能
            else if (sfd == i)
            {
                printf("触发客户端连接事件>>> \n");
                
                newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
                if (newfd < 0)
                {
                    ERR_MSG("accept");
                    return -1;
                }
                
                //备份客户端信息
                save_cin_info[newfd-4] = cin;

                //将newfd添加到集合中
                FD_SET(newfd, &readfds);

                //更新文件描述符最大值
                maxfd = maxfd>newfd?maxfd:newfd;
            }
            else
            {
                bzero(buf, sizeof(buf));
                res = recv(i, buf, sizeof(buf), 0);
                if (res < 0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if (0 == res)
                {
                    printf("客户端掉线\n");
                    //关闭文件描述符
                    close(i);

                    //将关闭的文件描述符从集合中踢出
                    FD_CLR(i, &readfds);

                    //更新maxfd
                    int j=0;
                    for (j=maxfd; j>=0; j--)
                    {
                        if (FD_ISSET(j, &readfds))
                        {
                            maxfd = j;
                            break;
                        }
                    }

                    if (j<0)
                        maxfd = -1;
                    continue;    
                }
                printf("[%s : %d] newfd = %d : %s\n", inet_ntoa(save_cin_info[i-4].sin_addr), ntohs(save_cin_info[i-4].sin_port), i, buf);
            }
        }
    }
    close(sfd);
    return 0;
}

#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <sys/select.h>
#include <sys/time.h>
 
#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d ", __LINE__);\
    perror(msg);\
}while(0)
 
#define IP "192.168.8.151"     //本机IP ifconfig
#define PORT 7777             
 
int main(int argc, const char *argv[])
{
    //创建流式套接字
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if(cfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("cfd = %d\n", cfd);
 
    //允许端口快速重用
    int reuse = 1;
    if(setsockopt(cfd, 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(PORT) ;     //端口号，
    sin.sin_addr.s_addr = inet_addr(IP);     //本机IP， ifconfig
    
    if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("连接成功\n");
    //设置一个读集合
    fd_set readfds,tmpfds;
 
    //清空readfds
    FD_ZERO(&readfds);
    FD_ZERO(&tmpfds);
 
    //将需要的文件描述符添加到集合中
    FD_SET(0,&readfds);
    FD_SET(cfd,&readfds);
 
    int s_res = 0;
    int maxfd = cfd;
    char buf[128] = "";
    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);
    ssize_t res = 0;
 
    while(1)
    {
        tmpfds = readfds;
        s_res =select(maxfd+1,&tmpfds,NULL,NULL,NULL);
        if(s_res < 0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(0 == s_res)
        {
            printf("超时\n");
            return -1;
        }
 
        if(FD_ISSET(0,&tmpfds))
        {
            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");
        }
        else if(FD_ISSET(cfd,&tmpfds))
        {
            //接受
            bzero(buf,sizeof(buf));
            res = recv(cfd,buf,sizeof(buf),0);
            if(res < 0)
            {
                ERR_MSG("recv");
                return -1;
            }
            else if(0 == res)
            {
                printf("cfd=%d 服务器下线 __%d__\n",cfd,__LINE__);
 
            }
            printf("cfd=%d :%s __%d__\n",cfd,buf,__LINE__);
 
        }
    
    }
 
    close(cfd);
 
    return -1;
}