aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/messagebus/socket_addr.cpp
blob: 0b4095abf05b23271c0e862199be29f31043033c (plain) (blame)
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
#include "socket_addr.h"

#include "netaddr.h"

#include <util/network/address.h>
#include <util/network/init.h>
#include <util/system/yassert.h>

using namespace NAddr;

using namespace NBus;
using namespace NBus::NPrivate;

static_assert(ADDR_UNSPEC == 0, "expect ADDR_UNSPEC == 0"); 

NBus::NPrivate::TBusSocketAddr::TBusSocketAddr(const NAddr::IRemoteAddr* addr)
    : IPv6ScopeID(0)
{
    const sockaddr* sa = addr->Addr();

    switch ((EAddrFamily)sa->sa_family) { 
        case AF_UNSPEC: { 
            IpAddr.Clear(); 
            Port = 0; 
            break; 
        } 
        case AF_INET: { 
            IpAddr.SetInAddr(((const sockaddr_in*)sa)->sin_addr); 
            Port = InetToHost(((const sockaddr_in*)sa)->sin_port); 
            break; 
        } 
        case AF_INET6: { 
            IpAddr.SetIn6Addr(((const sockaddr_in6*)sa)->sin6_addr); 
            Port = InetToHost(((const sockaddr_in*)sa)->sin_port); 
            IPv6ScopeID = InetToHost(((const sockaddr_in6*)sa)->sin6_scope_id); 
            break; 
        } 
        default: 
            Y_FAIL("unknown address family"); 
    }
}

NBus::NPrivate::TBusSocketAddr::TBusSocketAddr(TStringBuf host, unsigned port) { 
    *this = TNetAddr(host, port);
}

NBus::NPrivate::TBusSocketAddr::TBusSocketAddr(const TNetAddr& addr) { 
    *this = TBusSocketAddr(&addr);
}

TNetAddr NBus::NPrivate::TBusSocketAddr::ToNetAddr() const { 
    sockaddr_storage storage;
    Zero(storage);

    storage.ss_family = (ui16)IpAddr.GetAddrFamily(); 

    switch (IpAddr.GetAddrFamily()) {
        case ADDR_UNSPEC: 
            return TNetAddr(); 
        case ADDR_IPV4: { 
            ((sockaddr_in*)&storage)->sin_addr = IpAddr.GetInAddr(); 
            ((sockaddr_in*)&storage)->sin_port = HostToInet(Port); 
            break; 
        } 
        case ADDR_IPV6: { 
            ((sockaddr_in6*)&storage)->sin6_addr = IpAddr.GetIn6Addr(); 
            ((sockaddr_in6*)&storage)->sin6_port = HostToInet(Port); 
            ((sockaddr_in6*)&storage)->sin6_scope_id = HostToInet(IPv6ScopeID); 
            break; 
        } 
    }

    return TNetAddr(new TOpaqueAddr((sockaddr*)&storage)); 
}

template <>
void Out<TBusSocketAddr>(IOutputStream& out, const TBusSocketAddr& addr) {
    out << addr.ToNetAddr();
}