import base64
import re
+import socket
import urllib
import urllib2
from __version__ import CLIENT_VERSION
+from .errors import *
from i18n import _
# Initialize the logger.
# Connection to the core of the program.
self.core = core
+ # Find out on which distribution we are running.
+ self.distro = self._get_distro_identifier()
+ logger.debug(_("Running on distribution: %s") % self.distro)
+
@property
def proxy(self):
proxy = self.core.settings.get("proxy")
return proxy
- def guess_external_ipv6_address(self):
+ def _guess_external_ip_address(self, url, timeout=10):
"""
Sends a request to an external web server
to determine the current default IP address.
"""
- response = self.send_request("http://checkip6.dns.lightningwirelabs.com")
+ try:
+ response = self.send_request(url, timeout=timeout)
+
+ # If the server could not be reached, we will return nothing.
+ except DDNSNetworkError:
+ return
if not response.code == 200:
return
return match.group(1)
- def guess_external_ipv4_address(self):
+ def guess_external_ipv6_address(self):
"""
Sends a request to the internet to determine
- the public IP address.
-
- XXX does not work for IPv6.
+ the public IPv6 address.
"""
- response = self.send_request("http://checkip4.dns.lightningwirelabs.com")
+ return self._guess_external_ip_address("http://checkip6.dns.lightningwirelabs.com")
- if response.code == 200:
- match = re.search(r"Your IP address is: (\d+.\d+.\d+.\d+)", response.read())
- if match is None:
- return
-
- return match.group(1)
+ def guess_external_ipv4_address(self):
+ """
+ Sends a request to the internet to determine
+ the public IPv4 address.
+ """
+ return self._guess_external_ip_address("http://checkip4.dns.lightningwirelabs.com")
def send_request(self, url, method="GET", data=None, username=None, password=None, timeout=30):
assert method in ("GET", "POST")
query_args = self._format_query_args(data)
data = None
- url = "%s?%s" % (url, query_args)
+ if "?" in url:
+ url = "%s&%s" % (url, query_args)
+ else:
+ url = "%s?%s" % (url, query_args)
logger.debug("Sending request (%s): %s" % (method, url))
if data:
if username and password:
basic_auth_header = self._make_basic_auth_header(username, password)
- print repr(basic_auth_header)
req.add_header("Authorization", "Basic %s" % basic_auth_header)
# Set the user agent.
logger.debug(" %s: %s" % (k, v))
try:
- resp = urllib2.urlopen(req)
+ resp = urllib2.urlopen(req, timeout=timeout)
# Log response header.
logger.debug(_("Response header:"))
# Return the entire response object.
return resp
+ except urllib2.HTTPError, e:
+ # 503 - Service Unavailable
+ if e.code == 503:
+ raise DDNSServiceUnavailableError
+
+ # Raise all other unhandled exceptions.
+ raise
+
except urllib2.URLError, e:
+ if e.reason:
+ # Network Unreachable (e.g. no IPv6 access)
+ if e.reason.errno == 101:
+ raise DDNSNetworkUnreachableError
+ elif e.reason.errno == 111:
+ raise DDNSConnectionRefusedError
+
+ # Raise all other unhandled exceptions.
raise
+ except socket.timeout, e:
+ logger.debug(_("Connection timeout"))
+
+ raise DDNSConnectionTimeoutError
+
def _format_query_args(self, data):
args = []
# XXX TODO
assert False
+
+ def resolve(self, hostname, proto=None):
+ addresses = []
+
+ if proto is None:
+ family = 0
+ elif proto == "ipv6":
+ family = socket.AF_INET6
+ elif proto == "ipv4":
+ family = socket.AF_INET
+ else:
+ raise ValueError("Protocol not supported: %s" % proto)
+
+ # Resolve the host address.
+ try:
+ response = socket.getaddrinfo(hostname, None, family)
+ except socket.gaierror, e:
+ # Name or service not known
+ if e.errno == -2:
+ return []
+
+ # No record for requested family available (e.g. no AAAA)
+ elif e.errno == -5:
+ return []
+
+ raise
+
+ # Handle responses.
+ for family, socktype, proto, canonname, sockaddr in response:
+ # IPv6
+ if family == socket.AF_INET6:
+ address, port, flow_info, scope_id = sockaddr
+
+ # Only use the global scope.
+ if not scope_id == 0:
+ continue
+
+ # IPv4
+ elif family == socket.AF_INET:
+ address, port = sockaddr
+
+ # Ignore everything else...
+ else:
+ continue
+
+ # Add to repsonse list if not already in there.
+ if not address in addresses:
+ addresses.append(address)
+
+ return addresses
+
+ def _get_distro_identifier(self):
+ """
+ Returns a unique identifier for the distribution
+ we are running on.
+ """
+ os_release = self.__parse_os_release()
+ if os_release:
+ return os_release
+
+ system_release = self.__parse_system_release()
+ if system_release:
+ return system_release
+
+ # If nothing else could be found, we return
+ # just "unknown".
+ return "unknown"
+
+ def __parse_os_release(self):
+ """
+ Tries to parse /etc/os-release and
+ returns a unique distribution identifier
+ if the file exists.
+ """
+ try:
+ f = open("/etc/os-release", "r")
+ except IOError, e:
+ # File not found
+ if e.errno == 2:
+ return
+
+ raise
+
+ os_release = {}
+ with f:
+ for line in f.readlines():
+ m = re.match(r"^([A-Z\_]+)=(.*)$", line)
+ if m is None:
+ continue
+
+ os_release[m.group(1)] = m.group(2)
+
+ try:
+ return "%(ID)s-%(VERSION_ID)s" % os_release
+ except KeyError:
+ return
+
+ def __parse_system_release(self):
+ """
+ Tries to parse /etc/system-release and
+ returns a unique distribution identifier
+ if the file exists.
+ """
+ try:
+ f = open("/etc/system-release", "r")
+ except IOError, e:
+ # File not found
+ if e.errno == 2:
+ return
+
+ raise
+
+ with f:
+ # Read first line
+ line = f.readline()
+
+ # Check for IPFire systems
+ m = re.match(r"^IPFire (\d).(\d+)", line)
+ if m:
+ return "ipfire-%s" % m.group(1)
projects / oddments / ddns.git / blobdiff