open file only once on sending
This commit is contained in:
parent
21d0ca72a2
commit
ec20af6941
|
@ -351,7 +351,6 @@ class Socks5:
|
||||||
|
|
||||||
def open_file_for_reading(self):
|
def open_file_for_reading(self):
|
||||||
self.fd = open(self.file_props['file-name'])
|
self.fd = open(self.file_props['file-name'])
|
||||||
self.fd.seek(self.size)
|
|
||||||
|
|
||||||
def close_file(self):
|
def close_file(self):
|
||||||
try:
|
try:
|
||||||
|
@ -425,9 +424,7 @@ class Socks5:
|
||||||
buff = self.remaining_buff
|
buff = self.remaining_buff
|
||||||
self.remaining_buff = ''
|
self.remaining_buff = ''
|
||||||
else:
|
else:
|
||||||
self.open_file_for_reading()
|
|
||||||
buff = self.fd.read(MAX_BUFF_LEN)
|
buff = self.fd.read(MAX_BUFF_LEN)
|
||||||
self.close_file()
|
|
||||||
if len(buff) > 0:
|
if len(buff) > 0:
|
||||||
lenn = 0
|
lenn = 0
|
||||||
try:
|
try:
|
||||||
|
@ -656,6 +653,7 @@ class Socks5Sender(Socks5):
|
||||||
|
|
||||||
def send_file(self):
|
def send_file(self):
|
||||||
''' start sending the file over verified connection '''
|
''' start sending the file over verified connection '''
|
||||||
|
self.open_file_for_reading()
|
||||||
self.file_props['error'] = 0
|
self.file_props['error'] = 0
|
||||||
self.file_props['disconnect_cb'] = self.disconnect
|
self.file_props['disconnect_cb'] = self.disconnect
|
||||||
self.file_props['started'] = True
|
self.file_props['started'] = True
|
||||||
|
|
Loading…
Reference in New Issue