mirror of
https://github.com/JamesonHuang/OpenWrt_Luci_Lua.git
synced 2024-11-24 06:10:11 +00:00
95 lines
2.5 KiB
Lua
95 lines
2.5 KiB
Lua
module ("xiaoqiang.module.XQNetworkSpeedTest", package.seeall)
|
|
|
|
local LuciFs = require("luci.fs")
|
|
local LuciSys = require("luci.sys")
|
|
local LuciUtil = require("luci.util")
|
|
|
|
local XQFunction = require("xiaoqiang.common.XQFunction")
|
|
local XQConfigs = require("xiaoqiang.common.XQConfigs")
|
|
|
|
local DIR = "/tmp/"
|
|
-- Kbyte
|
|
local POST_FILESIZE = 512
|
|
-- Number of requests to perform
|
|
local REQUEST_TIMES = 40
|
|
-- Number of multiple requests to make at a time
|
|
local REQUEST_NUM = 4
|
|
|
|
local TIMELIMITE = 5
|
|
local TIMESTEP = 1
|
|
local AB_CMD = "/usr/bin/ab"
|
|
local DD_CMD = "/bin/dd"
|
|
|
|
local POST_URL = "http://netsp.master.qq.com/cgi-bin/netspeed"
|
|
|
|
function execl(command, times)
|
|
local io = require("io")
|
|
local pp = io.popen(command)
|
|
local line = ""
|
|
local data = {}
|
|
if times < 1 then
|
|
return nil
|
|
end
|
|
while true do
|
|
line = pp:read()
|
|
if not XQFunction.isStrNil(line) then
|
|
local speed = tonumber(line:match("tx:(%S+)"))
|
|
if speed > 0 then
|
|
table.insert(data, speed)
|
|
else
|
|
break
|
|
end
|
|
else
|
|
break
|
|
end
|
|
end
|
|
pp:close()
|
|
if #data > 2 then
|
|
return data[#data]
|
|
else
|
|
return execl(command, times - 1)
|
|
end
|
|
end
|
|
|
|
function uploadSpeedTest()
|
|
local postfile = DIR..LuciSys.uniqueid(8)..".dat"
|
|
local pfcmd = string.format("%s if=/dev/zero of=%s bs=1k count=%d >/dev/null 2>&1", DD_CMD, postfile, POST_FILESIZE)
|
|
os.execute(pfcmd)
|
|
if postfile and LuciFs.access(postfile) then
|
|
local cmd = string.format("%s -N -s %d -M %d -n %d -c %d -T 'multipart/form-data' -p %s '%s'",
|
|
AB_CMD, TIMESTEP, TIMELIMITE, REQUEST_TIMES, REQUEST_NUM, postfile, POST_URL)
|
|
local speed = execl(cmd, 2)
|
|
if postfile and LuciFs.access(postfile) then
|
|
LuciFs.unlink(postfile)
|
|
end
|
|
if speed then
|
|
return speed
|
|
else
|
|
return nil
|
|
end
|
|
else
|
|
local XQLog = require("xiaoqiang.XQLog")
|
|
XQLog.log(6, "create postfile error")
|
|
return nil
|
|
end
|
|
end
|
|
|
|
function downloadSpeedTest()
|
|
local result = {}
|
|
local cmd = "/usr/bin/speedtest"
|
|
for _, line in ipairs(LuciUtil.execl(cmd)) do
|
|
if not XQFunction.isStrNil(line) then
|
|
table.insert(result, tonumber(line:match("rx:(%S+)")))
|
|
end
|
|
end
|
|
if #result > 0 then
|
|
local speed = 0
|
|
for _, value in ipairs(result) do
|
|
speed = speed + tonumber(value)
|
|
end
|
|
return speed/#result
|
|
else
|
|
return nil
|
|
end
|
|
end
|