mirror of
				https://github.com/go-gitea/gitea
				synced 2025-10-31 11:28:24 +00:00 
			
		
		
		
	Backport #32881 by ExplodingDragon Incorrect content was introduced while generating the index, which has now been removed, and the missing fields have been added. Co-authored-by: Exploding Dragon <explodingfkl@gmail.com>
		
			
				
	
	
		
			403 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			403 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2024 The Gitea Authors. All rights reserved.
 | |
| // SPDX-License-Identifier: MIT
 | |
| 
 | |
| package arch
 | |
| 
 | |
| import (
 | |
| 	"archive/tar"
 | |
| 	"bytes"
 | |
| 	"compress/gzip"
 | |
| 	"context"
 | |
| 	"encoding/base64"
 | |
| 	"errors"
 | |
| 	"fmt"
 | |
| 	"io"
 | |
| 	"os"
 | |
| 	"strings"
 | |
| 
 | |
| 	packages_model "code.gitea.io/gitea/models/packages"
 | |
| 	arch_model "code.gitea.io/gitea/models/packages/arch"
 | |
| 	user_model "code.gitea.io/gitea/models/user"
 | |
| 	"code.gitea.io/gitea/modules/container"
 | |
| 	"code.gitea.io/gitea/modules/globallock"
 | |
| 	"code.gitea.io/gitea/modules/json"
 | |
| 	packages_module "code.gitea.io/gitea/modules/packages"
 | |
| 	arch_module "code.gitea.io/gitea/modules/packages/arch"
 | |
| 	"code.gitea.io/gitea/modules/util"
 | |
| 	packages_service "code.gitea.io/gitea/services/packages"
 | |
| 
 | |
| 	"github.com/keybase/go-crypto/openpgp"
 | |
| 	"github.com/keybase/go-crypto/openpgp/armor"
 | |
| 	"github.com/keybase/go-crypto/openpgp/packet"
 | |
| )
 | |
| 
 | |
| const (
 | |
| 	IndexArchiveFilename = "packages.db"
 | |
| )
 | |
| 
 | |
| func AquireRegistryLock(ctx context.Context, ownerID int64) (globallock.ReleaseFunc, error) {
 | |
| 	return globallock.Lock(ctx, fmt.Sprintf("packages_arch_%d", ownerID))
 | |
| }
 | |
| 
 | |
| // GetOrCreateRepositoryVersion gets or creates the internal repository package
 | |
| // The Arch registry needs multiple index files which are stored in this package.
 | |
| func GetOrCreateRepositoryVersion(ctx context.Context, ownerID int64) (*packages_model.PackageVersion, error) {
 | |
| 	return packages_service.GetOrCreateInternalPackageVersion(ctx, ownerID, packages_model.TypeArch, arch_module.RepositoryPackage, arch_module.RepositoryVersion)
 | |
| }
 | |
| 
 | |
| // GetOrCreateKeyPair gets or creates the PGP keys used to sign repository files
 | |
