mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-30 09:56:21 -05:00
86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/measurements"
|
|
"github.com/edgelesssys/constellation/v2/internal/logger"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// newMeasurementsMergeCmd creates a new merge command.
|
|
func newMeasurementsMergeCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "merge [flags] <measurements.json>...",
|
|
Short: "Merge OS image measurements",
|
|
Long: "Merge OS image measurements.",
|
|
Args: cobra.MinimumNArgs(1),
|
|
RunE: runMergeMeasurements,
|
|
}
|
|
|
|
cmd.SetOut(os.Stdout)
|
|
cmd.Flags().String("out", "", "Optional path to write the merge result to. If not set, the result is written to stdout.")
|
|
cmd.Flags().Bool("verbose", false, "Enable verbose output")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runMergeMeasurements(cmd *cobra.Command, args []string) error {
|
|
workdir := os.Getenv("BUILD_WORKING_DIRECTORY")
|
|
if len(workdir) > 0 {
|
|
must(os.Chdir(workdir))
|
|
}
|
|
|
|
flags, err := parseMergeMeasurementsFlags(cmd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log := logger.New(logger.PlainLog, flags.logLevel)
|
|
log.Debugf("Parsed flags: %+v", flags)
|
|
|
|
mergedMeasurements, err := readMeasurementsArgs(args)
|
|
if err != nil {
|
|
return fmt.Errorf("merging measurements: reading input files: %w", err)
|
|
}
|
|
|
|
out := cmd.OutOrStdout()
|
|
if len(flags.out) > 0 {
|
|
outF, err := os.Create(flags.out)
|
|
if err != nil {
|
|
return fmt.Errorf("merging measurements: opening output file: %w", err)
|
|
}
|
|
defer outF.Close()
|
|
out = outF
|
|
}
|
|
|
|
if err := json.NewEncoder(out).Encode(mergedMeasurements); err != nil {
|
|
return fmt.Errorf("merging measurements: writing output file: %w", err)
|
|
}
|
|
log.Infof("Merged image measurements")
|
|
return nil
|
|
}
|
|
|
|
func readMeasurementsArgs(paths []string) (measurements.ImageMeasurementsV2, error) {
|
|
measuremnts := make([]measurements.ImageMeasurementsV2, len(paths))
|
|
for i, path := range paths {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return measurements.ImageMeasurementsV2{}, err
|
|
}
|
|
defer f.Close()
|
|
if err := json.NewDecoder(f).Decode(&measuremnts[i]); err != nil {
|
|
return measurements.ImageMeasurementsV2{}, err
|
|
}
|
|
}
|
|
return measurements.MergeImageMeasurementsV2(measuremnts...)
|
|
}
|