Git-Mirrors/onionshare
1
0
Fork 0
mirror of https://github.com/onionshare/onionshare.git synced 2025-04-28 11:16:08 -04:00
Code Issues Packages Projects Releases Wiki Activity

Catch more exceptions in ReceiveModeFile that trigger on a full disk

Browse Source
This commit is contained in:
Micah Lee 2019-02-12 15:54:25 -08:00
parent 09c2696c27
commit e625d3b009
No known key found for this signature in database
GPG Key ID: 403C2657CD994F73
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
onionshare/web/receive_mode.py
View File

@ -198,21 +198,24 @@ class ReceiveModeFile(object):
try: try:
bytes_written = self.f.write(b) bytes_written = self.f.write(b)
self.onionshare_write_func(self.onionshare_filename, bytes_written)
except: except:
# If we can't write the file, close early
self.upload_error = True self.upload_error = True
return
self.onionshare_write_func(self.onionshare_filename, bytes_written)
def close(self): def close(self):
""" """
Custom close method that calls out to onionshare_close_func Custom close method that calls out to onionshare_close_func
""" """
self.f.close() try:
self.f.close()
if not self.upload_error: if not self.upload_error:
# Rename the in progress file to the final filename # Rename the in progress file to the final filename
os.rename(self.filename_in_progress, self.filename) os.rename(self.filename_in_progress, self.filename)
except:
self.upload_error = True
self.onionshare_close_func(self.onionshare_filename) self.onionshare_close_func(self.onionshare_filename)