_bw(): check sent count

bubbleboy14 · Apr 25, 2024 · 516ba92 · 516ba92
1 parent 1fbf05c
commit 516ba92
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/rel/rel.py b/rel/rel.py
@@ -298,7 +298,12 @@ def stop():
 def _bw(fn):
     wopts = writings[fn]
     try:
-        wopts["sender"](wopts["sock"], wopts["data"].pop(0))
+        wd = wopts["data"][0]
+        sent = wopts["sender"](wopts["sock"], wd)
+        if sent == len(wd):
+            wopts["data"].pop(0)
+        else:
+            wopts["data"][0] = wd[sent:]
         return wopts["data"]
     except OSError:
         wopts["err"]()