Move endpoint to download extracted pcap data

This commit is contained in:
Alfredo Cardigliano 2024-03-08 10:32:03 +01:00
parent 9549c69581
commit 236816677e

View file

@ -0,0 +1,50 @@
--
-- (C) 2013-24 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
local json = require "dkjson"
local recording_utils = require "recording_utils"
--
-- Download data produced by an extraction by job ID
-- Example: curl -u admin:admin http://localhost:3000/lua/rest/v2/get/pcap/extraction/data.lua?job_id=1&file_id=1
--
local function send_error(error_type)
local msg = ""
if error_type == "not_found" then
msg = i18n("traffic_recording.not_found")
elseif error_type == "not_granted" then
msg = i18n("traffic_recording.error_not_granted")
end
sendHTTPContentTypeHeader('application/json')
print(json.encode({error = msg}))
end
if not recording_utils.isAvailable() then
send_error("not_granted")
elseif isEmptyString(_GET["job_id"]) then
send_error("not_found")
else
local job_id = tonumber(_GET["job_id"])
local file_id = 1
if _GET["file_id"] ~= nil then
file_id = tonumber(_GET["file_id"])
end
local job_files = recording_utils.getJobFiles(job_id)
if job_files[file_id] == nil then
send_error("not_found")
else
local file = job_files[file_id]
sendHTTPContentTypeHeader('application/vnd.tcpdump.pcap', 'attachment; filename="extraction_'..job_id..'_'..file_id..'.pcap"')
ntop.dumpBinaryFile(file)
end
end