def command(self, command, data):
"""Sends the C side a command.
The flags can be found in SEASIDE.
Args:
command (int): The flag to send to the C side.
data (str): Any additional data to send with the flag.
TODO: Clean up the types / documentation.
TODO: Remove the eval.
"""
SEASIDE.send_SEASIDE(c_socket, c_socket_lock, int(command), eval(data))
def upload_pcap_file(self, file_data):
"""Takes a pcap file, and sends it to the C side for sending.
Args:
file_data (File): A CherryPy representation of a file.
TODO: Find a way to remove that NamedTemporaryFile middleman.
"""
pcap_file = tempfile.NamedTemporaryFile()
pcap_file.write(bytearray(file_data.file.read()))
pcap_file.seek(0)
packet = rdpcap(pcap_file.name)[0]
packet = conversions.convert_packet_int_array(packet)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock, 0, packet)
def load_pcap_file(self, filename):
"""Takes a filename, and attempts to load it from pcap_files/.
Reads the files, takes the first packet that is specified in that
file, and then sends it over to the C side for sending.
Args:
filename (str): Name of the file located in the pcap_files dir.
"""
if filename == '':
return
try:
packet = rdpcap('pcap_files/' + filename)[0]
except IOError:
print filename, "doesn't exist."
return
packet = conversions.convert_packet_int_array(packet)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock, 0, packet)
def packet_config(self, packet_layers):
"""Takes the configured layers and sends it to the C side.
Takes the packet_layers (in a JSON format, see configure_packet_layers
above), turns it into a legitimate Scapy packet, and then sends it to
the C side.
Args:
packet_layers (JSON): The packet layers in a string of JSON. See
configure_packet_layers above for an example of what this would
look like.
"""
# If the user didn't even configure a packet, but just clicked the
# "Done configuring" button like the silly user they are.
if packet_layers == '[]':
return
# Turn it into a Scapy packet.
packet = configure_packet_layers(packet_layers)
# Turn it into the corresponding int array.
packet = conversions.convert_packet_int_array(packet)
# Send it to the C side, using the SEASIDE format.
SEASIDE.send_SEASIDE(c_socket, c_socket_lock, 0, packet)
def user_interaction(lcd, lcd_lock, c_socket, c_socket_lock):
"""Uses the LCD screen to interact with the user.
Waits for the user to press a button and then performs the corresponding
action. While it is waiting, update_display_loop has control of the screen
and is displaying the statistics screen. When the user enters a
configuration option, that function takes control of the screen and
update_display_loop waits until it has released the lock to resume control.
The SEASIDE protocol is used for communication between the C and Python
programs. For more information about SEASIDE, see shared_files/SEASIDE.py.
Args:
lcd (LCD_Input_Wrapper object): the lcd screen to use.
lcd_lock (RLock object): the lock associated with the screen.
c_socket (socket object): the socket used for SEASIDE communication
c_socket_lock (RLock object): the lock associated with the socket.
"""
led_state = (0, 1, 0) # (r, g, b) for the screen LED
is_sending = False
global packet
delay_seconds, delay_useconds = 1, 0
delay_bytes = conversions.convert_delay_bytes(delay_seconds,
delay_useconds)
while True:
if lcd.is_pressed(LCD.SELECT): # Configure packet
packet_temp = pgen.configure_packet(lcd, lcd_lock)
if packet_temp is None:
time.sleep(0.3)
continue
packet = conversions.convert_packet_int_array(packet_temp)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.DELAY.value, delay_bytes)
time.sleep(1) # TODO: This separates sends. Make it unnecessary.
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.PACKET.value, packet)
led_state = (0, 1, 0)
elif lcd.is_pressed(LCD.UP): # Begin sending
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.START.value)
is_sending = True
elif lcd.is_pressed(LCD.DOWN): # Stop sending
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.STOP.value)
is_sending = False
elif lcd.is_pressed(LCD.LEFT): # Send single packet
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.SINGLE_PACKET.value)
threaded_lcd.flash_led(lcd, lcd_lock, *led_state)
elif lcd.is_pressed(LCD.RIGHT): # Configure delay
delay_seconds, delay_useconds = pgen.configure_delay(lcd, lcd_lock)
delay_bytes = conversions.convert_delay_bytes(delay_seconds,
delay_useconds)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.DELAY.value, delay_bytes)
threaded_lcd.flash_led(lcd, lcd_lock, 0, 0, 1)
if is_sending: # Ensures the LED always stays the right color.
threaded_lcd.lock_and_set_led_color(lcd, lcd_lock, *led_state)
else: # Turns it off if we've stopped sending
threaded_lcd.lock_and_set_led_color(lcd, lcd_lock, 0, 0, 0)
time.sleep(0.3) # Prevents duplicate presses
def user_interaction(lcd, lcd_lock, c_socket, c_socket_lock):
led_state = (0, 1, 0)
is_sending = False
global packet
delay_seconds, delay_useconds = 1, 0
delay_bytes = conversions.convert_delay_bytes(delay_seconds,
delay_useconds)
while True:
if lcd.is_pressed(LCD.SELECT): # Configure packet
packet_temp = pgen.configure_packet(lcd, lcd_lock)
if packet_temp is None:
time.sleep(0.3)
continue
packet = conversions.convert_packet_int_array(packet_temp)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.DELAY.value, delay_bytes)
time.sleep(1) # TODO: Fix needing this sleep function
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.PACKET.value, packet)
led_state = (0, 1, 0)
elif lcd.is_pressed(LCD.UP): # Begin sending
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.START.value)
is_sending = True
elif lcd.is_pressed(LCD.RIGHT): # Configure delay
delay_seconds, delay_useconds = pgen.configure_delay(lcd, lcd_lock)
delay_bytes = conversions.convert_delay_bytes(
delay_seconds, delay_useconds)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.DELAY.value, delay_bytes)
threaded_lcd.flash_led(lcd, lcd_lock, 0, 0, 1)
elif lcd.is_pressed(LCD.LEFT): # Send single packet
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.SINGLE_PACKET.value)
threaded_lcd.flash_led(lcd, lcd_lock, *led_state)
elif lcd.is_pressed(LCD.DOWN): # Stop sending
SEASIDE.send_SEASIDE(c_socket, c_socket_lock,
SEASIDE_FLAGS.STOP.value)
is_sending = False
if is_sending:
threaded_lcd.lock_and_set_led_color(lcd, lcd_lock, *led_state)
else:
threaded_lcd.lock_and_set_led_color(lcd, lcd_lock, 0, 0, 0)
time.sleep(0.3)
def load_pcap_file(self, filename):
if filename == '':
return
packet = rdpcap('pcap_files/' + filename)[0]
packet = conversions.convert_packet_int_array(packet)
SEASIDE.send_SEASIDE(c_socket, c_socket_lock, 0, packet)