From 6bfc230a7fc55a9cd6738280ad12e51e35165b89 Mon Sep 17 00:00:00 2001 From: Jared Dunbar Date: Sat, 8 Dec 2018 05:56:05 -0500 Subject: [PATCH] from talos, fixed --- index.py | 260 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 260 insertions(+) create mode 100755 index.py diff --git a/index.py b/index.py new file mode 100755 index 0000000..883f49b --- /dev/null +++ b/index.py @@ -0,0 +1,260 @@ +#!/usr/bin/python3 + +import datetime, bisect + +print("Content-type: text/html") +print("") + + +# Below code copied from a script from a forgotten source on the internet. + + +def parse_timestamp(raw_str): + tokens = raw_str.split() + if len(tokens) == 1: + if tokens[0].lower() == 'never': + return 'never'; + else: + raise Exception('Parse error in timestamp') + elif len(tokens) == 3: + return datetime.datetime.strptime(' '.join(tokens[1:]), + '%Y/%m/%d %H:%M:%S') + else: + raise Exception('Parse error in timestamp') + +def timestamp_is_ge(t1, t2): + if type(t2) == type(""): + return False + if t1 == 'never': + return True + elif t2 == 'never': + return False + else: + return t1 >= t2 + +def timestamp_is_lt(t1, t2): + if type(t2) == type(""): + return t1 != 'never' + if t1 == 'never': + return False + elif t2 == 'never': + return t1 != 'never' + else: + return t1 < t2 + +def timestamp_is_between(t, tstart, tend): + return timestamp_is_ge(t, tstart) and timestamp_is_lt(t, tend) + +def parse_hardware(raw_str): + tokens = raw_str.split() + if len(tokens) == 2: + return tokens[1] + else: + raise Exception('Parse error in hardware') + +def strip_endquotes(raw_str): + return raw_str.strip('"') + +def identity(raw_str): + return raw_str + +def parse_binding_state(raw_str): + tokens = raw_str.split() + if len(tokens) == 2: + return tokens[1] + else: + raise Exception('Parse error in binding state') + +def parse_next_binding_state(raw_str): + tokens = raw_str.split() + if len(tokens) == 3: + return tokens[2] + else: + raise Exception('Parse error in next binding state') + +def parse_rewind_binding_state(raw_str): + tokens = raw_str.split() + if len(tokens) == 3: + return tokens[2] + else: + raise Exception('Parse error in next binding state') + +def parse_leases_file(leases_file): + valid_keys = { + 'starts': parse_timestamp, + 'ends': parse_timestamp, + 'tstp': parse_timestamp, + 'tsfp': parse_timestamp, + 'atsfp': parse_timestamp, + 'cltt': parse_timestamp, + 'hardware': parse_hardware, + 'binding': parse_binding_state, + 'next': parse_next_binding_state, + 'rewind': parse_rewind_binding_state, + 'uid': strip_endquotes, + 'client-hostname': strip_endquotes, + 'option': identity, + 'set': identity, + 'on': identity, + 'abandoned': None, + 'bootp': None, + 'reserved': None, + } + leases_db = {} + lease_rec = {} + in_lease = False + in_failover = False + for line in leases_file: + if line.lstrip().startswith('#'): + continue + tokens = line.split() + if len(tokens) == 0: + continue + key = tokens[0].lower() + if key == 'lease': + if not in_lease: + ip_address = tokens[1] + lease_rec = {'ip_address' : ip_address} + in_lease = True + else: + raise Exception('Parse error in leases file') + elif key == 'failover': + in_failover = True + elif key == '}': + if in_lease: + for k in valid_keys: + if callable(valid_keys[k]): + lease_rec[k] = lease_rec.get(k, '') + else: + lease_rec[k] = False + ip_address = lease_rec['ip_address'] + if ip_address in leases_db: + leases_db[ip_address].insert(0, lease_rec) + else: + leases_db[ip_address] = [lease_rec] + lease_rec = {} + in_lease = False + elif in_failover: + in_failover = False + continue + else: + raise Exception('Parse error in leases file') + elif key in valid_keys: + if in_lease: + value = line[(line.index(key) + len(key)):] + value = value.strip().rstrip(';').rstrip() + if callable(valid_keys[key]): + lease_rec[key] = valid_keys[key](value) + else: + lease_rec[key] = True + else: + raise Exception('Parse error in leases file') + else: + if in_lease: + raise Exception('Parse error in leases file') + if in_lease: + raise Exception('Parse error in leases file') + return leases_db + +def round_timedelta(tdelta): + return datetime.timedelta(tdelta.days, + tdelta.seconds + (0 if tdelta.microseconds < 500000 else 1)) + +def timestamp_now(): + n = datetime.datetime.utcnow() + return datetime.datetime(n.year, n.month, n.day, n.hour, n.minute, + n.second + (0 if n.microsecond < 500000 else 1)) + +def lease_is_active(lease_rec, as_of_ts): + return timestamp_is_between(as_of_ts, lease_rec['starts'], + lease_rec['ends']) + +def ipv4_to_int(ipv4_addr): + parts = ipv4_addr.split('.') + return (int(parts[0]) << 24) + (int(parts[1]) << 16) + \ + (int(parts[2]) << 8) + int(parts[3]) + +def select_active_leases(leases_db, as_of_ts): + retarray = [] + sortedarray = [] + for ip_address in leases_db: + lease_rec = leases_db[ip_address][0] + if lease_is_active(lease_rec, as_of_ts): + ip_as_int = ipv4_to_int(ip_address) + insertpos = bisect.bisect(sortedarray, ip_as_int) + sortedarray.insert(insertpos, ip_as_int) + retarray.insert(insertpos, lease_rec) + return retarray + + +# End of the other code +############################# + +# Searches for insert segments in the HTML code, ex: +def insert_item(text, search, insert): + parts = text.split(search, 1) + # If this is 2 parts, then we have a split string. Otherwise, bail. + if len(parts) == 1: + return text + else: + before = parts[0].rsplit("", 1)[1] + return before + insert + after + +############################### + +try: + #myfile = open('leases.leases','r') + myfile = open('/var/lib/dhcp/dhcpd.leases', 'r') + leases = parse_leases_file(myfile) + myfile.close() + + now = timestamp_now() + report_dataset = select_active_leases(leases, now) + + + # read in the index file, and place it in idx + idx_file = open("index-src.html") + idx = "" + for line in idx_file: + idx += line + "\n" + + # Done reading in the file. Now to parse and replace, and return, if success + + repl = "" + + for lease in report_dataset: + try: + if lease['ends'] != 'never': + expires = str(lease['ends'] - now) + repl += "" + repl += lease['ip_address'] + repl += "" + repl += lease['hardware'] + repl += "" + if lease['ends'] == 'never': + repl += "never" + else: + repl += str(lease['ends'] - now) + repl += "" + repl += lease['client-hostname'] + repl += "\n" + except: + pass + + active = str(len(report_dataset)) + report_gen = str(now) + + stats = "

Total active leases: {}

Report generated {}

".format(active, report_gen) + + idx = insert_item(idx, "INSERT LEASE TABLE", repl) + idx = insert_item(idx, "INSERT STATS", stats) + print (idx) + + +except Exception as e: + print("") + print("There was an error on the server end. We are sorry.") + print("Perhaps try the static version of this page.") + print("The error encountered is listed below:

{}
".format(str(e).replace("\n", "
"))) + print("")