HTTP host is changeable

This commit is contained in:
leafee98 2022-09-13 10:38:14 +08:00
parent 08680de640
commit 0f5aa9ce29
2 changed files with 167 additions and 142 deletions

View file

@ -1,142 +0,0 @@
#!/usr/bin/python3
import urllib.parse
import urllib.request
import urllib.response
import logging
import re
import argparse
class login_manager:
logger = logging.getLogger('login_manager')
login_url = 'http://10.1.206.13/a70.htm'
logout_url = 'http://10.1.206.13/F.htm'
def __init__(self, username: str = '', password: str = ''):
self.username = username
self.password = password
@staticmethod
def generate_post_data(username: str, password: str):
data_template = {
'DDDDD': None,
'upass': None,
'R1': 0,
'R2': '',
'R3': 0,
'R6': 0,
'para': 00,
'0MKKey': 123456,
'buttonClicked': '',
'redirect_url': '',
'err_flag': '',
'username': '',
'password': '',
'user': '',
'cmd': '',
'Login': ''
}
data_template['DDDDD'] = username
data_template['upass'] = password
login_manager.generate_post_data_hook(data_template)
login_manager.logger.debug('the generated post data is %s', str(data_template))
return bytes(urllib.parse.urlencode(data_template), 'utf-8')
@staticmethod
def generate_post_data_hook(post_data: dict) -> None:
username = post_data['DDDDD']
username += '@bistu'
post_data['DDDDD'] = username
def login(self):
headers = {
'Content-Type': 'application/x-www-form-urlencoded',
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
'Accept-Language': 'en-US,zh-CN;q=0.5',
'Accept-Encoding': 'gzip, deflate'
}
post_data = self.generate_post_data(self.username, self.password)
self.logger.debug('the dest url is: %s', self.login_url)
self.logger.debug('the headers is: %s', str(headers))
self.logger.debug('the encoded post data is %s', str(post_data))
req = urllib.request.Request(url=self.login_url, data=post_data, headers=headers, method='POST')
self.logger.info('sending the login request...')
with urllib.request.urlopen(req) as response:
response_body = response.read().decode('gbk')
html_title = re.search('<title>(.*)</title>', response_body)[1]
self.logger.info('http_status: %d', response.status)
self.logger.info('title_of_response_html: %s', html_title)
self.logger.info('login_state: %s', 'success' if html_title == '认证成功页' else 'fail')
@staticmethod
def logout():
req = urllib.request.Request(url=login_manager.logout_url)
login_manager.logger.info('sending the logout request...')
with urllib.request.urlopen(req) as response:
response_body = response.read().decode('gbk')
html_title = re.search('<title>(.*)</title>', response_body)[1]
login_manager.logger.info('http_status: %d', response.status)
login_manager.logger.info('title_of_response_html: %s', html_title)
login_manager.logger.info('logout_state: %s', 'success' if html_title == '信息页' else 'fail')
def load_auth_info(path: str):
with open(path, 'r', encoding='utf-8') as file:
username = file.readline().strip()
password = file.readline().strip()
return username, password
def main():
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger('main')
parser = argparse.ArgumentParser()
parser.description = 'login or logout bistu wifi'
parser.epilog = 'the config file format(without quote): "{username}\\n{password}". The "\\n" is a newline charactor'
parser.add_argument('username', nargs='?',
type=str, help='username you use to login bistu wifi, logout if not provided')
parser.add_argument('password', nargs='?',
type=str, help='password you use to login bistu wifi, logout if not provided')
parser.add_argument('-o', '--logout', nargs='?', type=bool,
help='logout bistu wifi')
parser.add_argument('-c', '--config', nargs='?', type=str,
help='load username and password from config file')
args = parser.parse_args()
logger.debug('parse argument: username= %s', args.username)
logger.debug('parse argument: password= %s', args.password)
logger.debug('parse argument: logout= %s', args.logout)
if args.config is not None:
username, password = load_auth_info(args.config)
manager = login_manager(username, password)
manager.login()
elif args.username is not None and args.password is not None:
manager = login_manager(args.username, args.password)
manager.login()
elif args.logout:
login_manager.logout()
else:
parser.print_help()
if __name__ == '__main__':
main()

167
main.py Normal file
View file

