From e06943d20261f6991dc9ae1a75cae1ad52f4e9d4 Mon Sep 17 00:00:00 2001
From: Tim Strazzere <tim@strazzere.com>
Date: Tue, 3 Sep 2024 09:22:46 -0700
Subject: [PATCH] dump script in

---
 README.md        |  3 ++
 monitor.py       | 82 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  2 ++
 3 files changed, 87 insertions(+)
 create mode 100644 README.md
 create mode 100644 monitor.py
 create mode 100644 requirements.txt

diff --git a/README.md b/README.md
new file mode 100644
index 0000000..a6ee8ca
--- /dev/null
+++ b/README.md
@@ -0,0 +1,3 @@
+# whois-mx-monitor
+
+There are better solutions to this, but this was a quick and dirty thing written to run in a screen session over a weekend to monitor for when a domain and/or the mx records are nuked by the registrar/google.
\ No newline at end of file
diff --git a/monitor.py b/monitor.py
new file mode 100644
index 0000000..fe0acf8
--- /dev/null
+++ b/monitor.py
@@ -0,0 +1,82 @@
+import whois
+import dns.resolver
+import time
+import json
+import os
+import re
+
+DOMAIN = "derp.com"  # Replace with the domain you want to monitor
+CHECK_INTERVAL = 3600 / 2 
+DATA_FILE = "domain_data.json"
+INITIAL_BACKOFF = 60  # Initial backoff time in seconds (1 minute)
+
+def get_whois_data(domain):
+    try:
+        data = whois.whois(domain)
+        if 'Socket not responding' in data.text:
+            raise Exception("Socket not responding")
+        
+        # Remove the "Last update of whois database" line
+        data.text = re.sub(r"Last update of whois database:.*\r\n", "", data.text)
+        return data
+    except Exception as e:
+        print(f"WHOIS query failed: {e}")
+        return None
+
+# Sort the MX records to make comparison insensitive to order
+def get_mx_records(domain):
+    mx_records = []
+    try:
+        answers = dns.resolver.resolve(domain, 'MX')
+        for rdata in answers:
+            mx_records.append(rdata.exchange.to_text())
+    except Exception as e:
+        print(f"Error fetching MX records: {e}")
+    return sorted(mx_records)
+
+def load_previous_data():
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, 'r') as f:
+            return json.load(f)
+    return {}
+
+def save_current_data(data):
+    with open(DATA_FILE, 'w') as f:
+        json.dump(data, f)
+
+def check_for_changes(backoff_time=INITIAL_BACKOFF):
+    current_data = {
+        'whois': get_whois_data(DOMAIN),
+        'mx_records': get_mx_records(DOMAIN)
+    }
+
+    # Sort the MX records to make comparison insensitive to order
+    if current_data['whois'] is None:
+        print(f"Retrying WHOIS query in {backoff_time} seconds...")
+        time.sleep(backoff_time)
+        check_for_changes(backoff_time * 2)
+        return
+
+    previous_data = load_previous_data()
+
+    if previous_data:
+        whois_changed = previous_data['whois'] != current_data['whois'].__dict__
+        mx_changed = previous_data['mx_records'] != current_data['mx_records']
+
+        if whois_changed or mx_changed:
+            print(f"Changes detected for {DOMAIN}:")
+            if whois_changed:
+                print("\nWHOIS information has changed.")
+                print("Previous WHOIS data:", previous_data['whois'])
+                print("Current WHOIS data:", current_data['whois'].__dict__)
+            if mx_changed:
+                print("\nMX records have changed.")
+                print("Previous MX records:", previous_data['mx_records'])
+                print("Current MX records:", current_data['mx_records'])
+
+    save_current_data({'whois': current_data['whois'].__dict__, 'mx_records': current_data['mx_records']})
+
+if __name__ == "__main__":
+    while True:
+        check_for_changes()
+        time.sleep(CHECK_INTERVAL)
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..a24cb6a
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+python-whois==0.7.3
+dnspython==2.4.0
\ No newline at end of file