@@ -189,7 +189,9 @@ def authenticate(remote_addr, remote_port, password, md5_target, filename, conte
189189 return False , str (e )
190190
191191
192- def serve (remote_addr , local_addr , remote_port , local_port , password , md5_target , filename , command = FLASH ): # noqa: C901
192+ def serve (
193+ remote_addr , local_addr , remote_port , local_port , password , md5_target , filename , command = FLASH
194+ ): # noqa: C901
193195 # Create a TCP/IP socket
194196 sock = socket .socket (socket .AF_INET , socket .SOCK_STREAM )
195197 server_address = (local_addr , local_port )
@@ -220,7 +222,9 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, md5_target
220222 # Try authentication with the specified method first
221223 sys .stderr .write ("Authenticating..." )
222224 sys .stderr .flush ()
223- auth_success , auth_error = authenticate (remote_addr , remote_port , password , md5_target , filename , content_size , file_md5 , nonce )
225+ auth_success , auth_error = authenticate (
226+ remote_addr , remote_port , password , md5_target , filename , content_size , file_md5 , nonce
227+ )
224228
225229 if not auth_success :
226230 # If authentication failed and we're not already using MD5, try with MD5
@@ -229,7 +233,9 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, md5_target
229233 logging .warning ("Authentication failed with SHA256, retrying with MD5: %s" , auth_error )
230234
231235 # Restart the entire process with MD5 to get a fresh nonce
232- success , data , error = send_invitation_and_get_auth_challenge (remote_addr , remote_port , message , True )
236+ success , data , error = send_invitation_and_get_auth_challenge (
237+ remote_addr , remote_port , message , True
238+ )
233239 if not success :
234240 logging .error ("Failed to re-establish connection for MD5 retry: %s" , error )
235241 return 1
@@ -238,7 +244,9 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, md5_target
238244 nonce = data .split ()[1 ]
239245 sys .stderr .write ("Retrying with MD5..." )
240246 sys .stderr .flush ()
241- auth_success , auth_error = authenticate (remote_addr , remote_port , password , True , filename , content_size , file_md5 , nonce )
247+ auth_success , auth_error = authenticate (
248+ remote_addr , remote_port , password , True , filename , content_size , file_md5 , nonce
249+ )
242250 else :
243251 auth_success = False
244252 auth_error = "Expected AUTH challenge for MD5 retry, got: " + data
@@ -331,7 +339,9 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, md5_target
331339
332340 # After all attempts, provide detailed error information
333341 if received_any_response :
334- logging .warning ("Upload completed but device sent unexpected response(s). This may still be successful." )
342+ logging .warning (
343+ "Upload completed but device sent unexpected response(s). This may still be successful."
344+ )
335345 logging .warning ("Device might be rebooting to apply firmware - this is normal." )
336346 connection .close ()
337347 return 0 # Consider it successful if we got any response and upload completed
@@ -448,7 +458,7 @@ def main(args):
448458 options .auth ,
449459 options .md5_target ,
450460 options .image ,
451- command
461+ command ,
452462 )
453463
454464
0 commit comments