/usr/lib/python2.7/dist-packages/pyhsm/ksm/db_export.py is in yhsm-yubikey-ksm 1.2.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | #
# Copyright (c) 2013-2014 Yubico AB
# See the file COPYING for licence statement.
#
"""
Export AEAD from database.
"""
import os
import sys
import errno
import argparse
import sqlalchemy
import pyhsm.aead_cmd
def insert_slash(string, every=2):
"""insert_slash insert / every 2 char"""
return os.path.join(string[i:i+every] for i in xrange(0, len(string), every))
def mkdir_p(path):
"""mkdir -p: creates path like mkdir -p"""
try:
os.makedirs(path)
except OSError as exc:
if exc.errno == errno.EEXIST and os.path.isdir(path):
pass
else: raise
def main():
parser = argparse.ArgumentParser(description='Import AEADs into the database')
parser.add_argument('path', help='filesystem path of where to put AEADs')
parser.add_argument('dburl', help='connection URL for the database')
args = parser.parse_args()
#set the path
path = args.path
if not os.path.isdir(path):
print("\nInvalid path, make sure it exists.\n")
return 2
#mysql url
databaseUrl = args.dburl
try:
#check database connection
engine = sqlalchemy.create_engine(databaseUrl)
#SQLAlchemy voodoo
metadata = sqlalchemy.MetaData()
aeadobj = sqlalchemy.Table('aead_table', metadata, autoload=True, autoload_with=engine)
connection = engine.connect()
except:
print("FATAL: Database connect failure")
return 1
aead = None
nonce = None
key_handle = None
aead = pyhsm.aead_cmd.YHSM_GeneratedAEAD(nonce, key_handle, aead)
#get data from the database
result = connection.execute("SELECT * from aead_table")
#cycle through resutls
for row in result:
#read values row by row
aead.data = row['aead']
publicId = row['public_id']
aead.key_handle = row['keyhandle']
aead.nonce = row['nonce']
aead_dir = os.path.join(path, str(hex(aead.key_handle)).rstrip('L'), insert_slash(publicId))
#sanitize path
aead_dir = os.path.normpath(aead_dir)
#create path
mkdir_p(aead_dir)
#write the file in the path
pyhsm.aead_cmd.YHSM_GeneratedAEAD.save(aead, os.path.join(aead_dir, publicId))
#close connection
connection.close()
if __name__ == '__main__':
sys.exit(main())
|