mirror of https://github.com/hak5/ToorChat.git
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
|
|
USER_NAME_SIZE = 9
|
|
|
|
class ToorChatProtocol():
|
|
''' This is a class to allow for easy of use with anything to do with messaging '''
|
|
def __init__(self, device):
|
|
self.PACKET_START = "\xFF\xDE\xAD\xFF"
|
|
self.PACKET_END = "\xFF\xBE\xEF\xFF"
|
|
self.device = device
|
|
|
|
def send_message(self, message = "", user = None):
|
|
''' This is used to send a simple message over the toorchat protocol '''
|
|
msg = ToorChatMessage(message, user)
|
|
self.device.RFxmit(msg.to_string())
|
|
return msg
|
|
|
|
@classmethod
|
|
def parse_message(cls, raw_message):
|
|
message = ToorChatMessage()
|
|
message.raw = raw_message
|
|
start_index = raw_message.find(ToorChatProtocol.get_packet_start())
|
|
end_index = raw_message.find(ToorChatProtocol.get_packet_end())
|
|
if start_index == -1 or end_index == -1:
|
|
return None
|
|
message.start = raw_message[start_index:start_index + 4]
|
|
message.xid = raw_message[start_index + 4: start_index + 12]
|
|
message.user = raw_message[start_index + 12: start_index + 21]
|
|
message.data = raw_message[start_index + 21: end_index]
|
|
message.end = raw_message[end_index: end_index+4]
|
|
return message
|
|
|
|
@classmethod
|
|
def get_packet_start(cls):
|
|
return "\xFF\xDE\xAD\xFF"
|
|
|
|
@classmethod
|
|
def get_packet_end(cls):
|
|
return "\xFF\xBE\xEF\xFF"
|
|
|
|
class ToorChatMessage():
|
|
''' This is a simple Message object wrapper to make things cleaner '''
|
|
|
|
def __init__(self, message = "", user = None):
|
|
self.raw = None
|
|
self.start = ToorChatProtocol.get_packet_start()
|
|
self.xid = self.get_random_xid()
|
|
if user != None:
|
|
self.user = user
|
|
if len(self.user) < USER_NAME_SIZE:
|
|
self.user = self.user + " "* (USER_NAME_SIZE-len(self.user))
|
|
else:
|
|
self.user = "anonymous"
|
|
self.data = message
|
|
self.end = ToorChatProtocol.get_packet_end()
|
|
|
|
def get_random_xid(self):
|
|
return "44444444"
|
|
|
|
def __str__(self):
|
|
return self.start + self.xid + self.user + self.data + self.end
|
|
|
|
def to_string(self):
|
|
return self.__str__() |