ffpb.py 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762
  1. # -*- coding: utf-8 -*-
  2. from __future__ import print_function
  3. import willie
  4. import datetime
  5. import difflib
  6. from email.utils import mktime_tz
  7. import git
  8. import netaddr
  9. import json
  10. import urllib2
  11. import re
  12. import os
  13. import shelve
  14. import subprocess
  15. import time
  16. import dns.resolver,dns.reversename
  17. import socket
  18. import SocketServer
  19. import threading
  20. msgserver = None
  21. peers_repo = None
  22. monitored_nodes = None
  23. highscores = None
  24. alfred_method = None
  25. alfred_data = None
  26. alfred_update = datetime.datetime(1970,1,1,23,42)
  27. ffpb_resolver = dns.resolver.Resolver ()
  28. ffpb_resolver.nameservers = ['10.132.254.53']
  29. class MsgHandler(SocketServer.BaseRequestHandler):
  30. def handle(self):
  31. data = self.request.recv(2048).strip()
  32. sender = self._resolve_name (self.client_address[0])
  33. bot = self.server.bot
  34. if bot is None:
  35. print("ERROR: No bot in handle() :-(")
  36. return
  37. target = bot.config.core.owner
  38. if bot.config.has_section('ffpb'):
  39. is_public = data.lstrip().lower().startswith("public:")
  40. if is_public and not (bot.config.ffpb.msg_target_public is None):
  41. data = data[7:].lstrip()
  42. target = bot.config.ffpb.msg_target_public
  43. elif not (bot.config.ffpb.msg_target is None):
  44. target = bot.config.ffpb.msg_target
  45. bot.msg(target, "[{0}] {1}".format(sender, str(data)))
  46. def _resolve_name (self, ip):
  47. if ip.startswith ("127."):
  48. return "localhost"
  49. try:
  50. addr = dns.reversename.from_address (ip)
  51. return re.sub ("(.infra)?.ffpb.", "", str (ffpb_resolver.query (addr, "PTR")[0]))
  52. except dns.resolver.NXDOMAIN:
  53. return ip
  54. class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
  55. pass
  56. def setup(bot):
  57. global msgserver, peers_repo, alfred_method, highscores, monitored_nodes
  58. bot.memory['ffpb_in_setup'] = True
  59. highscores = shelve.open('highscoredata', writeback=True)
  60. if not 'nodes' in highscores:
  61. highscores['nodes'] = 0
  62. highscores['nodes_ts'] = time.time()
  63. if not 'clients' in highscores:
  64. highscores['clients'] = 0
  65. highscores['clients_ts'] = time.time()
  66. monitored_nodes = shelve.open('monitorednodes', writeback=True)
  67. if not bot.config.has_section('ffpb'):
  68. bot.memory['ffpb_in_setup'] = False
  69. return
  70. if not bot.config.ffpb.peers_directory is None:
  71. peers_repo = git.Repo(bot.config.ffpb.peers_directory)
  72. assert peers_repo.bare == False
  73. if int(bot.config.ffpb.msg_enable) == 1:
  74. host = "localhost"
  75. port = 2342
  76. if not bot.config.ffpb.msg_host is None: host = bot.config.ffpb.msg_host
  77. if not bot.config.ffpb.msg_port is None: port = int(bot.config.ffpb.msg_port)
  78. msgserver = ThreadingTCPServer((host,port), MsgHandler)
  79. msgserver.bot = bot
  80. ip, port = msgserver.server_address
  81. print("Messaging server listening on {}:{}".format(ip,port))
  82. msgserver_thread = threading.Thread(target=msgserver.serve_forever)
  83. msgserver_thread.daemon = True
  84. msgserver_thread.start()
  85. alfred_method = bot.config.ffpb.alfred_method
  86. ffpb_updatealfred(bot)
  87. bot.memory['ffpb_in_setup'] = False
  88. def shutdown(bot):
  89. global msgserver, highscores, monitored_nodes
  90. if not highscores is None:
  91. highscores.sync()
  92. highscores.close()
  93. highscores = None
  94. if not monitored_nodes is None:
  95. monitored_nodes.sync()
  96. monitored_nodes.close()
  97. monitored_nodes = None
  98. if not msgserver is None:
  99. msgserver.shutdown()
  100. print("Closed messaging server.")
  101. msgserver = None
  102. @willie.module.commands("help")
  103. @willie.module.commands("hilfe")
  104. @willie.module.commands("man")
  105. def ffpb_help(bot, trigger):
  106. functions = {
  107. "!ping <knoten>": "Prüfe ob der Knoten erreichbar ist.",
  108. "!status": "Aktuellen Status des Netzwerks (insb. Anzahl Knoten und Clients) ausgegeben.",
  109. "!info <knoten>": "Allgemeine Information zu dem Knoten anzeigen.",
  110. "!link <knoten>": "MAC-Adresse und Link zur Status-Seite des Knotens anzeigen.",
  111. "!exec-on-peer <knoten> <kommando>": "Befehl auf dem Knoten ausführen (nur möglich bei eigenen Knoten oder als Admin, in beiden Fällen auch nur wenn der SSH-Key des Bots hinterlegt wurde)",
  112. }
  113. param = trigger.group(2)
  114. if param is None:
  115. bot.say("Funktionen: " + str.join(", ", sorted(functions.keys())))
  116. return
  117. if param.startswith("!"): param = param[1:]
  118. for fun in functions.keys():
  119. if fun.startswith("!" + param + " "):
  120. bot.say("Hilfe zu '" + fun + "': " + functions[fun])
  121. return
  122. bot.say("Allgemeine Hilfe gibt's mit !help - ohne Parameter.")
  123. def ffpb_findnode(name):
  124. if name is None or len(name) == 0:
  125. return None
  126. name = str(name).strip()
  127. names = {}
  128. # try to match MAC
  129. m = re.search("^([0-9a-fA-F][0-9a-fA-F]:){5}[0-9a-fA-F][0-9a-fA-F]$", name)
  130. if (not m is None):
  131. mac = m.group(0).lower()
  132. if mac in alfred_data:
  133. return alfred_data[mac]
  134. # try to find alias MAC
  135. for nodeid in alfred_data:
  136. node = alfred_data[nodeid]
  137. if "network" in node:
  138. if "mac" in node["network"] and node["network"]["mac"].lower() == mac:
  139. return node
  140. if "mesh_interfaces" in node["network"]:
  141. for mim in node["network"]["mesh_interfaces"]:
  142. if mim.lower() == mac:
  143. return node
  144. return {
  145. 'hostname': '?-' + mac.replace(':','').lower(),
  146. 'network': { 'addresses': [ mac2ipv6(mac, 'fdca:ffee:ff12:132:') ], 'mac': mac, },
  147. 'hardware': { 'model': 'derived-from-mac' },
  148. }
  149. # look through the ALFRED peers
  150. for nodeid in alfred_data:
  151. node = alfred_data[nodeid]
  152. if 'hostname' in node:
  153. h = node['hostname']
  154. if h.lower() == name.lower():
  155. return node
  156. else:
  157. names[h] = nodeid
  158. # still not found -> try peers_repo
  159. if not peers_repo is None:
  160. peer_name = None
  161. peer_mac = None
  162. peer_file = None
  163. for b in peers_repo.heads.master.commit.tree.blobs:
  164. if b.name.lower() == name.lower():
  165. peer_name = b.name
  166. peer_file = b.abspath
  167. break
  168. if (not peer_file is None) and os.path.exists(peer_file):
  169. peerfile = open(peer_file, "r")
  170. for line in peerfile:
  171. if line.startswith("# MAC:"):
  172. peer_mac = line[6:].strip()
  173. peerfile.close()
  174. if not (peer_mac is None):
  175. return {
  176. 'hostname': peer_name,
  177. 'network': { 'addresses': [ mac2ipv6(peer_mac, 'fdca:ffee:ff12:132:') ], 'mac': peer_mac },
  178. 'hardware': { 'model': 'derived-from-vpnkeys' },
  179. }
  180. # do a similar name lookup in the ALFRED data
  181. possibilities = difflib.get_close_matches(name, [ x for x in names ], cutoff=0.8)
  182. print('findnode: Fuzzy matching \'{0}\' got {1} entries: {2}'.format(name, len(possibilities), ', '.join(possibilities)))
  183. if len(possibilities) == 1:
  184. # if we got exactly one candidate that might be it
  185. return alfred_data[names[possibilities[0]]]
  186. return None
  187. def ffpb_findnode_from_botparam(bot, name, ensure_recent_alfreddata = True):
  188. if (name is None or len(name) == 0):
  189. bot.reply("Grün.")
  190. return None
  191. if ensure_recent_alfreddata and alfred_data is None:
  192. bot.say("Informationen sind ausverkauft, kommen erst morgen wieder rein.")
  193. return None
  194. if ensure_recent_alfreddata and ffpb_alfred_data_outdated():
  195. bot.say("Ich habe gerade keine aktuellen Informationen, daher sage ich mal lieber nichts zu '" + name + "'.")
  196. return None
  197. node = ffpb_findnode(name)
  198. if node is None:
  199. bot.say("Kein Plan wer oder was mit '" + name + "' gemeint ist :(")
  200. return node
  201. def mac2ipv6(mac, prefix=None):
  202. result = str(netaddr.EUI(mac).ipv6_link_local())
  203. if (not prefix is None) and (result.startswith("fe80::")):
  204. result = prefix + result[6:]
  205. return result
  206. @willie.module.interval(30)
  207. def ffpb_updatealfred(bot):
  208. """Aktualisiere ALFRED-Daten"""
  209. global alfred_data, alfred_update
  210. if alfred_method is None or alfred_method == "None":
  211. return
  212. updated = None
  213. if alfred_method == "exec":
  214. rawdata = subprocess.check_output(['alfred-json', '-z', '-r', '158'])
  215. updated = datetime.datetime.now()
  216. elif alfred_method.startswith("http"):
  217. try:
  218. rawdata = urllib2.urlopen(alfred_method)
  219. except:
  220. print("Failed to download ALFRED data.")
  221. return
  222. updated = datetime.datetime.fromtimestamp(mktime_tz(rawdata.info().getdate_tz("Last-Modified")))
  223. else:
  224. print("Unknown ALFRED data method '", alfred_method, "', cannot load new data.", sep="")
  225. alfred_data = None
  226. return
  227. try:
  228. alfred_data = json.load(rawdata)
  229. #print("Fetched new ALFRED data:", len(alfred_data), "entries")
  230. alfred_update = updated
  231. except ValueError as e:
  232. print("Failed to parse ALFRED data: " + str(e))
  233. return
  234. def ffpb_alfred_data_outdated():
  235. timeout = datetime.datetime.now() - datetime.timedelta(minutes=5)
  236. is_outdated = timeout > alfred_update
  237. #print("ALFRED outdated? {0} (timeout={1} vs. lastupdate={2})".format(is_outdated, timeout, alfred_update))
  238. return is_outdated
  239. @willie.module.commands('debug-alfred')
  240. def ffpb_debug_alfred(bot, trigger):
  241. if alfred_data is None:
  242. bot.say("Keine ALFRED-Daten vorhanden.")
  243. else:
  244. bot.say("ALFRED Daten: count={0} lastupdate={1}".format(len(alfred_data), alfred_update))
  245. @willie.module.commands('alfred-data')
  246. def ffpb_peerdata(bot, trigger):
  247. if (not trigger.admin):
  248. bot.say('I wont leak (possibly) sensitive data to you.')
  249. return
  250. if (not trigger.is_privmsg) and (not trigger.nick in bot.ops[trigger.sender]):
  251. bot.say('Kein Keks? Keine Daten.')
  252. return
  253. target_name = trigger.group(2)
  254. node = ffpb_findnode_from_botparam(bot, target_name)
  255. if node is None: return
  256. for key in node:
  257. if key in [ 'hostname' ]: continue
  258. bot.say("{0}.{1} = {2}".format(node['hostname'], key, str(node[key])))
  259. @willie.module.commands('info')
  260. def ffpb_peerinfo(bot, trigger):
  261. target_name = trigger.group(2)
  262. node = ffpb_findnode_from_botparam(bot, target_name)
  263. if node is None: return
  264. info_mac = node["network"]["mac"]
  265. info_name = node["hostname"]
  266. info_hw = ""
  267. if "hardware" in node:
  268. if "model" in node["hardware"]:
  269. model = node["hardware"]["model"]
  270. info_hw = " model='" + model + "'"
  271. info_fw = ""
  272. info_update = ""
  273. if "software" in node:
  274. if "firmware" in node["software"]:
  275. fwinfo = str(node["software"]["firmware"]["release"]) if "release" in node["software"]["firmware"] else "unknown"
  276. info_fw = " firmware=" + fwinfo
  277. if "autoupdater" in node["software"]:
  278. autoupdater = node["software"]["autoupdater"]["branch"] if node["software"]["autoupdater"]["enabled"] else "off"
  279. info_update = " (autoupdater="+autoupdater+")"
  280. info_uptime = ""
  281. u = -1
  282. if "statistics" in node and "uptime" in node["statistics"]:
  283. u = int(float(node["statistics"]["uptime"]))
  284. elif 'uptime' in node:
  285. u = int(float(node['uptime']))
  286. if u > 0:
  287. d, r1 = divmod(u, 86400)
  288. h, r2 = divmod(r1, 3600)
  289. m, s = divmod(r2, 60)
  290. if d > 0:
  291. info_uptime = ' up {0}d {1}h'.format(d,h)
  292. elif h > 0:
  293. info_uptime = ' up {0}h {1}m'.format(h,m)
  294. else:
  295. info_uptime = ' up {0}m'.format(m)
  296. bot.say('[{1}]{2}{3}{4}{5}'.format(info_mac, info_name, info_hw, info_fw, info_update, info_uptime))
  297. @willie.module.commands('uptime')
  298. def ffpb_peeruptime(bot, trigger):
  299. target_name = trigger.group(2)
  300. node = ffpb_findnode_from_botparam(bot, target_name)
  301. if node is None: return
  302. info_name = node["hostname"]
  303. info_uptime = ''
  304. u_raw = None
  305. if 'statistics' in node and 'uptime' in node['statistics']:
  306. u_raw = node['statistics']['uptime']
  307. elif 'uptime' in node:
  308. u_raw = node['uptime']
  309. if not u_raw is None:
  310. u = int(float(u_raw))
  311. d, r1 = divmod(u, 86400)
  312. h, r2 = divmod(r1, 3600)
  313. m, s = divmod(r2, 60)
  314. if d > 0:
  315. info_uptime += '{0}d '.format(d)
  316. if h > 0:
  317. info_uptime += '{0}h '.format(h)
  318. info_uptime += '{0}m'.format(m)
  319. info_uptime += ' # raw: \'{0}\''.format(u_raw)
  320. else:
  321. info_uptime += '?'
  322. bot.say('uptime(\'{0}\') = {1}'.format(info_name, info_uptime))
  323. @willie.module.commands('link')
  324. def ffpb_peerlink(bot, trigger):
  325. target_name = trigger.group(2)
  326. node = ffpb_findnode_from_botparam(bot, target_name)
  327. if node is None: return
  328. info_mac = node["network"]["mac"]
  329. info_name = node["hostname"]
  330. info_v6 = mac2ipv6(info_mac, 'fdca:ffee:ff12:132:')
  331. bot.say('[{1}] mac {0} -> http://[{2}]/'.format(info_mac, info_name, info_v6))
  332. @willie.module.interval(60)
  333. def ffpb_updatepeers(bot):
  334. """Aktualisiere die Knotenliste und melde das Diff"""
  335. if peers_repo is None:
  336. print('WARNING: peers_repo is None')
  337. return
  338. old_head = peers_repo.head.commit
  339. peers_repo.remotes.origin.pull()
  340. new_head = peers_repo.head.commit
  341. if new_head != old_head:
  342. print('git pull: from ' + str(old_head) + ' to ' + str(new_head))
  343. added = []
  344. changed = []
  345. renamed = []
  346. deleted = []
  347. for f in old_head.diff(new_head):
  348. if f.new_file:
  349. added.append(f.b_blob.name)
  350. elif f.deleted_file:
  351. deleted.append(f.a_blob.name)
  352. elif f.renamed:
  353. renamed.append([f.rename_from, f.rename_to])
  354. else:
  355. changed.append(f.a_blob.name)
  356. response = "Knoten-Update (VPN +{0} %{1} -{2}): ".format(len(added), len(renamed)+len(changed), len(deleted))
  357. for f in added:
  358. response += " +'{}'".format(f)
  359. for f in changed:
  360. response += " %'{}'".format(f)
  361. for f in renamed:
  362. response += " '{}'->'{}'".format(f[0],f[1])
  363. for f in deleted:
  364. response += " -'{}'".format(f)
  365. bot.msg(bot.config.ffpb.msg_target, response)
  366. def ffpb_fetch_stats(bot, url, memoryid):
  367. response = urllib2.urlopen(url)
  368. data = json.load(response)
  369. nodes_active = 0
  370. nodes_total = 0
  371. clients_count = 0
  372. for node in data['nodes']:
  373. if node['flags']['gateway'] or node['flags']['client']:
  374. continue
  375. nodes_total += 1
  376. if node['flags']['online']:
  377. nodes_active += 1
  378. if 'legacy' in node['flags'] and node['flags']['legacy']:
  379. clients_count -= 1
  380. for link in data['links']:
  381. if link['type'] == 'client':
  382. clients_count += 1
  383. if not memoryid in bot.memory:
  384. bot.memory[memoryid] = { }
  385. stats = bot.memory[memoryid]
  386. stats["fetchtime"] = time.time()
  387. stats["nodes_active"] = nodes_active
  388. stats["nodes_total"] = nodes_total
  389. stats["clients"] = clients_count
  390. return (nodes_active, nodes_total, clients_count)
  391. @willie.module.interval(15)
  392. def ffpb_get_stats(bot):
  393. (nodes_active, nodes_total, clients_count) = ffpb_fetch_stats(bot, 'http://map.paderborn.freifunk.net/nodes.json', 'ffpb_stats')
  394. highscore_changed = False
  395. if nodes_active > highscores['nodes']:
  396. highscores['nodes'] = nodes_active
  397. highscores['nodes_ts'] = time.time()
  398. highscore_changed = True
  399. if clients_count > highscores['clients']:
  400. highscores['clients'] = clients_count
  401. highscores['clients_ts'] = time.time()
  402. highscore_changed = True
  403. if highscore_changed:
  404. print('HIGHSCORE changed: {0} nodes ({1}), {2} clients ({3})'.format(highscores['nodes'], highscores['nodes_ts'], highscores['clients'], highscores['clients_ts']))
  405. if not (bot.config.ffpb.msg_target is None):
  406. action_msg = 'notiert sich den neuen Highscore: {0} Knoten ({1}), {2} Clients ({3})'.format(highscores['nodes'], pretty_date(int(highscores['nodes_ts'])), highscores['clients'], pretty_date(int(highscores['clients_ts'])))
  407. action_target = bot.config.ffpb.msg_target
  408. if (not bot.config.ffpb.msg_target_public is None):
  409. action_target = bot.config.ffpb.msg_target_public
  410. bot.msg(action_target, '\x01ACTION %s\x01' % action_msg)
  411. @willie.module.commands('status')
  412. def ffpb_status(bot, trigger):
  413. """Status des FFPB-Netzes: Anzahl (aktiver) Knoten + Clients"""
  414. stats = bot.memory['ffpb_stats'] if 'ffpb_stats' in bot.memory else None
  415. if stats is None:
  416. bot.say('Uff, kein Plan wo der Zettel ist. Fragst du später nochmal?')
  417. return
  418. bot.say('Es sind {0} Knoten und ca. {1} Clients online.'.format(stats["nodes_active"], stats["clients"]))
  419. def pretty_date(time=False):
  420. """
  421. Get a datetime object or a int() Epoch timestamp and return a
  422. pretty string like 'an hour ago', 'Yesterday', '3 months ago',
  423. 'just now', etc
  424. """
  425. from datetime import datetime
  426. now = datetime.now()
  427. compare = None
  428. if type(time) is int:
  429. compare = datetime.fromtimestamp(time)
  430. elif type(time) is float:
  431. compare = datetime.fromtimestamp(int(time))
  432. elif isinstance(time,datetime):
  433. compare = time
  434. elif not time:
  435. compare = now
  436. diff = now - compare
  437. second_diff = diff.seconds
  438. day_diff = diff.days
  439. if day_diff < 0:
  440. return ''
  441. if day_diff == 0:
  442. if second_diff < 10:
  443. return "gerade eben"
  444. if second_diff < 60:
  445. return "vor " + str(second_diff) + " Sekunden"
  446. if second_diff < 120:
  447. return "vor einer Minute"
  448. if second_diff < 3600:
  449. return "vor " + str(second_diff / 60) + " Minuten"
  450. if second_diff < 7200:
  451. return "vor einer Stunde"
  452. if second_diff < 86400:
  453. return "vor " + str(second_diff / 3600) + " Stunden"
  454. if day_diff == 1:
  455. return "gestern"
  456. if day_diff < 7:
  457. return "vor " + str(day_diff) + " Tagen"
  458. return "am " + compare.strftime('%d.%m.%Y um %H:%M Uhr')
  459. @willie.module.commands('highscore')
  460. def ffpb_highscore(bot, trigger):
  461. bot.say('Highscore: {0} Knoten ({1}), {2} Clients ({3})'.format(
  462. highscores['nodes'], pretty_date(int(highscores['nodes_ts'])),
  463. highscores['clients'], pretty_date(int(highscores['clients_ts']))))
  464. @willie.module.commands('rollout-status')
  465. def ffpb_rolloutstatus(bot, trigger):
  466. result = { }
  467. for branch in [ 'stable', 'testing' ]:
  468. result[branch] = None
  469. skipped = 0
  470. if (not (trigger.admin and trigger.is_privmsg)) and (not trigger.nick in bot.ops[trigger.sender]):
  471. bot.say('Geh zur dunklen Seite, die haben Kekse - ohne Keks kein Rollout-Status.')
  472. return
  473. expected_release = trigger.group(2)
  474. if expected_release is None or len(expected_release) == 0:
  475. bot.say('Von welcher Firmware denn?')
  476. return
  477. for nodeid in alfred_data:
  478. item = alfred_data[nodeid]
  479. if (not 'software' in item) or (not 'firmware' in item['software']) or (not 'autoupdater' in item['software']):
  480. skipped+=1
  481. continue
  482. release = item['software']['firmware']['release']
  483. branch = item['software']['autoupdater']['branch']
  484. enabled = item['software']['autoupdater']['enabled']
  485. if not branch in result or result[branch] is None:
  486. result[branch] = { 'auto_count': 0, 'auto_not': 0, 'manual_count': 0, 'manual_not': 0, 'total': 0 }
  487. result[branch]['total'] += 1
  488. match = 'count' if release == expected_release else 'not'
  489. mode = 'auto' if enabled else 'manual'
  490. result[branch][mode+'_'+match] += 1
  491. output = "Rollout von '{0}':".format(expected_release)
  492. for branch in result:
  493. auto_count = result[branch]['auto_count']
  494. auto_total = auto_count + result[branch]['auto_not']
  495. manual_count = result[branch]['manual_count']
  496. manual_total = manual_count + result[branch]['manual_not']
  497. bot.say("Rollout von '{0}': {1} = {2}/{3} per Auto-Update, {4}/{5} manuell".format(expected_release, branch, auto_count, auto_total, manual_count, manual_total))
  498. if skipped > 0:
  499. bot.say("Rollout von '{0}': {1} Knoten unklar".format(expected_release, skipped))
  500. @willie.module.commands('ping')
  501. def ffpb_ping(bot, trigger=None, target_name=None):
  502. """Ping FFPB-Knoten"""
  503. if target_name is None: target_name = trigger.group(2)
  504. node = ffpb_findnode_from_botparam(bot, target_name, ensure_recent_alfreddata=False)
  505. if node is None: return None
  506. target = [x for x in node["network"]["addresses"] if not x.lower().startswith("fe80:")][0]
  507. target_alias = node["hostname"]
  508. print("pinging '{0}' at {1} ...".format(target_name, target))
  509. result = os.system('ping6 -c 2 -W 1 ' + target + ' >/dev/null')
  510. if result == 0:
  511. print("ping to '{0}' succeeded".format(target_name))
  512. if not bot is None: bot.say('Knoten "' + target_alias + '" antwortet \o/')
  513. return True
  514. elif result == 1 or result == 256:
  515. print("ping to '{0}' failed".format(target_name))
  516. if not bot is None: bot.say('Keine Antwort von "' + target_alias + '" :-(')
  517. return False
  518. else:
  519. print("ping to '{0}' broken: result='{1}'".format(target_name, result))
  520. if not bot is None: bot.say('Uh oh, irgendwas ist kaputt. Chef, ping result = ' + str(result) + ' - darf ich das essen?')
  521. return None
  522. @willie.module.interval(3*60)
  523. def ffpb_monitor_ping(bot):
  524. notify_target = bot.config.core.owner
  525. if (not bot.config.ffpb.msg_target is None):
  526. notify_target = bot.config.ffpb.msg_target
  527. for node in monitored_nodes:
  528. added = monitored_nodes[node][0]
  529. last_status = monitored_nodes[node][1]
  530. last_check = monitored_nodes[node][2]
  531. current_status = ffpb_ping(bot=None, target_name=node)
  532. monitored_nodes[node] = ( added, current_status, time.time() )
  533. print("Monitoring ({0}) {1} (last: {2} at {3})".format(node, current_status, last_status, time.strftime('%Y-%m-%d %H:%M', time.localtime(last_check))))
  534. if last_status != current_status and (last_status or current_status):
  535. if last_check is None:
  536. # erster Check, keine Ausgabe
  537. continue
  538. if current_status == True:
  539. bot.msg(notify_target, 'Monitoring: Knoten \'{0}\' pingt wieder (zuletzt {1})'.format(node, pretty_date(last_check)))
  540. else:
  541. bot.msg(notify_target, 'Monitoring: Knoten \'{0}\' DOWN'.format(node))
  542. @willie.module.commands('monitor')
  543. def ffpb_monitor(bot, trigger):
  544. if not trigger.admin:
  545. bot.say('Ich ping hier nicht für jeden durch die Weltgeschichte.')
  546. return
  547. if trigger.group(2) is None or len(trigger.group(2)) == 0:
  548. bot.say('Das Monitoring sagt du hast doofe Ohren.')
  549. return
  550. cmd = trigger.group(3)
  551. node = trigger.group(4)
  552. if not node is None: node = str(node)
  553. if cmd == "add":
  554. if node in monitored_nodes:
  555. bot.say('Knoten \'{0}\' wird bereits gemonitored.'.format(node))
  556. return
  557. monitored_nodes[node] = ( trigger.sender, None, None )
  558. bot.say('Knoten \'{0}\' wird jetzt ganz genau beobachtet.'.format(node))
  559. return
  560. if cmd == "del":
  561. if not node in monitored_nodes:
  562. bot.say('Knoten \'{0}\' war gar nicht im Monitoring?!?'.format(node))
  563. return
  564. del monitored_nodes[node]
  565. bot.say('Okidoki, \'{0}\' war mir sowieso egal.'.format(node))
  566. return
  567. if cmd == "info":
  568. if node in monitored_nodes:
  569. info = monitored_nodes[node]
  570. bot.say('Knoten \'{0}\' wurde zuletzt {1} gepingt (Ergebnis: {2}) - der Auftrag kam von {3}'.format(node, pretty_date(info[2]) if not info[2] is None else "^W noch nie", info[1], info[0]))
  571. else:
  572. bot.say('Knoten \'{0}\' ist nicht im Monitoring.'.format(node))
  573. return
  574. if cmd == "list":
  575. nodes = ""
  576. for node in monitored_nodes:
  577. nodes = nodes + " " + node
  578. bot.say('Monitoring aktiv für:' + nodes)
  579. return
  580. if cmd == "help":
  581. bot.say('Entweder "!monitor list" oder "!monitor {add|del|info} <node>"')
  582. return
  583. bot.say('Mit "' + str(cmd) + '" kann ich nix anfangen, probier doch mal "!monitor help".')
  584. @willie.module.commands('exec-on-peer')
  585. def ffpb_remoteexec(bot, trigger):
  586. """Remote Execution fuer FFPB_Knoten"""
  587. bot_params = trigger.group(2).split(' ',1)
  588. if len(bot_params) != 2:
  589. bot.say('Wenn du nicht sagst wo mach ich remote execution bei dir!')
  590. bot.say('Tipp: !exec-on-peer <peer> <cmd>')
  591. return
  592. target_name = bot_params[0]
  593. target_cmd = bot_params[1]
  594. if not trigger.admin:
  595. bot.say('I can haz sudo?')
  596. return
  597. if trigger.is_privmsg:
  598. bot.say('Bitte per Channel.')
  599. return
  600. if not trigger.nick in bot.ops[trigger.sender]:
  601. bot.say('Geh weg.')
  602. return
  603. node = ffpb_findnode_from_botparam(bot, target_name, ensure_recent_alfreddata=False)
  604. if node is None: return
  605. target = [x for x in node["network"]["addresses"] if not x.lower().startswith("fe80:")][0]
  606. target_alias = node["hostname"]
  607. cmd = 'ssh -6 -l root ' + target + ' -- "' + target_cmd + '"'
  608. print("REMOTE EXEC = " + cmd)
  609. try:
  610. result = subprocess.check_output(['ssh', '-6n', '-l', 'root', '-o', 'BatchMode=yes', '-o','StrictHostKeyChecking=no', target, target_cmd], stderr=subprocess.STDOUT, shell=False)
  611. lines = str(result).splitlines()
  612. if len(lines) == 0:
  613. bot.say('exec-on-peer(' + target_alias + '): No output')
  614. return
  615. msg = 'exec-on-peer(' + target_alias + '): ' + str(len(lines)) + ' Zeilen'
  616. if len(lines) > 8:
  617. msg += ' (zeige max. 8)'
  618. bot.say(msg + ':')
  619. for line in lines[0:8]:
  620. bot.say(line)
  621. except subprocess.CalledProcessError as e:
  622. bot.say('Fehler '+str(e.returncode)+' bei exec-on-peer('+target_alias+'): ' + e.output)