mirror of https://github.com/PurpleI2P/pyseeder
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
4.5 KiB
125 lines
4.5 KiB
#! /usr/bin/env python3 |
|
import os, os.path |
|
import sys |
|
from getpass import getpass |
|
import argparse |
|
|
|
from pyseeder.crypto import keygen |
|
from pyseeder.su3file import SU3File |
|
import pyseeder.transport |
|
|
|
from pyseeder.utils import PyseederException |
|
|
|
def keygen_action(args): |
|
"""Sub-command to generate keys""" |
|
priv_key_password = getpass("Set private key password: ").encode("utf-8") |
|
keygen(args.cert, args.private_key, priv_key_password, args.signer_id) |
|
|
|
def reseed_action(args): |
|
"""Sub-command to generate reseed file""" |
|
priv_key_password = input().encode("utf-8") |
|
su3file = SU3File(args.signer_id) |
|
su3file.reseed(args.netdb) |
|
su3file.write(args.outfile, args.private_key, priv_key_password) |
|
|
|
def transport_pull_action(args): |
|
"""Sub-command for downloading su3 file""" |
|
import random |
|
random.shuffle(args.urls) |
|
|
|
for u in args.urls: |
|
if pyseeder.transport.download(u, args.outfile): |
|
return True |
|
|
|
raise PyseederException("Failed to download su3 file") |
|
|
|
def transport_push_action(args): |
|
"""Sub-command for uploading su3 file with transports""" |
|
if not os.path.isfile(args.config): |
|
raise PyseederException("Can't read transports config file") |
|
|
|
import configparser |
|
config = configparser.ConfigParser() |
|
config.read(args.config) |
|
pyseeder.transport.upload(args.file, config) |
|
|
|
def main(): |
|
parser = argparse.ArgumentParser() |
|
subparsers = parser.add_subparsers(title="actions", |
|
help="Command to execute") |
|
|
|
kg_parser = subparsers.add_parser( |
|
"keygen", |
|
description="Generates keypair for your reseed", |
|
usage=""" |
|
%(prog)s --cert data/user_at_mail.i2p.crt \\ |
|
--private-key data/priv_key.pem --signer-id user@mail.i2p""" |
|
) |
|
kg_parser.add_argument("--signer-id", required=True, |
|
help="Identifier of certificate (example: user@mail.i2p)") |
|
kg_parser.add_argument("--private-key", default="data/priv_key.pem", |
|
help="RSA private key (default: data/priv_key.pem)") |
|
kg_parser.add_argument("--cert", required=True, |
|
help="Certificate (example: output/user_at_mail.i2p.crt)") |
|
kg_parser.set_defaults(func=keygen_action) |
|
|
|
|
|
rs_parser = subparsers.add_parser( |
|
"reseed", |
|
description="Creates su3 reseed file", |
|
usage=""" |
|
echo $YOUR_PASSWORD | %(prog)s --netdb /path/to/netDb \\ |
|
--private-key data/priv_key.pem --outfile output/i2pseeds.su3 \\ |
|
--signer-id user@mail.i2p""" |
|
) |
|
rs_parser.add_argument("--signer-id", required=True, |
|
help="Identifier of certificate (example: user@mail.i2p)") |
|
rs_parser.add_argument("--private-key", default="data/priv_key.pem", |
|
help="RSA private key (default: data/priv_key.pem)") |
|
rs_parser.add_argument("-o", "--outfile", default="output/i2pseeds.su3", |
|
help="Output file (default: output/i2pseeds.su3)") |
|
rs_parser.add_argument("--netdb", required=True, |
|
help="Path to netDb folder (example: ~/.i2pd/netDb)") |
|
rs_parser.set_defaults(func=reseed_action) |
|
|
|
|
|
tpull_parser = subparsers.add_parser( |
|
"transport.pull", |
|
description="Download su3 file from random reseed server", |
|
usage=""" |
|
%(prog)s --urls https://reseed.i2p-projekt.de/ \\ |
|
https://reseed.i2p.vzaws.com:8443/ \\ |
|
--outfile output/i2pseeds.su3""" |
|
) |
|
tpull_parser.add_argument("--urls", default=pyseeder.transport.RESEED_URLS, |
|
nargs="*", help="""Reseed URLs separated by space, default are |
|
mainline I2P (like https://reseed.i2p-projekt.de/)""") |
|
tpull_parser.add_argument("-o", "--outfile", default="output/i2pseeds.su3", |
|
help="Output file (default: output/i2pseeds.su3)") |
|
tpull_parser.set_defaults(func=transport_pull_action) |
|
|
|
|
|
tpush_parser = subparsers.add_parser( |
|
"transport.push", |
|
description="Upload su3 file with transports", |
|
usage="%(prog)s --config transports.ini --file output/i2pseeds.su3" |
|
) |
|
tpush_parser.add_argument("--config", default="transports.ini", |
|
help="Transports config file (default: transports.ini)") |
|
tpush_parser.add_argument("-f", "--file", default="output/i2pseeds.su3", |
|
help=".su3 file (default: output/i2pseeds.su3)") |
|
tpush_parser.set_defaults(func=transport_push_action) |
|
|
|
args = parser.parse_args() |
|
if hasattr(args, "func"): |
|
try: |
|
args.func(args) |
|
except PyseederException as pe: |
|
print("Pyseeder error: {}".format(pe)) |
|
sys.exit(1) |
|
else: |
|
parser.print_help() |
|
|
|
|
|
if __name__ == "__main__": |
|
main()
|
|
|