lokinet/llarp/net.hpp

#ifndef LLARP_NET_HPP
#define LLARP_NET_HPP
#include <llarp/net.h>

bool operator == (const sockaddr& a, const sockaddr& b)
{
  socklen_t sz = sizeof(a.sa_data);
  switch(a.sa_family)
  {
  case AF_INET:
    sz = sizeof(sockaddr_in);
    break;
  case AF_INET6:
    sz = sizeof(sockaddr_in6);
    break;
  case AF_PACKET:
    sz = sizeof(sockaddr_ll);
    break;
  default:
    break;
  }
  return a.sa_family == b.sa_family && memcmp(a.sa_data, b.sa_data, sz) == 0;
}

bool operator < (const sockaddr_in6 & a, const sockaddr_in6 & b)
{
  return memcmp(&a, &b, sizeof(sockaddr_in6)) < 0;
}

bool operator < (const in6_addr & a, const in6_addr b)
{
  return memcmp(&a, &b, sizeof(in6_addr)) < 0;
}

namespace llarp
{
  struct Addr
  {
    int af = AF_INET;
    in6_addr addr = {0};
    uint16_t port = 0;
    sockaddr saddr;

    Addr() {};
    
    Addr(const Addr & other) {
      af = other.af;
      port = other.port;
      memcpy(addr.s6_addr, other.addr.s6_addr, 16);
      memcpy(&saddr, &other.saddr, sizeof(sockaddr));
    }

    Addr(const llarp_ai & other) {
      af = AF_INET6;
      memcpy(addr.s6_addr, other.ip.s6_addr, 16);
      port = other.port;
      saddr.sa_family = af;
      memcpy(((sockaddr_in6 *)&saddr)->sin6_addr.s6_addr, addr.s6_addr, 16);
      ((sockaddr_in6 *)&saddr)->sin6_port = htons(port);
    }
    
    Addr(const sockaddr & other) {
      uint8_t * addrptr = addr.s6_addr;
      switch(other.sa_family)
      {
      case AF_INET:
        // SIIT
        af = AF_INET;
        memcpy(12+addrptr, &((const sockaddr_in*)(&other))->sin_addr, 4);
        addrptr[11] = 0xff;
        addrptr[10] = 0xff;
        port = ntohs(((sockaddr_in*)(&other))->sin_port);
        break;
      case AF_INET6:
        af = AF_INET6;
        memcpy(addrptr, &((const sockaddr_in6*)(&other))->sin6_addr, 16);
        port = ntohs(((sockaddr_in6*)(&other))->sin6_port);
        break;
        // TODO : sockaddr_ll
      default:
        break;
      }
      saddr.sa_family = af;
      memcpy(saddr.sa_data, other.sa_data, sizeof(saddr.sa_data));
    }

    operator const sockaddr * () const
    {
      return &saddr;
    }
    
    bool operator < (const Addr & other) const
    {
      return af < other.af && addr < other.addr && port < other.port;
    }
    
  };
}

#endif