path_config.go

package artifactory

import (
	"context"
	"crypto/sha256"
	"fmt"
	"time"

	"github.com/hashicorp/vault/sdk/framework"
	"github.com/hashicorp/vault/sdk/logical"
)

const configAdminPath = "config/admin"

func (b *backend) pathConfig() *framework.Path {
	return &framework.Path{
		Pattern: configAdminPath,
		Fields: map[string]*framework.FieldSchema{
			"access_token": {
				Type:        framework.TypeString,
				Description: "Administrator token to access Artifactory",
			},
			"url": {
				Type:        framework.TypeString,
				Required:    true,
				Description: "Address of the Artifactory instance",
			},
			"username_template": {
				Type:        framework.TypeString,
				Description: "Optional. Vault Username Template for dynamically generating usernames.",
			},
			"use_expiring_tokens": {
				Type:        framework.TypeBool,
				Default:     false,
				Description: "Optional. If Artifactory version >= 7.50.3, set expires_in to max_ttl and force_revocable.",
			},
			"force_revocable": {
				Type:        framework.TypeBool,
				Default:     false,
				Description: "Optional. When set to true, we will add the 'force_revocable' flag to the token's extension. In addition, a new configuration has been added that sets the default for setting the 'force_revocable' default when creating a new token - the default of this configuration will be 'false' to ensure that the Circle of Trust remains in place.",
			},
			"bypass_artifactory_tls_verification": {
				Type:        framework.TypeBool,
				Default:     false,
				Description: "Optional. Bypass certification verification for TLS connection with Artifactory. Default to `false`.",
			},
			"allow_scope_override": {
				Type:        framework.TypeBool,
				Default:     false,
				Description: "Optional. Determine if scoped tokens should be allowed. This is an advanced configuration option. Default to `false`.",
			},
			"revoke_on_delete": {
				Type:        framework.TypeBool,
				Default:     false,
				Description: "Optional. Revoke Administrator access token when this configuration is deleted. Default to `false`. Will be set to `true` if token is rotated.",
			},
		},
		Operations: map[logical.Operation]framework.OperationHandler{
			logical.UpdateOperation: &framework.PathOperation{
				Callback: b.pathConfigUpdate,
				Summary:  "Configure the Artifactory secrets backend.",
			},
			logical.DeleteOperation: &framework.PathOperation{
				Callback: b.pathConfigDelete,
				Summary:  "Delete the Artifactory secrets configuration.",
			},
			logical.ReadOperation: &framework.PathOperation{
				Callback: b.pathConfigRead,
				Summary:  "Examine the Artifactory secrets configuration.",
			},
		},
		HelpSynopsis: `Interact with the Artifactory secrets configuration.`,
		HelpDescription: `
Configure the parameters used to connect to the Artifactory server integrated with this backend.

The two main parameters are "url" which is the absolute URL to the Artifactory server. Note that "/artifactory/api"
is prepended by the individual calls, so do not include it in the URL here.

The second is "access_token" which must be an access token powerful enough to generate the other access tokens you'll
be using. This value is stored seal wrapped when available. Once set, the access token cannot be retrieved, but the backend
will send a sha256 hash of the token so you can compare it to your notes. If the token is a JWT Access Token, it will return
additional information such as jfrog_token_id, username and scope.

An optional "username_template" parameter will override the built-in default username_template for dynamically generating
usernames if a static one is not provided.

An optional "bypass_artifactory_tls_verification" parameter will enable bypassing the TLS connection verification with Artifactory.

An optional "allow_scope_override" parameter will enable issuing scoped tokens with Artifactory. This is an advanced option that must
have more sophisticated Vault policies. Please see README for an example.

No renewals or new tokens will be issued if the backend configuration (config/admin) is deleted.
`,
	}
}

type adminConfiguration struct {
	baseConfiguration
	UsernameTemplate                 string `json:"username_template,omitempty"`
	BypassArtifactoryTLSVerification bool   `json:"bypass_artifactory_tls_verification,omitempty"`
	AllowScopeOverride               bool   `json:"allow_scope_override,omitempty"`
	RevokeOnDelete                   bool   `json:"revoke_on_delete,omitempty"`
}

// fetchAdminConfiguration will return nil,nil if there's no configuration
func (b *backend) fetchAdminConfiguration(ctx context.Context, storage logical.Storage) (*adminConfiguration, error) {
	var config adminConfiguration

	// Read in the backend configuration
	entry, err := storage.Get(ctx, configAdminPath)
	if err != nil {
		return nil, err
	}

	if entry == nil {
		return nil, nil
	}

	if err := entry.DecodeJSON(&config); err != nil {
		return nil, err
	}

	return &config, nil
}

