ipyml/inet_address.h

#ifndef _IPYML_INET_ADDRESS_H
#define _IPYML_INET_ADDRESS_H

#include "yaml.h"

#include <arpa/inet.h>     // inet_ntop, socklen_t
#include <bits/socket.h>   // AF_*, socklen_t
#include <cassert>         // assert
#include <cstring>         // memcpy
#include <libmnl/libmnl.h> // mnl_attr_*
#include <linux/in.h>      // struct in_addr_t
#include <linux/netlink.h> // struct nlattr
#include <netinet/in.h>    // INET_ADDRSTRLEN
#include <ostream>         // std::ostream

/*
class may not have any virtual member due to:
> error: union member ‘Address::<anonymous union>::inet_addr’
> with non-trivial ‘constexpr InetAddress::InetAddress(InetAddress&&)’
*/

class InetAddress { // : public YamlObject {
  unsigned char bytes[sizeof(in_addr_t)];

public:
  InetAddress &operator=(const nlattr *attr) {
    assert(mnl_attr_validate(attr, MNL_TYPE_BINARY) == 0);
    assert(mnl_attr_get_payload_len(attr) == sizeof(bytes));
    memcpy(this->bytes, mnl_attr_get_payload(attr), sizeof(bytes));
    return *this;
  }

  void format(char *dst, socklen_t size) const {
    inet_ntop(AF_INET, bytes, dst, size);
  }

  void write_yaml(std::ostream &stream,
                  const yaml_indent_level_t indent_level = 0) const {
    char str[INET_ADDRSTRLEN];
    format(str, sizeof(str));
    stream << str << '\n';
  }
};

#endif