From 82fbfbfb697d2b88900dbf445bbe8bef31adefff Mon Sep 17 00:00:00 2001
From: root <root@localhost.localdomain>
Date: Tue, 18 Jul 2023 16:05:00 -0400
Subject: [PATCH] add automatic retrieval of domain if not given

---
 snaffcore/go_snaffle.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/snaffcore/go_snaffle.py b/snaffcore/go_snaffle.py
index c05a2e3..5104529 100644
--- a/snaffcore/go_snaffle.py
+++ b/snaffcore/go_snaffle.py
@@ -17,6 +17,19 @@ def begin_snaffle(options):
     print("Beginning the snaffle...")
     sleep(0.2)
 
+    if(not options.domain):
+        log.info("Domain not given, retrieving automatically.")
+        s = Server(options.targets[0], get_info = ALL)
+        c = Connection(s)
+        if(not c.bind()):
+             log.error("Could not get domain automatically")
+        else:  
+             try:
+                options.domain = s.info.other["ldapServiceName"][0].split("@")[1]
+             except Exception as e:
+                 log.error("Could not get domain automatically")
+        c.unbind()
+
     domain_names = []
     # TODO: Talk to AD via LDAP to get list of computers with file shares
     if options.disable_computer_discovery: