added argument parsing and banner

2026-09-24 18:22:23 +00:00 · 2023-06-22 17:05:23 -04:00
parent 1f5ded8299
commit 9c99b11e6b
2 changed files with 35 additions and 0 deletions
@@ -1,4 +1,39 @@
+import argparse
+import sys
+import ldap3
+
+
+def parse_arguments():
+    print("SnafflePy: A Port by @robert-todora")
+
+    parser = argparse.ArgumentParser(add_help=True, prog='snafflepy', description='A "port" of Snaffler in python')
+    parser.add_argument("-u","--username", metavar='username',type=str, nargs=1, help="domain username")
+    parser.add_argument("-p","--password", metavar='password',type=str, nargs=1, help="password for domain user")
+    parser.add_argument('--dc-ip', metavar='[IP addr]', required=True, nargs=1, help="IP address of domain controller")
+    parser.add_argument("-d", "--domain", metavar='domain', default="", help="FQDN domain to authenticate to")
+
+    if len(sys.argv) == 1:
+        parser.print_help()
+        sys.exit(1)
+
+
+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()
+    parse_arguments()
    print("I snaffled 'til the snafflin was done")