191 lines
7.8 KiB
Lua
191 lines
7.8 KiB
Lua
-- Copyright 2018-2020 Lienol <lawlienol@gmail.com>
|
|
module("luci.controller.ssr_mudb_server", package.seeall)
|
|
local http = require "luci.http"
|
|
local jsonc = require "luci.jsonc"
|
|
|
|
function index()
|
|
if not nixio.fs.access("/etc/config/ssr_mudb_server") then return end
|
|
entry({"admin", "vpn"}, firstchild(), "VPN", 45).dependent = false
|
|
if nixio.fs.access("/usr/share/ssr_mudb_server") then
|
|
entry({"admin", "vpn", "ssr_mudb_server"}, cbi("ssr_mudb_server/index"), _("SSR MuDB Server"), 2).dependent = true
|
|
end
|
|
|
|
entry({"admin", "vpn", "ssr_mudb_server", "user"}, template("ssr_mudb_server/user")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "config"}, cbi("ssr_mudb_server/config")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "status"}, call("status")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "user_save"}, call("user_save")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "user_list"}, call("user_list")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "user_get"}, call("user_get")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "get_link"}, call("get_link")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "clear_traffic"}, call("clear_traffic")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "clear_traffic_all_users"}, call("clear_traffic_all_users")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "remove_user"}, call("remove_user")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "get_log"}, call("get_log")).leaf = true
|
|
entry({"admin", "vpn", "ssr_mudb_server", "clear_log"}, call("clear_log")).leaf = true
|
|
end
|
|
|
|
local function http_write_json(content)
|
|
http.prepare_content("application/json")
|
|
http.write_json(content or {code = 1})
|
|
end
|
|
|
|
local function get_config_path()
|
|
return luci.sys.exec("echo -n $(cat /usr/share/ssr_mudb_server/userapiconfig.py | grep 'MUDB_FILE' | cut -d \"'\" -f 2)")
|
|
end
|
|
|
|
local function get_config_json()
|
|
return luci.sys.exec("cat " .. get_config_path()) or "[]"
|
|
end
|
|
|
|
function status()
|
|
local e = {}
|
|
e.status = luci.sys.call("busybox ps -w | grep -v grep | grep '/usr/share/ssr_mudb_server/server.py' >/dev/null") == 0
|
|
http_write_json(e)
|
|
end
|
|
|
|
function get_link()
|
|
local e = {}
|
|
local link = luci.sys.exec("cd /usr/share/ssr_mudb_server && python3 mujson_mgr.py -l -p " .. luci.http.formvalue("port") .. " | sed -n '$p'"):gsub("^%s*(.-)%s*$", "%1")
|
|
if link ~= "" then e.link = link end
|
|
http_write_json(e)
|
|
end
|
|
|
|
function clear_traffic()
|
|
local e = {}
|
|
e.status = luci.sys.call("cd /usr/share/ssr_mudb_server && python3 mujson_mgr.py -c -p '" .. luci.http.formvalue("port") .. "' >/dev/null") == 0
|
|
http_write_json(e)
|
|
end
|
|
|
|
function clear_traffic_all_users()
|
|
local e = {}
|
|
e.status = luci.sys.call("/usr/share/ssr_mudb_server/clear_traffic_all_users.sh >/dev/null") == 0
|
|
http_write_json(e)
|
|
end
|
|
|
|
function user_list()
|
|
luci.http.prepare_content("application/json")
|
|
luci.http.write(get_config_json())
|
|
end
|
|
|
|
function user_save()
|
|
local result = {code = 0}
|
|
local action = luci.http.formvalue("action")
|
|
local json_str = luci.http.formvalue("json")
|
|
if action and action == "add" or action == "edit" then
|
|
local user = jsonc.parse(json_str)
|
|
if user then
|
|
local json = jsonc.parse(get_config_json())
|
|
if json then
|
|
local port = user.port
|
|
local is_exist_port = 0
|
|
for index = 1, table.maxn(json) do
|
|
if json[index].port == tonumber(port) then
|
|
is_exist_port = 1
|
|
if user.old_port and user.old_port == tonumber(port) then
|
|
is_exist_port = 0
|
|
end
|
|
break
|
|
end
|
|
end
|
|
if is_exist_port == 0 then
|
|
if action == "add" then
|
|
local new_user = {
|
|
enable = tonumber(user.enable) or 0,
|
|
user = user.user or user.port,
|
|
port = tonumber(user.port),
|
|
passwd = user.passwd,
|
|
method = user.method,
|
|
protocol = user.protocol,
|
|
obfs = user.obfs,
|
|
protocol_param = user.protocol_param or "2",
|
|
speed_limit_per_con = tonumber(user.speed_limit_per_con) or 0,
|
|
speed_limit_per_user = tonumber(user.speed_limit_per_user) or 0,
|
|
forbidden_port = user.forbidden_port or "",
|
|
transfer_enable = tonumber(user.transfer_enable) or 1073741824,
|
|
d = 0,
|
|
u = 0
|
|
}
|
|
table.insert(json, new_user)
|
|
elseif action == "edit" then
|
|
for index = 1, table.maxn(json) do
|
|
if json[index].port == tonumber(user.old_port) then
|
|
json[index].enable = tonumber(user.enable) or 0
|
|
json[index].user = user.user
|
|
json[index].port = tonumber(user.port)
|
|
json[index].passwd = user.passwd
|
|
json[index].method = user.method
|
|
json[index].protocol = user.protocol
|
|
json[index].obfs = user.obfs
|
|
json[index].protocol_param = user.protocol_param
|
|
json[index].speed_limit_per_con = tonumber(user.speed_limit_per_con)
|
|
json[index].speed_limit_per_user = tonumber(user.speed_limit_per_user)
|
|
json[index].forbidden_port = user.forbidden_port
|
|
json[index].transfer_enable = tonumber(user.transfer_enable)
|
|
break
|
|
end
|
|
end
|
|
end
|
|
local f, err = io.open(get_config_path(), "w")
|
|
if f and err == nil then
|
|
f:write(jsonc.stringify(json, 1))
|
|
f:close()
|
|
luci.sys.call("/etc/init.d/ssr_mudb_server restart")
|
|
result = {code = 1}
|
|
end
|
|
else
|
|
result.msg = "端口已存在！"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
http_write_json(result)
|
|
end
|
|
|
|
function user_get()
|
|
local result = {}
|
|
local port = luci.http.formvalue("port")
|
|
local str = get_config_json()
|
|
local json = jsonc.parse(str)
|
|
if port and str and json then
|
|
for index = 1, table.maxn(json) do
|
|
local o = json[index]
|
|
if o.port == tonumber(port) then
|
|
result = o
|
|
break
|
|
end
|
|
end
|
|
end
|
|
http_write_json(result)
|
|
end
|
|
|
|
function remove_user()
|
|
local port = luci.http.formvalue("port")
|
|
local str = get_config_json()
|
|
local json = jsonc.parse(str)
|
|
if port and str and json then
|
|
for index = 1, table.maxn(json) do
|
|
local o = json[index]
|
|
if o.port == tonumber(port) then
|
|
json[index] = nil
|
|
break
|
|
end
|
|
end
|
|
local f, err = io.open(get_config_path(), "w")
|
|
if f and err == nil then
|
|
f:write(jsonc.stringify(json, 1))
|
|
f:close()
|
|
end
|
|
luci.http.status = 200
|
|
else
|
|
luci.http.status = 500
|
|
end
|
|
end
|
|
|
|
function get_log()
|
|
luci.http.write(luci.sys.exec("[ -f '/var/log/ssr_mudb_server.log' ] && cat /var/log/ssr_mudb_server.log"))
|
|
end
|
|
|
|
function clear_log()
|
|
luci.sys.call("echo '' > /var/log/ssr_mudb_server.log")
|
|
end
|