chore(shebang): python3; style(newline): crlf -> lf
This commit is contained in:
+212
-212
@@ -1,212 +1,212 @@
|
||||
|
||||
from sys import maxsize
|
||||
from enum import IntEnum, auto
|
||||
from urllib.error import URLError
|
||||
from urllib import request
|
||||
from errno import ENETUNREACH
|
||||
import json
|
||||
import random
|
||||
import subprocess
|
||||
|
||||
def request_get_text(url, headers={}):
|
||||
req = request.Request(url, headers=headers)
|
||||
with request.urlopen(req) as response:
|
||||
return response.read().decode("utf-8")
|
||||
|
||||
class LoginStatus(IntEnum):
|
||||
unknown = -maxsize-1
|
||||
no_wifi = -3
|
||||
not_unlimit = -2
|
||||
bad_pwd = -1
|
||||
succ = auto()
|
||||
used_online = auto()
|
||||
|
||||
class Loginer:
|
||||
def __init__(self,
|
||||
interface = None,
|
||||
online_device_limit = 2,
|
||||
log = print,
|
||||
):
|
||||
self.interface = interface
|
||||
self.online_device_limit = online_device_limit
|
||||
self.log = log
|
||||
self.interface_def = interface is None
|
||||
def check_connectivity(self):
|
||||
interface = self.interface
|
||||
if not self.interface_def:
|
||||
command = ["ping", "-I", interface]
|
||||
host = "223.5.5.5"
|
||||
command += ["-c", "1", host] # -n 1 on windows, -c 1 on linux
|
||||
try:
|
||||
result = subprocess.check_call(command, timeout=2)
|
||||
return result == 0
|
||||
except subprocess.TimeoutExpired:
|
||||
self.log(f"接口 {interface}：Ping to {host} timed out.")
|
||||
return False
|
||||
except subprocess.CalledProcessError:
|
||||
return False
|
||||
except Exception as e:
|
||||
self.log(f"接口 {interface}：Error during ping: {e}")
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
request_get_text("http://connect.rom.miui.com/generate_204")
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
referrer = "http://192.168.101.201"
|
||||
header = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36 Edg/105.0.1343.33",
|
||||
"Referer": referrer
|
||||
}
|
||||
|
||||
|
||||
@staticmethod
|
||||
def url(last, dr, v):
|
||||
return "/eportal/portal/" + last + "?callback=dr" + dr + "&v=" + v + \
|
||||
"&lang=zh-CN&wlan_user_mac=000000000000&jsVersion=4.1"
|
||||
|
||||
ip0 = "&wlan_user_ip=0.0.0.0"
|
||||
index_qs = "&program_index=ctshNw1713845951&page_index=V5fmKw1713845966"
|
||||
urlloadUserInfo = referrer + ":801" + url("page/loadUserInfo", "1004", "599") + ip0 + index_qs
|
||||
time_qs = "&start_time=2010-01-01&end_time=2100-01-01&start_rn=1&end_rn=5"
|
||||
urlloadOnlineRecord = referrer + ":801" + url("page/loadOnlineRecord", "1006", "2399") + time_qs + index_qs + "&wlan_user_ip=10.169.0.241"
|
||||
urllogin = "https://xha.ouc.edu.cn:802" + url("login", "1003", "2425") + ip0 + "&login_method=1&wlan_user_ipv6=&wlan_ac_ip=&wlan_ac_name=&terminal_type=1"
|
||||
|
||||
@staticmethod
|
||||
def json_from_drnnnn(t):
|
||||
"drNNNN(`result`); <- e.g. t.replace('dr1006(', '').replace(')', '').replace(';', '')"
|
||||
return json.loads(t[7:-2])
|
||||
def login(self, user, pwd):
|
||||
interface = self.interface
|
||||
interface_def = self.interface_def
|
||||
u = "&user_account=" + user
|
||||
|
||||
url = self.urlloadUserInfo + u
|
||||
try:
|
||||
t = request_get_text(url, headers=self.header)
|
||||
except URLError as e:
|
||||
# e.reason may be Error or str
|
||||
if getattr(e.reason, "errno", None) == ENETUNREACH:
|
||||
return LoginStatus.no_wifi
|
||||
raise
|
||||
|
||||
j = self.json_from_drnnnn(t)
|
||||
# 检查是否付费
|
||||
if j["user_info"]["user_state"] == "正常" and j["user_info"]["available_flow"] in ("0MB", "无限制"):
|
||||
url = self.urlloadOnlineRecord + u
|
||||
# 获取在线设备
|
||||
t = request_get_text(url, headers=self.header)
|
||||
j1 = self.json_from_drnnnn(t)
|
||||
result = LoginStatus.unknown
|
||||
if j1['count'] <= self.online_device_limit:
|
||||
# 判断密码是否正确
|
||||
url = self.urllogin + u + "&user_password=" + pwd
|
||||
if interface_def:
|
||||
res = request_get_text(url)
|
||||
else:
|
||||
command = ["curl", url, "--interface", interface]
|
||||
res = subprocess.check_output(command, text=True)
|
||||
j_res = self.json_from_drnnnn(res)
|
||||
"""
|
||||
dr1003({"result":0,"msg":"账号不存在","ret_code":1});
|
||||
dr1003({"result":0,"msg":"密码错误","ret_code":1});
|
||||
dr1003({"result":1,"msg":"Portal协议认证成功！"});
|
||||
dr1003({"result":0,"msg":"IP: 10.142.5.160 已经在线！","ret_code":2});
|
||||
"""
|
||||
msg = ""
|
||||
if not interface_def:
|
||||
self.log(f"使用接口 {interface} 进行请求")
|
||||
msg += f"接口 {interface}： "
|
||||
|
||||
j_msg = j_res["msg"]
|
||||
if "密码错误" in j_msg:
|
||||
msg += f"{user} {pwd} 密码错误"
|
||||
result = LoginStatus.bad_pwd
|
||||
elif "已经在线" in j_msg:
|
||||
msg += "正常在线！"
|
||||
result = LoginStatus.used_online
|
||||
elif "认证成功" in j_msg:
|
||||
msg += f"使用账号{user}登录成功!"
|
||||
result = LoginStatus.succ
|
||||
if msg != "":
|
||||
self.log(msg)
|
||||
return result
|
||||
else:
|
||||
return LoginStatus.not_unlimit
|
||||
|
||||
def login_till_succ(self, user_pwd_gen, bad_user_callback=lambda u: None):
|
||||
"""
|
||||
user_pwd_gen will be called multiply times until login succeeds.
|
||||
"""
|
||||
not_succ = True
|
||||
while not_succ:
|
||||
(user, pwd) = user_pwd_gen()
|
||||
ret = self.login(user, pwd)
|
||||
if ret == LoginStatus.no_wifi:
|
||||
raise RuntimeError("not connect to wifi yet")
|
||||
not_succ = ret < 0
|
||||
|
||||
if ret in {LoginStatus.bad_pwd, LoginStatus.not_unlimit}:
|
||||
bad_user_callback(user)
|
||||
|
||||
|
||||
def main(self, file, warn, sep=' '):
|
||||
interface_def = self.interface_def
|
||||
if interface_def:
|
||||
warn("no interface given by -i or --interface, use default route")
|
||||
|
||||
# 检测网络联通性
|
||||
if self.check_connectivity():
|
||||
return
|
||||
|
||||
with open(file, 'r') as f:
|
||||
user_pwd = [i.rstrip("\r\n").split(sep) for i in f.readlines()]
|
||||
|
||||
# login
|
||||
user_pwd_error_or_not_unlimit_combo_idx = set()
|
||||
cur_lku_user_idx = 0
|
||||
def user_pwd_getter():
|
||||
# 随机选择，防止前面有密码错误的用户卡死
|
||||
user_index = random.randrange(len(user_pwd))
|
||||
if user_index in user_pwd_error_or_not_unlimit_combo_idx:
|
||||
return user_pwd_getter()
|
||||
(user, pwd) = user_pwd[user_index]
|
||||
return (user, pwd)
|
||||
|
||||
def bad_user_callback(_):
|
||||
user_pwd_error_or_not_unlimit_combo_idx.add(cur_lku_user_idx)
|
||||
|
||||
self.login_till_succ(user_pwd_getter, bad_user_callback)
|
||||
|
||||
# 删除密码错误或不满足要求的用户
|
||||
with open(file, 'w') as f:
|
||||
for i in range(len(user_pwd)):
|
||||
if i in user_pwd_error_or_not_unlimit_combo_idx:
|
||||
user_pwd_error_or_not_unlimit_combo_idx.remove(i)
|
||||
else:
|
||||
t = user_pwd[i]
|
||||
f.write(t[0])
|
||||
f.write(' ')
|
||||
f.write(t[1])
|
||||
f.write('\n')
|
||||
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
import warnings
|
||||
import syslog
|
||||
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
||||
parser.add_argument("-i", "--interface", help="网卡接口名称. If not given, use default route", default=None)
|
||||
parser.add_argument("-f", "--file", help="用户密码文件路径, one record each line", default="user_pwd.txt")
|
||||
parser.add_argument("-S", "--no-syslog", action="store_true", help="use print over syslog for log")
|
||||
parser.add_argument("-s", "--sep", help="separator used in file between user and password", default=' ')
|
||||
parser.add_argument("-n", "--online-device-limit", help="only connect to it if its online device is not more than ONLINE_DEVICE_LIMIT", type=int, default=2)
|
||||
# TODO: support auto means 3 if only one account available but 0 otherwise
|
||||
# Or connect the account with least online device number
|
||||
args = parser.parse_args()
|
||||
|
||||
log = lambda *a: print(*a)
|
||||
if args.no_syslog:
|
||||
log = lambda *a: syslog.syslog(' '.join(map(str, a)))
|
||||
n = args.online_device_limit
|
||||
Loginer(args.interface, log=log, online_device_limit=n).main(args.file, warnings.warn, sep=args.sep)
|
||||
#!/usr/bin/env python3
|
||||
from sys import maxsize
|
||||
from enum import IntEnum, auto
|
||||
from urllib.error import URLError
|
||||
from urllib import request
|
||||
from errno import ENETUNREACH
|
||||
import json
|
||||
import random
|
||||
import subprocess
|
||||
|
||||
def request_get_text(url, headers={}):
|
||||
req = request.Request(url, headers=headers)
|
||||
with request.urlopen(req) as response:
|
||||
return response.read().decode("utf-8")
|
||||
|
||||
class LoginStatus(IntEnum):
|
||||
unknown = -maxsize-1
|
||||
no_wifi = -3
|
||||
not_unlimit = -2
|
||||
bad_pwd = -1
|
||||
succ = auto()
|
||||
used_online = auto()
|
||||
|
||||
class Loginer:
|
||||
def __init__(self,
|
||||
interface = None,
|
||||
online_device_limit = 2,
|
||||
log = print,
|
||||
):
|
||||
self.interface = interface
|
||||
self.online_device_limit = online_device_limit
|
||||
self.log = log
|
||||
self.interface_def = interface is None
|
||||
def check_connectivity(self):
|
||||
interface = self.interface
|
||||
if not self.interface_def:
|
||||
command = ["ping", "-I", interface]
|
||||
host = "223.5.5.5"
|
||||
command += ["-c", "1", host] # -n 1 on windows, -c 1 on linux
|
||||
try:
|
||||
result = subprocess.check_call(command, timeout=2)
|
||||
return result == 0
|
||||
except subprocess.TimeoutExpired:
|
||||
self.log(f"接口 {interface}：Ping to {host} timed out.")
|
||||
return False
|
||||
except subprocess.CalledProcessError:
|
||||
return False
|
||||
except Exception as e:
|
||||
self.log(f"接口 {interface}：Error during ping: {e}")
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
request_get_text("http://connect.rom.miui.com/generate_204")
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
referrer = "http://192.168.101.201"
|
||||
header = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36 Edg/105.0.1343.33",
|
||||
"Referer": referrer
|
||||
}
|
||||
|
||||
|
||||
@staticmethod
|
||||
def url(last, dr, v):
|
||||
return "/eportal/portal/" + last + "?callback=dr" + dr + "&v=" + v + \
|
||||
"&lang=zh-CN&wlan_user_mac=000000000000&jsVersion=4.1"
|
||||
|
||||
ip0 = "&wlan_user_ip=0.0.0.0"
|
||||
index_qs = "&program_index=ctshNw1713845951&page_index=V5fmKw1713845966"
|
||||
urlloadUserInfo = referrer + ":801" + url("page/loadUserInfo", "1004", "599") + ip0 + index_qs
|
||||
time_qs = "&start_time=2010-01-01&end_time=2100-01-01&start_rn=1&end_rn=5"
|
||||
urlloadOnlineRecord = referrer + ":801" + url("page/loadOnlineRecord", "1006", "2399") + time_qs + index_qs + "&wlan_user_ip=10.169.0.241"
|
||||
urllogin = "https://xha.ouc.edu.cn:802" + url("login", "1003", "2425") + ip0 + "&login_method=1&wlan_user_ipv6=&wlan_ac_ip=&wlan_ac_name=&terminal_type=1"
|
||||
|
||||
@staticmethod
|
||||
def json_from_drnnnn(t):
|
||||
"drNNNN(`result`); <- e.g. t.replace('dr1006(', '').replace(')', '').replace(';', '')"
|
||||
return json.loads(t[7:-2])
|
||||
def login(self, user, pwd):
|
||||
interface = self.interface
|
||||
interface_def = self.interface_def
|
||||
u = "&user_account=" + user
|
||||
|
||||
url = self.urlloadUserInfo + u
|
||||
try:
|
||||
t = request_get_text(url, headers=self.header)
|
||||
except URLError as e:
|
||||
# e.reason may be Error or str
|
||||
if getattr(e.reason, "errno", None) == ENETUNREACH:
|
||||
return LoginStatus.no_wifi
|
||||
raise
|
||||
|
||||
j = self.json_from_drnnnn(t)
|
||||
# 检查是否付费
|
||||
if j["user_info"]["user_state"] == "正常" and j["user_info"]["available_flow"] in ("0MB", "无限制"):
|
||||
url = self.urlloadOnlineRecord + u
|
||||
# 获取在线设备
|
||||
t = request_get_text(url, headers=self.header)
|
||||
j1 = self.json_from_drnnnn(t)
|
||||
result = LoginStatus.unknown
|
||||
if j1['count'] <= self.online_device_limit:
|
||||
# 判断密码是否正确
|
||||
url = self.urllogin + u + "&user_password=" + pwd
|
||||
if interface_def:
|
||||
res = request_get_text(url)
|
||||
else:
|
||||
command = ["curl", url, "--interface", interface]
|
||||
res = subprocess.check_output(command, text=True)
|
||||
j_res = self.json_from_drnnnn(res)
|
||||
"""
|
||||
dr1003({"result":0,"msg":"账号不存在","ret_code":1});
|
||||
dr1003({"result":0,"msg":"密码错误","ret_code":1});
|
||||
dr1003({"result":1,"msg":"Portal协议认证成功！"});
|
||||
dr1003({"result":0,"msg":"IP: 10.142.5.160 已经在线！","ret_code":2});
|
||||
"""
|
||||
msg = ""
|
||||
if not interface_def:
|
||||
self.log(f"使用接口 {interface} 进行请求")
|
||||
msg += f"接口 {interface}： "
|
||||
|
||||
j_msg = j_res["msg"]
|
||||
if "密码错误" in j_msg:
|
||||
msg += f"{user} {pwd} 密码错误"
|
||||
result = LoginStatus.bad_pwd
|
||||
elif "已经在线" in j_msg:
|
||||
msg += "正常在线！"
|
||||
result = LoginStatus.used_online
|
||||
elif "认证成功" in j_msg:
|
||||
msg += f"使用账号{user}登录成功!"
|
||||
result = LoginStatus.succ
|
||||
if msg != "":
|
||||
self.log(msg)
|
||||
return result
|
||||
else:
|
||||
return LoginStatus.not_unlimit
|
||||
|
||||
def login_till_succ(self, user_pwd_gen, bad_user_callback=lambda u: None):
|
||||
"""
|
||||
user_pwd_gen will be called multiply times until login succeeds.
|
||||
"""
|
||||
not_succ = True
|
||||
while not_succ:
|
||||
(user, pwd) = user_pwd_gen()
|
||||
ret = self.login(user, pwd)
|
||||
if ret == LoginStatus.no_wifi:
|
||||
raise RuntimeError("not connect to wifi yet")
|
||||
not_succ = ret < 0
|
||||
|
||||
if ret in {LoginStatus.bad_pwd, LoginStatus.not_unlimit}:
|
||||
bad_user_callback(user)
|
||||
|
||||
|
||||
def main(self, file, warn, sep=' '):
|
||||
interface_def = self.interface_def
|
||||
if interface_def:
|
||||
warn("no interface given by -i or --interface, use default route")
|
||||
|
||||
# 检测网络联通性
|
||||
if self.check_connectivity():
|
||||
return
|
||||
|
||||
with open(file, 'r') as f:
|
||||
user_pwd = [i.rstrip("\r\n").split(sep) for i in f.readlines()]
|
||||
|
||||
# login
|
||||
user_pwd_error_or_not_unlimit_combo_idx = set()
|
||||
cur_lku_user_idx = 0
|
||||
def user_pwd_getter():
|
||||
# 随机选择，防止前面有密码错误的用户卡死
|
||||
user_index = random.randrange(len(user_pwd))
|
||||
if user_index in user_pwd_error_or_not_unlimit_combo_idx:
|
||||
return user_pwd_getter()
|
||||
(user, pwd) = user_pwd[user_index]
|
||||
return (user, pwd)
|
||||
|
||||
def bad_user_callback(_):
|
||||
user_pwd_error_or_not_unlimit_combo_idx.add(cur_lku_user_idx)
|
||||
|
||||
self.login_till_succ(user_pwd_getter, bad_user_callback)
|
||||
|
||||
# 删除密码错误或不满足要求的用户
|
||||
with open(file, 'w') as f:
|
||||
for i in range(len(user_pwd)):
|
||||
if i in user_pwd_error_or_not_unlimit_combo_idx:
|
||||
user_pwd_error_or_not_unlimit_combo_idx.remove(i)
|
||||
else:
|
||||
t = user_pwd[i]
|
||||
f.write(t[0])
|
||||
f.write(' ')
|
||||
f.write(t[1])
|
||||
f.write('\n')
|
||||
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
import warnings
|
||||
import syslog
|
||||
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
||||
parser.add_argument("-i", "--interface", help="网卡接口名称. If not given, use default route", default=None)
|
||||
parser.add_argument("-f", "--file", help="用户密码文件路径, one record each line", default="user_pwd.txt")
|
||||
parser.add_argument("-S", "--no-syslog", action="store_true", help="use print over syslog for log")
|
||||
parser.add_argument("-s", "--sep", help="separator used in file between user and password", default=' ')
|
||||
parser.add_argument("-n", "--online-device-limit", help="only connect to it if its online device is not more than ONLINE_DEVICE_LIMIT", type=int, default=2)
|
||||
# TODO: support auto means 3 if only one account available but 0 otherwise
|
||||
# Or connect the account with least online device number
|
||||
args = parser.parse_args()
|
||||
|
||||
log = lambda *a: print(*a)
|
||||
if args.no_syslog:
|
||||
log = lambda *a: syslog.syslog(' '.join(map(str, a)))
|
||||
n = args.online_device_limit
|
||||
Loginer(args.interface, log=log, online_device_limit=n).main(args.file, warnings.warn, sep=args.sep)
|
||||
|
||||
Reference in New Issue
Block a user