Kumi
a568cd2669
Introduce 'onion_importer.py' for importing .onion keys from directories to a MariaDB database and deleting the sources. Add 'onionator.py' to interface with mkp224o and store generated keys in the database based on filtering criteria. Include sample configurations and update .gitignore to exclude virtual environments, config files, and filters.
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
import argparse
|
|
import mysql.connector
|
|
import configparser
|
|
import subprocess
|
|
|
|
def main():
|
|
"""
|
|
Entry point of the script.
|
|
"""
|
|
# Set up the argument parser
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-c', '--config', help='path to the configuration file', default="config.ini")
|
|
args = parser.parse_args()
|
|
|
|
# Read the MariaDB server details from the configuration file
|
|
config = configparser.ConfigParser()
|
|
config.read(args.config)
|
|
|
|
# Connect to the database
|
|
conn = mysql.connector.connect(
|
|
host=config['database']['host'],
|
|
user=config['database']['user'],
|
|
password=config['database']['password'],
|
|
database=config['database']['database']
|
|
)
|
|
cursor = conn.cursor()
|
|
|
|
# Create the table if it doesn't already exist
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS onion_keys (hostname text, public_key text, secret_key text)
|
|
''')
|
|
|
|
# Start mkp224o
|
|
process = subprocess.Popen(['mkp224o', '-y', '-f', 'filter'], stdout=subprocess.PIPE)
|
|
|
|
# Run mkp224o until the user interrupts the script
|
|
while True:
|
|
try:
|
|
# Read the output from mkp224o in blocks of four lines
|
|
hostname = None
|
|
public_key = None
|
|
secret_key = None
|
|
while True:
|
|
line = process.stdout.readline().decode().strip()
|
|
if line == '---':
|
|
break
|
|
|
|
# Split the line into fields
|
|
fields = line.split(':', 1)
|
|
if len(fields) != 2:
|
|
continue
|
|
key, value = fields
|
|
key = key.strip()
|
|
value = value.strip()
|
|
|
|
# Extract the hostname, public key, and secret key from the output
|
|
if key == 'hostname':
|
|
hostname = value
|
|
elif key == 'hs_ed25519_public_key':
|
|
public_key = value
|
|
elif key == 'hs_ed25519_secret_key':
|
|
secret_key = value
|
|
|
|
if hostname:
|
|
# Insert the data into the database
|
|
cursor.execute('''
|
|
INSERT INTO onion_keys (hostname, public_key, secret_key) VALUES (%s, %s, %s)
|
|
''', (hostname, public_key, secret_key))
|
|
|
|
# Save the changes to the database
|
|
conn.commit()
|
|
except KeyboardInterrupt:
|
|
# If the user interrupts the script, break out of the loop
|
|
break
|
|
|
|
main()
|