proxy support

postfix_mta_sts_resolver/resolver.py

@@ -23,6 +23,13 @@ def __init__(self, *, timeout=defaults.TIMEOUT, loop):
         self._timeout = timeout
         self._resolver = aiodns.DNSResolver(timeout=timeout, loop=loop)
         self._http_timeout = aiohttp.ClientTimeout(total=timeout)
+        self._proxy_info = aiohttp.helpers.proxies_from_env().get('https', None)
+        if self._proxy_info is None:
+            self._proxy = None
+            self._proxy_auth = None
+        else:
+            self._proxy = self._proxy_info.proxy
+            self._proxy_auth = self._proxy_info.proxy_auth
 
     async def resolve(self, domain, last_known_id=None):
         if domain.startswith('.'):
@@ -69,8 +76,12 @@ async def resolve(self, domain, last_known_id=None):
 
         # Fetch actual policy
         try:
-            async with aiohttp.ClientSession(loop=self._loop, timeout=self._http_timeout) as session:
-                async with session.get(sts_policy_url, allow_redirects=False) as resp:
+            async with aiohttp.ClientSession(loop=self._loop,
+                                             timeout=self._http_timeout) as session:
+                async with session.get(sts_policy_url,
+                                       allow_redirects=False,
+                                       proxy=self._proxy,
+                                       proxy_auth=self._proxy_auth) as resp:
                     if resp.status != 200:
                         raise BadSTSPolicy()
                     if not is_plaintext(resp.headers.get('Content-Type', '')):

setup.py

@@ -6,7 +6,7 @@
     long_description = f.read()
 
 setup(name='postfix_mta_sts_resolver',
-      version='0.2.0',
+      version='0.2.1',
       description='Daemon which provides TLS client policy for Postfix via socketmap, according to domain MTA-STS policy',
       url='https://github.com/Snawoot/postfix-mta-sts-resolver',
       author='Vladislav Yarmak',