Skip to content

Commit

Permalink
Add ability to publish package indices using AptByHash format
Browse files Browse the repository at this point in the history
closes #795
  • Loading branch information
Andrew Cheng committed Sep 25, 2023
1 parent 3264630 commit fe55235
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 0 deletions.
2 changes: 2 additions & 0 deletions CHANGES/795.feature
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Mitigating the Hash Sum Mismatch error in debian repos by generating additional metadata files named by checksum.
Wiki of the feature: https://wiki.ubuntu.com/AptByHash
1 change: 1 addition & 0 deletions pulp_deb/app/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,6 @@
http://docs.pulpproject.org/en/3.0/nightly/plugins/plugin-writer/index.html
"""

APT_BY_HASH = False
FORBIDDEN_CHECKSUM_WARNINGS = True
FORCE_IGNORE_MISSING_PACKAGE_INDICES = False
27 changes: 27 additions & 0 deletions pulp_deb/app/tasks/publishing.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import os
import shutil
from contextlib import suppress
from pathlib import Path

from datetime import datetime, timezone
from debian import deb822
Expand Down Expand Up @@ -39,6 +40,7 @@
CHECKSUM_TYPE_MAP,
)

from pulp_deb.app.settings import APT_BY_HASH

import logging
from gettext import gettext as _
Expand Down Expand Up @@ -300,6 +302,23 @@ def finish(self):
publication=self.parent.publication, file=File(open(gz_package_index_path, "rb"))
)
gz_package_index.save()

# Generating metadata files using checksum
if APT_BY_HASH:
for path, index in (
(package_index_path, package_index),
(gz_package_index_path, gz_package_index),
):
for checksum in settings.ALLOWED_CONTENT_CHECKSUMS:
if checksum in CHECKSUM_TYPE_MAP:
hashed_index_path = _fetch_file_checksum(path, index, checksum)
hashed_index = PublishedMetadata.create_from_file(
publication=self.parent.publication,
file=File(open(path, "rb")),
relative_path=hashed_index_path,
)
hashed_index.save()

self.parent.add_metadata(package_index)
self.parent.add_metadata(gz_package_index)

Expand Down Expand Up @@ -339,6 +358,7 @@ def __init__(
self.release["Components"] = "" # Will be set later
if release.description != NULL_VALUE:
self.release["Description"] = release.description
self.release["Acquire-By-Hash"] = "yes" if APT_BY_HASH else "no"

for checksum_type, deb_field in CHECKSUM_TYPE_MAP.items():
if checksum_type in settings.ALLOWED_CONTENT_CHECKSUMS:
Expand Down Expand Up @@ -413,3 +433,10 @@ def _zip_file(file_path):
with GzipFile(gz_file_path, "wb") as f_out:
shutil.copyfileobj(f_in, f_out)
return gz_file_path


def _fetch_file_checksum(file_path, index, checksum):
digest = getattr(index.contentartifact_set.first().artifact, checksum)
checksum_type = CHECKSUM_TYPE_MAP[checksum]
hashed_path = Path(file_path).parents[0] / "by-hash" / checksum_type / digest
return hashed_path

0 comments on commit fe55235

Please sign in to comment.