Handles LOADS of protobufs now. Major credit to AT and wintersoldier for this update

This commit is contained in:
Josh Conway 2024-09-11 06:11:51 +00:00
parent f1b0b4fc9a
commit 597e63543c

View File

@ -11,11 +11,14 @@ import socket
import zmq import zmq
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
from meshtastic import protocols, mesh_pb2 from meshtastic import protocols, mesh_pb2, admin_pb2, portnums_pb2, telemetry_pb2, mqtt_pb2
from datetime import datetime
# SDR output example data: ffffffffb45463dab971aa8c6308000078aacf76587a5a4cf4a20e2c1d0349ab3f72 # SDR output example data: ffffffffb45463dab971aa8c6308000078aacf76587a5a4cf4a20e2c1d0349ab3f72
# Use default key. Result should be: b'\x08\x01\x12\x0eTestingCLU1234' # Use default key. Result should be: b'\x08\x01\x12\x0eTestingCLU1234'
debug=False
##### START FUNCTIONS BLOCK ##### ##### START FUNCTIONS BLOCK #####
# Takes in a string encoded as hex, and emits them as a bytes encoded of the same hex representation # Takes in a string encoded as hex, and emits them as a bytes encoded of the same hex representation
@ -39,6 +42,7 @@ parser.add_argument('-i', '--input', action='store', dest='input', help='SDR cap
parser.add_argument('-k', '--key', action='store',dest='key', help='AES key override in Base64') parser.add_argument('-k', '--key', action='store',dest='key', help='AES key override in Base64')
parser.add_argument('-n', '--net', action='store',dest='net', help='Network TCP in ip or DNS. ZeroMQ protocol.') parser.add_argument('-n', '--net', action='store',dest='net', help='Network TCP in ip or DNS. ZeroMQ protocol.')
parser.add_argument('-p', '--port', action='store',dest='port', help='Network port') parser.add_argument('-p', '--port', action='store',dest='port', help='Network port')
parser.add_argument('-d', '--debug', action='store_true',dest='debug', help='Print more debug messages')
args = parser.parse_args() args = parser.parse_args()
##### END PARSE COMMANDLINE INPUT ##### ##### END PARSE COMMANDLINE INPUT #####
@ -70,6 +74,7 @@ def parseAESKey(aesKey):
if (aesKeyLength == 32 or aesKeyLength == 64): if (aesKeyLength == 32 or aesKeyLength == 64):
pass pass
else: else:
if debug:
print("The included AES key appears to be invalid. The key length is" , aesKeyLength , "and is not the key length of 128 or 256 bits.") print("The included AES key appears to be invalid. The key length is" , aesKeyLength , "and is not the key length of 128 or 256 bits.")
sys.exit() sys.exit()
@ -105,6 +110,12 @@ def dataExtractor(data):
'reserved' : hexStringToBinary(data[28:32]), 'reserved' : hexStringToBinary(data[28:32]),
'data' : hexStringToBinary(data[32:len(data)]) 'data' : hexStringToBinary(data[32:len(data)])
} }
if debug:
print("dest " + str(meshPacketHex['dest'].hex()) + " sender " + str(meshPacketHex['sender'].hex()))
print("id " + str(int(meshPacketHex['packetID'].hex(),16)))
print("flags " + str(meshPacketHex['flags'].hex()))
print("chash " + str(meshPacketHex['channelHash'].hex()))
print("data " + str(meshPacketHex['data'].hex()))
return meshPacketHex return meshPacketHex
##### END DATA EXTRACTION BLOCK ##### ##### END DATA EXTRACTION BLOCK #####
@ -121,7 +132,8 @@ def dataDecryptor(meshPacketHex, aesKey):
aesNonce = meshPacketHex['packetID'] + b'\x00\x00\x00\x00' + meshPacketHex['sender'] + b'\x00\x00\x00\x00' aesNonce = meshPacketHex['packetID'] + b'\x00\x00\x00\x00' + meshPacketHex['sender'] + b'\x00\x00\x00\x00'
# print("Nonce binary is:", aesNonce) if debug:
print("Nonce binary is:", aesNonce.hex())
# print("Nonce length is:", len(aesNonce) ) # print("Nonce length is:", len(aesNonce) )
@ -131,30 +143,140 @@ def dataDecryptor(meshPacketHex, aesKey):
# Do the decryption. Note, that this cipher is reversible, so running the cipher on encrypted gives decrypted, and running the cipher on decrypted gives encrypted. # Do the decryption. Note, that this cipher is reversible, so running the cipher on encrypted gives decrypted, and running the cipher on decrypted gives encrypted.
decryptedOutput = decryptor.update(meshPacketHex['data']) + decryptor.finalize() decryptedOutput = decryptor.update(meshPacketHex['data']) + decryptor.finalize()
if debug:
print("dec: "+ decryptedOutput.hex())
return decryptedOutput return decryptedOutput
##### END DECRYPTION PROCESS ##### ###### END DECRYPTION PROCESS #####
#
#
#
###### START PROTOBUF DECODER #####
#
#def decodeProtobuf(packetData):
# # print("Packet data:", packetData)
# data = mesh_pb2.Data()
# try:
# data.ParseFromString(packetData)
#
# handler = protocols.get(data.portnum)
# if handler.protobufFactory is None:
# pass
# else:
# pb = handler.protobufFactory()
# pb.ParseFromString(data.payload)
# except:
# data = "INVALID PROTOBUF:"
# return data
#
###### END PROTOBUF DECODER #####
##### START PROTOBUF DECODER ##### ##### START PROTOBUF DECODER #####
def decodeProtobuf(packetData): def decodeProtobuf(packetData, sourceID, destID):
# print("Packet data:", packetData)
data = mesh_pb2.Data() data = mesh_pb2.Data()
try: try:
data.ParseFromString(packetData) data.ParseFromString(packetData)
handler = protocols.get(data.portnum)
if handler.protobufFactory is None:
pass
else:
pb = handler.protobufFactory()
pb.ParseFromString(data.payload)
except: except:
data = "INVALID PROTOBUF:" data = "INVALID PROTOBUF: " + str(packetData)
return data return data
match data.portnum :
case 0 : # UNKNOWN_APP
data = "UNKNOWN_APP To be implemented"
case 1 : # TEXT_MESSAGE_APP
text_payload = data.payload.decode('utf-8')
data = "TEXT_MESSAGE_APP " + str(sourceID) + " -> " + str(destID) + " " + str(text_payload)
case 2 : # REMOTE_HARDWARE_APP
data = "REMOTE_HARDWARE_APP To be implemented"
case 3 : # POSITION_APP
pos = mesh_pb2.Position()
pos.ParseFromString(data.payload)
latitude = pos.latitude_i * 1e-7
longitude = pos.longitude_i * 1e-7
data="POSITION_APP " + str(sourceID) + " -> " + str(destID) + " " + str(latitude) +"," + str(longitude)
case 4 : # NODEINFO_APP
info = mesh_pb2.User()
info.ParseFromString(data.payload)
data = "NODEINFO_APP " + str(info)
case 5 : # ROUTING_APP
rtng = mesh_pb2.Routing()
rtng.ParseFromString(data.payload)
data = "TELEMETRY_APP " + str(rtng)
case 6 : # ADMIN_APP
admn = admin_pb2.AdminMessage()
admn.ParseFromString(data.payload)
data = "ADMIN_APP " + str(admn)
case 7 : # TEXT_MESSAGE_COMPRESSED_APP
data = "TEXT_MESSAGE_COMPRESSED_APP To be implemented"
case 10 : # DETECTION_SENSOR_APP
data = "DETECTION_SENSOR_APP To be implemented"
case 32 : # REPLY_APP
data = "REPLY_APP To be implemented"
case 33 : # IP_TUNNEL_APP
data = "IP_TUNNEL_APP To be implemented"
case 34 : # PAXCOUNTER_APP
data = "PAXCOUNTER_APP To be implemented"
case 64 : # SERIAL_APP
print(" ")
case 65 : # STORE_FORWARD_APP
sfwd = mesh_pb2.StoreAndForward()
sfwd.ParseFromString(data.payload)
data = "STORE_FORWARD_APP " + str(sfwd)
case 67 : # TELEMETRY_APP
env = telemetry_pb2.Telemetry()
env.ParseFromString(data.payload)
data = "TELEMETRY_APP " + str(env)
case 68 : # ZPS_APP
z_info = mesh_pb2.zps()
z_info.ParseFromString(data.payload)
data = "ZPS_APP " + str(z_info)
case 69 : # SIMULATOR_APP
data = "SIMULATOR_APP To be implemented"
case 70 : # TRACEROUTE_APP
trct= mesh_pb2.RouteDiscovery()
trct.ParseFromString(data.payload)
data = "TRACEROUTE_APP " + str(sourceID) + " -> " + str(destID) + " " + str(trct)
case 71 : # NEIGHBORINFO_APP
ninfo = mesh_pb2.NeighborInfo()
ninfo.ParseFromString(data.payload)
data = "NEIGHBORINFO_APP " + str(ninfo)
case 72 : # ATAK_PLUGIN
data = "ATAK_PLUGIN To be implemented"
case 73 : # MAP_REPORT_APP
mrpt = mesh_pb2.MapReport()
mrpt.ParseFromString(data.payload)
data = "MAP_REPORT_APP " + str(mrpt)
case 74 : # POWERSTRESS_APP
data = "POWERSTRESS_APP To be implemented"
case 256 : # PRIVATE_APP
data = "PRIVATE_APP To be implemented"
case 257 : # ATAK_FORWARDER
data = "ATAK_FORWARD To be implemented"
case _:
data = "UNKNOWN PROTOBUF"
return data
# # print("Packet data:", packetData)
# data = mesh_pb2.Data()
# try:
# data.ParseFromString(packetData)
#
# handler = protocols.get(data.portnum)
# if handler.protobufFactory is None:
# pass
# else:
# pb = handler.protobufFactory()
# pb.ParseFromString(data.payload)
# except:
# data = "INVALID PROTOBUF:"
# return data
##### END PROTOBUF DECODER ##### ##### END PROTOBUF DECODER #####
@ -171,16 +293,12 @@ def networkParse(ipAddr, port, aesKey):
while True: while True:
if socket.poll(10) != 0: if socket.poll(10) != 0:
msg = socket.recv() msg = socket.recv()
# Extracts data from network socket
extractedData = dataExtractor(msg.hex()) extractedData = dataExtractor(msg.hex())
# Decrypts the payload PacketID = extractedData['packetID'].hex()
if debug:
print("Packet: " + msg.hex())
decryptedData = dataDecryptor(extractedData, aesKey) decryptedData = dataDecryptor(extractedData, aesKey)
# Decodes the Protobuf if possible protobufMessage = decodeProtobuf(decryptedData, extractedData['sender'].hex(), extractedData['dest'].hex())
protobufMessage = decodeProtobuf(decryptedData)
if(protobufMessage == "INVALID PROTOBUF:"):
print("INVALID PROTOBUF: ", end = '')
print(decryptedData)
else:
print(protobufMessage) print(protobufMessage)
else: else:
@ -194,19 +312,24 @@ if __name__ == "__main__":
# Network branch. Doesnt exit, so we need IP Port and AES key # Network branch. Doesnt exit, so we need IP Port and AES key
try: try:
if args.debug:
debug=True
if len(args.net) > 0 and len(args.port) > 0: if len(args.net) > 0 and len(args.port) > 0:
if debug:
print(args.net, args.port) print(args.net, args.port)
networkParse(args.net, args.port, meshtasticFullKeyHex) networkParse(args.net, args.port, meshtasticFullKeyHex)
except: except:
# If we get a payload on commandline, decrypt and exit. # If we get a payload on commandline, decrypt and exit.
meshPacketHex = dataExtractor(args.input) meshPacketHex = dataExtractor(args.input)
if debug:
print(meshPacketHex)
decryptedData = dataDecryptor(meshPacketHex, meshtasticFullKeyHex) decryptedData = dataDecryptor(meshPacketHex, meshtasticFullKeyHex)
protobufMessage = decodeProtobuf(decryptedData) protobufMessage = decodeProtobuf(decryptedData)
if(protobufMessage == "INVALID PROTOBUF:"): if(protobufMessage == "INVALID PROTOBUF:"):
if debug:
print("INVALID PROTOBUF: ", end = '') print("INVALID PROTOBUF: ", end = '')
if debug:
print(decryptedData) print(decryptedData)
else: else:
print(protobufMessage) print(protobufMessage)