def format_pks(keys):
# First, add header.
# This comes in the format of "info:1:{number-of-keys}"
data = ""
data += "info:1:{}\n".format(keys.total)
# Then, add the strings for each key.
for key in keys.query.all():
# Load keyinfo.
newkey = KeyInfo.pgp_dump(key.armored)
s1, s2 = newkey.to_pks()
data += s1 + '\n' + s2 + '\n'
return data
def force_reevaluate_all_keys():
"""Force the database to re-calculate all details from the armored key data. Useful if pgpdump has an update, or the internal calculations change."""
# Query the database for keys.
for key in db.Key.query.all():
print("Re-calculating {}".format(key.key_fp_id))
# Re-calculate key.
kin = KeyInfo.pgp_dump(key.armored)
# Create a new database object.
newkey = db.Key.from_keyinfo(kin)
# Re-set the ID.
newkey.id = key.id
# Add the key.
db.db.session.merge(newkey)
# Commit.
db.db.session.commit()
def import_from_dump(file):
"""Import from an SKS dump."""
# First, load a new GPG instance.
gpg = GPG(gnupghome="/tmp/skier-import")
# Read data in
dumpf = open(file, 'rb')
data = dumpf.read()
dump = pgpdump.BinaryData(data)
dumpf.close()
# Import the keydump into gpg
try:
gpg.import_keys(data)
except:
# Yeah, whatever.
pass
"""
# Read in packets
packets_flat = []
try:
for packet in dump.packets():
packets_flat.append(packet)
except Exception as e:
# Some SKS data is malformed.
print("SKS data in {} is malformed - truncating at packet {}".format(file, len(packets_flat)+1))
# Unflatten list into sublists, starting at each new Publickey packet.
packets = []
tmpl = []
for packet in packets_flat:
if isinstance(packet, pgpdump.packet.PublicKeyPacket) and not isinstance(packet, pgpdump.packet.PublicSubkeyPacket) and tmpl:
tmpl.insert(0, packet)
gpg_output = gpg.export_keys(keyids=[packet.fingerprint.decode()])
if gpg_output:
tmpl.append(gpg_output)
packets.append(tmpl)
tmpl = []
continue
else:
tmpl.append(packet)
packets.append(tmpl)
"""
armored = []
for key in gpg.list_keys():
tmpkey = gpg.export_keys(keyids=key['fingerprint'])
if not tmpkey:
print("{} is malformed - not importing".format(key['fingerprint']))
armored.append(tmpkey)
print("Importing {} keys".format(len(armored)))
"""
for gpack in packets:
keyinfo = KeyInfo.pgp_dump(None, packets=gpack)
dbob = db.Key.from_keyinfo(keyinfo)
db.db.session.add(dbob)
db.db.session.commit()
"""
for key in armored:
kyinfo = KeyInfo.pgp_dump(key)
if not kyinfo.fingerprint:
print("Key {} is malformed - cannot be imported".format(kyinfo))
else:
dbob = db.Key.from_keyinfo(kyinfo)
db.db.session.add(dbob)
db.db.session.commit()
shutil.rmtree("/tmp/skier-import")
