venus/aperisolve
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #155 from aradhyacp/fix/png-PCRT-issue

Browse Source 
fix: PCRT repaired PNG files corruption and filename extension #152
This commit is contained in:
Zeecka
2026-01-10 00:17:54 +01:00
committed by GitHub
parent c02e02b967 75ad5916af
commit 21a8361067
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aperisolve/analyzers/pcrt.py
+1 -1
View File
@@ -18,7 +18,7 @@ class PCRTAnalyzer(SubprocessAnalyzer):
def _write_repaired_data(self, data: bytes) -> str: def _write_repaired_data(self, data: bytes) -> str:
"""Write recovered image.""" """Write recovered image."""
img_name = f"pcrt_recoverd_{self.input_img.name}.png" img_name = f"pcrt_recovered_{self.input_img.stem}.png"
output_path = self.output_dir / img_name output_path = self.output_dir / img_name
saved_img_url = "/image/" + str(Path(self.output_dir.name) / img_name) saved_img_url = "/image/" + str(Path(self.output_dir.name) / img_name)
with output_path.open("wb") as f: with output_path.open("wb") as f:
aperisolve/utils/png.py
+3 -2
View File
@@ -57,8 +57,9 @@ class PNG:
pos = data.find(b"IHDR") pos = data.find(b"IHDR")
if pos == -1: if pos == -1:
return -1, b"" return -1, b""
idat_begin = data.find(b"IDAT") # IHDR chunk is always 25 bytes: 4 (length) + 4 (type) + 13 (data) + 4 (CRC)
return pos, data[pos - 4 : idat_begin - 4 if idat_begin != -1 else pos + 21] # Only return the IHDR chunk itself, not everything up to IDAT
return pos, data[pos - 4 : pos + 21]
def check_header(self) -> bool: def check_header(self) -> bool:
"""Check and fix PNG header.""" """Check and fix PNG header."""