@ -0,0 +1,167 @@
#!/usr/bin/python3
import urllib.parse
import urllib.request
import urllib.response
import logging
import re
import argparse
logger = logging.getLogger('login_manager')
def generate_post_data(username: str, password: str):
'''
Generate bytes to send to through HTTP POST, an example from browser:
DDDDD=2021020000&upass=Bistu000000&R1=0&R2=&R3=0&R6=0&para=00&0MKKey=123456&buttonClicked=&redirect_url=&err_flag=&username=&password=&user=&cmd=&Login=
'''
data_template = {
'DDDDD': None,
'upass': None,
'R1': 0,
'R2': '',
'R3': 0,
'R6': 0,
'para': 00,
'0MKKey': 123456,
'buttonClicked': '',
'redirect_url': '',
'err_flag': '',
'username': '',
'password': '',
'user': '',
'cmd': '',
'Login': ''
}
data_template['DDDDD'] = username
data_template['upass'] = password
# login_manager.generate_post_data_hook(data_template)
logger.debug('the generated post data is %s', str(data_template))
return bytes(urllib.parse.urlencode(data_template), 'utf-8')
def login(username: str, password: str, host: str = '10.1.206.13'):
'''
There are some other host, the known are:
- 10.1.206.3 (default)
- 192.168.211.3
'''
headers = {
'Content-Type': 'application/x-www-form-urlencoded',
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
'Accept-Language': 'en-US,zh-CN;q=0.5',
'Accept-Encoding': 'gzip, deflate'
}
post_data = generate_post_data(username, password)
login_url = f'http://{host}/a70.htm'
logger.debug('the dest url is: %s', login_url)
logger.debug('the headers is: %s', str(headers))
logger.debug('the encoded post data is %s', str(post_data))
req = urllib.request.Request(url=login_url, data=post_data, headers=headers, method='POST')
logger.info('sending the login request...')
with urllib.request.urlopen(req) as response:
response_body = response.read().decode('gbk')
html_title = re.search('<title>(.*)</title>', response_body)[1]
logger.info('http_status: %d', response.status)
logger.info('title_of_response_html: %s', html_title)
logger.info('login_state: %s', 'success' if html_title == '认证成功页' else 'fail')
def logout(host: str = '10.1.206.13'):
'''
param host: see login host
'''
logout_url = f'http://{host}/F.htm'
req = urllib.request.Request(url=logout_url)
logger.info('sending the logout request...')
with urllib.request.urlopen(req) as response:
response_body = response.read().decode('gbk')
html_title = re.search('<title>(.*)</title>', response_body)[1]
logger.info('http_status: %d', response.status)
logger.info('title_of_response_html: %s', html_title)
logger.info('logout_state: %s', 'success' if html_title == '信息页' else 'fail')
#@staticmethod
#def generate_post_data_hook(post_data: dict) -> None:
# username = post_data['DDDDD']
# username += '@bistu'
# post_data['DDDDD'] = username
def load_auth_info(path: str):
with open(path, 'r', encoding='utf-8') as file:
username = file.readline().strip()
password = file.readline().strip()
host = file.readline().strip()
if host == '':
host = None
return username, password, host
def main():
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger('main')
parser = argparse.ArgumentParser()
parser.description = 'login or logout bistu wifi'
parser.epilog = 'the config file format(without quote): "{username}\\n{password}". The "\\n" is a newline charactor'
parser.add_argument('-u', '--username', nargs='?',
type=str, help='username you use to login bistu wifi, logout if not provided')
parser.add_argument('-p', '--password', nargs='?',
type=str, help='password you use to login bistu wifi, logout if not provided')
parser.add_argument('--host', nargs='?', default=None,
type=str, help='host to send login/logout request to, default use 10.1.206.13')
parser.add_argument('-o', '--logout', nargs='?', type=bool, default=False,
help='logout bistu wifi')
parser.add_argument('-c', '--config', nargs='?', type=str,
help='load info(username, password, host) from config file')
args = parser.parse_args()
logger.debug('parse argument: username= %s', args.username)
logger.debug('parse argument: password= %s', args.password)
logger.debug('parse argument: host= %s', args.host)
logger.debug('parse argument: logout= %s', args.logout)
if args.config is not None:
username, password, host = load_auth_info(args.config)
if host is None:
login(username, password)
else:
login(username, password, host)
elif args.username is not None and args.password is not None:
if args.host is None:
login(args.username, args.password)
else:
login(args.username, args.password, args.host)
elif args.logout:
if args.host is None:
logout()
else:
logout(args.host)
else:
parser.print_help()
if __name__ == '__main__':
main()