ntopng/scripts/lua/system_setup_ui/dns.lua

--
-- (C) 2013-20 - ntop.org
--

local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path

local system_setup_ui_utils = require "system_setup_ui_utils"
local json = require("dkjson")
require "lua_utils"
require "prefs_utils"
prefsSkipRedis(true)

local is_nedge = ntop.isnEdge()
local is_appliance = ntop.isAppliance()

if not (is_nedge or is_appliance) or not isAdministrator() then
   return
end

local sys_config
if is_nedge then
   package.path = dirs.installdir .. "/pro/scripts/lua/nedge/modules/system_config/?.lua;" .. package.path
   sys_config = require("nf_config"):create(true)
else -- ntop.isAppliance()
   package.path = dirs.installdir .. "/scripts/lua/modules/system_config/?.lua;" .. package.path
   sys_config = require("appliance_config"):create(true)
end

system_setup_ui_utils.process_apply_discard_config(sys_config)

local function findDnsPreset(preset_name)
  for _, preset in pairs(DNS_PRESETS) do
    if preset.id == preset_name then
      return preset
    end
  end

  return nil
end

if table.len(_POST) > 0 then
  local dns_config = sys_config:getDnsConfig()
  local changed = false

  if _POST["global_dns_preset"] ~= nil then
    local primary_dns = _POST["global_primary_dns"]
    local secondary_dns = _POST["global_secondary_dns"]
    local preset = findDnsPreset(_POST["global_dns_preset"])

    if preset ~= nil then
      primary_dns = preset.primary_dns
      secondary_dns = preset.secondary_dns
    end

    if not isEmptyString(primary_dns) and (secondary_dns ~= nil) then
      dns_config.global_preset = ternary(preset ~= nil, _POST["global_dns_preset"], "custom")
      dns_config.global = primary_dns
      dns_config.secondary = secondary_dns
      changed = true
    end
  end

  if is_nedge then
    if _POST["child_dns_preset"] ~= nil then
      local child_dns = _POST["child_primary_dns"]
      local preset = findDnsPreset(_POST["child_dns_preset"])

      if preset ~= nil then
        child_dns = preset.primary_dns
      end

      if not isEmptyString(child_dns) then
        dns_config.child_preset = ternary(preset ~= nil, _POST["child_dns_preset"], "custom")
        dns_config.child_safe = child_dns
        changed = true
      end
    end
  end

  if _POST["forge_global_dns"] ~= nil then
    dns_config.forge_global = _POST["forge_global_dns"] == "1"
    changed = true
  end

  if changed then
    sys_config:setDnsConfig(dns_config)
    sys_config:save()
  end
end

local function getDnsPresets(childsafe)
  -- Init dns dropdown
  local dns_keys = {}
  local dns_values = {}

  for _, dns in ipairs(DNS_PRESETS) do
    if (childsafe and dns.child_safe) or (not childsafe and not dns.child_safe) then
      dns_keys[#dns_keys + 1] = dns.id
      dns_values[#dns_values + 1] = dns.label
    end
  end

  dns_keys[#dns_keys + 1] = "custom"
  dns_values[#dns_values + 1] = i18n("nedge.custom")

  return dns_keys, dns_values
end

local print_page_body = function()
  local dns_config = sys_config:getDnsConfig()

  -- Global DNS
  local dns_forging = sys_config:isGlobalDnsForgingEnabled()

  printPageSection(i18n("nedge.global_dns"))
  prefsToggleButton(subpage_active, {
    title = i18n("nedge.enforce_global_dns"),
    description = i18n("nedge.enforce_global_dns_description"),
    content = "",
    field = "forge_global_dns",
    pref = "",
    redis_prefix = "",
    default = ternary(dns_forging, "1", "0"),
    to_switch = nil,
  })

  local dns_keys, dns_values = getDnsPresets(false)
  prefsDropdownFieldPrefs(i18n("nedge.dns_server_preset"), i18n("nedge.dns_server_preset_descr"), "global_dns_preset", dns_values, dns_config.global_preset or "custom", true, {keys=dns_keys})

  prefsInputFieldPrefs(i18n("prefs.primary_dns"), i18n("nedge.the_primary_dns_server"),
          "", "global_primary_dns", dns_config.global or "0.0.0.0", nil, true, nil, nil,
          {required=true, pattern=getIPv4Pattern()})

  prefsInputFieldPrefs(i18n("prefs.secondary_dns"), i18n("nedge.the_secondary_dns_server"),
          "", "global_secondary_dns", dns_config.secondary or "0.0.0.0", nil, true, nil, nil,
          {required=false, pattern=getIPv4Pattern()})

  if is_nedge then
    -- Child Safe DNS
    local dns_keys, dns_values = getDnsPresets(true)

    printPageSection(i18n("prefs.safe_search_dns_title"))
    prefsDropdownFieldPrefs(i18n("nedge.dns_server_preset"), i18n("nedge.dns_server_preset_descr"), "child_dns_preset", dns_values, dns_config.child_preset or "custom", true, {keys=dns_keys})

    prefsInputFieldPrefs(i18n("prefs.safe_search_dns_title"), i18n("nedge.the_primary_dns_server"),
            "", "child_primary_dns", dns_config.child_safe or "0.0.0.0", nil, true, nil, nil,
            {required=true, pattern=getIPv4Pattern()})
  end

  printSaveButton()

  print[[
    <script>
      var dns_presets = ]] print(json.encode(DNS_PRESETS)) print[[;

      function handlePreset(key) {
        var selector = $("#" + key + "_dns_preset");
        var checkDns = function() {
          var selection = $(selector).val();

          for (var i=0; i<dns_presets.length; i++) {
            if (dns_presets[i].id == selection) {
              $("#" + key + "_primary_dns")
                .attr("disabled", "disabled")
                .val(dns_presets[i].primary_dns);

              $("#" + key + "_secondary_dns")
                .attr("disabled", "disabled")
                .val(dns_presets[i].secondary_dns);
              return;
            }
          }

          // not found, assuming custom
          $("#" + key + "_primary_dns").removeAttr("disabled");
          $("#" + key + "_secondary_dns").removeAttr("disabled");
        }

        selector.change(checkDns);
        checkDns();
      }

      handlePreset("id_input_global");
      handlePreset("id_input_child");
    </script>
  ]]
end

system_setup_ui_utils.print_setup_page(print_page_body, sys_config)