mirror of
				https://github.com/optim-enterprises-bv/vault.git
				synced 2025-11-03 20:17:59 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			136 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			136 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package aws
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
 | 
						|
	"github.com/aws/aws-sdk-go/aws"
 | 
						|
	"github.com/hashicorp/vault/sdk/framework"
 | 
						|
	"github.com/hashicorp/vault/sdk/logical"
 | 
						|
)
 | 
						|
 | 
						|
func pathConfigRoot(b *backend) *framework.Path {
 | 
						|
	return &framework.Path{
 | 
						|
		Pattern: "config/root",
 | 
						|
		Fields: map[string]*framework.FieldSchema{
 | 
						|
			"access_key": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeString,
 | 
						|
				Description: "Access key with permission to create new keys.",
 | 
						|
			},
 | 
						|
 | 
						|
			"secret_key": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeString,
 | 
						|
				Description: "Secret key with permission to create new keys.",
 | 
						|
			},
 | 
						|
 | 
						|
			"region": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeString,
 | 
						|
				Description: "Region for API calls.",
 | 
						|
			},
 | 
						|
			"iam_endpoint": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeString,
 | 
						|
				Description: "Endpoint to custom IAM server URL",
 | 
						|
			},
 | 
						|
			"sts_endpoint": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeString,
 | 
						|
				Description: "Endpoint to custom STS server URL",
 | 
						|
			},
 | 
						|
			"max_retries": &framework.FieldSchema{
 | 
						|
				Type:        framework.TypeInt,
 | 
						|
				Default:     aws.UseServiceDefaultRetries,
 | 
						|
				Description: "Maximum number of retries for recoverable exceptions of AWS APIs",
 | 
						|
			},
 | 
						|
		},
 | 
						|
 | 
						|
		Callbacks: map[logical.Operation]framework.OperationFunc{
 | 
						|
			logical.ReadOperation:   b.pathConfigRootRead,
 | 
						|
			logical.UpdateOperation: b.pathConfigRootWrite,
 | 
						|
		},
 | 
						|
 | 
						|
		HelpSynopsis:    pathConfigRootHelpSyn,
 | 
						|
		HelpDescription: pathConfigRootHelpDesc,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func (b *backend) pathConfigRootRead(ctx context.Context, req *logical.Request, data *framework.FieldData) (*logical.Response, error) {
 | 
						|
	b.clientMutex.RLock()
 | 
						|
	defer b.clientMutex.RUnlock()
 | 
						|
 | 
						|
	entry, err := req.Storage.Get(ctx, "config/root")
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if entry == nil {
 | 
						|
		return nil, nil
 | 
						|
	}
 | 
						|
 | 
						|
	var config rootConfig
 | 
						|
 | 
						|
	if err := entry.DecodeJSON(&config); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	configData := map[string]interface{}{
 | 
						|
		"access_key":   config.AccessKey,
 | 
						|
		"region":       config.Region,
 | 
						|
		"iam_endpoint": config.IAMEndpoint,
 | 
						|
		"sts_endpoint": config.STSEndpoint,
 | 
						|
		"max_retries":  config.MaxRetries,
 | 
						|
	}
 | 
						|
	return &logical.Response{
 | 
						|
		Data: configData,
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
func (b *backend) pathConfigRootWrite(ctx context.Context, req *logical.Request, data *framework.FieldData) (*logical.Response, error) {
 | 
						|
	region := data.Get("region").(string)
 | 
						|
	iamendpoint := data.Get("iam_endpoint").(string)
 | 
						|
	stsendpoint := data.Get("sts_endpoint").(string)
 | 
						|
	maxretries := data.Get("max_retries").(int)
 | 
						|
 | 
						|
	b.clientMutex.Lock()
 | 
						|
	defer b.clientMutex.Unlock()
 | 
						|
 | 
						|
	entry, err := logical.StorageEntryJSON("config/root", rootConfig{
 | 
						|
		AccessKey:   data.Get("access_key").(string),
 | 
						|
		SecretKey:   data.Get("secret_key").(string),
 | 
						|
		IAMEndpoint: iamendpoint,
 | 
						|
		STSEndpoint: stsendpoint,
 | 
						|
		Region:      region,
 | 
						|
		MaxRetries:  maxretries,
 | 
						|
	})
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	if err := req.Storage.Put(ctx, entry); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	// clear possible cached IAM / STS clients after successfully updating
 | 
						|
	// config/root
 | 
						|
	b.iamClient = nil
 | 
						|
	b.stsClient = nil
 | 
						|
 | 
						|
	return nil, nil
 | 
						|
}
 | 
						|
 | 
						|
type rootConfig struct {
 | 
						|
	AccessKey   string `json:"access_key"`
 | 
						|
	SecretKey   string `json:"secret_key"`
 | 
						|
	IAMEndpoint string `json:"iam_endpoint"`
 | 
						|
	STSEndpoint string `json:"sts_endpoint"`
 | 
						|
	Region      string `json:"region"`
 | 
						|
	MaxRetries  int    `json:"max_retries"`
 | 
						|
}
 | 
						|
 | 
						|
const pathConfigRootHelpSyn = `
 | 
						|
Configure the root credentials that are used to manage IAM.
 | 
						|
`
 | 
						|
 | 
						|
const pathConfigRootHelpDesc = `
 | 
						|
Before doing anything, the AWS backend needs credentials that are able
 | 
						|
to manage IAM policies, users, access keys, etc. This endpoint is used
 | 
						|
to configure those credentials. They don't necessarily need to be root
 | 
						|
keys as long as they have permission to manage IAM.
 | 
						|
`
 |