mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-28 18:02:38 +02:00
cbbd6afcc1
This commit removes code that's not being used, it can be exceptions with the `as e` where the exception itself is not used, the same with `with` statements, and some other parts where there were duplicated code.
287 lines
10 KiB
Python
287 lines
10 KiB
Python
import asyncio
|
|
import os
|
|
import sys
|
|
from typing import Dict, Optional
|
|
|
|
import fitz
|
|
import magic
|
|
|
|
from . import errors
|
|
from .common import DEFAULT_DPI, DangerzoneConverter, running_on_qubes
|
|
|
|
|
|
class DocumentToPixels(DangerzoneConverter):
|
|
async def write_page_count(self, count: int) -> None:
|
|
return await self.write_int(count)
|
|
|
|
async def write_page_width(self, width: int) -> None:
|
|
return await self.write_int(width)
|
|
|
|
async def write_page_height(self, height: int) -> None:
|
|
return await self.write_int(height)
|
|
|
|
async def write_page_data(self, data: bytes) -> None:
|
|
return await self.write_bytes(data)
|
|
|
|
def update_progress(self, text: str, *, error: bool = False) -> None:
|
|
print(text, file=sys.stderr)
|
|
|
|
async def convert(self) -> None:
|
|
conversions: Dict[str, Dict[str, Optional[str]]] = {
|
|
# .pdf
|
|
"application/pdf": {"type": "PyMuPDF"},
|
|
# .docx
|
|
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .doc
|
|
"application/msword": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .docm
|
|
"application/vnd.ms-word.document.macroEnabled.12": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .xlsx
|
|
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .xls
|
|
"application/vnd.ms-excel": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .pptx
|
|
"application/vnd.openxmlformats-officedocument.presentationml.presentation": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .ppt
|
|
"application/vnd.ms-powerpoint": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .odt
|
|
"application/vnd.oasis.opendocument.text": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .odg
|
|
"application/vnd.oasis.opendocument.graphics": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .odp
|
|
"application/vnd.oasis.opendocument.presentation": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .ods
|
|
"application/vnd.oasis.opendocument.spreadsheet": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .ods / .ots
|
|
"application/vnd.oasis.opendocument.spreadsheet-template": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .odt / .ott
|
|
"application/vnd.oasis.opendocument.text-template": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .hwp
|
|
# Commented MIMEs are not used in `file` and don't conform to the rules.
|
|
# Left them for just in case
|
|
# PR: https://github.com/freedomofpress/dangerzone/pull/460
|
|
# "application/haansofthwp": {
|
|
# "type": "libreoffice",
|
|
# "libreoffice_ext": "h2orestart.oxt",
|
|
# },
|
|
# "application/vnd.hancom.hwp": {
|
|
# "type": "libreoffice",
|
|
# "libreoffice_ext": "h2orestart.oxt",
|
|
# },
|
|
"application/x-hwp": {
|
|
"type": "libreoffice",
|
|
"libreoffice_ext": "h2orestart.oxt",
|
|
},
|
|
# .hwpx
|
|
# "application/haansofthwpx": {
|
|
# "type": "libreoffice",
|
|
# "libreoffice_ext": "h2orestart.oxt",
|
|
# },
|
|
# "application/vnd.hancom.hwpx": {
|
|
# "type": "libreoffice",
|
|
# "libreoffice_ext": "h2orestart.oxt",
|
|
# },
|
|
"application/x-hwp+zip": {
|
|
"type": "libreoffice",
|
|
"libreoffice_ext": "h2orestart.oxt",
|
|
},
|
|
"application/hwp+zip": {
|
|
"type": "libreoffice",
|
|
"libreoffice_ext": "h2orestart.oxt",
|
|
},
|
|
# At least .odt, .docx, .odg, .odp, .ods, and .pptx
|
|
"application/zip": {
|
|
"type": "libreoffice",
|
|
},
|
|
# At least .doc, .docx, .odg, .odp, .odt, .pdf, .ppt, .pptx, .xls, and .xlsx
|
|
"application/octet-stream": {
|
|
"type": "libreoffice",
|
|
},
|
|
# At least .doc, .ppt, and .xls
|
|
"application/x-ole-storage": {
|
|
"type": "libreoffice",
|
|
},
|
|
# .epub
|
|
"application/epub+zip": {"type": "PyMuPDF"},
|
|
# .svg
|
|
"image/svg+xml": {"type": "PyMuPDF"},
|
|
# .bmp
|
|
"image/bmp": {"type": "PyMuPDF"},
|
|
# .pnm
|
|
"image/x-portable-anymap": {"type": "PyMuPDF"},
|
|
# .pbm
|
|
"image/x-portable-bitmap": {"type": "PyMuPDF"},
|
|
# .ppm
|
|
"image/x-portable-pixmap": {"type": "PyMuPDF"},
|
|
# .jpg
|
|
"image/jpeg": {"type": "PyMuPDF"},
|
|
# .gif
|
|
"image/gif": {"type": "PyMuPDF"},
|
|
# .png
|
|
"image/png": {"type": "PyMuPDF"},
|
|
# .tif
|
|
"image/tiff": {"type": "PyMuPDF"},
|
|
"image/x-tiff": {"type": "PyMuPDF"},
|
|
}
|
|
|
|
# Detect MIME type
|
|
mime_type = self.detect_mime_type("/tmp/input_file")
|
|
|
|
# Validate MIME type
|
|
if mime_type not in conversions:
|
|
raise errors.DocFormatUnsupported()
|
|
|
|
# Temporary fix for the HWPX format
|
|
# Should be removed after new release of `file' (current release 5.44)
|
|
if mime_type == "application/zip":
|
|
file_type = self.detect_mime_type("/tmp/input_file")
|
|
hwpx_file_type = 'Zip data (MIME type "application/hwp+zip"?)'
|
|
if file_type == hwpx_file_type:
|
|
mime_type = "application/x-hwp+zip"
|
|
|
|
# Convert input document to PDF
|
|
conversion = conversions[mime_type]
|
|
if conversion["type"] == "PyMuPDF":
|
|
try:
|
|
doc = fitz.open("/tmp/input_file", filetype=mime_type)
|
|
except (ValueError, fitz.FileDataError):
|
|
raise errors.DocCorruptedException()
|
|
elif conversion["type"] == "libreoffice":
|
|
libreoffice_ext = conversion.get("libreoffice_ext", None)
|
|
# Disable conversion for HWP/HWPX on specific platforms. See:
|
|
#
|
|
# https://github.com/freedomofpress/dangerzone/issues/494
|
|
# https://github.com/freedomofpress/dangerzone/issues/498
|
|
if libreoffice_ext == "h2orestart.oxt" and running_on_qubes():
|
|
raise errors.DocFormatUnsupportedHWPQubes()
|
|
if libreoffice_ext:
|
|
await self.install_libreoffice_ext(libreoffice_ext)
|
|
self.update_progress("Converting to PDF using LibreOffice")
|
|
args = [
|
|
"libreoffice",
|
|
"--headless",
|
|
"--safe-mode",
|
|
"--convert-to",
|
|
"pdf",
|
|
"--outdir",
|
|
"/tmp",
|
|
"/tmp/input_file",
|
|
]
|
|
await self.run_command(
|
|
args,
|
|
error_message="Conversion to PDF with LibreOffice failed",
|
|
)
|
|
pdf_filename = "/tmp/input_file.pdf"
|
|
# XXX: Sometimes, LibreOffice can fail with status code 0. So, we need to
|
|
# always check if the file exists. See:
|
|
#
|
|
# https://github.com/freedomofpress/dangerzone/issues/494
|
|
if not os.path.exists(pdf_filename):
|
|
raise errors.LibreofficeFailure()
|
|
try:
|
|
doc = fitz.open(pdf_filename)
|
|
except (ValueError, fitz.FileDataError):
|
|
raise errors.DocCorruptedException()
|
|
else:
|
|
# NOTE: This should never be reached
|
|
raise errors.DocFormatUnsupported()
|
|
|
|
# Obtain number of pages
|
|
if doc.page_count > errors.MAX_PAGES:
|
|
raise errors.MaxPagesException()
|
|
await self.write_page_count(doc.page_count)
|
|
|
|
for page in doc.pages():
|
|
# TODO check if page.number is doc-controlled
|
|
page_num = page.number + 1 # pages start in 1
|
|
|
|
self.update_progress(
|
|
f"Converting page {page_num}/{doc.page_count} to pixels"
|
|
)
|
|
pix = page.get_pixmap(dpi=DEFAULT_DPI)
|
|
rgb_buf = pix.samples_mv
|
|
await self.write_page_width(pix.width)
|
|
await self.write_page_height(pix.height)
|
|
await self.write_page_data(rgb_buf)
|
|
|
|
self.update_progress("Converted document to pixels")
|
|
|
|
async def install_libreoffice_ext(self, libreoffice_ext: str) -> None:
|
|
self.update_progress(f"Installing LibreOffice extension '{libreoffice_ext}'")
|
|
unzip_args = [
|
|
"unzip",
|
|
"-d",
|
|
f"/usr/lib/libreoffice/share/extensions/{libreoffice_ext}/",
|
|
f"/libreoffice_ext/{libreoffice_ext}",
|
|
]
|
|
await self.run_command(
|
|
unzip_args,
|
|
error_message="LibreOffice extension installation failed (unzipping)",
|
|
)
|
|
|
|
def detect_mime_type(self, path: str) -> str:
|
|
"""Detect MIME types in a platform-agnostic type.
|
|
|
|
Detect the MIME type of a file, either on Qubes or container platforms.
|
|
"""
|
|
try:
|
|
mime = magic.Magic(mime=True)
|
|
mime_type = mime.from_file("/tmp/input_file")
|
|
except TypeError:
|
|
mime_type = magic.detect_from_filename("/tmp/input_file").mime_type
|
|
|
|
return mime_type
|
|
|
|
|
|
async def main() -> None:
|
|
try:
|
|
data = await DocumentToPixels.read_bytes()
|
|
except EOFError:
|
|
sys.exit(1)
|
|
|
|
with open("/tmp/input_file", "wb") as f:
|
|
f.write(data)
|
|
|
|
try:
|
|
converter = DocumentToPixels()
|
|
await converter.convert()
|
|
except errors.ConversionException as e:
|
|
await DocumentToPixels.write_bytes(str(e).encode(), file=sys.stderr)
|
|
sys.exit(e.error_code)
|
|
except Exception as e:
|
|
await DocumentToPixels.write_bytes(str(e).encode(), file=sys.stderr)
|
|
error_code = errors.UnexpectedConversionError.error_code
|
|
sys.exit(error_code)
|
|
|
|
# Write debug information
|
|
await DocumentToPixels.write_bytes(converter.captured_output, file=sys.stderr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(asyncio.run(main()))
|