| func GetOrCreateKeyPair(ctx context.Context, ownerID int64) (string, string, error) {
 | |
| 	priv, err := user_model.GetSetting(ctx, ownerID, arch_module.SettingKeyPrivate)
 | |
| 	if err != nil && !errors.Is(err, util.ErrNotExist) {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 
 | |
| 	pub, err := user_model.GetSetting(ctx, ownerID, arch_module.SettingKeyPublic)
 | |
| 	if err != nil && !errors.Is(err, util.ErrNotExist) {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 
 | |
| 	if priv == "" || pub == "" {
 | |
| 		priv, pub, err = generateKeypair()
 | |
| 		if err != nil {
 | |
| 			return "", "", err
 | |
| 		}
 | |
| 
 | |
| 		if err := user_model.SetUserSetting(ctx, ownerID, arch_module.SettingKeyPrivate, priv); err != nil {
 | |
| 			return "", "", err
 | |
| 		}
 | |
| 
 | |
| 		if err := user_model.SetUserSetting(ctx, ownerID, arch_module.SettingKeyPublic, pub); err != nil {
 | |
| 			return "", "", err
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return priv, pub, nil
 | |
| }
 | |
| 
 | |
| func generateKeypair() (string, string, error) {
 | |
| 	e, err := openpgp.NewEntity("", "Arch Registry", "", nil)
 | |
| 	if err != nil {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 
 | |
| 	var priv strings.Builder
 | |
| 	var pub strings.Builder
 | |
| 
 | |
| 	w, err := armor.Encode(&priv, openpgp.PrivateKeyType, nil)
 | |
| 	if err != nil {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 	if err := e.SerializePrivate(w, nil); err != nil {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 	w.Close()
 | |
| 
 | |
| 	w, err = armor.Encode(&pub, openpgp.PublicKeyType, nil)
 | |
| 	if err != nil {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 	if err := e.Serialize(w); err != nil {
 | |
| 		return "", "", err
 | |
| 	}
 | |
| 	w.Close()
 | |
| 
 | |
| 	return priv.String(), pub.String(), nil
 | |
| }
 | |
| 
 | |
| func SignData(ctx context.Context, ownerID int64, r io.Reader) ([]byte, error) {
 | |
| 	priv, _, err := GetOrCreateKeyPair(ctx, ownerID)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	block, err := armor.Decode(strings.NewReader(priv))
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	e, err := openpgp.ReadEntity(packet.NewReader(block.Body))
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	buf := &bytes.Buffer{}
 | |
| 	if err := openpgp.DetachSign(buf, e, r, nil); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	return buf.Bytes(), nil
 | |
| }
 | |
| 
 | |
| // BuildAllRepositoryFiles (re)builds all repository files for every available repositories and architectures
 | |
| func BuildAllRepositoryFiles(ctx context.Context, ownerID int64) error {
 | |
| 	pv, err := GetOrCreateRepositoryVersion(ctx, ownerID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	// 1. Delete all existing repository files
 | |
| 	pfs, err := packages_model.GetFilesByVersionID(ctx, pv.ID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	for _, pf := range pfs {
 | |
| 		if err := packages_service.DeletePackageFile(ctx, pf); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// 2. (Re)Build repository files for existing packages
 | |
| 	repositories, err := arch_model.GetRepositories(ctx, ownerID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	for _, repository := range repositories {
 | |
| 		architectures, err := arch_model.GetArchitectures(ctx, ownerID, repository)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		for _, architecture := range architectures {
 | |
| 			if err := buildPackagesIndex(ctx, ownerID, pv, repository, architecture); err != nil {
 | |
| 				return fmt.Errorf("failed to build repository files [%s/%s]: %w", repository, architecture, err)
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // BuildSpecificRepositoryFiles builds index files for the repository
 | |
| func BuildSpecificRepositoryFiles(ctx context.Context, ownerID int64, repository, architecture string) error {
 | |
| 	pv, err := GetOrCreateRepositoryVersion(ctx, ownerID)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	architectures := container.SetOf(architecture)
 | |
| 	if architecture == arch_module.AnyArch {
 | |
| 		// Update all other architectures too when updating the any index
 | |
| 		additionalArchitectures, err := arch_model.GetArchitectures(ctx, ownerID, repository)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		architectures.AddMultiple(additionalArchitectures...)
 | |
| 	}
 | |
| 
 | |
| 	for architecture := range architectures {
 | |
| 		if err := buildPackagesIndex(ctx, ownerID, pv, repository, architecture); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func searchPackageFiles(ctx context.Context, ownerID int64, repository, architecture string) ([]*packages_model.PackageFile, error) {
 | |
| 	pfs, _, err := packages_model.SearchFiles(ctx, &packages_model.PackageFileSearchOptions{
 | |
| 		OwnerID:     ownerID,
 | |
| 		PackageType: packages_model.TypeArch,
 | |
| 		Query:       "%.pkg.tar.%",
 | |
| 		Properties: map[string]string{
 | |
| 			arch_module.PropertyRepository:   repository,
 | |
| 			arch_module.PropertyArchitecture: architecture,
 | |
| 		},
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	return pfs, nil
 | |
| }
 | |
| 
 | |
| func buildPackagesIndex(ctx context.Context, ownerID int64, repoVersion *packages_model.PackageVersion, repository, architecture string) error {
 | |
| 	pfs, err := searchPackageFiles(ctx, ownerID, repository, architecture)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	if architecture != arch_module.AnyArch {
 | |
| 		// Add all any packages too
 | |
| 		anyarchFiles, err := searchPackageFiles(ctx, ownerID, repository, arch_module.AnyArch)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		pfs = append(pfs, anyarchFiles...)
 | |
| 	}
 | |
| 
 | |
| 	// Delete the package indices if there are no packages
 | |
| 	if len(pfs) == 0 {
 | |
| 		pf, err := packages_model.GetFileForVersionByName(ctx, repoVersion.ID, IndexArchiveFilename, fmt.Sprintf("%s|%s", repository, architecture))
 | |
| 		if err != nil && !errors.Is(err, util.ErrNotExist) {
 | |
| 			return err
 | |
| 		} else if pf == nil {
 | |
| 			return nil
 | |
| 		}
 | |
| 
 | |
| 		return packages_service.DeletePackageFile(ctx, pf)
 | |
| 	}
 | |
| 
 | |
| 	indexContent, _ := packages_module.NewHashedBuffer()
 | |
| 	defer indexContent.Close()
 | |
| 
 | |
| 	gw := gzip.NewWriter(indexContent)
 | |
| 	tw := tar.NewWriter(gw)
 | |
| 
 | |
| 	cache := make(map[int64]*packages_model.Package)
 | |
| 
 | |
| 	for _, pf := range pfs {
 | |
| 		opts := &entryOptions{
 | |
| 			File: pf,
 | |
| 		}
 | |
| 
 | |
| 		opts.Version, err = packages_model.GetVersionByID(ctx, pf.VersionID)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		if err := json.Unmarshal([]byte(opts.Version.MetadataJSON), &opts.VersionMetadata); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		opts.Package = cache[opts.Version.PackageID]
 | |
| 		if opts.Package == nil {
 | |
| 			opts.Package, err = packages_model.GetPackageByID(ctx, opts.Version.PackageID)
 | |
| 			if err != nil {
 | |
| 				return err
 | |
| 			}
 | |
| 			cache[opts.Package.ID] = opts.Package
 | |
| 		}
 | |
| 		opts.Blob, err = packages_model.GetBlobByID(ctx, pf.BlobID)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 
 | |
| 		sig, err := packages_model.GetPropertiesByName(ctx, packages_model.PropertyTypeFile, pf.ID, arch_module.PropertySignature)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		if len(sig) == 0 {
 | |
| 			return util.ErrNotExist
 | |
| 		}
 | |
| 		opts.Signature = sig[0].Value
 | |
| 
 | |
| 		meta, err := packages_model.GetPropertiesByName(ctx, packages_model.PropertyTypeFile, pf.ID, arch_module.PropertyMetadata)
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		if len(meta) == 0 {
 | |
| 			return util.ErrNotExist
 | |
| 		}
 | |
| 		if err := json.Unmarshal([]byte(meta[0].Value), &opts.FileMetadata); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 
 | |
| 		if err := writeFiles(tw, opts); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 		if err := writeDescription(tw, opts); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	tw.Close()
 | |
| 	gw.Close()
 | |
| 
 | |
| 	signature, err := SignData(ctx, ownerID, indexContent)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	if _, err := indexContent.Seek(0, io.SeekStart); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	_, err = packages_service.AddFileToPackageVersionInternal(
 | |
| 		ctx,
 | |
| 		repoVersion,
 | |
| 		&packages_service.PackageFileCreationInfo{
 | |
| 			PackageFileInfo: packages_service.PackageFileInfo{
 | |
| 				Filename:     IndexArchiveFilename,
 | |
| 				CompositeKey: fmt.Sprintf("%s|%s", repository, architecture),
 | |
| 			},
 | |
| 			Creator:           user_model.NewGhostUser(),
 | |
| 			Data:              indexContent,
 | |
| 			IsLead:            false,
 | |
| 			OverwriteExisting: true,
 | |
| 			Properties: map[string]string{
 | |
| 				arch_module.PropertyRepository:   repository,
 | |
| 				arch_module.PropertyArchitecture: architecture,
 | |
| 				arch_module.PropertySignature:    base64.StdEncoding.EncodeToString(signature),
 | |
| 			},
 | |
| 		},
 | |
| 	)
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| type entryOptions struct {
 | |
| 	Package         *packages_model.Package
 | |
| 	Version         *packages_model.PackageVersion
 | |
| 	VersionMetadata *arch_module.VersionMetadata
 | |
| 	File            *packages_model.PackageFile
 | |
| 	FileMetadata    *arch_module.FileMetadata
 | |
| 	Blob            *packages_model.PackageBlob
 | |
| 	Signature       string
 | |
| }
 | |
| 
 | |
| type keyValue struct {
 | |
| 	Key   string
 | |
| 	Value string
 | |
| }
 | |
| 
 | |
| func writeFiles(tw *tar.Writer, opts *entryOptions) error {
 | |
| 	return writeFields(tw, fmt.Sprintf("%s-%s/files", opts.Package.Name, opts.Version.Version), []keyValue{
 | |
| 		{"FILES", strings.Join(opts.FileMetadata.Files, "\n")},
 | |
| 	})
 | |
| }
 | |
| 
 | |
| // https://gitlab.archlinux.org/pacman/pacman/-/blob/master/lib/libalpm/be_sync.c#L562
 | |
| func writeDescription(tw *tar.Writer, opts *entryOptions) error {
 | |
| 	return writeFields(tw, fmt.Sprintf("%s-%s/desc", opts.Package.Name, opts.Version.Version), []keyValue{
 | |
| 		{"FILENAME", opts.File.Name},
 | |
| 		{"MD5SUM", opts.Blob.HashMD5},
 | |
| 		{"SHA256SUM", opts.Blob.HashSHA256},
 | |
| 		{"PGPSIG", opts.Signature},
 | |
| 		{"CSIZE", fmt.Sprintf("%d", opts.Blob.Size)},
 | |
| 		{"ISIZE", fmt.Sprintf("%d", opts.FileMetadata.InstalledSize)},
 | |
| 		{"NAME", opts.Package.Name},
 | |
| 		{"BASE", opts.FileMetadata.Base},
 | |
| 		{"ARCH", opts.FileMetadata.Architecture},
 | |
| 		{"VERSION", opts.Version.Version},
 | |
| 		{"DESC", opts.VersionMetadata.Description},
 | |
| 		{"URL", opts.VersionMetadata.ProjectURL},
 | |
| 		{"LICENSE", strings.Join(opts.VersionMetadata.Licenses, "\n")},
 | |
| 		{"GROUPS", strings.Join(opts.FileMetadata.Groups, "\n")},
 | |
| 		{"BUILDDATE", fmt.Sprintf("%d", opts.FileMetadata.BuildDate)},
 | |
| 		{"PACKAGER", opts.FileMetadata.Packager},
 | |
| 		{"PROVIDES", strings.Join(opts.FileMetadata.Provides, "\n")},
 | |
| 		{"REPLACES", strings.Join(opts.FileMetadata.Replaces, "\n")},
 | |
| 		{"CONFLICTS", strings.Join(opts.FileMetadata.Conflicts, "\n")},
 | |
| 		{"DEPENDS", strings.Join(opts.FileMetadata.Depends, "\n")},
 | |
| 		{"OPTDEPENDS", strings.Join(opts.FileMetadata.OptDepends, "\n")},
 | |
| 		{"MAKEDEPENDS", strings.Join(opts.FileMetadata.MakeDepends, "\n")},
 | |
| 		{"CHECKDEPENDS", strings.Join(opts.FileMetadata.CheckDepends, "\n")},
 | |
| 	})
 | |
| }
 | |
| 
 | |
| func writeFields(tw *tar.Writer, filename string, fields []keyValue) error {
 | |
| 	buf := &bytes.Buffer{}
 | |
| 	for _, kv := range fields {
 | |
| 		if kv.Value == "" {
 | |
| 			continue
 | |
| 		}
 | |
| 		fmt.Fprintf(buf, "%%%s%%\n%s\n\n", kv.Key, kv.Value)
 | |
| 	}
 | |
| 
 | |
| 	if err := tw.WriteHeader(&tar.Header{
 | |
| 		Name: filename,
 | |
| 		Size: int64(buf.Len()),
 | |
| 		Mode: int64(os.ModePerm),
 | |
| 	}); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	_, err := io.Copy(tw, buf)
 | |
| 	return err
 | |
| }
 |