func (b *backend) pathConfigUpdate(ctx context.Context, req *logical.Request, data *framework.FieldData) (*logical.Response, error) {
	b.configMutex.Lock()
	defer b.configMutex.Unlock()

	config, err := b.fetchAdminConfiguration(ctx, req.Storage)
	if err != nil {
		return nil, err
	}

	if config == nil {
		config = &adminConfiguration{
			baseConfiguration: baseConfiguration{
				UseNewAccessAPI: true,
			},
		}
	}

	if val, ok := data.GetOk("url"); ok {
		config.ArtifactoryURL = val.(string)
		config.AccessToken = "" // clear access token if URL changes, requires setting access_token and url together for security reasons
	}

	if val, ok := data.GetOk("access_token"); ok {
		config.AccessToken = val.(string)
	}

	if val, ok := data.GetOk("username_template"); ok {
		config.UsernameTemplate = val.(string)
		up, err := testUsernameTemplate(config.UsernameTemplate)
		if err != nil {
			return logical.ErrorResponse("username_template error"), err
		}
		b.usernameProducer = up
	}

	if val, ok := data.GetOk("use_expiring_tokens"); ok {
		config.UseExpiringTokens = val.(bool)
	}

	if val, ok := data.GetOk("force_revocable"); ok {
		temp := val.(bool)
		config.ForceRevocable = &temp
	} else {
		config.ForceRevocable = nil
	}

	if val, ok := data.GetOk("bypass_artifactory_tls_verification"); ok {
		config.BypassArtifactoryTLSVerification = val.(bool)
	}

	if val, ok := data.GetOk("allow_scope_override"); ok {
		config.AllowScopeOverride = val.(bool)
	}

	if val, ok := data.GetOk("revoke_on_delete"); ok {
		config.RevokeOnDelete = val.(bool)
	}

	if config.ArtifactoryURL == "" {
		return logical.ErrorResponse("url is required"), nil
	}

	b.InitializeHttpClient(config)

	go b.sendUsage(config.baseConfiguration, "pathConfigRotateUpdate")

	config.UseNewAccessAPI = b.useNewAccessAPI(config.baseConfiguration)

	entry, err := logical.StorageEntryJSON(configAdminPath, config)
	if err != nil {
		return nil, err
	}

	err = req.Storage.Put(ctx, entry)
	if err != nil {
		return nil, err
	}

	return nil, nil
}

func (b *backend) pathConfigDelete(ctx context.Context, req *logical.Request, _ *framework.FieldData) (*logical.Response, error) {
	b.configMutex.Lock()
	defer b.configMutex.Unlock()

	config, err := b.fetchAdminConfiguration(ctx, req.Storage)
	if err != nil {
		return nil, err
	}

	if config == nil {
		return logical.ErrorResponse("backend not configured"), nil
	}

	go b.sendUsage(config.baseConfiguration, "pathConfigDelete")

	if err := req.Storage.Delete(ctx, configAdminPath); err != nil {
		return nil, err
	}

	if config.RevokeOnDelete {
		b.Logger().Info("config.RevokeOnDelete is 'true'. Attempt to revoke access token.")

		token, err := b.getTokenInfo(config.baseConfiguration, config.AccessToken)
		if err != nil {
			b.Logger().Warn("error parsing existing access token", "err", err)
			return nil, nil
		}

		err = b.RevokeToken(config.baseConfiguration, token.TokenID)
		if err != nil {
			b.Logger().Warn("error revoking existing access token", "tokenId", token.TokenID, "err", err)
			return nil, nil
		}
	}

	return nil, nil
}

func (b *backend) pathConfigRead(ctx context.Context, req *logical.Request, _ *framework.FieldData) (*logical.Response, error) {
	b.configMutex.RLock()
	defer b.configMutex.RUnlock()

	logger := b.Logger().With("func", "pathConfigRead")

	config, err := b.fetchAdminConfiguration(ctx, req.Storage)
	if err != nil {
		return nil, err
	}

	if config == nil {
		return logical.ErrorResponse("backend not configured"), nil
	}

	go b.sendUsage(config.baseConfiguration, "pathConfigRead")

	version, err := b.getVersion(config.baseConfiguration)
	if err != nil {
		logger.Error("failed to get system version", "err", err)
		return nil, err
	}

	// I'm not sure if I should be returning the access token, so I'll hash it.
	accessTokenHash := sha256.Sum256([]byte(config.AccessToken))

	configMap := map[string]interface{}{
		"access_token_sha256":                 fmt.Sprintf("%x", accessTokenHash[:]),
		"url":                                 config.ArtifactoryURL,
		"version":                             version,
		"use_expiring_tokens":                 config.UseExpiringTokens,
		"force_revocable":                     config.ForceRevocable,
		"bypass_artifactory_tls_verification": config.BypassArtifactoryTLSVerification,
		"allow_scope_override":                config.AllowScopeOverride,
		"revoke_on_delete":                    config.RevokeOnDelete,
	}

	// Optionally include username_template
	if len(config.UsernameTemplate) > 0 {
		configMap["username_template"] = config.UsernameTemplate
	}

	// Optionally include token info if it parses properly
	token, err := b.getTokenInfo(config.baseConfiguration, config.AccessToken)
	if err != nil {
		logger.Warn("Error parsing AccessToken", "err", err.Error())
	} else {
		configMap["token_id"] = token.TokenID
		configMap["username"] = token.Username
		configMap["scope"] = token.Scope
		if token.Expires > 0 {
			configMap["exp"] = token.Expires
			tm := time.Unix(token.Expires, 0)
			configMap["expires"] = tm.Local()
		}
	}

	supportForceRevocable, err := b.supportForceRevocable(config.baseConfiguration)
	if err != nil {
		logger.Warn("failed to determine if force_revocable is supported. Set 'supportForceRevocable' to 'false'.", "err", err)
		supportForceRevocable = false
	}

	if supportForceRevocable {
		configMap["use_expiring_tokens"] = config.UseExpiringTokens
	}

	return &logical.Response{
		Data: configMap,
	}, nil
}