from mgr_module import MgrModule import datetime import errno import json from collections import defaultdict from prettytable import PrettyTable from threading import Event, Lock DATEFMT = '%Y-%m-%d %H:%M:%S.%f' MAX_WAIT = 600 MIN_WAIT = 60 class Module(MgrModule): MODULE_OPTIONS = [ { 'name': 'warn_recent_interval', 'type': 'secs', 'default': 60*60*24*14, 'desc': 'time interval in which to warn about recent crashes', 'runtime': True, }, { 'name': 'retain_interval', 'type': 'secs', 'default': 60*60*24 * 365, 'desc': 'how long to retain crashes before pruning them', 'runtime': True, }, ] def __init__(self, *args, **kwargs): super(Module, self).__init__(*args, **kwargs) self.crashes = None self.crashes_lock = Lock() self.run = True self.event = Event() def shutdown(self): self.run = False self.event.set() def serve(self): self.config_notify() while self.run: with self.crashes_lock: self._refresh_health_checks() self._prune(self.retain_interval) wait = min(MAX_WAIT, max(self.warn_recent_interval / 100, MIN_WAIT)) self.event.wait(wait) self.event.clear() def config_notify(self): for opt in self.MODULE_OPTIONS: setattr(self, opt['name'], self.get_module_option(opt['name'])) self.log.debug(' mgr option %s = %s', opt['name'], getattr(self, opt['name'])) def _load_crashes(self): raw = self.get_store_prefix('crash/') self.crashes = {k[6:]: json.loads(m) for (k, m) in raw.items()} def _refresh_health_checks(self): if not self.crashes: self._load_crashes() cutoff = datetime.datetime.utcnow() - datetime.timedelta( seconds=self.warn_recent_interval) recent = { crashid: crash for crashid, crash in self.crashes.items() if self.time_from_string(crash['timestamp']) > cutoff and 'archived' not in crash } num = len(recent) health_checks = {} if recent: detail = [ '%s crashed on host %s at %s' % ( crash.get('entity_name', 'unidentified daemon'), crash.get('utsname_hostname', '(unknown)'), crash.get('timestamp', 'unknown time')) for (_, crash) in recent.items()] if num > 30: detail = detail[0:30] detail.append('and %d more' % (num - 30)) self.log.debug('detail %s' % detail) health_checks['RECENT_CRASH'] = { 'severity': 'warning', 'summary': '%d daemons have recently crashed' % (num), 'detail': detail, } self.set_health_checks(health_checks) def handle_command(self, inbuf, command): with self.crashes_lock: if not self.crashes: self._load_crashes() for cmd in self.COMMANDS: if cmd['cmd'].startswith(command['prefix']): handler = cmd['handler'] break if handler is None: return errno.EINVAL, '', 'unknown command %s' % command['prefix'] return handler(self, command, inbuf) def time_from_string(self, timestr): # drop the 'Z' timezone indication, it's always UTC timestr = timestr.rstrip('Z') return datetime.datetime.strptime(timestr, DATEFMT) def validate_crash_metadata(self, inbuf): # raise any exceptions to caller metadata = json.loads(inbuf) for f in ['crash_id', 'timestamp']: if f not in metadata: raise AttributeError("missing '%s' field" % f) time = self.time_from_string(metadata['timestamp']) return metadata def timestamp_filter(self, f): """ Filter crash reports by timestamp. :param f: f(time) return true to keep crash report :returns: crash reports for which f(time) returns true """ def inner(pair): _, crash = pair time = self.time_from_string(crash["timestamp"]) return f(time) return filter(inner, self.crashes.items()) # command handlers def do_info(self, cmd, inbuf): crashid = cmd['id'] crash = self.crashes.get(crashid) if not crash: return errno.EINVAL, '', 'crash info: %s not found' % crashid val = json.dumps(crash, indent=4) return 0, val, '' def do_post(self, cmd, inbuf): try: metadata = self.validate_crash_metadata(inbuf) except Exception as e: return errno.EINVAL, '', 'malformed crash metadata: %s' % e crashid = metadata['crash_id'] if crashid not in self.crashes: self.crashes[crashid] = metadata key = 'crash/%s' % crashid self.set_store(key, json.dumps(metadata)) self._refresh_health_checks() return 0, '', '' def ls(self): if not self.crashes: self._load_crashes() return self.do_ls({'prefix': 'crash ls'}, '') def do_ls(self, cmd, inbuf): if cmd['prefix'] == 'crash ls': t = self.crashes.values() else: t = [crash for crashid, crash in self.crashes.items() if 'archived' not in crash] r = sorted(t, key=lambda i: i.get('crash_id')) if cmd.get('format') == 'json' or cmd.get('format') == 'json-pretty': return 0, json.dumps(r, indent=4), '' else: table = PrettyTable(['ID', 'ENTITY', 'NEW'], border=False) table.left_padding_width = 0 table.right_padding_width = 1 table.align['ID'] = 'l' table.align['ENTITY'] = 'l' for c in r: table.add_row([c.get('crash_id'), c.get('entity_name','unknown'), '' if 'archived' in c else '*']) return 0, table.get_string(), '' def do_rm(self, cmd, inbuf): crashid = cmd['id'] if crashid in self.crashes: del self.crashes[crashid] key = 'crash/%s' % crashid self.set_store(key, None) # removes key self._refresh_health_checks() return 0, '', '' def do_prune(self, cmd, inbuf): keep = cmd['keep'] try: keep = int(keep) except ValueError: return errno.EINVAL, '', 'keep argument must be integer' self._prune(keep * 60*60*24) return 0, '', '' def _prune(self, seconds): now = datetime.datetime.utcnow() cutoff = now - datetime.timedelta(seconds=seconds) removed_any = False # make a copy of the list, since we'll modify self.crashes below to_prune = list(self.timestamp_filter(lambda ts: ts <= cutoff)) for crashid, crash in to_prune: del self.crashes[crashid] key = 'crash/%s' % crashid self.set_store(key, None) removed_any = True if removed_any: self._refresh_health_checks() def do_archive(self, cmd, inbuf): crashid = cmd['id'] crash = self.crashes.get(crashid) if not crash: return errno.EINVAL, '', 'crash info: %s not found' % crashid if not crash.get('archived'): crash['archived'] = str(datetime.datetime.utcnow()) self.crashes[crashid] = crash key = 'crash/%s' % crashid self.set_store(key, json.dumps(crash)) self._refresh_health_checks() return 0, '', '' def do_archive_all(self, cmd, inbuf): for crashid, crash in self.crashes.items(): if not crash.get('archived'): crash['archived'] = str(datetime.datetime.utcnow()) self.crashes[crashid] = crash key = 'crash/%s' % crashid self.set_store(key, json.dumps(crash)) self._refresh_health_checks() return 0, '', '' def do_stat(self, cmd, inbuf): # age in days for reporting, ordered smallest first bins = [1, 3, 7] retlines = list() def binstr(bindict): binlines = list() count = len(bindict['idlist']) if count: binlines.append( '%d older than %s days old:' % (count, bindict['age']) ) for crashid in bindict['idlist']: binlines.append(crashid) return '\n'.join(binlines) total = 0 now = datetime.datetime.utcnow() for i, age in enumerate(bins): agelimit = now - datetime.timedelta(days=age) bins[i] = { 'age': age, 'agelimit': agelimit, 'idlist': list() } for crashid, crash in self.crashes.items(): total += 1 stamp = self.time_from_string(crash['timestamp']) for i, bindict in enumerate(bins): if stamp <= bindict['agelimit']: bindict['idlist'].append(crashid) # don't count this one again continue retlines.append('%d crashes recorded' % total) for bindict in bins: retlines.append(binstr(bindict)) return 0, '\n'.join(retlines), '' def do_json_report(self, cmd, inbuf): """ Return a machine readable summary of recent crashes. """ try: hours = int(cmd['hours']) except ValueError: return errno.EINVAL, '', ' argument must be integer' report = defaultdict(lambda: 0) for crashid, crash in self.crashes.items(): pname = crash.get("process_name", "unknown") if not pname: pname = "unknown" report[pname] += 1 return 0, '', json.dumps(report) def self_test(self): # test time conversion timestr = '2018-06-22 20:35:38.058818Z' dt = self.time_from_string(timestr) if dt != datetime.datetime(2018, 6, 22, 20, 35, 38, 58818): raise RuntimeError('time_from_string() failed') COMMANDS = [ { 'cmd': 'crash info name=id,type=CephString', 'desc': 'show crash dump metadata', 'perm': 'r', 'handler': do_info, }, { 'cmd': 'crash ls', 'desc': 'Show new and archived crash dumps', 'perm': 'r', 'handler': do_ls, }, { 'cmd': 'crash ls-new', 'desc': 'Show new crash dumps', 'perm': 'r', 'handler': do_ls, }, { 'cmd': 'crash post', 'desc': 'Add a crash dump (use -i )', 'perm': 'rw', 'handler': do_post, }, { 'cmd': 'crash prune name=keep,type=CephString', 'desc': 'Remove crashes older than days', 'perm': 'rw', 'handler': do_prune, }, { 'cmd': 'crash rm name=id,type=CephString', 'desc': 'Remove a saved crash ', 'perm': 'rw', 'handler': do_rm, }, { 'cmd': 'crash stat', 'desc': 'Summarize recorded crashes', 'perm': 'r', 'handler': do_stat, }, { 'cmd': 'crash json_report name=hours,type=CephString', 'desc': 'Crashes in the last hours', 'perm': 'r', 'handler': do_json_report, }, { 'cmd': 'crash archive name=id,type=CephString', 'desc': 'Acknowledge a crash and silence health warning(s)', 'perm': 'w', 'handler': do_archive, }, { 'cmd': 'crash archive-all', 'desc': 'Acknowledge all new crashes and silence health warning(s)', 'perm': 'w', 'handler': do_archive_all, }, ]