netframe/Socket.cpp at master · lfxy/netframe · 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
#include "Socket.h"
#include "InetAddress.h"
#include "SocketOps.h"
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <strings.h>

Socket::~Socket()
{
    SocketOps::close(m_sockfd);
}

void Socket::bindAddress(const InetAddress& localaddr)
{
    SocketOps::bindOrDie(m_sockfd, localaddr.getSockAddrInet());
}

void Socket::listen()
{
    SocketOps::listenOrDie(m_sockfd);
}

int Socket::accept(InetAddress* peeraddr)
{
    struct sockaddr_in addr;
    bzero(&addr, sizeof(addr));
    int connfd = SocketOps::accept(m_sockfd, &addr);
    if(connfd >= 0)
    {
        peeraddr->setSockAddrInet(addr);
    }
    return connfd;
}

void Socket::shutdownWrite()
{
    SocketOps::shutdownWrite(m_sockfd);
}

void Socket::setTcpNoDelay(bool on)
{
    int optval = on ? 1: 0;
    ::setsockopt(m_sockfd, IPPROTO_TCP, TCP_NODELAY, &optval, sizeof(optval));
}

void Socket::setReuseAddr(bool on)
{
    int optval = on ? 1: 0;
    ::setsockopt(m_sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
}

void Socket::setKeepAlive(bool on)
{
    int optval = on ? 1: 0;
    ::setsockopt(m_sockfd, SOL_SOCKET, SO_KEEPALIVE, &optval, sizeof(optval));
}