Merge pull request #157 from bashclub/dev

Create zmb-ad_activate-trashbin.py

scripts/zmb-ad_activate-trashbin.py

@@ -0,0 +1,118 @@
+#!/usr/bin/env python3
+import argparse
+import os
+import sys
+from samba.samdb import SamDB
+from samba.param import LoadParm
+from samba.auth import system_session
+import ldb
+
+
+def get_samdb():
+    if os.geteuid() != 0:
+        print("[!] Dieses Skript muss als root ausgeführt werden.", file=sys.stderr)
+        sys.exit(2)
+
+    lp = LoadParm()
+    try:
+        lp.load_default()
+    except Exception:
+        if os.path.exists("/etc/samba/smb.conf"):
+            lp.load("/etc/samba/smb.conf")
+        else:
+            print("[!] Konnte smb.conf nicht laden.", file=sys.stderr)
+            sys.exit(2)
+
+    samdb_path = lp.samdb_url()
+    if not samdb_path:
+        private_dir = lp.get("private dir") or "/var/lib/samba/private"
+        samdb_path = os.path.join(private_dir, "sam.ldb")
+
+    try:
+        session = system_session(lp)
+        samdb = SamDB(
+            url=samdb_path,
+            session_info=session,
+            credentials=None,
+            lp=lp,
+        )
+        return samdb
+    except Exception as e:
+        print(f"[!] Fehler beim Öffnen der SAM-Datenbank ({samdb_path}): {e}", file=sys.stderr)
+        sys.exit(2)
+
+
+def is_recycle_bin_enabled(samdb):
+    config_dn = samdb.get_config_basedn().get_linearized()
+    partitions_dn = f"CN=Partitions,{config_dn}"
+
+    res = samdb.search(
+        base=partitions_dn,
+        scope=ldb.SCOPE_BASE,
+        attrs=["msDS-EnabledFeature"],
+    )
+
+    if len(res) > 0 and "msDS-EnabledFeature" in res[0]:
+        enabled_features = [
+            f.decode("utf-8") if isinstance(f, bytes) else str(f)
+            for f in res[0]["msDS-EnabledFeature"]
+        ]
+        for feat in enabled_features:
+            if "Recycle Bin Feature" in feat:
+                return True
+    return False
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Active Directory Recycle Bin Status prüfen und aktivieren (Samba AD)."
+    )
+    parser.add_argument(
+        "--check",
+        action="store_true",
+        help="Prüft nur, ob der Papierkorb aktiv ist (Exit 0 = aktiv, 1 = inaktiv).",
+    )
+    args = parser.parse_args()
+
+    samdb = get_samdb()
+    enabled = is_recycle_bin_enabled(samdb)
+
+    if args.check:
+        if enabled:
+            print("[+] Active Directory Papierkorb ist AKTIVIERT.")
+            sys.exit(0)
+        else:
+            print("[-] Active Directory Papierkorb ist NICHT aktiviert.")
+            sys.exit(1)
+
+    # Standard-Ablauf: Aktivieren falls noch nicht aktiv
+    if enabled:
+        print("[+] Der Active Directory Papierkorb ist BEREITS AKTIV!")
+        return
+
+    config_dn = samdb.get_config_basedn().get_linearized()
+    partitions_dn = f"CN=Partitions,{config_dn}"
+    feature_dn = f"CN=Recycle Bin Feature,CN=Optional Features,CN=Directory Service,CN=Windows NT,CN=Services,{config_dn}"
+
+    print(f"[*] Configuration DN : {config_dn}")
+    print(f"[*] Partitions DN    : {partitions_dn}")
+    print(f"[*] Feature DN       : {feature_dn}")
+    print("\n[*] Aktiviere Active Directory Recycle Bin...")
+
+    msg = ldb.Message()
+    msg.dn = ldb.Dn(samdb, partitions_dn)
+    msg["msDS-EnabledFeature"] = ldb.MessageElement(
+        [feature_dn], ldb.FLAG_MOD_ADD, "msDS-EnabledFeature"
+    )
+
+    try:
+        samdb.modify(msg, controls=["relax:0"])
+        print("[+] ERFOLG: Active Directory Papierkorb wurde erfolgreich aktiviert!")
+    except ldb.LdbError as err:
+        num, text = err.args
+        print(f"[!] Fehler bei der Aktivierung (LDB Error {num}): {text}", file=sys.stderr)
+        sys.exit(2)
+
+
+if __name__ == "__main__":
+    main()