webssh

commit 95e73328523fec6c79820fef0d16e948b415f04c
parent d6083a9608a5e2f7a95b224b93bbf2830ae8eda9
Author: Sheng <webmaster0115@gmail.com>
Date:   Thu, 15 Mar 2018 01:46:54 +0800

Raise exception when empty host_keys combined with reject policy

Diffstat:
M
main.py
 | 
6
+++++-

1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/main.py b/main.py
@@ -317,6 +317,10 @@ def main():
     base_dir = os.path.dirname(__file__)
     host_file = os.path.join(base_dir, 'known_hosts')
     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 = {
         'template_path': os.path.join(base_dir, 'templates'),
@@ -335,7 +339,7 @@ def main():
         debug=options.debug,
         host_file=host_file,
         host_keys=host_keys,
-        policy=get_policy_class(options.policy)()
+        policy=policy_class()
     )
     app = tornado.web.Application(handlers, **settings)
     app.listen(options.port, options.address)