]> git.ipfire.org Git - ddns.git/blobdiff - src/ddns/system.py
ddns: Port to python3
[ddns.git] / src / ddns / system.py
index 9214cce6ab287068fdd72df1263e13e81880c081..299ed078f85dde61ae52be3e7e4bf1a61f3ba05a 100644 (file)
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/python3
 ###############################################################################
 #                                                                             #
 # ddns - A dynamic DNS client for IPFire                                      #
 
 import base64
 import re
+import ssl
 import socket
-import urllib
-import urllib2
+import urllib.request
+import urllib.parse
+import urllib.error
 
-from __version__ import CLIENT_VERSION
+from .__version__ import CLIENT_VERSION
 from .errors import *
-from i18n import _
+from .i18n import _
 
 # Initialize the logger.
 import logging
@@ -65,13 +67,20 @@ class DDNSSystem(object):
                return proxy
 
        def get_local_ip_address(self, proto):
+               ip_address = self._get_local_ip_address(proto)
+
+               # Check if the IP address is usable and only return it then
+               if self._is_usable_ip_address(proto, ip_address):
+                       return ip_address
+
+       def _get_local_ip_address(self, proto):
                # Legacy code for IPFire 2.
                if self.distro == "ipfire-2" and proto == "ipv4":
                        try:
                                with open("/var/ipfire/red/local-ipaddress") as f:
                                        return f.readline()
 
-                       except IOError, e:
+                       except IOError as e:
                                # File not found
                                if e.errno == 2:
                                        return
@@ -104,9 +113,9 @@ class DDNSSystem(object):
 
        def guess_external_ip_address(self, family, **kwargs):
                if family == "ipv6":
-                       url = "http://checkip6.dns.lightningwirelabs.com"
+                       url = "https://checkip6.dns.lightningwirelabs.com"
                elif family == "ipv4":
-                       url = "http://checkip4.dns.lightningwirelabs.com"
+                       url = "https://checkip4.dns.lightningwirelabs.com"
                else:
                        raise ValueError("unknown address family")
 
@@ -129,7 +138,7 @@ class DDNSSystem(object):
                if data:
                        logger.debug("  data: %s" % data)
 
-               req = urllib2.Request(url, data=data)
+               req = urllib.request.Request(url, data=data)
 
                if username and password:
                        basic_auth_header = self._make_basic_auth_header(username, password)
@@ -155,7 +164,7 @@ class DDNSSystem(object):
                        logger.debug("  %s: %s" % (k, v))
 
                try:
-                       resp = urllib2.urlopen(req, timeout=timeout)
+                       resp = urllib.request.urlopen(req, timeout=timeout)
 
                        # Log response header.
                        logger.debug(_("Response header (Status Code %s):") % resp.code)
@@ -165,7 +174,12 @@ class DDNSSystem(object):
                        # Return the entire response object.
                        return resp
 
-               except urllib2.HTTPError, e:
+               except urllib.error.HTTPError as e:
+                       # Log response header.
+                       logger.debug(_("Response header (Status Code %s):") % e.code)
+                       for k, v in e.hdrs.items():
+                               logger.debug("  %s: %s" % (k, v))
+
                        # 400 - Bad request
                        if e.code == 400:
                                raise DDNSRequestError(e.reason)
@@ -175,6 +189,16 @@ class DDNSSystem(object):
                        elif e.code in (401, 403):
                                raise DDNSAuthenticationError(e.reason)
 
+                       # 404 - Not found
+                       # Either the provider has changed the API, or
+                       # there is an error on the server
+                       elif e.code == 404:
+                               raise DDNSNotFound(e.reason)
+
+                       # 429 - Too Many Requests
+                       elif e.code == 429:
+                               raise DDNSTooManyRequests(e.reason)
+
                        # 500 - Internal Server Error
                        elif e.code == 500:
                                raise DDNSInternalServerError(e.reason)
@@ -186,18 +210,38 @@ class DDNSSystem(object):
                        # Raise all other unhandled exceptions.
                        raise
 
-               except urllib2.URLError, e:
+               except urllib.error.URLError as e:
                        if e.reason:
