X-Git-Url: http://plrg.eecs.uci.edu/git/?p=folly.git;a=blobdiff_plain;f=folly%2FIPAddressV4.h;h=061f53fc8a2523b5a0a1821e4f362e8969b69255;hp=26a2dba38acdbf0efd1ecc049dd8754f755bd787;hb=e96129da65d3ad2b20aae5a2bf2d22d2d72b8feb;hpb=ed8c80a0e0988e4ce687f51ca832a00e4a6b7930 diff --git a/folly/IPAddressV4.h b/folly/IPAddressV4.h index 26a2dba3..061f53fc 100644 --- a/folly/IPAddressV4.h +++ b/folly/IPAddressV4.h @@ -1,5 +1,5 @@ /* - * Copyright 2017 Facebook, Inc. + * Copyright 2014-present Facebook, Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -22,9 +22,12 @@ #include #include -#include +#include +#include +#include #include #include +#include namespace folly { @@ -54,8 +57,12 @@ typedef std::array ByteArray4; */ class IPAddressV4 { public: + // Max size of std::string returned by toFullyQualified. + static constexpr size_t kMaxToFullyQualifiedSize = + 4 /*words*/ * 3 /*max chars per word*/ + 3 /*separators*/; + // returns true iff the input string can be parsed as an ipv4-address - static bool validate(StringPiece ip); + static bool validate(StringPiece ip) noexcept; // create an IPAddressV4 instance from a uint32_t (network byte order) static IPAddressV4 fromLong(uint32_t src); @@ -66,19 +73,36 @@ class IPAddressV4 { * Create a new IPAddress instance from the provided binary data. * @throws IPAddressFormatException if the input length is not 4 bytes. */ - static IPAddressV4 fromBinary(ByteRange bytes) { - IPAddressV4 addr; - addr.setFromBinary(bytes); - return addr; - } + static IPAddressV4 fromBinary(ByteRange bytes); + + /** + * Non-throwing version of fromBinary(). + * On failure returns IPAddressFormatError. + */ + static Expected tryFromBinary( + ByteRange bytes) noexcept; + + /** + * Tries to create a new IPAddressV4 instance from provided string and + * returns it on success. Returns IPAddressFormatError on failure. + */ + static Expected tryFromString( + StringPiece str) noexcept; /** * Returns the address as a Range. */ ByteRange toBinary() const { - return ByteRange((const unsigned char *) &addr_.inAddr_.s_addr, 4); + return ByteRange((const unsigned char*)&addr_.inAddr_.s_addr, 4); } + /** + * Create a new IPAddress instance from the in-addr.arpa representation. + * @throws IPAddressFormatException if the input is not a valid in-addr.arpa + * representation + */ + static IPAddressV4 fromInverseArpaName(const std::string& arpaname); + /** * Convert a IPv4 address string to a long in network byte order. * @param [in] ip the address to convert @@ -101,10 +125,10 @@ class IPAddressV4 { explicit IPAddressV4(StringPiece ip); // ByteArray4 constructor - explicit IPAddressV4(const ByteArray4& src); + explicit IPAddressV4(const ByteArray4& src) noexcept; // in_addr constructor - explicit IPAddressV4(const in_addr src); + explicit IPAddressV4(const in_addr src) noexcept; // Return the V6 mapped representation of the address. IPAddressV6 createIPv6() const; @@ -129,7 +153,9 @@ class IPAddressV4 { * @see IPAddress#bitCount * @returns 32 */ - static size_t bitCount() { return 32; } + static constexpr size_t bitCount() { + return 32; + } /** * @See IPAddress#toJson @@ -183,8 +209,12 @@ class IPAddressV4 { // @see IPAddress#str std::string str() const; + std::string toInverseArpaName() const; + // return underlying in_addr structure - in_addr toAddr() const { return addr_.inAddr_; } + in_addr toAddr() const { + return addr_.inAddr_; + } sockaddr_in toSockAddr() const { sockaddr_in addr; @@ -201,10 +231,17 @@ class IPAddressV4 { } // @see IPAddress#toFullyQualified - std::string toFullyQualified() const { return str(); } + std::string toFullyQualified() const { + return str(); + } + + // @see IPAddress#toFullyQualifiedAppend + void toFullyQualifiedAppend(std::string& out) const; // @see IPAddress#version - uint8_t version() const { return 4; } + uint8_t version() const { + return 4; + } /** * Return the mask associated with the given number of bits. @@ -222,44 +259,48 @@ class IPAddressV4 { const CIDRNetworkV4& one, const CIDRNetworkV4& two); // Number of bytes in the address representation. - static size_t byteCount() { return 4; } - //get nth most significant bit - 0 indexed + static size_t byteCount() { + return 4; + } + // get nth most significant bit - 0 indexed bool getNthMSBit(size_t bitIndex) const { return detail::getNthMSBitImpl(*this, bitIndex, AF_INET); } - //get nth most significant byte - 0 indexed + // get nth most significant byte - 0 indexed uint8_t getNthMSByte(size_t byteIndex) const; - //get nth bit - 0 indexed + // get nth bit - 0 indexed bool getNthLSBit(size_t bitIndex) const { return getNthMSBit(bitCount() - bitIndex - 1); } - //get nth byte - 0 indexed + // get nth byte - 0 indexed uint8_t getNthLSByte(size_t byteIndex) const { return getNthMSByte(byteCount() - byteIndex - 1); } - const unsigned char* bytes() const { return addr_.bytes_.data(); } + const unsigned char* bytes() const { + return addr_.bytes_.data(); + } private: union AddressStorage { - static_assert(sizeof(in_addr) == sizeof(ByteArray4), - "size of in_addr and ByteArray4 are different"); + static_assert( + sizeof(in_addr) == sizeof(ByteArray4), + "size of in_addr and ByteArray4 are different"); in_addr inAddr_; ByteArray4 bytes_; AddressStorage() { std::memset(this, 0, sizeof(AddressStorage)); } - explicit AddressStorage(const ByteArray4 bytes): bytes_(bytes) {} - explicit AddressStorage(const in_addr addr): inAddr_(addr) {} + explicit AddressStorage(const ByteArray4 bytes) : bytes_(bytes) {} + explicit AddressStorage(const in_addr addr) : inAddr_(addr) {} } addr_; - static const std::array masks_; - /** * Set the current IPAddressV4 object to have the address specified by bytes. - * @throws IPAddressFormatException if bytes.size() is not 4. + * Returns IPAddressFormatError if bytes.size() is not 4. */ - void setFromBinary(ByteRange bytes); + Expected trySetFromBinary( + ByteRange bytes) noexcept; }; // boost::hash uses hash_value() so this allows boost::hash to work @@ -294,13 +335,13 @@ inline bool operator>=(const IPAddressV4& a, const IPAddressV4& b) { return !(a < b); } -} // folly +} // namespace folly namespace std { -template<> +template <> struct hash { size_t operator()(const folly::IPAddressV4 addr) const { return addr.hash(); } }; -} // std +} // namespace std