resource_aws_auth_backend_role.go

package vault

import (
	"encoding/json"
	"fmt"
	"log"
	"regexp"
	"strings"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/hashicorp/vault/api"
)

var (
	awsAuthBackendRoleBackendFromPathRegex = regexp.MustCompile("^auth/(.+)/role/.+$")
	awsAuthBackendRoleNameFromPathRegex    = regexp.MustCompile("^auth/.+/role/(.+)$")
)

func awsAuthBackendRoleResource() *schema.Resource {
	return &schema.Resource{
		Create: awsAuthBackendRoleCreate,
		Read:   awsAuthBackendRoleRead,
		Update: awsAuthBackendRoleUpdate,
		Delete: awsAuthBackendRoleDelete,
		Exists: awsAuthBackendRoleExists,
		Importer: &schema.ResourceImporter{
			State: schema.ImportStatePassthrough,
		},

		Schema: map[string]*schema.Schema{
			"role": {
				Type:        schema.TypeString,
				Required:    true,
				Description: "Name of the role.",
				ForceNew:    true,
			},
			"auth_type": {
				Type:        schema.TypeString,
				Optional:    true,
				Default:     "iam",
				Description: "The auth type permitted for this role.",
				ForceNew:    true,
			},
			"bound_ami_id": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances using this AMI ID will be permitted to log in.",
			},
			"bound_account_id": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances with this account ID in their identity document will be permitted to log in.",
			},
			"bound_region": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances in this region will be permitted to log in.",
			},
			"bound_vpc_id": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances associated with this VPC ID will be permitted to log in.",
			},
			"bound_subnet_id": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances associated with this subnet ID will be permitted to log in.",
			},
			"bound_iam_role_arn": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances that match this IAM role ARN will be permitted to log in.",
			},
			"bound_iam_instance_profile_arn": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Only EC2 instances associated with an IAM instance profile ARN that matches this value will be permitted to log in.",
			},
			"role_tag": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The key of the tag on EC2 instance to use for role tags.",
			},
			"bound_iam_principal_arn": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The IAM principal that must be authenticated using the iam auth method.",
			},
			"inferred_entity_type": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The type of inferencing Vault should do.",
			},
			"inferred_aws_region": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The region to search for the inferred entities in.",
			},
			"resolve_aws_unique_ids": {
				Type:        schema.TypeBool,
				Optional:    true,
				Description: "Whether or not Vault should resolve the bound_iam_principal_arn to an AWS Unique ID. When true, deleting a principal and recreating it with the same name won't automatically grant the new principal the same roles in Vault that the old principal had.",
				Computed:    true,
			},
			"ttl": {
				Type:        schema.TypeInt,
				Optional:    true,
				Description: "The TTL period of tokens issued using this role, provided as the number of seconds.",
			},
			"max_ttl": {
				Type:        schema.TypeInt,
				Optional:    true,
				Description: "The maximum allowed lifetime of tokens issued using this role, provided as the number of seconds.",
			},
			"period": {
				Type:        schema.TypeInt,
				Optional:    true,
				Description: "If set, indicates that the token generated using this role should never expire. The token should be renewed within the duration specified by this value. At each renewal, the token's TTL will be set to the value of this field. The maximum allowed lifetime of token issued using this role. Specified as a number of seconds.",
			},
			"policies": {
				Type:     schema.TypeList,
				Optional: true,
				Elem: &schema.Schema{
					Type: schema.TypeString,
				},
				Description: "Policies to be set on tokens issued using this role.",
			},
			"allow_instance_migration": {
				Type:        schema.TypeBool,
				Optional:    true,
				Description: "When true, allows migration of the underlying instance where the client resides. Use with caution.",
				Computed:    true,
			},
			"disallow_reauthentication": {
				Type:        schema.TypeBool,
				Optional:    true,
				Description: "When true, only allows a single token to be granted per instance ID.",
			},
			"backend": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Unique name of the auth backend to configure.",
				ForceNew:    true,
				Default:     "aws",
				// standardise on no beginning or trailing slashes
				StateFunc: func(v interface{}) string {
					return strings.Trim(v.(string), "/")
				},
			},
		},
	}
}

