diff options
Diffstat (limited to 'Monitoring/nagios/plugins/nsca_client.py')
| -rwxr-xr-x | Monitoring/nagios/plugins/nsca_client.py | 184 |
1 files changed, 184 insertions, 0 deletions
diff --git a/Monitoring/nagios/plugins/nsca_client.py b/Monitoring/nagios/plugins/nsca_client.py new file mode 100755 index 00000000..76e551db --- /dev/null +++ b/Monitoring/nagios/plugins/nsca_client.py @@ -0,0 +1,184 @@ +### This is a very quick and dirty code for David so he can work on its sikuli agent +# and report as nsca the results. +# This need to be clean a lot, it's still a server and should be a client class :) +# I can do it after my "new baby holidays" are finished ;) +# J.Gabes + + + +import time +import select +import socket +import struct +import sys +import random + + +def decrypt_xor(data, key): + keylen = len(key) + crypted = [chr(ord(data[i]) ^ ord(key[i % keylen])) for i in xrange(len(data))] + return ''.join(crypted) + + + +#Just print some stuff +class NSCA_client(): + def __init__(self, host, port, encryption_method, password): + self.host = host + self.port = port + self.encryption_method = encryption_method + self.password = password + self.rng = random.Random(password) + + + #Ok, main function that is called in the CONFIGURATION phase + def get_objects(self): + print "[Dummy] ask me for objects to return" + r = {'hosts' : []} + h = {'name' : 'dummy host from dummy arbiter module', + 'register' : '0', + } + + r['hosts'].append(h) + print "[Dummy] Returning to Arbiter the hosts:", r + return r + + def send_init_packet(self, socket): + ''' + Build an init packet + 00-127 : IV + 128-131 : unix timestamp + ''' + iv = ''.join([chr(self.rng.randrange(256)) for i in xrange(128)]) + init_packet = struct.pack("!128sI", iv, int(time.mktime(time.gmtime()))) + socket.send(init_packet) + return iv + + def read_check_result(self, data, iv): + ''' + Read the check result + 00-01 : Version + 02-05 : CRC32 + 06-09 : Timestamp + 10-11 : Return code + 12-75 : hostname + 76-203 : service + 204-715 : output of the plugin + 716-720 : padding + ''' + if len(data) != 720: + return None + + if self.encryption_method == 1: + data = decrypt_xor(data,self.password) + data = decrypt_xor(data,iv) + + (version, pad1, crc32, timestamp, rc, hostname_dirty, service_dirty, output_dirty, pad2) = struct.unpack("!hhIIh64s128s512sh",data) + hostname = hostname_dirty.partition("\0", 1)[0] + service = service_dirty.partition("\0", 1)[0] + output = output_dirty.partition("\0", 1)[0] + return (timestamp, rc, hostname, service, output) + + def post_command(self, timestamp, rc, hostname, service, output): + ''' + Send a check result command to the arbiter + ''' + if len(service) == 0: + extcmd = "[%lu] PROCESS_HOST_CHECK_RESULT;%s;%d;%s\n" % (timestamp,hostname,rc,output) + else: + extcmd = "[%lu] PROCESS_SERVICE_CHECK_RESULT;%s;%s;%d;%s\n" % (timestamp,hostname,service,rc,output) + + print "want to send", extcmd + + #e = ExternalCommand(extcmd) + #self.from_q.put(e) + + + # When you are in "external" mode, that is the main loop of your process + def main(self): + #self.set_exit_handler() + self.interrupted = False + backlog = 5 + size = 8192 + server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + #server.setblocking(0) + server.connect((self.host, self.port)) + #server.listen(backlog) + input = [server] + databuffer = {} + IVs = {} + + init = server.recv(size) + print "got init", init + + #init_packet = struct.pack("!128sI",iv,int(time.mktime(time.gmtime()))) + (iv, t) = struct.unpack("!128sI",init) + print "IV", iv + print "T", t + + version = 0 + pad1 = 0 + crc32= 0 + timestamp = int(time.time()) + rc = 2 + hostname_dirty = "moncul" + service_dirty = "fonctionnne" + output_dirty = "blablalba" + pad2=0 + ''' + Read the check result + 00-01 : Version + 02-05 : CRC32 + 06-09 : Timestamp + 10-11 : Return code + 12-75 : hostname + 76-203 : service + 204-715 : output of the plugin + 716-720 : padding + ''' + init_packet = struct.pack("!hhIIh64s128s512sh", version, pad1, crc32, timestamp, rc, hostname_dirty, service_dirty, output_dirty, pad2) + print "Create packent len", len(init_packet) + #(version, pad1, crc32, timestamp, rc, hostname_dirty, service_dirty, output_dirty, pad2) = struct.unpack("!hhIIh64s128s512sh",data) + + data = decrypt_xor(init_packet,iv) + data = decrypt_xor(data,self.password) + + + server.send(data) + sys.exit(0) + + while not self.interrupted: + print "Loop" + inputready,outputready,exceptready = select.select(input,[],[], 1) + + for s in inputready: + if s == server: + # handle the server socket + #client, address = server.accept() + iv = self.send_init_packet(client) + IVs[client] = iv + input.append(client) + else: + # handle all other sockets + data = s.recv(size) + if s in databuffer: + databuffer[s] += data + else: + databuffer[s] = data + if len(databuffer[s]) == 720: + # end-of-transmission or an empty line was received + (timestamp, rc, hostname, service, output)=self.read_check_result(databuffer[s],IVs[s]) + del databuffer[s] + del IVs[s] + self.post_command(timestamp,rc,hostname,service,output) + try: + s.shutdown(2) + except Exception , exp: + print exp + s.close() + input.remove(s) + + + +nsca = NSCA_client('localhost', 5667, 1, 'toto') +nsca.main() |