mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
121 lines
4.0 KiB
Go
121 lines
4.0 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
// package measurementsuploader is used to upload measurements (v2) JSON files (and signatures) to S3.
|
|
package measurementsuploader
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"net/url"
|
|
"time"
|
|
|
|
s3manager "github.com/aws/aws-sdk-go-v2/feature/s3/manager"
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
s3types "github.com/aws/aws-sdk-go-v2/service/s3/types"
|
|
"github.com/edgelesssys/constellation/v2/internal/api/versionsapi"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/measurements"
|
|
"github.com/edgelesssys/constellation/v2/internal/constants"
|
|
"github.com/edgelesssys/constellation/v2/internal/staticupload"
|
|
)
|
|
|
|
// Uploader uploads image info to S3.
|
|
type Uploader struct {
|
|
uploadClient uploadClient
|
|
uploadClientClose func(ctx context.Context) error
|
|
// bucket is the name of the S3 bucket to use.
|
|
bucket string
|
|
|
|
log *slog.Logger
|
|
}
|
|
|
|
// New creates a new Uploader.
|
|
func New(ctx context.Context, region, bucket, distributionID string, log *slog.Logger) (*Uploader, CloseFunc, error) {
|
|
staticUploadClient, staticUploadClientClose, err := staticupload.New(ctx, staticupload.Config{
|
|
Region: region,
|
|
Bucket: bucket,
|
|
DistributionID: distributionID,
|
|
CacheInvalidationStrategy: staticupload.CacheInvalidateBatchOnFlush,
|
|
CacheInvalidationWaitTimeout: 10 * time.Minute,
|
|
}, log)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
uploader := &Uploader{
|
|
uploadClient: staticUploadClient,
|
|
uploadClientClose: staticUploadClientClose,
|
|
bucket: bucket,
|
|
log: log,
|
|
}
|
|
uploaderClose := func(ctx context.Context) error {
|
|
return uploader.Close(ctx)
|
|
}
|
|
return uploader, uploaderClose, nil
|
|
}
|
|
|
|
// Close closes the uploader.
|
|
// It invalidates the CDN cache for all uploaded files.
|
|
func (a *Uploader) Close(ctx context.Context) error {
|
|
if a.uploadClientClose == nil {
|
|
return nil
|
|
}
|
|
return a.uploadClientClose(ctx)
|
|
}
|
|
|
|
// Upload uploads the measurements v2 JSON file and its signature to S3.
|
|
func (a *Uploader) Upload(ctx context.Context, rawMeasurement, signature io.ReadSeeker) (string, string, error) {
|
|
// parse the measurements to get the ref, stream, and version
|
|
var measurements measurements.ImageMeasurementsV2
|
|
if err := json.NewDecoder(rawMeasurement).Decode(&measurements); err != nil {
|
|
return "", "", err
|
|
}
|
|
if _, err := rawMeasurement.Seek(0, io.SeekStart); err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
ver, err := versionsapi.NewVersion(measurements.Ref, measurements.Stream, measurements.Version, versionsapi.VersionKindImage)
|
|
if err != nil {
|
|
return "", "", fmt.Errorf("creating version: %w", err)
|
|
}
|
|
key, err := url.JoinPath(ver.ArtifactPath(versionsapi.APIV2), ver.Kind().String(), constants.CDNMeasurementsFile)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
sigKey, err := url.JoinPath(ver.ArtifactPath(versionsapi.APIV2), ver.Kind().String(), constants.CDNMeasurementsSignature)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
a.log.Debug(fmt.Sprintf("Archiving image measurements to s3://%v/%v and s3://%v/%v", a.bucket, key, a.bucket, sigKey))
|
|
if _, err = a.uploadClient.Upload(ctx, &s3.PutObjectInput{
|
|
Bucket: &a.bucket,
|
|
Key: &key,
|
|
Body: rawMeasurement,
|
|
ChecksumAlgorithm: s3types.ChecksumAlgorithmSha256,
|
|
}); err != nil {
|
|
return "", "", fmt.Errorf("uploading measurements: %w", err)
|
|
}
|
|
if _, err = a.uploadClient.Upload(ctx, &s3.PutObjectInput{
|
|
Bucket: &a.bucket,
|
|
Key: &sigKey,
|
|
Body: signature,
|
|
ChecksumAlgorithm: s3types.ChecksumAlgorithmSha256,
|
|
}); err != nil {
|
|
return "", "", fmt.Errorf("uploading measurements signature: %w", err)
|
|
}
|
|
return constants.CDNRepositoryURL + "/" + key, constants.CDNRepositoryURL + "/" + sigKey, nil
|
|
}
|
|
|
|
type uploadClient interface {
|
|
Upload(ctx context.Context, input *s3.PutObjectInput, opts ...func(*s3manager.Uploader)) (*s3manager.UploadOutput, error)
|
|
}
|
|
|
|
// CloseFunc is a function that closes the client.
|
|
type CloseFunc func(ctx context.Context) error
|