kopia lustrzana https://github.com/micropython/micropython-lib
49 wiersze
1.2 KiB
Python
49 wiersze
1.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import re
|
||
|
|
import csv
|
||
|
|
|
||
|
|
|
||
|
|
def add_with_filter(crts_path, filter_path):
|
||
|
|
filter_set = set()
|
||
|
|
name_owner = {}
|
||
|
|
with open(filter_path, "r", encoding="utf-8") as f:
|
||
|
|
csv_reader = csv.reader(f, delimiter=",")
|
||
|
|
|
||
|
|
# Skip header
|
||
|
|
next(csv_reader)
|
||
|
|
for row in csv_reader:
|
||
|
|
filter_set.add(row[1])
|
||
|
|
name_owner[row[1]] = row[0]
|
||
|
|
|
||
|
|
crt_str = []
|
||
|
|
with open(crts_path, "r", encoding="utf-8") as f:
|
||
|
|
crt_str = f.read()
|
||
|
|
|
||
|
|
# Split all certs into a list of (name, certificate string) tuples
|
||
|
|
pem_crts = re.findall(
|
||
|
|
r"(^.+?)\n(=+\n[\s\S]+?END CERTIFICATE-----\n)", crt_str, re.MULTILINE
|
||
|
|
)
|
||
|
|
|
||
|
|
filtered_crts = ""
|
||
|
|
filtered_crts = "CACERTS = b'''"
|
||
|
|
for name, crt in pem_crts:
|
||
|
|
if name in filter_set:
|
||
|
|
filtered_crts += f"\n{name_owner[name]}:{name}\n"
|
||
|
|
filtered_crts += crt
|
||
|
|
|
||
|
|
filtered_crts += "'''"
|
||
|
|
|
||
|
|
return filtered_crts
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# print(sys.argv)
|
||
|
|
cert_path, filt_path = sys.argv[1:3]
|
||
|
|
cert_bundle = add_with_filter(cert_path, filt_path)
|
||
|
|
with open("ca_bundle_mp.py", "w") as cafile:
|
||
|
|
cafile.write(cert_bundle)
|
||
|
|
print(cert_bundle)
|