bin_putty_log_provide_message_length.py

#-------------------------------------------------------------------------------
# Name:        module1
# Purpose:
#
# Author:      User
#
# Created:     22/11/2015
# Copyright:   (c) User 2015
# Licence:     <your licence>
#-------------------------------------------------------------------------------
import os



#
input_file_path = os.path.join("logs", "2015-11-16_play_game_fo_a_while_putty.log")
#
lists_output_dir = os.path.join("binned_log", "by_length_byte", "lists")
if not os.path.exists(lists_output_dir):
    os.makedirs(lists_output_dir)
#
raw_output_dir = os.path.join("binned_log", "by_length_byte", "raw")
if not os.path.exists(raw_output_dir):
    os.makedirs(raw_output_dir)
#
WYLDSTYLE_UID = "04 9A 74 6A 0B 40 80"






def convert_to_byte_list(packet_line_string):
    """
    Take line from usb-mitm stdout and make it into a python libusb statement
    01[32]: 55 14 c6 16 01 1e 01 ff 00 18 01 1e 01 ff 00 18 01 1e 01 ff 00 18 ea 00 00 00 00 00 00 00 00 00
    becomes
    dev.write(1, [0x55, 0x06, 0xc0, 0x02, 0x00, 0xff, 0xff, 0xff, 0x1a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00])# Switch to blue
    """
    packet_byte_strings = packet_line_string.split(" ")[1:]
    formatted_packet_bytes = "["
    for packet_byte_string in packet_byte_strings:
        formatted_packet_bytes += "0x"+packet_byte_string+", "
    formatted_packet_bytes = formatted_packet_bytes[:-2]
    formatted_packet_bytes += "]"
    return formatted_packet_bytes



with open(input_file_path, "r") as input_file:
    for line in input_file:
        if "[32]:" not in line:# All USB endpoints for the device take 32-byte packets
            continue
        #print(line)
        #81[32]: 55 01 15 6b 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00
        endpoint_number = line.split("[")[0]
        byte_strings = line.split(" ")[1:]
        length_byte = "0x"+byte_strings[1]
        # Save raw
        output_file_name = endpoint_number+"_"+length_byte+".txt"
        output_file_path = os.path.join(raw_output_dir, output_file_name)
        with open(output_file_path, "a") as output_file:
            output_file.write(line)
        # Save as byte list
        bytelist = convert_to_byte_list(line[:-1])
        bytelist_output_file_name = endpoint_number+"_"+length_byte+".bytelists"
        bytelist_output_file_path = os.path.join(lists_output_dir, bytelist_output_file_name)
        with open(bytelist_output_file_path, "a") as bytelist_output_file:
            bytelist_output_file.write(bytelist+",\n")

def main():
    pass

if __name__ == '__main__':
    main()