Adding Shield Tables

aws/plugin.go

@@ -489,6 +489,13 @@ func Plugin(ctx context.Context) *plugin.Plugin {
 			"aws_sfn_state_machine":                                        tableAwsStepFunctionsStateMachine(ctx),
 			"aws_sfn_state_machine_execution":                              tableAwsStepFunctionsStateMachineExecution(ctx),
 			"aws_sfn_state_machine_execution_history":                      tableAwsStepFunctionsStateMachineExecutionHistory(ctx),
+			"aws_shield_attack":                                            tableAwsShieldAttack(ctx),
+			"aws_shield_attack_statistic":                                  tableAwsShieldAttackStatistic(ctx),
+			"aws_shield_drt_access":                                        tableAwsShieldDRTAccess(ctx),
+			"aws_shield_emergency_contact":                                 tableAwsShieldEmergencyContact(ctx),
+			"aws_shield_protection":                                        tableAwsShieldProtection(ctx),
+			"aws_shield_protection_group":                                  tableAwsShieldProtectionGroup(ctx),
+			"aws_shield_subscription":                                      tableAwsShieldSubscription(ctx),
 			"aws_simspaceweaver_simulation":                                tableAwsSimSpaceWeaverSimulation(ctx),
 			"aws_sns_subscription":                                         tableAwsSnsSubscription(ctx),
 			"aws_sns_topic":                                                tableAwsSnsTopic(ctx),

aws/service.go

@@ -125,6 +125,7 @@ import (
 	"github.com/aws/aws-sdk-go-v2/service/servicequotas"
 	"github.com/aws/aws-sdk-go-v2/service/ses"
 	"github.com/aws/aws-sdk-go-v2/service/sfn"
+	"github.com/aws/aws-sdk-go-v2/service/shield"
 	"github.com/aws/aws-sdk-go-v2/service/simspaceweaver"
 	"github.com/aws/aws-sdk-go-v2/service/sns"
 	"github.com/aws/aws-sdk-go-v2/service/sqs"
@@ -1494,6 +1495,18 @@ func StepFunctionsClient(ctx context.Context, d *plugin.QueryData) (*sfn.Client,
 	return sfn.NewFromConfig(*cfg), nil
 }
 
+func ShieldClient(ctx context.Context, d *plugin.QueryData) (*shield.Client, error) {
+	cfg, err := getClientForDefaultRegion(ctx, d)
+	if err != nil {
+		return nil, err
+	}
+
+	if cfg == nil {
+		return nil, nil
+	}
+	return shield.NewFromConfig(*cfg), nil
+}
+
 func SNSClient(ctx context.Context, d *plugin.QueryData) (*sns.Client, error) {
 	cfg, err := getClientForQueryRegion(ctx, d)
 	if err != nil {

aws/table_aws_shield_attack.go

@@ -0,0 +1,244 @@
+package aws
+
+import (
+	"context"
+	"time"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/shield"
+	"github.com/aws/aws-sdk-go-v2/service/shield/types"
+
+	"github.com/turbot/steampipe-plugin-sdk/v5/grpc/proto"
+	"github.com/turbot/steampipe-plugin-sdk/v5/plugin"
+	"github.com/turbot/steampipe-plugin-sdk/v5/plugin/transform"
+)
+
+//// TABLE DEFINITION
+
+func tableAwsShieldAttack(_ context.Context) *plugin.Table {
+	return &plugin.Table{
+		Name:        "aws_shield_attack",
+		Description: "AWS Shield Attack",
+		Get: &plugin.GetConfig{
+			Hydrate:    getAttack,
+			KeyColumns: plugin.SingleColumn("attack_id"),
+			Tags:       map[string]string{"service": "shield", "action": "DescribeAttack"},
+		},
+		List: &plugin.ListConfig{
+			Hydrate:    listAttacks,
+			KeyColumns: []*plugin.KeyColumn{
+				{
+					Name:    "resource_arn",
+					Require: plugin.Optional,
+					Operators: []string{"="},
+				},
+			},
+			Tags:       map[string]string{"service": "shield", "action": "ListAttacks"},
+		},
+		Columns: awsGlobalRegionColumns([]*plugin.Column{
+			{
+				Name:        "attack_id",
+				Description: "The unique identifier (ID) of the attack.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "resource_arn",
+				Description: "The ARN (Amazon Resource Name) of the Amazon Web Services resource that was attacked.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "start_time",
+				Description: "The start time of the attack.",
+				Type:        proto.ColumnType_TIMESTAMP,
+			},
+			{
+				Name:        "end_time",
+				Description: "The end time of the attack.",
+				Type:        proto.ColumnType_TIMESTAMP,
+			},
+			{
+				Name:        "attack_vectors",
+				Description: "The list of attacks for the time period.",
+				Type:        proto.ColumnType_JSON,
+			},
+			{
+				Name:        "sub_resources",
+				Description: "If applicable, additional detail about the resource being attacked, for example, IP address or URL.",
+				Hydrate:     getAttack,
+				Type:        proto.ColumnType_JSON,
+			},
+			{
+				Name:        "attack_counters",
+				Description: "List of counters that describe the attack for the specified time period.",
+				Hydrate:     getAttack,
+				Type:        proto.ColumnType_JSON,
+			},
+			{
+				Name:        "attack_properties",
+				Description: "The array of objects that provide details of the Shield event.",
+				Hydrate:     getAttack,
+				Type:        proto.ColumnType_JSON,
+			},
+			{
+				Name:        "mitigations",
+				Description: "List of mitigation actions taken for the attack.",
+				Hydrate:     getAttack,
+				Type:        proto.ColumnType_JSON,
+			},
+			// Steampipe standard columns
+			{
+				Name:        "title",
+				Description: resourceInterfaceDescription("title"),
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AttackId"),
+			},
+		}),
+	}
+}
+
+//// LIST FUNCTION
+
+func listAttacks(ctx context.Context, d *plugin.QueryData, _ *plugin.HydrateData) (interface{}, error) {
+	// Create session
+	svc, err := ShieldClient(ctx, d)
+	if err != nil {
+		plugin.Logger(ctx).Error("aws_shield_attack.listAttacks", "connection_error", err)
+		return nil, err
+	}
+
+	if svc == nil {
+		// Unsupported region, return no data
+		return nil, nil
+	}
+
+	// Limiting the results
+	queryResultLimit := int32(1000)
+	if d.QueryContext.Limit != nil {
+		queryResultLimit = min(queryResultLimit, int32(*d.QueryContext.Limit))
+	}
+
+	input := &shield.ListAttacksInput{
+		MaxResults: aws.Int32(queryResultLimit),
+	}
+
+	if d.Quals["resource_arn"] != nil {
+		input.ResourceArns = []string{}
+		for _, q := range d.Quals["resource_arn"].Quals {
+			input.ResourceArns = append(input.ResourceArns, q.Value.GetStringValue())
+		}
+	}
+
+	paginator := shield.NewListAttacksPaginator(svc, input, func(o *shield.ListAttacksPaginatorOptions) {
+		o.Limit = queryResultLimit
+		o.StopOnDuplicateToken = true
+	})
+
+	// List call
+	for paginator.HasMorePages() {
+		// apply rate limiting
+		d.WaitForListRateLimit(ctx)
+
+		output, err := paginator.NextPage(ctx)
+		if err != nil {
+			plugin.Logger(ctx).Error("aws_shield_attack.listAttacks", "api_error", err)
+			return nil, err
+		}
+
+		for _, items := range output.AttackSummaries {
+			d.StreamListItem(ctx, &items)
+			// Context can be cancelled due to manual cancellation or the limit has been hit
+			if d.RowsRemaining(ctx) == 0 {
+				return nil, nil
+			}
+		}
+	}
+
+	return nil, nil
+}
+
+//// HYDRATE FUNCTIONS
+
+func getAttack(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	var attackId string
+	if h.Item != nil {
+		attackId = *h.Item.(*types.AttackSummary).AttackId
+	} else {
+		attackId = d.EqualsQualString("attack_id")
+	}
+
+	if attackId == "" {
+		return nil, nil
+	}
+
+	// Create session
+	svc, err := ShieldClient(ctx, d)
+	if err != nil {
+		plugin.Logger(ctx).Error("aws_shield_attack.getAttack", "connection_error", err)
+		return nil, err
+	}
+
+	if svc == nil {
+		// Unsupported region, return no data
+		return nil, nil
+	}
+
+	data, err := svc.DescribeAttack(ctx, &shield.DescribeAttackInput{
+		AttackId: aws.String(attackId),
+	})
+
+	if err != nil {
+		plugin.Logger(ctx).Error("aws_shield_attack.getAttack", "api_error", err)
+		return nil, err
+	}
+
+	if data.Attack != nil {
+		return &AttackExtended{
+			AttackVectors: getAttackVectors(*data),
+			AttackCounters: data.Attack.AttackCounters,
+			AttackId: data.Attack.AttackId,
+			AttackProperties: data.Attack.AttackProperties,
+			EndTime: data.Attack.EndTime,
+			Mitigations: data.Attack.Mitigations,
+			ResourceArn: data.Attack.ResourceArn,
+			StartTime: data.Attack.StartTime,
+			SubResources: data.Attack.SubResources,
+		}, nil
+	}
+
+	return nil, nil
+}
+
+//// HELPER FUNCTIONS
+
+type AttackExtended struct {
+	AttackVectors []types.AttackVectorDescription
+	AttackCounters []types.SummarizedCounter
+	AttackId *string
+	AttackProperties []types.AttackProperty
+	EndTime *time.Time
+	Mitigations []types.Mitigation
+	ResourceArn *string
+	StartTime *time.Time
+	SubResources []types.SubResourceSummary
+}
+
+func getAttackVectors(attack shield.DescribeAttackOutput) ([]types.AttackVectorDescription) {
+	attackVectors := []types.AttackVectorDescription{}
+
+	if attack.Attack.SubResources == nil {
+		return attackVectors
+	}
+
+	for _, subResource := range attack.Attack.SubResources {
+		if subResource.AttackVectors == nil {
+			continue
+		}
+		for _, attackVector := range subResource.AttackVectors {
+			attackVectors = append(attackVectors, types.AttackVectorDescription{
+				VectorType: attackVector.VectorType,
+			})
+		}
+	}
+
+	return attackVectors
+}