diff options
Diffstat (limited to 'remote/lib/support.lua')
| -rw-r--r-- | remote/lib/support.lua | 669 |
1 files changed, 669 insertions, 0 deletions
diff --git a/remote/lib/support.lua b/remote/lib/support.lua new file mode 100644 index 0000000..984d677 --- /dev/null +++ b/remote/lib/support.lua @@ -0,0 +1,669 @@ +require "luarocks.require" +require "md5" +require "lfs" + +home = os.getenv( "HOME" ) + +os.execute("mkdir -p " .. home .. "/.conky/requests"); +os.execute("mkdir -p " .. home .. "/.conky/info"); +c_timer_init = -1 + +do + function conky_eval(...) + return conky_parse(table.concat(arg, " ")) + end + + function conky_multiply_line(times, ...) + local str = table.concat(arg, " ") + local tbl = {} + for i=1,times,1 do + tbl[i] = string.gsub(str,"@i@", i-1) + end + local res = table.concat(tbl,"\n") + return res; + end + + function split(delimiter, text) + local list = {} + local pos = 1 + if string.find("", delimiter, 1) then -- this would result in endless loops + error("delimiter matches empty string!") + end + while 1 do + local first, last = string.find(text, delimiter, pos) + if first then -- found? + table.insert(list, string.sub(text, pos, first-1)) + pos = last+1 + else + table.insert(list, string.sub(text, pos)) + break + end + end + return list + end + + function conky_filter_real_ip(iface) + local req = string.format('${addrs %s}', iface) + local ips = conky_parse(req); +--[[ + print(ips) +--]] + local list = split(",", ips) + for i, item in ipairs(list) do + if (string.match(item, "141\.52\.64")) then + return item + end + end + return "Unknown" + end + + user = split('/', home) + user = user[#user] + + result_check_server_status = {} + result_check_adei_source = {} + result_check_server_ = {} + result_check_service_ = {} + online = {} + status = {} + tested = {} + c_timer = c_timer_init + width = 80 + update_time = 0 + outdate_time = 0 + timeout = 2 + + function conky_set_width (w) + width = tonumber(w) + return "" + end + + function conky_set_timeout (t) + timeout = tonumber(t) + return "" + end + + function cmd_popen(server, port, cmd) + if (c_timer_init == 0) then + return io.popen(cmd) + else + local cmd_md5 = md5.sumhexa(cmd) + local fn = string.format("%s/.conky/info/%s:%s-%s.srv", home, server, port, cmd_md5) + stat = lfs.attributes(fn) + if (stat == nil) then + return nil + end + + local since = os.difftime(stat['modification'], update_time) + if (since < 0) then + return nil + end + since = os.difftime(outdate_time, os.time()) + if (since < 0) then + return nil + end + return io.open(fn, "r") + end + end + + function conky_check_server_status(server, port, ...) + port = port or 22 + local name = string.format("%s:%i", server, port); + + if (c_timer < 0) then + if (c_timer == -1) then + local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port) + + tested[string.format("%s:%i", server, port)] = false; + + local f = assert(io.open(fn, "w")) + + local cmd = string.format("/etc/conky/scripts/ping.pl %s:%i %i", server, port, timeout) + f:write(cmd .. "\n") + + if (arg[1] ~= "-") then + cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_status.sh 2>&1 | fold -w %i", port, timeout, server, width) + f:write(cmd .. "\n") + + for i, service in ipairs(arg) + do + cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_%s_status.sh 2>&1 | fold -w %i", port, timeout, server, service, width) + f:write(cmd .. "\n") + end + end + f:close() + end + if (result_check_server_status[name] == nil) then + local res + if (math.abs(c_timer)%2 > 0) then + res = "${color yellow}?${color}" + else + res = "${color yellow}.${color}" + end + + if (arg[1] ~= "-") then + res = res .. " " + for i, service in ipairs(arg) do + res = res .. " " + end + end + --result_check_server_status[server] = res + --status[server] = false + return res + end + elseif (c_timer == 0) then + result_check_server_status[name] = check_server_status(server, port, arg) + end + + return result_check_server_status[name]; + end + + function check_server_status(server, port, additional) + port = port or 22 + + local res + local cmd = string.format("/etc/conky/scripts/ping.pl %s:%i %i", server, port, timeout) + local cmdf = cmd_popen(server, port, cmd) + if (cmdf == nil) then + res = -1 + else + local data = cmdf:read("*a") + res = tonumber(data) + cmdf:close() + end + + if (additional[1] == "-") then + if (res > 0) then + online[server] = true + return "${color green}*" + elseif (res < 0) then + return "${color red}?" + else + online[server] = false + return "${color red}*" + end + end + + if (res > 0) then + online[server] = true + + local output + + cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_status.sh 2>&1 | fold -w %i", port, timeout, server, width) + cmdf = cmd_popen(server, port, cmd) + if (cmdf == nil) then + output = string.format("${color green}* ${color red}?") + else + local res = {} + local line = cmdf:read("*l") + while line do + local m1 = string.match(line, "^\(.*[^%s]\)%s*$") + if (m1) then + table.insert(res, m1) + end + line = cmdf:read("*l") + end +-- res = cmdf:read("*a") + cmdf:close() + + res = table.concat(res, "\n") + + if (string.match(res, "[%a%d]")) then + table.insert(status, string.format("${color yellow}%s:%i${color gray}\n%s", server, port, res)) + output = string.format("${color green}* ${color red}*") + else + output = string.format("${color green}* *") + end + end + + for i, service in ipairs(additional) + do + cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_%s_status.sh 2>&1 | fold -w %i", port, timeout, server, service, width) + cmdf = cmd_popen(server, port, cmd) + if (cmdf == nil) then + output = output .. string.format(" ${color red}?") + else + local res = {} + local line = cmdf:read("*l") + while line do + local m1 = string.match(line, "^\(.*[^%s]\)%s*$") + if (m1) then + table.insert(res, m1) + end + line = cmdf:read("*l") + end + cmdf:close() + + if (res[1]) then + local info = table.remove(res) + local s = 0 + local extra = "" + + local m1, m2 = string.match(info, "^\(%d+\)\(%s.+\)$") + if (m1 == nil) then + m1 = string.match(info, "^\%d+$"); + end + if (m1 == nil) then + table.insert(res, info) + else + s = tonumber(m1) + if (m2 ~= nil) then + extra = string.sub(m2,2) + end + end + + if (res[1]) then + res = table.concat(res, "\n") + table.insert(status, string.format("${color yellow}%s:%i:%s${color gray}\n%s", server, port, service, res)) + end + if (s == 0) then + output = output .. string.format(" ${color red}*") + elseif (s == 1) then + output = output .. string.format(" ${color green}*") + else + output = output .. string.format(" ${color yellow}*") + end + output=output .. " ${color white}" .. extra; + else + output = output .. string.format(" ${color green}*") + end + end + end + + return output + else + if (res < 0) then + res = string.format("${color red}? ") + else + online[server] = false + res = string.format("${color red}* ") + end + for i, service in ipairs(additional) do + res = res .. " " + end + + return res + end + end + + function conky_check_adei_source(server, port, config, setup, db_server, db_name, ignore_list) + port = port or 22 + local name = string.format("%s:%i__%s__%s", server, port, db_server, db_name) + if (c_timer < 0) then + if (c_timer == -1) then + ignore_list = ignore_list or "" + + local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port) + + local f = assert(io.open(fn, "a+")) + local cmd = string.format('ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_adei_source.sh %s %s %s %s "%s" 2>&1', port, timeout, server, config, setup, db_server, db_name, ignore_list) + f:write(cmd .. "\n") + f:close() + end + + if (result_check_adei_source[name] == nil) then + return "${color yellow}?${color}" + else + if (online[server]) then + return result_check_adei_source[name]; + else + return "${color red}?${color}" + end + end + elseif (online[server]) then + if (c_timer == 0) then + result_check_adei_source[name] = check_adei_source(server, port, config, setup, db_server, db_name, ignore_list) + end + return result_check_adei_source[name]; + else + return "${color red}?${color}" + end + end + + function check_adei_source(server, port, config, setup, db_server, db_name, ignore_list) + ignore_list = ignore_list or "" + cmd = string.format('ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_adei_source.sh %s %s %s %s "%s" 2>&1', port, timeout, server, config, setup, db_server, db_name, ignore_list) + cmdf = cmd_popen(server, port, cmd) + if (cmdf == nil) then + return "${color red}?" + end + + local res = {} + line = cmdf:read("*l") + while line do + local m1 = string.match(line, "^\(.*[^%s]\)%s*$") + if (m1) then + table.insert(res, m1) + end + line = cmdf:read("*l") + end + cmdf:close() + + local info = table.remove(res) + + local t1, t2, t3 = string.match(info, "^\(%d+\)%s+\(%d+\)%s+\(%d+\)") + + local s = tonumber(t1) + if (s == nil) then + table.insert(res, info) + s = 0 + end + + res = table.concat(res, "\n") + + local output + if (s == 1) then + output="${color green}*" + else +--[[ + table.insert(status, string.format("${color yellow}%s -- %s${color gray}\n%s\n", db_server, db_name, res)) +--]] + if (s == 0) then output="${color red}*" + else output="${color yellow}*" end + end + +-- if (s ~= 0) then + local groups = tonumber(t2) + if (groups) then + output = output .. string.format("${color white} %i groups", groups); + else + output = output .. string.format("${color red} ? groups"); + end + + local size = tonumber(t3) + if (groups and size) then + output = output .. string.format("${color white}, %i GB", size); + else + output = output .. string.format("${color red}, ? GB"); + end +-- end + + if (string.match(res, "[%a%d]")) then + output = output .. "${color gray}\n ${font Bitstream Vera Sans Mono:size=7}" .. string.gsub(res,"\n", "${font}\n ${font Bitstream Vera Sans Mono:size=7}") .. "${font}${color white}" + end + + return output + end + + function conky_check_server_(service, server, port, ...) + port = port or 22 + local name = string.format("%s:%i:%s", server, port, service) + if (c_timer < 0) then + if (c_timer == -1) then + ignore_list = ignore_list or "" + + local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port) + + local f = assert(io.open(fn, "a+")) + local cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_%s.sh 2>&1", port, timeout, server, service) + f:write(cmd .. "\n") + f:close() + end + + if (result_check_server_[name] == nil) then + return "${color yellow}?${color}" + else + if (online[server]) then + return result_check_server_[name]; + else + return "${color red}?${color}" + end + end + elseif (online[server]) then + if (c_timer == 0) then + result_check_server_[name] = check_server_(service, server, port, arg) + end + return result_check_server_[name]; + else + return "${color red}?${color}" + end + end + + function conky_check_service_(service, id, ...) + local name = string.format("%s:%s", service, id) + if (c_timer < 0) then + if (c_timer == -1) then + ignore_list = ignore_list or "" + + tested[string.format("%s:%s", service, id)] = false; + + local fn = string.format("%s/.conky/requests/%s:%s.srv", home, service, id) + + local f = assert(io.open(fn, "a+")) + local cmd = string.format("/etc/conky/service/check_%s.sh %s %s 2>&1", service, id, table.concat(arg, " ")) + f:write(cmd .. "\n") + f:close() + end + + if (result_check_server_[name] == nil) then + return "${color yellow}?${color}" + else + return result_check_server_[name]; + end + else + if (c_timer == 0) then + result_check_server_[name] = check_service_(service, id, arg) + end + return result_check_server_[name]; + end + end + + + function format_traffic(value, yellow, red) + yellow = yellow or 100 + red = red or 1000 + + local value = math.floor(tonumber(value) / 1073741824) + + local res + if (value > red) then res="${color red}" + elseif (value > yellow) then res="${color yellow}" + else res="${color white}" end + + if (value < 10) then res = res .. " " .. value + elseif (value < 100) then res = res .. " " .. value + else res = res .. value end + + return res .. " GB${color gray}" + end + + function check_server_(service, server, port, opts) + port = port or 22 + local cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_%s.sh 2>&1", port, timeout, server, service) + cmdf = cmd_popen(server, port, cmd) + if (cmdf == nil) then + return "${color red}?" + end + + local res = {} + line = cmdf:read("*l") + while line do + local m1 = string.match(line, "^\(.*[^%s]\)%s*$") + if (m1) then + table.insert(res, m1) + end + line = cmdf:read("*l") + end + cmdf:close() + + if (service == "traffic") then + if (res[1]) then + local t1, t2, t3 = string.match(res[1], "^\(%d+\)%s+\(%d+\)%s+\(%d+\)") + + if (t1 and t2 and t3) then + if ((opts) and (opts[1])) then yellow = tonumber(opts[1]) + else yellow = nil end + if ((opts) and (opts[2])) then red = tonumber(opts[2]) + else red = nil end + + t1=format_traffic(t1,yellow,red) + t2=format_traffic(t2,yellow,red) + t3=format_traffic(t3,yellow,red) + return string.format("${color gray}i: %s, o: %s, f: %s", t1, t2, t3) + else + return "${color red}" .. res[1] + end + end + end + + + return "" + end + + + function check_service_(service, id, opts) + local cmd = string.format("/etc/conky/service/check_%s.sh %s %s 2>&1", service, id, table.concat(opts, " ")) + cmdf = cmd_popen(service, id, cmd) + if (cmdf == nil) then + return "${color red}?" + end + + local res = {} + local local_res = {} + line = cmdf:read("*l") + while line do + local t, m1 = string.match(line, "^\(\*?\)\(.*[^%s]\)%s*$") + if (m1) then + if (t == nil) or (t == '') then + table.insert(res, m1) + else + table.insert(local_res, string.sub(m1,1)) + end + end + line = cmdf:read("*l") + end + cmdf:close() + + local output + local info = table.remove(res) + local t1 = 0 + local t2 = nil + local msg = '' + + if (info) then + t1, t2, msg = string.match(info, "^\(%d+\)%s+\(%d+\)\(.*\)") + end + + local s = tonumber(t1) + if (s == nil) then + table.insert(res, info) + s = 0 + end + + if (s == 1) then + output="${color green}*" + else + if (s == 0) then output="${color red}*" + else output="${color yellow}*" end + end + + s = tonumber(t2) + if (s == nil) then + s = 0 + else + if (s == 1) then + output=output .. " ${color green}*" + else + if (s == 0) then output=output .. " ${color red}*" + else output=output .. " ${color yellow}*" end + end + end + + output = output .. " ${color white}" .. msg + + res = table.concat(res, "\n") + local_res = table.concat(local_res, "\n") + + + if (string.match(res, "[%a%d]")) then + table.insert(status, string.format("${color yellow}%s:%s${color gray}\n%s", service, id, res)) + end + + if (string.match(local_res, "[%a%d]")) then + output = output .. "${color gray}\n ${font Bitstream Vera Sans Mono:size=7}" .. string.gsub(local_res,"\n", "${font}\n ${font Bitstream Vera Sans Mono:size=7}") .. "${font}${color white}" + end + + return output + end + + + + function conky_print_server_errors(interval, tag) + local res = table.concat(status, "\n") + + if (c_timer < 0) then + tag = user .. "_" .. tag + + if (c_timer == -1) then + update_time = os.time() + + local result = 0; + while (result == 0) do + result = os.execute("ps x | grep conky_process_requests | grep " .. tag .. " | grep -v grep > /dev/null") + end + + for item, s in pairs(tested) do + os.execute("/etc/conky/scripts/conky_process_requests.pl \"" .. item .. ".srv\" " .. tag .. " &") + end + end + + local result=os.execute("ps x | grep conky_process_requests | grep " .. tag .. " | grep -v grep > /dev/null") + if (result == 0) then + c_timer = c_timer - 1; + local since = os.difftime(os.time(), update_time) + if (since > tonumber(interval)) then + result_check_server_status = {} + result_check_adei_source = {} + result_check_server_ = {} + end + else + outdate_time = os.time() + 2 * tonumber(interval) + c_timer = 0 + end + return "" + else + c_timer = c_timer + conky_info.update_interval + if (c_timer > tonumber(interval)) then + status = {} + tested = {} + c_timer = c_timer_init + end + end + + if (string.match(res, "[%a%d]")) then + return string.format("\n\n%s", res) + else + return "" + end + end + + function conky_outcon(pos1, pos2) + local n = tonumber(conky_parse('${tcp_portmon 32768 65535 count}')) + local res = {} + local str = "" + for i=1,n do + local val = tostring(conky_parse(string.format('${tcp_portmon 32768 65535 rhost %i}:${tcp_portmon 32768 65535 rport %i}',i - 1,i - 1))) + + if (res[val] == nil) then + res[val] = 1 + else + res[val] = res[val] + 1 + end + end + + + local names = {} + for addr,num in pairs(res) do + table.insert(names, addr) + end + + table.sort(names, function(a,b) return res[a]>res[b] end) + + for i=1,#names do + local name = names[i] + str = str .. string.format("${goto %i}%s${goto %i}cnt %3i\n",pos1,name,pos2,res[name]) + end + + return str + end +end |