mirror of
https://github.com/ytdl-org/youtube-dl.git
synced 2024-12-22 16:57:40 +00:00
[utils] Fix _windows_write_string (Fixes #2779)
It turns out that the function did not work for outputs longer than 1024 UCS-2 tokens. Write non-BMP characters one by one to ensure that we count correctly.
This commit is contained in:
parent
edec83a025
commit
d1b9c912a4
1 changed files with 16 additions and 7 deletions
|
@ -923,9 +923,6 @@ def _windows_write_string(s, out):
|
||||||
2: -12,
|
2: -12,
|
||||||
}
|
}
|
||||||
|
|
||||||
def ucs2_len(s):
|
|
||||||
return sum((2 if ord(c) > 0xffff else 1) for c in s)
|
|
||||||
|
|
||||||
fileno = out.fileno()
|
fileno = out.fileno()
|
||||||
if fileno not in WIN_OUTPUT_IDS:
|
if fileno not in WIN_OUTPUT_IDS:
|
||||||
return False
|
return False
|
||||||
|
@ -959,13 +956,25 @@ def _windows_write_string(s, out):
|
||||||
if not_a_console(h):
|
if not_a_console(h):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
remaining = ucs2_len(s)
|
def next_nonbmp_pos(s):
|
||||||
while remaining > 0:
|
try:
|
||||||
|
return next(i for i, c in enumerate(s) if ord(c) > 0xffff)
|
||||||
|
except StopIteration:
|
||||||
|
return len(s)
|
||||||
|
|
||||||
|
while s:
|
||||||
|
count = min(next_nonbmp_pos(s), 1024)
|
||||||
|
|
||||||
ret = WriteConsoleW(
|
ret = WriteConsoleW(
|
||||||
h, s, min(remaining, 1024), ctypes.byref(written), None)
|
h, s, count if count else 2, ctypes.byref(written), None)
|
||||||
if ret == 0:
|
if ret == 0:
|
||||||
raise OSError('Failed to write string')
|
raise OSError('Failed to write string')
|
||||||
remaining -= written.value
|
if not count: # We just wrote a non-BMP character
|
||||||
|
assert written.value == 2
|
||||||
|
s = s[1:]
|
||||||
|
else:
|
||||||
|
assert written.value > 0
|
||||||
|
s = s[written.value:]
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue