@@ -94,7 +94,8 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, filename,
94
94
return 1
95
95
96
96
content_size = os .path .getsize (filename )
97
- file_md5 = hashlib .md5 (open (filename , "rb" ).read ()).hexdigest ()
97
+ with open (filename , "rb" ) as f :
98
+ file_md5 = hashlib .md5 (f .read ()).hexdigest ()
98
99
logging .info ("Upload size: %d" , content_size )
99
100
message = "%d %d %d %s\n " % (command , local_port , content_size , file_md5 )
100
101
@@ -163,6 +164,7 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, filename,
163
164
sock2 .close ()
164
165
165
166
logging .info ("Waiting for device..." )
167
+
166
168
try :
167
169
sock .settimeout (10 )
168
170
connection , client_address = sock .accept ()
@@ -172,6 +174,7 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, filename,
172
174
logging .error ("No response from device" )
173
175
sock .close ()
174
176
return 1
177
+
175
178
try :
176
179
with open (filename , "rb" ) as f :
177
180
if PROGRESS :
@@ -225,7 +228,8 @@ def serve(remote_addr, local_addr, remote_port, local_port, password, filename,
225
228
logging .error ("Error response from device" )
226
229
connection .close ()
227
230
return 1
228
-
231
+ except Exception as e : # noqa: E722
232
+ logging .error ("Error: %s" , str (e ))
229
233
finally :
230
234
connection .close ()
231
235
0 commit comments