def get_host_keys_settings(options): if not options.hostfile: host_keys_filename = os.path.join(base_dir, 'known_hosts') else: host_keys_filename = options.hostfile host_keys = load_host_keys(host_keys_filename) if not options.syshostfile: filename = os.path.expanduser('~/.ssh/known_hosts') else: filename = options.syshostfile system_host_keys = load_host_keys(filename) if not options.certs: certs_home = os.path.join(base_dir, 'webssh', 'certs') else: certs_home = options.certs ip_trusted_list = [] ip_black_list = [] for line in open(os.path.join(certs_home, "ip_trusted.conf")): list = line.split('=') if len(list) == 2: if list[1].strip() == '0': ip_black_list.append(list[0].strip()) else: ip_trusted_list.append(list[0].strip()) settings = dict(host_keys=host_keys, system_host_keys=system_host_keys, host_keys_filename=host_keys_filename, certs_home=certs_home, ip_trusted_list=ip_trusted_list, ip_black_list=ip_black_list) return settings
def test_load_host_keys(self): path = '/path-not-exists' host_keys = load_host_keys(path) self.assertFalse(host_keys) path = '/tmp' host_keys = load_host_keys(path) self.assertFalse(host_keys) path = os.path.join(base_dir, 'tests', 'known_hosts_example') host_keys = load_host_keys(path) self.assertEqual(host_keys, paramiko.hostkeys.HostKeys(path))
def test_load_host_keys(self): path = '/path-not-exists' host_keys = load_host_keys(path) self.assertFalse(host_keys) path = '/tmp' host_keys = load_host_keys(path) self.assertFalse(host_keys) path = 'tests/known_hosts_example' host_keys = load_host_keys(path) self.assertEqual(host_keys, paramiko.hostkeys.HostKeys(path))
def get_host_keys_settings(options): if not options.hostfile: host_keys_filename = os.path.join(base_dir, 'known_hosts') else: host_keys_filename = options.hostfile host_keys = load_host_keys(host_keys_filename) if not options.syshostfile: filename = os.path.expanduser('~/.ssh/known_hosts') else: filename = options.syshostfile system_host_keys = load_host_keys(filename) settings = dict(host_keys=host_keys, system_host_keys=system_host_keys, host_keys_filename=host_keys_filename) return settings
def test_get_host_keys_settings(self): options.hostfile = '' options.syshostfile = '' dic = get_host_keys_settings(options) filename = os.path.join(base_dir, 'known_hosts') self.assertEqual(dic['host_keys'], load_host_keys(filename)) self.assertEqual(dic['host_keys_filename'], filename) self.assertEqual( dic['system_host_keys'], load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))) options.hostfile = make_tests_data_path('known_hosts_example') options.syshostfile = make_tests_data_path('known_hosts_example2') dic2 = get_host_keys_settings(options) self.assertEqual(dic2['host_keys'], load_host_keys(options.hostfile)) self.assertEqual(dic2['host_keys_filename'], options.hostfile) self.assertEqual(dic2['system_host_keys'], load_host_keys(options.syshostfile))
def test_get_host_keys_settings(self): options.hostFile = '' options.sysHostFile = '' dic = get_host_keys_settings(options) filename = os.path.join(base_dir, 'known_hosts') self.assertEqual(dic['host_keys'], load_host_keys(filename)) self.assertEqual(dic['host_keys_filename'], filename) self.assertEqual( dic['system_host_keys'], load_host_keys(os.path.expanduser('~/.ssh/known_hosts')) ) options.hostFile = 'tests/known_hosts_example' options.sysHostFile = 'tests/known_hosts_example2' dic2 = get_host_keys_settings(options) self.assertEqual(dic2['host_keys'], load_host_keys(options.hostFile)) self.assertEqual(dic2['host_keys_filename'], options.hostFile) self.assertEqual(dic2['system_host_keys'], load_host_keys(options.sysHostFile))