library/cpp/netliba/v6/udp_socket.h


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

#pragma once

#include <util/generic/ptr.h>
#include <util/generic/utility.h>
#include <library/cpp/netliba/socket/socket.h>

namespace NNetliba {
    bool IsLocalIPv4(ui32 ip);
    bool IsLocalIPv6(ui64 network, ui64 iface);
    bool InitLocalIPList();

    constexpr int UDP_LOW_LEVEL_HEADER_SIZE = 4;

    using NNetlibaSocket::EFragFlag;
    using NNetlibaSocket::FF_ALLOW_FRAG;
    using NNetlibaSocket::FF_DONT_FRAG;

    class TNetSocket: public TNonCopyable {
        TIntrusivePtr<NNetlibaSocket::ISocket> s;
        ui32 PortCrc;
        mutable int Tos;

    public:
        enum ESendError {
            SEND_OK,
            SEND_BUFFER_OVERFLOW,
            SEND_NO_ROUTE_TO_HOST,
        };
        TNetSocket()
            : PortCrc(0)
            , Tos(0)
        {
        }
        ~TNetSocket() {
        }

        void Open(int port);
        void Open(const TIntrusivePtr<NNetlibaSocket::ISocket>& socket);
        void Close();
        void SendSelfFakePacket() const;
        bool IsValid() const {
            return s.Get() ? s->IsValid() : false;
        }
        int GetNetworkOrderPort() const {
            return s->GetNetworkOrderPort();
        }
        ESendError SendTo(const char* buf, int size, const sockaddr_in6& toAddress, const EFragFlag frag) const;
        bool RecvFrom(char* buf, int* size, sockaddr_in6* fromAddress) const;
        void Wait(float timeoutSec) const;
        void SetTOS(int n) const;

        // obtaining icmp host unreachable in convoluted way
        bool Connect(const sockaddr_in6& addr);
        void SendEmptyPacket();
        bool IsHostUnreachable();
    };
}