285 lines
5.9 KiB
Python
285 lines
5.9 KiB
Python
|
|
"""
|
|
host and connection class shared between hbd and
|
|
the websit's heartbeat.py
|
|
|
|
"""
|
|
|
|
import time
|
|
import json
|
|
import copy
|
|
|
|
num = 0
|
|
|
|
MAXRTTS = 10
|
|
|
|
#
|
|
class Connection:
|
|
# map of addrs to names
|
|
|
|
htab = {}
|
|
unknown = "unknown"
|
|
up = "up"
|
|
down = "down"
|
|
overdue = "overdue"
|
|
|
|
def __init__(self, host, cid, addr, afam):
|
|
self.host = host
|
|
self.cid = cid
|
|
self.addr = addr
|
|
self.afam = afam
|
|
self.rtts = [0]
|
|
self.lastbeat = time.time()
|
|
self.statetime = self.lastbeat
|
|
self.state = Connection.unknown
|
|
|
|
if host:
|
|
r = "new addr %s" % (addr)
|
|
Connection.htab[addr] = self.host.name
|
|
if self.host.isDynDns():
|
|
Host.dnsQ.put((self.host.name, self.addr))
|
|
|
|
def registerDns(self):
|
|
Host.dnsQ.put((self.host.name, self.addr))
|
|
|
|
|
|
def statedict(self, Null=False):
|
|
d = {}
|
|
if not Null:
|
|
d['addr'] = self.addr
|
|
if self.rtts[-1]:
|
|
d['rtt'] = "%0.1f" % self.rtts[-1]
|
|
else:
|
|
d['rtt'] = ""
|
|
d['lastbeat'] = self.lastbeat
|
|
if self.state == Connection.overdue:
|
|
d['state'] = "<b>%s</b>" % self.state
|
|
else:
|
|
d['state'] = self.state
|
|
d['statetime'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(self.statetime))
|
|
|
|
else:
|
|
d['addr'] = ''
|
|
d['rtt'] = ""
|
|
d['lastbeat'] = ''
|
|
d['state'] = ''
|
|
d['statetime'] = ''
|
|
return d
|
|
|
|
|
|
def headerdict(self, afam):
|
|
d = {}
|
|
d['addr'] = '%s Addr' % afam
|
|
d['rtt'] = 'Latencey'
|
|
d['lastbeat'] = 'Last Contact'
|
|
d['state'] = 'State'
|
|
d['statetime'] = 'Last State'
|
|
return d
|
|
|
|
|
|
def jsons(self):
|
|
return(json.dumps(self.__dict__))
|
|
|
|
|
|
# set new state, return number of secs in previous state
|
|
def newstate(self, state, now, when=0):
|
|
self.state = state
|
|
delta = now - when
|
|
s = delta - self.statetime
|
|
self.statetime = delta
|
|
return s
|
|
|
|
|
|
def getstate(self):
|
|
return self.state
|
|
|
|
|
|
def newaddr(self, addr, rtt, now):
|
|
self.lastbeat = now
|
|
self.rtts.append(rtt)
|
|
if len(self.rtts) > MAXRTTS:
|
|
del self.rtts[0]
|
|
|
|
if self.addr == addr:
|
|
r = None
|
|
else:
|
|
r = "changed from %s to %s" % (self.addr, addr)
|
|
try:
|
|
del Connection.htab[self.addr]
|
|
except:
|
|
pass
|
|
self.addr = addr
|
|
Connection.htab[addr] = self.host.name
|
|
if self.host.isDynDns():
|
|
Host.dnsQ.put((self.host.name, self.addr))
|
|
return r
|
|
|
|
|
|
#
|
|
class Host:
|
|
# Table of Hosts
|
|
hosts = {}
|
|
|
|
def __init__(self, name):
|
|
global num
|
|
self.name = name
|
|
if name:
|
|
num += 1
|
|
Host.hosts[name] = self
|
|
self.num = num
|
|
self.dyn = False
|
|
self.upcount = 0
|
|
self.interval = 0
|
|
self.doesack = -1
|
|
self.cmds = []
|
|
self.cver = 0
|
|
self.connections = {}
|
|
self.hdwcounts = [[0,0],[0,0],[0,0]]
|
|
|
|
|
|
def statedict(self):
|
|
d = {}
|
|
d['name'] = self.name
|
|
d['dyn'] = str(self.dyn)
|
|
d['ver'] = str(self.cver)
|
|
d['num'] = self.num
|
|
for c in ['IPv4', 'IPv6']:
|
|
if c in self.connections:
|
|
cs = self.connections[c].statedict()
|
|
else:
|
|
cs = ubConnection.statedict(True)
|
|
for csv in cs:
|
|
d['%s.%s' % (c, csv) ] = cs[csv]
|
|
|
|
return d
|
|
|
|
|
|
def headerdict(self):
|
|
d = {}
|
|
d['name'] = 'Name'
|
|
d['dyn'] = 'Dyn'
|
|
d['ver'] = 'Ver'
|
|
d['num'] = '??'
|
|
for c in ['IPv4', 'IPv6']:
|
|
cs = ubConnection.headerdict(c)
|
|
for csv in cs:
|
|
d['%s.%s' % (c, csv) ] = cs[csv]
|
|
return d
|
|
|
|
|
|
def registerDns(self):
|
|
for af in self.connections:
|
|
self.connections[af].registerDns()
|
|
|
|
|
|
def jsons(self):
|
|
ddict = {}
|
|
for d in self.__dict__:
|
|
if d == 'connections':
|
|
cl = []
|
|
for c in self.connections:
|
|
# dirty ugly hack: fix conn to host backpointer
|
|
cld = copy.deepcopy(self.connections[c].__dict__)
|
|
cld['host'] = cld['host'].name
|
|
cl.append(cld)
|
|
ddict[d] = cl
|
|
else:
|
|
ddict[d] = self.__dict__[d]
|
|
return json.dumps(ddict)
|
|
|
|
|
|
def setcver(self, cver):
|
|
self.cver = cver
|
|
|
|
|
|
def isDynDns(self):
|
|
return self.dyn
|
|
|
|
|
|
def isIPv4(self, addr):
|
|
if type(addr) == type(()):
|
|
return addr[0].find('.') > 0
|
|
else:
|
|
return addr.find('.') > 0
|
|
|
|
|
|
def conndata(self, cid, addr, rtt, now):
|
|
if self.isIPv4(addr):
|
|
afam = "IPv4"
|
|
else:
|
|
afam = "IPv6"
|
|
|
|
if afam not in self.connections:
|
|
self.connections[afam] = Connection(self, cid, addr, afam)
|
|
|
|
conn = self.connections[afam]
|
|
res = conn.newaddr(addr, rtt, now)
|
|
return conn, res
|
|
|
|
|
|
# called when reloading class from pickle, add new fields here
|
|
def fixup(self):
|
|
pass
|
|
|
|
|
|
def dispstate(self):
|
|
if self.state in ["down", "overdue"]:
|
|
state = "<b>%s</b>" % self.state
|
|
elif self.state in ["up", "UP"]:
|
|
state = ""
|
|
for x in self.connections.keys():
|
|
try:
|
|
state += " %5.1f" % (self.connections[x].rtts[-1])
|
|
except:
|
|
state += " %5s" % (self.connections[x].rtts[-1])
|
|
else:
|
|
state = "%s" % self.state
|
|
return state
|
|
|
|
|
|
def dispstats(self):
|
|
if self.doesack != -1:
|
|
if self.upcount > 0:
|
|
# return "(%0.1f%%) %s %s %s " % ((self.doesack * 100.0) / self.upcount, self.doesack, self.upcount, self.hdwcounts)
|
|
r = ""
|
|
for v in xrange(3):
|
|
a,u = self.hdwcounts[v]
|
|
if (self.upcount - u) != 0:
|
|
vs = "%0.0f" % (100.0 - (((self.doesack - a) * 100.0) / (self.upcount - u)))
|
|
if vs == "0":
|
|
vs=""
|
|
else:
|
|
vs="-"
|
|
r+= '<td align="right">%s</td>' % vs
|
|
return r
|
|
else:
|
|
return "<td>(%s)</td><td></td><td></td>" % (self.doesack)
|
|
return '<td align="right">N/A</td><td></td<td></td>>'
|
|
|
|
|
|
hostfields = ['name', 'IPv4.addr', 'IPv4.state', 'IPv4.rtt', 'IPv4.statetime', 'IPv6.addr', 'IPv6.state', 'IPv6.rtt', 'IPv6.statetime', 'ver']
|
|
def htmlheaders(self):
|
|
h = []
|
|
hd = ubHost.headerdict()
|
|
for f in Host.hostfields:
|
|
h.append(hd[f])
|
|
return "<tr><th>"+"</th><th>".join(h)+"</th></tr>\n"
|
|
|
|
|
|
def htmldisp(self):
|
|
h = []
|
|
hd = self.statedict()
|
|
for f in Host.hostfields:
|
|
h.append(hd[f])
|
|
return "<tr><td>"+"</td><td>".join(h)+"</td></tr>\n"
|
|
|
|
# return "<tr><td>%-24s</td><td>%-7s</td>%s<td>%-16s</td><td>%-16s</td><td>%-17s</td><td>%s</td></tr>\n" % \
|
|
# (self.name, self.dispstate(), self.dispstats(), ipv4addr, ipv6addr, lastts, self.cver)
|
|
|
|
|
|
|
|
# create fake "unbound objects", remove in Python 3.0
|
|
ubHost = Host(None)
|
|
ubConnection = Connection(None, "", "", "")
|
|
|