From 95e73328523fec6c79820fef0d16e948b415f04c Mon Sep 17 00:00:00 2001 From: Sheng Date: Thu, 15 Mar 2018 01:46:54 +0800 Subject: [PATCH] Raise exception when empty host_keys combined with reject policy --- main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index f0f5564..f80982b 100644 --- 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)