HttpServer/SocketOps.cpp at main · cellphonef/HttpServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#include "SocketOps.h"
#include "Logger.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <fcntl.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/errno.h>


int createNonblockingOrDie() {

    int fd = socket(AF_INET, SOCK_STREAM, 0);
    if (fd == -1)
        LOG_ERROR("%s : %s", "socket error", strerror(errno));


    // 常用代码片段，将fd设置为非阻塞
    // 先获取旧的标志集合再或上新的标志得到新的标志集合
    int flags = fcntl(fd, F_GETFL, 0);
    if (flags == -1)
        LOG_ERROR("%s : %s", "fcntl F_GETFL error", strerror(errno));

    flags |= O_NONBLOCK;
    if (fcntl(fd, F_SETFL, flags) == -1)
        LOG_ERROR("%s : %s", "fcntl F_SETFL error", strerror(errno));

    return fd;
}

void bindOrDie(int sockFd, int port) {
    /*
    * sockaddr_in IPv4套接字地址结构
    *
    * struct sockaddr_in {
    *     uint8_t sin_len;  // 结构长度，除非涉及路由套接字否则无需设置和检查它
    *     sa_family_t sin_family;
    *     in_port_t sin_port;
    *     struct in_addr sin_addr;
    *     char sin_zero[8];
    * }
    */

    struct sockaddr_in servAddr;
    bzero(&servAddr, sizeof(servAddr));
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servAddr.sin_port = htons(port);

    if (bind(sockFd, (sockaddr*)&servAddr, sizeof(servAddr)) == -1)
        LOG_ERROR("%s : %s", "bind error", strerror(errno));


}

void listenOrDie(int sockFd) {
    if (listen(sockFd, 4096) == -1)
        LOG_ERROR("%s : %s", "listen error", strerror(errno));

}

int acceptOrDie(int listenFd, struct sockaddr_in* clientAddr) {
    socklen_t clientLen = sizeof(*clientAddr);
    int connFd = accept(listenFd, (sockaddr*)(&clientAddr), &clientLen);  // FIXME: c++11 cast

    if (connFd < 0) {
        if (errno != EAGAIN)
            LOG_ERROR("%s : %s", "accept error", strerror(errno));

    }

    return connFd;

}

void setNonBlocking(int sockFd) {
    int flags = fcntl(sockFd, F_GETFL, 0);
    if (flags == -1)
        LOG_ERROR("%s : %s", "fcntl F_GETFL error", strerror(errno));

    flags |= O_NONBLOCK;
    if (fcntl(sockFd, F_SETFL, flags) == -1) {
        LOG_ERROR("%s : %s", "fcntl F_SETFL error", strerror(errno));
    }
}