// Copyright 2019 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "shill/routing_table_entry.h"

#include <linux/rtnetlink.h>

#include <string>

#include <base/strings/stringprintf.h>

namespace shill {

// static
const int RoutingTableEntry::kDefaultTag = -1;

RoutingTableEntry::RoutingTableEntry()
    : metric(0),
      scope(RT_SCOPE_UNIVERSE),
      table(RT_TABLE_MAIN),
      type(RTN_UNICAST),
      protocol(RTPROT_BOOT),
      tag(kDefaultTag) {}

// static
RoutingTableEntry RoutingTableEntry::Create(IPAddress::Family family) {
  return Create(IPAddress(family), IPAddress(family), IPAddress(family));
}

// static
RoutingTableEntry RoutingTableEntry::Create(const IPAddress& dst_in,
                                            const IPAddress& src_in,
                                            const IPAddress& gateway_in) {
  RoutingTableEntry entry;
  entry.dst = dst_in;
  entry.src = src_in;
  entry.gateway = gateway_in;
  return entry;
}

RoutingTableEntry& RoutingTableEntry::SetMetric(uint32_t metric_in) {
  metric = metric_in;
  return *this;
}

RoutingTableEntry& RoutingTableEntry::SetScope(unsigned char scope_in) {
  scope = scope_in;
  return *this;
}

RoutingTableEntry& RoutingTableEntry::SetTable(uint32_t table_in) {
  table = table_in;
  return *this;
}

RoutingTableEntry& RoutingTableEntry::SetType(unsigned char type_in) {
  type = type_in;
  return *this;
}

RoutingTableEntry& RoutingTableEntry::SetTag(uint8_t tag_in) {
  tag = tag_in;
  return *this;
}

// clang-format off
bool RoutingTableEntry::operator==(const RoutingTableEntry& b) const {
  return (dst == b.dst &&
          src == b.src &&
          gateway == b.gateway &&
          metric == b.metric &&
          scope == b.scope &&
          table == b.table &&
          type == b.type &&
          tag == b.tag);
}
// clang-format on

std::ostream& operator<<(std::ostream& os, const RoutingTableEntry& entry) {
  std::string dest_address =
      entry.dst.IsDefault() ? "default" : entry.dst.ToString();
  const char* dest_prefix;
  switch (entry.type) {
    case RTN_LOCAL:
      dest_prefix = "local ";
      break;
    case RTN_BROADCAST:
      dest_prefix = "broadcast ";
      break;
    case RTN_BLACKHOLE:
      dest_prefix = "blackhole";
      dest_address = "";  // Don't print the address.
      break;
    case RTN_UNREACHABLE:
      dest_prefix = "unreachable";
      dest_address = "";  // Don't print the address.
      break;
    default:
      dest_prefix = "";
      break;
  }
  std::string gateway;
  if (!entry.gateway.IsDefault()) {
    gateway = " via " + entry.gateway.ToString();
  }
  std::string src;
  if (!entry.src.IsDefault()) {
    src = " src " + entry.src.ToString();
  }

  os << base::StringPrintf(
      "%s%s%s metric %d %s table %d%s", dest_prefix, dest_address.c_str(),
      gateway.c_str(), entry.metric,
      IPAddress::GetAddressFamilyName(entry.dst.family()).c_str(),
      static_cast<int>(entry.table), src.c_str());
  return os;
}

}  // namespace shill