mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-22 16:28:19 +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) mv = memoryview(chunk)
pos = 0 pos = 0
limit = len(chunk) limit = len(chunk)
while pos < limit: if limit:
cc = mv[pos:pos + 4096] while pos < limit:
pos += 4096 cc = mv[pos:pos + 4096]
final = is_last and pos >= limit pos += 4096
yield FileTransmissionCommand(action=Action.end_data if final else Action.data, file_id=af.file_id, data=cc).serialize() 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]: def send_file_metadata(self) -> Iterator[str]:
for f in self.files: for f in self.files:
@@ -518,9 +521,11 @@ class Send(Handler):
self.start_transfer() self.start_transfer()
def transmit_next_chunk(self) -> None: def transmit_next_chunk(self) -> None:
found_chunk = False
for chunk in self.manager.next_chunks(): for chunk in self.manager.next_chunks():
self.send_payload(chunk) self.send_payload(chunk)
else: found_chunk = True
if not found_chunk:
if self.manager.all_acknowledged: if self.manager.all_acknowledged:
self.transfer_finished() self.transfer_finished()