mirror of https://github.com/hak5/ToorChat.git
Added the channel and frequency options
parent
12959f2641
commit
916d34d938
|
@ -14,6 +14,21 @@ class ToorChatProtocol():
|
|||
self.device.RFxmit(msg.to_string())
|
||||
return msg
|
||||
|
||||
def change_channel(self, channel):
|
||||
''' This is used to change the channel that the user operates in '''
|
||||
self.device.setChannel(int(channel))
|
||||
if channel != int
|
||||
return None
|
||||
else
|
||||
return channel
|
||||
|
||||
def change_frequency(self, frequency):
|
||||
''' This is used to change the frequency that the user operates in '''
|
||||
self.device.setFreq(int(change_frequency))
|
||||
if frequency != int
|
||||
return None
|
||||
else
|
||||
return frequency
|
||||
@classmethod
|
||||
def parse_message(cls, raw_message):
|
||||
message = ToorChatMessage()
|
||||
|
|
16
toorchat.py
16
toorchat.py
|
@ -34,6 +34,8 @@ class Visualizer():
|
|||
self.protocol = ToorChatProtocol(self.badge)
|
||||
self.message_queue = []
|
||||
self.user = None
|
||||
self.channel = None
|
||||
self.frequency = None
|
||||
#This when set to True will kill the thread
|
||||
self.exit = False
|
||||
|
||||
|
@ -54,7 +56,7 @@ class Visualizer():
|
|||
self.last_message_index = 0
|
||||
while True:
|
||||
self.screen_max_y, self.screen_max_x = self.screen.getmaxyx()
|
||||
self.screen.addstr(0, 1, "[S] Send Message [U] Set User Name")
|
||||
self.screen.addstr(0, 1, "[S] Send Message [U] Set User Name [C] Set Channel [F] Set Frequency")
|
||||
self.__add_message_to_screen__()
|
||||
entry = self.screen.getch()
|
||||
if entry == curses.KEY_RESIZE:
|
||||
|
@ -72,6 +74,18 @@ class Visualizer():
|
|||
self.user = user_input[:USER_NAME_SIZE]
|
||||
self.screen.nodelay(1)
|
||||
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
||||
if entry == ord('c'):
|
||||
self.screen.nodelay(0)
|
||||
user_input = self.screen.getstr(1, 1, 60)
|
||||
self.channel = self.protocol.change_channel(user_input)
|
||||
self.screen.nodelay(1)
|
||||
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
||||
if entry == ord('f'):
|
||||
self.screen.nodelay(0)
|
||||
user_input = self.screen.getstr(1, 1, 60)
|
||||
self.frequency = self.protocol.change_frequency(user_input)
|
||||
self.screen.nodelay(1)
|
||||
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.exit = True
|
||||
|
|
Loading…
Reference in New Issue