ip_address.hpp

Go to the documentation of this file.

#pragma once

#include "hebi.h"

#include <iostream>

#ifdef WIN32

#define NOMINMAX
#include <ws2tcpip.h>
#include <stdlib.h>
#define bswap_32(x) _byteswap_ulong(x)
using in_addr_t = in_addr;

#elif defined(__APPLE__)

// Mac OS X / Darwin features
#include <libkern/OSByteOrder.h>
#define bswap_32(x) OSSwapInt32(x)
#include <arpa/inet.h>

#else // Linux

#include <arpa/inet.h>
#include <byteswap.h>

#endif

namespace hebi {

class IpAddress final {
public:
  IpAddress() = default;

  // Create IP Address from network order uint32_t
  IpAddress(uint32_t address_raw) : address_network_order_(address_raw) {}

  static IpAddress fromBytes(uint8_t a, uint8_t b, uint8_t c, uint8_t d) {
    // TODO: depends on byte order:
    uint32_t tmp = (a << 24) | (b << 16) | (c << 8) | (d);
    return IpAddress(tmp);
  }
  
  static IpAddress fromLittleEndian(uint32_t raw) {
    return IpAddress(bswap_32(raw));
  }

  bool setToString(const std::string& ip_str) {
    in_addr_t addr;
    if (inet_pton(AF_INET, ip_str.c_str(), &addr) != 1)
      return false;
  #ifdef WIN32
    address_network_order_ = addr.S_un.S_addr;
  #else
    address_network_order_ = addr;
  #endif
    return true;
  }

  std::string toString() const {
    std::string res;
    res.resize(INET_ADDRSTRLEN);
    inet_ntop(AF_INET, &address_network_order_, &res[0], INET_ADDRSTRLEN);
    return res;
  }

  uint32_t getLittleEndian() const {
    return bswap_32(address_network_order_);
  }

  uint32_t getBigEndian() const {
    return address_network_order_;
  }

private:
  uint32_t address_network_order_{};
};

} // namespace hebi