This commit is contained in:
darkclip 2024-01-27 03:17:59 +08:00
parent f8ff038ea1
commit bd4c7b8fef
2 changed files with 100 additions and 2 deletions

View File

@ -66,8 +66,9 @@ deb https://$sec_site/debian-security $OS_CODE-security main contrib non-free no
curl -fsSL "https://packagecloud.io/ookla/speedtest-cli/gpgkey" | \
gpg --dearmor | tee /etc/apt/trusted.gpg.d/ookla_speedtest-cli-archive-keyring.gpg > /dev/null
cp "$SCRIPTPATH/.zshrc" $HOME
usermod -s /usr/bin/zsh $(whoami)
cp "$SCRIPTPATH/.zshrc" $HOME
cp "$SCRIPTPATH/*.py" $HOME
}
install_pkg(){
@ -110,4 +111,4 @@ done;
if [ $INSTALL_ONLY -eq 0 ]; then
configure;
fi;
install_pkg;
install_pkg;

97
sort_known_hosts.py Executable file
View File

@ -0,0 +1,97 @@
#!/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('known_hosts', type=Path, 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)