2022-09-05 07:06:08 +00:00
|
|
|
/*
|
|
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
*/
|
|
|
|
|
2022-05-02 11:21:07 +00:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
2022-09-08 12:45:27 +00:00
|
|
|
"context"
|
2022-06-09 14:10:42 +00:00
|
|
|
"fmt"
|
2022-09-08 12:45:27 +00:00
|
|
|
"io"
|
2022-05-02 11:21:07 +00:00
|
|
|
|
2022-06-08 06:26:08 +00:00
|
|
|
"github.com/edgelesssys/constellation/cli/internal/cloudcmd"
|
2022-06-07 09:05:52 +00:00
|
|
|
"github.com/edgelesssys/constellation/cli/internal/proto"
|
2022-09-08 12:45:27 +00:00
|
|
|
"github.com/edgelesssys/constellation/internal/atls"
|
2022-06-07 09:08:44 +00:00
|
|
|
"github.com/edgelesssys/constellation/internal/cloud/cloudprovider"
|
2022-05-02 11:21:07 +00:00
|
|
|
"github.com/edgelesssys/constellation/internal/constants"
|
2022-05-16 15:32:00 +00:00
|
|
|
"github.com/edgelesssys/constellation/internal/file"
|
2022-05-02 11:21:07 +00:00
|
|
|
"github.com/edgelesssys/constellation/internal/state"
|
|
|
|
"github.com/spf13/afero"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
)
|
|
|
|
|
2022-09-08 12:45:27 +00:00
|
|
|
type recoveryClient interface {
|
|
|
|
Connect(endpoint string, validators atls.Validator) error
|
|
|
|
Recover(ctx context.Context, masterSecret, salt []byte) error
|
|
|
|
io.Closer
|
|
|
|
}
|
2022-05-02 11:21:07 +00:00
|
|
|
|
2022-06-08 06:14:28 +00:00
|
|
|
// NewRecoverCmd returns a new cobra.Command for the recover command.
|
|
|
|
func NewRecoverCmd() *cobra.Command {
|
2022-05-02 11:21:07 +00:00
|
|
|
cmd := &cobra.Command{
|
|
|
|
Use: "recover",
|
2022-05-09 15:02:47 +00:00
|
|
|
Short: "Recover a completely stopped Constellation cluster",
|
2022-05-04 07:13:46 +00:00
|
|
|
Long: "Recover a Constellation cluster by sending a recovery key to an instance in the boot stage." +
|
2022-05-02 11:21:07 +00:00
|
|
|
"\nThis is only required if instances restart without other instances available for bootstrapping.",
|
|
|
|
Args: cobra.ExactArgs(0),
|
|
|
|
RunE: runRecover,
|
|
|
|
}
|
2022-05-06 15:51:41 +00:00
|
|
|
cmd.Flags().StringP("endpoint", "e", "", "endpoint of the instance, passed as HOST[:PORT] (required)")
|
2022-05-06 11:56:02 +00:00
|
|
|
must(cmd.MarkFlagRequired("endpoint"))
|
2022-07-29 07:52:47 +00:00
|
|
|
cmd.Flags().String("master-secret", constants.MasterSecretFilename, "path to master secret file")
|
2022-05-02 11:21:07 +00:00
|
|
|
return cmd
|
|
|
|
}
|
|
|
|
|
2022-07-29 07:52:47 +00:00
|
|
|
func runRecover(cmd *cobra.Command, _ []string) error {
|
2022-05-02 11:21:07 +00:00
|
|
|
fileHandler := file.NewHandler(afero.NewOsFs())
|
2022-09-08 12:45:27 +00:00
|
|
|
recoveryClient := &proto.RecoverClient{}
|
2022-05-02 11:21:07 +00:00
|
|
|
defer recoveryClient.Close()
|
2022-06-28 09:19:03 +00:00
|
|
|
return recover(cmd, fileHandler, recoveryClient)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
|
|
|
|
2022-06-28 09:19:03 +00:00
|
|
|
func recover(cmd *cobra.Command, fileHandler file.Handler, recoveryClient recoveryClient) error {
|
2022-07-29 07:52:47 +00:00
|
|
|
flags, err := parseRecoverFlags(cmd)
|
2022-05-02 11:21:07 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-07-29 07:52:47 +00:00
|
|
|
var masterSecret masterSecret
|
|
|
|
if err := fileHandler.ReadJSON(flags.secretPath, &masterSecret); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-05-23 13:01:39 +00:00
|
|
|
var stat state.ConstellationState
|
|
|
|
if err := fileHandler.ReadJSON(constants.StateFilename, &stat); err != nil {
|
2022-05-02 11:21:07 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-05-23 13:01:39 +00:00
|
|
|
provider := cloudprovider.FromString(stat.CloudProvider)
|
2022-09-07 13:38:29 +00:00
|
|
|
config, err := readConfig(cmd.OutOrStdout(), fileHandler, flags.configPath)
|
2022-05-23 13:01:39 +00:00
|
|
|
if err != nil {
|
2022-06-09 14:10:42 +00:00
|
|
|
return fmt.Errorf("reading and validating config: %w", err)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
|
|
|
|
2022-08-09 12:04:40 +00:00
|
|
|
validators, err := cloudcmd.NewValidator(provider, config)
|
2022-05-02 11:21:07 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-08-12 13:59:45 +00:00
|
|
|
if err := recoveryClient.Connect(flags.endpoint, validators.V(cmd)); err != nil {
|
2022-05-02 11:21:07 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-09-08 12:45:27 +00:00
|
|
|
if err := recoveryClient.Recover(cmd.Context(), masterSecret.Key, masterSecret.Salt); err != nil {
|
2022-05-02 11:21:07 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
cmd.Println("Pushed recovery key.")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-07-29 07:52:47 +00:00
|
|
|
func parseRecoverFlags(cmd *cobra.Command) (recoverFlags, error) {
|
2022-05-06 11:56:02 +00:00
|
|
|
endpoint, err := cmd.Flags().GetString("endpoint")
|
2022-05-02 11:21:07 +00:00
|
|
|
if err != nil {
|
2022-06-09 14:10:42 +00:00
|
|
|
return recoverFlags{}, fmt.Errorf("parsing endpoint argument: %w", err)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
2022-07-29 06:24:13 +00:00
|
|
|
endpoint, err = addPortIfMissing(endpoint, constants.BootstrapperPort)
|
2022-05-06 11:56:02 +00:00
|
|
|
if err != nil {
|
2022-06-09 14:10:42 +00:00
|
|
|
return recoverFlags{}, fmt.Errorf("validating endpoint argument: %w", err)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
masterSecretPath, err := cmd.Flags().GetString("master-secret")
|
|
|
|
if err != nil {
|
2022-06-09 14:10:42 +00:00
|
|
|
return recoverFlags{}, fmt.Errorf("parsing master-secret path argument: %w", err)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
|
|
|
|
2022-05-13 09:56:43 +00:00
|
|
|
configPath, err := cmd.Flags().GetString("config")
|
2022-05-02 11:21:07 +00:00
|
|
|
if err != nil {
|
2022-06-09 14:10:42 +00:00
|
|
|
return recoverFlags{}, fmt.Errorf("parsing config path argument: %w", err)
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return recoverFlags{
|
2022-07-29 07:52:47 +00:00
|
|
|
endpoint: endpoint,
|
|
|
|
secretPath: masterSecretPath,
|
|
|
|
configPath: configPath,
|
2022-05-02 11:21:07 +00:00
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type recoverFlags struct {
|
2022-07-29 07:52:47 +00:00
|
|
|
endpoint string
|
|
|
|
secretPath string
|
|
|
|
configPath string
|
2022-05-02 11:21:07 +00:00
|
|
|
}
|