parser = argparse.ArgumentParser(description='PostGreScan')

parser.add_argument('targets', type=str)

parser.add_argument('-p', metavar='ports', type=str_ports, nargs='?', help='target port', default='5432', dest='port')

parser.add_argument('-u', metavar='username', type=str, nargs='?', help='Username', default=None, dest='username')

parser.add_argument('--pass', metavar='password', type=str, nargs='?', help='Password', default=None, dest='password')

parser.add_argument('--timeout', metavar='timeout', nargs='?', type=int, help='Connect timeout', default=5, dest='timeout')

# Actions

parser.add_argument("--dbs", action='store_true', help='List databases')

parser.add_argument('--cmd', metavar='command', type=str, nargs='?', help='Execute a command via PostgreSQL RCE techniques', default=None, dest='cmd')

# Bruteforce

parser.add_argument("--bruteforce", action='store_true', help='Enable bruteforce')

parser.add_argument('-U', metavar='username file', type=str, nargs='?', help='Username file (format username or username:password)', default=None, dest='username_file')

parser.add_argument('-P', metavar='password file', type=str, nargs='?', help='Password file', default=None, dest='password_file')

parser.add_argument('-W', metavar='number worker', nargs='?', type=int, help='Number of concurent workers for the bruteforce', default=5, dest='bruteforce_workers')

# Dispatcher arguments

parser.add_argument('-w', metavar='number worker', nargs='?', type=int, help='Number of concurent workers', default=10, dest='workers')

args = parser.parse_args()

static_inputs = {}

if args.port:

static_inputs['port'] = args.port

creds = {}

if args.username:

creds['username'] = args.username

if args.password:

creds['password'] = args.password

actions = {}

if args.dbs:

actions['list_dbs'] = {}

if args.cmd:

actions['cmd'] = {'command': args.cmd}

if args.bruteforce:

actions['bruteforce'] ={'username_file': args.username_file, 'password_file': args.password_file, 'workers': args.bruteforce_workers}

Output.setup()

postgrescan(args.targets, static_inputs, args.workers, actions, creds, args.timeout)

args = (actions, creds, timeout)