new project
This commit is contained in:
Ville Rantanen
116
nando
Executable file
116
nando
Executable file
@@ -0,0 +1,116 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import socket,sys,time,os
|
||||
|
||||
def setup_options():
|
||||
''' Setup the command line options '''
|
||||
from argparse import ArgumentParser
|
||||
|
||||
parser=ArgumentParser(description="Alive notifier.")
|
||||
parser.add_argument("--host",action='store', dest='HOST',type=str,required=True,
|
||||
help="Host name/IP of the server")
|
||||
parser.add_argument("--port", action='store', dest='PORT', default=13370,type=int,
|
||||
help="Port number of the server")
|
||||
parser.add_argument("-t",action='store', dest='TIMEOUT',default=5, type=int,
|
||||
help="Connection timeout")
|
||||
parser.add_argument("-i",action='store', dest='INTERVAL',default=0, type=int,
|
||||
help="Send signal every i seconds. If 0, send only once and exit.")
|
||||
parser.add_argument("-m",action='store', dest='MSGFILE',type=str,default=False,
|
||||
help="Read a text file as description. Max length 512 chars")
|
||||
parser.add_argument("--query-ip",action='store', dest='QUERY_HOSTNAME',default=None,type=str,
|
||||
help="Query the IP of a host name")
|
||||
parser.add_argument("--query-host",action='store', dest='QUERY_IP',default=None,type=str,
|
||||
help="Query the hostname of an IP")
|
||||
parser.add_argument("command",type=str,action="store",default="",nargs="?",
|
||||
choices=['','list','alive','lost'],
|
||||
help="Command to send. If empty, send standard alive-notice. Commands: list, alive")
|
||||
options=parser.parse_args()
|
||||
return options
|
||||
|
||||
def print_table(data):
|
||||
lengths=[0,0,0,0]
|
||||
for row in data.split("\n"):
|
||||
cols=row.split("|",4)
|
||||
if len(cols)!=4:
|
||||
continue
|
||||
for c in range(4):
|
||||
lengths[c]=max(lengths[c], len(cols[c]))
|
||||
lengths=[str(x) for x in lengths]
|
||||
for row in data.split("\n"):
|
||||
cols=row.split("|",4)
|
||||
if len(cols)!=4:
|
||||
continue
|
||||
print(("{0:<"+lengths[0]+"} | {1:<"+lengths[1]+"} | {2:>"+lengths[2]+"} | {3:<"+lengths[3]+"}").format(*cols))
|
||||
|
||||
def query_ip(sock,opts):
|
||||
sock.sendto("list", (opts.HOST, opts.PORT))
|
||||
received = sock.recv(1024)
|
||||
for row in received.split("\n"):
|
||||
cols=row.split("|",4)
|
||||
if cols[0]==opts.QUERY_HOSTNAME:
|
||||
sys.stdout.write(cols[1])
|
||||
sys.exit(0)
|
||||
sys.exit(1)
|
||||
|
||||
def query_host(sock,opts):
|
||||
sock.sendto("list", (opts.HOST, opts.PORT))
|
||||
received = sock.recv(1024)
|
||||
for row in received.split("\n"):
|
||||
cols=row.split("|",4)
|
||||
if cols[1]==opts.QUERY_IP:
|
||||
sys.stdout.write(cols[0])
|
||||
sys.exit(0)
|
||||
sys.exit(1)
|
||||
|
||||
def read_desc(opts):
|
||||
|
||||
if not opts.MSGFILE:
|
||||
return ""
|
||||
if not os.path.isfile(opts.MSGFILE):
|
||||
return ""
|
||||
|
||||
DESC=open(opts.MSGFILE,"rb").read(512)
|
||||
DESC=''.join([ c for c in DESC if c not in ['\n','\r','|'] ])
|
||||
return DESC
|
||||
|
||||
opts=setup_options()
|
||||
|
||||
errors=0
|
||||
while True:
|
||||
try:
|
||||
MYNAME = socket.gethostname()
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
sock.settimeout(opts.TIMEOUT)
|
||||
sock.connect((opts.HOST,opts.PORT))
|
||||
MYIP=sock.getsockname()[0]
|
||||
|
||||
if opts.QUERY_HOSTNAME:
|
||||
query_ip(sock,opts)
|
||||
if opts.QUERY_IP:
|
||||
query_host(sock,opts)
|
||||
|
||||
if opts.command=="":
|
||||
DESC=read_desc(opts)
|
||||
MSG="{0}|{1}|{2}\n".format(MYNAME,MYIP,DESC)
|
||||
else:
|
||||
MSG=opts.command
|
||||
|
||||
sock.sendto(MSG, (opts.HOST, opts.PORT))
|
||||
received = sock.recv(1024)
|
||||
if opts.command!="":
|
||||
print_table(received)
|
||||
if opts.INTERVAL==0 and opts.command=="":
|
||||
sock.sendto("list", (opts.HOST, opts.PORT))
|
||||
received = sock.recv(1024)
|
||||
print_table(received)
|
||||
errors=0
|
||||
except socket.error:
|
||||
print("Didn't get reply from {0}:{1}".format(opts.HOST,opts.PORT))
|
||||
errors=1
|
||||
except IndexError:
|
||||
print(received)
|
||||
if opts.INTERVAL==0:
|
||||
break
|
||||
time.sleep(opts.INTERVAL)
|
||||
|
||||
sys.exit(errors)
|
||||
43
nando-desc
Executable file
43
nando-desc
Executable file
@@ -0,0 +1,43 @@
|
||||
#!/bin/bash
|
||||
|
||||
#~ [ $UID = 0 ] || {
|
||||
#~ echo Must be run as root
|
||||
#~ exit 1
|
||||
#~ }
|
||||
function filesize {
|
||||
# Return a human readable size from integer of bytes
|
||||
# Usage: filesize 10000
|
||||
|
||||
[ "$1" = "0" ] && {
|
||||
echo "0 B"
|
||||
return 0
|
||||
}
|
||||
|
||||
awk 'BEGIN{ x = '$1'
|
||||
split("B KB MB GB TB PB",type)
|
||||
for(i=5;y < 1;i--)
|
||||
y = x / (2^(10*i))
|
||||
str=int(y*10)/10 " " type[i+2]
|
||||
if (x==0) { str = "0 B" }
|
||||
print str
|
||||
}' || return $?
|
||||
}
|
||||
|
||||
|
||||
# Gather list of last user of machine
|
||||
|
||||
LASTUSER=$( who | tail -n 1 | awk '{ print $1 }' )
|
||||
FULLNAME=$( getent passwd "$LASTUSER" | cut -d ':' -f 5 )
|
||||
|
||||
DESC_CORES=$( grep -c ^processor /proc/cpuinfo | tr -d -c [:digit:] )
|
||||
DESC_TYPE=$( grep -m 1 "model name" /proc/cpuinfo | cut -d: -f2 | sed -e 's/^ *//' )
|
||||
DESC_MEM=$( grep MemTotal /proc/meminfo | cut -f2 -d: | tr -d -c [:digit:] )
|
||||
DESC_MEM=$( filesize $(( ${DESC_MEM}*1024 )) )
|
||||
DESC_VERS=$( lsb_release -r | awk '{print $2}' )
|
||||
DESC_UP=$( uptime | sed s/,.*// )
|
||||
DESC_SERIAL=$( dmidecode | grep "Serial Number:" | head -n 1 | cut -d: -f2 | sed -e 's/^ *//' )
|
||||
DESC="$LASTUSER/$FULLNAME/${DESC_CORES}x ${DESC_TYPE}/RAM ${DESC_MEM}/v.${DESC_VERS}/${DESC_UP}/SN ${DESC_SERIAL}"
|
||||
echo "$DESC"
|
||||
|
||||
# exit okay
|
||||
true
|
||||
204
nandod
Executable file
204
nandod
Executable file
@@ -0,0 +1,204 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import SocketServer,threading
|
||||
import sqlite3
|
||||
import time,os,sys
|
||||
|
||||
ALIVE=60*30
|
||||
TABLE_HEAD="Host|IP|Age(d h:m:s)|Description\n"
|
||||
|
||||
def setup_options():
|
||||
''' Setup the command line options '''
|
||||
from argparse import ArgumentParser
|
||||
|
||||
parser=ArgumentParser(description="Alive notifier.")
|
||||
parser.add_argument("--host",action='store', dest='HOST',default='0.0.0.0',type=str,
|
||||
help="Bind to address")
|
||||
parser.add_argument("--port", action='store', dest='PORT', default=13370,type=int,
|
||||
help="Bind to port")
|
||||
parser.add_argument("--db",action='store', dest='DB',default='/tmp/nando.sqlite',type=str,
|
||||
help="Sqlite file for database.")
|
||||
parser.add_argument("--quiet","-q",action='store_true', dest='quiet',default=False,
|
||||
help="Quiet operation.")
|
||||
options=parser.parse_args()
|
||||
return options
|
||||
|
||||
|
||||
class UDPHandler(SocketServer.BaseRequestHandler):
|
||||
def handle(self):
|
||||
data = self.request[0].strip()
|
||||
socket = self.request[1]
|
||||
if not opts.quiet:
|
||||
print "{0} wrote: {1}".format(self.client_address[0],data)
|
||||
if data=="list":
|
||||
reply=DB.list_all()
|
||||
elif data=="alive":
|
||||
reply=DB.list_alive()
|
||||
elif data=="lost":
|
||||
reply=DB.list_lost()
|
||||
else:
|
||||
DB.update(data)
|
||||
reply="OK"
|
||||
socket.sendto(reply, self.client_address)
|
||||
|
||||
class TCPHandler(SocketServer.BaseRequestHandler):
|
||||
def handle(self):
|
||||
if not opts.quiet:
|
||||
print "TCP request: {0}".format(self.client_address[0])
|
||||
self.request.send('HTTP/1.0 200 OK\r\n')
|
||||
self.request.send("Content-Type: text/html\r\n\r\n")
|
||||
self.request.sendall(HTMLDB.HTML_list())
|
||||
self.request.close()
|
||||
|
||||
class DataBase:
|
||||
def __init__(self,DB):
|
||||
self.DBfile=DB
|
||||
self.conn=None
|
||||
self.db=None
|
||||
if not os.path.exists(self.DBfile):
|
||||
self.createDB()
|
||||
self.conn_init()
|
||||
|
||||
def conn_init(self):
|
||||
self.conn=sqlite3.connect(self.DBfile)
|
||||
self.db=self.conn.cursor()
|
||||
self.conn.text_factory=str
|
||||
|
||||
def conn_end(self):
|
||||
self.conn.commit()
|
||||
#self.conn.close()
|
||||
|
||||
def createDB(self):
|
||||
self.conn_init()
|
||||
self.db.execute('CREATE TABLE alive (id TEXT PRIMARY KEY,\
|
||||
ip TEXT,\
|
||||
desc TEXT,\
|
||||
date INTEGER)')
|
||||
self.conn_end()
|
||||
|
||||
def update(self,id):
|
||||
try:
|
||||
host,ip,desc=id.split("|",3)
|
||||
desc=''.join([ c for c in desc if c not in ['\n','\r','|'] ])
|
||||
if len(ip.split("."))!=4:
|
||||
return 1
|
||||
except:
|
||||
return
|
||||
self.db=self.conn.cursor()
|
||||
self.db.execute("INSERT OR REPLACE INTO alive(id,ip,date,desc) \
|
||||
VALUES(?,?,?,?)",(host,ip,int(time.time()),desc))
|
||||
self.conn_end()
|
||||
|
||||
def list_all(self):
|
||||
self.db=self.conn.cursor()
|
||||
self.db.execute("SELECT id,ip,date,desc FROM alive ORDER BY id")
|
||||
alive=[]
|
||||
lost=[]
|
||||
for row in self.db:
|
||||
age=int(time.time())-row[2]
|
||||
if age<ALIVE:
|
||||
alive.append("{0}|{1}|{2}|{3}".format(row[0],row[1], humanize_time(age),row[3]))
|
||||
else:
|
||||
lost.append( "{0}|{1}|{2}|{3}".format(row[0],row[1], humanize_time(age),row[3]))
|
||||
return TABLE_HEAD+"\n".join(alive)+"\n---|---|---|---\n"+"\n".join(lost)
|
||||
|
||||
def list_alive(self):
|
||||
self.db=self.conn.cursor()
|
||||
self.db.execute("SELECT id,ip,date,desc FROM alive ORDER BY date DESC")
|
||||
msg=[]
|
||||
for row in self.db:
|
||||
age=int(time.time())-row[2]
|
||||
if age<ALIVE:
|
||||
msg.append("{0}|{1}|{2}|{3}".format(row[0],row[1], humanize_time(age),row[3]))
|
||||
return TABLE_HEAD+"\n".join(msg)
|
||||
|
||||
def list_lost(self):
|
||||
self.db=self.conn.cursor()
|
||||
self.db.execute("SELECT id,ip,date,desc FROM alive ORDER BY date")
|
||||
msg=[]
|
||||
for row in self.db:
|
||||
age=int(time.time())-row[2]
|
||||
if age>=ALIVE:
|
||||
msg.append("{0}|{1}|{2}|{3}".format(row[0],row[1], humanize_time(age),row[3]))
|
||||
return TABLE_HEAD+"\n".join(msg)
|
||||
|
||||
def HTML_list(self):
|
||||
msg=[]
|
||||
msg.append('''<html>
|
||||
<head>
|
||||
<style>
|
||||
body {
|
||||
font-family: monospace;
|
||||
}
|
||||
table {
|
||||
background-color: gray;
|
||||
}
|
||||
td, th {
|
||||
text-align: left;
|
||||
background-color: lightgray;
|
||||
padding: 0.5ex;
|
||||
}
|
||||
.right {
|
||||
text-align: right;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body><h1>Alive</h1><table>''')
|
||||
self.db=self.conn.cursor()
|
||||
self.db.execute("SELECT id,ip,date,desc FROM alive ORDER BY id")
|
||||
msg.append('<tr><th>{0}</th><th>{1}</th><th class="right">{2}</th><th>{3}</th></tr>'.format('Host','IP','Age[d h:m:s]','Description'))
|
||||
for row in self.db:
|
||||
age=int(time.time())-row[2]
|
||||
if age <ALIVE:
|
||||
msg.append('<tr><td>{0}</td><td>{1}</td><td class="right">{2}</td><td>{3}</td></tr>'.format(row[0],row[1],humanize_time(age),row[3]))
|
||||
msg.append('</table><h1>Lost</h1><table>')
|
||||
msg.append('<tr><th>{0}</th><th>{1}</th><th class="right">{2}</th><th>{3}</th></tr>'.format('Host','IP','Age[d h:m:s]','Description'))
|
||||
self.db.execute("SELECT id,ip,date,desc FROM alive ORDER BY id")
|
||||
for row in self.db:
|
||||
age=int(time.time())-row[2]
|
||||
if age >=ALIVE:
|
||||
msg.append('<tr><td>{0}</td><td>{1}</td><td class="right">{2}</td><td>{3}</td></tr>'.format(row[0],row[1],humanize_time(age),row[3]))
|
||||
msg.append('</table></body></html>')
|
||||
|
||||
return "\n\r".join(msg)
|
||||
|
||||
def UDPserve():
|
||||
global DB
|
||||
global server
|
||||
DB=DataBase(opts.DB)
|
||||
server = SocketServer.UDPServer((opts.HOST, opts.PORT), UDPHandler)
|
||||
server.serve_forever()
|
||||
def TCPserve():
|
||||
global HTMLDB
|
||||
global HTMLserver
|
||||
HTMLDB=DataBase(opts.DB)
|
||||
HTMLserver = SocketServer.TCPServer((opts.HOST, opts.PORT), TCPHandler)
|
||||
HTMLserver.serve_forever()
|
||||
|
||||
def humanize_time(secs):
|
||||
mins, secs = divmod(secs, 60)
|
||||
hours, mins = divmod(mins, 60)
|
||||
if hours < 24:
|
||||
return '%02d:%02d:%02d' % (hours, mins, secs)
|
||||
else:
|
||||
days, hours = divmod(hours, 24)
|
||||
return '%dd %02d:%02d:%02d' % (days, hours, mins, secs)
|
||||
|
||||
if __name__ == "__main__":
|
||||
opts=setup_options()
|
||||
if not opts.quiet:
|
||||
print("Starting NandoD {0}:{1}".format(opts.HOST,opts.PORT))
|
||||
UDP=threading.Thread(target=UDPserve)
|
||||
TCP=threading.Thread(target=TCPserve)
|
||||
UDP.daemon=True
|
||||
TCP.daemon=True
|
||||
UDP.start()
|
||||
TCP.start()
|
||||
while True:
|
||||
try:
|
||||
time.sleep(1)
|
||||
except KeyboardInterrupt:
|
||||
print("Exiting..")
|
||||
server.shutdown()
|
||||
HTMLserver.shutdown()
|
||||
sys.exit(0)
|
||||
Reference in New Issue
Block a user