mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-16 13:34:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix end of data not being sent when using rsync

Browse Source
This commit is contained in:
Kovid Goyal
2021-10-02 06:15:06 +05:30
parent 1d1d55e2b4
commit ffd7b5779b
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
kittens/transfer/send.py
View File

@@ -349,11 +349,14 @@ class SendManager:
mv = memoryview(chunk)
pos = 0
limit = len(chunk)
while pos < limit:
cc = mv[pos:pos + 4096]
pos += 4096
final = is_last and pos >= limit
yield FileTransmissionCommand(action=Action.end_data if final else Action.data, file_id=af.file_id, data=cc).serialize()
if limit:
while pos < limit:
cc = mv[pos:pos + 4096]
pos += 4096
final = is_last and pos >= limit
yield FileTransmissionCommand(action=Action.end_data if final else Action.data, file_id=af.file_id, data=cc).serialize()
elif is_last:
yield FileTransmissionCommand(action=Action.end_data, file_id=af.file_id, data=b'').serialize()
def send_file_metadata(self) -> Iterator[str]:
for f in self.files:
@@ -518,9 +521,11 @@ class Send(Handler):
self.start_transfer()
def transmit_next_chunk(self) -> None:
found_chunk = False
for chunk in self.manager.next_chunks():
self.send_payload(chunk)
else:
found_chunk = True
if not found_chunk:
if self.manager.all_acknowledged:
self.transfer_finished()