83 lines
2.6 KiB
Python
Executable File
83 lines
2.6 KiB
Python
Executable File
# This file will help to serve as a starting point for using the rest of the tools
|
|
# Things we want to figure out
|
|
# 1) Is your key active?
|
|
# 2) If active, can you read monitoring configs, can you write?
|
|
# 3) Okay, you can read monitoring configs. We recommend things to avoid. Want to go further? Use write access to disable (if applicable)
|
|
# 4) Don't want to do anything with monitoring? That's fine, let's guide you through figuring out what your access looks like
|
|
# 5) Help with a printout of options from this point forward
|
|
|
|
import boto3
|
|
import argparse
|
|
import os
|
|
from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY
|
|
from botocore.exceptions import ClientError
|
|
from modules import *
|
|
import sys
|
|
|
|
sys.path.append("modules")
|
|
for module in all_modules:
|
|
exec("from %s import *"%module)
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-s", "--step", help="list the step you would like to run",
|
|
action="store", type=str, required=True)
|
|
parser.add_argument("-t", "--target", help="Give your target a name so we can track results",
|
|
action="store", type=str, required=True)
|
|
parser.add_argument("-a", "--arguments", help="Provide a list of arguments, comma separated. Ex: arg1,arg2,arg3",
|
|
action="store", type=str, required=False)
|
|
parser.add_argument("-l", "--list", help="list steps", action="store_true")
|
|
parser.add_argument("-v", "--verbosity", help="increase output verbosity",
|
|
action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
def perform_credential_check():
|
|
try:
|
|
client = boto3.client("sts", aws_access_key_id=AWS_ACCESS_KEY_ID, aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
|
|
account_id = client.get_caller_identity()["Account"]
|
|
except ClientError as e:
|
|
print("The AWS Access Keys are not valid/active")
|
|
#exit(1)
|
|
|
|
def step_recon():
|
|
print("!!!")
|
|
|
|
def method_create():
|
|
try:
|
|
arg = globals()["step_" + args.step]
|
|
return arg
|
|
except KeyError:
|
|
print("That step does not exist")
|
|
exit(1)
|
|
|
|
# Need to figure out if we have keys in the ENV or not
|
|
if AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY:
|
|
perform_credential_check()
|
|
else:
|
|
print("Please supply keys as outlined in our README.md file")
|
|
#exit(1)
|
|
|
|
if (args.list):
|
|
pass
|
|
|
|
|
|
# arg_list has to be defined otherwise will cause an exception
|
|
arg_list = None
|
|
|
|
if (args.arguments):
|
|
arg_list = args.arguments.split(',')
|
|
|
|
# We need the user to tell us the step they want to proceed on
|
|
if (args.step):
|
|
arg = method_create()
|
|
if callable(arg):
|
|
if arg_list:
|
|
arg(arg_list)
|
|
else:
|
|
arg()
|
|
|
|
|
|
# Allow the user to specify verbosity for debugging
|
|
if (args.verbosity):
|
|
print("Verbosity is enabled")
|