summaryrefslogtreecommitdiffhomepage
path: root/NetEx/NetAddr.h
blob: 1b69c256218bf4ebe226b67e45c3fbf6d713c6a4 (plain)
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
/*  Starshatter: The Open Source Project
    Copyright (c) 2021-2022, Starshatter: The Open Source Project Contributors
    Copyright (c) 2011-2012, Starshatter OpenSource Distribution Contributors
    Copyright (c) 1997-2006, Destroyer Studios LLC.

    AUTHOR:       John DiCamillo


    OVERVIEW
    ========
    Network Address (specifically, Internet Protocol)
*/

#ifndef NetAddr_h
#define NetAddr_h

#ifdef _WIN32
#include <winsock2.h>
#else
#include <arpa/inet.h>
#include <sys/socket.h>
#endif

#include <cstdint>


class NetAddr
{
public:
    static const char* TYPENAME() { return "NetAddr"; }

    NetAddr(const char* a, std::uint16_t p=0);
    NetAddr(std::uint32_t a=0, std::uint16_t p=0);
    NetAddr(const NetAddr& n);

    int operator == (const NetAddr& a) const { return addr==a.addr && port==a.port; }

    std::uint32_t  IPAddr() const { return addr; }
    std::uint8_t   B4()     const { return static_cast<std::uint8_t>((addr & 0xff000000) >> 24); }
    std::uint8_t   B3()     const { return static_cast<std::uint8_t>((addr & 0x00ff0000) >> 16); }
    std::uint8_t   B2()     const { return static_cast<std::uint8_t>((addr & 0x0000ff00) >>  8); }
    std::uint8_t   B1()     const { return static_cast<std::uint8_t>((addr & 0x000000ff)      ); }

    std::uint16_t  Port()   const    { return port; }
    void           SetPort(std::uint16_t p)   { port = p;    }

    sockaddr*      GetSockAddr()       const;
    size_t         GetSockAddrLength() const;

    void           SetSockAddr(sockaddr* s, int size);
    void           InitFromSockAddr();

private:
    void        Init();

    std::uint32_t addr; // IP addr in host byte order
    std::uint16_t port; // IP port in host byte order
    sockaddr_in sadr;
};


#endif  // NetAddr_h