Merge pull request #18 from robert-todora/dev

Dev

snaffcore/classifier.py

@@ -25,7 +25,8 @@ class Rules:
 
 
 # TODO
-def is_interest(file, rules:Rules):
+
+def is_interest(file):
     # massive_wordlist = prepare_classifiers()
     # print(massive_wordlist)
     # for root, dirs, files in os.walk(snafflepy_path, topdown=False):

snaffcore/go_snaffle.py

@@ -27,6 +27,21 @@ def begin_snaffle(options):
     print("Beginning the snaffle...")
     sleep(0.2)
 
+    if(not options.domain):
+        log.info("Domain not provided, retrieving automatically.")
+        s = Server(options.targets[0], get_info = ALL)
+        c = Connection(s)
+        if(not c.bind()):
+             log.error("Could not get domain automatically")
+             sys.exit(1)
+        else:  
+             try:
+                options.domain = str(s.info.other["ldapServiceName"][0].split("@")[1]).lower()
+             except Exception as e:
+                 log.error("Could not get domain automatically")
+                 sys.exit(1)
+        c.unbind()
+
     domain_names = []
     # TODO: Talk to AD via LDAP to get list of computers with file shares
     if options.disable_computer_discovery: