@@ -24,7 +24,7 @@ def compute_crc16(data):
2424class FirmwareUpdateFactory :
2525 def __init__ (self ):
2626 self .buffer = bytearray (256 )
27-
27+
2828 def create_fwug_start_msg (self ):
2929 msg_len = 2 + 2 # Total length: 2 bytes header + 2 bytes footer
3030 msg_header = struct .pack ('BB' , COM_PROTO_MSG_TYPE_FWUG_START , msg_len )
@@ -34,61 +34,61 @@ def create_fwug_start_msg(self):
3434 message = msg_header + struct .pack ('>H' , crc16 )
3535 self .buffer [:len (message )] = message
3636 return self .buffer [:len (message )]
37-
37+
3838 def create_fwug_data_msg (self , packet_number , payload ):
3939 if len (payload ) != FIRMWARE_UPDATE_PACKET_SIZE :
4040 raise ValueError (f"Payload must be { FIRMWARE_UPDATE_PACKET_SIZE } )
41-
41+
4242 msg_len = 2 + 2 + FIRMWARE_UPDATE_PACKET_SIZE + 2 # Header + packet number + payload + footer
4343 msg_header = struct .pack ('BB' , COM_PROTO_MSG_TYPE_FWUG_DATA , msg_len )
4444 packet_num = struct .pack ('<H' , packet_number ) # Big-endian packet number
4545 msg_footer = struct .pack ('H' , 0 ) # Placeholder for CRC16
46-
46+
4747 message = msg_header + packet_num + payload + msg_footer
4848 crc16 = compute_crc16 (message [:- 2 ])
4949 message = msg_header + packet_num + payload + struct .pack ('>H' , crc16 )
50-
50+
5151 self .buffer [:len (message )] = message
5252 return self .buffer [:len (message )]
53-
53+
5454 def create_fwug_cancel_msg (self ):
5555 msg_len = 2 + 2 # 2 bytes header + 2 bytes footer
5656 msg_header = struct .pack ('BB' , COM_PROTO_MSG_TYPE_FWUG_CANCEL , msg_len )
5757 msg_footer = struct .pack ('H' , 0 )
5858 message = msg_header + msg_footer
5959 crc16 = compute_crc16 (message [:- 2 ])
6060 message = msg_header + struct .pack ('>H' , crc16 )
61-
61+
6262 self .buffer [:len (message )] = message
6363 return self .buffer [:len (message )]
6464
65- def send_message_via_serial (message , port = 'COM9 ' , baudrate = 115200 ):
65+ def send_message_via_serial (message , port = '/dev/ttyACM0 ' , baudrate = 115200 ):
6666 # Open serial port
6767 ser = serial .Serial (port , baudrate )
68-
68+
6969 # Create the buffer of 256 bytes
7070 buffer_size = 256
7171 buffer = bytearray (buffer_size )
72-
72+
7373 # Copy message into the buffer
7474 buffer [:len (message )] = message
75-
75+
7676 # Fill the rest of the buffer with 0xFF
7777 for i in range (len (message ), buffer_size ):
7878 buffer [i ] = 0xFF
79-
79+
8080 # Convert buffer to hex string
8181 hex_buffer = buffer .hex ()
82-
82+
8383 # Print hex buffer for verification
8484 #print("Hex Buffer:", hex_buffer)
8585 ser .write (bytes .fromhex (hex_buffer ))
86-
86+
8787 # Initialize variables for response handling
8888 max_wait_time = 15 # Maximum wait time in seconds
89- check_interval = 0.001 # Interval to check for response in seconds
89+ check_interval = 0.01 # Interval to check for response in seconds
9090 total_wait_time = 0
91-
91+
9292 # Wait for the response
9393 response = bytearray ()
9494 while total_wait_time < max_wait_time :
@@ -97,15 +97,16 @@ def send_message_via_serial(message, port='COM9', baudrate=115200):
9797 break
9898 time .sleep (check_interval )
9999 total_wait_time += check_interval
100-
100+
101101 # Print the response in hex format for verification
102102 # if response:
103103 # print("Response (Hex):", response.hex())
104104 # else:
105105 # print("No response received within the maximum wait time")
106-
106+
107107 # Close serial port
108108 ser .close ()
109+ return response
109110
110111# Function to send firmware update data in chunks
111112def send_firmware_update (file_path , factory ):
@@ -116,12 +117,12 @@ def send_firmware_update(file_path, factory):
116117 data_chunk = f .read (FIRMWARE_UPDATE_PACKET_SIZE )
117118 if not data_chunk :
118119 break # No more data to send
119-
120+
120121 # Create FWUG_DATA message
121122 data_msg = factory .create_fwug_data_msg (packet_number , data_chunk )
122123 print (f"Sending FWUG_DATA Message { packet_number } )
123- send_message_via_serial (data_msg )
124-
124+ response = send_message_via_serial (data_msg )
125+ print ( "Response:" , response . hex ())
125126 packet_number += 1
126127
127128if __name__ == "__main__" :
0 commit comments