Browse Source

flatten if

David Baker 4 years ago
parent
commit
1c821a6db7
1 changed files with 17 additions and 18 deletions
  1. 17 18
      sydent/http/servlets/threepidbindservlet.py

+ 17 - 18
sydent/http/servlets/threepidbindservlet.py

@@ -116,25 +116,24 @@ class ThreePidBindServlet(Resource):
         if isV2:
             logger.info("Allowing any domain because path is v2")
             allow_mxid_domain = True
+        elif isMxidDomainAllowed(mxid_domain):
+            logger.info("Allowing domain %s" % (mxid_domain,))
+            allow_mxid_domain = True
         else:
-            if isMxidDomainAllowed(mxid_domain):
-                logger.info("Allowing domain %s" % (mxid_domain,))
-                allow_mxid_domain = True
-            else:
-                # do the same check on the .well-known lookup
-                httpClient = SimpleHttpClient(self.sydent)
-                try:
-                    wellKnown = yield httpClient.get_json("https://%s/.well-known/matrix/server" % (mxid_domain,))
-                    if 'm.server' in wellKnown:
-                        if isMxidDomainAllowed(wellKnown['m.server']):
-                            logger.info("Allowing domain %s due to .well-known: %s" % (mxid_domain, wellKnown['m.server']))
-                            allow_mxid_domain = True
-                        else:
-                            logger.info("Not allowing domain %s from to .well-known: %s" % (mxid_domain, wellKnown['m.server']))
-                except Exception as e:
-                    # we could be more specific with our errors but we're not going to allow it in any case
-                    logger.info(".well-known lookup failed for %s: %r" % (mxid_domain, e))
-                    pass
+            # do the same check on the .well-known lookup
+            httpClient = SimpleHttpClient(self.sydent)
+            try:
+                wellKnown = yield httpClient.get_json("https://%s/.well-known/matrix/server" % (mxid_domain,))
+                if 'm.server' in wellKnown:
+                    if isMxidDomainAllowed(wellKnown['m.server']):
+                        logger.info("Allowing domain %s due to .well-known: %s" % (mxid_domain, wellKnown['m.server']))
+                        allow_mxid_domain = True
+                    else:
+                        logger.info("Not allowing domain %s from to .well-known: %s" % (mxid_domain, wellKnown['m.server']))
+            except Exception as e:
+                # we could be more specific with our errors but we're not going to allow it in any case
+                logger.info(".well-known lookup failed for %s: %r" % (mxid_domain, e))
+                pass
 
         if allow_mxid_domain:
             res = self.sydent.threepidBinder.addBinding(s.medium, s.address, mxid)