Entire code is in a class now for modular usage
This commit is contained in:
parent
d33930ee37
commit
053c5e9d32
GPG Key ID:
EF4B922DB85DC9DE
194
meshmqtt.py
194
meshmqtt.py
@ -6,8 +6,8 @@ import base64
|
||||
import logging
|
||||
|
||||
try:
|
||||
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
|
||||
from cryptography.hazmat.backends import default_backend
|
||||
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
|
||||
except ImportError:
|
||||
raise SystemExit('missing the cryptography module (pip install cryptography)')
|
||||
|
||||
@ -26,16 +26,55 @@ except ImportError:
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %I:%M:%S')
|
||||
|
||||
|
||||
def decode_encrypted(message_packet):
|
||||
class MeshtasticMQTT(object):
|
||||
def __init__(self):
|
||||
'''Initialize the Meshtastic MQTT client'''
|
||||
|
||||
self.broadcast_id = 4294967295
|
||||
self.key = None
|
||||
|
||||
|
||||
def connect(self, broker: str, port: int, root: str, tls: bool, username: str, password: str, key: str):
|
||||
'''
|
||||
Connect to the MQTT broker
|
||||
|
||||
:param broker: The MQTT broker address
|
||||
:param port: The MQTT broker port
|
||||
:param root: The root topic to subscribe to
|
||||
:param tls: Enable TLS/SSL
|
||||
:param username: The MQTT username
|
||||
:param password: The MQTT password
|
||||
:param key: The encryption key
|
||||
'''
|
||||
|
||||
client = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2, client_id='', clean_session=True, userdata=None)
|
||||
client.username_pw_set(username=username, password=password)
|
||||
|
||||
self.key = '1PG7OiApB1nwvP+rz05pAQ==' if key == 'AQ==' else key
|
||||
|
||||
if tls:
|
||||
client.tls_set()
|
||||
#client.tls_insecure_set(False)
|
||||
|
||||
client.on_connect = self.on_connect
|
||||
client.on_message = self.on_message
|
||||
client.on_subscribe = self.on_subscribe
|
||||
client.on_unsubscribe = self.on_unsubscribe
|
||||
|
||||
client.connect(broker, port, 60)
|
||||
client.subscribe(root, 0)
|
||||
client.loop_forever()
|
||||
|
||||
|
||||
def decrypt_message_packet(self, message_packet):
|
||||
'''
|
||||
Decrypt an encrypted message packet.
|
||||
|
||||
:param message_packet: The message packet to decrypt
|
||||
'''
|
||||
|
||||
try:
|
||||
# Ensure the key is formatted and padded correctly before turning it into bytes
|
||||
padded_key = args.key.ljust(len(args.key) + ((4 - (len(args.key) % 4)) % 4), '=')
|
||||
padded_key = self.key.ljust(len(self.key) + ((4 - (len(self.key) % 4)) % 4), '=')
|
||||
key = padded_key.replace('-', '+').replace('_', '/')
|
||||
key_bytes = base64.b64decode(key.encode('ascii'))
|
||||
|
||||
@ -54,6 +93,56 @@ def decode_encrypted(message_packet):
|
||||
data.ParseFromString(decrypted_bytes)
|
||||
message_packet.decoded.CopyFrom(data)
|
||||
|
||||
return message_packet
|
||||
|
||||
|
||||
def on_connect(self, client, userdata, flags, rc, properties):
|
||||
'''
|
||||
Callback for when the client receives a CONNACK response from the server.
|
||||
|
||||
:param client: The client instance for this callback
|
||||
:param userdata: The private user data as set in Client() or user_data_set()
|
||||
:param flags: Response flags sent by the broker
|
||||
:param rc: The connection result
|
||||
:param properties: The properties returned by the broker
|
||||
'''
|
||||
|
||||
if rc == 0:
|
||||
logging.info('Connected to MQTT broker')
|
||||
else:
|
||||
logging.error(f'Failed to connect to MQTT broker: {rc}')
|
||||
|
||||
|
||||
def on_message(self, client, userdata, msg):
|
||||
'''
|
||||
Callback for when a message is received from the server.
|
||||
|
||||
:param client: The client instance for this callback
|
||||
:param userdata: The private user data as set in Client() or user_data_set()
|
||||
:param msg: An instance of MQTTMessage. This is a
|
||||
'''
|
||||
|
||||
# Define the service envelope
|
||||
service_envelope = mqtt_pb2.ServiceEnvelope()
|
||||
|
||||
try:
|
||||
# Parse the message payload
|
||||
service_envelope.ParseFromString(msg.payload)
|
||||
|
||||
# logging.info('Received a packet:')
|
||||
# logging.info(service_envelope)
|
||||
|
||||
# Extract the message packet from the service envelope
|
||||
message_packet = service_envelope.packet
|
||||
|
||||
except Exception as e:
|
||||
#logging.error(f'Failed to parse message: {str(e)}')
|
||||
return
|
||||
|
||||
# Check if the message is encrypted before decrypting it
|
||||
if message_packet.HasField('encrypted') and not message_packet.HasField('decoded'):
|
||||
message_packet = self.decrypt_message_packet(message_packet)
|
||||
|
||||
if message_packet.decoded.portnum == portnums_pb2.TEXT_MESSAGE_APP:
|
||||
text_payload = message_packet.decoded.payload.decode('utf-8')
|
||||
text = {
|
||||
@ -65,7 +154,9 @@ def decode_encrypted(message_packet):
|
||||
logging.info('Received text message:')
|
||||
logging.info(text)
|
||||
|
||||
elif message_packet.decoded.portnum == portnums_pb2.MAP_REPORT_APP:
|
||||
h = [portnums_pb2.REMOTE_HARDWARE_APP, portnums_pb2.ROUTING_APP,
|
||||
|
||||
elif message_packet.decoded.portnum == portnums_pb2.MAP_REPORT_APP: # comes unencrypted
|
||||
pos = mesh_pb2.Position()
|
||||
pos.ParseFromString(message_packet.decoded.payload)
|
||||
logging.info('Received map report:')
|
||||
@ -89,56 +180,15 @@ def decode_encrypted(message_packet):
|
||||
logging.info('Received telemetry:')
|
||||
logging.info(env)
|
||||
|
||||
except Exception as e:
|
||||
logging.error(f'Failed to decrypt message: {str(e)}')
|
||||
elif message_packet.decoded.portnum == portnums_pb2.TRACEROUTE_APP:
|
||||
routeDiscovery = mesh_pb2.RouteDiscovery()
|
||||
routeDiscovery.ParseFromString(message_packet.decoded.payload)
|
||||
logging.info('Received traceroute:')
|
||||
logging.info(routeDiscovery)
|
||||
|
||||
|
||||
def on_connect(client, userdata, flags, rc, properties):
|
||||
'''
|
||||
Callback for when the client receives a CONNACK response from the server.
|
||||
|
||||
:param client: The client instance for this callback
|
||||
:param userdata: The private user data as set in Client() or user_data_set()
|
||||
:param flags: Response flags sent by the broker
|
||||
:param rc: The connection result
|
||||
:param properties: The properties returned by the broker
|
||||
'''
|
||||
|
||||
if rc == 0:
|
||||
logging.info('Connected to MQTT broker')
|
||||
else:
|
||||
logging.error(f'Failed to connect to MQTT broker: {rc}')
|
||||
|
||||
|
||||
def on_message(client, userdata, msg):
|
||||
'''
|
||||
Callback for when a message is received from the server.
|
||||
|
||||
:param client: The client instance for this callback
|
||||
:param userdata: The private user data as set in Client() or user_data_set()
|
||||
:param msg: An instance of MQTTMessage. This is a
|
||||
'''
|
||||
|
||||
# Define the service envelope
|
||||
service_envelope = mqtt_pb2.ServiceEnvelope()
|
||||
|
||||
try:
|
||||
# Parse the message payload
|
||||
service_envelope.ParseFromString(msg.payload)
|
||||
|
||||
logging.info('Received a packet:')
|
||||
logging.info(service_envelope)
|
||||
|
||||
# Extract the message packet from the service envelope
|
||||
message_packet = service_envelope.packet
|
||||
|
||||
except Exception as e:
|
||||
#logging.error(f'Failed to parse message: {str(e)}')
|
||||
return
|
||||
|
||||
# Check if the message is encrypted before decrypting it
|
||||
if message_packet.HasField('encrypted') and not message_packet.HasField('decoded'):
|
||||
decode_encrypted(message_packet)
|
||||
logging.warning('Received an unknown message:')
|
||||
logging.info(message_packet)
|
||||
|
||||
# If the message is not encrypted, log the payload (this should not happen)
|
||||
else:
|
||||
@ -146,7 +196,7 @@ def on_message(client, userdata, msg):
|
||||
logging.info(f'Payload: {message_packet}')
|
||||
|
||||
|
||||
def on_subscribe(client, userdata, mid, reason_code_list, properties):
|
||||
def on_subscribe(self, client, userdata, mid, reason_code_list, properties):
|
||||
'''
|
||||
Callback for when the client receives a SUBACK response from the server.
|
||||
|
||||
@ -164,7 +214,7 @@ def on_subscribe(client, userdata, mid, reason_code_list, properties):
|
||||
logging.info(f'Broker granted the following QoS: {reason_code_list[0].value}')
|
||||
|
||||
|
||||
def on_unsubscribe(client, userdata, mid, reason_code_list, properties):
|
||||
def on_unsubscribe(self, client, userdata, mid, reason_code_list, properties):
|
||||
'''
|
||||
Callback for when the client receives a UNSUBACK response from the server.
|
||||
|
||||
@ -187,7 +237,7 @@ def on_unsubscribe(client, userdata, mid, reason_code_list, properties):
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
parser = argparse.ArgumentParser(description='Mesh MQTT')
|
||||
parser = argparse.ArgumentParser(description='Meshtastic MQTT Interface')
|
||||
parser.add_argument('--broker', default='mqtt.meshtastic.org', help='MQTT broker address')
|
||||
parser.add_argument('--port', default=1883, type=int, help='MQTT broker port')
|
||||
parser.add_argument('--root', default='#', help='Root topic')
|
||||
@ -197,33 +247,5 @@ if __name__ == '__main__':
|
||||
parser.add_argument('--key', default='AQ==', help='Encryption key')
|
||||
args = parser.parse_args()
|
||||
|
||||
# Set the broadcast ID (Do we need to change this for a custom channel?)
|
||||
broadcast_id = 4294967295
|
||||
|
||||
# Create the MQTT client
|
||||
client = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2, client_id='', clean_session=True, userdata=None) # Defaults to mqtt.MQTTv311 (change with protocol=mqtt.MQTTv5)
|
||||
|
||||
# Set the authentication details
|
||||
client.username_pw_set(username=args.username, password=args.password)
|
||||
|
||||
# Enable TLS/SSL if the --tls flag is set
|
||||
if args.tls:
|
||||
import ssl
|
||||
client.tls_set()
|
||||
#client.tls_set(cert_reqs=ssl.CERT_REQUIRED, tls_version=ssl.PROTOCOL_TLSv1_2)
|
||||
#client.tls_insecure_set(False)
|
||||
|
||||
# Set the callbacks
|
||||
client.on_connect = on_connect
|
||||
client.on_message = on_message
|
||||
client.on_subscribe = on_subscribe
|
||||
client.on_unsubscribe = on_unsubscribe
|
||||
|
||||
# Connect to the broker
|
||||
client.connect(args.broker, args.port, 60)
|
||||
|
||||
# Subscribe to the root topic
|
||||
client.subscribe(args.root, 0)
|
||||
|
||||
# Start the keep-alive loop
|
||||
client.loop_forever()
|
||||
client = MeshtasticMQTT()
|
||||
client.connect(args.broker, args.port, args.root, args.tls, args.username, args.password, args.key)
|
||||
Loading…
Reference in New Issue
Block a user