mirror of
https://github.com/JamesonHuang/OpenWrt_Luci_Lua.git
synced 2024-11-27 15:30:11 +00:00
572 lines
21 KiB
Lua
572 lines
21 KiB
Lua
module ("xiaoqiang.util.XQDeviceUtil", package.seeall)
|
||
|
||
local Json = require("cjson")
|
||
|
||
local XQConfigs = require("xiaoqiang.common.XQConfigs")
|
||
local XQFunction = require("xiaoqiang.common.XQFunction")
|
||
local XQEquipment = require("xiaoqiang.XQEquipment")
|
||
|
||
function getDeviceCompany(mac)
|
||
local companyInfo = { name = "", icon = "" }
|
||
if XQFunction.isStrNil(mac) or string.len(mac) < 8 then
|
||
return companyInfo
|
||
end
|
||
return XQEquipment.identifyDevice(mac, nil)
|
||
end
|
||
|
||
function getDeviceInfoFromDB()
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local result = {}
|
||
local deviceList = XQDBUtil.fetchAllDeviceInfo()
|
||
if #deviceList > 0 then
|
||
for _,device in ipairs(deviceList) do
|
||
result[device.mac] = device
|
||
end
|
||
end
|
||
return result
|
||
end
|
||
|
||
function saveDeviceName(mac,name)
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local code = XQDBUtil.updateDeviceNickname(XQFunction.macFormat(mac),name)
|
||
if code == 0 then
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
--
|
||
-- Get DHCP list
|
||
--
|
||
|
||
function getDHCPList()
|
||
local NixioFs = require("nixio.fs")
|
||
local LuciUci = require("luci.model.uci")
|
||
local uci = LuciUci.cursor()
|
||
local result = {}
|
||
local leasefile = XQConfigs.DHCP_LEASE_FILEPATH
|
||
uci:foreach("dhcp", "dnsmasq",
|
||
function(s)
|
||
if s.leasefile and NixioFs.access(s.leasefile) then
|
||
leasefile = s.leasefile
|
||
return false
|
||
end
|
||
end)
|
||
local dhcp = io.open(leasefile, "r")
|
||
if dhcp then
|
||
for line in dhcp:lines() do
|
||
if line then
|
||
local ts, mac, ip, name = line:match("^(%d+) (%S+) (%S+) (%S+)")
|
||
if name == "*" then
|
||
name = ""
|
||
end
|
||
if ts and mac and ip and name then
|
||
result[#result+1] = {
|
||
mac = XQFunction.macFormat(mac),
|
||
ip = ip,
|
||
name = name
|
||
}
|
||
end
|
||
end
|
||
end
|
||
dhcp:close()
|
||
return result
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
|
||
function getDHCPDict()
|
||
local dhcpDict = {}
|
||
local dhcpList = getDHCPList()
|
||
for _,value in ipairs(dhcpList) do
|
||
dhcpDict[value.mac] = value
|
||
end
|
||
return dhcpDict
|
||
end
|
||
|
||
function getDHCPIpDict()
|
||
local dhcpDict = {}
|
||
local dhcpList = getDHCPList()
|
||
for _,value in ipairs(dhcpList) do
|
||
dhcpDict[value.ip] = value
|
||
end
|
||
return dhcpDict
|
||
end
|
||
|
||
function getMacfilterInfoList()
|
||
local LuciUtil = require("luci.util")
|
||
local macFilterInfo = {}
|
||
local metaData = LuciUtil.execi("/usr/sbin/sysapi macfilter get")
|
||
for filterInfo in metaData do
|
||
filterInfo = filterInfo..";"
|
||
local mac = filterInfo:match('mac=(%S-);') or ""
|
||
local wan = filterInfo:match('wan=(%S-);') or ""
|
||
local lan = filterInfo:match('lan=(%S-);') or ""
|
||
local admin = filterInfo:match('admin=(%S-);') or ""
|
||
local pridisk = filterInfo:match('pridisk=(%S-);') or ""
|
||
local entry = {}
|
||
if (not XQFunction.isStrNil(mac)) then
|
||
entry["mac"] = XQFunction.macFormat(mac)
|
||
entry["wan"] = (string.upper(wan) == "YES" and true or false)
|
||
entry["lan"] = (string.upper(lan) == "YES" and true or false)
|
||
entry["admin"] = (string.upper(admin) == "YES" and true or false)
|
||
entry["pridisk"] = (string.upper(pridisk) == "YES" and true or false)
|
||
table.insert(macFilterInfo, entry)
|
||
end
|
||
end
|
||
return macFilterInfo
|
||
end
|
||
|
||
function getMacfilterInfoDict()
|
||
local macFilterDict = {}
|
||
local macFilterList = getMacfilterInfoList()
|
||
for _,value in ipairs(macFilterList) do
|
||
macFilterDict[value.mac] = value
|
||
end
|
||
return macFilterDict
|
||
end
|
||
|
||
--[[
|
||
@param devName : lan/wan，其他情况 DEVNAME = DEV
|
||
]]--
|
||
function getWanLanNetworkStatistics(devName)
|
||
local LuciUtil = require("luci.util")
|
||
local tracmd = ""
|
||
if devName == "lan" then
|
||
tracmd = "ubus call trafficd lan"
|
||
elseif devName == "wan" then
|
||
tracmd = "ubus call trafficd wan"
|
||
end
|
||
local statistics = {
|
||
["upload"] = "0",
|
||
["upspeed"] = "0",
|
||
["download"] = "0",
|
||
["downspeed"] = "0",
|
||
["devname"] = "",
|
||
["maxuploadspeed"] = "0",
|
||
["maxdownloadspeed"] = "0"
|
||
}
|
||
|
||
local ubusinfo = LuciUtil.exec(tracmd)
|
||
if XQFunction.isStrNil(ubusinfo) then
|
||
return statistics
|
||
end
|
||
local ubusinfo = Json.decode(ubusinfo)
|
||
if devName == "wan" then
|
||
statistics.devname = tostring(ubusinfo.ifname)
|
||
statistics.upload = tostring(ubusinfo.tx_bytes)
|
||
statistics.download = tostring(ubusinfo.rx_bytes)
|
||
statistics.upspeed = tostring(math.floor(ubusinfo.tx_rate or 0))
|
||
statistics.downspeed = tostring(math.floor(ubusinfo.rx_rate or 0))
|
||
statistics.maxuploadspeed = tostring(math.floor(ubusinfo.max_tx_rate or 0))
|
||
statistics.maxdownloadspeed = tostring(math.floor(ubusinfo.max_rx_rate or 0))
|
||
|
||
local history = LuciUtil.exec("ubus call trafficd list_wan_rate")
|
||
if not XQFunction.isStrNil(history) then
|
||
historylist = {}
|
||
history = Json.decode(history)
|
||
for _, rate in ipairs(history.rate) do
|
||
if rate then
|
||
table.insert(historylist, tostring(rate))
|
||
end
|
||
end
|
||
statistics.history = table.concat(historylist, ",")
|
||
end
|
||
else
|
||
statistics.devname = tostring(ubusinfo.ifname)
|
||
statistics.upload = tostring(ubusinfo.rx_bytes)
|
||
statistics.download = tostring(ubusinfo.tx_bytes)
|
||
statistics.upspeed = tostring(math.floor(ubusinfo.rx_rate or 0))
|
||
statistics.downspeed = tostring(math.floor(ubusinfo.tx_rate or 0))
|
||
statistics.maxuploadspeed = tostring(math.floor(ubusinfo.max_rx_rate or 0))
|
||
statistics.maxdownloadspeed = tostring(math.floor(ubusinfo.max_tx_rate or 0))
|
||
end
|
||
return statistics
|
||
end
|
||
|
||
--[[
|
||
@param mac=B8:70:F4:27:0C:1B 网卡mac地址
|
||
@param upload=14745 主机当前累计上传数据总量（byte）
|
||
@param upspeed=54 主机5秒平均上传速度（byte/s）
|
||
@param download=25777 主机当前累计下载数据总量（byte）
|
||
@param downspeed=120 主机5秒平均下载速度（byte/s）
|
||
@param oneline=169 主机在线时长（秒）
|
||
@param devname 设备名
|
||
@param maxuploadspeed 上传峰值
|
||
@param maxdownloadspeed 下载峰值
|
||
]]--
|
||
function getDevNetStatisticsList()
|
||
local LuciUtil = require("luci.util")
|
||
local statList = {}
|
||
local dhcpNameDict = getDHCPDict()
|
||
local deviceInfoDict = getDeviceInfoFromDB()
|
||
|
||
local deviceinfo = LuciUtil.exec("ubus call trafficd hw")
|
||
if XQFunction.isStrNil(deviceinfo) then
|
||
return statList
|
||
else
|
||
deviceinfo = Json.decode(deviceinfo)
|
||
end
|
||
for key, dev in pairs(deviceinfo) do
|
||
if dev then
|
||
local item = {}
|
||
local mac = XQFunction.macFormat(key)
|
||
local name, nickName, oriName
|
||
if dhcpNameDict[mac] then
|
||
oriName = dhcpNameDict[mac].name
|
||
end
|
||
local device = deviceInfoDict[mac]
|
||
if device then
|
||
if XQFunction.isStrNil(oriName) then
|
||
oriName = device.oName
|
||
end
|
||
nickName = device.nickname
|
||
end
|
||
local company = XQEquipment.identifyDevice(mac, oriName)
|
||
local dtype = company["type"]
|
||
if not XQFunction.isStrNil(nickName) then
|
||
name = nickName
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(dtype.n) then
|
||
name = dtype.n
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(oriName) then
|
||
name = oriName
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(company.name) then
|
||
name = company.name
|
||
end
|
||
if XQFunction.isStrNil(name) then
|
||
name = mac
|
||
end
|
||
|
||
local tx_bytes, tx_rate, rx_bytes, rx_rate, max_tx_rate, max_rx_rate = 0, 0, 0, 0, 0, 0
|
||
local iplist = dev.ip_list
|
||
if #iplist > 0 then
|
||
for _, ip in ipairs(iplist) do
|
||
tx_bytes = tx_bytes + ip.tx_bytes or 0
|
||
rx_bytes = rx_bytes + ip.rx_bytes or 0
|
||
tx_rate = tx_rate + ip.tx_rate or 0
|
||
rx_rate = rx_rate + ip.rx_rate or 0
|
||
max_tx_rate = max_tx_rate + ip.max_tx_rate or 0
|
||
max_rx_rate = max_rx_rate + ip.max_rx_rate or 0
|
||
end
|
||
end
|
||
item["mac"] = mac
|
||
item["upload"] = tostring(tx_bytes)
|
||
item["upspeed"] = tostring(math.floor(tx_rate))
|
||
item["download"] = tostring(rx_bytes)
|
||
item["downspeed"] = tostring(math.floor(rx_rate))
|
||
item["online"] = tostring(dev.online_timer or 0)
|
||
item["devname"] = name
|
||
item["maxuploadspeed"] = tostring(math.floor(max_tx_rate))
|
||
item["maxdownloadspeed"] = tostring(math.floor(max_rx_rate))
|
||
statList[#statList+1] = item
|
||
end
|
||
end
|
||
return statList
|
||
end
|
||
|
||
function getDevNetStatisticsDict()
|
||
local statDict = {}
|
||
local statlist = getDevNetStatisticsList()
|
||
for _, item in ipairs(statlist) do
|
||
if item then
|
||
statDict[item.mac] = item
|
||
end
|
||
end
|
||
return statDict
|
||
end
|
||
|
||
function getConnectDeviceCount()
|
||
local LuciUtil = require("luci.util")
|
||
local count = 0
|
||
local deviceinfo = LuciUtil.exec("ubus call trafficd hw")
|
||
if XQFunction.isStrNil(deviceinfo) then
|
||
return count
|
||
else
|
||
deviceinfo = Json.decode(deviceinfo)
|
||
end
|
||
for _,device in pairs(deviceinfo) do
|
||
if device and device.ip_list and #device.ip_list > 0 then
|
||
local dev = device.ifname
|
||
if not XQFunction.isStrNil(dev) and (tonumber(device.assoc) == 1 or not dev:match("wl")) then
|
||
local tcount = 0
|
||
for _,ip in ipairs(device.ip_list) do
|
||
if ip.ageing_timer <= 300 then
|
||
if tcount >= 1 then
|
||
break
|
||
end
|
||
count = count + 1
|
||
tcount = tcount + 1
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
return count
|
||
end
|
||
|
||
--[[
|
||
@return online: 0 (offline) 1 (online)
|
||
@return ip: ip address
|
||
@return mac: mac address
|
||
@return type: wifi/line
|
||
@return tag: 1 (normal) 2 (in denylist)
|
||
@return port: 1 (2.4G wifi) 2 (5G wifi)
|
||
@return name: name for show
|
||
@return origin_name: origin name
|
||
@return signal: wifi signal
|
||
@return statistics:
|
||
]]--
|
||
function getConnectDeviceList()
|
||
local XQWifiUtil = require("xiaoqiang.util.XQWifiUtil")
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local XQEquipment = require("xiaoqiang.XQEquipment")
|
||
local XQLog = require("xiaoqiang.XQLog")
|
||
|
||
local LuciUtil = require("luci.util")
|
||
local deviceList = {}
|
||
|
||
local deviceinfo = LuciUtil.exec("ubus call trafficd hw")
|
||
if XQFunction.isStrNil(deviceinfo) then
|
||
return deviceList
|
||
else
|
||
deviceinfo = Json.decode(deviceinfo)
|
||
end
|
||
|
||
local macFilterDict = getMacfilterInfoDict()
|
||
local dhcpDeviceDict = getDHCPDict()
|
||
local dhcpDevIpDict = getDHCPIpDict()
|
||
local deviceInfoDict = getDeviceInfoFromDB()
|
||
local wifiDeviceDict = XQWifiUtil.getAllWifiConnetDeviceDict()
|
||
|
||
for key ,item in pairs(deviceinfo) do
|
||
if item and item.ip_list and #item.ip_list > 0 then
|
||
local devicesignal, devicetype, deviceport
|
||
local dev = item.ifname
|
||
local mac = XQFunction.macFormat(key)
|
||
|
||
if not XQFunction.isStrNil(dev) and (tonumber(item.assoc) == 1 or not dev:match("wl")) then
|
||
-- 信号强度
|
||
local signal = wifiDeviceDict[mac]
|
||
if signal and signal.signal then
|
||
devicesignal = signal.signal
|
||
else
|
||
devicesignal = ""
|
||
end
|
||
if dev:match("eth") then
|
||
devicetype = "line"
|
||
deviceport = 0
|
||
elseif dev == "wl0" then
|
||
devicetype = "wifi"
|
||
deviceport = 2
|
||
elseif dev == "wl1" then
|
||
devicetype = "wifi"
|
||
deviceport = 1
|
||
elseif dev == "wl1.2" then
|
||
devicetype = "wifi"
|
||
deviceport = 3
|
||
end
|
||
|
||
-- 访问权限
|
||
local authority = {}
|
||
if (macFilterDict[mac]) then
|
||
authority["wan"] = macFilterDict[mac]["wan"] and 1 or 0
|
||
authority["lan"] = macFilterDict[mac]["lan"] and 1 or 0
|
||
authority["admin"] = macFilterDict[mac]["admin"] and 1 or 0
|
||
authority["pridisk"] = macFilterDict[mac]["pridisk"] and 1 or 0
|
||
else
|
||
authority["wan"] = 1
|
||
authority["lan"] = 1
|
||
authority["admin"] = 1
|
||
-- private disk deny access default
|
||
authority["pridisk"] = 0
|
||
end
|
||
|
||
local deviceInfo = deviceInfoDict[mac]
|
||
local dhcpinfo = dhcpDeviceDict[mac]
|
||
if not deviceInfo and dhcpinfo then
|
||
XQDBUtil.saveDeviceInfo(mac,dhcpinfo.name,"","","")
|
||
end
|
||
local count = 0
|
||
for _,ip in ipairs(item.ip_list) do
|
||
if ip.ageing_timer <= 300 then
|
||
if count >= 1 then
|
||
break
|
||
end
|
||
-- 设备名称
|
||
local name, originName, nickName, company
|
||
if dhcpDevIpDict[ip.ip] ~= nil then
|
||
originName = dhcpDevIpDict[ip.ip].name
|
||
end
|
||
|
||
if originName and originName:match("^xiaomi%-ir") then -- fix miio model string
|
||
originName = originName:gsub("%-",".")
|
||
end
|
||
|
||
if deviceInfo then
|
||
if XQFunction.isStrNil(originName) then
|
||
originName = deviceInfo.oName
|
||
end
|
||
nickName = deviceInfo.nickname
|
||
end
|
||
if not XQFunction.isStrNil(nickName) then
|
||
name = nickName
|
||
end
|
||
company = XQEquipment.identifyDevice(mac, originName)
|
||
local dtype = company["type"]
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(dtype.n) then
|
||
name = dtype.n
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(originName) then
|
||
name = originName
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(company.name) then
|
||
name = company.name
|
||
end
|
||
if XQFunction.isStrNil(name) then
|
||
name = mac
|
||
end
|
||
if dtype.c == 3 and XQFunction.isStrNil(nickName) then
|
||
name = dtype.n
|
||
end
|
||
local device = {
|
||
["ip"] = ip.ip,
|
||
["mac"] = mac,
|
||
["online"] = 1,
|
||
["type"] = devicetype,
|
||
["port"] = deviceport,
|
||
["ctype"] = dtype.c,
|
||
["ptype"] = dtype.p,
|
||
["origin_name"] = originName or "",
|
||
["name"] = name,
|
||
["company"] = company,
|
||
["authority"] = authority
|
||
}
|
||
local statistics = {}
|
||
statistics["dev"] = dev
|
||
statistics["mac"] = mac
|
||
statistics["ip"] = ip.ip
|
||
statistics["upload"] = tostring(ip.tx_bytes or 0)
|
||
statistics["upspeed"] = tostring(math.floor(ip.tx_rate or 0))
|
||
statistics["download"] = tostring(ip.rx_bytes or 0)
|
||
statistics["downspeed"] = tostring(math.floor(ip.rx_rate or 0))
|
||
statistics["online"] = tostring(item.online_timer or 0)
|
||
statistics["maxuploadspeed"] = tostring(math.floor(ip.max_tx_rate or 0))
|
||
statistics["maxdownloadspeed"] = tostring(math.floor(ip.max_rx_rate or 0))
|
||
device["statistics"] = statistics
|
||
table.insert(deviceList, device)
|
||
count = count + 1
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
-- if #deviceList > 0 then
|
||
-- table.sort(deviceList,
|
||
-- function(a, b)
|
||
-- return b.statistics.onlinets < a.statistics.onlinets
|
||
-- end
|
||
-- )
|
||
-- end
|
||
return deviceList
|
||
end
|
||
|
||
function getConDevices(withbrlan)
|
||
local XQDBUtil = require("xiaoqiang.util.XQDBUtil")
|
||
local XQEquipment = require("xiaoqiang.XQEquipment")
|
||
local XQLog = require("xiaoqiang.XQLog")
|
||
|
||
local LuciUtil = require("luci.util")
|
||
local deviceList = {}
|
||
|
||
local deviceinfo = LuciUtil.exec("ubus call trafficd hw")
|
||
if XQFunction.isStrNil(deviceinfo) then
|
||
return deviceList
|
||
else
|
||
deviceinfo = Json.decode(deviceinfo)
|
||
end
|
||
|
||
local dhcpDevIpDict = getDHCPIpDict()
|
||
--local dhcpDeviceDict = getDHCPDict()
|
||
local deviceInfoDict = getDeviceInfoFromDB()
|
||
|
||
for key ,item in pairs(deviceinfo) do
|
||
if item and item.ip_list and #item.ip_list > 0 then
|
||
local dev = item.ifname
|
||
local mac = XQFunction.macFormat(key)
|
||
|
||
if not XQFunction.isStrNil(dev) and ((not dev:match("wl") and withbrlan) or (dev:match("wl") and tonumber(item.assoc) == 1)) then
|
||
local count = 0
|
||
for _,ip in ipairs(item.ip_list) do
|
||
if count >= 1 then
|
||
break
|
||
end
|
||
local name, originName, nickName, company
|
||
if dhcpDevIpDict[ip.ip] ~= nil then
|
||
originName = dhcpDevIpDict[ip.ip].name
|
||
end
|
||
local deviceInfo = deviceInfoDict[mac]
|
||
if deviceInfo then
|
||
if XQFunction.isStrNil(originName) then
|
||
originName = deviceInfo.oName
|
||
end
|
||
nickName = deviceInfo.nickname
|
||
end
|
||
if not XQFunction.isStrNil(nickName) then
|
||
name = nickName
|
||
end
|
||
company = XQEquipment.identifyDevice(mac, originName)
|
||
|
||
local dtype = company["type"]
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(dtype.n) then
|
||
name = dtype.n
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(originName) then
|
||
name = originName
|
||
end
|
||
if XQFunction.isStrNil(name) and not XQFunction.isStrNil(company.name) then
|
||
name = company.name
|
||
end
|
||
if XQFunction.isStrNil(name) then
|
||
name = mac
|
||
end
|
||
if dtype.c == 3 and XQFunction.isStrNil(nickName) then
|
||
name = dtype.n
|
||
end
|
||
if ip.ageing_timer <= 300 then
|
||
local device = {
|
||
["ip"] = ip.ip,
|
||
["mac"] = mac,
|
||
["online"] = 1,
|
||
["ctype"] = dtype.c,
|
||
["ptype"] = dtype.p,
|
||
["origin_name"] = originName or "",
|
||
["name"] = name,
|
||
["company"] = company
|
||
}
|
||
local statistics = {}
|
||
statistics["dev"] = dev
|
||
statistics["mac"] = mac
|
||
statistics["ip"] = ip.ip
|
||
statistics["upload"] = tostring(ip.tx_bytes or 0)
|
||
statistics["upspeed"] = tostring(math.floor(ip.tx_rate or 0))
|
||
statistics["download"] = tostring(ip.rx_bytes or 0)
|
||
statistics["downspeed"] = tostring(math.floor(ip.rx_rate or 0))
|
||
statistics["online"] = tostring(item.online_timer or 0)
|
||
statistics["maxuploadspeed"] = tostring(math.floor(ip.max_tx_rate or 0))
|
||
statistics["maxdownloadspeed"] = tostring(math.floor(ip.max_rx_rate or 0))
|
||
device["statistics"] = statistics
|
||
table.insert(deviceList, device)
|
||
count = count + 1
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
return deviceList
|
||
end
|