123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260 |
- #!/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: <!-- INSERT X HERE -->
- 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)[0]
- after = parts[1].split("-->", 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 += "<tr><td>"
- repl += lease['ip_address']
- repl += "</td><td>"
- repl += lease['hardware']
- repl += "</td><td>"
- if lease['ends'] == 'never':
- repl += "never"
- else:
- repl += str(lease['ends'] - now)
- repl += "</td><td>"
- repl += lease['client-hostname']
- repl += "</td></tr>\n"
- except:
- pass
-
- active = str(len(report_dataset))
- report_gen = str(now)
-
- stats = "<h4>Total active leases: {}</h4><h4>Report generated {}</h4>".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("<html><body>")
- print("There was an error on the server end. We are sorry.")
- print("Perhaps try the <a href=\"index-src.html\">static version</a> of this page.")
- print("The error encountered is listed below: <br/><br/>{}<br/>".format(str(e).replace("\n", "<br/>")))
- print("</body></html>")
|