60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
"""Foremost Analyzer for Image Submissions."""
|
|
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from .utils import MAX_PENDING_TIME, update_data
|
|
|
|
|
|
def analyze_foremost(input_img: Path, output_dir: Path) -> None:
|
|
"""Analyze an image submission using foremost."""
|
|
|
|
try:
|
|
foremost_dir = output_dir / "foremost"
|
|
foremost_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
stderr = ""
|
|
# Run foremost
|
|
data = subprocess.run(
|
|
["foremost", "-o", foremost_dir, "-i", input_img],
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
timeout=MAX_PENDING_TIME,
|
|
)
|
|
# Note: foremost use stderr as standard output for results :o)
|
|
# So we can't use it as error detection :
|
|
# stderr += data.stderr
|
|
|
|
# Zip extracted files
|
|
zip_data = subprocess.run(
|
|
["7z", "a", "../foremost.7z", "*"],
|
|
cwd=foremost_dir,
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
timeout=MAX_PENDING_TIME,
|
|
)
|
|
|
|
stderr += zip_data.stderr
|
|
|
|
# Remove the extracted directory
|
|
shutil.rmtree(foremost_dir)
|
|
|
|
data_strings = data.stdout.split("\n") if data else []
|
|
data_strings = [s for s in data_strings if s]
|
|
|
|
update_data(
|
|
output_dir,
|
|
{
|
|
"foremost": {
|
|
"status": "ok",
|
|
"output": data_strings,
|
|
"download": f"/download/{output_dir.name}/foremost",
|
|
}
|
|
},
|
|
)
|
|
except Exception as e:
|
|
update_data(output_dir, {"foremost": {"status": "error", "error": str(e)}})
|