configuration backup (tar download)

scripts/lua/get_config.lua

@@ -0,0 +1,112 @@
+--
+-- (C) 2013-18 - ntop.org
+--
+
+local prefs = ntop.getPrefs()
+local dirs = ntop.getDirs()
+
+package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
+
+local os_utils = require "os_utils"
+
+require "lua_utils"
+
+local json = require("dkjson")
+
+local function send_error(error_type)
+   local msg = i18n("error")
+   if error_type == "not_granted" then
+     msg = i18n("conf_backup.not_granted")
+   elseif error_type == "tar_not_found" then
+     msg = i18n("conf_backup.tar_not_counf")
+   end
+
+   sendHTTPContentTypeHeader('application/json')
+   print(json.encode({error = msg}))
+end
+
+user_group = ntop.getUserGroup()
+if user_group ~= "administrator" then
+  send_error("not_granted")
+else
+
+  if ntop.isWindows() then
+
+    sendHTTPContentTypeHeader('application/json', 'attachment; filename="runtimeprefs.json"')
+
+    local runtimeprefs_path = os_utils.fixPath(dirs.workingdir.."/runtimeprefs.json")
+    local runtimeprefs = io.open(runtimeprefs_path, "r")
+    
+    print(runtimeprefs:read "*a") 
+
+  else -- Unix
+
+    -- -- Old way (single json)
+    -- sendHTTPContentTypeHeader('application/json', 'attachment; filename="runtimeprefs.json"')
+    -- local config_backup = {}
+    -- if not isEmptyString(prefs.config_file) then
+    --   local ntopng_conf = io.open(prefs.config_file, "r")  
+    --   if ntopng_conf ~= nil then
+    --     config_backup[prefs.config_file] = (ntopng_conf:read "*a")
+    --   end
+    -- end
+    -- print(json.encode(config_backup or {}))
+
+    local bin_tar = "/bin/tar"
+    if not ntop.exists(bin_tar) then
+      bin_tar = "/usr/bin/tar"
+    end
+    if not ntop.exists(bin_tar) then
+      send_error("tar_not_found")
+    else
+
+      local config_files = "/etc/ntopng"
+
+      local license_path = "/etc/ntopng.license"
+      if ntop.isnEdge() then
+        license_path = "/etc/nedge.license"
+      end
+      if ntop.exists(license_path) then
+        config_files = config_files .. " " .. license_path
+      end
+
+      if not isEmptyString(prefs.config_file) then
+        if ntop.exists(prefs.config_file) then
+          config_files = config_files .. " " .. prefs.config_file
+        end
+      end
+
+      local runtimeprefs_path = os_utils.fixPath(dirs.workingdir.."/runtimeprefs.json")
+      if ntop.exists(runtimeprefs_path) then
+        config_files = config_files .. " " .. runtimeprefs_path
+      end
+
+      if ntop.isnEdge() then
+        local system_config_path = os_utils.fixPath(dirs.workingdir.."/system.config")
+        if ntop.exists(system_config_path) then
+          config_files = config_files .. " " .. system_config_path
+        end
+      end
+
+      if not isEmptyString(prefs.ndpi_proto_file) then
+        if ntop.exists(prefs.ndpi_proto_file) then
+          config_files = config_files .. " " .. prefs.ndpi_proto_file
+        end
+      end
+
+      sendHTTPContentTypeHeader('application/x-tar-gz', 'attachment; filename="ntopng_conf_backup.tar.gz"')
+
+      local cmd = bin_tar .. " czP " .. config_files 
+
+      -- Note: we are using os.execute / ntop.dumpBinaryFile as io.popen / print 
+      -- cannot be used for dumping binary files directly to the connection
+
+      tmp = os.tmpname()
+      os.execute(cmd .. " > " .. tmp)
+      ntop.dumpBinaryFile(tmp)
+      os.remove(tmp)
+
+    end
+  end
+end
+