InetAddr.h

Go to the documentation of this file.

#ifndef HYPERTABLE_INETADDR_H
#define HYPERTABLE_INETADDR_H

extern "C" {
#include <netinet/in.h>
}

#include "Common/String.h"

namespace Hypertable {

  struct Endpoint {
    Endpoint(const String &host, uint16_t port) : host(host), port(port) {}

    String host;
    uint16_t port;
  };

  std::ostream &operator<<(std::ostream &, const Endpoint &);

  struct InetAddr : sockaddr_in {
    InetAddr();
    InetAddr(const String &host, uint16_t port);
    InetAddr(const String &endpoint);
    InetAddr(uint32_t ip32, uint16_t port);
    InetAddr(const sockaddr_in &addr) { operator=(addr); }

    size_t encoded_length() const;
    void encode(uint8_t **bufp) const;
    void decode(const uint8_t **bufp, size_t *remainp);

    InetAddr &operator=(const sockaddr_in &addr) {
      if (this != &addr)
        memcpy(this, &addr, sizeof(sockaddr_in));

      return *this;
    }

    bool operator==(const InetAddr &other) const {
      return (bool)!memcmp(this, &other, sizeof(InetAddr));
    }

    bool operator!=(const InetAddr &other) const {
      return !(*this == other);
    }

    bool operator<(const InetAddr &other) const {
      if (sin_family != other.sin_family)
        return sin_family < other.sin_family;
      if (sin_addr.s_addr != other.sin_addr.s_addr)
        return sin_addr.s_addr < other.sin_addr.s_addr;
      return sin_port < other.sin_port;
    }

    String format(int sep = ':') { return InetAddr::format(*this, sep); }
    String format_ipaddress() { return InetAddr::format_ipaddress(*this); }
    String hex(int sep = ':') { return InetAddr::hex(*this, sep); }

    // convenient/legacy static methods
    static bool initialize(sockaddr_in *addr, const char *host, uint16_t port);

    static bool initialize(sockaddr_in *addr, const char *addr_str);

    static Endpoint parse_endpoint(const char *endpoint, int defport = 0);
    static Endpoint parse_endpoint(const String &endpoint, int defport = 0) {
      return parse_endpoint(endpoint.c_str(), defport);
    }

    static bool parse_ipv4(const char *ip, uint16_t port, sockaddr_in &addr,
                           int base = 0);

    static bool initialize(sockaddr_in *addr, uint32_t haddr, uint16_t port);

    static const char *string_format(String &addr_str, const sockaddr_in &addr);
    static String format(const sockaddr_in &addr, int sep = ':');
    static String format_ipaddress(const sockaddr_in &addr);
    static String hex(const sockaddr_in &addr, int sep = ':');
  };

  std::ostream &operator<<(std::ostream &, const sockaddr_in &);

} // namespace Hypertable

#endif // HYPERTABLE_INETADDR_H