hak5
/
ToorChat
mirror of https://github.com/hak5/ToorChat.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added users

master
haddaway 2012-10-20 21:44:30 -07:00
parent c3299508a6
commit 9a43cb7ca1
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libtoorchat.py
View File

@ -10,7 +10,6 @@ class ToorChatProtocol():
''' This is used to send a simple message over the toorchat protocol ''' ''' This is used to send a simple message over the toorchat protocol '''
msg = ToorChatMessage(message, user) msg = ToorChatMessage(message, user)
self.device.RFxmit(msg.to_string()) self.device.RFxmit(msg.to_string())
print msg.to_string()
@classmethod @classmethod
def parse_message(cls, raw_message): def parse_message(cls, raw_message):

19
toorchat.py
View File

@ -33,6 +33,7 @@ class Visualizer():
self.badge.setModeRX() self.badge.setModeRX()
self.protocol = ToorChatProtocol(self.badge) self.protocol = ToorChatProtocol(self.badge)
self.message_queue = [] self.message_queue = []
self.user = None
#This when set to True will kill the thread #This when set to True will kill the thread
self.exit = False self.exit = False
@ -53,10 +54,11 @@ class Visualizer():
last_message_index = 0 last_message_index = 0
while True: while True:
self.screen_max_y, self.screen_max_x = self.screen.getmaxyx() self.screen_max_y, self.screen_max_x = self.screen.getmaxyx()
self.screen.addstr(0, 1, "[S] Send Message ") self.screen.addstr(0, 1, "[S] Send Message [U] Set User Name")
if len(self.message_queue) > 0: if len(self.message_queue) > 0:
message = time.ctime()+":"+ str(self.message_queue[len(self.message_queue)-1].data) message = self.message_queue[len(self.message_queue)-1]
self.screen.addstr(last_message_index+3,1, message + " "*(self.screen_max_x-(2+len(message)))) message_string = str(message.user)+":"+ str(message.data)
self.screen.addstr(last_message_index+3,1, message_string + " "*(self.screen_max_x-(2+len(message_string))))
print self.screen_max_x print self.screen_max_x
last_message_index +=1 last_message_index +=1
self.message_queue.pop() self.message_queue.pop()
@ -68,10 +70,15 @@ class Visualizer():
if entry == ord('s'): if entry == ord('s'):
self.screen.nodelay(0) self.screen.nodelay(0)
user_input = self.screen.getstr(1, 1, 60) user_input = self.screen.getstr(1, 1, 60)
self.protocol.send_message(user_input) self.protocol.send_message(user_input, self.user)
self.screen.nodelay(1) self.screen.nodelay(1)
self.__draw_frame__() self.screen.addstr(1,1," "*(self.screen_max_x-3))
self.screen.refresh() if entry == ord('u'):
self.screen.nodelay(0)
user_input = self.screen.getstr(1, 1, 60)
self.user = user_input
self.screen.nodelay(1)
self.screen.addstr(1,1," "*(self.screen_max_x-3))
except KeyboardInterrupt: except KeyboardInterrupt:
self.exit = True self.exit = True