linux socket编程2-服务器代码示例

﹏ヽ暗。殇╰゛Y 2022-09-29 11:57 232阅读 0赞

#include <stdio.h>
    #include <stdlib.h>
    #include <memory.h>
    #include <signal.h>
    #include <time.h>
    #include <string.h>
    #include <unistd.h>
    #include <sys/socket.h>
    #include <netdb.h>
    
    int sockfd;
    
    void sig_handler(int signo)
    {
        if(signo == SIGINT)
        {
           printf("server close!\n");
           /*步骤6：关闭socket*/
           close(sockfd);
           exit(1);
        }
    }
    
    /*输出连接上来到客户端相关信息*/
    void out_addr(struct sockaddr_in *clientaddr)
    {
        //将端口从网络字节序转换成主机字节序列
        int port = ntohs(clientaddr->sin_port);
        char ip[16];
        memset(ip, 0, sizeof(ip));
        //将IP地址从网络字节序转换成点分十进制
        inet_ntop(AF_INET, &clientaddr->sin_addr.s_addr, ip, sizeof(ip));
        printf("client: %s(%d) connected\n", ip, port);
    }
    
    void do_service(int fd)
    {
        //获得系统时间
        long t = time(0);
        char* s = ctime(&t);
        size_t size = strlen(s) * sizeof(char);
        //将服务器获得的系统时间写会到客户端
        if(write(fd, s, size) != size)
        {
            perror("write error");
        }
    }
    
    int main(int argc, char* argv[])
    {
        if(argc < 2)
        {
            printf("usage: %s #port\n", argv[0]);
            exit(1);
        }
    
        if(signal(SIGINT, sig_handler) == SIG_ERR)
        {
            perror("signal sigint error");
            exit(1);
        }
        
        /*步骤1：创建socket(套接字)
         *注：socket创建在内核中是一个结构体.
         *AF_INET: IPV4
         *SOCK_STREAM: TCP协议
         */
        sockfd = socket(AF_INET, SOCK_STREAM, 0);    
        
        /*
         *步骤2：调用bind函数将socket与地址(包括ip、port)进行绑定
         */
        struct sockaddr_in serveraddr;
        memset(&serveraddr, 0, sizeof(serveraddr));
        //往地址中填入ip、port、internet地址族类型
        serveraddr.sin_family = AF_INET; //IPV4
        serveraddr.sin_port = htons(atoi(argv[1]));//将主机字节序转化为网络字节序
        serveraddr.sin_addr.s_addr = INADDR_ANY;
        if(bind(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr)) < 0)
        {
            perror("bind error");
            exit(1);
        }
        
        /*
         *步骤3:调用listen函数启动监听(指定port监听)
         *      通知系统去接收来自客户端的连接请求
         *      (将接收到到客户端连接请求放置到对应到队列中)
         *第二个参数：指定队列的长度
         */
        if(listen(sockfd, 10) < 0)
        {
            perror("listen error");
            exit(1);
        }
        
        /*
         *步骤4：调用accept函数从队列中获得一个客户端的请求连接,并返回新的socket描述符
         *注意：若没有客户端到连接，调用此函数后会阻塞直到获得一个客户端的连接。
         */
        struct sockaddr_in clientaddr;
        socklen_t clientaddr_len = sizeof(clientaddr);
        while(1)
        {
            int fd = accept(sockfd, (struct sockaddr*)&clientaddr, &clientaddr_len);
            if(fd < 0)
            {
                perror("accept error");
                continue;
            }
            
            /*
             *步骤5：调用IO函数(read/write)和连接到客户端进行双向到通讯
            */
            out_addr(&clientaddr);
            do_service(fd);
            
            /*
             *步骤6：关闭socket
            */
        }
        
        return 0;
    }