MyMuduo/Socket.cpp at main · cellphonef/MyMuduo · 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
#include "Socket.h"
#include "InetAddress.h"
#include "SocketOps.h"

#include <unistd.h>
#include <string.h>  // for memset
#include <netinet/in.h>  // for IPPROTO_xxx
#include <netinet/tcp.h>  // for TCP_xxx

Socket::Socket(int sockfd)
    : sockfd_(sockfd) {}

Socket::~Socket() {
    close(sockfd_);
}

int Socket::fd() const {
    return sockfd_;
}

void Socket::bindAddress(const InetAddress& localAddr) {
    bindOrDie(sockfd_, (sockaddr*)localAddr.getSockAddr());  // FIXME: use implicit_cast
}

void Socket::listen() {
    listenOrDie(sockfd_);
}


int Socket::accept(InetAddress* peerAddr) {
    sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));
    int connfd = accpet(sockfd_, (sockaddr*)&addr);
    if (connfd > 0) {
        peerAddr->setSockAddr(addr);
    }

    return connfd;

}


void Socket::shutdownWrite() {
    ::shutdownWrite(sockfd_);
}

void Socket::setReuseAddr(bool on) {
    int optval = on? 1 : 0;
    ::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEADDR, &optval, static_cast<socklen_t>(sizeof(optval)));

    // 错误处理
}


void Socket::setReusePort(bool on) {
    int optval = on? 1 : 0;
    // >=Linux 3.9才支持SO_REUSEPORT
    ::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEPORT, &optval, static_cast<socklen_t>(sizeof(optval)));

}


void Socket::setKeepAlive(bool on) {
    int optval = on? 1 : 0;
    ::setsockopt(sockfd_, SOL_SOCKET, SO_KEEPALIVE, &optval, static_cast<socklen_t>(sizeof(optval)));

}

void Socket::setTcpNoDelay(bool on) {
    int optval = on? 1 : 0;
    ::setsockopt(sockfd_, IPPROTO_TCP, TCP_NODELAY, &optval, static_cast<socklen_t>(sizeof(optval)));
}