network-monitor-impl-rtnl.hpp

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
#ifndef NDN_UTIL_NETWORK_MONITOR_IMPL_RTNL_HPP
#define NDN_UTIL_NETWORK_MONITOR_IMPL_RTNL_HPP

#include "ndn-cxx-config.hpp"
#include "../network-monitor.hpp"

#ifndef NDN_CXX_HAVE_RTNETLINK
#error "This file should not be compiled ..."
#endif

#include <boost/asio/posix/stream_descriptor.hpp>

#include <array>
#include <map>

#include <linux/netlink.h>
#include <linux/rtnetlink.h>
#include <linux/if_addr.h>
#include <linux/if_link.h>

namespace ndn {
namespace util {

class NetworkMonitor::Impl
{
public:
  Impl(NetworkMonitor& nm, boost::asio::io_service& io);

  ~Impl();

  uint32_t
  getCapabilities() const
  {
    return NetworkMonitor::CAP_ENUM |
           NetworkMonitor::CAP_IF_ADD_REMOVE |
           NetworkMonitor::CAP_STATE_CHANGE |
           NetworkMonitor::CAP_MTU_CHANGE |
           NetworkMonitor::CAP_ADDR_ADD_REMOVE;
  }

  shared_ptr<NetworkInterface>
  getNetworkInterface(const std::string& ifname) const;

  std::vector<shared_ptr<NetworkInterface>>
  listNetworkInterfaces() const;

private:
  struct RtnlRequest
  {
    nlmsghdr nlh;
    ifinfomsg ifi;
    rtattr rta __attribute__((aligned(NLMSG_ALIGNTO))); // rtattr has to be aligned
    uint32_t rtext;                                     // space for IFLA_EXT_MASK
  };

  bool
  isEnumerating() const;

  void
  initSocket();

  void
  sendDumpRequest(uint16_t nlmsgType);

  void
  asyncRead();

  void
  handleRead(const boost::system::error_code& error, size_t nBytesReceived,
             const shared_ptr<boost::asio::posix::stream_descriptor>& socket);

  void
  parseNetlinkMessage(const nlmsghdr* nlh, size_t len);

  void
  parseLinkMessage(const nlmsghdr* nlh, const ifinfomsg* ifi);

  void
  parseAddressMessage(const nlmsghdr* nlh, const ifaddrmsg* ifa);

  void
  parseRouteMessage(const nlmsghdr* nlh, const rtmsg* rtm);

  static void
  updateInterfaceState(NetworkInterface& interface, uint8_t operState);

private:
  NetworkMonitor& m_nm;
  std::map<int /*ifindex*/, shared_ptr<NetworkInterface>> m_interfaces; 
  std::array<uint8_t, 16384> m_buffer; 
  shared_ptr<boost::asio::posix::stream_descriptor> m_socket; 
  uint32_t m_pid; 
  uint32_t m_sequenceNo; 
  bool m_isEnumeratingLinks; 
  bool m_isEnumeratingAddresses; 
};

} // namespace util
} // namespace ndn

#endif // NDN_UTIL_NETWORK_MONITOR_IMPL_RTNL_HPP