+                               # Handle SSL errors
+                               if isinstance(e.reason, ssl.SSLError):
+                                       e = e.reason
+
+                                       if e.reason == "CERTIFICATE_VERIFY_FAILED":
+                                               raise DDNSCertificateError
+
+                                       # Raise all other SSL errors
+                                       raise DDNSSSLError(e.reason)
+
+                               # Name or service not known
+                               if e.reason.errno == -2:
+                                       raise DDNSResolveError
+
                                # Network Unreachable (e.g. no IPv6 access)
                                if e.reason.errno == 101:
                                        raise DDNSNetworkUnreachableError
+
+                               # Connection Refused
                                elif e.reason.errno == 111:
                                        raise DDNSConnectionRefusedError
 
+                               # No route to host
+                               elif e.reason.errno == 113:
+                                       raise DDNSNoRouteToHostError(req.host)
+
                        # Raise all other unhandled exceptions.
                        raise
 
-               except socket.timeout, e:
+               except socket.timeout as e:
                        logger.debug(_("Connection timeout"))
 
                        raise DDNSConnectionTimeoutError
@@ -206,7 +250,7 @@ class DDNSSystem(object):
                args = []
 
                for k, v in data.items():
-                       arg = "%s=%s" % (k, urllib.quote(v))
+                       arg = "%s=%s" % (k, urllib.parse.quote(v))
                        args.append(arg)
 
                return "&".join(args)
@@ -215,7 +259,7 @@ class DDNSSystem(object):
                authstring = "%s:%s" % (username, password)
 
                # Encode authorization data in base64.
-               authstring = base64.encodestring(authstring)
+               authstring = base64.encodebytes(authstring)
 
                # Remove any newline characters.
                authstring = authstring.replace("\n", "")
@@ -249,16 +293,18 @@ class DDNSSystem(object):
                guess_ip = self.core.settings.get("guess_external_ip", "true")
                guess_ip = guess_ip in ("true", "yes", "1")
 
-               # If the external IP address should be used, we just do that.
-               if guess_ip:
-                       return self.guess_external_ip_address(proto)
-
                # Get the local IP address.
-               local_ip_address = self.get_local_ip_address(proto)
+               local_ip_address = None
+
+               if not guess_ip:
+                       try:
+                               local_ip_address = self.get_local_ip_address(proto)
+                       except NotImplementedError:
+                               logger.warning(_("Falling back to check the IP address with help of a public server"))
 
-               # If the local IP address is not usable, we must guess
-               # the correct IP address...
-               if not self._is_usable_ip_address(proto, local_ip_address):
+               # If no local IP address could be determined, we will fall back to the guess
+               # it with help of an external server...
+               if not local_ip_address:
                        local_ip_address = self.guess_external_ip_address(proto)
 
                return local_ip_address
@@ -280,7 +326,7 @@ class DDNSSystem(object):
                                r"^172\.(1[6-9]|2[0-9]|31)\.\d+\.\d+$",
 
                                # Dual Stack Lite address space
-                               r"^100\.(6[4-9]|[7-9][0-9]|1[01][0-9]|12[0-7])\.\d+\.\d+",
+                               r"^100\.(6[4-9]|[7-9][0-9]|1[01][0-9]|12[0-7])\.\d+\.\d+$",
                        )
 
                        for match in matches:
@@ -309,11 +355,15 @@ class DDNSSystem(object):
                # Resolve the host address.
                try:
                        response = socket.getaddrinfo(hostname, None, family)
-               except socket.gaierror, e:
+               except socket.gaierror as e:
                        # Name or service not known
                        if e.errno == -2:
                                return []
 
+                       # Temporary failure in name resolution
+                       elif e.errno == -3:
+                               raise DDNSResolveError(hostname)
+
                        # No record for requested family available (e.g. no AAAA)
                        elif e.errno == -5:
                                return []
@@ -339,7 +389,7 @@ class DDNSSystem(object):
                                continue
 
                        # Add to repsonse list if not already in there.
-                       if not address in addresses:
+                       if address not in addresses:
                                addresses.append(address)
 
                return addresses
@@ -369,7 +419,7 @@ class DDNSSystem(object):
                """
                try:
                        f = open("/etc/os-release", "r")
-               except IOError, e:
+               except IOError as e:
                        # File not found
                        if e.errno == 2:
                                return
@@ -398,7 +448,7 @@ class DDNSSystem(object):
                """
                try:
                        f = open("/etc/system-release", "r")
-               except IOError, e:
+               except IOError as e:
                        # File not found
                        if e.errno == 2:
                                return