mirror of https://github.com/hak5/ToorChat.git
121 lines
3.5 KiB
Python
Executable File
121 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
from rflib import *
|
|
import rflib.chipcon_nic as rfnic
|
|
import atexit
|
|
from libtoorchat import *
|
|
from os import system
|
|
import curses
|
|
import time
|
|
from threading import Thread
|
|
|
|
def thread_run(visual):
|
|
''' This is our function for the thread '''
|
|
#Thread should run until exit
|
|
while not visual.exit:
|
|
try:
|
|
msg, timestamp = visual.badge.RFrecv()
|
|
toor_message = ToorChatProtocol.parse_message(msg)
|
|
if toor_message != None:
|
|
if toor_message.type == ToorChatProtocol.get_chat_type():
|
|
visual.message_queue.append(toor_message)
|
|
if toor_message.type == ToorChatProtocol.get_web_request_type():
|
|
#If we are registered as a server, lets typ to make that request
|
|
if visual.server:
|
|
ToorChatProtocol.get_web_messages(toor_message.data, visual)
|
|
|
|
except ChipconUsbTimeoutException:
|
|
pass
|
|
|
|
class Visualizer():
|
|
|
|
def __init__(self):
|
|
self.screen = curses.initscr()
|
|
self.screen.nodelay(1)
|
|
self.badge = RfCat(idx=0)
|
|
self.badge.setModeRX()
|
|
self.protocol = ToorChatProtocol(self.badge)
|
|
self.message_queue = []
|
|
self.user = None
|
|
#This when set to True will kill the thread
|
|
self.exit = False
|
|
self.server = False
|
|
|
|
def set_server(self):
|
|
self.server = True
|
|
|
|
def start(self):
|
|
self.__start_recv_thread__()
|
|
self.__run__()
|
|
|
|
def __start_recv_thread__(self):
|
|
'''This spins off a thread to deal with recving information from the rf device '''
|
|
self.thread = Thread(target=thread_run, args=(self,))
|
|
self.thread.start()
|
|
|
|
def __run__(self):
|
|
try:
|
|
self.screen_max_y, self.screen_max_x = self.screen.getmaxyx()
|
|
self.__draw_frame__()
|
|
self.screen.refresh()
|
|
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 [W] Load Website")
|
|
self.__add_message_to_screen__()
|
|
entry = self.screen.getch()
|
|
if entry == curses.KEY_RESIZE:
|
|
self.__draw_frame__(False)
|
|
if entry == ord('s'):
|
|
self.screen.nodelay(0)
|
|
user_input = self.screen.getstr(1, 1, 60)
|
|
old_message = self.protocol.send_chat_message(user_input, self.user)
|
|
self.message_queue.append(old_message)
|
|
self.screen.nodelay(1)
|
|
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
|
if entry == ord('u'):
|
|
self.screen.nodelay(0)
|
|
user_input = self.screen.getstr(1, 1, 60)
|
|
self.user = user_input[:USER_NAME_SIZE]
|
|
self.screen.nodelay(1)
|
|
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
|
if entry == ord('w'):
|
|
self.screen.nodelay(0)
|
|
user_input = self.screen.getstr(1, 1, 60)
|
|
self.protocol.send_web_request(user_input)
|
|
self.screen.nodelay(1)
|
|
self.screen.addstr(1,1," "*(self.screen_max_x-3))
|
|
|
|
except KeyboardInterrupt:
|
|
self.exit = True
|
|
self.stop()
|
|
|
|
def __add_message_to_screen__(self):
|
|
if len(self.message_queue) > 0:
|
|
message = self.message_queue[len(self.message_queue)-1]
|
|
message_string = str(message.user)+":"+ str(message.data)
|
|
self.screen.addstr(self.last_message_index+3,1, message_string + " "*(self.screen_max_x-(2+len(message_string))))
|
|
self.last_message_index +=1
|
|
self.message_queue.pop()
|
|
if self.last_message_index > self.screen_max_y-5:
|
|
self.last_message_index = 0
|
|
self.__draw_frame__()
|
|
|
|
def __draw_frame__(self, clear = True):
|
|
if clear:
|
|
self.screen.clear()
|
|
self.screen.border(0)
|
|
self.screen.addstr(2,1,"_"*(self.screen_max_x-2))
|
|
|
|
def stop(self):
|
|
curses.endwin()
|
|
|
|
if __name__ == '__main__':
|
|
visual = Visualizer()
|
|
if len(sys.argv) > 1:
|
|
if sys.argv[1] == "-s":
|
|
visual.set_server()
|
|
visual.start()
|
|
|
|
|