/usr/share/pyshared/PreludeNotify/heartbeat.py is in prelude-notify 0.9.1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | # Copyright (C) 2008 PreludeIDS Technologies. All Rights Reserved.
# Author: Yoann Vandoorselaere <yoann.v@prelude-ids.com>
#
# This file is part of the Prelude-Notify program.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING. If not, write to
# the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
import time
import gobject
import pynotify
class HeartbeatSource:
def _update_fields(self, idmef):
self._heartbeat_interval = idmef.Get("heartbeat.heartbeat_interval")
self._last = time.time()
self._name = idmef.Get("heartbeat.analyzer(-1).name")
self._model = idmef.Get("heartbeat.analyzer(-1).model")
self._analyzerid = idmef.Get("heartbeat.analyzer(-1).analyzerid")
self._node_name = idmef.Get("heartbeat.analyzer(-1).node.name")
self._node_location = idmef.Get("heartbeat.analyzer(-1).node.location")
self._node_addresses = idmef.Get("heartbeat.analyzer(-1).node.address(*).address")
def __init__(self, env, idmef):
self._env = env
self._delta = 3
self._timer = None
self._is_dead = False
self.update(idmef)
def _genid(self):
str = ""
if self._name:
str += self._name
if self._model:
str += " (%s)" % self._model
elif self._model:
str += self._model
elif self._analyzerid:
str += self._analyzerid
str += " on "
if self._node_name:
str += self._node_name
if self._node_location:
str += ", " + self._node_location
return str
def _heartbeat_timer_cb(self):
self._is_dead = True
self._env.notify.run("high", None, "Missing agent",
"No heartbeat received for %s" % self._genid(), aggregate=False)
return False
def update(self, idmef):
if self._is_dead:
self._is_dead = False
self._env.notify.run(None, None, "Agent back online",
"Missing analyzer %s is back online" % self._genid(), aggregate=False)
if self._timer:
gobject.source_remove(self._timer)
self._update_fields(idmef)
if not self._is_dead:
self._timer = gobject.timeout_add((self._heartbeat_interval + self._delta) * 1000, self._heartbeat_timer_cb)
class HeartbeatMonitor:
def __init__(self, env):
self._env = env
self._table = {}
def heartbeat(self, idmef):
analyzerid = idmef.Get("heartbeat.analyzer(-1).analyzerid")
if self._table.has_key(analyzerid):
self._table[analyzerid].update(idmef)
else:
self._table[analyzerid] = HeartbeatSource(self._env, idmef)
|