-
Notifications
You must be signed in to change notification settings - Fork 0
/
fitnew.lua
171 lines (143 loc) · 6.52 KB
/
fitnew.lua
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
local http = require "luci.http"
local util = require "luci.util"
local ltn12 = require "luci.ltn12"
local cjson = require "cjson"
local sys = require("luci.sys")
local http=require("socket.http")
--local ap = require("ap")
function syslog(log)
os.execute(string.format([[logger "%s"]],log))
end
function os.capture(cmd)
local s = sys.exec(cmd)
return s
end
function request_server(server,client,token)
syslog("server="..server)
syslog("client="..client)
syslog("token="..token)
local request_body_prefix = [[
<?xml version="1.0" encoding="utf-8"?>
<s:Envelope xmlns:s="http:https://schemas.xmlsoap.org/soap/envelope/" s:encodingStyle="http:https://schemas.xmlsoap.org/soap/encoding/">
<s:Body>
<u:SetDeviceAuthorization xmlns:u="urn:schemas-upnp-org:service:ConnectionManager:1">
<DeviceIP>%s</DeviceIP>
<Token>%s</Token>
<Enable>1</Enable>
</u:SetDeviceAuthorization>
</s:Body>
</s:Envelope>
]]
local request_body = string.format(request_body_prefix,client,token)
--syslog(request_body)
local response_body = {}
local res, code, response_headers = http.request{
url = server.."/_urn:schemas-upnp-org:service:ConnectionManager_control",
method = "POST",
headers =
{
["SOAPACTION"] = [["urn:schemas-upnp-org:service:ConnectionManager:1#SetDeviceAuthorization"]];
["Content-Type"] = [[text/xml]];
["Content-Length"] = #request_body;
},
source = ltn12.source.string(request_body),
sink = ltn12.sink.table(response_body),
}
--syslog("response_headers.length="..response_headers.content-length)
return code
end
function parse_data_from_phone(data,env)
syslog("type=1")
local ip_client = env.REMOTE_HOST
local server = data.server
local token = data.token
local result = request_server(server,ip_client,token)
return result
end
function get_phone_mac_from_record_file_by_userid(userid)
local cmd_get_phone_mac_by_user_id_prefix = [[cat /var/run/fitnew_userid_mac | grep %s | awk -F";" '{print $2}']]
local cmd_get_phone_mac_by_user_id = string.format(cmd_get_phone_mac_by_user_id_prefix,userid)
local phone_mac = sys.exec(cmd_get_phone_mac_by_user_id)
if(string.len(phone_mac) == 0 or string.len(phone_mac) == 1) then
return 0
else
return string.gsub(phone_mac,"\n","")
end
end
function get_server_mac_from_record_file_by_userid(userid)
--local record_userid_server_mac_file = "/var/run/fitnew_userid_mac"
local cmd_get_server_mac_by_user_id_prefix = [[cat /var/run/fitnew_userid_mac | grep %s | awk -F";" '{print $3}']]
local cmd_get_server_mac_by_user_id = string.format(cmd_get_server_mac_by_user_id_prefix,userid)
local server_mac = sys.exec(cmd_get_server_mac_by_user_id)
return string.gsub(server_mac,"\n","")
end
function parse_data_from_pad(data,env)
local ip_client = env.REMOTE_HOST
syslog("ip client: "..ip_client)
syslog("ip_client="..ip_client)
--local mac_server = get_mac_by_ip(ip_client) -- 在这里为平板mac地址
--syslog("mac_server: "..mac_server)
local userid = data.userid
if(data.action == 0) then --用户扫二维码时，平板发出消息(消息中有client地址则从黑名单中移除,主要是为了将用户ID和平板mac地址记录下来)
local result = 0
local mac_phone = get_phone_mac_from_record_file_by_userid(userid)
if(mac_phone ~= 0) then
--不是第一次登录,将mac地址从黑名单中移除(文件中已有完整的 userid:client_mac:server_mac映射关系)
--将手机mac地址从黑名单中移除,并更新关系映射文件,以防止用户是从不同的平板电脑登录导致无法一对一投屏
syslog("type=0,action=0,mac_phone="..mac_phone)
local mac_server_history = get_server_mac_from_record_file_by_userid(userid)
rm_rule(mac_phone,mac_server_history)
--update_record_file(userid,"",mac_server)
update_record_file(userid,mac_phone,mac_server)
result = remove_mac_from_blacklist(mac_phone)
else
syslog("type=0,action=0,mac_phone=0")
result = record_userid_servermac_to_file(userid,mac_server) -- 第一次登录,将用户ID和平板mac地址写入文件
end
return cjson.encode({errcode=result})
elseif (data.action == 1) then --消息从平板发出，请求路由器将设备踢下线并将mac地址拉入黑名单
local mac_phone = get_phone_mac_from_record_file_by_userid(userid)
local mac_server_history = get_server_mac_from_record_file_by_userid(userid)
if(mac_phone == 0) then -- 用户扫码,但是没有使用小程序连接wifi
syslog("type=0,action=1,mac_phone=0")
local result_delete_userid_entry = delete_userid_entry(userid)
return result_delete_userid_entry
end
syslog("type=0,action=1,mac_phone="..mac_phone)
local result_dis = disassociate(mac_phone) --踢掉用户
local result_rm_rule = rm_rule(mac_phone,mac_server) --删除规则(本应是mac_client,为了方便测试改为data.client)
syslog("mac_server: "..mac_server)
syslog("mac_server_history: "..mac_server_history)
if(mac_server == mac_server_history) then
local result_add_mac_to_blacklist = add_mac_to_blacklist(mac_phone) --将用户加入黑名单
end
--local result_delete_userid_entry = delete_userid_entry(userid)
return result_dis
elseif (data.action == 2) then --平板向路由器请求获取wifi的SSID及密码
syslog("type=0,action=2")
local cmd_get_wifi_ssid = "uci get wireless.default_ra.ssid"
local cmd_get_wifi_key = "uci get wireless.default_ra.key"
local result_ssid = sys.exec(cmd_get_wifi_ssid)
local result_key = sys.exec(cmd_get_wifi_key)
local result = {ssid= string.gsub(result_ssid,"\n",""),key= string.gsub(result_key,"\n","")}
return cjson.encode(result)
end
end
function handle_request(env)
uhttpd.send("Status: 200 OK\r\n")
uhttpd.send("Content-Type: application/json\r\n\r\n")
local client_data = io.read("*all")
local data = cjson.decode(client_data)
if(data.type == 0) then
syslog("type="..data.type)
syslog("action="..data.action)
local result = parse_data_from_pad(data,env)
syslog("result="..result)
uhttpd.send(result)
elseif(data.type == 1) then
syslog("token ="..data.token)
local result = parse_data_from_phone(data,env)
local res = {err=tostring(result)}
uhttpd.send(cjson.encode(res))
end
end