dece/Bebop
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Activity

tofu: fix bug in save_cert_stash

Browse source
This commit is contained in:
dece 2021-03-13 20:37:53 +01:00
parent 05c54eff48
commit 1ff23e58e8
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
bebop/tofu.py
View file

@ -8,6 +8,7 @@ import datetime
import hashlib import hashlib
import re import re
from enum import Enum from enum import Enum
from pathlib import Path
import asn1crypto.x509 import asn1crypto.x509
@ -15,7 +16,7 @@ import asn1crypto.x509
STASH_LINE_RE = re.compile(r"(\S+) (\S+) (\S+) (\d+)") STASH_LINE_RE = re.compile(r"(\S+) (\S+) (\S+) (\d+)")
def load_cert_stash(stash_path): def load_cert_stash(stash_path: Path):
"""Load the certificate stash from the file, or None on error. """Load the certificate stash from the file, or None on error.
The stash is a dict with host names as keys and tuples as values. Tuples The stash is a dict with host names as keys and tuples as values. Tuples
@ -42,7 +43,7 @@ def load_cert_stash(stash_path):
return stash return stash
def save_cert_stash(stash, stash_path): def save_cert_stash(stash: dict, stash_path: Path):
"""Save the certificate stash.""" """Save the certificate stash."""
try: try:
with open(stash_path, "wt") as stash_file: with open(stash_path, "wt") as stash_file:
@ -51,7 +52,7 @@ def save_cert_stash(stash, stash_path):
if not is_permanent: if not is_permanent:
continue continue
entry_line = f"{name} {algo} {fingerprint} {timestamp}\n" entry_line = f"{name} {algo} {fingerprint} {timestamp}\n"
stash_path.write(entry_line) stash_file.write(entry_line)
except (OSError, ValueError): except (OSError, ValueError):
pass pass