/*  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