summaryrefslogtreecommitdiffhomepage
path: root/NetEx/NetAddr.h
blob: 82f3420764e0be8d81910e2f33730cf8163f3462 (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
/*  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

#include <windows.h>
#include <winsock.h>

// +-------------------------------------------------------------------+

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

    NetAddr(const char* a, WORD p=0);
    NetAddr(DWORD a=0, WORD p=0);
    NetAddr(const NetAddr& n);

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

    DWORD          IPAddr() const { return addr; }
    BYTE           B4()     const { return (BYTE) ((addr & 0xff000000) >> 24); }
    BYTE           B3()     const { return (BYTE) ((addr & 0x00ff0000) >> 16); }
    BYTE           B2()     const { return (BYTE) ((addr & 0x0000ff00) >>  8); }
    BYTE           B1()     const { return (BYTE) ((addr & 0x000000ff)      ); }

    WORD           Port()   const    { return port; }
    void           SetPort(WORD p)   { port = p;    }

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

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

private:
    void        Init();

    DWORD       addr; // IP addr in host byte order
    WORD        port; // IP port in host byte order
    sockaddr_in sadr;
};


#endif  // NetAddr_h