mirror of
https://github.com/cisagov/snafflepy
synced 2026-09-24 18:22:23 +00:00
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
import argparse
|
|
import sys
|
|
import logging
|
|
|
|
from snaffcore.go_snaffle import *
|
|
from snaffcore.utilities import *
|
|
from snaffcore.logger import *
|
|
|
|
log = logging.getLogger('snafflepy')
|
|
log.setLevel(logging.INFO)
|
|
|
|
|
|
def parse_arguments():
|
|
syntax_error = False
|
|
print("SnafflePy by @robert-todora")
|
|
|
|
parser = argparse.ArgumentParser(
|
|
add_help=True, prog='snaffler.py', description='A "port" of Snaffler in python')
|
|
parser.add_argument("targets", nargs='+', type=make_targets,
|
|
help="IPs, hostnames, CIDR ranges, or files contains targets to snaffle")
|
|
parser.add_argument("-u", "--username", metavar='username',
|
|
type=str, help="domain username")
|
|
parser.add_argument("-p", "--password", metavar='password',
|
|
type=str, help="password for domain user")
|
|
parser.add_argument("-d", "--domain", metavar='domain',
|
|
default="", help="FQDN domain to authenticate to")
|
|
parser.add_argument("-H", "--hash", metavar='hash',
|
|
default="", help="NT hash for authentication")
|
|
parser.add_argument("-v", "--verbose",
|
|
action='store_true', help="Show more info")
|
|
# parser.add_argument("-e", "--exclude", )
|
|
|
|
# TODO
|
|
parser.add_argument("-i", "--no-discovery", action='store_true',
|
|
help="Disables computer and share discovery (more stealthy)")
|
|
parser.add_argument("-n", "--disable-computer-discovery", action='store_true',
|
|
help="Disable computer discovery, requires a list of hosts to do discovery on")
|
|
|
|
try:
|
|
if len(sys.argv) <= 1:
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
except argparse.ArgumentError as e:
|
|
syntax_error = True
|
|
log.error(e)
|
|
log.error('Check your syntax')
|
|
|
|
finally:
|
|
if syntax_error:
|
|
parser.print_help()
|
|
sys.exit(2)
|
|
else:
|
|
options = parser.parse_args()
|
|
|
|
# TODO
|
|
if options.verbose:
|
|
log.setLevel('DEBUG')
|
|
|
|
# TODO
|
|
if options.hash:
|
|
pass
|
|
|
|
# TODO
|
|
if options.no_discovery:
|
|
pass
|
|
|
|
# TODO
|
|
if options.disable_computer_discovery:
|
|
pass
|
|
|
|
targets = set()
|
|
[[targets.add(t) for t in g] for g in options.targets]
|
|
options.targets = list(targets)
|
|
|
|
return options
|
|
|
|
|
|
def print_banner():
|
|
print(r'''
|
|
O~~ ~~ O~~ O~~ O~~ O~~~~~~~
|
|
O~~ O~~ O~ O~ O~~ O~~ O~~
|
|
O~~ O~~ O~~ O~~ O~O~ O~O~O~ O~ O~~ O~~ O~~ O~~O~~ O~~
|
|
O~~ O~~ O~~ O~~ O~~ O~~ O~~ O~~ O~ O~~ O~~~~~~~ O~~ O~~
|
|
O~~ O~~ O~~O~~ O~~ O~~ O~~ O~~O~~~~~ O~~O~~ O~~~
|
|
O~~ O~~ O~~ O~~O~~ O~~ O~~ O~~ O~~O~ O~~ O~~
|
|
O~~ ~~ O~~~ O~~ O~~ O~~~ O~~ O~~ O~~~ O~~~~ O~~ O~~
|
|
O~~ ''')
|
|
|
|
print("")
|
|
print("")
|
|
|
|
|
|
def main():
|
|
print_banner()
|
|
snaffle_options = parse_arguments()
|
|
begin_snaffle(snaffle_options)
|
|
|
|
print("\nI snaffled 'til the snafflin was done")
|
|
sleep(0.2)
|
|
print("View log file at ~/.snafflepy/logs/")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|