linux网络模型非阻塞i/o模型代码实现

客官°小女子只卖身不卖艺 2022-06-01 03:44 167阅读 0赞

#include <iostream>
    #include <stdlib.h>
    #include <sys/socket.h>
    #include <sys/types.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <unistd.h>
    #include <error.h>
    #include <pthread.h>
    #include <stdio.h>
    #define ErrFlag -1
    
    #define ShowErrMsg(msg)\
        do{ perror(msg); exit(1);  }while(0)
    
    
    void* myFunc(void* ptrNewFd)
    {
        int newfd = (int)(intptr_t)ptrNewFd;
        if((newfd < 0)  || (newfd == 0))
        {
            ShowErrMsg("thread function newfd");
            exit(1);
        }
    
        char buf[1024];
        read(newfd, buf, sizeof(buf));
    
        std::cout << buf << std::endl;
    
    }
    
    int main(int argc, char* argv[])
    {
        int retStatus = 0;
        int listenNums = 50;
        int fd = socket(AF_INET, SOCK_STREAM, 0);
        if(ErrFlag == fd)
        {
            ShowErrMsg("socket");
            exit(1);
        }
    
        struct sockaddr_in serverAddr;
        serverAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
        serverAddr.sin_family = AF_INET;
        serverAddr.sin_port  = htons(19999);
        
        retStatus =  bind(fd, (struct sockaddr *)&serverAddr, sizeof(serverAddr));
        if(ErrFlag == retStatus)
        {
            ShowErrMsg("bind");
            exit(1);
        }
    
        retStatus = listen(fd,listenNums);
        if(ErrFlag == retStatus)
        {
            ShowErrMsg("listen");
            exit(1);
        }
    
        while(1)
        {
            int newfd = accept(fd, NULL, NULL);
            if(ErrFlag == newfd)
            {
                ShowErrMsg("accept");
                exit(1);
            }
    
            pthread_t pid;
            pthread_create(&pid,NULL, myFunc,(void*)(intptr_t)newfd);
        }
        close(fd);
        return 0;
    }