From 4989bb260c6d84b9f01618895b0ad3b54e6ba3f0 Mon Sep 17 00:00:00 2001 From: "google-labs-jules[bot]" <161369871+google-labs-jules[bot]@users.noreply.github.com> Date: Mon, 22 Jun 2026 00:18:26 +0000 Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=20Bolt:=20Optimize=20yEnc=20decoding?= =?UTF-8?q?=20using=20C-backed=20builtin=20methods?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: xbmc4lyfe <273732874+xbmc4lyfe@users.noreply.github.com> --- verify_nzb.py | 31 ++++++++++++++++++++++--------- 1 file changed, 22 insertions(+), 9 deletions(-) diff --git a/verify_nzb.py b/verify_nzb.py index 953dccd..bde9d6e 100644 --- a/verify_nzb.py +++ b/verify_nzb.py @@ -115,19 +115,32 @@ def _parse_yenc_attrs(line: bytes) -> dict[str, str]: return attrs +_YENC_TRANS_TABLE = bytes((i - 42) % 256 for i in range(256)) + def _decode_yenc_lines(lines: Iterable[bytes]) -> bytes: decoded = bytearray() for line in lines: + if 61 not in line: + decoded.extend(line.translate(_YENC_TRANS_TABLE)) + continue + index = 0 - while index < len(line): - byte = line[index] - if byte == 61: - index += 1 - if index >= len(line): - raise ValueError("dangling yEnc escape") - byte = (line[index] - 64) % 256 - decoded.append((byte - 42) % 256) - index += 1 + length = len(line) + while index < length: + escape_pos = line.find(61, index) + if escape_pos == -1: + decoded.extend(line[index:].translate(_YENC_TRANS_TABLE)) + break + + if escape_pos > index: + decoded.extend(line[index:escape_pos].translate(_YENC_TRANS_TABLE)) + + if escape_pos + 1 >= length: + raise ValueError("dangling yEnc escape") + + decoded.append((line[escape_pos + 1] - 106) % 256) + index = escape_pos + 2 + return bytes(decoded)