104 lines
3.2 KiB
Python
Executable File
104 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
from collections import namedtuple
|
|
|
|
|
|
class Entry(namedtuple('Entry', ['host', 'algorithm', 'hash'])):
|
|
separator: str = '.'
|
|
|
|
def __str__(self):
|
|
return f'{self.host} {self.algorithm} {self.hash}'
|
|
|
|
@property
|
|
def key(self):
|
|
return f'{self.algorithm}{self.separator}{self.hash}'
|
|
|
|
|
|
def main(args):
|
|
filename: Path | None = getattr(args, 'known_hosts', None)
|
|
file_path = filename.resolve()
|
|
if not file_path.exists() or not file_path.is_file():
|
|
sys.exit(1)
|
|
entryMap = {}
|
|
|
|
with file_path.open(encoding='utf-8') as fp:
|
|
for lineNumber, line in enumerate(fp):
|
|
content = line.strip().split()
|
|
if len(content) == 0:
|
|
continue # empty line -> skip
|
|
if line[0].startswith('#') or len(content) != 3:
|
|
if not line[0].startswith('#'):
|
|
sys.stderr.write(f'could not parse a line {lineNumber}: "{line}"\n')
|
|
sys.stdout.write(line)
|
|
continue # comment or erroneous line -> skip
|
|
entry = Entry(*content)
|
|
entryMap.setdefault(entry.key, []).append(entry)
|
|
|
|
compactEntryList = []
|
|
|
|
for key, entryList in entryMap.items():
|
|
domainList = []
|
|
ipList = []
|
|
entry: Entry
|
|
for entry in entryList:
|
|
for domainOrIp in entry.host.split(','):
|
|
if len(domainOrIp) < 2:
|
|
sys.stderr.write(f'could not parse {domainOrIp}\n')
|
|
continue
|
|
start_loc = 0
|
|
if domainOrIp[0] == '[':
|
|
start_loc = 1
|
|
if 'a' <= domainOrIp[start_loc].lower() <= 'z':
|
|
domainList.append(domainOrIp)
|
|
else:
|
|
ipList.append(domainOrIp)
|
|
host = ','.join(sorted(domainList) + sorted(ipList))
|
|
compactEntryList.append(Entry(host, *key.split(Entry.separator)))
|
|
|
|
output: Path | None = getattr(args, 'output', None)
|
|
output_lines = '\n'.join(
|
|
map(
|
|
str,
|
|
sorted(
|
|
compactEntryList,
|
|
key=lambda entry: (
|
|
entry.host[1:] if entry.host.startswith('[') else entry.host
|
|
),
|
|
),
|
|
)
|
|
)
|
|
if output is None:
|
|
sys.stdout.write(output_lines)
|
|
else:
|
|
output_file = output.resolve()
|
|
if not output_file.parent.exists():
|
|
os.makedirs(output_file.parent)
|
|
if output_file.is_dir():
|
|
print('Can not write to directory.')
|
|
with output_file.open('w', encoding='utf-8') as wfp:
|
|
wfp.writelines(output_lines)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
|
)
|
|
parser.add_argument(
|
|
'-i',
|
|
'--known_hosts',
|
|
type=Path,
|
|
default=f"{os.path.expanduser('~')}/.ssh/known_hosts",
|
|
help='known_hosts file to read',
|
|
)
|
|
parser.add_argument('-o', '--output', type=Path, help='output file to write')
|
|
parser.set_defaults(func=main)
|
|
args = parser.parse_args()
|
|
args.func(args)
|