1/* 2 * Copyright 2011 The WebRTC Project Authors. All rights reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11#ifndef WEBRTC_BASE_IPADDRESS_H_ 12#define WEBRTC_BASE_IPADDRESS_H_ 13 14#if defined(WEBRTC_POSIX) 15#include <netinet/in.h> 16#include <sys/socket.h> 17#include <arpa/inet.h> 18#include <netdb.h> 19#endif 20#if defined(WEBRTC_WIN) 21#include <winsock2.h> 22#include <ws2tcpip.h> 23#endif 24#include <string.h> 25#include <string> 26#include <vector> 27 28#include "webrtc/base/basictypes.h" 29#include "webrtc/base/byteorder.h" 30#if defined(WEBRTC_WIN) 31#include "webrtc/base/win32.h" 32#endif 33 34namespace rtc { 35 36enum IPv6AddressFlag { 37 IPV6_ADDRESS_FLAG_NONE = 0x00, 38 39 // Temporary address is dynamic by nature and will not carry MAC 40 // address. 41 IPV6_ADDRESS_FLAG_TEMPORARY = 1 << 0, 42 43 // Temporary address could become deprecated once the preferred 44 // lifetime is reached. It is still valid but just shouldn't be used 45 // to create new connection. 46 IPV6_ADDRESS_FLAG_DEPRECATED = 1 << 1, 47}; 48 49// Version-agnostic IP address class, wraps a union of in_addr and in6_addr. 50class IPAddress { 51 public: 52 IPAddress() : family_(AF_UNSPEC) { 53 ::memset(&u_, 0, sizeof(u_)); 54 } 55 56 explicit IPAddress(const in_addr& ip4) : family_(AF_INET) { 57 memset(&u_, 0, sizeof(u_)); 58 u_.ip4 = ip4; 59 } 60 61 explicit IPAddress(const in6_addr& ip6) : family_(AF_INET6) { 62 u_.ip6 = ip6; 63 } 64 65 explicit IPAddress(uint32 ip_in_host_byte_order) : family_(AF_INET) { 66 memset(&u_, 0, sizeof(u_)); 67 u_.ip4.s_addr = HostToNetwork32(ip_in_host_byte_order); 68 } 69 70 IPAddress(const IPAddress& other) : family_(other.family_) { 71 ::memcpy(&u_, &other.u_, sizeof(u_)); 72 } 73 74 virtual ~IPAddress() {} 75 76 const IPAddress & operator=(const IPAddress& other) { 77 family_ = other.family_; 78 ::memcpy(&u_, &other.u_, sizeof(u_)); 79 return *this; 80 } 81 82 bool operator==(const IPAddress& other) const; 83 bool operator!=(const IPAddress& other) const; 84 bool operator <(const IPAddress& other) const; 85 bool operator >(const IPAddress& other) const; 86 friend std::ostream& operator<<(std::ostream& os, const IPAddress& addr); 87 88 int family() const { return family_; } 89 in_addr ipv4_address() const; 90 in6_addr ipv6_address() const; 91 92 // Returns the number of bytes needed to store the raw address. 93 size_t Size() const; 94 95 // Wraps inet_ntop. 96 std::string ToString() const; 97 98 // Same as ToString but anonymizes it by hiding the last part. 99 std::string ToSensitiveString() const; 100 101 // Returns an unmapped address from a possibly-mapped address. 102 // Returns the same address if this isn't a mapped address. 103 IPAddress Normalized() const; 104 105 // Returns this address as an IPv6 address. 106 // Maps v4 addresses (as ::ffff:a.b.c.d), returns v6 addresses unchanged. 107 IPAddress AsIPv6Address() const; 108 109 // For socketaddress' benefit. Returns the IP in host byte order. 110 uint32 v4AddressAsHostOrderInteger() const; 111 112 static void set_strip_sensitive(bool enable); 113 114 private: 115 int family_; 116 union { 117 in_addr ip4; 118 in6_addr ip6; 119 } u_; 120 121 static bool strip_sensitive_; 122}; 123 124// IP class which could represent IPv6 address flags which is only 125// meaningful in IPv6 case. 126class InterfaceAddress : public IPAddress { 127 public: 128 InterfaceAddress() : ipv6_flags_(IPV6_ADDRESS_FLAG_NONE) {} 129 130 InterfaceAddress(IPAddress ip) 131 : IPAddress(ip), ipv6_flags_(IPV6_ADDRESS_FLAG_NONE) {} 132 133 InterfaceAddress(IPAddress addr, int ipv6_flags) 134 : IPAddress(addr), ipv6_flags_(ipv6_flags) {} 135 136 InterfaceAddress(const in6_addr& ip6, int ipv6_flags) 137 : IPAddress(ip6), ipv6_flags_(ipv6_flags) {} 138 139 const InterfaceAddress & operator=(const InterfaceAddress& other); 140 141 bool operator==(const InterfaceAddress& other) const; 142 bool operator!=(const InterfaceAddress& other) const; 143 144 int ipv6_flags() const { return ipv6_flags_; } 145 friend std::ostream& operator<<(std::ostream& os, 146 const InterfaceAddress& addr); 147 148 private: 149 int ipv6_flags_; 150}; 151 152bool IPFromAddrInfo(struct addrinfo* info, IPAddress* out); 153bool IPFromString(const std::string& str, IPAddress* out); 154bool IPFromString(const std::string& str, int flags, 155 InterfaceAddress* out); 156bool IPIsAny(const IPAddress& ip); 157bool IPIsLoopback(const IPAddress& ip); 158bool IPIsPrivate(const IPAddress& ip); 159bool IPIsUnspec(const IPAddress& ip); 160size_t HashIP(const IPAddress& ip); 161 162// These are only really applicable for IPv6 addresses. 163bool IPIs6Bone(const IPAddress& ip); 164bool IPIs6To4(const IPAddress& ip); 165bool IPIsSiteLocal(const IPAddress& ip); 166bool IPIsTeredo(const IPAddress& ip); 167bool IPIsULA(const IPAddress& ip); 168bool IPIsV4Compatibility(const IPAddress& ip); 169bool IPIsV4Mapped(const IPAddress& ip); 170 171// Returns the precedence value for this IP as given in RFC3484. 172int IPAddressPrecedence(const IPAddress& ip); 173 174// Returns 'ip' truncated to be 'length' bits long. 175IPAddress TruncateIP(const IPAddress& ip, int length); 176 177// Returns the number of contiguously set bits, counting from the MSB in network 178// byte order, in this IPAddress. Bits after the first 0 encountered are not 179// counted. 180int CountIPMaskBits(IPAddress mask); 181 182} // namespace rtc 183 184#endif // WEBRTC_BASE_IPADDRESS_H_ 185