prepare for using program_options; sanitize namespaces a bit
[dyn-nsupdate.git] / client-scripts / dyn-ns-client
1 #!/usr/bin/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 # The views and conclusions contained in the software and documentation are those
26 # of the authors and should not be interpreted as representing official policies, 
27 # either expressed or implied, of the FreeBSD Project.
28
29 import urllib.request, socket, sys, argparse
30
31 # configuration variables
32 server = 'ipv4.ns.ralfj.de'
33 domains = ['domain.dyn.ralfj.de'] # list of domains to update
34 password = 'yourpassword'
35 # END of configuration variables
36
37 # allow overwriting some values on the command-line
38 parser = argparse.ArgumentParser(description='Update a domain managed by a dyn-nsupdate server')
39 parser.add_argument("-s", "--server",
40                     dest="server", default=server,
41                     help="The dyn-nsupdate server")
42 parser.add_argument("-p", "--password",
43                     dest="password", default=password,
44                     help="The password used to update the domains")
45 parser.add_argument("-v", "--verbose",
46                     action="store_true", dest="verbose",
47                     help="Be more verbose")
48 parser.add_argument("domains",  metavar='DOMAIN', nargs='*', default=domains,
49                     help="The domains to update")
50 args = parser.parse_args()
51
52 def urlopen(url):
53     return urllib.request.urlopen(url).read().decode('utf-8').strip()
54
55 myip = urlopen('https://'+args.server+'/checkip')
56
57 def update_domain(domain):
58     '''Update the given domain, using the global server, user, password. Returns True on success, False on failure.'''
59     global myip, args
60     # check if the domain is already mapped to our current IP
61     try:
62         domainip = socket.gethostbyname(domain)
63         if myip == domainip:
64             # nothing to do
65             if args.verbose:
66                 print("Domain",domain,"already up-to-date, not doing anything")
67             return True
68     except socket.gaierror: # domain not found
69         pass
70
71     # we need to update the IP
72     result = urlopen('https://'+args.server+'/update?password='+urllib.parse.quote(args.password)+'&domain='+urllib.parse.quote(domain)+'&ip='+urllib.parse.quote(myip))
73     if 'good '+myip == result: 
74         print("Successfully updated domain",domain,"to",myip)
75         # all went all right
76         return True
77     else:
78         # Something went wrong
79         print("Unexpected answer from server",server,"while updating",domain,"to",myip)
80         print(result)
81         return False
82
83 exitcode = 0
84 for domain in args.domains:
85     if not update_domain(domain):
86         exitcode = 1
87 sys.exit(exitcode)