/*

 * This file is part of PowerDNS or dnsdist.

 * Copyright -- PowerDNS.COM B.V. and its contributors

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of version 2 of the GNU General Public License as

 * published by the Free Software Foundation.

 *

 * In addition, for the avoidance of any doubt, permission is granted to

 * link this program with OpenSSL and to (re)distribute the binaries

 * produced as the result of such linking.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#include "svc-records.hh"

#include "misc.hh"

#include "base64.hh"

const std::map<std::string, SvcParam::SvcParamKey> SvcParam::SvcParams = {

  {"mandatory", SvcParam::SvcParamKey::mandatory},

  {"alpn", SvcParam::SvcParamKey::alpn},

  {"no-default-alpn", SvcParam::SvcParamKey::no_default_alpn},

  {"port", SvcParam::SvcParamKey::port},

  {"ipv4hint", SvcParam::SvcParamKey::ipv4hint},

  {"ech", SvcParam::SvcParamKey::ech},

  {"ipv6hint", SvcParam::SvcParamKey::ipv6hint},

  {"dohpath", SvcParam::SvcParamKey::dohpath},

  {"ohttp", SvcParam::SvcParamKey::ohttp},

  {"tls-supported-groups", SvcParam::SvcParamKey::tls_supported_groups},

};

SvcParam::SvcParamKey SvcParam::keyFromString(const std::string& k) {

  bool ignored;

  return SvcParam::keyFromString(k, ignored);

}

SvcParam::SvcParamKey SvcParam::keyFromString(const std::string& k, bool &generic) {

  auto it = SvcParams.find(k);

  if (it != SvcParams.end()) {

    generic = false;

    return it->second;

  }

  if (k.substr(0, 3) == "key") {

    try {

      generic = true;

      return SvcParam::SvcParamKey(pdns::checked_stoi<uint16_t>(k.substr(3)));

    }

    catch (...) {

    }

  }

  throw std::invalid_argument("SvcParam '" + k + "' is not recognized or in keyNNNN format");

}

std::string SvcParam::keyToString(const SvcParam::SvcParamKey& k) {

  auto ret = std::find_if(SvcParams.begin(), SvcParams.end(), [&](const std::pair<std::string, SvcParam::SvcParamKey>& e) { return e.second == k; });

  if (ret != SvcParams.end()) {

    return ret->first;

  }

  return "key" + std::to_string(k);

}

SvcParam::SvcParam(const SvcParamKey &key) {

  d_key = key;

  if (d_key != SvcParamKey::no_default_alpn && d_key != SvcParamKey::ohttp) {

    throw std::invalid_argument("can not create non-empty SvcParam for key '" + keyToString(key) + "'");

  }

}

SvcParam::SvcParam(const SvcParamKey &key, const std::string &value) {

  d_key = key;

  if (d_key != SvcParamKey::ech && d_key != SvcParamKey::dohpath && d_key < 10) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with a string value");

  }

  if (d_key == SvcParamKey::ech) {

    std::string d;

    // TODO check Base64 decode

    d_ech = value;

    return;

  }

  d_value = value;

}

SvcParam::SvcParam(const SvcParamKey &key, std::vector<std::string> &&value) {

  d_key = key;

  if (d_key != SvcParamKey::alpn) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with a string-set value");

  }

  if (d_key == SvcParamKey::alpn) {

    d_alpn = std::move(value);

  }

}

SvcParam::SvcParam(const SvcParamKey &key, std::set<std::string> &&value) {

  d_key = key;

  if (d_key != SvcParamKey::mandatory) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with a string-set value");

  }

  if (d_key == SvcParamKey::mandatory) {

    for (auto const &v: value) {

      d_mandatory.insert(keyFromString(v));

    }

  }

}

SvcParam::SvcParam(const SvcParamKey &key, std::set<SvcParam::SvcParamKey> &&value) {

  d_key = key;

  if (d_key != SvcParamKey::mandatory) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with a SvcParamKey-set value");

  }

  d_mandatory = std::move(value);

}

SvcParam::SvcParam(const SvcParamKey &key, std::vector<ComboAddress> &&value) {

  d_key = key;

  if (d_key != SvcParamKey::ipv6hint && d_key != SvcParamKey::ipv4hint) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with an IP address value");

  }

  for (auto const &addr : value) {

    if (d_key == SvcParam::ipv6hint && !addr.isIPv6()) {

      throw std::invalid_argument("non-IPv6 address ('" + addr.toString() + "') passed for " + keyToString(key));

    }

    if (d_key == SvcParam::ipv4hint && !addr.isIPv4()) {

      throw std::invalid_argument("non-IPv4 address ('" + addr.toString() + "') passed for " + keyToString(key));

    }

  }

  d_ipHints = std::move(value);

}

SvcParam::SvcParam(const SvcParamKey &key, std::vector<uint16_t> &&value) : d_key(key) {

  if (d_key != SvcParamKey::tls_supported_groups) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with a uint16 value-vector");

  }

  d_tls_supported_groups = std::move(value);

}

SvcParam::SvcParam(const SvcParamKey &key, const uint16_t value) {

  d_key = key;

  if (d_key != SvcParamKey::port) {

    throw std::invalid_argument("can not create SvcParam for " + keyToString(key) + " with an port value");

  }

  d_port = value;

}

//! This ensures an std::set<SvcParam> will be sorted by key (section 2.2 mandates this for wire format)

bool SvcParam::operator<(const SvcParam& other) const {

  return this->d_key < other.getKey();

}

const std::vector<ComboAddress>& SvcParam::getIPHints() const {

  if (d_key != SvcParamKey::ipv6hint && d_key != SvcParamKey::ipv4hint) {

    throw std::invalid_argument("getIPHints called for non-IP address key '" + keyToString(d_key) + "'");

  }

  return d_ipHints;

}

uint16_t SvcParam::getPort() const {

  if (d_key != SvcParam::port) {

    throw std::invalid_argument("getPort called for non-port key '" + keyToString(d_key) + "'");

  }

  return d_port;

}

const std::vector<std::string>& SvcParam::getALPN() const {

  if (d_key != SvcParam::alpn) {

    throw std::invalid_argument("getALPN called for non-alpn key '" + keyToString(d_key) + "'");

  }

  return d_alpn;

}

const std::set<SvcParam::SvcParamKey>& SvcParam::getMandatory() const {

  if (d_key != SvcParam::mandatory) {

    throw std::invalid_argument("getMandatory called for non-mandatory key '" + keyToString(d_key) + "'");

  }

  return d_mandatory;

}

const std::string& SvcParam::getECH() const {

  if (d_key != SvcParam::ech) {

    throw std::invalid_argument("getECH called for non-ech key '" + keyToString(d_key) + "'");

  }

  return d_ech;

}

const std::string& SvcParam::getValue() const {

  if (d_key != SvcParamKey::dohpath && d_key < 10) {

    throw std::invalid_argument("getValue called for non-single value key '" + keyToString(d_key) + "'");

  }

  return d_value;

}

const std::vector<uint16_t>& SvcParam::getTLSSupportedGroups() const {

  if (d_key != SvcParam::tls_supported_groups) {

    throw std::invalid_argument("getTLSSupportedGroups called for non-tls-supported-groups key '" + keyToString(d_key) + "'");

  }

  return d_tls_supported_groups;

}