From 3de8c2a65152a9985b04f11a64c6568c93a70dbf Mon Sep 17 00:00:00 2001
From: Jacob Hoffman-Andrews <jsha@eff.org>
Date: Wed, 6 Aug 2014 17:08:30 -0400
Subject: [PATCH] Do a better job finding address domain from mx doman.

---
 ConfigParser.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/ConfigParser.py b/ConfigParser.py
index 69779548f..eff4d3557 100755
--- a/ConfigParser.py
+++ b/ConfigParser.py
@@ -72,11 +72,11 @@ class Config:
         sys.stderr.write("Unknown attribute: " + `atr` + "\n")
 
   def get_address_domains(self, mx_hostname):
-    for mx_domain, address_domains in self.mx_domain_to_address_domains.items():
-      # TODO: write this better
-      if (mx_hostname.find(mx_domain) > 0 and
-          mx_hostname.find(mx_domain) == len(mx_hostname) - len(mx_domain)):
-        return address_domains
+    labels = mx_hostname.split(".")
+    for n in range(1, len(labels)):
+      parent = "." + ".".join(labels[n:])
+      if parent in self.mx_domain_to_address_domains:
+        return self.mx_domain_to_address_domains[parent]
     return None
 
   def check_tls_policy_domains(self, val):