123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252 |
- #!/usr/bin/env python3
- #
- # Copyright (c) 2017-2020 Joe Clarke <jclarke@cisco.com>
- # All rights reserved.
- #
- # Redistribution and use in source and binary forms, with or without
- # modification, are permitted provided that the following conditions
- # are met:
- # 1. Redistributions of source code must retain the above copyright
- # notice, this list of conditions and the following disclaimer.
- # 2. Redistributions in binary form must reproduce the above copyright
- # notice, this list of conditions and the following disclaimer in the
- # documentation and/or other materials provided with the distribution.
- #
- # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
- # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- # SUCH DAMAGE.
- from __future__ import print_function
- from builtins import str
- import requests
- from requests.packages.urllib3.exceptions import InsecureRequestWarning
- requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
- import json
- import sys
- import re
- import os
- import argparse
- import CLEUCreds
- from cleu.config import Config as C
- CNR_HEADERS = {"authorization": CLEUCreds.JCLARKE_BASIC, "accept": "application/json", "content-type": "application/json"}
- CACHE_FILE = "dns_records.dat"
- def get_devs():
- url = "http://{}/get/switches/json".format(C.TOOL)
- devices = []
- response = requests.request("GET", url)
- code = response.status_code
- if code == 200:
- j = response.json()
- for dev in j:
- dev_dic = {}
- if dev["IPAddress"] == "0.0.0.0":
- continue
- if not re.search(r"^0", dev["Hostname"]):
- continue
- dev_dic["name"] = dev["Hostname"]
- dev_dic["aliases"] = [str("{}.{}.".format(dev["Name"], C.DNS_DOMAIN)), str("{}.{}.".format(dev["AssetTag"], C.DNS_DOMAIN))]
- dev_dic["ip"] = dev["IPAddress"]
- devices.append(dev_dic)
- return devices
- def purge_rr(name, url, zone):
- params = {"zoneOrigin": zone}
- try:
- response = requests.request("DELETE", url, headers=CNR_HEADERS, params=params, verify=False)
- response.raise_for_status()
- print("Purged entry for {}".format(name))
- except Exception as e:
- sys.stderr.write("Error purging entry for {}: {}\n".format(name, e))
- def purge_rrs(hname, dev):
- aname = hname
- cnames = []
- for alias in dev["aliases"]:
- cnames.append(alias.split(".")[0])
- pname = ".".join(dev["ip"].split(".")[::-1][0:3])
- ubase = C.DNS_BASE + "CCMRRSet" + "/{}"
- url = ubase.format(aname)
- purge_rr(aname, url, C.DNS_DOMAIN)
- for cname in cnames:
- url = ubase.format(cname)
- purge_rr(cname, url, C.DNS_DOMAIN)
- url = ubase.format(pname)
- purge_rr(pname, url, "10.in-addr.arpa")
- def add_entry(url, hname, dev):
- global CNR_HEADERS
- try:
- rrset = [
- "IN 0 A {}".format(dev["ip"]),
- ]
- rrset_obj = {"name": hname, "rrs": {"stringItem": rrset}, "zoneOrigin": C.DNS_DOMAIN}
- response = requests.request("PUT", url, headers=CNR_HEADERS, json=rrset_obj, verify=False)
- response.raise_for_status()
- print("Added entry for {} ==> {}".format(hname, dev["ip"]))
- except Exception as e:
- sys.stderr.write("Error adding entry for {}: {}\n".format(hname, e))
- return
- for alias in dev["aliases"]:
- aname = alias.split(".")[0]
- alias_rrset_obj = {
- "name": aname,
- "rrs": {"stringItem": ["IN 0 CNAME {}.{}.".format(hname, C.DNS_DOMAIN)]},
- "zoneOrigin": C.DNS_DOMAIN,
- }
- url = C.DNS_BASE + "CCMRRSet" + "/{}".format(aname)
- try:
- response = requests.request("PUT", url, headers=CNR_HEADERS, json=alias_rrset_obj, verify=False)
- response.raise_for_status()
- print("Added CNAME entry {} ==> {}".format(alias, hname))
- except Exception as e:
- sys.stderr.write("Error adding CNAME {} for {}: {}\n".format(alias, hname, e))
- try:
- ptr_rrset = ["IN 0 PTR {}.{}.".format(hname, C.DNS_DOMAIN)]
- rip = ".".join(dev["ip"].split(".")[::-1][0:3])
- ptr_rrset_obj = {"name": rip, "rrs": {"stringItem": ptr_rrset}, "zoneOrigin": "10.in-addr.arpa."}
- url = C.DNS_BASE + "CCMRRSet" + "/{}".format(rip)
- response = requests.request("PUT", url, headers=CNR_HEADERS, json=ptr_rrset_obj, verify=False)
- response.raise_for_status()
- print("Added PTR entry {} ==> {}".format(rip, hname))
- except Exception as e:
- sys.stderr.write("Error adding PTR entry for {}: {}\n".format(rip, e))
- if __name__ == "__main__":
- parser = argparse.ArgumentParser(description="Usage:")
- # script arguments
- parser.add_argument("--purge", help="Purge previous records", action="store_true")
- args = parser.parse_args()
- prev_records = []
- if os.path.exists(CACHE_FILE):
- fd = open(CACHE_FILE, "r")
- prev_records = json.load(fd)
- fd.close()
- devs = get_devs()
- for record in prev_records:
- found_record = False
- for dev in devs:
- hname = dev["name"].replace(".{}".format(C.DNS_DOMAIN), "")
- if record == hname:
- found_record = True
- break
- if found_record:
- continue
- url = C.DNS_BASE + "CCMHost" + "/{}".format(record)
- try:
- response = requests.request("DELETE", url, headers=CNR_HEADERS, params={"zoneOrigin": C.DNS_DOMAIN}, verify=False)
- response.raise_for_status()
- except Exception as e:
- sys.stderr.write("Failed to delete entry for {}\n".format(record))
- records = []
- for dev in devs:
- hname = dev["name"].replace(".{}".format(C.DNS_DOMAIN), "")
- records.append(hname)
- if args.purge:
- purge_rrs(hname, dev)
- url = C.DNS_BASE + "CCMHost" + "/{}".format(hname)
- response = requests.request("GET", url, headers=CNR_HEADERS, params={"zoneOrigin": C.DNS_DOMAIN}, verify=False)
- url = C.DNS_BASE + "CCMRRSet" + "/{}".format(hname)
- if response.status_code == 404:
- iurl = C.DNS_BASE + "CCMHost"
- response = requests.request(
- "GET", iurl, params={"zoneOrigin": C.DNS_DOMAIN, "addrs": dev["ip"] + "$"}, headers=CNR_HEADERS, verify=False
- )
- cur_entry = []
- if response.status_code != 404:
- cur_entry = response.json()
- if len(cur_entry) > 0:
- print("Found entry for {}: {}".format(dev["ip"], response.status_code))
- cur_entry = response.json()
- if len(cur_entry) > 1:
- print("ERROR: Found multiple entries for IP {}".format(dev["ip"]))
- continue
- print("Found old entry for IP {} => {}".format(dev["ip"], cur_entry[0]["name"]))
- durl = C.DNS_BASE + "CCMHost" + "/{}".format(cur_entry[0]["name"])
- try:
- response = requests.request("DELETE", durl, params={"zoneOrigin": C.DNS_DOMAIN}, headers=CNR_HEADERS, verify=False)
- response.raise_for_status()
- except Exception as e:
- sys.stderr.write("Failed to delete stale entry for {} ({})\n".format(cur_entry[0]["name"], dev["ip"]))
- continue
- add_entry(url, hname, dev)
- else:
- cur_entry = response.json()
- create_new = True
- for addr in cur_entry["addrs"]["stringItem"]:
- if addr == dev["ip"]:
- if "aliases" in dev and "aliases" in cur_entry:
- if (len(dev["aliases"]) > 0 and "stringItem" not in cur_entry["aliases"]) or (
- len(dev["aliases"]) != len(cur_entry["aliases"]["stringItem"])
- ):
- break
- common = set(dev["aliases"]) & set(cur_entry["aliases"]["stringItem"])
- if len(common) != len(dev["aliases"]):
- break
- create_new = False
- break
- elif ("aliases" in dev and "aliases" not in cur_entry) or ("aliases" in cur_entry and "aliases" not in dev):
- break
- else:
- create_new = False
- break
- if create_new:
- print("Deleting entry for {}".format(hname))
- purge_rrs(hname, dev)
- add_entry(url, hname, dev)
- else:
- # print("Not creating a new entry for {} as it already exists".format(dev["name"]))
- pass
- fd = open(CACHE_FILE, "w")
- json.dump(records, fd, indent=4)
- fd.close()
|