func awsAuthBackendRoleCreate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*api.Client)

	backend := d.Get("backend").(string)
	role := d.Get("role").(string)

	path := awsAuthBackendRolePath(backend, role)

	log.Printf("[DEBUG] Writing AWS auth backend role %q", path)
	iPolicies := d.Get("policies").([]interface{})
	policies := make([]string, 0, len(iPolicies))
	for _, iPolicy := range iPolicies {
		policies = append(policies, iPolicy.(string))
	}

	authType := d.Get("auth_type").(string)
	inferred := d.Get("inferred_entity_type").(string)

	data := map[string]interface{}{
		"auth_type": authType,
	}
	if v, ok := d.GetOk("ttl"); ok {
		data["ttl"] = v.(int)
	}
	if v, ok := d.GetOk("max_ttl"); ok {
		data["max_ttl"] = v.(int)
	}
	if v, ok := d.GetOk("period"); ok {
		data["period"] = v.(int)
	}
	if len(policies) > 0 {
		data["policies"] = policies
	}
	if authType == "ec2" || (inferred == "ec2_instance" && authType == "iam") {
		if v, ok := d.GetOk("bound_ami_id"); ok {
			data["bound_ami_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_account_id"); ok {
			data["bound_account_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_region"); ok {
			data["bound_region"] = v.(string)
		}
		if v, ok := d.GetOk("bound_vpc_id"); ok {
			data["bound_vpc_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_subnet_id"); ok {
			data["bound_subnet_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_iam_role_arn"); ok {
			data["bound_iam_role_arn"] = v.(string)
		}
		if v, ok := d.GetOk("bound_iam_instance_profile_arn"); ok {
			data["bound_iam_instance_profile_arn"] = v.(string)
		}
	}
	if authType == "ec2" {
		if v, ok := d.GetOk("role_tag"); ok {
			data["role_tag"] = v.(string)
		}
		if v, ok := d.GetOk("allow_instance_migration"); ok {
			data["allow_instance_migration"] = v.(bool)
		}
		if v, ok := d.GetOk("disallow_reauthentication"); ok {
			data["disallow_reauthentication"] = v.(bool)
		}
	}
	if authType == "iam" {
		if inferred != "" {
			data["inferred_entity_type"] = inferred
		}
		if v, ok := d.GetOk("bound_iam_principal_arn"); ok {
			data["bound_iam_principal_arn"] = v.(string)
		}
		if v, ok := d.GetOk("inferred_aws_region"); ok {
			data["inferred_aws_region"] = v.(string)
		}
		if v, ok := d.GetOk("resolve_aws_unique_ids"); ok {
			data["resolve_aws_unique_ids"] = v.(bool)
		}
	}
	_, err := client.Logical().Write(path, data)

	d.SetId(path)

	if err != nil {
		d.SetId("")
		return fmt.Errorf("error writing AWS auth backend role %q: %s", path, err)
	}
	log.Printf("[DEBUG] Wrote AWS auth backend role %q", path)

	return awsAuthBackendRoleRead(d, meta)
}

func awsAuthBackendRoleRead(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*api.Client)
	path := d.Id()

	backend, err := awsAuthBackendRoleBackendFromPath(path)
	if err != nil {
		return fmt.Errorf("invalid path %q for AWS auth backend role: %s", path, err)
	}

	role, err := awsAuthBackendRoleNameFromPath(path)
	if err != nil {
		return fmt.Errorf("invalid path %q for AWS auth backend role: %s", path, err)
	}

	log.Printf("[DEBUG] Reading AWS auth backend role %q", path)
	resp, err := client.Logical().Read(path)
	if err != nil {
		return fmt.Errorf("error reading AWS auth backend role %q: %s", path, err)
	}
	log.Printf("[DEBUG] Read AWS auth backend role %q", path)
	if resp == nil {
		log.Printf("[WARN] AWS auth backend role %q not found, removing from state", path)
		d.SetId("")
		return nil
	}
	iPolicies := resp.Data["policies"].([]interface{})
	policies := make([]string, 0, len(iPolicies))
	for _, iPolicy := range iPolicies {
		policies = append(policies, iPolicy.(string))
	}

	ttl, err := resp.Data["ttl"].(json.Number).Int64()
	if err != nil {
		return fmt.Errorf("expected ttl %q to be a number, isn't", resp.Data["ttl"])
	}

	maxTTL, err := resp.Data["max_ttl"].(json.Number).Int64()
	if err != nil {
		return fmt.Errorf("expected max_ttl %q to be a number, isn't", resp.Data["max_ttl"])
	}

	period, err := resp.Data["period"].(json.Number).Int64()
	if err != nil {
		return fmt.Errorf("expected period %q to be a number, isn't", resp.Data["period"])
	}

	d.Set("backend", backend)
	d.Set("role", role)
	d.Set("auth_type", resp.Data["auth_type"])
	d.Set("bound_ami_id", resp.Data["bound_ami_id"])
	d.Set("bound_account_id", resp.Data["bound_account_id"])
	d.Set("bound_region", resp.Data["bound_region"])
	d.Set("bound_vpc_id", resp.Data["bound_vpc_id"])
	d.Set("bound_subnet_id", resp.Data["bound_subnet_id"])
	d.Set("bound_iam_role_arn", resp.Data["bound_iam_role_arn"])
	d.Set("bound_iam_instance_profile_arn", resp.Data["bound_iam_instance_profile_arn"])
	d.Set("role_tag", resp.Data["role_tag"])
	d.Set("bound_iam_principal_arn", resp.Data["bound_iam_principal_arn"])
	d.Set("inferred_entity_type", resp.Data["inferred_entity_type"])
	d.Set("inferred_aws_region", resp.Data["inferred_aws_region"])
	d.Set("resolve_aws_unique_ids", resp.Data["resolve_aws_unique_ids"])
	d.Set("ttl", ttl)
	d.Set("max_ttl", maxTTL)
	d.Set("period", period)
	d.Set("policies", policies)
	d.Set("allow_instance_migration", resp.Data["allow_instance_migration"])
	d.Set("disallow_reauthentication", resp.Data["disallow_reauthentication"])

	return nil
}

func awsAuthBackendRoleUpdate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*api.Client)
	path := d.Id()

	log.Printf("[DEBUG] Updating AWS auth backend role %q", path)
	iPolicies := d.Get("policies").([]interface{})
	policies := make([]string, 0, len(iPolicies))
	for _, iPolicy := range iPolicies {
		policies = append(policies, iPolicy.(string))
	}

	authType := d.Get("auth_type").(string)
	inferred := d.Get("inferred_entity_type").(string)

	data := map[string]interface{}{}
	if v, ok := d.GetOk("ttl"); ok {
		data["ttl"] = v.(int)
	}
	if v, ok := d.GetOk("max_ttl"); ok {
		data["max_ttl"] = v.(int)
	}
	if v, ok := d.GetOk("period"); ok {
		data["period"] = v.(int)
	}
	if len(policies) > 0 {
		data["policies"] = policies
	}
	if authType == "ec2" || (inferred == "ec2_instance" && authType == "iam") {
		if v, ok := d.GetOk("bound_ami_id"); ok {
			data["bound_ami_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_account_id"); ok {
			data["bound_account_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_region"); ok {
			data["bound_region"] = v.(string)
		}
		if v, ok := d.GetOk("bound_vpc_id"); ok {
			data["bound_vpc_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_subnet_id"); ok {
			data["bound_subnet_id"] = v.(string)
		}
		if v, ok := d.GetOk("bound_iam_role_arn"); ok {
			data["bound_iam_role_arn"] = v.(string)
		}
		if v, ok := d.GetOk("bound_iam_instance_profile_arn"); ok {
			data["bound_iam_instance_profile_arn"] = v.(string)
		}
	}
	if authType == "ec2" {
		if v, ok := d.GetOk("role_tag"); ok {
			data["role_tag"] = v.(string)
		}
		if v, ok := d.GetOk("allow_instance_migration"); ok {
			data["allow_instance_migration"] = v.(bool)
		}
		if v, ok := d.GetOk("disallow_reauthentication"); ok {
			data["disallow_reauthentication"] = v.(bool)
		}
	}
	if authType == "iam" {
		if inferred != "" {
			data["inferred_entity_type"] = inferred
		}
		if v, ok := d.GetOk("bound_iam_principal_arn"); ok {
			data["bound_iam_principal_arn"] = v.(string)
		}
		if v, ok := d.GetOk("inferred_aws_region"); ok {
			data["inferred_aws_region"] = v.(string)
		}
		if v, ok := d.GetOk("resolve_aws_unique_ids"); ok {
			data["resolve_aws_unique_ids"] = v.(bool)
		}
	}

	_, err := client.Logical().Write(path, data)
	if err != nil {
		return fmt.Errorf("error updating AWS auth backend role %q: %s", path, err)
	}
	log.Printf("[DEBUG] Updated AWS auth backend role %q", path)

	return awsAuthBackendRoleRead(d, meta)
}

func awsAuthBackendRoleDelete(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*api.Client)
	path := d.Id()

	log.Printf("[DEBUG] Deleting AWS auth backend role %q", path)
	_, err := client.Logical().Delete(path)
	if err != nil {
		return fmt.Errorf("error deleting AWS auth backend role %q", path)
	}
	log.Printf("[DEBUG] Deleted AWS auth backend role %q", path)

	return nil
}

func awsAuthBackendRoleExists(d *schema.ResourceData, meta interface{}) (bool, error) {
	client := meta.(*api.Client)

	path := d.Id()
	log.Printf("[DEBUG] Checking if AWS auth backend role %q exists", path)

	resp, err := client.Logical().Read(path)
	if err != nil {
		return true, fmt.Errorf("error checking if AWS auth backend role %q exists: %s", path, err)
	}
	log.Printf("[DEBUG] Checked if AWS auth backend role %q exists", path)

	return resp != nil, nil
}

func awsAuthBackendRolePath(backend, role string) string {
	return "auth/" + strings.Trim(backend, "/") + "/role/" + strings.Trim(role, "/")
}

func awsAuthBackendRoleNameFromPath(path string) (string, error) {
	if !awsAuthBackendRoleNameFromPathRegex.MatchString(path) {
		return "", fmt.Errorf("no role found")
	}
	res := awsAuthBackendRoleNameFromPathRegex.FindStringSubmatch(path)
	if len(res) != 2 {
		return "", fmt.Errorf("unexpected number of matches (%d) for role", len(res))
	}
	return res[1], nil
}

func awsAuthBackendRoleBackendFromPath(path string) (string, error) {
	if !awsAuthBackendRoleBackendFromPathRegex.MatchString(path) {
		return "", fmt.Errorf("no backend found")
	}
	res := awsAuthBackendRoleBackendFromPathRegex.FindStringSubmatch(path)
	if len(res) != 2 {
		return "", fmt.Errorf("unexpected number of matches (%d) for backend", len(res))
	}
	return res[1], nil
}