Something went wrong. Try again.
Cloudflare Tunnel client with NetBSD support
Something went wrong. Try again.
cloudflared release_pkgs.py
18 kB · 462 lines
Python
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463""" This is a utility for creating deb and rpm packages, signing them and uploading them to a storage and adding metadata to workers KV.
It has two over-arching responsiblities: 1. Create deb and yum repositories from .deb and .rpm files. This is also responsible for signing the packages and generally preparing them to be in an uploadable state. 2. Upload these packages to a storage in a format that apt and yum expect."""import argparseimport base64import loggingimport osimport shutilfrom pathlib import Pathfrom subprocess import Popen, PIPE
import boto3import gnupgfrom botocore.client import Configfrom botocore.exceptions import ClientError
# The front facing R2 URL to access assets from.R2_ASSET_URL = 'https://demo-r2-worker.cloudflare-tunnel.workers.dev/'
class PkgUploader: def __init__(self, account_id, bucket_name, client_id, client_secret): self.account_id = account_id self.bucket_name = bucket_name self.client_id = client_id self.client_secret = client_secret
def upload_pkg_to_r2(self, filename, upload_file_path): endpoint_url = f"https://{self.account_id}.r2.cloudflarestorage.com"
config = Config( region_name='auto', s3={ "addressing_style": "path", } )
r2 = boto3.client( "s3", endpoint_url=endpoint_url, aws_access_key_id=self.client_id, aws_secret_access_key=self.client_secret, config=config, )
print(f"uploading asset: {filename} to {upload_file_path} in bucket {self.bucket_name}...") try: r2.upload_file(filename, self.bucket_name, upload_file_path) except ClientError as e: raise e
class PkgCreator: """ The distribution conf is what dictates to reprepro, the debian packaging building and signing tool we use, what distros to support, what GPG key to use for signing and what to call the debian binary etc. This function creates it "./conf/distributions".
origin - name of your package (String) label - label of your package (could be same as the name) (String) release - release you want this to be distributed for (List of Strings) components - could be a channel like main/stable/beta archs - Architecture (List of Strings) description - (String) gpg_key_id - gpg key id of what you want to use to sign the packages.(String) """
def create_distribution_conf(self, file_path, origin, label, releases, archs, components, description, gpg_key_id): with open(file_path, "w+") as distributions_file: for release in releases: distributions_file.write(f"Origin: {origin}\n") distributions_file.write(f"Label: {label}\n") distributions_file.write(f"Codename: {release}\n") archs_list = " ".join(archs) distributions_file.write(f"Architectures: {archs_list}\n") distributions_file.write(f"Components: {components}\n") distributions_file.write(f"Description: {description} - {release}\n") distributions_file.write(f"SignWith: {gpg_key_id}\n") distributions_file.write("\n") return distributions_file
""" Uses the reprepro tool to generate packages, sign them and create the InRelease as specified by the distribution_conf file.
This function creates three folders db, pool and dist. db and pool contain information and metadata about builds. We can ignore these. dist: contains all the pkgs and signed releases that are necessary for an apt download. """
def create_deb_pkgs(self, release, deb_file): print(f"creating deb pkgs: {release} : {deb_file}") p = Popen(["reprepro", "includedeb", release, deb_file], stdout=PIPE, stderr=PIPE) out, err = p.communicate() if p.returncode != 0: print(f"create deb_pkgs result => {out}, {err}") raise
def create_rpm_pkgs(self, artifacts_path, gpg_key_name): self._setup_rpm_pkg_directories(artifacts_path, gpg_key_name) p = Popen(["createrepo_c", "./rpm"], stdout=PIPE, stderr=PIPE) out, err = p.communicate() if p.returncode != 0: print(f"create rpm_pkgs result => {out}, {err}") raise
self._sign_repomd()
""" creates a <binary>.repo file with details like so [cloudflared-stable] name=cloudflared-stable baseurl=https://pkg.cloudflare.com/cloudflared/rpm enabled=1 type=rpm gpgcheck=1 gpgkey=https://pkg.cloudflare.com/cloudflare-main.gpg """
def create_repo_file(self, file_path, binary_name, baseurl, gpgkey_url): repo_file_path = os.path.join(file_path, binary_name + '.repo') with open(repo_file_path, "w+") as repo_file: repo_file.write(f"[{binary_name}-stable]\n") repo_file.write(f"name={binary_name}-stable\n") repo_file.write(f"baseurl={baseurl}/rpm\n") repo_file.write("enabled=1\n") repo_file.write("type=rpm\n") repo_file.write("gpgcheck=1\n") repo_file.write(f"gpgkey={gpgkey_url}\n") return repo_file_path
def _sign_rpms(self, file_path, gpg_key_name): p = Popen(["rpm", "--define", f"_gpg_name {gpg_key_name}", "--addsign", file_path], stdout=PIPE, stderr=PIPE) out, err = p.communicate() if p.returncode != 0: print(f"rpm sign result result => {out}, {err}") raise
def _sign_repomd(self): p = Popen(["gpg", "--batch", "--yes", "--detach-sign", "--armor", "./rpm/repodata/repomd.xml"], stdout=PIPE, stderr=PIPE) out, err = p.communicate() if p.returncode != 0: print(f"sign repomd result => {out}, {err}") raise
""" sets up and signs the RPM directories in the following format: - rpm - aarch64 - x86_64 - 386
this assumes the assets are in the format <prefix>-<aarch64/x86_64/386>.rpm """
def _setup_rpm_pkg_directories(self, artifacts_path, gpg_key_name, archs=["aarch64", "x86_64", "386"]): for arch in archs: for root, _, files in os.walk(artifacts_path): for file in files: if file.endswith(f"{arch}.rpm"): new_dir = f"./rpm/{arch}" os.makedirs(new_dir, exist_ok=True) old_path = os.path.join(root, file) new_path = os.path.join(new_dir, file) shutil.copyfile(old_path, new_path) self._sign_rpms(new_path, gpg_key_name)
""" imports gpg keys into the system so reprepro and createrepo can use it to sign packages. it returns the GPG ID after a successful import """
def import_gpg_keys(self, private_key, public_key): gpg = gnupg.GPG() private_key = base64.b64decode(private_key) import_result = gpg.import_keys(private_key) if not import_result.fingerprints: raise Exception("Failed to import private key")
public_key = base64.b64decode(public_key) gpg.import_keys(public_key)
imported_fingerprint = import_result.fingerprints[0] data = gpg.list_keys(secret=True)
# Find the specific key we just imported by comparing fingerprints for key in data: if key["fingerprint"] == imported_fingerprint: return (key["fingerprint"], key["uids"][0])
raise Exception(f"Could not find imported key with fingerprint {imported_fingerprint}")
def import_multiple_gpg_keys(self, primary_private_key, primary_public_key, secondary_private_key=None, secondary_public_key=None): """ Import one or two GPG keypairs. Returns a list of (fingerprint, uid) with the primary first. """ results = [] if primary_private_key and primary_public_key: results.append(self.import_gpg_keys(primary_private_key, primary_public_key)) if secondary_private_key and secondary_public_key: # Ensure secondary is imported and appended results.append(self.import_gpg_keys(secondary_private_key, secondary_public_key)) return results
""" basically rpm --import <key_file> This enables us to sign rpms. """
def import_rpm_key(self, public_key): file_name = "pb.key" with open(file_name, "wb") as f: public_key = base64.b64decode(public_key) f.write(public_key)
p = Popen(["rpm", "--import", file_name], stdout=PIPE, stderr=PIPE) out, err = p.communicate() if p.returncode != 0: print(f"create rpm import result => {out}, {err}") raise
""" Walks through a directory and uploads it's assets to R2. directory : root directory to walk through (String). release: release string. If this value is none, a specific release path will not be created and the release will be uploaded to the default path. binary: name of the binary to upload"""
def upload_from_directories(pkg_uploader, directory, release, binary): for root, _, files in os.walk(directory): for file in files: upload_file_name = os.path.join(binary, root, file) if release: upload_file_name = os.path.join(release, upload_file_name) filename = os.path.join(root, file) try: pkg_uploader.upload_pkg_to_r2(filename, upload_file_name) except ClientError as e: logging.error(e) return
""" 1. looks into a artifacts folder for cloudflared debs 2. creates Packages.gz, InRelease (signed) files 3. uploads them to Cloudflare R2
pkg_creator, pkg_uploader: are instantiations of the two classes above.
gpg_key_id: is an id indicating the key the package should be signed with. The public key of this id will be uploaded to R2 so it can be presented to apt downloaders.
release_version: is the cloudflared release version. Only provide this if you want a permanent backup."""
def create_deb_packaging(pkg_creator, pkg_uploader, releases, primary_gpg_key_id, secondary_gpg_key_id, binary_name, archs, package_component, release_version): # set configuration for package creation. print(f"initialising configuration for {binary_name} , {archs}") Path("./conf").mkdir(parents=True, exist_ok=True) # If in rollover mode (secondary provided), tell reprepro to sign with both keys. sign_with_ids = primary_gpg_key_id if not secondary_gpg_key_id else f"{primary_gpg_key_id} {secondary_gpg_key_id}" pkg_creator.create_distribution_conf( "./conf/distributions", binary_name, binary_name, releases, archs, package_component, f"apt repository for {binary_name}", sign_with_ids)
# create deb pkgs for release in releases: for arch in archs: print(f"creating deb pkgs for {release} and {arch}...") pkg_creator.create_deb_pkgs(release, f"./artifacts/cloudflared-linux-{arch}.deb")
print("uploading latest to r2...") upload_from_directories(pkg_uploader, "dists", None, binary_name) upload_from_directories(pkg_uploader, "pool", None, binary_name)
if release_version: print(f"uploading versioned release {release_version} to r2...") upload_from_directories(pkg_uploader, "dists", release_version, binary_name) upload_from_directories(pkg_uploader, "pool", release_version, binary_name)
def create_rpm_packaging( pkg_creator, pkg_uploader, artifacts_path, release_version, binary_name, gpg_key_name, base_url, gpg_key_url, upload_repo_file=False,): print(f"creating rpm pkgs...") pkg_creator.create_rpm_pkgs(artifacts_path, gpg_key_name) repo_file = pkg_creator.create_repo_file(artifacts_path, binary_name, base_url, gpg_key_url)
print("Uploading repo file") pkg_uploader.upload_pkg_to_r2(repo_file, binary_name + ".repo")
print("uploading latest to r2...") upload_from_directories(pkg_uploader, "rpm", None, binary_name)
if upload_repo_file: print(f"uploading versioned release {release_version} to r2...") upload_from_directories(pkg_uploader, "rpm", release_version, binary_name)
def parse_args(): parser = argparse.ArgumentParser( description="Creates linux releases and uploads them in a packaged format" )
parser.add_argument( "--bucket", default=os.environ.get("R2_BUCKET"), help="R2 Bucket name" ) parser.add_argument( "--id", default=os.environ.get("R2_CLIENT_ID"), help="R2 Client ID" ) parser.add_argument( "--secret", default=os.environ.get("R2_CLIENT_SECRET"), help="R2 Client Secret" ) parser.add_argument( "--account", default=os.environ.get("R2_ACCOUNT_ID"), help="R2 Account Tag" ) parser.add_argument( "--release-tag", default=os.environ.get("RELEASE_VERSION"), help="Release version you want your pkgs to be\ prefixed with. Leave empty if you don't want tagged release versions backed up to R2." )
parser.add_argument( "--binary", default=os.environ.get("BINARY_NAME"), help="The name of the binary the packages are for" )
parser.add_argument( "--gpg-private-key", default=os.environ.get("LINUX_SIGNING_PRIVATE_KEY"), help="GPG private key to sign the\ packages" )
parser.add_argument( "--gpg-public-key", default=os.environ.get("LINUX_SIGNING_PUBLIC_KEY"), help="GPG public key used for\ signing packages" )
# Optional secondary keypair for key rollover parser.add_argument( "--gpg-private-key-2", default=os.environ.get("LINUX_SIGNING_PRIVATE_KEY_2"), help="Secondary GPG private key for rollover" ) parser.add_argument( "--gpg-public-key-2", default=os.environ.get("LINUX_SIGNING_PUBLIC_KEY_2"), help="Secondary GPG public key for rollover" )
parser.add_argument( "--gpg-public-key-url", default=os.environ.get("GPG_PUBLIC_KEY_URL"), help="GPG public key url that\ downloaders can use to verify signing" )
parser.add_argument( "--pkg-upload-url", default=os.environ.get("PKG_URL"), help="URL to be used by downloaders" )
parser.add_argument( "--deb-based-releases", default=["any", "bookworm", "noble", "jammy", "focal", "bionic", "xenial"], help="list of debian based releases that need to be packaged for" )
parser.add_argument( "--archs", default=["amd64", "386", "arm64", "arm", "armhf"], help="list of architectures we want to package for. Note that\ it is the caller's responsiblity to ensure that these debs are already present in a directory. This script\ will not build binaries or create their debs." )
parser.add_argument( "--upload-repo-file", action='store_true', help="Upload RPM repo file to R2" ) args = parser.parse_args()
return args
if __name__ == "__main__": try: args = parse_args() except Exception as e: logging.exception(e) exit(1)
pkg_creator = PkgCreator() # Import one or two keypairs; primary first key_results = pkg_creator.import_multiple_gpg_keys( args.gpg_private_key, args.gpg_public_key, args.gpg_private_key_2, args.gpg_public_key_2, ) if not key_results or len(key_results) == 0: raise SystemExit("No GPG keys were provided for signing") primary_gpg_key_id, primary_gpg_key_name = key_results[0] secondary_gpg_key_id = None secondary_gpg_key_name = None if len(key_results) > 1: secondary_gpg_key_id, secondary_gpg_key_name = key_results[1]
if args.gpg_private_key_2: print(f"signing RPM with secondary gpg_key: {secondary_gpg_key_id}") pkg_creator.import_rpm_key(args.gpg_public_key_2) else: print(f"signing RPM with primary gpg_key: {primary_gpg_key_name}") pkg_creator.import_rpm_key(args.gpg_public_key)
pkg_uploader = PkgUploader(args.account, args.bucket, args.id, args.secret) print(f"signing deb with primary gpg_key: {primary_gpg_key_id} and secondary gpg_key: {secondary_gpg_key_id}") create_deb_packaging( pkg_creator, pkg_uploader, args.deb_based_releases, primary_gpg_key_id, secondary_gpg_key_id, args.binary, args.archs, "main", args.release_tag, )
create_rpm_packaging( pkg_creator, pkg_uploader, "./artifacts", args.release_tag, args.binary, secondary_gpg_key_name, args.pkg_upload_url, args.gpg_public_key_url, args.upload_repo_file, )