src/network/address.h

   1 /*
   2 Minetest
   3 Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
   4
   5 This program is free software; you can redistribute it and/or modify
   6 it under the terms of the GNU Lesser General Public License as published by
   7 the Free Software Foundation; either version 2.1 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU Lesser General Public License for more details.
  14
  15 You should have received a copy of the GNU Lesser General Public License along
  16 with this program; if not, write to the Free Software Foundation, Inc.,
  17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #pragma once
  21
  22 #ifdef _WIN32
  23 #ifndef _WIN32_WINNT
  24 #define _WIN32_WINNT 0x0501
  25 #endif
  26 #include <windows.h>
  27 #include <winsock2.h>
  28 #include <ws2tcpip.h>
  29 #else
  30 #include <netinet/in.h>
  31 #include <sys/socket.h>
  32 #endif
  33
  34 #include <ostream>
  35 #include <cstring>
  36 #include "irrlichttypes.h"
  37 #include "networkexceptions.h"
  38
  39 struct IPv6AddressBytes
  40 {
  41         u8 bytes[16];
  42         IPv6AddressBytes() { memset(bytes, 0, 16); }
  43 };
  44
  45 class Address
  46 {
  47 public:
  48         Address();
  49         Address(u32 address, u16 port);
  50         Address(u8 a, u8 b, u8 c, u8 d, u16 port);
  51         Address(const IPv6AddressBytes *ipv6_bytes, u16 port);
  52
  53         bool operator==(const Address &address);
  54         bool operator!=(const Address &address) { return !(*this == address); }
  55
  56         struct in_addr getAddress() const;
  57         struct in6_addr getAddress6() const;
  58         u16 getPort() const;
  59         int getFamily() const { return m_addr_family; }
  60         bool isIPv6() const { return m_addr_family == AF_INET6; }
  61         bool isZero() const;
  62         void print(std::ostream &s) const;
  63         std::string serializeString() const;
  64         bool isLocalhost() const;
  65
  66         // Resolve() may throw ResolveError (address is unchanged in this case)
  67         void Resolve(const char *name);
  68
  69         void setAddress(u32 address);
  70         void setAddress(u8 a, u8 b, u8 c, u8 d);
  71         void setAddress(const IPv6AddressBytes *ipv6_bytes);
  72         void setPort(u16 port);
  73
  74 private:
  75         unsigned short m_addr_family = 0;
  76         union
  77         {
  78                 struct in_addr ipv4;
  79                 struct in6_addr ipv6;
  80         } m_address;
  81         u16 m_port = 0; // Port is separate from sockaddr structures
  82 };