1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
--
-- (C) 2019-22 - ntop.org
--
local file_utils = {}
local os_utils = require("os_utils")
local json = require "dkjson"
-- ##############################################
function file_utils.read_file(file_path)
local content = nil
local file = io.open(file_path, "r")
if file ~= nil then
content = file:read("*all")
file:close()
end
return content
end
-- ##############################################
function file_utils.read_json_file(file_path)
local json_content = nil
local content = file_utils.read_file(file_path)
if content ~= nil then
json_content = json.decode(content)
end
return json_content
end
-- ##############################################
function file_utils.copy_file(fname, src_path, dst_path)
local src
local dst
if(fname == nil) then
src = os_utils.fixPath(src_path)
dst = os_utils.fixPath(dst_path)
else
src = os_utils.fixPath(src_path .. "/" .. fname)
dst = os_utils.fixPath(dst_path .. "/" .. fname)
end
local infile, err = io.open(src, "rb")
-- NOTE: Do not forget the 'b' flag [https://www.lua.org/pil/21.2.2.html]
-- as this is compulsory on windows
traceError(TRACE_INFO, TRACE_CONSOLE, string.format("Copying file %s -> %s", src, dst))
if(do_trace) then
io.write(string.format("\tLoad [%s]\n", fname))
end
if(ntop.exists(dst)) then
-- NOTE: overwriting is not allowed as it means that a file was already provided by
-- another plugin
-- io.write(debug.traceback())
traceError(TRACE_ERROR, TRACE_CONSOLE, string.format("Trying to overwrite existing file %s", dst))
return(false)
end
if(infile == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, string.format("Could not open file %s for read: %s", src, err or ""))
return(false)
end
local instr = infile:read("*all")
infile:close()
local outfile, err = io.open(dst, "wb")
if(outfile == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, string.format("Could not open file %s for write", dst, err or ""))
return(false)
end
if (instr == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, string.format("Could not write the string into file %s", dst))
return (false)
end
outfile:write(instr)
outfile:close()
ntop.setDefaultFilePermissions(dst)
return(true)
end
-- #########################################################
function file_utils.recursive_copy(src_path, dst_path, path_map, required_extension)
for fname in pairs(ntop.readdir(src_path)) do
local do_copy = true
if((required_extension ~= nil) and not(ends(fname, ".lua"))) then
-- Don't copy
do_copy = false
traceError(TRACE_INFO, TRACE_CONSOLE, "SKIP file_utils.recursive_copy("..fname.." ["..src_path.." -> "..dst_path.."])\n")
end
if(do_copy) then
traceError(TRACE_INFO, TRACE_CONSOLE, "COPY file_utils.recursive_copy("..fname.." ["..src_path.." -> "..dst_path.."])\n")
if not file_utils.copy_file(fname, src_path, dst_path) then
return(false)
end
if path_map then
path_map[os_utils.fixPath(dst_path .. "/" .. fname)] = os_utils.fixPath(src_path .. "/" .. fname)
end
end
end
return(true)
end
-- #########################################################
return(file_utils)
|
