11e45127d2430a8ab7da2fd6d930bb70b940fa07
[dyn-nsupdate.git] / client-scripts / dyn-ns-client
1 #!/usr/bin/env python3
2 # Copyright (c) 2014, Ralf Jung <post@ralfj.de>
3 # All rights reserved.
4
5 # Redistribution and use in source and binary forms, with or without
6 # modification, are permitted provided that the following conditions are met:
7
8 # 1. Redistributions of source code must retain the above copyright notice, this
9 #    list of conditions and the following disclaimer. 
10 # 2. Redistributions in binary form must reproduce the above copyright notice,
11 #    this list of conditions and the following disclaimer in the documentation
12 #    and/or other materials provided with the distribution.
13
14 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
15 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
18 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
19 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
20 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
21 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
23 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 #==============================================================================
25
26 import urllib.request, socket, sys, argparse, os, configparser, itertools, subprocess, re, ssl
27
28 VERBOSE_CHANGE = 1
29 VERBOSE_FULL   = 2
30
31 def sslContext(config):
32     if config['DEFAULT'].get('ssl_check_cert', 'yes').lower() in ('0', 'false', 'no'):
33         context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
34         return context
35     else:
36         return None
37
38 def readConfig(fname, defSection = 'DEFAULT'):
39     config = configparser.ConfigParser()
40     with open(fname) as file:
41         stream = itertools.chain(("["+defSection+"]\n",), file)
42         config.read_file(stream)
43     return config
44
45 def getConfigDir():
46     try:
47         from xdg import BaseDirectory
48         return os.path.join(BaseDirectory.xdg_config_home, "dyn-nsupdate")
49     except ImportError:
50         return os.path.expanduser("~/.config/dyn-nsupdate")
51
52 def urlopen(url, config):
53     if sys.version_info >= (3, 4, 3):
54         return urllib.request.urlopen(url, context=sslContext(config)).read().decode('utf-8').strip()
55     else:
56         return urllib.request.urlopen(url).read().decode('utf-8').strip()
57
58 def getMyIP(family, config, methods = {}, verbose = 0):
59     '''Returns our current IP address (<family> can be "IPv4" or "IPv6"), detected as given by the configuration.
60        Additional detection methods can be supplied via <methods>.'''
61     method = config[family]['method']
62     if method == 'none':
63         return None
64     elif method == 'remove':
65         return ""
66     elif method == 'web':
67         server = config[family].get('server', config['DEFAULT']['server'])
68         url = 'https://'+server+'/checkip'
69         try:
70             ip = urlopen(url, config)
71         except urllib.error.URLError:
72             raise Exception("Error fetching {}, make sure the URL is correct and the internet connection actually works.".format(url))
73         if verbose >= VERBOSE_FULL:
74             print("Server",server,"says my",family,"is",ip)
75         return ip
76     elif method in methods:
77         return methods[method]()
78     else:
79         raise Exception("Unsupported "+family+" detection method: "+method)
80
81 def getMyIPv4(config, verbose = 0):
82     '''Returns our current IPv4 address, detected as given by the configuration'''
83     return getMyIP("IPv4", config, verbose=verbose)
84
85 def getMyIPv6(config, verbose = 0):
86     '''Returns our current IPv6 address, detected as given by the configuration'''
87     def local():
88         out = subprocess.check_output(["ip", "addr"])
89         for line in out.decode('utf-8').split('\n'):
90             m = re.search('inet6 ([a-fA-F0-9:]+)/64 ([a-zA-Z0-9 ]*)', line)
91             if m is not None:
92                 ip = m.group(1)
93                 flags = m.group(2).split()
94                 if not 'temporary' in flags and not 'deprecated' in flags and not "link" in flags:
95                     if verbose >= VERBOSE_FULL:
96                         print("Local IPv6 detected to be",ip)
97                     return ip
98         raise Exception("Unable to detect correct local IPv6 address")
99     return getMyIP("IPv6", config, methods={'local': local}, verbose=verbose)
100
101 def getCurIP(domain, family):
102     '''Return the current IP of the given <domain>. <family> can be socket.AF_INET or socket.AF_INET6.'''
103     try:
104         addr = socket.getaddrinfo(domain, None, family=family)
105         return addr[0][4][0]
106     except socket.gaierror: # domain not found
107         return ""
108
109 def getCurIPv4(domain):
110     '''Returns the current IPv4 address of the given domain'''
111     return getCurIP(domain, socket.AF_INET)
112
113 def getCurIPv6(domain):
114     '''Returns the current IPv6 address of the given domain'''
115     return getCurIP(domain, socket.AF_INET6)
116
117 def updateDomain(server, domain, ipv4, ipv6, password, config, verbose):
118     '''Update the given domain, using the server, password. ipv4 or ipv6 can be None to not update that record, or strings with the respective addresses.
119        Updates ae only performed if necessary.
120        Returns True on success, False on failure.'''
121     assert ipv4 is not None or ipv6 is not None
122     
123     # check what the domain is currently mapped to
124     curIPv4 = getCurIPv4(domain)
125     curIPv6 = getCurIPv6(domain)
126     if verbose >= VERBOSE_FULL:
127         print("Current status of domain {} is: IPv4 address '{}', IPv6 address '{}'".format(domain, curIPv4, curIPv6))
128     
129     # check if there's something to do
130     needUpdate = (ipv4 is not None and curIPv4 != ipv4) or (ipv6 is not None and curIPv6 != ipv6)
131     if not needUpdate:
132         if verbose >= VERBOSE_FULL:
133             print("Everything already up-to-date, nothing to do")
134         return True
135
136     # we need to update the IP
137     url = 'https://'+server+'/update?password='+urllib.parse.quote(password)+'&domain='+urllib.parse.quote(domain)
138     expected = "good"
139     if ipv4 is not None:
140         url += '&ip='+urllib.parse.quote(ipv4)
141         expected += " "+ipv4
142     if ipv6 is not None:
143         url += '&ipv6='+urllib.parse.quote(ipv6)
144         expected += " "+ipv6
145     result = urlopen(url, config)
146     
147     # did everything go as planned?
148     if result == expected:
149         if verbose >= VERBOSE_CHANGE:
150             msg = "Successfully updated domain {} on {}:".format(domain, server)
151             if ipv4 is not None:
152                 if curIPv4 == ipv4:
153                     msg += " IPv4={} (unchanged)".format(curIPv4)
154                 else:
155                     msg += " IPv4={} -> {}".format(curIPv4, ipv4)
156             if ipv4 is not None and ipv6 is not None:
157                 msg += ","
158             if ipv6 is not None:
159                 if curIPv6 == ipv6:
160                     msg += " IPv6={} (unchanged)".format(curIPv6)
161                 else:
162                     msg += " IPv6={} -> {}".format(curIPv6, ipv6)
163             print(msg)
164         # all went all right
165         return True
166     else:
167         # Something went wrong
168         print("Unexpected answer from server",server,"while updating",domain)
169         print(result)
170         return False
171
172 if __name__ == "__main__":
173     # allow overwriting some values on the command-line
174     parser = argparse.ArgumentParser(description='Update a domain managed by a dyn-nsupdate server')
175     parser.add_argument("-c", "--config",
176                         dest="config", default=os.path.join(getConfigDir(), "dyn-ns-client.conf"),
177                         help="The configuration file")
178     parser.add_argument("-v", "--verbose",
179                         action="count", dest="verbose", default=0,
180                         help="Be more verbose")
181     args = parser.parse_args()
182     
183     # read config
184     if not os.path.isfile(args.config):
185         raise Exception("The config file does not exist: "+args.config)
186     config = readConfig(args.config)
187
188     # get our own addresses
189     myIPv4 = getMyIPv4(config, args.verbose)
190     myIPv6 = getMyIPv6(config, args.verbose)
191
192     # update all the domains
193     exitcode = 0
194     domains = map(str.strip, config['DEFAULT']['domains'].split(','))
195     if not domains:
196         raise Exception("No domain given to update!")
197     for domain in domains:
198         if not updateDomain(config['DEFAULT']['server'], domain, myIPv4, myIPv6, config['DEFAULT']['password'], config, verbose=args.verbose):
199             exitcode = 1
200     sys.exit(exitcode)