88 regels
3.1 KiB
Python
88 regels
3.1 KiB
Python
import bottombar as bb
|
|
import argparse
|
|
from console_bottombar import BottomBar
|
|
import getpass
|
|
import json
|
|
import mqtt
|
|
import os
|
|
import re
|
|
import time
|
|
|
|
VERSION = "0.1.0"
|
|
STARTTIME = time.time()
|
|
logfile = None
|
|
|
|
HELPTEXT = """
|
|
F1: Get this help message
|
|
F2: Set a filter (regular expression) for the topic of a message
|
|
Examples:
|
|
* "/gfw/.*/main_light/" to get everything with "/gfw/" before "/main_light/"
|
|
* "^zigbee.*(?>!logging)$" to get everything starting with "zigbee" and not ending with "logging"
|
|
* "^(?!shellies).*/dirk/.*temperature$" to get everything not starting with "shellies" followed by "/dirk/" and ends with "temperature"
|
|
F9: Start / Stop logging to mqtt-sniffer.log
|
|
F12: Quit the mqtt sniffer
|
|
|
|
'c': Clear screen
|
|
'q': Quit
|
|
"""
|
|
|
|
|
|
def rx_mqtt(mc, userdata, message):
|
|
global my_bb
|
|
global logfile
|
|
try:
|
|
match = len(re.findall(my_bb.get_entry('msg_re'), message.topic)) > 0
|
|
except re.error:
|
|
print('No valid regular expression (%s). No filter active.' % my_bb.get_entry('msg_re'))
|
|
match = True
|
|
|
|
ts = time.time()-STARTTIME
|
|
data = None
|
|
try:
|
|
data = json.loads(message.payload)
|
|
except:
|
|
pass
|
|
try:
|
|
data = message.payload.decode("utf-8")
|
|
except:
|
|
data = message.payload
|
|
|
|
if match:
|
|
print("%9.04f::%75s::%s" % (ts, message.topic, data))
|
|
if my_bb.get_entry('log2file'):
|
|
if logfile is None:
|
|
logfile = open('mqtt_sniffer.log', 'w')
|
|
logfile.write("%9.04f::%s::%s\n" % (ts, message.topic, data))
|
|
logfile.flush()
|
|
bb.redraw()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description='This is a mqtt sniffer.')
|
|
parser.add_argument('-f', dest='hostname', default='localhost', help='Hostname of the mqtt server')
|
|
parser.add_argument('-p', dest='port', default=1883, help='Port of the mqtt server')
|
|
parser.add_argument('-n', dest='no_credentials', action='store_true', help='Avoid asking for credentials')
|
|
parser.add_argument('-u', dest='username', default=None, help='Set username for mqtt server')
|
|
parser.add_argument('-t', dest='topicfilter', default="", help='Set topic filter')
|
|
parser.add_argument('-l', dest='logtofile', action='store_true', help='Enable logging to file')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.no_credentials:
|
|
if args.username == None:
|
|
args.username = input("Username: ")
|
|
password = getpass.getpass(prompt='Password: ', stream=None)
|
|
else:
|
|
args.username = None
|
|
password = None
|
|
|
|
my_bb = BottomBar(VERSION, label='MQTT-Sniffer')
|
|
my_bb.add_entry('help', 1, my_bb.FUNC_INFO, label='[F1] Help', infotext=HELPTEXT)
|
|
my_bb.add_entry('msg_re', 2, my_bb.FUNC_TEXT, label='[F2] Filter', default=args.topicfilter)
|
|
my_bb.add_entry('quit', 12, my_bb.FUNC_QUIT, "Quit", label='[F12]', right=True)
|
|
my_bb.add_entry('log2file', 9, my_bb.FUNC_BOOL, label='[F9] Log2File', default=args.logtofile, right=True)
|
|
|
|
mc = mqtt.mqtt_client("mqtt_sniffer", args.hostname, port=args.port, username=args.username, password=password)
|
|
mc.add_callback("#", rx_mqtt)
|
|
my_bb.run()
|