Files
dnsupdate/dnsupdate.py
Fredrik Wahlberg 18077c99ea Added options using the optparse module
TTL can now be set
Global variables are a mess and must be fixed!
2004-01-11 22:11:13 +00:00

94 lines
2.6 KiB
Python

import dns.query
import dns.tsigkeyring
import dns.update
from optparse import OptionParser
import re
import string
import sys
# dnsupdate.py
# Updates a dynamic dns-record using a TSIG key.
domain = "lubcke.se"
keyname = "lubcke.se."
keysecret = "ZGhNJ05b8ThmHOXhJvkvMw=="
ttl = 60
def main():
usage = "usage: %prog [-n] hostname [-i] ip adress [OPTIONS]"
parser = OptionParser(usage)
parser.add_option("-d", "--domain",
type="string",
help="Name of the domain to update")
parser.add_option("-i", "--ipaddress",
type="string",
help="IP-address of the host")
parser.add_option("-k", "--keyname",
type="string",
help="Name of the TSIG key")
parser.add_option("-n", "--name",
type="string",
help="Name of the host to update")
parser.add_option("-p", "--password",
type="string",
help="The TSIG key")
parser.add_option("-t", "--ttl",
type="int",
help="TTL in seconds")
(options, args) = parser.parse_args()
global domain, keyname, keysecret, ttl
if options.domain:
domain = options.domain
if options.keyname:
keyname = options.keyname
if options.password:
keysecret = options.password
if options.ttl:
ttl = options.ttl
if not options.ipaddress or not options.name:
print "IP address and hostname are required for operation.\nEnter '%s -h' for usage information" % sys.argv[0]
return
else:
global hostname, ipaddress
ipaddress = options.ipaddress
hostname = options.name
update()
def update():
# Sanity check incoming data
global hostname, ipaddress, ttl
hostname = string.replace(hostname, "."+domain, '')
if not re.search('^[12]?[0-9]?[0-9](\.[12]?[0-9]?[0-9]){3}$', ipaddress):
print "Invalid ip address '%s'" % ipaddress
return
# The name of the key and the secret
keyring = dns.tsigkeyring.from_text({
keyname: keysecret
})
# dns.update.Update(name of domain, keyring, keyname)
update = dns.update.Update(domain, keyring=keyring, keyname=keyname)
# update.replace(hostname, ttl, record-type, new ip)
update.replace(hostname, ttl, 'a', ipaddress)
# doit, servername
response = dns.query.tcp(update, 'nic.wahlberg.se')
# Verify response
if response.rcode() == 0:
print "Host '%s.%s' has been added with ip address %s" % (hostname, domain, ipaddress)
else:
print "An error has occurred, the server returned:\n%s" % response
if __name__=="__main__":
main()