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

Lint part 2

Browse Source
This commit is contained in:
Zeecka
2025-12-10 22:30:46 +01:00
parent d1fb8b3760
commit d86416c97d
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aperisolve/analyzers/image_resize.py
+6 -7
View File
@@ -177,8 +177,8 @@ def analyze_image_resize(input_img: Path, output_dir: Path) -> None:
b = bytearray(f) b = bytearray(f)
# Find IHDR start # Find IHDR start
IHDR = b.find(b"\x49\x48\x44\x52") ihdr = b.find(b"\x49\x48\x44\x52")
if IHDR == -1: if ihdr == -1:
logs.append("Failure: PNG header (IHDR) not found.") logs.append("Failure: PNG header (IHDR) not found.")
update_data( update_data(
output_dir, output_dir,
@@ -187,13 +187,13 @@ def analyze_image_resize(input_img: Path, output_dir: Path) -> None:
return return
# Calculate Chunk Length (The 4 bytes BEFORE IHDR tag) # Calculate Chunk Length (The 4 bytes BEFORE IHDR tag)
chunk_length = int.from_bytes((b[IHDR - 4 : IHDR]), byteorder="big") + 4 chunk_length = int.from_bytes((b[ihdr - 4 : ihdr]), byteorder="big") + 4
# Extract Target CRC (The 4 bytes AFTER the chunk data) # Extract Target CRC (The 4 bytes AFTER the chunk data)
target_crc = b[IHDR + chunk_length : IHDR + chunk_length + 4] target_crc = b[ihdr + chunk_length : ihdr + chunk_length + 4]
# Isolate the header chunk (Type + Data) # Isolate the header chunk (Type + Data)
header_chunk = b[IHDR : IHDR + chunk_length] header_chunk = b[ihdr : ihdr + chunk_length]
logs.append(f"Target CRC found: 0x{target_crc.hex()}") logs.append(f"Target CRC found: 0x{target_crc.hex()}")
@@ -221,7 +221,7 @@ def analyze_image_resize(input_img: Path, output_dir: Path) -> None:
# Splicing: [Start...IHDR+4] + [W] + [H] + [IHDR+12...End] # Splicing: [Start...IHDR+4] + [W] + [H] + [IHDR+12...End]
full_png_data = ( full_png_data = (
b[: IHDR + 4] + width_bytes + height_bytes + b[IHDR + 12 :] b[: ihdr + 4] + width_bytes + height_bytes + b[ihdr + 12 :]
) )
# Write the recovered file to disk # Write the recovered file to disk
@@ -247,4 +247,3 @@ def analyze_image_resize(input_img: Path, output_dir: Path) -> None:
except Exception as e: except Exception as e:
update_data(output_dir, {"image_resize": {"status": "error", "error": str(e)}}) update_data(output_dir, {"image_resize": {"status": "error", "error": str(e)}})
return None