jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Raise exception when empty host_keys combined with reject policy

pull/12/head
Sheng 2018-03-15 01:46:54 +08:00
parent d6083a9608
commit 95e7332852
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -317,6 +317,10 @@ def main():
base_dir = os.path.dirname(__file__) base_dir = os.path.dirname(__file__)
host_file = os.path.join(base_dir, 'known_hosts') host_file = os.path.join(base_dir, 'known_hosts')
host_keys = create_host_file(host_file) host_keys = create_host_file(host_file)
policy_class = get_policy_class(options.policy)
if not host_keys and policy_class is paramiko.client.RejectPolicy:
raise ValueError('Empty known_hosts with reject policy?')
settings = { settings = {
'template_path': os.path.join(base_dir, 'templates'), 'template_path': os.path.join(base_dir, 'templates'),
@ -335,7 +339,7 @@ def main():
debug=options.debug, debug=options.debug,
host_file=host_file, host_file=host_file,
host_keys=host_keys, host_keys=host_keys,
policy=get_policy_class(options.policy)() policy=policy_class()
) )
app = tornado.web.Application(handlers, **settings) app = tornado.web.Application(handlers, **settings)
app.listen(options.port, options.address) app.listen(options.port, options.address)